vectorvein 0.2.8__tar.gz → 0.2.10__tar.gz

{vectorvein-0.2.8 → vectorvein-0.2.10}/PKG-INFO

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

{vectorvein-0.2.8 → vectorvein-0.2.10}/pyproject.toml

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

{vectorvein-0.2.8 → vectorvein-0.2.10}/src/vectorvein/chat_clients/base_client.py

@@ -411,6 +411,15 @@ class BaseChatClient(ABC):
             timeout=timeout,
         )
 
+    def model_list(self):
+        _raw_client = self.raw_client
+        if isinstance(_raw_client, (OpenAI, AzureOpenAI)):
+            return _raw_client.models.list().model_dump()
+        elif isinstance(_raw_client, Anthropic):
+            return _raw_client.models.list(limit=1000).model_dump()
+        else:
+            raise ValueError(f"Unsupported client type: {type(_raw_client)}")
+
 
 class BaseAsyncChatClient(ABC):
     DEFAULT_MODEL: str
@@ -788,3 +797,12 @@ class BaseAsyncChatClient(ABC):
             extra_body=extra_body,
             timeout=timeout,
         )
+
+    async def model_list(self):
+        _raw_client = self.raw_client
+        if isinstance(_raw_client, (AsyncOpenAI, AsyncAzureOpenAI)):
+            return (await _raw_client.models.list()).model_dump()
+        elif isinstance(_raw_client, AsyncAnthropic):
+            return (await _raw_client.models.list(limit=1000)).model_dump()
+        else:
+            raise ValueError(f"Unsupported client type: {type(_raw_client)}")

{vectorvein-0.2.8 → vectorvein-0.2.10}/src/vectorvein/types/llm_parameters.py

@@ -8,10 +8,12 @@ from pydantic import BaseModel, Field
 from anthropic._types import NotGiven as AnthropicNotGiven
 from anthropic._types import NOT_GIVEN as ANTHROPIC_NOT_GIVEN
 from anthropic.types import ToolParam as AnthropicToolParam
+from anthropic.types import ThinkingConfigParam, ThinkingConfigEnabledParam
 from anthropic.types.message_create_params import ToolChoice as AnthropicToolChoice
 
 from openai._types import NotGiven as OpenAINotGiven
 from openai._types import NOT_GIVEN as OPENAI_NOT_GIVEN
+from openai.types.chat.completion_create_params import ResponseFormat
 from openai.types.chat.chat_completion_chunk import ChoiceDeltaToolCall
 from openai.types.chat.chat_completion_tool_param import ChatCompletionToolParam
 from openai.types.completion_usage import CompletionTokensDetails, PromptTokensDetails
@@ -153,4 +155,7 @@ __all__ = [
     "AnthropicToolChoice",
     "OPENAI_NOT_GIVEN",
     "ANTHROPIC_NOT_GIVEN",
+    "ResponseFormat",
+    "ThinkingConfigParam",
+    "ThinkingConfigEnabledParam",
 ]