Jarvis-Brain 0.1.7.3__py3-none-any.whl → 0.1.7.6__py3-none-any.whl

{jarvis_brain-0.1.7.3.dist-info → jarvis_brain-0.1.7.6.dist-info}/METADATA

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

jarvis_brain-0.1.7.6.dist-info/RECORD

@@ -0,0 +1,11 @@
+mcp_tools/__init__.py,sha256=nqhlHRalTYQF8gU5RBfanOn-zSONLYON1mjPIvP-f4w,113
+mcp_tools/dp_tools.py,sha256=KVeIEGSTMLhBsGXRA9n01zH2WSBVuRl3nzgJ-R1GTgk,10796
+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.6.dist-info/METADATA,sha256=rsmZj5uJPcxFZh52xRxwf3TPjyFMMq3ppwV4fGwUGQA,241
+jarvis_brain-0.1.7.6.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+jarvis_brain-0.1.7.6.dist-info/entry_points.txt,sha256=YFQT4xpkUqt5dM5wlKPQQOqcjMuFrT9iuRAzIpAyH7U,51
+jarvis_brain-0.1.7.6.dist-info/RECORD,,

mcp_tools/__init__.py

@@ -0,0 +1,4 @@
+a = ["a", "b", "c", "d", "e", "f", "g", "h"]
+print(len(a))
+for index, num in enumerate(a):
+    print(index, num)

mcp_tools/dp_tools.py

@@ -15,7 +15,7 @@ waf_status_code_dict = {
     521: "加速乐"
 }
 # 一轮最大输入，以免单个html最大长度超过ai最大输入
-ont_turn_max_token = 25000
+one_turn_max_token = 25000
 
 
 def register_visit_url(mcp: FastMCP, browser_manager: BrowserManager, client_manager: DPProxyClientManager):
@@ -46,7 +46,7 @@ def register_get_html(mcp: FastMCP, browser_manager, client_manager: DPProxyClie
         if not os.path.exists(html_source_code_local_save_path):
             os.makedirs(html_source_code_local_save_path)
         min_html, compress_rate = compress_html(tab.html)
-        html_str_list = [min_html[i:i + ont_turn_max_token] for i in range(0, len(min_html), ont_turn_max_token)]
+        html_str_list = [min_html[i:i + one_turn_max_token] for i in range(0, len(min_html), one_turn_max_token)]
         html_file_list = []
         for index, html_str in enumerate(html_str_list):
             file_name = file_name_prefix + f"_{tab_id}_segment{index}.html"
@@ -105,6 +105,8 @@ 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]
         return dp_mcp_message_pack(
             f"已完成tab页:【{tab_id}】对：【{css_selector}】的检查",
             tab_id=tab_id,
@@ -129,7 +131,7 @@ def register_quit_browser(mcp: FastMCP, browser_manager, client_manager: DPProxy
 
 def register_pop_first_packet(mcp: FastMCP, browser_manager, client_manager: DPProxyClientManager):
     @mcp.tool(name="pop_first_packet",
-              description="每调用一次就会弹出传入的tab页所监听到的数据包中的第一个packet_message")
+              description="每调用一次就会弹出传入的tab页所监听到的数据包中的第一个packet_message，当一个packet_message的response body过长时会被切分成多个包，具体一个请求是否还有下一个包，可以参考body_completed字段")
     async def pop_first_packet(browser_port: int, tab_id: str) -> dict[str, Any]:
         _browser = browser_manager.get_browser(browser_port)
         client = client_manager.get_client(tab_id)

tools/browser_proxy.py

@@ -7,6 +7,8 @@ from DrissionPage._units.listener import DataPacket
 from typing import Tuple, Optional
 import json
 
+one_turn_max_token = 20000
+
 
 class DPProxyClient:
     def __init__(self, driver: ChromiumTab, self_kill=False):
@@ -134,16 +136,22 @@ def check_data_packet(packet: DataPacket, client: DPProxyClient):
     if packet.request.hasPostData:
         data = packet.request.postData
     body = packet.response.body
-    temp_dict = {
-        "url": url,
-        "method": method,
-        "request_data": data,
-        "request_headers": dict(packet.request.headers),
-        "response_headers": dict(packet.response.headers),
-        "response_body": body,
-    }
-    # print(temp_dict)
-    client.packet_queue.append(temp_dict)
+    body_str = json.dumps(body, ensure_ascii=False)
+    body_str_list = [body_str[i:i + one_turn_max_token] for i in range(0, len(body_str), one_turn_max_token)]
+    body_completed = True
+    for index, body_str in enumerate(body_str_list):
+        if (index + 1) != len(body_str_list):
+            body_completed = False
+        temp_dict = {
+            "url": url,
+            "body_completed": body_completed,
+            "method": method,
+            "request_data": data,
+            "request_headers": dict(packet.request.headers),
+            "response_headers": dict(packet.response.headers),
+            "response_body_segment": body_str,
+        }
+        client.packet_queue.append(temp_dict)
 
 
 client_manager = DPProxyClientManager()

jarvis_brain-0.1.7.3.dist-info/RECORD

@@ -1,11 +0,0 @@
-mcp_tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mcp_tools/dp_tools.py,sha256=LT2x9edWmmDqNoy66zPgwjUSLpYbMPUPluvDoAXs0FU,10421
-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=ii4GCKOU-R6Uoj1I3k4RMqtuLOiddd4WAjG5WGc51F8,6359
-tools/tools.py,sha256=TaWs-CNXy-py9BFmCnJrQ09ke938xXpImf-N2Qo_Rvc,4708
-jarvis_brain-0.1.7.3.dist-info/METADATA,sha256=62mF8nqQZcrtOJtlTzUODlHGpUvpg9FcuisQDAtnwIs,241
-jarvis_brain-0.1.7.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-jarvis_brain-0.1.7.3.dist-info/entry_points.txt,sha256=YFQT4xpkUqt5dM5wlKPQQOqcjMuFrT9iuRAzIpAyH7U,51
-jarvis_brain-0.1.7.3.dist-info/RECORD,,