vectorvein 0.2.9__tar.gz → 0.2.10__tar.gz

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

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

{vectorvein-0.2.9 → 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.9"
+version = "0.2.10"
 
 [project.license]
 text = "MIT"

{vectorvein-0.2.9 → 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)}")