pycoze 0.1.279__py3-none-any.whl → 0.1.281__py3-none-any.whl

pycoze/api/lib/view.py

@@ -5,6 +5,7 @@ from pycoze import utils
 
 socket = utils.socket
 
+
 class ViewCls:
 
     def __init__(self, location: list[str], nodeIds: list[str] = []):
@@ -15,7 +16,7 @@ class ViewCls:
 
     def __getitem__(self, key):
         return self.location[key]
-    
+
     def wait_for_tab_open(self):
         times = 0
         while not self.is_tab_open():
@@ -38,7 +39,7 @@ class ViewCls:
         assert self.is_tab
         result = socket.post_and_recv_result("is-tab-open", {"location": self.location})
         return result
-    
+
     def pin_tab(self):
         assert self.is_tab
         self.wait_for_tab_open()
@@ -57,55 +58,63 @@ class WorkflowCls(ViewCls):
         pass
 
     def run(self, wait_for_end=False):
-        result = socket.post_and_recv_result("workflow-run", {"location": self.location, "wait_for_end": wait_for_end})
+        result = socket.post_and_recv_result(
+            "workflow-run", {"location": self.location, "wait_for_end": wait_for_end}
+        )
         if not result["ok"]:
             raise Exception(result["value"])
         return result["value"]  # task_id
 
     def stop(self):
-        result = socket.post_and_recv_result("workflow-stop", {"location": self.location})
+        result = socket.post_and_recv_result(
+            "workflow-stop", {"location": self.location}
+        )
         if not result["ok"]:
             raise Exception(result["value"])
 
     def get_state(self):
-        result = socket.post_and_recv_result("workflow-get-state", {"location": self.location})
+        result = socket.post_and_recv_result(
+            "workflow-get-state", {"location": self.location}
+        )
         if not result["ok"]:
             raise Exception(result["value"])
         return result["value"]
 
     def get_nodes(self):
-        result = socket.post_and_recv_result("workflow-get-graph-nodes", {"location": self.location})
+        result = socket.post_and_recv_result(
+            "workflow-get-graph-nodes", {"location": self.location}
+        )
         if not result["ok"]:
             raise Exception(result["value"])
         return result["value"]
-        
+
     def get_edges(self):
-        result = socket.post_and_recv_result("workflow-get-graph-edges", {"location": self.location})
+        result = socket.post_and_recv_result(
+            "workflow-get-graph-edges", {"location": self.location}
+        )
         return result["value"]
-       
-    def add_text_node(self, text: str, position='center'):
-        result = socket.post_and_recv_result("workflow-add-text-node", {
-            "location": self.location,
-            "text": text,
-            "position": position
-        })
+
+    def add_text_node(self, text: str, position="center"):
+        result = socket.post_and_recv_result(
+            "workflow-add-text-node",
+            {"location": self.location, "text": text, "position": position},
+        )
         if not result["ok"]:
             raise Exception(result["value"])
         time.sleep(0.01)
         return result["value"]  # node
 
-    def add_markdown_node(self, text: str, position='center'):
-        result = socket.post_and_recv_result("workflow-add-markdown-node", {
-            "location": self.location,
-            "text": text,
-            "position": position
-        })
+    def add_markdown_node(self, text: str, position="center"):
+        result = socket.post_and_recv_result(
+            "workflow-add-markdown-node",
+            {"location": self.location, "text": text, "position": position},
+        )
         if not result["ok"]:
             raise Exception(result["value"])
         time.sleep(0.01)
         return result["value"]  # node
 
-    def add_file_node(self, file_path: str, position='center'):
+    def add_file_node(self, file_path: str, position="center"):
         is_step = False
         if file_path.lower().endswith(".py"):
             with open(file_path, "r", encoding="utf-8") as f:
@@ -114,40 +123,36 @@ class WorkflowCls(ViewCls):
                     is_step = True
         if is_step:
             return self.add_step_file_node(file_path, position)
-        result = socket.post_and_recv_result("workflow-add-file-node", {
-            "location": self.location,
-            "file_path": file_path,
-            "position": position
-        })
+        result = socket.post_and_recv_result(
+            "workflow-add-file-node",
+            {"location": self.location, "file_path": file_path, "position": position},
+        )
         if not result["ok"]:
             raise Exception(result["value"])
         time.sleep(0.01)
         return result["value"]  # node
 
