pycoze 0.1.265__tar.gz → 0.1.267__tar.gz

{pycoze-0.1.265 → pycoze-0.1.267}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pycoze
-Version: 0.1.265
+Version: 0.1.267
 Summary: Package for pycoze only!
 Author: Yuan Jie Xiong
 Author-email: aiqqqqqqq@qq.com

{pycoze-0.1.265 → pycoze-0.1.267}/pycoze/api/__init__.py

@@ -1,8 +1,10 @@
 from .lib.window_cls import WindowCls
+from .lib.tab_cls import TabCls
 
 class Api:
     def __init__(self) -> None:
         self.window = WindowCls()
+        self.tab = TabCls()
 
 api = Api()
 

pycoze-0.1.267/pycoze/api/lib/tab_cls.py

@@ -0,0 +1,103 @@
+import os
+from uuid import uuid4
+from view import ViewCls, WebsiteViewCls, FileViewCls, DirectoryViewCls, WorkflowCls
+import time
+from typing import Union
+from pycoze import utils
+
+
+socket = utils.socket
+
+class TabCls:
+
+    def open_path(self, item_path: str) -> FileViewCls | DirectoryViewCls:
+        if not os.path.isabs(item_path):
+            item_path = os.path.abspath(item_path)
+        if os.path.isdir(item_path):
+            return self.open_dir(item_path)
+        else:
+            return self.open_file(item_path)
+
+    def open_file(self, item_path: str) -> Union[FileViewCls, WebsiteViewCls, WorkflowCls]:
+        if not os.path.isabs(item_path):
+            item_path = os.path.abspath(item_path)
+        location = ["FileOrDir", item_path]
+        socket.post("add-tab", {"location": location, "name": os.path.basename(item_path)})
+        lower_suffix = item_path.lower().split(".")[-1]
+        if lower_suffix == 'workflow':
+            return WorkflowCls(location)
+        if len(location) > 1 and (location[0] == "Website" or location[1].endswith(".website") or location[1].endswith(".html") or location[1].endswith(".ipynb")):
+            return WebsiteViewCls(location)
+        return FileViewCls(location)
+
+    def open_dir(self, item_path: str) -> DirectoryViewCls:
+        if not os.path.isabs(item_path):
+            item_path = os.path.abspath(item_path)
+        location = ["FileOrDir", item_path]
+        socket.post("add-tab", {"location": location, "name": os.path.basename(item_path)})
+        return DirectoryViewCls(location)
+
+    def open_website(self, url: str, uid:str=None) -> WebsiteViewCls:
+        if uid is None:
+            uid = str(uuid4())
+        location = ["Website", url, uid]
+        socket.post("add-tab", {"location": location, "name": url})
+        return WebsiteViewCls(location)
+    
+    def get_active(self) -> ViewCls:
+        result = socket.post_and_recv_result("get-active-tab", {})
+        return self._result_to_view(result)
+
+    def _result_to_view(self, result: list[str]) -> ViewCls:
+        if result[0] == "Website":
+            return WebsiteViewCls(result)
+        elif result[0] == "FileOrDir":
+            if os.path.isdir(result[1]):
+                return DirectoryViewCls(result)
+            else:
+                if result[1].lower().split(".")[-1] == 'workflow':
+                    return WorkflowCls(result)
+                return FileViewCls(result)
+        return ViewCls(result)
+    
+    def wait_for_tab_open(self, location: list[str] | ViewCls):
+        times = 0
+        while not self.is_tab_open(location):
+            time.sleep(0.01)
+            times += 1
+            if times > 1000:
+                raise Exception("Tab open timeout")
+
+    def get_all(self) -> list[ViewCls]:
+        results = socket.post_and_recv_result("get-all-tabs", {})
+        return [self._result_to_view(result) for result in results]
+
+    def close_tab(self, location: list[str] | ViewCls):
+        if isinstance(location, ViewCls):
+            location = location.location
+        self.wait_for_tab_open(location)
+        socket.post("close-tab", {"location": location})
+
+    def switch_tab(self, location: list[str] | ViewCls):
+        if isinstance(location, ViewCls):
+            location = location.location
+        self.wait_for_tab_open(location)
+        socket.post("switchTab", {"location": location})
+
+    def is_tab_open(self, location: list[str] | ViewCls):
+        if isinstance(location, ViewCls):
+            location = location.location
+        result = socket.post_and_recv_result("is-tab-open", {"location": location})
+        return result
+
+    def pin_tab(self, location: list[str] | ViewCls):
+        if isinstance(location, ViewCls):
+            location = location.location
+        self.wait_for_tab_open(location)
+        socket.post("pin-tab", {"location": location})
+
+    def unpin_tab(self, location: list[str] | ViewCls):
+        if isinstance(location, ViewCls):
+            location = location.location
+        self.wait_for_tab_open(location)
+        socket.post("unpin-tab", {"location": location})

