Jarvis-Brain 0.1.7.7__py3-none-any.whl → 0.1.7.8__py3-none-any.whl

{jarvis_brain-0.1.7.7.dist-info → jarvis_brain-0.1.7.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Jarvis_Brain
-Version: 0.1.7.7
+Version: 0.1.7.8
 Summary: Jarvis brain mcp
 Requires-Python: >=3.10
 Requires-Dist: beautifulsoup4

{jarvis_brain-0.1.7.7.dist-info → jarvis_brain-0.1.7.8.dist-info}/RECORD

@@ -1,11 +1,11 @@
 mcp_tools/__init__.py,sha256=nqhlHRalTYQF8gU5RBfanOn-zSONLYON1mjPIvP-f4w,113
-mcp_tools/dp_tools.py,sha256=O51cz2ectsbJJ6fN9X1YarSC7JereXeUWFtehx71E3Q,10796
+mcp_tools/dp_tools.py,sha256=LUnBMk6fjY-q30YUIy4sjwobAKeSmDOkZfAhToi2BT0,11485
 mcp_tools/main.py,sha256=Fdt2N3oKGwvruuno_ywnuWSlm1BexE9ZY669H2LTo9w,1056
 tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tools/browser_manager.py,sha256=EXM7n-sDOtdQGpWkVTAZHWhepVU-7PAoUTDNgGF9_fQ,1938
 tools/browser_proxy.py,sha256=cdMRxcUYyaOqGU17lldltHOvt9rxXD5Dwh7hBXEBby4,6780
 tools/tools.py,sha256=TaWs-CNXy-py9BFmCnJrQ09ke938xXpImf-N2Qo_Rvc,4708
-jarvis_brain-0.1.7.7.dist-info/METADATA,sha256=u_pB128wsC-0P2-hKsYlgkZFaDZ45dCoM81ZdnZDRdU,241
-jarvis_brain-0.1.7.7.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-jarvis_brain-0.1.7.7.dist-info/entry_points.txt,sha256=YFQT4xpkUqt5dM5wlKPQQOqcjMuFrT9iuRAzIpAyH7U,51
-jarvis_brain-0.1.7.7.dist-info/RECORD,,
+jarvis_brain-0.1.7.8.dist-info/METADATA,sha256=1FAwCfT0kxBMKNUGmq1JnYH_4yzSfipZD81BOLe4-0Y,241
+jarvis_brain-0.1.7.8.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+jarvis_brain-0.1.7.8.dist-info/entry_points.txt,sha256=YFQT4xpkUqt5dM5wlKPQQOqcjMuFrT9iuRAzIpAyH7U,51
+jarvis_brain-0.1.7.8.dist-info/RECORD,,

mcp_tools/dp_tools.py

@@ -88,8 +88,12 @@ def register_close_tab(mcp: FastMCP, browser_manager, client_manager: DPProxyCli
 
 
 def register_check_selector(mcp: FastMCP, browser_manager, client_manager: DPProxyClientManager):
-    @mcp.tool(name="check_selector", description="查找tab页中是否包含元素，并返回元素attr_name所对应的值")
-    async def check_selector(browser_port: int, tab_id: str, css_selector: str, attr_name: str = "text") -> dict[
+    @mcp.tool(name="check_selector",
+              description="查找tab页中是否包含元素，并返回元素attr_name所对应的值。"
+                          "当要选择的元素包含过多元素时，需要传入offset和page_size来分批查看元素，一般不建议调整page_size，更推荐你调整offset"
+                          "同时如果单个元素属性值太长，函数会进行截断。一般的单个元素的属性值超过300个字符的就会触发截断，截断后会在最后拼接'...'")
+    async def check_selector(browser_port: int, tab_id: str, css_selector: str, attr_name: str = "text",
+                             offset: int = 0, page_size: int = 10) -> dict[
         str, Any]:
         _browser = browser_manager.get_browser(browser_port)
         target_tab = _browser.get_tab(tab_id)
@@ -105,13 +109,18 @@ def register_check_selector(mcp: FastMCP, browser_manager, client_manager: DPPro
             attr_output = "\n".join(ele_text_list)
         else:
             attr_output = json.dumps([i.attr(attr_name) for i in target_eles])
-        # 对attr_output逐个截断，截断的长度为：一轮最大token除以元素个数+6个点
-        attr_output = [attr_str[:one_turn_max_token // (len(attr_str) + 6)] + "......" for attr_str in attr_output]
+        if len(attr_output) > page_size:
+            attr_output = attr_output[offset:offset + page_size]
+        # 对attr_output逐个截断，截断的长度为：一轮最大token除以元素个数+3个点+两个引号和逗号
+        attr_output = [attr_str[:min(300, one_turn_max_token // (len(attr_str) + 6))] + "..."
+                       for attr_str in attr_output]
         return dp_mcp_message_pack(
             f"已完成tab页:【{tab_id}】对：【{css_selector}】的检查",
             tab_id=tab_id,
             selector=css_selector,
             selector_ele_exist=exist_flag,
+            page_size=page_size,
+            offset=offset,
             attr_output=attr_output
         )