vectorvein 0.2.3__py3-none-any.whl → 0.2.4__py3-none-any.whl

vectorvein/api/client.py

@@ -99,7 +99,6 @@ class VectorVeinClient:
                 headers=headers,
                 **kwargs,
             )
-            response.raise_for_status()
             result = response.json()
 
             if result["status"] in [401, 403]:
@@ -191,6 +190,16 @@ class VectorVeinClient:
 
             time.sleep(5)
 
+    @overload
+    def check_workflow_status(
+        self, rid: str, wid: Optional[str] = None, api_key_type: Literal["WORKFLOW"] = "WORKFLOW"
+    ) -> WorkflowRunResult: ...
+
+    @overload
+    def check_workflow_status(
+        self, rid: str, wid: str, api_key_type: Literal["VAPP"] = "VAPP"
+    ) -> WorkflowRunResult: ...
+
     def check_workflow_status(
         self, rid: str, wid: Optional[str] = None, api_key_type: Literal["WORKFLOW", "VAPP"] = "WORKFLOW"
     ) -> WorkflowRunResult:
@@ -198,13 +207,18 @@ class VectorVeinClient:
 
         Args:
             rid: 工作流运行记录ID
-            wid: 工作流ID，非必填
+            wid: 工作流ID，非必填，api_key_type 为 'VAPP' 时必填
             api_key_type: 密钥类型，可选值：'WORKFLOW' 或 'VAPP'
 
         Returns:
             WorkflowRunResult: 工作流运行结果
+
+        Raises:
+            VectorVeinAPIError: 工作流错误
         """
         payload = {"rid": rid}
+        if api_key_type == "VAPP" and not wid:
+            raise VectorVeinAPIError("api_key_type 为 'VAPP' 时工作流 ID 不能为空")
         if wid:
             payload["wid"] = wid
         response = self._request("POST", "workflow/check-status", json=payload, api_key_type=api_key_type)
@@ -526,7 +540,6 @@ class AsyncVectorVeinClient:
                 headers=headers,
                 **kwargs,
             )
-            response.raise_for_status()
             result = response.json()
 
             if result["status"] in [401, 403]:
@@ -618,6 +631,16 @@ class AsyncVectorVeinClient:
 
             await asyncio.sleep(5)
 
+    @overload
+    async def check_workflow_status(
+        self, rid: str, wid: Optional[str] = None, api_key_type: Literal["WORKFLOW"] = "WORKFLOW"
+    ) -> WorkflowRunResult: ...
+
+    @overload
+    async def check_workflow_status(
+        self, rid: str, wid: str, api_key_type: Literal["VAPP"] = "VAPP"
+    ) -> WorkflowRunResult: ...
+
     async def check_workflow_status(
         self, rid: str, wid: Optional[str] = None, api_key_type: Literal["WORKFLOW", "VAPP"] = "WORKFLOW"
     ) -> WorkflowRunResult:
@@ -625,13 +648,15 @@ class AsyncVectorVeinClient:
 
         Args:
             rid: 工作流运行记录ID
-            wid: 工作流ID，非必填
+            wid: 工作流ID，非必填，api_key_type 为 'VAPP' 时必填
             api_key_type: 密钥类型，可选值：'WORKFLOW' 或 'VAPP'
 
-        Returns:
-            WorkflowRunResult: 工作流运行结果
+        Raises:
+            VectorVeinAPIError: 工作流错误
         """
         payload = {"rid": rid}
+        if api_key_type == "VAPP" and not wid:
+            raise VectorVeinAPIError("api_key_type 为 'VAPP' 时工作流 ID 不能为空")
         if wid:
             payload["wid"] = wid
         response = await self._request("POST", "workflow/check-status", json=payload, api_key_type=api_key_type)

{vectorvein-0.2.3.dist-info → vectorvein-0.2.4.dist-info}/METADATA

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

{vectorvein-0.2.3.dist-info → vectorvein-0.2.4.dist-info}/RECORD

@@ -1,9 +1,9 @@
-vectorvein-0.2.3.dist-info/METADATA,sha256=c-trHoS2s_Hz1zCnciGNk6AdPwlNWZp0PBxzNaRCdhg,4413
-vectorvein-0.2.3.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
-vectorvein-0.2.3.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+vectorvein-0.2.4.dist-info/METADATA,sha256=TFMh4mGtTFrgaRTpXSj5wa9HMcVJ2j74Z1AwTockagI,4413
+vectorvein-0.2.4.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+vectorvein-0.2.4.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=C0TvmjzvZjYz9IpvDwGx5nMQzR9GyTJEU7aav4epeBc,31710
+vectorvein/api/client.py,sha256=K93a-EvmBqh6sz_iFkMK1VlM54mgZsjOabnRzJJnbWM,32732
 vectorvein/api/exceptions.py,sha256=btfeXfNfc7zLykMKklpJePLnmJie5YSxCYHyMReCC9s,751
 vectorvein/api/models.py,sha256=z5MeXMxWFHlNkP5vjVz6gEn5cxD1FbQ8pQvbx9KtgkE,1422
 vectorvein/chat_clients/__init__.py,sha256=omQuG4PRRPNflSAgtdU--rwsWG6vMpwMEyIGZyFVHVQ,18596
@@ -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.3.dist-info/RECORD,,
+vectorvein-0.2.4.dist-info/RECORD,,