vectorvein 0.2.1__py3-none-any.whl → 0.2.2__py3-none-any.whl

vectorvein/api/client.py

@@ -166,30 +166,34 @@ class VectorVeinClient:
             if time.time() - start_time > timeout:
                 raise TimeoutError(f"Workflow execution timed out after {timeout} seconds")
 
-            status = self.check_workflow_status(rid)
-            if status["status"] == 200:
-                return WorkflowRunResult(
-                    rid=rid,
-                    status=status["status"],
-                    msg=status["msg"],
-                    data=[WorkflowOutput(**output) for output in status["data"]],
-                )
-            elif status["status"] == 500:
-                raise WorkflowError(f"Workflow execution failed: {status['msg']}")
+            result = self.check_workflow_status(rid)
+            if result.status == 200:
+                return result
+            elif result.status == 500:
+                raise WorkflowError(f"Workflow execution failed: {result.msg}")
 
             time.sleep(5)
 
-    def check_workflow_status(self, rid: str) -> Dict:
+    def check_workflow_status(self, rid: str) -> WorkflowRunResult:
         """检查工作流运行状态
 
         Args:
             rid: 工作流运行记录ID
 
         Returns:
-            Dict: 工作流状态信息
+            WorkflowRunResult: 工作流运行结果
         """
         payload = {"rid": rid}
-        return self._request("POST", "workflow/check-status", json=payload)
+        response = self._request("POST", "workflow/check-status", json=payload)
+        if response["status"] in [200, 202]:
+            return WorkflowRunResult(
+                rid=rid,
+                status=response["status"],
+                msg=response["msg"],
+                data=[WorkflowOutput(**output) for output in response["data"]],
+            )
+        else:
+            raise WorkflowError(f"Workflow execution failed: {response['msg']}")
 
     def get_access_keys(
         self, access_keys: Optional[List[str]] = None, get_type: Literal["selected", "all"] = "selected"
@@ -566,30 +570,34 @@ class AsyncVectorVeinClient:
             if time.time() - start_time > timeout:
                 raise TimeoutError(f"Workflow execution timed out after {timeout} seconds")
 
-            status = await self.check_workflow_status(rid)
-            if status["status"] == 200:
-                return WorkflowRunResult(
-                    rid=rid,
-                    status=status["status"],
-                    msg=status["msg"],
-                    data=[WorkflowOutput(**output) for output in status["data"]],
-                )
-            elif status["status"] == 500:
-                raise WorkflowError(f"Workflow execution failed: {status['msg']}")
+            result = await self.check_workflow_status(rid)
+            if result.status == 200:
+                return result
+            elif result.status == 500:
+                raise WorkflowError(f"Workflow execution failed: {result.msg}")
 
             await asyncio.sleep(5)
 
-    async def check_workflow_status(self, rid: str) -> Dict:
+    async def check_workflow_status(self, rid: str) -> WorkflowRunResult:
         """异步检查工作流运行状态
 
         Args:
             rid: 工作流运行记录ID
 
         Returns:
-            Dict: 工作流状态信息
+            WorkflowRunResult: 工作流运行结果
         """
         payload = {"rid": rid}
-        return await self._request("POST", "workflow/check-status", json=payload)
+        response = await self._request("POST", "workflow/check-status", json=payload)
+        if response["status"] in [200, 202]:
+            return WorkflowRunResult(
+                rid=rid,
+                status=response["status"],
+                msg=response["msg"],
+                data=[WorkflowOutput(**output) for output in response["data"]],
+            )
+        else:
+            raise WorkflowError(f"Workflow execution failed: {response['msg']}")
 
     async def get_access_keys(
         self, access_keys: Optional[List[str]] = None, get_type: Literal["selected", "all"] = "selected"

{vectorvein-0.2.1.dist-info → vectorvein-0.2.2.dist-info}/METADATA

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

{vectorvein-0.2.1.dist-info → vectorvein-0.2.2.dist-info}/RECORD

@@ -1,9 +1,9 @@
-vectorvein-0.2.1.dist-info/METADATA,sha256=MuvWxw_gudRvlueGAhNhgl-4rHhMpb2sygCZlqA-tjM,4413
-vectorvein-0.2.1.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
-vectorvein-0.2.1.dist-info/entry_points.txt,sha256=6OYgBcLyFCUgeqLgnvMyOJxPCWzgy7se4rLPKtNonMs,34
+vectorvein-0.2.2.dist-info/METADATA,sha256=wS-CyMabwgi8OjyI4c9bcTRekuY2fmfOvekZB6Dh_PQ,4413
+vectorvein-0.2.2.dist-info/WHEEL,sha256=thaaA2w1JzcGC48WYufAs8nrYZjJm8LqNfnXFOFyCC4,90
+vectorvein-0.2.2.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=THL2epmILfYk5ZTzWbyeWgkFndlD8NNwDGHK2ammQyE,29110
+vectorvein/api/client.py,sha256=bym9p462CxtY3pkOyZb28EmjLUYaQQn9_eTWJSKS8mk,29464
 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.1.dist-info/RECORD,,
+vectorvein-0.2.2.dist-info/RECORD,,