vectorvein 0.1.12__tar.gz → 0.1.13__tar.gz

{vectorvein-0.1.12 → vectorvein-0.1.13}/PKG-INFO

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

{vectorvein-0.1.12 → vectorvein-0.1.13}/pyproject.toml

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

{vectorvein-0.1.12 → vectorvein-0.1.13}/src/vectorvein/chat_clients/base_client.py

@@ -52,6 +52,7 @@ class BaseChatClient(ABC):
         max_tokens: int | None = None,
         tools: list | NotGiven = NOT_GIVEN,
         tool_choice: str | NotGiven = NOT_GIVEN,
+        **kwargs,
     ) -> ChatCompletionMessage | Generator[ChatCompletionDeltaMessage, Any, None]:
         pass
 
@@ -63,6 +64,7 @@ class BaseChatClient(ABC):
         max_tokens: int | None = None,
         tools: list | NotGiven = NOT_GIVEN,
         tool_choice: str | NotGiven = NOT_GIVEN,
+        **kwargs,
     ) -> Generator[ChatCompletionDeltaMessage, Any, None]:
         return self.create_completion(
             messages=messages,
@@ -72,6 +74,7 @@ class BaseChatClient(ABC):
             max_tokens=max_tokens,
             tools=tools,
             tool_choice=tool_choice,
+            **kwargs,
         )
 
 
@@ -115,6 +118,7 @@ class BaseAsyncChatClient(ABC):
         max_tokens: int | None = None,
         tools: list | NotGiven = NOT_GIVEN,
         tool_choice: str | NotGiven = NOT_GIVEN,
+        **kwargs,
     ) -> ChatCompletionMessage | AsyncGenerator[ChatCompletionDeltaMessage, None]:
         pass
 
@@ -126,6 +130,7 @@ class BaseAsyncChatClient(ABC):
         max_tokens: int | None = None,
         tools: list | NotGiven = NOT_GIVEN,
         tool_choice: str | NotGiven = NOT_GIVEN,
+        **kwargs,
     ) -> AsyncGenerator[ChatCompletionDeltaMessage, None]:
         return await self.create_completion(
             messages=messages,
@@ -135,4 +140,5 @@ class BaseAsyncChatClient(ABC):
             max_tokens=max_tokens,
             tools=tools,
             tool_choice=tool_choice,
+            **kwargs,
         )