Metadata-Version: 2.4
Name: peagen
Version: 0.2.1.dev10
Summary: Swarmauri's Peagan - An AI-driven contextual, dependency-based scaffolding tool for rapid content generation.
License-Expression: Apache-2.0
License-File: LICENSE
Keywords: sdk,standards,peagen
Author: Jacob Stewart
Author-email: jacob@swarmauri.com
Requires-Python: >=3.10,<3.13
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Development Status :: 3 - Alpha
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3 :: Only
Requires-Dist: GitPython
Requires-Dist: PyGithub
Requires-Dist: aiosqlite (>=0.19.0)
Requires-Dist: alembic
Requires-Dist: anyio
Requires-Dist: asyncpg (>=0.30.0)
Requires-Dist: colorama
Requires-Dist: fastapi (>=0.111)
Requires-Dist: filelock
Requires-Dist: h11 (>=0.16.0)
Requires-Dist: httpx[http2] (>=0.27)
Requires-Dist: inflect
Requires-Dist: jinja2 (>=3.1.6)
Requires-Dist: jsonpatch (>=1.33)
Requires-Dist: jsonschema (>=4.18.5)
Requires-Dist: minio
Requires-Dist: pgpy (>=0.6.0)
Requires-Dist: pika
Requires-Dist: psutil (>=6.0)
Requires-Dist: psycopg2-binary
Requires-Dist: pydantic-settings (>=2.2)
Requires-Dist: pydantic[email] (>=2.7)
Requires-Dist: pytest-monitor (>=1.6.6)
Requires-Dist: python-dotenv
Requires-Dist: python-gnupg
Requires-Dist: pyyaml
Requires-Dist: redis
Requires-Dist: redis[hiredis] (>=5.0)
Requires-Dist: rich
Requires-Dist: s3fs
Requires-Dist: sqlalchemy (>=2.0)
Requires-Dist: swarmauri
Requires-Dist: swarmauri_base
Requires-Dist: swarmauri_core
Requires-Dist: swarmauri_crypto_paramiko
Requires-Dist: swarmauri_crypto_pgp
Requires-Dist: swarmauri_gitfilter_file
Requires-Dist: swarmauri_keyprovider_ssh
Requires-Dist: swarmauri_prompt_j2prompttemplate
Requires-Dist: swarmauri_standard
Requires-Dist: swarmauri_storage_file
Requires-Dist: swarmauri_storage_file
Requires-Dist: textual
Requires-Dist: tigrbl
Requires-Dist: tigrbl_auth
Requires-Dist: tigrbl_client
Requires-Dist: typer
Requires-Dist: uvicorn[standard] (>=0.29)
Requires-Dist: websockets (>=12)
Description-Content-Type: text/markdown


