fireworks2bedrock-migrate 0.0.1__tar.gz

fireworks2bedrock_migrate-0.0.1/.gitignore

@@ -0,0 +1,28 @@
+# Python
+__pycache__/
+*.py[cod]
+*.egg-info/
+.venv/
+venv/
+dist/
+build/
+.pytest_cache/
+.mypy_cache/
+.ruff_cache/
+
+# Node
+node_modules/
+js/dist/
+*.tsbuildinfo
+
+# Environment and secrets
+.env
+.env.*
+!.env.example
+*.pem
+*accessKeys*.csv
+
+# OS / editors
+.DS_Store
+.idea/
+.vscode/

fireworks2bedrock_migrate-0.0.1/PKG-INFO

@@ -0,0 +1,105 @@
+Metadata-Version: 2.5
+Name: fireworks2bedrock-migrate
+Version: 0.0.1
+Summary: One-shot migration of Kimi K3 inference from Fireworks to Amazon Bedrock for Python repositories
+Project-URL: Repository, https://github.com/HolboxAI/fireworks2bedrock-migrate
+License-Expression: Apache-2.0
+Requires-Python: >=3.9
+Requires-Dist: libcst>=1.1.0
+Provides-Extra: dev
+Requires-Dist: pytest; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# fireworks2bedrock-migrate
+
+One-shot migration of **Kimi K3** inference from **Fireworks** to **Amazon Bedrock**, for **Python** repositories. One command scans the repository, rewrites the inference calls, and opens a pull request. The model does not change — only the backend does.
+
+> JavaScript / TypeScript repositories use the npm package instead: `npx -y fireworks2bedrock-migrate migrate --pr`.
+
+---
+
+## For coding agents (use this README as a skill)
+
+This package is meant to be driven by a coding agent. When asked to *"migrate this repository from Fireworks Kimi K3 to Bedrock"*, follow this workflow exactly and do not improvise on the parts it covers.
+
+### Step 1 — Scan
+
+```bash
+uvx fireworks2bedrock-migrate scan <repo> --json
+```
+
+Read the `findings` and bucket them:
+
+| classification | meaning |
+|---|---|
+| `MIGRATE` | will be rewritten |
+| `BLOCK` | left untouched (fine-tuned models, non-inference APIs, adapters, dedicated deployments) |
+| `REVIEW` | needs a decision (runtime `base_url` clients, computed model IDs, raw HTTP) |
+
+### Step 2 — Migrate and open the pull request
+
+```bash
+uvx fireworks2bedrock-migrate migrate <repo> --pr --json
+```
+
+This creates the branch `f2b/kimi-k3-bedrock`, rewrites the `MIGRATE` files, drops in the `llm_client.py` helper and `f2b.json` routing config, adds the runtime dependencies, writes `BEDROCK_MIGRATION.md`, and opens the pull request.
+
+### Step 3 — Act on the exit code
+
+| code | meaning | what to do |
+|---|---|---|
+| `0` | migration applied (and PR opened with `--pr`) | go to Step 4 |
+| `20` | nothing to migrate, or everything is `BLOCK` | report why; edit nothing |
+| `30` | environment problem (dirty git tree, no `origin` remote) | report the message; edit nothing |
+
+### Step 4 — Verify
+
+The tool does not run the repository's own tests yet (`checks` in the JSON is reserved for that), so the agent verifies:
+
+1. Read `review_items` from the JSON output — each one needs a decision before deploy.
+2. Run the repository's own checks: `pytest -q`, plus lint/type-check/build from `.github/workflows/*.yml` or `pyproject.toml`.
+3. Confirm the diff touches only `MIGRATE` files plus the helper and config.
+4. Fix only failures in the files the tool listed, then push to the branch.
+
+### Step 5 — Report
+
+Report the pull request URL, the `BLOCK` items that were deliberately left alone, and any `REVIEW` items still needing attention.
+
+### Rules — never break these
+
+- Never edit a file or finding the tool classified `BLOCK`.
+- Never hand-edit a model ID; the tool and `f2b.json` own that mapping.
+- Never touch infrastructure files unless the tool lists them as `REVIEW`.
+- Do not add setup steps for the developer — one command is the whole experience.
+
+---
+
+## What the migration changes
+
+- Model ID `accounts/fireworks/models/kimi-k3` → `global.moonshotai.kimi-k3` (`us.moonshotai.kimi-k3` with `--us-only`)
+- `OpenAI(...)` / `Fireworks(...)` / `AsyncFireworks(...)` / `ChatFireworks(...)` → `get_llm_client()` / `get_async_llm_client()` / `ChatOpenAI(model=KIMI_K3, ...)`
+- Adds the `llm_client.py` helper, `f2b.json` routing config, and the runtime deps `openai` + `aws-bedrock-token-generator`
+- Writes `BEDROCK_MIGRATION.md`: full change log, everything left untouched, the IAM policy to attach, and rollback steps
+
+## Rollback
+
+Set `F2B_BACKEND=fireworks` to route back to Fireworks with no code change.
+
+## CLI reference
+
+```
+fireworks2bedrock-migrate scan    [path] [--json]
+fireworks2bedrock-migrate migrate [path] [--pr] [--us-only] [--region <r>] [--json] [--no-git]
+```
+
+- `--pr` pushes the branch and opens a pull request (without it the commit stays local).
+- `--us-only` targets the `us.moonshotai.kimi-k3` profile instead of `global.`.
+- `--no-git` skips branch/commit/push (for testing).
+
+## Scope
+
+| Supported | Not supported (left as `BLOCK`) |
+|---|---|
+| Fireworks **base** Kimi K3 inference | fine-tuned models, LoRA adapters, dedicated deployments |
+| OpenAI SDK, Fireworks SDK, LangChain, raw HTTP inference calls | fine-tuning, embeddings, datasets, batch, rerank, other non-inference APIs |
+| Python repositories | other models and providers |

