vectorvein 0.1.55__py3-none-any.whl → 0.1.56__py3-none-any.whl

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/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.55.dist-info → vectorvein-0.1.56.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vectorvein
-Version: 0.1.55
+Version: 0.1.56
 Summary: VectorVein python SDK
 Author-Email: Anderson <andersonby@163.com>
 License: MIT

{vectorvein-0.1.55.dist-info → vectorvein-0.1.56.dist-info}/RECORD

@@ -1,13 +1,13 @@
-vectorvein-0.1.55.dist-info/METADATA,sha256=HKXCRg6-WfX1Um_Dx9GKaiIy97uA0yDiyePzdyTYicw,633
-vectorvein-0.1.55.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
-vectorvein-0.1.55.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+vectorvein-0.1.56.dist-info/METADATA,sha256=3YTghMyf-ZK9VzwhaqGwf_xuhKmVI5IA7SKI1O73Sww,633
+vectorvein-0.1.56.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+vectorvein-0.1.56.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
 vectorvein/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vectorvein/chat_clients/__init__.py,sha256=Oev7Lv1DIEWCMD-2Pm7e2cwzX7JFQTnIK-j6o4iUuyQ,17725
 vectorvein/chat_clients/anthropic_client.py,sha256=G68JGM98E0pYyi8Tjvo4VQtnug9ncugFbb4d0DrPVQo,36122
 vectorvein/chat_clients/baichuan_client.py,sha256=CVMvpgjdrZGv0BWnTOBD-f2ufZ3wq3496wqukumsAr4,526
 vectorvein/chat_clients/base_client.py,sha256=0Uj0e-JR0a68sRS_WfUMVd91Av7lzJh6-DukjutlaD0,9497
 vectorvein/chat_clients/deepseek_client.py,sha256=3qWu01NlJAP2N-Ff62d5-CZXZitlizE1fzb20LNetig,526
-vectorvein/chat_clients/gemini_client.py,sha256=TJZUP8dbufuDnqkRI_gxIDDCYHrIhMPCqKGI6sSwqZo,19860
+vectorvein/chat_clients/gemini_client.py,sha256=DlQI5p8qG5erThMb17ojB2gofzTkwVy3veFC8_Cbpc4,19902
 vectorvein/chat_clients/groq_client.py,sha256=Uow4pgdmFi93ZQSoOol2-0PhhqkW-S0XuSldvppz5U4,498
 vectorvein/chat_clients/local_client.py,sha256=55nOsxzqUf79q3Y14MKROA71zxhsT7p7FsDZ89rts2M,422
 vectorvein/chat_clients/minimax_client.py,sha256=3HetFZbmgvEzWFaH_Gbj99lsh9DmIOhfpVdbV4HxrSI,19116
@@ -18,7 +18,7 @@ vectorvein/chat_clients/openai_compatible_client.py,sha256=D2VmhpDVct4w2y58s87An
 vectorvein/chat_clients/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vectorvein/chat_clients/qwen_client.py,sha256=-ryh-m9PgsO0fc4ulcCmPTy1155J8YUy15uPoJQOHA0,513
 vectorvein/chat_clients/stepfun_client.py,sha256=zsD2W5ahmR4DD9cqQTXmJr3txrGuvxbRWhFlRdwNijI,519
-vectorvein/chat_clients/utils.py,sha256=J-lv-XNlpFwcvLREqVZWKOA4QkN1t9XsOU6h0kxkWO4,26553
+vectorvein/chat_clients/utils.py,sha256=UqDV4ljwZRbndZdkyHkIBis1ciRt5gjPFOJ0vnR2Uas,26669
 vectorvein/chat_clients/xai_client.py,sha256=eLFJJrNRJ-ni3DpshODcr3S1EJQLbhVwxyO1E54LaqM,491
 vectorvein/chat_clients/yi_client.py,sha256=RNf4CRuPJfixrwLZ3-DEc3t25QDe1mvZeb9sku2f8Bc,484
 vectorvein/chat_clients/zhipuai_client.py,sha256=Ys5DSeLCuedaDXr3PfG1EW2zKXopt-awO2IylWSwY0s,519
@@ -33,4 +33,4 @@ vectorvein/types/llm_parameters.py,sha256=5o-C_yXxxQWZy_e8OWowB2107GTS-Eawx4Mvb1
 vectorvein/types/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vectorvein/utilities/media_processing.py,sha256=CTRq-lGlFkFgP_FSRhNwF_qUgmOrXPf2_1Ok9HY42_g,5887
 vectorvein/utilities/retry.py,sha256=6KFS9R2HdhqM3_9jkjD4F36ZSpEx2YNFGOVlpOsUetM,2208
-vectorvein-0.1.55.dist-info/RECORD,,
+vectorvein-0.1.56.dist-info/RECORD,,