![Swarmauri Logo](https://github.com/swarmauri/swarmauri-sdk/blob/3d4d1cfa949399d7019ae9d8f296afba773dfb7f/assets/swarmauri.brand.theme.svg)

<p align="center">
    <a href="https://pypi.org/project/peagen/">
        <img src="https://img.shields.io/pypi/dm/peagen" alt="PyPI - Downloads"/></a>
    <a href="https://hits.sh/github.com/swarmauri/swarmauri-sdk/tree/master/pkgs/standards/peagen/">
        <img alt="Hits" src="https://hits.sh/github.com/swarmauri/swarmauri-sdk/tree/master/pkgs/standards/peagen.svg"/></a>
    <a href="https://pypi.org/project/peagen/">
        <img src="https://img.shields.io/pypi/pyversions/peagen" alt="PyPI - Python Version"/></a>
    <a href="https://pypi.org/project/peagen/">
        <img src="https://img.shields.io/pypi/l/peagen" alt="PyPI - License"/></a>
    <a href="https://pypi.org/project/peagen/">
        <img src="https://img.shields.io/pypi/v/peagen?label=peagen&color=green" alt="PyPI - peagen"/></a>
</p>

---

# Peagen: a Template‑Driven Workflow

## Terminology

- **Tenant** – a namespace used to group related resources, such as repositories.
- **Principal** – an owner of resources (for example, an individual user or an organization).

## Why Use the Peagen CLI?

#### Reduced Variance in LLM‑Driven Generation  
- While LLMs inherently introduce some nondeterminism, Peagen’s structured prompts, injected examples, and dependency‑aware ordering significantly reduce output variance. You’ll still see slight variations on each run, but far less than with ad‑hoc prompt calls.

#### Consistency & Repeatability  
- By centralizing file definitions in a YAML payload plus Jinja2 `ptree.yaml.j2` templates, Peagen ensures every project run follows the same logic. Changes to templates or project YAML immediately propagate on the next `peagen` invocation.

#### No Vector Store—Pure DAG + Jinja2  
- Peagen does *not* rely on a vector store or similarity search. Instead, it constructs a directed acyclic graph (DAG) of inter‑file dependencies, then topologically sorts files to determine processing order. Dependencies and example snippets are injected directly into prompt templates via Jinja2.

#### Built‑In Dependency Management  
- The CLI’s `--transitive` flag toggles between strict and transitive dependency sorts, so you can include or exclude indirect dependencies in your generation run.

#### Seamless LLM Integration  
- In GENERATE mode, the CLI automatically fills agent‑prompt templates with context and dependency examples, calls your configured LLM (e.g. OpenAI’s GPT‑4), and writes back the generated content. All model parameters (provider, model name, temperature) flow through CLI flags and environment variables—no extra scripting needed.

---

## When to Choose CLI over the Programmatic API

### Interactive Iteration  
- Quickly regenerate after tweaking templates or YAML with a single shell command—faster than editing and running a Python script.

### CI/CD Enforcement
- Embed `peagen local sort` and `peagen local process` in pipelines (GitHub Actions, Jenkins, etc.) to ensure generated artifacts stay up to date. Exit codes and verbosity flags integrate seamlessly with automation tools.

### Polyglot & Minimal Overhead  
- Teams in Java, Rust, Go, or any language can use Peagen by installing and invoking the CLI—no Python API import paths to manage.

### What Is Peagen?
#### Core Concepts
> Peagen is a template‑driven orchestration engine that transforms high‑level project definitions into concrete files - statically rendered or LLM‑generated - while respecting inter‑file dependencies.

Peagen’s orchestration layer now lives in the handler modules that back each CLI command. Rather than instantiating a `Peagen` class, the CLI builds `peagen.orm.Task` models and forwards them to functions such as `peagen.handlers.process_handler.process_handler`. Those handlers manage the Git work-tree supplied in `task.args["worktree"]`, resolve configuration via `PluginManager`, and delegate to helper routines in `peagen.core`.

Key primitives you will encounter in the codebase are:

- `load_projects_payload(projects_payload)`
  Parses YAML (or an already loaded document) into a list of project dictionaries and validates it against the current schema.
- `process_all_projects(projects_payload, cfg, transitive=False)`
  Renders every project inside the Git work-tree referenced by `cfg["worktree"]`.
- `process_single_project(project, cfg, start_idx=0, start_file=None, transitive=False)`
  Executes the render → dependency sort → file processing pipeline for a single project and returns both the processed records and the next index.
- `sort_file_records(file_records, *, start_idx=0, start_file=None, transitive=False)`
  Implements the deterministic topological ordering used by both the CLI and worker processes. Dependencies are expressed via `record["EXTRAS"]["DEPENDENCIES"]`.

Every call into `process_core` expects a configuration dictionary containing a `pathlib.Path` under `cfg["worktree"]`. That path is where copy templates are materialised, generated files are written, and—if a `GitVCS` instance is available—commits are staged and pushed. Understanding these functions is the quickest way to extend Peagen programmatically because the CLI is a thin wrapper over them.

---

## Prerequisites & Setup

### Installing Peagen

```bash
# From PyPI (recommended)
pip install peagen

# With Poetry
poetry add peagen

# With uv managing pyproject dependencies
uv add peagen

# Install the CLI globally with uv
uv tool install peagen

# From source (latest development)
git clone https://github.com/swarmauri/swarmauri-sdk.git
cd pkgs/standards/peagen
pip install .

peagen --help
```

### Executing `peagen --help`

```bash
peagen --help
```

![image](https://github.com/user-attachments/assets/52ce6b30-c7cc-4f9e-b9e4-8dde8f034b9c)


### Configuring `OPENAI_API_KEY`

```bash
export OPENAI_API_KEY="sk-…"
```

### CLI Defaults via `.peagen.toml`

Create a `.peagen.toml` in your project root to store provider credentials and
command defaults. A typical configuration might look like:

```toml
# .peagen.toml
[llm]
default_provider = "openai"
default_model_name = "gpt-4"

[llm.api_keys]
openai = "sk-..."

[storage]
default_filter = "file"

[storage.filters.file]
output_dir = "./peagen_artifacts"

[vcs]

default_vcs = "git"


[vcs.adapters.git]
mirror_git_url = "${MIRROR_GIT_URL}"
mirror_git_token = "${MIRROR_GIT_TOKEN}"
owner = "${OWNER}"


[vcs.adapters.git.remotes]
origin = "${GITEA_REMOTE}"
upstream = "${GITHUB_REMOTE}"
```

With these values in place you can omit `--provider`, `--model-name`, and other
flags when running the CLI.
If `--provider` is omitted and no `default_provider` is configured (or the
`PROVIDER` environment variable is unset), Peagen will raise an error.

### Project YAML Schema Overview

```yaml
# projects_payload.yaml
PROJECTS:
  - NAME: "ExampleParserProject"
    ROOT: "pkgs"
    TEMPLATE_SET: "swarmauri_base"
    PACKAGES:
      - NAME: "base/swarmauri_base"
        MODULES:
          - NAME: "ParserBase"
            EXTRAS:
              PURPOSE: "Provide a base implementation of the interface class."
              DESCRIPTION: "Base implementation of the interface class"
              REQUIREMENTS:
                - "Should inherit from the interface first and ComponentBase second."
              RESOURCE_KIND: "parsers"
              INTERFACE_NAME: "IParser"
              INTERFACE_FILE: "pkgs/core/swarmauri_core/parsers/IParser.py"
```

---

## CLI Entry Point Overview

Peagen’s CLI is organised into four top-level groups:

* `peagen fetch` – materialise workspaces and template sets on the local machine.
* `peagen local …` – run handlers directly against the current environment.
* `peagen remote …` – submit tasks to a JSON-RPC gateway.
* `peagen tui` – launch the textual dashboard.

### `peagen fetch`

Synchronise workspaces locally (optionally cloning source packages and installing template sets).

```bash
peagen fetch [WORKSPACE_URI ...] \
  [--no-source/--with-source] \
  [--install-template-sets/--no-install-template-sets] \
  [--repo <GIT_URL>] \
  [--ref <REF>]
```

### `peagen local process`

Render and/or generate files for one or more projects inside an existing Git work-tree.

```bash
peagen local process <PROJECTS_YAML> \
  --repo <PATH_OR_URL> \
  [--ref <REF>] \
  [--project-name <NAME>] \
  [--start-idx <NUM>] \
  [--start-file <PATH>] \
  [--transitive/--no-transitive] \
  [--agent-env '{"provider": "openai", "model_name": "gpt-4"}'] \
  [--output-base <PATH>]
```

Pass `--repo $(pwd)` when operating on a local checkout so the handler can construct the work-tree that `process_core` expects.

![image](https://github.com/user-attachments/assets/1f52f066-8caa-4070-ab63-63350f95b0ee)

### `peagen remote process`

Submit a processing task to a Peagen gateway.

```bash
peagen remote process <PROJECTS_YAML> \
  --repo <GIT_URL> \
  [--ref <REF>] \
  [--project-name <NAME>] \
  [--start-idx <NUM>] \
  [--start-file <PATH>] \
  [--transitive/--no-transitive] \
  [--agent-env <JSON>] \
  [--output-base <PATH>] \
  [--watch/-w] \
  [--interval/-i <SECONDS>]
```

### `peagen local sort`

Inspect the dependency order without writing files.

```bash
peagen local sort <PROJECTS_YAML> \
  [--repo <PATH_OR_URL>] \
  [--ref <REF>] \
  [--project-name <NAME>] \
  [--start-idx <NUM>] \
  [--start-file <PATH>] \
  [--transitive/--no-transitive] \
  [--show-dependencies]
```

![image](https://github.com/user-attachments/assets/216369fb-b9e9-4ab9-84ca-5f13f7dfbc88)

### `peagen remote sort`

```bash
peagen remote sort <PROJECTS_YAML> \
  --repo <GIT_URL> \
  [--ref <REF>] \
  [--project-name <NAME>] \
  [--start-idx <NUM>] \
  [--start-file <PATH>] \
  [--transitive/--no-transitive] \
  [--show-dependencies] \
  [--watch/-w] \
  [--interval/-i <SECONDS>]
```

### `peagen local template-set list`

List available template sets and their directories:

```bash
peagen local template-set list
```

![image](https://github.com/user-attachments/assets/d0757543-87df-45d5-8962-e7580bd3738a)

Remote equivalents (`peagen remote template-set …`) provide the same operations via the gateway.

### `peagen local doe gen`

Expand a Design-of-Experiments spec into a `project_payloads.yaml` bundle.

```bash
peagen local doe gen <DOE_SPEC_YML> <TEMPLATE_PROJECT> \
  [--output project_payloads.yaml] \
  [-c PATH | --config PATH] \
  [--dry-run] \
  [--force]
```

Craft `doe_spec.yml` using the scaffold created by `peagen init doe-spec`. Follow the editing guidelines in [`peagen/scaffold/doe_spec/README.md`](peagen/scaffold/doe_spec/README.md): update factor levels, run `peagen validate doe-spec doe_spec.yml`, bump the version in `spec.yaml`, and never mutate published versions. Remote execution is available via `peagen remote doe gen` with the same flags plus `--repo/--ref` when targeting a gateway.

### `peagen local db upgrade`

Apply Alembic migrations to the latest version. Run this command before starting the gateway to ensure the database schema is current.

```bash
peagen local db upgrade
```

Run migrations on a gateway instance:

```bash
peagen remote db upgrade
```

### Remote Processing with Multi-Tenancy

```bash
peagen remote process projects.yaml \
  --gateway-url http://localhost:8000/rpc \
  --pool acme-lab \
  --repo https://example.com/org/repo.git
```

Pass `--pool` to target a specific tenant or workspace when submitting tasks to the gateway. The shared handler surfaces `--repo` and `--ref` so workflows can operate on any Git repository and reference.

---

## Examples & Walkthroughs

### Single‑Project Processing Example

```bash
peagen local process projects.yaml \
  --repo $(pwd) \
  --project-name MyProject \
  --agent-env '{"provider": "openai", "model_name": "gpt-4"}' \
  -v
```

* Loads only `MyProject` from `projects.yaml`.
* Renders its `ptree.yaml.j2` into file records.
* Builds the dependency DAG and topologically sorts it.
* Processes each record: static or LLM‑generated.

### Batch Processing All Projects

```bash
peagen local process projects.yaml \
  --repo $(pwd) \
  --agent-env '{"provider": "openai", "model_name": "gpt-4"}' \
  -vv
```

* Iterates every project in `projects.yaml`.
* Processes them sequentially (load → render → sort → generate).
* Uses DEBUG logs to print full DAGs and rendered prompts.

### Transitive Dependency Sorting with Resumption

```bash
peagen local process projects.yaml \
  --repo $(pwd) \
  --project-name AnalyticsService \
  --transitive \
  --start-file services/data_pipeline.py \
  -v
```

* Builds full DAG including indirect dependencies.
* Topologically sorts all records.
* Skips ahead to `services/data_pipeline.py` and processes from there.

### Python API: dependency ordering

```python
from peagen.core.sort_core import sort_file_records

file_records = [
    {
        "RENDERED_FILE_NAME": "components/db.py",
        "EXTRAS": {"DEPENDENCIES": ["README.md"]},
    },
    {
        "RENDERED_FILE_NAME": "README.md",
        "EXTRAS": {"DEPENDENCIES": []},
    },
    {
        "RENDERED_FILE_NAME": "components/service.py",
        "EXTRAS": {"DEPENDENCIES": ["components/db.py"]},
    },
]

ordered, next_idx = sort_file_records(file_records=file_records)
print([rec["RENDERED_FILE_NAME"] for rec in ordered])
print(next_idx)
```

---

## Advanced Tips

### Resuming at a Specific Point

* `--start-file <PATH>`: begin at a given file record.
* `--start-idx <NUM>`: jump to a zero‑based index in the sorted list.

### Custom Agent‑Prompt Templates

```bash
peagen local process projects.yaml \
  --repo $(pwd) \
  --agent-env '{"provider": "openai", "model_name": "gpt-4"}' \
  --agent-prompt-template-file ./custom_prompts/my_agent.j2
```

### Integrating into CI/CD Pipelines

```yaml
# .github/workflows/generate.yml
name: Generate Files
on:
  push:
    paths:
      - 'templates/**'
      - 'packages/**'
      - 'projects.yaml'
jobs:
  build:
    runs-on: ubuntu-latest
    steps:
      - uses: actions/checkout@v3
      - name: Set up Python
        uses: actions/setup-python@v4
        with:
          python-version: '3.x'
      - name: Install dependencies
        run: pip install peagen
      - name: Generate files
        env:
          OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
        run: |
          peagen local process projects.yaml \
            --repo "$GITHUB_WORKSPACE" \
            --agent-env '{"provider": "openai", "model_name": "gpt-4"}' \
            --transitive \
            -v
      - name: Commit changes
        run: |
          git config user.name "github-actions[bot]"
          git config user.email "actions@github.com"
          git add .
          git diff --quiet || git commit -m "chore: update generated files"
```


## Conclusion & Next Steps

### Embedding Peagen Programmatically

```python
from peagen.core import Peagen
import os

agent_env = {
    "provider": "openai",
    "model_name": "gpt-4",
    "api_key": os.environ["OPENAI_API_KEY"],
}
pea = Peagen(
    projects_payload_path="projects.yaml",
    agent_env=agent_env,
)

projects = pea.load_projects()
result, idx = pea.process_single_project(projects[0], start_idx=0)
```

### Transport Models

Runtime RPC payloads should be validated using the Pydantic schemas generated
in `peagen.orm.schemas`. For example, use
`TaskRead.model_validate_json()` when decoding a task received over the network:

```python
from peagen.orm.schemas import TaskRead

task = TaskRead.model_validate_json(raw_json)
```

The gateway and worker components rely on these schema classes rather than the
ORM models under `peagen.orm`.

> **Important**
> JSON-RPC methods such as `Task.submit` **only** accept `TaskCreate`,
> `TaskUpdate`, or `TaskRead` instances. Passing dictionaries or nested `dto`
> mappings is unsupported and will trigger a `TypeError`.

> **Note**
> Earlier versions exposed these models under ``peagen.models`` and the
> transport schemas under ``peagen.models.schemas``. Update any imports to use
> ``peagen.orm`` and ``peagen.orm.schemas`` going forward.

### Git Filters & Publishers

Peagen's artifact output and event publishing are pluggable. Use the `git_filter` argument to control where files are saved and optionally provide a publisher for notifications. Built‑ins live under the `peagen.plugins` namespace. Available filters include `S3FSFilter` and `MinioFilter`, while publisher options cover `RedisPublisher`, `RabbitMQPublisher`, and `WebhookPublisher`. See [docs/storage_adapters_and_publishers.md](docs/storage_adapters_and_publishers.md) for details.


For the event schema and routing key conventions, see [docs/eda_protocol.md](docs/eda_protocol.md). Events can also be emitted directly from the CLI using `--notify`:

```bash
peagen local process projects.yaml \
  --repo $(pwd) \
  --notify redis://localhost:6379/0/custom.events
```

For a walkthrough of encrypted secrets and key management, see
[docs/secure_secrets_tutorial.md](docs/secure_secrets_tutorial.md).

### Parallel Processing & Artifact Storage Options

Peagen can accelerate generation by spawning multiple workers. Set `--workers <N>`
on the CLI (or `workers = N` in `.peagen.toml`) to enable a thread pool that
renders files concurrently while still honoring dependency order. Leaving the
flag unset or `0` processes files sequentially.

Artifact locations are resolved via the `--artifacts` flag. Targets may be a
local directory (`file:///./peagen_artifacts`) using `S3FSFilter` or an
S3/MinIO endpoint (`s3://host:9000`) handled by `MinioFilter`. Custom
filters and publishers can be supplied programmatically:

```python
from peagen.core import Peagen
from swarmauri_gitfilter_minio import MinioFilter
from peagen.plugins.publishers.webhook_publisher import WebhookPublisher

store = MinioFilter.from_uri("s3://localhost:9000/peagen")
bus = WebhookPublisher("https://example.com/peagen")
```

Another Example:

```
from peagen.plugins.publishers.redis_publisher import RedisPublisher
from peagen.plugins.publishers.rabbitmq_publisher import RabbitMQPublisher

store = MinioFilter.from_uri("s3://localhost:9000/peagen")
bus = RedisPublisher("redis://localhost:6379/0")
# bus = RabbitMQPublisher(host="localhost", port=5672, routing_key="peagen.events")

pea = Peagen(
    projects_payload_path="projects.yaml",
    git_filter=store,
    agent_env={"provider": "openai", "model_name": "gpt-4"},
)

bus.publish("peagen.events", {"type": "process.started"})
pea.process_all_projects()
```

### Contributing & Extending Templates

* **Template Conventions:** Place new Jinja2 files under your `TEMPLATE_BASE_DIR` as `*.j2`, using the same context variables (`projects`, `packages`, `modules`) that core templates rely on.
* **Adding New Commands:** Define a new subcommand in `cli.py`, wire it into the parser, instantiate `Peagen`, and call core methods.
* **Submitting Pull Requests:** Fork the repo, add/update templates under `peagen/templates/`, update docs/README, and open a PR tagging maintainers.

### Textual TUI

Run `peagen tui` to launch an experimental dashboard that
subscribes to the gateway's `/ws/tasks` WebSocket. The gateway now emits
`task.update`, `worker.update` and `queue.update` events. Use the tab keys to
switch between task lists, logs and opened files. The footer shows system
metrics and current time. Remote artifact paths are downloaded via their git
filter and re-uploaded when saving.

### Streaming Events with wscat

Use the [`wscat`](https://github.com/websockets/wscat) CLI to inspect the
gateway's WebSocket events directly from the terminal:

```bash
npx wscat -c https://gw.peagen.com/ws/tasks
```

Incoming JSON messages mirror those displayed in the TUI, providing a quick way
to monitor `task.update`, `worker.update`, and `queue.update` events.

## Results Backends
Peagen supports pluggable results backends. Built-in options include `local_fs`, `postgres`, and `in_memory`.