fireworks2bedrock_migrate-0.0.1/README.md

@@ -0,0 +1,93 @@
+# fireworks2bedrock-migrate
+
+One-shot migration of **Kimi K3** inference from **Fireworks** to **Amazon Bedrock**, for **Python** repositories. One command scans the repository, rewrites the inference calls, and opens a pull request. The model does not change — only the backend does.
+
+> JavaScript / TypeScript repositories use the npm package instead: `npx -y fireworks2bedrock-migrate migrate --pr`.
+
+---
+
+## For coding agents (use this README as a skill)
+
+This package is meant to be driven by a coding agent. When asked to *"migrate this repository from Fireworks Kimi K3 to Bedrock"*, follow this workflow exactly and do not improvise on the parts it covers.
+
+### Step 1 — Scan
+
+```bash
+uvx fireworks2bedrock-migrate scan <repo> --json
+```
+
+Read the `findings` and bucket them:
+
+| classification | meaning |
+|---|---|
+| `MIGRATE` | will be rewritten |
+| `BLOCK` | left untouched (fine-tuned models, non-inference APIs, adapters, dedicated deployments) |
+| `REVIEW` | needs a decision (runtime `base_url` clients, computed model IDs, raw HTTP) |
+
+### Step 2 — Migrate and open the pull request
+
+```bash
+uvx fireworks2bedrock-migrate migrate <repo> --pr --json
+```
+
+This creates the branch `f2b/kimi-k3-bedrock`, rewrites the `MIGRATE` files, drops in the `llm_client.py` helper and `f2b.json` routing config, adds the runtime dependencies, writes `BEDROCK_MIGRATION.md`, and opens the pull request.
+
+### Step 3 — Act on the exit code
+
+| code | meaning | what to do |
+|---|---|---|
+| `0` | migration applied (and PR opened with `--pr`) | go to Step 4 |
+| `20` | nothing to migrate, or everything is `BLOCK` | report why; edit nothing |
+| `30` | environment problem (dirty git tree, no `origin` remote) | report the message; edit nothing |
+
+### Step 4 — Verify
+
+The tool does not run the repository's own tests yet (`checks` in the JSON is reserved for that), so the agent verifies:
+
+1. Read `review_items` from the JSON output — each one needs a decision before deploy.
+2. Run the repository's own checks: `pytest -q`, plus lint/type-check/build from `.github/workflows/*.yml` or `pyproject.toml`.
+3. Confirm the diff touches only `MIGRATE` files plus the helper and config.
+4. Fix only failures in the files the tool listed, then push to the branch.
+
+### Step 5 — Report
+
+Report the pull request URL, the `BLOCK` items that were deliberately left alone, and any `REVIEW` items still needing attention.
+
+### Rules — never break these
+
+- Never edit a file or finding the tool classified `BLOCK`.
+- Never hand-edit a model ID; the tool and `f2b.json` own that mapping.
+- Never touch infrastructure files unless the tool lists them as `REVIEW`.
+- Do not add setup steps for the developer — one command is the whole experience.
+
+---
+
+## What the migration changes
+
+- Model ID `accounts/fireworks/models/kimi-k3` → `global.moonshotai.kimi-k3` (`us.moonshotai.kimi-k3` with `--us-only`)
+- `OpenAI(...)` / `Fireworks(...)` / `AsyncFireworks(...)` / `ChatFireworks(...)` → `get_llm_client()` / `get_async_llm_client()` / `ChatOpenAI(model=KIMI_K3, ...)`
+- Adds the `llm_client.py` helper, `f2b.json` routing config, and the runtime deps `openai` + `aws-bedrock-token-generator`
+- Writes `BEDROCK_MIGRATION.md`: full change log, everything left untouched, the IAM policy to attach, and rollback steps
+
+## Rollback
+
+Set `F2B_BACKEND=fireworks` to route back to Fireworks with no code change.
+
+## CLI reference
+
+```
+fireworks2bedrock-migrate scan    [path] [--json]
+fireworks2bedrock-migrate migrate [path] [--pr] [--us-only] [--region <r>] [--json] [--no-git]
+```
+
+- `--pr` pushes the branch and opens a pull request (without it the commit stays local).
+- `--us-only` targets the `us.moonshotai.kimi-k3` profile instead of `global.`.
+- `--no-git` skips branch/commit/push (for testing).
+
+## Scope
+
+| Supported | Not supported (left as `BLOCK`) |
+|---|---|
+| Fireworks **base** Kimi K3 inference | fine-tuned models, LoRA adapters, dedicated deployments |
+| OpenAI SDK, Fireworks SDK, LangChain, raw HTTP inference calls | fine-tuning, embeddings, datasets, batch, rerank, other non-inference APIs |
+| Python repositories | other models and providers |

