vectorvein 0.1.57__tar.gz → 0.1.58__tar.gz

{vectorvein-0.1.57 → vectorvein-0.1.58}/PKG-INFO

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

{vectorvein-0.1.57 → vectorvein-0.1.58}/pyproject.toml

@@ -17,7 +17,7 @@ description = "VectorVein python SDK"
 name = "vectorvein"
 readme = "README.md"
 requires-python = ">=3.10"
-version = "0.1.57"
+version = "0.1.58"
 
 [project.license]
 text = "MIT"

{vectorvein-0.1.57 → vectorvein-0.1.58}/src/vectorvein/chat_clients/gemini_client.py

@@ -197,7 +197,7 @@ class GeminiChatClient(BaseChatClient):
                 if self.http_client:
                     client = self.http_client
                 else:
-                    client = httpx.Client()
+                    client = httpx.Client(timeout=300)
                 with client.stream("POST", url, headers=headers, params=params, json=request_body) as response:
                     for chunk in response.iter_lines():
                         message = {"content": "", "tool_calls": []}
@@ -236,7 +236,7 @@ class GeminiChatClient(BaseChatClient):
             if self.http_client:
                 client = self.http_client
             else:
-                client = httpx.Client()
+                client = httpx.Client(timeout=300)
             response = client.post(url, json=request_body, headers=headers, params=params, timeout=None).json()
             if "error" in response:
                 raise Exception(response["error"])
@@ -444,7 +444,7 @@ class AsyncGeminiChatClient(BaseAsyncChatClient):
                 if self.http_client:
                     client = self.http_client
                 else:
-                    client = httpx.AsyncClient()
+                    client = httpx.AsyncClient(timeout=300)
                 async with client.stream("POST", url, headers=headers, params=params, json=request_body) as response:
                     async for chunk in response.aiter_lines():
                         message = {"content": "", "tool_calls": []}
@@ -483,7 +483,7 @@ class AsyncGeminiChatClient(BaseAsyncChatClient):
             if self.http_client:
                 client = self.http_client
             else:
-                client = httpx.AsyncClient()
+                client = httpx.AsyncClient(timeout=300)
             async with client:
                 response = await client.post(url, json=request_body, headers=headers, params=params, timeout=None)
                 response = response.json()