Jarvis-Brain 0.1.11.2__tar.gz → 0.1.11.4__tar.gz

{jarvis_brain-0.1.11.2 → jarvis_brain-0.1.11.4}/PKG-INFO

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

{jarvis_brain-0.1.11.2 → jarvis_brain-0.1.11.4}/mcp_tools/dp_tools.py

@@ -308,13 +308,15 @@ def register_scroll_action(mcp: FastMCP, browser_manager):
 
 
 def register_get_screenshot(mcp: FastMCP, browser_manager):
-    @mcp.tool(name="get_tab_screenshot", description="尝试对传入tab页进行截图，并将截图压缩为1M大小png图片，会返回截图保存路径")
+    @mcp.tool(name="get_tab_screenshot",
+              description="尝试对传入tab页进行截图，并将截图压缩为1M大小png图片，会返回截图保存路径")
     async def get_tab_screenshot(browser_port: int, tab_id: str) -> dict[str, Any]:
         _browser = browser_manager.get_browser(browser_port)
         target_tab = _browser.get_tab(tab_id)
         if not os.path.exists(html_source_code_local_save_path):
             os.makedirs(html_source_code_local_save_path)
         timestamp = int(time.time() * 1000)
+        time.sleep(3)
         origin_png = target_tab.get_screenshot(as_bytes="png")
         compress_png = compress_image_bytes(origin_png)
         image_path = os.path.join(html_source_code_local_save_path, f"{browser_port}_{tab_id}_{timestamp}.png")

