forgeoptimizer 1.0.0__tar.gz

forgeoptimizer-1.0.0/.github/workflows/ci.yml

@@ -0,0 +1,197 @@
+name: ci
+env:
+  PYTHONUTF8: "1"
+on:
+  push:
+    branches: [main]
+    tags:
+      - "v*"
+  pull_request:
+    branches: [main]
+  workflow_dispatch:
+
+jobs:
+  test:
+    name: test (${{ matrix.os }} / py${{ matrix.python-version }})
+    runs-on: ${{ matrix.os }}
+    strategy:
+      fail-fast: false
+      matrix:
+        os: [ubuntu-latest, macos-latest, windows-latest]
+        python-version: ["3.12"]
+
+    steps:
+      - name: Check out
+        uses: actions/checkout@v4
+
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+          cache: pip
+
+      - name: Install
+        run: |
+          python -m pip install --upgrade pip
+          python -m pip install -e ".[dev]"
+
+      - name: Smoke test CLI
+        run: |
+          forge --version
+          forge --help
+          forge graph --help
+
+      - name: Lint
+        run: python -m ruff check forgecli tests
+
+      - name: Tests
+        run: python -m pytest -q
+
+      - name: Cross-platform shell smoke
+        if: runner.os != 'Windows'
+        shell: bash
+        run: |
+          python -c "from forgecli.platform import run; r = run(['echo', 'cross-platform']); print('OK' if r.ok else 'FAIL')"
+
+      - name: Cross-platform shell smoke (Windows)
+        if: runner.os == 'Windows'
+        shell: pwsh
+        run: |
+          python -c "from forgecli.platform import run; r = run(['cmd.exe', '/c', 'echo cross-platform']); print('OK' if r.ok else 'FAIL')"
+
+  build:
+    name: build sdist + wheel
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+      - name: Install build
+        run: python -m pip install --upgrade build twine
+      - name: Build
+        run: python -m build
+      - name: Verify package metadata
+        run: |
+          twine check dist/*
+          python - <<'PY'
+          import zipfile
+          from pathlib import Path
+          wheel = next(Path("dist").glob("*.whl"))
+          with zipfile.ZipFile(wheel) as zf:
+              names = zf.namelist()
+          assert any(n.startswith("forgecli/") for n in names), "missing forgecli package"
+          print(f"OK: {wheel.name}")
+          PY
+      - name: Upload artifacts
+        uses: actions/upload-artifact@v4
+        with:
+          name: dist
+          path: dist/
+
+  uv-install:
+    name: uv tool install (${{ matrix.os }})
+    needs: build
+    runs-on: ${{ matrix.os }}
+    strategy:
+      fail-fast: false
+      matrix:
+        os: [ubuntu-latest, macos-latest, windows-latest]
+    steps:
+      - name: Check out
+        uses: actions/checkout@v4
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+
+      - name: Install uv
+        uses: astral-sh/setup-uv@v6
+
+      - uses: actions/download-artifact@v4
+        with:
+          name: dist
+          path: dist/
+
+      - name: Install forgectx via uv tool
+        shell: bash
+        run: |
+          uv tool install dist/*.whl
+
+      - name: Verify forge --help
+        run: forge --help
+
+      - name: Verify wrapper commands are registered
+        run: |
+          forge claude --help
+          forge codex --help
+          forge cursor --help
+          forge opencode --help
+          forge commandcode --help
+
+
+  publish-testpypi:
+    name: publish to TestPyPI (on tag or manual)
+    if: startsWith(github.ref, 'refs/tags/v') || github.event_name == 'workflow_dispatch'
+    needs: [test, build, uv-install]
+    runs-on: ubuntu-latest
+    environment: testpypi
+    permissions:
+      id-token: write
+      contents: read
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+      - name: Rename package to match TestPyPI Trusted Publisher
+        run: |
+          python -c "
+          p = 'pyproject.toml'
+          content = open(p).read()
+          content = content.replace('name = \"forgectx\"', 'name = \"forgekit\"')
+          open(p, 'w').write(content)
+          "
+      - name: Build sdist and wheel
+        run: |
+          pip install --upgrade build
+          python -m build
+      - name: Publish
+        uses: pypa/gh-action-pypi-publish@release/v1
+        with:
+          repository-url: https://test.pypi.org/legacy/
+          skip-existing: true
+
+  publish-pypi:
+    name: publish to PyPI (on tag or manual)
+    if: startsWith(github.ref, 'refs/tags/v') || github.event_name == 'workflow_dispatch'
+    needs: [test, build, uv-install]
+    runs-on: ubuntu-latest
+    environment: pypi
+    permissions:
+      id-token: write
+      contents: read
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+      - name: Rename package to match PyPI Trusted Publisher
+        run: |
+          python -c "
+          p = 'pyproject.toml'
+          content = open(p).read()
+          content = content.replace('name = \"forgectx\"', 'name = \"forgeoptimizer\"')
+          open(p, 'w').write(content)
+          "
+      - name: Build sdist and wheel
+        run: |
+          pip install --upgrade build
+          python -m build
+      - name: Publish
+        uses: pypa/gh-action-pypi-publish@release/v1
+        with:
+          skip-existing: true
+
+

forgeoptimizer-1.0.0/.gitignore

@@ -0,0 +1,51 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+.commandcode/
+# C extensions
+*.so
+
+# Distribution / packaging
+build/
+graphify-out/
+dist/
+*.egg-info/
+*.egg
+
+# But the in-tree `forgecli/build/` package is part of our source.
+!forgecli/build/
+
+# Virtual environments
+.venv/
+venv/
+env/
+.env
+
+# Test / coverage
+.pytest_cache/
+.coverage
+htmlcov/
+.mypy_cache/
+.ruff_cache/
+
+# Editor
+.idea/
+.vscode/
+*.swp
+.DS_Store
+
+# Local config / secrets
+.env
+.env.*
+!.env.example
+*.local.toml
+
+# Local data
+*.db
+*.sqlite
+*.sqlite3
+
+# Test artifacts (prevent CI failures)
+foo.py
+main.go

forgeoptimizer-1.0.0/Forge.toml

@@ -0,0 +1,17 @@
+[app]
+log_level = "INFO"
+
+[git]
+default_branch = "main"
+
+[graph]
+enabled = true
+languages = ["python", "typescript"]
+
+[prompt_optimizer]
+backend = "ruleset"
+enabled = true
+intensity = "lite"
+
+[providers]
+allowed = ["mock"]

forgeoptimizer-1.0.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Forge Authors
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

forgeoptimizer-1.0.0/PKG-INFO

@@ -0,0 +1,169 @@
+Metadata-Version: 2.4
+Name: forgeoptimizer
+Version: 1.0.0
+Summary: Forge — AI optimization runtime. Fast prompt and token optimization for Claude Code, Codex, Cursor, OpenCode, and CommandCode CLI.
+Project-URL: Homepage, https://github.com/mdshzb04/ForgeCli
+Project-URL: Repository, https://github.com/mdshzb04/ForgeCli
+Project-URL: Documentation, https://github.com/mdshzb04/ForgeCli#readme
+Project-URL: Issues, https://github.com/mdshzb04/ForgeCli/issues
+Author: Forge Authors
+License: MIT License
+        
+        Copyright (c) 2025 Forge Authors
+        
+        Permission is hereby granted, free of charge, to any person obtaining a copy
+        of this software and associated documentation files (the "Software"), to deal
+        in the Software without restriction, including without limitation the rights
+        to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+        copies of the Software, and to permit persons to whom the Software is
+        furnished to do so, subject to the following conditions:
+        
+        The above copyright notice and this permission notice shall be included in all
+        copies or substantial portions of the Software.
+        
+        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+        AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+        LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+        OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+        SOFTWARE.
+License-File: LICENSE
+Keywords: ai,claude,cli,codex,cursor,developer-tools
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Software Development
+Requires-Python: >=3.12
+Requires-Dist: aiofiles>=23.2.1
+Requires-Dist: click>=8.1.7
+Requires-Dist: cryptography>=42.0.5
+Requires-Dist: gitpython>=3.1.42
+Requires-Dist: httpx>=0.27.0
+Requires-Dist: keyring>=25.2.0
+Requires-Dist: platformdirs>=4.2.0
+Requires-Dist: pydantic-settings>=2.2.0
+Requires-Dist: pydantic>=2.6.0
+Requires-Dist: pyyaml>=6.0
+Requires-Dist: rich>=13.7.0
+Requires-Dist: tree-sitter-languages>=1.10.0
+Requires-Dist: tree-sitter>=0.21.0
+Requires-Dist: typer>=0.12.0
+Provides-Extra: dev
+Requires-Dist: build>=1.0.0; extra == 'dev'
+Requires-Dist: mypy>=1.10.0; extra == 'dev'
+Requires-Dist: pytest-asyncio>=0.23.0; extra == 'dev'
+Requires-Dist: pytest-cov>=4.1.0; extra == 'dev'
+Requires-Dist: pytest-timeout>=2.3.0; extra == 'dev'
+Requires-Dist: pytest>=8.0.0; extra == 'dev'
+Requires-Dist: respx>=0.21.0; extra == 'dev'
+Requires-Dist: ruff>=0.4.0; extra == 'dev'
+Requires-Dist: twine>=5.0.0; extra == 'dev'
+Requires-Dist: types-pyyaml; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# Forge
+
+Forge is an AI optimization runtime. It prepares your repository context at light speed — prompt optimization and token compression — then launches the AI CLI you already use.
+
+Install:
+
+```bash
+uv tool install forgectx
+```
+
+The command is `forge`.
+
+## What it does
+
+When you run `forge claude`, `forge codex`, `forge cursor`, `forge opencode`, or `forge commandcode`, Forge:
+
+1. Detects your git repository (or current directory)
+2. Builds a **shallow** repo snapshot — it does **not** index your whole codebase
+3. Runs prompt optimization (Ponytail ruleset) and token compression
+4. Reuses cached results when nothing important changed
+5. Launches the selected CLI with `FORGE_CONTEXT` and `FORGE_CONTEXT_FILE` set
+
+Wrappers feel instant because Forge skips full graph builds during normal use.
+
+## Commands
+
+| Command | Description |
+| -------- | ------------ |
+| `forge claude` | Launch Claude Code with optimized context |
+| `forge codex` | Launch Codex CLI with optimized context |
+| `forge cursor` | Launch Cursor CLI with optimized context |
+| `forge opencode` | Launch OpenCode CLI with optimized context |
+| `forge commandcode` | Launch CommandCode CLI with optimized context |
+| `forge graph build` | Optionally build a full knowledge graph via Graphify |
+| `forge --version` | Show version |
+
+Pass through any flags your CLI supports:
+
+```bash
+forge claude -- "fix the failing test in tests/test_foo.py"
+forge codex --help
+forge cursor --refresh
+```
+
+Use `--refresh` to bypass Forge's context cache.
+
+## Optional: knowledge graph
+
+If you want a full codebase knowledge graph (separate from the fast wrapper path):
+
+```bash
+uv tool install graphifyy
+forge graph build
+```
+
+Graphify runs only when you ask for it — not on every `forge claude` / `codex` / `cursor` invocation.
+
+## Environment variables
+
+Wrappers export:
+
+| Variable | Purpose |
+| -------- | -------- |
+| `FORGE_CONTEXT` | Optimized text context |
+| `FORGE_CONTEXT_FILE` | Path to the optimized context file |
+| `FORGE_REPO_ROOT` | Detected repository root |
+
+## Development
+
+```bash
+git clone https://github.com/mdshzb04/ForgeCli
+cd ForgeCli
+python -m venv .venv && source .venv/bin/activate
+pip install -e ".[dev]"
+pytest
+ruff check forgecli tests
+```
+
+## Release
+
+Push a version tag to publish to TestPyPI and PyPI:
+
+```bash
+git tag v1.0.0
+git push origin v1.0.0
+```
+
+Publish jobs run **only on `v*` tags**. On regular pushes to `main`, those jobs show as **Skipped** — that is expected, not a failure.
+
+Configure GitHub environments `testpypi` and `pypi` with PyPI trusted publishing for package name `forgectx`.
+
+---
+
+<img width="1854" height="1005" alt="image" src="https://github.com/user-attachments/assets/03f3c2e2-424c-4784-8a59-b2b0f4b99447" />
+
+<img width="1854" height="1005" alt="image" src="https://github.com/user-attachments/assets/6eb06d10-6f1f-4648-b679-028368362c24" />
+
+## License
+
+[MIT](LICENSE)

forgeoptimizer-1.0.0/README.md

@@ -0,0 +1,100 @@
+# Forge
+
+Forge is an AI optimization runtime. It prepares your repository context at light speed — prompt optimization and token compression — then launches the AI CLI you already use.
+
+Install:
+
+```bash
+uv tool install forgectx
+```
+
+The command is `forge`.
+
+## What it does
+
+When you run `forge claude`, `forge codex`, `forge cursor`, `forge opencode`, or `forge commandcode`, Forge:
+
+1. Detects your git repository (or current directory)
+2. Builds a **shallow** repo snapshot — it does **not** index your whole codebase
+3. Runs prompt optimization (Ponytail ruleset) and token compression
+4. Reuses cached results when nothing important changed
+5. Launches the selected CLI with `FORGE_CONTEXT` and `FORGE_CONTEXT_FILE` set
+
+Wrappers feel instant because Forge skips full graph builds during normal use.
+
+## Commands
+
+| Command | Description |
+| -------- | ------------ |
+| `forge claude` | Launch Claude Code with optimized context |
+| `forge codex` | Launch Codex CLI with optimized context |
+| `forge cursor` | Launch Cursor CLI with optimized context |
+| `forge opencode` | Launch OpenCode CLI with optimized context |
+| `forge commandcode` | Launch CommandCode CLI with optimized context |
+| `forge graph build` | Optionally build a full knowledge graph via Graphify |
+| `forge --version` | Show version |
+
+Pass through any flags your CLI supports:
+
+```bash
+forge claude -- "fix the failing test in tests/test_foo.py"
+forge codex --help
+forge cursor --refresh
+```
+
+Use `--refresh` to bypass Forge's context cache.
+
+## Optional: knowledge graph
+
+If you want a full codebase knowledge graph (separate from the fast wrapper path):
+
+```bash
+uv tool install graphifyy
+forge graph build
+```
+
+Graphify runs only when you ask for it — not on every `forge claude` / `codex` / `cursor` invocation.
+
+## Environment variables
+
+Wrappers export:
+
+| Variable | Purpose |
+| -------- | -------- |
+| `FORGE_CONTEXT` | Optimized text context |
+| `FORGE_CONTEXT_FILE` | Path to the optimized context file |
+| `FORGE_REPO_ROOT` | Detected repository root |
+
+## Development
+
+```bash
+git clone https://github.com/mdshzb04/ForgeCli
+cd ForgeCli
+python -m venv .venv && source .venv/bin/activate
+pip install -e ".[dev]"
+pytest
+ruff check forgecli tests
+```
+
+## Release
+
+Push a version tag to publish to TestPyPI and PyPI:
+
+```bash
+git tag v1.0.0
+git push origin v1.0.0
+```
+
+Publish jobs run **only on `v*` tags**. On regular pushes to `main`, those jobs show as **Skipped** — that is expected, not a failure.
+
+Configure GitHub environments `testpypi` and `pypi` with PyPI trusted publishing for package name `forgectx`.
+
+---
+
+<img width="1854" height="1005" alt="image" src="https://github.com/user-attachments/assets/03f3c2e2-424c-4784-8a59-b2b0f4b99447" />
+
+<img width="1854" height="1005" alt="image" src="https://github.com/user-attachments/assets/6eb06d10-6f1f-4648-b679-028368362c24" />
+
+## License
+
+[MIT](LICENSE)

forgeoptimizer-1.0.0/forgecli/__init__.py

@@ -0,0 +1,4 @@
+"""Forge — AI optimization runtime."""
+
+__version__ = "1.0.0"
+__app_name__ = "Forge"

forgeoptimizer-1.0.0/forgecli/build/__init__.py

@@ -0,0 +1,135 @@
+"""Build pipeline: Graphify retrieval -> Ponytail -> LLM -> diff -> apply -> test -> summary.
+
+Each stage is a small async function that takes a :class:`BuildContext` and
+mutates it, returning the same context. The :class:`BuildPipeline` runs the
+stages in order and records per-stage status; on failure it short-circuits
+unless ``continue_on_error`` is set.
+"""
+
+from __future__ import annotations
+
+from collections.abc import Awaitable, Callable
+from dataclasses import dataclass, field
+from enum import Enum
+from pathlib import Path
+from typing import Any
+
+from forgecli.providers.base import ChatResponse
+from forgecli.providers.router import RouteDecision
+
+
+class StageStatus(str, Enum):
+    """Lifecycle status of a single pipeline stage."""
+
+    PENDING = "pending"
+    RUNNING = "running"
+    SUCCEEDED = "succeeded"
+    FAILED = "failed"
+    SKIPPED = "skipped"
+
+
+@dataclass
+class StageRecord:
+    """One row in the pipeline's per-stage log."""
+
+    name: str
+    status: StageStatus = StageStatus.PENDING
+    started_at: float | None = None
+    finished_at: float | None = None
+    notes: tuple[str, ...] = ()
+    error: str | None = None
+
+    @property
+    def duration_seconds(self) -> float | None:
+        if self.started_at is None or self.finished_at is None:
+            return None
+        return self.finished_at - self.started_at
+
+
+@dataclass
+class BuildContext:
+    """Mutable state shared by every pipeline stage."""
+
+    prompt: str
+    root: Path
+    decision: RouteDecision | None = None
+    retrieval: str = ""               # graph-derived context
+    caveman_optimized_request: Any = None
+    caveman_optimized_notes: tuple[str, ...] = ()
+    optimized_request: Any = None
+    optimized_notes: tuple[str, ...] = ()
+    response: ChatResponse | None = None
+    diff_text: str = ""
+    applied_files: list[Path] = field(default_factory=list)
+    test_stdout: str = ""
+    test_stderr: str = ""
+    test_returncode: int | None = None
+    summary: str = ""
+    extras: dict[str, Any] = field(default_factory=dict)
+    stages: list[StageRecord] = field(default_factory=list)
+
+
+PipelineStage = Callable[[BuildContext], Awaitable[BuildContext]]
+
+
+@dataclass
+class BuildResult:
+    """The output of :meth:`BuildPipeline.run`."""
+
+    success: bool
+    context: BuildContext
+    failure_stage: str | None = None
+
+
+class BuildPipeline:
+    """A composable async pipeline of named stages."""
+
+    def __init__(
+        self,
+        stages: list[tuple[str, PipelineStage]],
+        *,
+        continue_on_error: bool = False,
+    ) -> None:
+        self._stages = list(stages)
+        self._continue_on_error = continue_on_error
+
+    @property
+    def stage_names(self) -> list[str]:
+        return [name for name, _ in self._stages]
+
+    async def run(self, context: BuildContext) -> BuildResult:
+        context.stages = [
+            StageRecord(name=name) for name, _ in self._stages
+        ]
+        for index, (name, stage) in enumerate(self._stages):
+            record = context.stages[index]
+            record.status = StageStatus.RUNNING
+            import time
+
+            record.started_at = time.perf_counter()
+            try:
+                context = await stage(context)
+                record.status = StageStatus.SUCCEEDED
+            except Exception as exc:  # noqa: BLE001 - we want to capture everything
+                record.status = StageStatus.FAILED
+                record.error = repr(exc)
+                record.finished_at = time.perf_counter()
+                if not self._continue_on_error:
+                    return BuildResult(
+                        success=False,
+                        context=context,
+                        failure_stage=name,
+                    )
+            else:
+                record.finished_at = time.perf_counter()
+        return BuildResult(success=True, context=context)
+
+
+__all__ = [
+    "BuildContext",
+    "BuildPipeline",
+    "BuildResult",
+    "PipelineStage",
+    "StageRecord",
+    "StageStatus",
+]