pycoze-0.1.267/pycoze/api/lib/view.py

@@ -0,0 +1,242 @@
+import time
+from typing import Union
+from pycoze import utils
+
+
+socket = utils.socket
+
+class ViewCls:
+
+    def __init__(self, location: list[str], nodeIds: list[str] = []):
+        self.location = location
+        self.nodeIds = nodeIds
+        self.is_in_workflow = len(nodeIds) > 0
+        self.is_tab = len(nodeIds) == 0
+
+    def __getitem__(self, key):
+        return self.location[key]
+    
+    def wait_for_tab_open(self):
+        times = 0
+        while not self.is_tab_open():
+            time.sleep(0.01)
+            times += 1
+            if times > 1000:
+                raise Exception("Tab open timeout")
+
+    def close_tab(self):
+        assert self.is_tab
+        self.wait_for_tab_open()
+        socket.post("close-tab", {"location": self.location})
+
+    def set_activate(self):
+        assert self.is_tab
+        self.wait_for_tab_open()
+        socket.post("switchTab", {"location": self.location})
+
+    def is_tab_open(self):
+        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()
+        socket.post("pin-tab", {"location": self.location})
+
+    def unpin_tab(self):
+        assert self.is_tab
+        self.wait_for_tab_open()
+        socket.post("unpin-tab", {"location": self.location})
+
+
+class WebsiteViewCls(ViewCls):
+    def execute_javaScript(self, js_code: str):
+        result = socket.post_and_recv_result("webview.executeJavaScript", {"location": self.location, "code": js_code})
+        if not result["ok"]:
+            raise Exception(result["value"])
+        return result["value"] if "value" in result else None
+    
+    def capture_page(self, rect=None):
+        result = socket.post_and_recv_result("webview.capturePage", {"location": self.location, "rect": rect})
+        if not result["ok"]:
+            raise Exception(result["value"])
+        return result["value"]
+    
+    def get_webview_height(self):
+        result = socket.post_and_recv_result("get-webview-height", {"location": self.location})
+        if not result["ok"]:
+            raise Exception(result["value"])
+        return result["value"]
+    
+    def set_webview_height(self, height: int):
+        result = socket.post_and_recv_result("set-webview-height", {"location": self.location, "height": height})
+        if not result["ok"]:
+            raise Exception(result["value"])
+        return result["value"]
+    
+    def set_webview_allow_new_window(self, allow: bool):
+        result = socket.post_and_recv_result("set-webview-allow-new-window", {"location": self.location, "allow": allow})
+        if not result["ok"]:
+            raise Exception(result["value"])
+        return result["value"]
+
+class FileViewCls(ViewCls):
+
+    def get_file_path(self):
+        return self.location[1]
+
+
+class DirectoryViewCls(FileViewCls):
+
+    def get_dir_path(self):
+        return self.location[1]
+
+
+class WorkflowCls(FileViewCls):
+
+    def layout_arrangement(self):
+        # NetworkX layout arrangement，直接用python的布局即可，虽然没有输入输出port的，但是用节点坐标，再缩放几倍一般情况就可以了
+        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})
+        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})
+        if not result["ok"]:
+            raise Exception(result["value"])
+
+    def get_state(self):
+        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})
+        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})
+        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
+        })
+        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
+        })
+        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'):
+        is_step = False
+        if file_path.lower().endswith(".py"):
+            with open(file_path, "r", encoding="utf-8") as f:
+                content = f.read()
+                if "ps_workflow" in content:
+                    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
+        })
+        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
+        })
+        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):
+        result = socket.post_and_recv_result(
+            "workflow-add-edge", {
+                "location": self.location,
+                "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)
+        return result["value"]  # edge
+
+    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})
+        if not result["ok"]:
+            raise Exception(result["value"])
+        return result["value"]
+
+    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})
+        if not result["ok"]:
+            raise Exception(result["value"])
+        return result["value"]
+
+    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})
+        if not result["ok"]:
+            raise Exception(result["value"])
+        return result["value"]
+
+    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})
+        if not result["ok"]:
+            raise Exception(result["value"])
+        return result["value"]
+    
+    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})
+        if not result["ok"]:
+            raise Exception(result["value"])
+        return result["value"]
+    
+    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)

