vectorvein 0.1.26__tar.gz → 0.1.27__tar.gz

{vectorvein-0.1.26 → vectorvein-0.1.27}/PKG-INFO

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

{vectorvein-0.1.26 → vectorvein-0.1.27}/pyproject.toml

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

{vectorvein-0.1.26 → vectorvein-0.1.27}/src/vectorvein/chat_clients/base_client.py

@@ -88,6 +88,22 @@ class BaseChatClient(ABC):
     ) -> Generator[ChatCompletionDeltaMessage, Any, None]:
         pass
 
+    @overload
+    @abstractmethod
+    def create_completion(
+        self,
+        messages: list,
+        model: str | None = None,
+        stream: bool = False,
+        temperature: float = 0.7,
+        max_tokens: int | None = None,
+        tools: Iterable[ToolParam] | NotGiven = NOT_GIVEN,
+        tool_choice: ToolChoice | NotGiven = NOT_GIVEN,
+        response_format: dict | None = None,
+        **kwargs,
+    ) -> ChatCompletionMessage | Generator[ChatCompletionDeltaMessage, Any, None]:
+        pass
+
     @abstractmethod
     def create_completion(
         self,
@@ -196,6 +212,22 @@ class BaseAsyncChatClient(ABC):
     ) -> AsyncGenerator[ChatCompletionDeltaMessage, None]:
         pass
 
+    @overload
+    @abstractmethod
+    async def create_completion(
+        self,
+        messages: list,
+        model: str | None = None,
+        stream: bool = False,
+        temperature: float = 0.7,
+        max_tokens: int | None = None,
+        tools: list | NotGiven = NOT_GIVEN,
+        tool_choice: ToolChoice | NotGiven = NOT_GIVEN,
+        response_format: dict | None = None,
+        **kwargs,
+    ) -> ChatCompletionMessage | AsyncGenerator[ChatCompletionDeltaMessage, None]:
+        pass
+
     @abstractmethod
     async def create_completion(
         self,

{vectorvein-0.1.26 → vectorvein-0.1.27}/src/vectorvein/chat_clients/gemini_client.py

@@ -82,6 +82,21 @@ class GeminiChatClient(BaseChatClient):
     ) -> Generator[ChatCompletionDeltaMessage, None, None]:
         pass
 
+    @overload
+    def create_completion(
+        self,
+        messages: list,
+        model: str | None = None,
+        stream: bool | None = None,
+        temperature: float | None = None,
+        max_tokens: int | None = None,
+        tools: Iterable[ToolParam] | NotGiven = NOT_GIVEN,
+        tool_choice: ToolChoice | NotGiven = NOT_GIVEN,
+        response_format: dict | None = None,
+        **kwargs,
+    ) -> ChatCompletionMessage | Generator[ChatCompletionDeltaMessage, Any, None]:
+        pass
+
     def create_completion(
         self,
         messages: list,
@@ -295,6 +310,21 @@ class AsyncGeminiChatClient(BaseAsyncChatClient):
     ) -> AsyncGenerator[ChatCompletionDeltaMessage, Any]:
         pass
 
+    @overload
+    async def create_completion(
+        self,
+        messages: list,
+        model: str | None = None,
+        stream: bool | None = None,
+        temperature: float | None = None,
+        max_tokens: int | None = None,
+        tools: Iterable[ToolParam] | NotGiven = NOT_GIVEN,
+        tool_choice: ToolChoice | NotGiven = NOT_GIVEN,
+        response_format: dict | None = None,
+        **kwargs,
+    ) -> ChatCompletionMessage | AsyncGenerator[ChatCompletionDeltaMessage, Any]:
+        pass
+
     async def create_completion(
         self,
         messages: list,

{vectorvein-0.1.26 → vectorvein-0.1.27}/src/vectorvein/chat_clients/minimax_client.py

@@ -107,6 +107,21 @@ class MiniMaxChatClient(BaseChatClient):
     ) -> Generator[ChatCompletionDeltaMessage, None, None]:
         pass
 
+    @overload
+    def create_completion(
+        self,
+        messages: list,
+        model: str | None = None,
+        stream: bool | None = None,
+        temperature: float | None = None,
+        max_tokens: int | None = None,
+        tools: Iterable[ToolParam] | NotGiven = NOT_GIVEN,
+        tool_choice: ToolChoice | NotGiven = NOT_GIVEN,
+        response_format: dict | None = None,
+        **kwargs,
+    ) -> ChatCompletionMessage | Generator[ChatCompletionDeltaMessage, Any, None]:
+        pass
+
     def create_completion(
         self,
         messages: list,
@@ -307,6 +322,21 @@ class AsyncMiniMaxChatClient(BaseAsyncChatClient):
     ) -> AsyncGenerator[ChatCompletionDeltaMessage, Any]:
         pass
 
+    @overload
+    async def create_completion(
+        self,
+        messages: list,
+        model: str | None = None,
+        stream: bool | None = None,
+        temperature: float | None = None,
+        max_tokens: int | None = None,
+        tools: Iterable[ToolParam] | NotGiven = NOT_GIVEN,
+        tool_choice: ToolChoice | NotGiven = NOT_GIVEN,
+        response_format: dict | None = None,
+        **kwargs,
+    ) -> ChatCompletionMessage | AsyncGenerator[ChatCompletionDeltaMessage, Any]:
+        pass
+
     async def create_completion(
         self,
         messages: list,

{vectorvein-0.1.26 → vectorvein-0.1.27}/src/vectorvein/chat_clients/openai_compatible_client.py

@@ -107,6 +107,21 @@ class OpenAICompatibleChatClient(BaseChatClient):
     ) -> Generator[ChatCompletionDeltaMessage, None, None]:
         pass
 
+    @overload
+    def create_completion(
+        self,
+        messages: list,
+        model: str | None = None,
+        stream: bool | None = None,
+        temperature: float | None = None,
+        max_tokens: int | None = None,
+        tools: Iterable[ToolParam] | NotGiven = NOT_GIVEN,
+        tool_choice: ToolChoice | NotGiven = NOT_GIVEN,
+        response_format: dict | None = None,
+        **kwargs,
+    ) -> ChatCompletionMessage | Generator[ChatCompletionDeltaMessage, Any, None]:
+        pass
+
     def create_completion(
         self,
         messages: list,
@@ -314,6 +329,21 @@ class AsyncOpenAICompatibleChatClient(BaseAsyncChatClient):
     ) -> AsyncGenerator[ChatCompletionDeltaMessage, Any]:
         pass
 
+    @overload
+    async def create_completion(
+        self,
+        messages: list,
+        model: str | None = None,
+        stream: bool | None = None,
+        temperature: float | None = None,
+        max_tokens: int | None = None,
+        tools: Iterable[ToolParam] | NotGiven = NOT_GIVEN,
+        tool_choice: ToolChoice | NotGiven = NOT_GIVEN,
+        response_format: dict | None = None,
+        **kwargs,
+    ) -> ChatCompletionMessage | AsyncGenerator[ChatCompletionDeltaMessage, Any]:
+        pass
+
     async def create_completion(
         self,
         messages: list,