fireworks2bedrock_migrate-0.0.1/pyproject.toml

@@ -0,0 +1,21 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "fireworks2bedrock-migrate"
+version = "0.0.1"
+description = "One-shot migration of Kimi K3 inference from Fireworks to Amazon Bedrock for Python repositories"
+readme = "README.md"
+license = "Apache-2.0"
+requires-python = ">=3.9"
+dependencies = ["libcst>=1.1.0"]
+
+[project.optional-dependencies]
+dev = ["pytest"]
+
+[project.scripts]
+fireworks2bedrock-migrate = "fireworks2bedrock_migrate.cli:main"
+
+[project.urls]
+Repository = "https://github.com/HolboxAI/fireworks2bedrock-migrate"

fireworks2bedrock_migrate-0.0.1/src/fireworks2bedrock_migrate/__init__.py

@@ -0,0 +1,3 @@
+"""One-shot migration of Kimi K3 inference from Fireworks to Amazon Bedrock."""
+
+__version__ = "0.0.1"

fireworks2bedrock_migrate-0.0.1/src/fireworks2bedrock_migrate/cli.py

@@ -0,0 +1,145 @@
+"""Command-line entry point."""
+
+import argparse
+import json
+import os
+import sys
+from collections import Counter
+
+from . import __version__
+from .migrate import migrate_repo
+from .scan import detect_language, scan_repo
+
+
+def _build_parser():
+    parser = argparse.ArgumentParser(
+        prog="fireworks2bedrock-migrate",
+        description="Migrate Kimi K3 inference from Fireworks to Amazon Bedrock.",
+    )
+    parser.add_argument("--version", action="version", version=__version__)
+    sub = parser.add_subparsers(dest="command", required=True)
+
+    scan = sub.add_parser("scan", help="List Fireworks usage and how each item would be handled")
+    scan.add_argument("path", nargs="?", default=".")
+    scan.add_argument("--json", action="store_true", help="Machine-readable output on stdout")
+
+    migrate = sub.add_parser("migrate", help="Apply the migration on a new branch")
+    migrate.add_argument("path", nargs="?", default=".")
+    migrate.add_argument("--pr", action="store_true", help="Push the branch and open a pull request")
+    migrate.add_argument("--us-only", action="store_true", help="Use the us. inference profile instead of global.")
+    migrate.add_argument("--region", help="Default AWS region for the helper fallback")
+    migrate.add_argument("--json", action="store_true", help="Machine-readable output on stdout")
+    migrate.add_argument("--no-git", action="store_true", help="Do not branch/commit/push (for testing)")
+
+    return parser
+
+
+def _cmd_scan(args) -> int:
+    path = os.path.abspath(args.path)
+    if not os.path.isdir(path):
+        print(f"error: not a directory: {path}", file=sys.stderr)
+        return 30
+
+    language = detect_language(path)
+    findings = scan_repo(path)
+    code = _exit_code(findings)
+
+    if args.json:
+        payload = {
+            "version": __version__,
+            "repo": path,
+            "language": language,
+            "branch": None,
+            "pr_url": None,
+            "findings": [f.to_dict() for f in findings],
+            "checks": [],
+            "review_items": [],
+            "exit_code": code,
+        }
+        print(json.dumps(payload, indent=2))
+    else:
+        _print_human(language, path, findings)
+
+    return code
+
+
+def _exit_code(findings) -> int:
+    if not findings:
+        return 20
+    if all(f.classification == "BLOCK" for f in findings):
+        return 20
+    return 0
+
+
+def _print_human(language: str, path: str, findings) -> None:
+    print(f"repo:     {path}")
+    print(f"language: {language}")
+    print(f"findings: {len(findings)}")
+    print()
+    if not findings:
+        print("No Fireworks Kimi K3 usage found.")
+        return
+
+    file_width = min(max(len(f.file) for f in findings), 36)
+    header = f"{'CLASS':<8} {'LINE':>5}  {'KIND':<22}  {'FILE':<{file_width}}  DETAIL"
+    print(header)
+    print("-" * len(header))
+    for f in findings:
+        file_col = f.file if len(f.file) <= file_width else "…" + f.file[-(file_width - 1):]
+        print(f"{f.classification:<8} {f.line:>5}  {f.kind:<22}  {file_col:<{file_width}}  {f.detail}")
+
+    counts = Counter(f.classification for f in findings)
+    print()
+    parts = [f"{cls}={counts[cls]}" for cls in ("MIGRATE", "BLOCK", "REVIEW") if counts[cls]]
+    print("summary: " + ", ".join(parts))
+
+
+def _cmd_migrate(args) -> int:
+    path = os.path.abspath(args.path)
+    if not os.path.isdir(path):
+        print(f"error: not a directory: {path}", file=sys.stderr)
+        return 30
+
+    result = migrate_repo(
+        path,
+        us_only=args.us_only,
+        pr=args.pr,
+        region=args.region,
+        apply_git=not args.no_git,
+    )
+
+    if args.json:
+        payload = {
+            "version": __version__,
+            "repo": path,
+            "language": result.language,
+            "branch": result.branch,
+            "pr_url": result.pr_url,
+            "findings": [f.to_dict() for f in result.findings],
+            "checks": result.checks,
+            "review_items": result.review_items,
+            "exit_code": result.exit_code,
+        }
+        print(json.dumps(payload, indent=2))
+    else:
+        print(f"repo:       {path}")
+        print(f"branch:     {result.branch or '(local only)'}")
+        print(f"pr:         {result.pr_url or '(not opened)'}")
+        print(f"touched:    {len(result.touched)} files")
+        print(f"exit_code:  {result.exit_code}")
+
+    return result.exit_code
+
+
+def main(argv=None):
+    parser = _build_parser()
+    args = parser.parse_args(argv)
+    if args.command == "scan":
+        return _cmd_scan(args)
+    if args.command == "migrate":
+        return _cmd_migrate(args)
+    return 2
+
+
+if __name__ == "__main__":
+    sys.exit(main())

fireworks2bedrock_migrate-0.0.1/src/fireworks2bedrock_migrate/findings.py

@@ -0,0 +1,24 @@
+"""A scan finding: one Fireworks touchpoint and how the migration will handle it."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import Any, Dict
+
+
+@dataclass
+class Finding:
+    classification: str  # MIGRATE | BLOCK | REVIEW
+    file: str
+    line: int
+    kind: str
+    detail: str
+
+    def to_dict(self) -> Dict[str, Any]:
+        return {
+            "class": self.classification,
+            "file": self.file,
+            "line": self.line,
+            "kind": self.kind,
+            "detail": self.detail,
+        }