vectorvein 0.1.54__tar.gz → 0.1.56__tar.gz

{vectorvein-0.1.54 → vectorvein-0.1.56}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.1
 Name: vectorvein
-Version: 0.1.54
-Summary: Default template for PDM package
+Version: 0.1.56
+Summary: VectorVein python SDK
 Author-Email: Anderson <andersonby@163.com>
 License: MIT
 Requires-Python: >=3.10

{vectorvein-0.1.54 → vectorvein-0.1.56}/pyproject.toml

@@ -13,11 +13,11 @@ dependencies = [
     "qwen-tokenizer>=0.2.0",
     "google-auth>=2.35.0",
 ]
-description = "Default template for PDM package"
+description = "VectorVein python SDK"
 name = "vectorvein"
 readme = "README.md"
 requires-python = ">=3.10"
-version = "0.1.54"
+version = "0.1.56"
 
 [project.license]
 text = "MIT"

{vectorvein-0.1.54 → vectorvein-0.1.56}/src/vectorvein/chat_clients/gemini_client.py

@@ -216,9 +216,9 @@ class GeminiChatClient(BaseChatClient):
                             ]
 
                         result["usage"] = message["usage"] = {
-                            "prompt_tokens": data["usageMetadata"]["promptTokenCount"],
-                            "completion_tokens": data["usageMetadata"]["candidatesTokenCount"],
-                            "total_tokens": data["usageMetadata"]["totalTokenCount"],
+                            "prompt_tokens": data["usageMetadata"].get("promptTokenCount", 0),
+                            "completion_tokens": data["usageMetadata"].get("candidatesTokenCount", 0),
+                            "total_tokens": data["usageMetadata"].get("totalTokenCount", 0),
                         }
                         yield ChatCompletionDeltaMessage(**message)
 
@@ -453,9 +453,9 @@ class AsyncGeminiChatClient(BaseAsyncChatClient):
                             ]
 
                         result["usage"] = message["usage"] = {
-                            "prompt_tokens": data["usageMetadata"]["promptTokenCount"],
-                            "completion_tokens": data["usageMetadata"]["candidatesTokenCount"],
-                            "total_tokens": data["usageMetadata"]["totalTokenCount"],
+                            "prompt_tokens": data["usageMetadata"].get("promptTokenCount", 0),
+                            "completion_tokens": data["usageMetadata"].get("candidatesTokenCount", 0),
+                            "total_tokens": data["usageMetadata"].get("totalTokenCount", 0),
                         }
                         yield ChatCompletionDeltaMessage(**message)
 

{vectorvein-0.1.54 → vectorvein-0.1.56}/src/vectorvein/chat_clients/utils.py

@@ -130,10 +130,12 @@ def get_token_counts(text: str | dict, model: str = "", use_token_server_first:
             .sleep_time(1)
             .run()
         )
-        if response is None:
-            return 1000
-        result = response.json()
-        return result["total_tokens"]
+        if response is not None:
+            try:
+                result = response.json()
+                return result["total_tokens"]
+            except Exception as e:
+                print(f"Failed to parse token count response: {e}")
 
     if not isinstance(text, str):
         text = str(text)

{vectorvein-0.1.54 → vectorvein-0.1.56}/src/vectorvein/types/defaults.py

@@ -558,7 +558,7 @@ ANTHROPIC_MODELS: Final[Dict[str, Dict[str, Any]]] = {
         "max_output_tokens": 8192,
         "function_call_available": True,
         "response_format_available": False,
-        "native_multimodal": True,
+        "native_multimodal": False,
     },
     "claude-3-5-sonnet-20240620": {
         "id": "claude-3-5-sonnet-20240620",