gac 1.0.0__py3-none-any.whl → 1.0.1__py3-none-any.whl

gac/__version__.py

@@ -1,3 +1,3 @@
 """Version information for gac package."""
 
-__version__ = "1.0.0"
+__version__ = "1.0.1"

gac/ai.py

@@ -5,9 +5,11 @@ It consolidates all AI-related functionality including token counting and commit
 """
 
 import logging
+import os
 from functools import lru_cache
 from typing import Any
 
+import httpx
 import tiktoken
 
 from gac.ai_providers import (
@@ -30,19 +32,10 @@ def count_tokens(content: str | list[dict[str, str]] | dict[str, Any], model: st
         return 0
 
     if model.startswith("anthropic"):
-        import anthropic
-
-        try:
-            client = anthropic.Anthropic()
-
-            # Use the messages.count_tokens API for accurate counting
-            model_name = model.split(":", 1)[1] if ":" in model else "claude-3-5-haiku-latest"
-            response = client.messages.count_tokens(model=model_name, messages=[{"role": "user", "content": text}])
-
-            return response.input_tokens
-        except Exception:
-            # Fallback to simple estimation for Anthropic models
-            return len(text) // 4
+        anthropic_tokens = anthropic_count_tokens(text, model)
+        if anthropic_tokens is not None:
+            return anthropic_tokens
+        return len(text) // 4
 
     try:
         encoding = get_encoding(model)
@@ -52,6 +45,60 @@ def count_tokens(content: str | list[dict[str, str]] | dict[str, Any], model: st
         return len(text) // 4
 
 
+def anthropic_count_tokens(text: str, model: str) -> int | None:
+    """Call Anthropic's token count endpoint and return the token usage.
+
+    Returns the token count when successful, otherwise ``None`` so callers can
+    fall back to a heuristic estimate.
+    """
+    api_key = os.getenv("ANTHROPIC_API_KEY")
+    if not api_key:
+        logger.debug("ANTHROPIC_API_KEY not set; using heuristic token estimation for Anthropic model")
+        return None
+
+    model_name = model.split(":", 1)[1] if ":" in model else "claude-3-5-haiku-latest"
+    headers = {
+        "Content-Type": "application/json",
+        "x-api-key": api_key,
+        "anthropic-version": "2023-06-01",
+    }
+    payload = {
+        "model": model_name,
+        "messages": [
+            {
+                "role": "user",
+                "content": [
+                    {
+                        "type": "text",
+                        "text": text,
+                    }
+                ],
+            }
+        ],
+    }
+
+    try:
+        response = httpx.post(
+            "https://api.anthropic.com/v1/messages/count_tokens",
+            headers=headers,
+            json=payload,
+            timeout=30.0,
+        )
+        response.raise_for_status()
+        data = response.json()
+
+        if "input_tokens" in data:
+            return data["input_tokens"]
+        if "usage" in data and "input_tokens" in data["usage"]:
+            return data["usage"]["input_tokens"]
+
+        logger.warning("Unexpected response format from Anthropic token count API: %s", data)
+    except Exception as exc:
+        logger.warning("Failed to retrieve Anthropic token count via HTTP: %s", exc)
+
+    return None
+
+
 def extract_text_content(content: str | list[dict[str, str]] | dict[str, Any]) -> str:
     """Extract text content from various input formats."""
     if isinstance(content, str):

{gac-1.0.0.dist-info → gac-1.0.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gac
-Version: 1.0.0
+Version: 1.0.1
 Summary: AI-powered Git commit message generator with multi-provider support
 Project-URL: Homepage, https://github.com/cellwebb/gac
 Project-URL: Documentation, https://github.com/cellwebb/gac#readme

{gac-1.0.0.dist-info → gac-1.0.1.dist-info}/RECORD

@@ -1,6 +1,6 @@
 gac/__init__.py,sha256=T3KAW47ZmvB5AozG_uL92ryBYgp-2LNEztBaxaY3dJE,674
-gac/__version__.py,sha256=dFDGI6kSyoMIM0lfv_qWhBcmDf6A-bxqO14NKF-1bbg,66
-gac/ai.py,sha256=7rqXXXXNZiQe1vsPNoFU-jnLXfbnTTmnGKUYBZXpCFI,4928
+gac/__version__.py,sha256=8Yc4NMnplKKb2pLEFCgKeKRVX9Q-31V40n444l7Ao5M,66
+gac/ai.py,sha256=E0vfkWyqYvNJFAbOYvHG-tnwcQakAYm34oHMbKM5GIk,6267
 gac/ai_providers.py,sha256=QiVSspn0cauxl7m1Chn6nw1kAO1ByAuPiQqZWyZZCys,13210
 gac/cli.py,sha256=eQS8S7v6p0CfN9wtr239ujYGTi9rKl-KV7STX2U-C3w,4581
 gac/config.py,sha256=wSgEDjtis7Vk1pv5VPvYmJyD9-tymDS6GiUHjnCMbIM,1486
@@ -14,8 +14,8 @@ gac/main.py,sha256=WI7mxIbL05neQr1VfoopOeZKIonwpwFeZCt_4VFewPY,11987
 gac/preprocess.py,sha256=4igtZ9OTHgTpqwlJmbcGaqzmdD0HHCZJwsZ9eG118Gk,15360
 gac/prompt.py,sha256=_fv24XU3DZE_S72vcdUYnNkmy-_KXnr1Vlc-9okop7E,17263
 gac/utils.py,sha256=W3ladtmsH01MNLdckQYTzYrYbTGEdzCKI36he9C-y_E,3945
-gac-1.0.0.dist-info/METADATA,sha256=OOLz0xmLbHsSAgfKa0830-zIGhwT_9VMqlI7b72cV6Y,8351
-gac-1.0.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-gac-1.0.0.dist-info/entry_points.txt,sha256=tdjN-XMmcWfL92swuRAjT62bFLOAwk9bTMRLGP5Z4aI,36
-gac-1.0.0.dist-info/licenses/LICENSE,sha256=vOab37NouL1PNs5BswnPayrMCqaN2sqLfMQfqPDrpZg,1103
-gac-1.0.0.dist-info/RECORD,,
+gac-1.0.1.dist-info/METADATA,sha256=iEfp3b1Mx12iwon4aD6QDE8foNYrn-dN9yCmwhuKQTU,8351
+gac-1.0.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+gac-1.0.1.dist-info/entry_points.txt,sha256=tdjN-XMmcWfL92swuRAjT62bFLOAwk9bTMRLGP5Z4aI,36
+gac-1.0.1.dist-info/licenses/LICENSE,sha256=vOab37NouL1PNs5BswnPayrMCqaN2sqLfMQfqPDrpZg,1103
+gac-1.0.1.dist-info/RECORD,,