-    def add_step_file_node(self, file_path: str, position='center'):
-        result = socket.post_and_recv_result("workflow-add-step-file-node", {
-            "location": self.location,
-            "file_path": file_path,
-            "position": position
-        })
+    def add_step_file_node(self, file_path: str, position="center"):
+        result = socket.post_and_recv_result(
+            "workflow-add-step-file-node",
+            {"location": self.location, "file_path": file_path, "position": position},
+        )
         if not result["ok"]:
             raise Exception(result["value"])
         time.sleep(0.01)
         return result["value"]  # node
 
-    def add_edge(self, from_node_id: str, from_port_id: str, to_node_id: str, to_port_id: str):
+    def add_edge(
+        self, from_node_id: str, from_port_id: str, to_node_id: str, to_port_id: str
+    ):
         result = socket.post_and_recv_result(
-            "workflow-add-edge", {
+            "workflow-add-edge",
+            {
                 "location": self.location,
-                "source": {
-                    "cell": from_node_id,
-                    "port": from_port_id
-                },
-                "target": {
-                    "cell": to_node_id,
-                    "port": to_port_id
-                }
-            })
+                "source": {"cell": from_node_id, "port": from_port_id},
+                "target": {"cell": to_node_id, "port": to_port_id},
+            },
+        )
         if not result["ok"]:
             raise Exception(result["value"])
         time.sleep(0.01)
@@ -156,7 +161,9 @@ class WorkflowCls(ViewCls):
     def get_node_info(self, node: Union[str, object]):
         if isinstance(node, object):
             node_id = node["id"]
-        result = socket.post_and_recv_result("workflow-get-node-info", {"location": self.location, "node_id": node_id})
+        result = socket.post_and_recv_result(
+            "workflow-get-node-info", {"location": self.location, "node_id": node_id}
+        )
         if not result["ok"]:
             raise Exception(result["value"])
         return result["value"]
@@ -164,7 +171,9 @@ class WorkflowCls(ViewCls):
     def get_node_state(self, node: Union[str, object]):
         if isinstance(node, object):
             node_id = node["id"]
-        result = socket.post_and_recv_result("workflow-get-node-state", {"location": self.location, "node_id": node_id})
+        result = socket.post_and_recv_result(
+            "workflow-get-node-state", {"location": self.location, "node_id": node_id}
+        )
         if not result["ok"]:
             raise Exception(result["value"])
         return result["value"]
@@ -172,28 +181,36 @@ class WorkflowCls(ViewCls):
     def get_node_log(self, node: Union[str, object]):
         if isinstance(node, object):
             node_id = node["id"]
-        result = socket.post_and_recv_result("workflow-get-node-log", {"location": self.location, "node_id": node_id})
+        result = socket.post_and_recv_result(
+            "workflow-get-node-log", {"location": self.location, "node_id": node_id}
+        )
         if not result["ok"]:
             raise Exception(result["value"])
         return result["value"]
 
-    def get_node_input_data(self, node: Union[str, object], port:str):
+    def get_node_input_data(self, node: Union[str, object], port: str):
         if isinstance(node, object):
             node_id = node["id"]
-        result = socket.post_and_recv_result("workflow-get-node-input-data", {"location": self.location, "node_id": node_id, "port":port})
+        result = socket.post_and_recv_result(
+            "workflow-get-node-input-data",
+            {"location": self.location, "node_id": node_id, "port": port},
+        )
         if not result["ok"]:
             raise Exception(result["value"])
         return result["value"]
-    
-    def get_node_output_data(self, node: Union[str, object], port:str):
+
+    def get_node_output_data(self, node: Union[str, object], port: str):
         if isinstance(node, object):
             node_id = node["id"]
-        result = socket.post_and_recv_result("workflow-get-node-output-data", {"location": self.location, "node_id": node_id, "port":port})
+        result = socket.post_and_recv_result(
+            "workflow-get-node-output-data",
+            {"location": self.location, "node_id": node_id, "port": port},
+        )
         if not result["ok"]:
             raise Exception(result["value"])
         return result["value"]
-    
-    def parse_node_data(self, data: Union[dict|list]):
+
+    def parse_node_data(self, data: Union[dict, list]):
         if isinstance(data, list):
             return [parse_data(d) for d in data]
-        return parse_data(data)
+        return parse_data(data)

pycoze/bot/agent/agent.py

@@ -15,7 +15,7 @@ from .agent_types.const import HumanToolString
 async def run_agent(agent, inputs: list, tool_compatibility_mode: bool):
     exist_ids = set()
     content_list = []
-    async for event in agent.astream_events(inputs, version="v1"):
+    async for event in agent.astream_events(inputs, version="v2"):
         kind = event["event"]
         if kind == "on_chain_end":
             if "data" in event:

