Jarvis-Brain 0.0.3__py3-none-any.whl → 0.0.5__py3-none-any.whl

{jarvis_brain-0.0.3.dist-info → jarvis_brain-0.0.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Jarvis_Brain
-Version: 0.0.3
+Version: 0.0.5
 Summary: Jarvis brain
 Requires-Python: >=3.10
 Requires-Dist: drissionpage

jarvis_brain-0.0.5.dist-info/RECORD

@@ -0,0 +1,7 @@
+mcp_tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mcp_tools/dp_tools.py,sha256=f5jl6BHfuvqVjFplnUEvlzLbK0k0bJnMpR7cVHxWsy8,6017
+mcp_tools/main.py,sha256=QS5GUQQC-zfMqQ_rFm6Kc1X4ATfalrGl_FPMR6yCp1U,380
+jarvis_brain-0.0.5.dist-info/METADATA,sha256=krtKk3icbi2KWLn6bxivwtUSi-b7mOdZn_VgF4M3XVE,176
+jarvis_brain-0.0.5.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+jarvis_brain-0.0.5.dist-info/entry_points.txt,sha256=YFQT4xpkUqt5dM5wlKPQQOqcjMuFrT9iuRAzIpAyH7U,51
+jarvis_brain-0.0.5.dist-info/RECORD,,

mcp_tools/dp_tools.py

@@ -9,9 +9,8 @@ from DrissionPage._elements.none_element import NoneElement
 from fastmcp import FastMCP
 from lxml import html, etree
 
-# mcp = FastMCP("JARVIS_DP_TOOLS")
-
 html_source_code_local_save_path = os.path.join(os.getcwd(), "html-source-code")
+browser_pool = {}
 
 
 # 压缩html
@@ -75,6 +74,7 @@ def register_visit_url(mcp: FastMCP):
             }]
         }
 
+
 def register_get_html(mcp: FastMCP):
     @mcp.tool(name="get_html", description="使用Drissionpage获取某一个tab页的html")
     async def get_html(browser_port: int, tab_id: str) -> dict[str, Any]:
@@ -95,6 +95,7 @@ def register_get_html(mcp: FastMCP):
             }]
         }
 
+
 def register_get_new_tab(mcp: FastMCP):
     @mcp.tool(name="get_new_tab", description="使用Drissionpage创建一个新的tab页，在新的tab页中打开url")
     async def get_new_tab(browser_port: int, url: str) -> dict[str, Any]:
@@ -112,6 +113,7 @@ def register_get_new_tab(mcp: FastMCP):
             }]
         }
 
+
 def register_switch_tab(mcp: FastMCP):
     @mcp.tool(name="switch_tab", description="根据传入的tab_id切换到对应的tab页", )
     async def switch_tab(browser_port: int, tab_id: str) -> dict[str, Any]:
@@ -126,6 +128,7 @@ def register_switch_tab(mcp: FastMCP):
             }]
         }
 
+
 def register_close_tab(mcp: FastMCP):
     @mcp.tool(name="close_tab", description="根据传入的tab_id关闭tab页", )
     async def close_tab(browser_port, tab_id) -> dict[str, Any]:
@@ -140,6 +143,7 @@ def register_close_tab(mcp: FastMCP):
             }]
         }
 
+
 def register_check_selector(mcp: FastMCP):
     @mcp.tool(name="check_selector", description="查找tab页中是否包含元素")
     async def check_selector(browser_port: int, tab_id: str, css_selector: str) -> dict[str, Any]:
@@ -150,17 +154,16 @@ def register_check_selector(mcp: FastMCP):
             css_selector = "css:" + css_selector
         target_ele = target_tab.ele(css_selector)
         return {
-        "content": [{
-            "type": "text",
-            "text": json.dumps({
-                "message": f"已完成tab页:【{tab_id}】对：【{css_selector}】的检查",
-                "tab_id": tab_id,
-                "selector": css_selector,
-                "selector_ele_exist": not isinstance(target_ele, NoneElement),
-            }, ensure_ascii=False)
-        }]
-    }
-
+            "content": [{
+                "type": "text",
+                "text": json.dumps({
+                    "message": f"已完成tab页:【{tab_id}】对：【{css_selector}】的检查",
+                    "tab_id": tab_id,
+                    "selector": css_selector,
+                    "selector_ele_exist": not isinstance(target_ele, NoneElement),
+                }, ensure_ascii=False)
+            }]
+        }
 
 # def main():
 #     mcp.run()

mcp_tools/main.py

@@ -1,6 +1,5 @@
 # main.py
-import os
-from src.mcp_tools.dp_tools import *
+from mcp_tools.dp_tools import *
 from fastmcp import FastMCP
 
 mcp = FastMCP("Jarvis Brain DrissionPage mcp")

jarvis_brain-0.0.3.dist-info/RECORD

@@ -1,7 +0,0 @@
-mcp_tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mcp_tools/dp_tools.py,sha256=TsBJM_Fdr6IreZvqoUZnDv2OJi-oMEYCLzO5cIIUVe4,5991
-mcp_tools/main.py,sha256=udTCzVeYLhYK_sFBuKQ9_idbF_qKY-0oTfJaigZc2_g,394
-jarvis_brain-0.0.3.dist-info/METADATA,sha256=OVAipWel585XsqFvWoETCsYA_RQVEakHo2l_EAa1N_I,176
-jarvis_brain-0.0.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-jarvis_brain-0.0.3.dist-info/entry_points.txt,sha256=YFQT4xpkUqt5dM5wlKPQQOqcjMuFrT9iuRAzIpAyH7U,51
-jarvis_brain-0.0.3.dist-info/RECORD,,