vectorvein 0.2.1__tar.gz → 0.2.2__tar.gz

{vectorvein-0.2.1 → vectorvein-0.2.2}/PKG-INFO

@@ -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 → vectorvein-0.2.2}/pyproject.toml

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

{vectorvein-0.2.1 → vectorvein-0.2.2}/src/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"