{jarvis_brain-0.1.11.2 → jarvis_brain-0.1.11.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "Jarvis_Brain"  # 别人下载时用的名字，必须在 PyPI 上唯一
-version = "0.1.11.2"
+version = "0.1.11.4"
 description = "Jarvis brain mcp"
 dependencies = [
     "fastmcp",

{jarvis_brain-0.1.11.2 → jarvis_brain-0.1.11.4}/tools/tools.py

@@ -11,84 +11,95 @@ from PIL import Image
 import io
 
 compress_html_js = """
-    function getSimplifiedDOM(node) {
-        // 1. 处理文本节点
-        if (node.nodeType === Node.TEXT_NODE) {
-            const text = node.textContent.trim();
-            // 限制文本长度，避免大段文章消耗 token，保留前100个字符通常足够定位
-            return text ? text.slice(0, 100) + (text.length > 100 ? '...' : '') : null;
-        }
-
-        // 2. 过滤无用标签
-        const ignoreTags = ['SCRIPT', 'STYLE', 'NOSCRIPT', 'IFRAME', 'SVG', 'LINK', 'META'];
-        if (ignoreTags.includes(node.tagName)) return null;
-        if (node.nodeType !== Node.ELEMENT_NODE) return null;
-
-        // 3. 过滤不可见元素
-        const style = window.getComputedStyle(node);
-        if (style.display === 'none' || style.visibility === 'hidden' || style.opacity === '0') return null;
-        // 过滤宽高太小的元素（往往是埋点空像素）
-        const rect = node.getBoundingClientRect();
-        if (rect.width === 0 || rect.height === 0) return null;
-
-        // --- 开始构建标签字符串 ---
-        const tagName = node.tagName.toLowerCase();
-        let tagStr = tagName;
-
-        // A. 基础标识符 (ID 和 Class)
-        if (node.id) tagStr += `#${node.id}`;
-        if (node.className && typeof node.className === 'string') {
-            // 过滤掉 Tailwind 等太长且无语义的 class，保留有意义的业务 class
-            // 这里简单处理，全部保留，让 LLM 自己判断
-            const classes = node.className.trim().split(/\s+/);
-            if (classes.length > 0) tagStr += `.${classes.join('.')}`;
-        }
-
-        // B. 关键属性白名单 (这是你指出问题的核心修复)
-        const props = [];
-
-        // 通用重要属性
-        if (node.getAttribute('role')) props.push(`role="${node.getAttribute('role')}"`);
-        if (node.getAttribute('aria-label')) props.push(`aria-label="${node.getAttribute('aria-label')}"`);
-        if (node.getAttribute('title')) props.push(`title="${node.getAttribute('title')}"`);
-
-        // 特定标签的特定属性
-        if (tagName === 'a') {
-            const href = node.getAttribute('href');
-            // 只保留有意义的链接，忽略 javascript:;
-            if (href && !href.startsWith('javascript')) props.push(`href="${href}"`);
-        } else if (tagName === 'input' || tagName === 'textarea' || tagName === 'select') {
-            if (node.getAttribute('type')) props.push(`type="${node.getAttribute('type')}"`);
-            if (node.getAttribute('name')) props.push(`name="${node.getAttribute('name')}"`);
-            if (node.getAttribute('placeholder')) props.push(`placeholder="${node.getAttribute('placeholder')}"`);
-            if (node.disabled) props.push('disabled');
-            if (node.checked) props.push('checked');
-        } else if (tagName === 'button') {
-            if (node.getAttribute('type')) props.push(`type="${node.getAttribute('type')}"`);
-        } else if (tagName === 'img') {
-            if (node.getAttribute('alt')) props.push(`alt="${node.getAttribute('alt')}"`);
-        }
-
-        if (props.length > 0) {
-            tagStr += ` ${props.join(' ')}`;
-        }
-
-        // 4. 递归子节点
-        const children = Array.from(node.childNodes)
-            .map(getSimplifiedDOM)
-            .filter(n => n !== null);
-
-        // 5. 组装输出
-        // 如果没有子节点，也没有ID/Class，也不是输入框/图片/链接，那这个标签可能只是布局用的 div，可以考虑跳过它直接返回子节点内容
-        // 但为了保持结构完整，我们暂时保留它
-        if (children.length === 0) {
-            // 自闭合标签或空标签
-            return `<${tagStr} />`;
-        }
-        return `<${tagStr}>${children.join('')}</${tagName}>`; // 结束标签只保留 tagName 节省 token
+function getSimplifiedDOM(node) {
+    // 1. 处理文本节点
+    if (node.nodeType === Node.TEXT_NODE) {
+        const text = node.textContent.trim();
+        return text ? text.slice(0, 100) + (text.length > 100 ? '...' : '') : null;
     }
 
-    return getSimplifiedDOM(document.body);
+    // 2. 过滤无用标签
+    const ignoreTags = ['SCRIPT', 'STYLE', 'NOSCRIPT', 'IFRAME', 'SVG', 'LINK', 'META'];
+    if (ignoreTags.includes(node.tagName)) return null;
+    if (node.nodeType !== Node.ELEMENT_NODE) return null;
+
+    // 3. 过滤不可见元素
+    // 【注意】这里声明了第一次 style
+    const style = window.getComputedStyle(node);
+    
+    if (style.display === 'none' || style.visibility === 'hidden' || style.opacity === '0') return null;
+    
+    // 过滤宽高太小的元素（往往是埋点空像素）
+    const rect = node.getBoundingClientRect();
+    
+    // 【修复点】删除了这里重复的 const style = ... 代码
+    // 直接使用上面已经定义好的 style 变量即可
+    
+    // 如果宽高为0，但溢出可见，说明可能有定位的子元素显示在外面
+    if ((rect.width === 0 || rect.height === 0) && style.overflow !== 'visible') return null;
+
+    // --- 开始构建标签字符串 ---
+    const tagName = node.tagName.toLowerCase();
+    let tagStr = tagName;
+
+    // A. 基础标识符 (ID 和 Class)
+    if (node.id) tagStr += `#${node.id}`;
+    if (node.className && typeof node.className === 'string') {
+        const classes = node.className.trim().split(/\s+/);
+        if (classes.length > 0) tagStr += `.${classes.join('.')}`;
+    }
+
+    // B. 关键属性白名单
+    const props = [];
+
+    // 通用重要属性
+    if (node.getAttribute('role')) props.push(`role="${node.getAttribute('role')}"`);
+    if (node.getAttribute('aria-label')) props.push(`aria-label="${node.getAttribute('aria-label')}"`);
+    if (node.getAttribute('title')) props.push(`title="${node.getAttribute('title')}"`);
+    // 建议增加这个，很多弹窗用这个属性
+    if (node.getAttribute('aria-modal')) props.push(`aria-modal="${node.getAttribute('aria-modal')}"`);
+
+    // 特定标签的特定属性
+    if (tagName === 'a') {
+        const href = node.getAttribute('href');
+        if (href && !href.startsWith('javascript')) props.push(`href="${href}"`);
+    } else if (tagName === 'input' || tagName === 'textarea' || tagName === 'select') {
+        if (node.getAttribute('type')) props.push(`type="${node.getAttribute('type')}"`);
+        if (node.getAttribute('name')) props.push(`name="${node.getAttribute('name')}"`);
+        if (node.getAttribute('placeholder')) props.push(`placeholder="${node.getAttribute('placeholder')}"`);
+        if (node.disabled) props.push('disabled');
+        if (node.checked) props.push('checked');
+    } else if (tagName === 'button') {
+        if (node.getAttribute('type')) props.push(`type="${node.getAttribute('type')}"`);
+    } else if (tagName === 'img') {
+        if (node.getAttribute('alt')) props.push(`alt="${node.getAttribute('alt')}"`);
+    } else if (tagName === 'dialog') {
+        // 保留 open 属性
+        if (node.open) props.push('open');
+    }
+
+    if (props.length > 0) {
+        tagStr += ` ${props.join(' ')}`;
+    }
+
+    // 4. 递归子节点 (包含 Shadow DOM 处理)
+    let childNodes = Array.from(node.childNodes);
+    if (node.shadowRoot) {
+        childNodes = [...childNodes, ...Array.from(node.shadowRoot.childNodes)];
+    }
+    
+    const children = childNodes
+        .map(getSimplifiedDOM)
+        .filter(n => n !== null);
+    
+    // 5. 组装输出
+    if (children.length === 0) {
+        return `<${tagStr} />`;
+    }
+    return `<${tagStr}>${children.join('')}</${tagName}>`; 
+}
+
+return getSimplifiedDOM(document.body);
 """