{pycoze-0.1.279.dist-info → pycoze-0.1.281.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pycoze
-Version: 0.1.279
+Version: 0.1.281
 Summary: Package for pycoze only!
 Home-page: UNKNOWN
 Author: Yuan Jie Xiong

{pycoze-0.1.279.dist-info → pycoze-0.1.281.dist-info}/RECORD

@@ -8,13 +8,13 @@ pycoze/ai/llm/think.py,sha256=sUgTBdGzcZtL3r-Wx8M3lDuVUmDVz8g3qC0VU8uiKAI,5143
 pycoze/api/__init__.py,sha256=GGRRRPop0ZxdXe5JRhg2XvHITGIWfNcHA25opJZ0f1w,313
 pycoze/api/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pycoze/api/lib/tab_cls.py,sha256=nPbTfWF_Mmz3tKAmGrVdgK21HE9g1074V59nrteiG8c,2612
-pycoze/api/lib/view.py,sha256=sl3qOC-IbfJaUg1-nLtEpm-0sDxyI9UOhhrnITaFsFs,7258
+pycoze/api/lib/view.py,sha256=_PIpTfeuTPPlMDKshMGsqFQYMq7ZiO4Hg5XwHwDoU60,7357
 pycoze/api/lib/window_cls.py,sha256=Yezy-SR_EwVtUjrgXnHRC69WGgXVaXeJHjaxn09Ophg,1737
 pycoze/bot/__init__.py,sha256=JxnRoCCqx_LFyVb3pLu0qYCsH8ZLuAaMXAtvVUuCHuE,78
 pycoze/bot/agent_chat.py,sha256=ARXXsIVCTpmBojz2C4BR69nB0QhM6gkEngL3iq34JPo,4178
 pycoze/bot/bot.py,sha256=_qmUTZ09FmRLifHrW5stDZWGVK6yuMEMBC3fmeYJnqk,844
 pycoze/bot/agent/__init__.py,sha256=3wE8_FFQS8j2BY-g9Cr-onV0POEvDRZaw_NCzpqrNus,265
-pycoze/bot/agent/agent.py,sha256=rdmc1GD3LVWNDpt2BD4_5N3w5c8dHkxoOlvAWKq3b1E,3653
+pycoze/bot/agent/agent.py,sha256=qkLIRgSMNT1VD_UD0e6kYUuOTOqylQiYSCay6HZ12LA,3653
 pycoze/bot/agent/assistant.py,sha256=5LIgPIVVzx6uIOWT5S_XDDyPPjPHRBBNpIU3GiOkVHc,1186
 pycoze/bot/agent/chat.py,sha256=mubOCAHvA6VtyE6N40elI6KrP6A69uB_G6ihE3G_Vi4,860
 pycoze/bot/agent/agent_types/__init__.py,sha256=zmU2Kmrv5mCdfg-QlPn2H6pWxbGeq8s7YTqLhpzJC6k,179
@@ -35,8 +35,8 @@ pycoze/utils/arg.py,sha256=jop1tBfe5hYkHW1NSpCeaZBEznkgguBscj_7M2dWfrs,503
 pycoze/utils/env.py,sha256=5pWlXfM1F5ZU9hhv1rHlDEanjEW5wf0nbyez9bNRqqA,559
 pycoze/utils/socket.py,sha256=bZbFFRH4mfThzRqt55BAAGQ6eICx_ja4x8UGGrUdAm8,2428
 pycoze/utils/text_or_file.py,sha256=gpxZVWt2DW6YiEg_MnMuwg36VNf3TX383QD_1oZNB0Y,551
-pycoze-0.1.279.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
-pycoze-0.1.279.dist-info/METADATA,sha256=kCMpc0ItijrWiZigjVP1T_126Zn5d7CrB2DYSQI-sK8,755
-pycoze-0.1.279.dist-info/WHEEL,sha256=ewwEueio1C2XeHTvT17n8dZUJgOvyCWCt0WVNLClP9o,92
-pycoze-0.1.279.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
-pycoze-0.1.279.dist-info/RECORD,,
+pycoze-0.1.281.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
+pycoze-0.1.281.dist-info/METADATA,sha256=gRDvFXAP3MK6MxGBidAvlkFmo8cm-xBMMGCeKSyVetY,755
+pycoze-0.1.281.dist-info/WHEEL,sha256=ewwEueio1C2XeHTvT17n8dZUJgOvyCWCt0WVNLClP9o,92
+pycoze-0.1.281.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
+pycoze-0.1.281.dist-info/RECORD,,