driftfix 0.1.0__py3-none-any.whl

driftfix/cli.py

@@ -0,0 +1,138 @@
+"""driftfix: a dependency upgrade broke your tests, Claude fixes the code."""
+
+from __future__ import annotations
+
+import argparse
+import asyncio
+import subprocess
+import sys
+from dataclasses import dataclass
+from pathlib import Path
+
+# Pinning back to the old version is not a fix.
+MANIFESTS = {
+    "requirements.txt", "pyproject.toml", "setup.py", "setup.cfg", "Pipfile", "Pipfile.lock",
+    "poetry.lock", "uv.lock", "package.json", "package-lock.json", "yarn.lock", "pnpm-lock.yaml",
+}
+
+PROMPT = """\
+The dependency `{package}` was upgraded{versions}. The test suite now fails:
+
+$ {test_command}
+{output}
+
+Fix this repository's code so it works with the new version of `{package}`.
+- Read the package's changelog / migration guide for the relevant versions if you need to.
+- Change application code. Change tests only where they use `{package}`'s API directly.
+- Never pin, downgrade, or edit dependency files ({manifests}).
+- Never delete or skip tests to make them pass.
+- Run `{test_command}` to confirm the fix.
+Your final message must be only a short markdown summary of what changed and why, for a PR description."""
+
+
+@dataclass
+class TestRun:
+    passed: bool
+    output: str
+
+
+def run_tests(command: str, repo: Path, timeout: int) -> TestRun:
+    try:
+        proc = subprocess.run(
+            command, shell=True, cwd=repo, capture_output=True, text=True, timeout=timeout
+        )
+    except subprocess.TimeoutExpired:
+        return TestRun(False, f"timed out after {timeout}s")
+    lines = (proc.stdout + proc.stderr).splitlines()
+    return TestRun(proc.returncode == 0, "\n".join(lines[-150:]))
+
+
+def changed_files(repo: Path) -> list[str]:
+    out = subprocess.run(
+        ["git", "status", "--porcelain"], cwd=repo, capture_output=True, text=True, check=True
+    ).stdout
+    return [line[3:].split(" -> ")[-1] for line in out.splitlines()]
+
+
+def build_prompt(package: str, old: str | None, new: str | None, command: str, out: str) -> str:
+    versions = f" from {old or '?'} to {new}" if new else (f" from {old}" if old else "")
+    return PROMPT.format(
+        package=package, versions=versions, test_command=command, output=out,
+        manifests=", ".join(sorted(MANIFESTS)),
+    )
+
+
+async def run_agent(prompt: str, repo: Path, model: str, max_turns: int, budget: float) -> tuple[str, float]:
+    from claude_agent_sdk import ClaudeAgentOptions, ResultMessage, query
+
+    options = ClaudeAgentOptions(
+        cwd=repo,
+        model=model,
+        allowed_tools=["Read", "Edit", "Write", "Glob", "Grep", "Bash", "WebFetch", "WebSearch"],
+        permission_mode="bypassPermissions",  # headless; run in CI or a throwaway checkout
+        max_turns=max_turns,
+        max_budget_usd=budget,
+        setting_sources=["project"],  # repo's CLAUDE.md, not the user's personal config
+        skills=[],
+    )
+    summary, cost = "", 0.0
+    async for message in query(prompt=prompt, options=options):
+        if isinstance(message, ResultMessage):
+            summary, cost = message.result or "", message.total_cost_usd or 0.0
+    return summary, cost
+
+
+def fix(args: argparse.Namespace) -> int:
+    repo = args.repo.resolve()
+    if changed_files(repo):
+        print("Working tree is dirty. Commit or stash first so driftfix's edits are reviewable.")
+        return 2
+    before = run_tests(args.test, repo, args.timeout)
+    if before.passed:
+        print(f"Tests pass with {args.package} as installed. Nothing to fix.")
+        return 0
+    baseline = set(changed_files(repo))  # test-run artifacts like __pycache__
+
+    print(f"Tests fail after upgrading {args.package}. Handing off to Claude ({args.model})...")
+    prompt = build_prompt(args.package, args.from_version, args.to_version, args.test, before.output)
+    try:
+        summary, cost = asyncio.run(run_agent(prompt, repo, args.model, args.max_turns, args.budget))
+    except Exception as exc:  # still report on the PR instead of dying silently
+        summary, cost = f"Agent error: `{exc}`", 0.0
+
+    touched = [f for f in changed_files(repo) if f not in baseline]
+    pinned = [f for f in touched if Path(f).name in MANIFESTS]
+    after = run_tests(args.test, repo, args.timeout)  # don't trust the agent's word
+    ok = after.passed and bool(touched) and not pinned
+
+    status = "✅ Fixed" if ok else "❌ Not fixed"
+    report = f"## driftfix: {args.package}\n\n**{status}** · cost ${cost:.2f}\n\n{summary}\n"
+    if pinned:
+        report += f"\nRejected: agent edited dependency files {pinned}.\n"
+    if not after.passed:
+        report += f"\nTests still failing:\n```\n{after.output[-3000:]}\n```\n"
+    print(report)
+    if args.summary:
+        args.summary.write_text(report)
+    return 0 if ok else 1
+
+
+def main(argv: list[str] | None = None) -> int:
+    p = argparse.ArgumentParser(prog="driftfix", description=__doc__)
+    sub = p.add_subparsers(dest="command", required=True)
+    f = sub.add_parser("fix", help="fix code broken by a dependency upgrade")
+    f.add_argument("--package", required=True)
+    f.add_argument("--from", dest="from_version")
+    f.add_argument("--to", dest="to_version")
+    f.add_argument("--test", default="pytest -q", help="test command (default: pytest -q)")
+    f.add_argument("--repo", type=Path, default=Path("."))
+    f.add_argument("--model", default="claude-opus-5")
+    f.add_argument("--max-turns", type=int, default=50)
+    f.add_argument("--budget", type=float, default=5.0, help="max spend in USD (default: 5)")
+    f.add_argument("--timeout", type=int, default=600, help="test timeout in seconds")
+    f.add_argument("--summary", type=Path, help="write the markdown report here")
+    return fix(p.parse_args(argv))
+
+
+if __name__ == "__main__":
+    sys.exit(main())

