vectorvein 0.2.9__py3-none-any.whl → 0.2.10__py3-none-any.whl

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.9.dist-info → vectorvein-0.2.10.dist-info}/METADATA

@@ -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.dist-info → vectorvein-0.2.10.dist-info}/RECORD

@@ -1,6 +1,6 @@
-vectorvein-0.2.9.dist-info/METADATA,sha256=0-oCeh7DA-fSPf3gKR0i79tcp_qLi5iOf1w4-mW9WaU,4413
-vectorvein-0.2.9.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
-vectorvein-0.2.9.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+vectorvein-0.2.10.dist-info/METADATA,sha256=Drs6uKwNLAN9Gh-FmMJnOfEcoNBviytd2HSxJ_OFe6c,4414
+vectorvein-0.2.10.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+vectorvein-0.2.10.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
 vectorvein/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 vectorvein/api/__init__.py,sha256=lfY-XA46fgD2iIZTU0VYP8i07AwA03Egj4Qua0vUKrQ,738
 vectorvein/api/client.py,sha256=xF-leKDQzVyyy9FnIRaz0k4eElYW1XbbzeRLcpnyk90,33047
@@ -9,7 +9,7 @@ vectorvein/api/models.py,sha256=xtPWMsB0yIJI7i-gY4B6MtvXv0ZIXnoeKspmeInH6fU,1449
 vectorvein/chat_clients/__init__.py,sha256=1ElJPca32YqSxD4dWTHkTEWhiJE_N6J-e5ZGilpT7ks,19178
 vectorvein/chat_clients/anthropic_client.py,sha256=Yx0H2kW-XWOZId9Dnzke5eWWS8FqMf6OfgczZF7yeuM,58192
 vectorvein/chat_clients/baichuan_client.py,sha256=CVMvpgjdrZGv0BWnTOBD-f2ufZ3wq3496wqukumsAr4,526
-vectorvein/chat_clients/base_client.py,sha256=RcAJwwJ83qC73t3W7GoPNvQXb-qPJ7d8U1J5nwzcYls,37044
+vectorvein/chat_clients/base_client.py,sha256=35K_4kS9UnZA68utauh6GcV3IW-GutAMYr_jpb3eYt0,37863
 vectorvein/chat_clients/deepseek_client.py,sha256=3qWu01NlJAP2N-Ff62d5-CZXZitlizE1fzb20LNetig,526
 vectorvein/chat_clients/gemini_client.py,sha256=ufovIZrmAE3RLEe8h5WXombf7bARAZxnkj6ydNK2FQM,475
 vectorvein/chat_clients/groq_client.py,sha256=Uow4pgdmFi93ZQSoOol2-0PhhqkW-S0XuSldvppz5U4,498
@@ -59,4 +59,4 @@ vectorvein/workflow/nodes/vector_db.py,sha256=t6I17q6iR3yQreiDHpRrksMdWDPIvgqJs0
 vectorvein/workflow/nodes/video_generation.py,sha256=qmdg-t_idpxq1veukd-jv_ChICMOoInKxprV9Z4Vi2w,4118
 vectorvein/workflow/nodes/web_crawlers.py,sha256=LsqomfXfqrXfHJDO1cl0Ox48f4St7X_SL12DSbAMSOw,5415
 vectorvein/workflow/utils/json_to_code.py,sha256=F7dhDy8kGc8ndOeihGLRLGFGlquoxVlb02ENtxnQ0C8,5914
-vectorvein-0.2.9.dist-info/RECORD,,
+vectorvein-0.2.10.dist-info/RECORD,,