{pycoze-0.1.265 → pycoze-0.1.267}/pycoze/api/lib/window_cls.py

@@ -8,11 +8,7 @@ socket = utils.socket
 
 class WindowCls:
 
-    def get_slected_text(self) -> str:
-        result = socket.post_and_recv_result("get-selected-text", {})
-        return result
-
-    def append_msg(self, message: str, type: str = 'info'):
+    def message(self, message: str, type: str = 'info'):
         assert type in ['info', 'warning', 'success', 'error'], 'type must be info, warning, success or error'
         if not isinstance(message, str):
             message = repr(message)
@@ -24,6 +20,10 @@ class WindowCls:
     def maximize(self):
         socket.post("maximize", {})
 
+    def get_slected_text(self) -> str:
+        result = socket.post_and_recv_result("get-selected-text", {})
+        return result
+
     def open_file_with_system(self, file_path, wait: bool):
         process_fn = subprocess.run if wait else subprocess.Popen
         if sys.platform.startswith('linux'):
@@ -35,9 +35,9 @@ class WindowCls:
         else:
             raise OSError('Unsupported operating system')
 
-    def open_program(self, file_path, wait: bool):
+    def open_program(self, program_path, wait: bool):
         process_fn = subprocess.run if wait else subprocess.Popen
-        process_fn([file_path])
+        process_fn([program_path])
 
     def execute_javaScript(self, js_code: str):
         result = socket.post_and_recv_result("executeJavaScript", {"code": js_code})

{pycoze-0.1.265 → pycoze-0.1.267}/pycoze/reference/workflow.py

@@ -23,7 +23,7 @@ def _ref_workflows(workflow_id, as_agent_tool=False, workspace_path=None):
 
     try:
         with ModuleManager(module_path) as manager:
-            module = importlib.import_module("workflow")
+            module = importlib.import_module("tool")
             export_tools = getattr(module, "export_tools")
             valid_tools = []
             for tool in export_tools:

{pycoze-0.1.265 → pycoze-0.1.267}/pycoze.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pycoze
-Version: 0.1.265
+Version: 0.1.267
 Summary: Package for pycoze only!
 Author: Yuan Jie Xiong
 Author-email: aiqqqqqqq@qq.com

{pycoze-0.1.265 → pycoze-0.1.267}/pycoze.egg-info/SOURCES.txt

@@ -14,6 +14,8 @@ pycoze/ai/llm/text_to_image_prompt.py
 pycoze/ai/llm/think.py
 pycoze/api/__init__.py
 pycoze/api/lib/__init__.py
+pycoze/api/lib/tab_cls.py
+pycoze/api/lib/view.py
 pycoze/api/lib/window_cls.py
 pycoze/bot/__init__.py
 pycoze/bot/agent_chat.py

{pycoze-0.1.265 → pycoze-0.1.267}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="pycoze",
-    version="0.1.265",
+    version="0.1.267",
     packages=find_packages(),
     install_requires=[],
     author="Yuan Jie Xiong",