driftfix-0.1.0.dist-info/METADATA

@@ -0,0 +1,104 @@
+Metadata-Version: 2.5
+Name: driftfix
+Version: 0.1.0
+Summary: Dependabot bumps it. CI breaks. driftfix fixes the code.
+Project-URL: Repository, https://github.com/Gthejesraj/driftfix
+License-Expression: MIT
+License-File: LICENSE
+Requires-Python: >=3.10
+Requires-Dist: claude-agent-sdk>=0.2
+Provides-Extra: dev
+Requires-Dist: pytest>=8; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# driftfix
+
+**Dependabot bumps it. CI breaks. driftfix fixes the code.**
+
+Dependency bots open the upgrade PR, then leave you to fix whatever broke.
+driftfix finishes the job: it reads the failing tests, has Claude adapt your
+code to the new version, reruns your tests itself, and pushes the fix onto
+the same PR. If it can't get to green, it comments what it tried and commits
+nothing.
+
+```
+Dependabot PR (openai 0.28 → 1.x) → tests fail → Claude fixes code → tests rerun → ✅ commit on the PR
+                                                                              └→ ❌ comment only
+```
+
+## Benchmark
+
+Real breaking upgrades, each a small app with tests that pass on the old version and fail on the new one ([`bench/`](bench/)):
+
+| Package | Upgrade | Fixed | Cost | Diff |
+|---|---|---|---|---|
+| httpx | 0.27.2 → 0.28.1 | ✅ | $0.12 | 1 file changed, 4 insertions(+), 1 deletion(-) |
+| numpy | 1.26.4 → 2.5.3 | ✅ | $0.14 | 1 file changed, 5 insertions(+), 5 deletions(-) |
+| openai | 0.28.1 → 3.24.0 | ✅ | $0.29 | 1 file changed, 4 insertions(+), 3 deletions(-) |
+| pydantic | 1.10.21 → 2.13.5 | ✅ | $0.20 | 1 file changed, 11 insertions(+), 11 deletions(-) |
+| sqlalchemy | 1.4.54 → 2.1.3 | ✅ | $0.19 | 1 file changed, 7 insertions(+), 4 deletions(-) |
+| pydantic (6-file app, shared base model) | 1.10.21 → 2.13.5 | ✅ | $0.51 | 4 files changed, 47 insertions(+), 29 deletions(-) |
+
+6/6 fixed for $1.45 total, no test edits. The 6-file case runs with deprecation warnings as errors and includes a silent behavior change (v2 stops coercing numeric SKUs to strings). Rerun: `python bench/run.py` (or `--check` to validate cases for free).
+More cases welcome: add a folder under `bench/cases/`.
+
+## GitHub Action
+
+```yaml
+# .github/workflows/driftfix.yml
+name: driftfix
+on: pull_request
+permissions: { contents: write, pull-requests: write }
+jobs:
+  fix:
+    if: github.actor == 'dependabot[bot]'
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+        with: { ref: "${{ github.head_ref }}" }
+      - uses: Gthejesraj/driftfix@main
+        with:
+          anthropic-api-key: ${{ secrets.ANTHROPIC_API_KEY }}
+          setup: pip install -r requirements.txt
+          test: pytest -q
+```
+
+**Important:** workflows triggered by Dependabot can't read your normal repo
+secrets. Add `ANTHROPIC_API_KEY` under **Settings → Secrets → Dependabot**.
+
+Commits pushed with the default `GITHUB_TOKEN` don't re-trigger CI. Use a PAT
+or GitHub App token in `actions/checkout` if you want CI to rerun on the fix.
+
+## CLI
+
+```bash
+pip install driftfix
+# after upgrading a dependency and seeing tests fail:
+driftfix fix --package openai --from 0.28 --to 1.40 --test "pytest -q"
+```
+
+| Flag | Default | |
+|---|---|---|
+| `--test` | `pytest -q` | any command; exit 0 = green |
+| `--model` | `claude-opus-5` | |
+| `--budget` | `5` | max USD per run |
+| `--max-turns` | `50` | |
+| `--summary FILE` | | write the markdown report |
+
+## Guardrails
+
+- Your tests decide. driftfix reruns them after the agent finishes; the agent's word doesn't count.
+- Edits to dependency files (pinning back) are rejected.
+- Requires a clean git tree, so every change is a reviewable diff.
+- Never merges anything. The PR is still yours to review.
+- The agent runs with full tool access inside the checkout. Run it in CI or a throwaway clone.
+
+## Roadmap
+
+- More benchmark cases (pandas 2, Django, langchain), harder multi-file repos
+- npm / Renovate support
+- Provider mode: packages ship migration notes that driftfix reads
+
+## License
+
+MIT

driftfix-0.1.0.dist-info/RECORD

@@ -0,0 +1,7 @@
+driftfix/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+driftfix/cli.py,sha256=HY8x1AI1HTWbhEnhnKx5ZOTYTkN--Y39cLojhcT-_3E,5723
+driftfix-0.1.0.dist-info/METADATA,sha256=EsExwlUUHt-Dkvy6cvC6VQ_htJfAMhDnYzhQyWpLm_Y,3969
+driftfix-0.1.0.dist-info/WHEEL,sha256=W3fkpkm7-wf9vBI5Z-7s0eWkeM-spu78I8Neb98DeEg,87
+driftfix-0.1.0.dist-info/entry_points.txt,sha256=Gn3xBJyokXKDUsya24AbEE8W_6Gj-Vt00IRbvNeR4tU,47
+driftfix-0.1.0.dist-info/licenses/LICENSE,sha256=uewuVeXSAUsHUGdfCwBJow1jblK9GtiJwMRdVBSvXzQ,1078
+driftfix-0.1.0.dist-info/RECORD,,

driftfix-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.32.4
+Root-Is-Purelib: true
+Tag: py3-none-any

driftfix-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+driftfix = driftfix.cli:main

driftfix-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 driftfix contributors
+
+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.