models-dev 1.0.54__tar.gz → 1.0.56__tar.gz

models_dev-1.0.56/.gitignore

@@ -0,0 +1 @@
+/.env

{models_dev-1.0.54 → models_dev-1.0.56}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: models-dev
-Version: 1.0.54
+Version: 1.0.56
 Summary: Typed Python interface to models.dev API data
 Project-URL: Homepage, https://github.com/vklimontovich/models-dev
 Project-URL: Documentation, https://models.dev

{models_dev-1.0.54 → models_dev-1.0.56}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "models-dev"
-version = "1.0.54"
+version = "1.0.56"
 description = "Typed Python interface to models.dev API data"
 readme = "README.md"
 requires-python = ">=3.10"

{models_dev-1.0.54 → models_dev-1.0.56}/scripts/update.py

@@ -1,5 +1,5 @@
 # /// script
-# dependencies = ["anthropic", "httpx", "gitpython"]
+# dependencies = ["anthropic", "httpx", "gitpython", "models-dev"]
 # ///
 """Update data.json.gz from models.dev.
 
@@ -23,6 +23,7 @@ DATA_URL = "https://models.dev/api.json"
 ROOT = Path(__file__).parent.parent.parent
 DATA_PATH = ROOT / "python" / "src" / "models_dev" / "data.json.gz"
 PYPROJECT_PATH = ROOT / "python" / "pyproject.toml"
+NODE_PACKAGE_PATH = ROOT / "node" / "package.json"
 
 
 def fetch_latest() -> str:
@@ -48,11 +49,14 @@ def compute_diff(old: str, new: str) -> str:
     return "".join(diff)
 
 
-def generate_commit_message(diff: str) -> str:
-    """Generate commit message using Anthropic API."""
+MODEL_ID = "claude-haiku-4-5-20251001"
+
+
+def generate_commit_message(diff: str) -> tuple[str, int, int]:
+    """Generate commit message using Anthropic API. Returns (message, in_tokens, out_tokens)."""
     api_key = os.environ.get("ANTHROPIC_API_KEY")
     if not api_key:
-        return "chore: update models data"
+        return "chore: update models data", 0, 0
 
     import anthropic
 
@@ -64,11 +68,28 @@ def generate_commit_message(diff: str) -> str:
     )
     prompt = f"{prompt}\n\n{diff[:10000]}"
     resp = client.messages.create(
-        model="claude-haiku-4-5",
+        model=MODEL_ID,
         max_tokens=1000,
         messages=[{"role": "user", "content": prompt}],
     )
-    return resp.content[0].text.strip()
+    return resp.content[0].text.strip(), resp.usage.input_tokens, resp.usage.output_tokens
+
+
+def print_token_stats(input_tokens: int, output_tokens: int) -> None:
+    """Print token usage and cost using models-dev pricing."""
+    from models_dev import get_model_by_id
+
+    model = get_model_by_id("anthropic", MODEL_ID)
+    if not model.cost:
+        print(f"Tokens: {input_tokens} in, {output_tokens} out (no pricing data)")
+        return
+
+    input_cost = (input_tokens / 1_000_000) * (model.cost.input or 0)
+    output_cost = (output_tokens / 1_000_000) * (model.cost.output or 0)
+    total_cost = input_cost + output_cost
+
+    print(f"Tokens: {input_tokens} in, {output_tokens} out")
+    print(f"Cost: ${total_cost:.6f} (${model.cost.input}/M in, ${model.cost.output}/M out)")
 
 
 def get_version() -> str:
@@ -78,7 +99,7 @@ def get_version() -> str:
 
 
 def bump_version() -> str:
-    """Bump patch version and return new version."""
+    """Bump patch version in pyproject.toml and package.json. Returns new version."""
     content = PYPROJECT_PATH.read_text()
     old = get_version()
     parts = old.split(".")
@@ -86,6 +107,12 @@ def bump_version() -> str:
     new = ".".join(parts)
     content = re.sub(r'^version = ".+"', f'version = "{new}"', content, flags=re.MULTILINE)
     PYPROJECT_PATH.write_text(content)
+
+    # Update node package.json
+    pkg = json.loads(NODE_PACKAGE_PATH.read_text())
+    pkg["version"] = new
+    NODE_PACKAGE_PATH.write_text(json.dumps(pkg, indent=2) + "\n")
+
     return new
 
 
@@ -95,10 +122,28 @@ def save_data(data_str: str) -> None:
         f.write(json.dumps(json.loads(data_str), separators=(",", ":")))
 
 
+def get_action_url() -> str | None:
+    """Get GitHub Action run URL from environment."""
+    server = os.environ.get("GITHUB_SERVER_URL")
+    repo = os.environ.get("GITHUB_REPOSITORY")
+    run_id = os.environ.get("GITHUB_RUN_ID")
+    if server and repo and run_id:
+        return f"{server}/{repo}/actions/runs/{run_id}"
+    return None
+
+
 def commit(message: str) -> None:
     """Commit changes."""
+    action_url = get_action_url()
+    if action_url:
+        message = f"{message}\n\nAction: {action_url}"
+
     repo = git.Repo(ROOT)
-    repo.index.add(["python/pyproject.toml", "python/src/models_dev/data.json.gz"])
+    repo.index.add([
+        "python/pyproject.toml",
+        "python/src/models_dev/data.json.gz",
+        "node/package.json",
+    ])
     repo.index.commit(message)
 
 
@@ -122,8 +167,9 @@ def main() -> int:
     print(diff[:500])
 
     print("Generating commit message...")
-    message = generate_commit_message(diff)
+    message, input_tokens, output_tokens = generate_commit_message(diff)
     print(f"Message: {message}")
+    print_token_stats(input_tokens, output_tokens)
 
     if args.dry_run:
         print("Dry run, stopping")

{models_dev-1.0.54 → models_dev-1.0.56}/uv.lock

@@ -34,7 +34,7 @@ wheels = [
 
 [[package]]
 name = "models-dev"
-version = "1.0.54"
+version = "1.0.56"
 source = { editable = "." }
 
 [package.dev-dependencies]

models_dev-1.0.54/.gitignore

@@ -1,10 +0,0 @@
-.venv/
-__pycache__/
-*.pyc
-*.egg-info/
-dist/
-build/
-.pytest_cache/
-.ruff_cache/
-.idea/
-python/.venv/