gac 0.17.6__tar.gz → 0.18.0__tar.gz

{gac-0.17.6 → gac-0.18.0}/.gitignore

@@ -97,7 +97,6 @@ venv.bak/
 
 # Package management
 poetry.toml
-uv.lock
 
 # =========================
 # Python tools and linters

{gac-0.17.6 → gac-0.18.0}/PKG-INFO

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

@@ -1,3 +1,3 @@
 """Version information for gac package."""
 
-__version__ = "0.17.6"
+__version__ = "0.18.0"

{gac-0.17.6 → gac-0.18.0}/src/gac/ai.py

@@ -28,10 +28,17 @@ def count_tokens(content: str | list[dict[str, str]] | dict[str, Any], model: st
     if model.startswith("anthropic"):
         import anthropic
 
-        client = anthropic.Anthropic()
-        # Use the simple count_tokens method
-        # Note: This is only a rough estimate for newer models
-        return client.count_tokens(text)
+        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
 
     try:
         encoding = get_encoding(model)