vectorvein 0.1.0__tar.gz → 0.1.1__tar.gz

{vectorvein-0.1.0 → vectorvein-0.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vectorvein
-Version: 0.1.0
+Version: 0.1.1
 Summary: Default template for PDM package
 Author-Email: Anderson <andersonby@163.com>
 License: MIT

{vectorvein-0.1.0 → vectorvein-0.1.1}/pyproject.toml

@@ -1,7 +1,4 @@
 [project]
-name = "vectorvein"
-version = "0.1.0"
-description = "Default template for PDM package"
 authors = [
     { name = "Anderson", email = "andersonby@163.com" },
 ]
@@ -13,17 +10,20 @@ dependencies = [
     "pydantic>=2.8.2",
     "Pillow>=10.4.0",
 ]
-requires-python = ">=3.8"
+description = "Default template for PDM package"
+name = "vectorvein"
 readme = "README.md"
+requires-python = ">=3.8"
+version = "0.1.1"
 
 [project.license]
 text = "MIT"
 
 [build-system]
+build-backend = "pdm.backend"
 requires = [
     "pdm-backend",
 ]
-build-backend = "pdm.backend"
 
 [tool.pdm]
 distribution = true

{vectorvein-0.1.0 → vectorvein-0.1.1}/src/vectorvein/chat_clients/openai_compatible_client.py

@@ -120,6 +120,8 @@ class OpenAICompatibleChatClient(BaseChatClient):
                 for chunk in response:
                     if len(chunk.choices) == 0:
                         continue
+                    if not chunk.choices[0].delta:
+                        continue
                     if self.model_setting.function_call_available:
                         yield chunk.choices[0].delta.model_dump()
                     else:
@@ -255,6 +257,8 @@ class AsyncOpenAICompatibleChatClient(BaseAsyncChatClient):
                 async for chunk in response:
                     if len(chunk.choices) == 0:
                         continue
+                    if not chunk.choices[0].delta:
+                        continue
                     if self.model_setting.function_call_available:
                         yield chunk.choices[0].delta.model_dump()
                     else:

{vectorvein-0.1.0 → vectorvein-0.1.1}/tests/test_create_chat_client.py

@@ -181,14 +181,16 @@ backend = BackendType.MiniMax
 model = "abab6.5s-chat"
 backend = BackendType.Gemini
 model = "gemini-1.5-flash"
+backend = BackendType.OpenAI
+model = "gpt-35-turbo"
 start_time = time.perf_counter()
 # test_sync(backend=backend, model=model, stream=False, use_tool=False)
-test_sync(backend=backend, model=model, stream=False, use_tool=True)
+# test_sync(backend=backend, model=model, stream=False, use_tool=True)
 # test_sync(backend=backend, model=model, stream=True, use_tool=False)
-test_sync(backend=backend, model=model, stream=True, use_tool=True)
+# test_sync(backend=backend, model=model, stream=True, use_tool=True)
 # asyncio.run(test_async(backend=backend, model=model, stream=False, use_tool=False))
-asyncio.run(test_async(backend=backend, model=model, stream=False, use_tool=True))
-# asyncio.run(test_async(backend=backend, model=model, stream=True, use_tool=False))
-asyncio.run(test_async(backend=backend, model=model, stream=True, use_tool=True))
+# asyncio.run(test_async(backend=backend, model=model, stream=False, use_tool=True))
+asyncio.run(test_async(backend=backend, model=model, stream=True, use_tool=False))
+# asyncio.run(test_async(backend=backend, model=model, stream=True, use_tool=True))
 end_time = time.perf_counter()
 print(f"Stream time elapsed: {end_time - start_time} seconds")