gac 0.17.5__tar.gz → 0.17.7__tar.gz

{gac-0.17.5 → gac-0.17.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gac
-Version: 0.17.5
+Version: 0.17.7
 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-0.17.5 → gac-0.17.7}/src/gac/__version__.py

@@ -1,3 +1,3 @@
 """Version information for gac package."""
 
-__version__ = "0.17.5"
+__version__ = "0.17.7"

{gac-0.17.5 → gac-0.17.7}/src/gac/ai.py

@@ -28,7 +28,19 @@ def count_tokens(content: str | list[dict[str, str]] | dict[str, Any], model: st
     if model.startswith("anthropic"):
         import anthropic
 
-        return anthropic.Client().count_tokens(text)
+        try:
+            client = anthropic.Anthropic()
+            # Extract the actual model name after the provider prefix
+            model_name = model.split(":", 1)[1] if ":" in model else "claude-3-5-haiku-latest"
+
+            # Use the beta messages.count_tokens API for accurate counting
+            response = client.messages.count_tokens(
+                model=model_name, messages=[{"role": "user", "content": text}], betas=["token-counting-2024-11-01"]
+            )
+            return response.input_tokens
+        except Exception:
+            # Fallback to simple estimation for Anthropic models
+            return len(text) // 4
 
     try:
         encoding = get_encoding(model)