vectorvein 0.1.19__tar.gz → 0.1.21__tar.gz

{vectorvein-0.1.19 → vectorvein-0.1.21}/PKG-INFO

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

{vectorvein-0.1.19 → vectorvein-0.1.21}/pyproject.toml

@@ -16,7 +16,7 @@ description = "Default template for PDM package"
 name = "vectorvein"
 readme = "README.md"
 requires-python = ">=3.10"
-version = "0.1.19"
+version = "0.1.21"
 
 [project.license]
 text = "MIT"

{vectorvein-0.1.19 → vectorvein-0.1.21}/src/vectorvein/chat_clients/gemini_client.py

@@ -39,6 +39,10 @@ class GeminiChatClient(BaseChatClient):
             **kwargs,
         )
 
+    @property
+    def raw_client(self):
+        return self.http_client
+
     def create_completion(
         self,
         messages: list = list,
@@ -48,6 +52,7 @@ class GeminiChatClient(BaseChatClient):
         max_tokens: int | None = None,
         tools: list | None = None,
         tool_choice: str | None = None,
+        **kwargs,
     ):
         if model is not None:
             self.model = model
@@ -95,6 +100,7 @@ class GeminiChatClient(BaseChatClient):
                 "maxOutputTokens": max_tokens,
             },
             **tools_params,
+            **kwargs,
         }
         if system_prompt:
             request_body["systemInstruction"] = {"parts": [{"text": system_prompt}]}
@@ -209,6 +215,10 @@ class AsyncGeminiChatClient(BaseAsyncChatClient):
             **kwargs,
         )
 
+    @property
+    def raw_client(self):
+        return self.http_client
+
     async def create_completion(
         self,
         messages: list = list,
@@ -218,6 +228,7 @@ class AsyncGeminiChatClient(BaseAsyncChatClient):
         max_tokens: int | None = None,
         tools: list | None = None,
         tool_choice: str | None = None,
+        **kwargs,
     ):
         if model is not None:
             self.model = model
@@ -265,6 +276,7 @@ class AsyncGeminiChatClient(BaseAsyncChatClient):
                 "maxOutputTokens": max_tokens,
             },
             **tools_params,
+            **kwargs,
         }
         if system_prompt:
             request_body["systemInstruction"] = {"parts": [{"text": system_prompt}]}

{vectorvein-0.1.19 → vectorvein-0.1.21}/src/vectorvein/chat_clients/minimax_client.py

@@ -66,6 +66,10 @@ class MiniMaxChatClient(BaseChatClient):
         else:
             self.http_client = httpx.Client()
 
+    @property
+    def raw_client(self):
+        return self.http_client
+
     def create_completion(
         self,
         messages: list = list,
@@ -231,6 +235,10 @@ class AsyncMiniMaxChatClient(BaseAsyncChatClient):
         else:
             self.http_client = httpx.AsyncClient()
 
+    @property
+    def raw_client(self):
+        return self.http_client
+
     async def create_completion(
         self,
         messages: list = list,