jarvis-ai-assistant 0.1.138__py3-none-any.whl → 0.1.141__py3-none-any.whl

jarvis/jarvis_tools/registry.py

@@ -1,7 +1,9 @@
 import json
+import os
 from pathlib import Path
 import re
 import sys
+import tempfile
 from typing import Any, Callable, Dict, List, Optional, Tuple
 
 import yaml
@@ -10,7 +12,7 @@ from yaspin import yaspin
 from jarvis.jarvis_agent.output_handler import OutputHandler
 from jarvis.jarvis_platform.registry import PlatformRegistry
 from jarvis.jarvis_tools.base import Tool
-from jarvis.jarvis_utils.config import get_max_token_count
+from jarvis.jarvis_utils.config import get_max_input_token_count
 from jarvis.jarvis_utils.embedding import get_context_token_count
 from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 from jarvis.jarvis_utils.utils import ct, ot, init_env
@@ -23,6 +25,7 @@ tool_call_help = f"""
 
 # 📋 工具调用格式
 {ot("TOOL_CALL")}
+want: 想要从执行结果中获取到的信息，如果工具输出内容过长，会根据此字段尝试提取有效信息
 name: 工具名称
 arguments:
     param1: 值1
@@ -55,9 +58,11 @@ arguments:
 始终使用 | 语法表示字符串参数：
 
 {ot("TOOL_CALL")}
-name: execute_shell
+want: 当前的git状态，期望获取xxx的提交记录
+name: execute_script
 arguments:
-    command: |
+    interpreter: bash
+    script_cotent: |
         git status --porcelain
 {ct("TOOL_CALL")}
 
@@ -82,9 +87,7 @@ class ToolRegistry(OutputHandler):
         return "TOOL_CALL"
 
     def can_handle(self, response: str) -> bool:
-        if self._extract_tool_calls(response):
-            return True
-        return False
+        return ToolRegistry._has_tool_calls_block(response)
 
     def prompt(self) -> str:
         """加载工具"""
@@ -119,13 +122,9 @@ class ToolRegistry(OutputHandler):
         return ""
 
     def handle(self, response: str, agent: Any) -> Tuple[bool, Any]:
-        tool_calls = self._extract_tool_calls(response)
-        if len(tool_calls) > 1:
-            PrettyOutput.print(f"操作失败：检测到多个操作。一次只能执行一个操作。尝试执行的操作：{', '.join([tool_call['name'] for tool_call in tool_calls])}", OutputType.WARNING)
-            return False, f"调用失败：请一次只处理一个工具调用。"
-        if len(tool_calls) == 0:
-            return False, ""
-        tool_call = tool_calls[0]
+        tool_call, err_msg = self._extract_tool_calls(response)
+        if err_msg:
+            return False, err_msg
         return False, self.handle_tool_calls(tool_call, agent)
 
     def __init__(self):
@@ -134,8 +133,7 @@ class ToolRegistry(OutputHandler):
         # 加载内置工具和外部工具
         self._load_builtin_tools()
         self._load_external_tools()
-        # 确保max_token_count是整数
-        self.max_token_count = int(get_max_token_count() * 0.8)
+        self.max_input_token_count = get_max_input_token_count() - 2048
 
     def use_tools(self, name: List[str]):
         """使用指定工具"""
@@ -239,8 +237,14 @@ class ToolRegistry(OutputHandler):
         except Exception as e:
             PrettyOutput.print(f"从 {Path(file_path).name} 加载工具失败: {str(e)}", OutputType.ERROR)
             return False
+        
     @staticmethod
-    def _extract_tool_calls(content: str) -> List[Dict]:
+    def _has_tool_calls_block(content: str) -> bool:
+        """从内容中提取工具调用块"""
+        return re.search(ot("TOOL_CALL")+r'(.*?)'+ct("TOOL_CALL"), content, re.DOTALL) is not None
+        
+    @staticmethod
+    def _extract_tool_calls(content: str) -> Tuple[Dict, str]:
         """从内容中提取工具调用。
 
         参数:
@@ -258,11 +262,19 @@ class ToolRegistry(OutputHandler):
         for item in data:
             try:
                 msg = yaml.safe_load(item)
-                if 'name' in msg and 'arguments' in msg:
+                if 'name' in msg and 'arguments' in msg and 'want' in msg:
                     ret.append(msg)
+                else:
+                    return {}, f"""工具调用格式错误，请检查工具调用格式。 
+                    
+                    {tool_call_help}""" 
             except Exception as e:
-                continue
-        return ret
+                return {}, f"""工具调用格式错误，请检查工具调用格式。 
+                
+                {tool_call_help}""" 
+        if len(ret) > 1:
+            return {}, "检测到多个工具调用，请一次只处理一个工具调用。"
+        return ret[0] if ret else {}, ""
 
     def register_tool(self, name: str, description: str, parameters: Dict, func: Callable):
         """注册新工具"""
@@ -283,74 +295,32 @@ class ToolRegistry(OutputHandler):
             return {"success": False, "stderr": f"工具 {name} 不存在，可用的工具有: {', '.join(self.tools.keys())}", "stdout": ""}
         return tool.execute(arguments)
 
+    def _format_tool_output(self, stdout: str, stderr: str) -> str:
+        """格式化工具输出为可读字符串
+
+        Args:
+            stdout: 标准输出
+            stderr: 标准错误
+
+        Returns:
+            str: 格式化后的输出
+        """
+        output_parts = []
+        if stdout:
+            output_parts.append(f"输出:\n{stdout}")
+        if stderr:
+            output_parts.append(f"错误:\n{stderr}")
+        output = "\n\n".join(output_parts)
+        return "无输出和错误" if not output else output
+
+
     def handle_tool_calls(self, tool_call: Dict, agent: Any) -> str:
-        """处理工具调用，只处理第一个工具"""
         try:
-            # 只处理第一个工具调用
             name = tool_call["name"]
             args = tool_call["arguments"]
+            want = tool_call["want"]
             args["agent"] = agent
 
-            tool_call_help = f"""
-# 🛠️ 工具使用系统
-您正在使用一个需要精确格式和严格规则的工具执行系统。
-
-# 📋 工具调用格式
-
-{ot("TOOL_CALL")}
-name: 工具名称
-arguments:
-    param1: 值1
-    param2: 值2
-{ct("TOOL_CALL")}
-
-# ❗ 关键规则
-1. 每次只使用一个工具
-   - 一次只执行一个工具
-   - 等待结果后再进行下一步
-
-2. 严格遵守格式
-   - 完全按照上述格式
-   - 使用正确的YAML缩进
-   - 包含所有必需参数
-
-3. 结果处理
-   - 等待执行结果
-   - 不要假设结果
-   - 不要创建虚假响应
-   - 不要想象对话
-
-4. 信息管理
-   - 如果信息不足，询问用户
-   - 跳过不必要的步骤
-   - 如果卡住，请求指导
-   - 不要在没有完整信息的情况下继续
-
-# 📝 字符串参数格式
-始终使用 | 语法表示字符串参数：
-
-{ot("TOOL_CALL")}
-name: execute_shell
-arguments:
-    command: |
-        git status --porcelain
-{ct("TOOL_CALL")}
-
-# 💡 最佳实践
-- 准备好后立即开始执行
-- 无需请求许可即可开始
-- 使用正确的字符串格式
-- 监控进度并调整
-- 遇到困难时请求帮助
-
-# ⚠️ 常见错误
-- 同时调用多个工具
-- 字符串参数缺少 |
-- 假设工具结果
-- 创建虚构对话
-- 在没有所需信息的情况下继续
-"""
-
             if isinstance(args, str):
                 try:
                     args = json.loads(args)
@@ -358,58 +328,25 @@ arguments:
                     PrettyOutput.print(f"工具参数格式无效: {name} {tool_call_help}", OutputType.ERROR)
                     return ""
 
-            # Execute tool call
+            # 执行工具调用
             result = self.execute_tool(name, args)
 
-            stdout = result["stdout"]
-            stderr = result.get("stderr", "")
-            output_parts = []
-            if stdout:
-                output_parts.append(f"输出:\n{stdout}")
-            if stderr:
-                output_parts.append(f"错误:\n{stderr}")
-            output = "\n\n".join(output_parts)
-            output = "无输出和错误" if not output else output
-
-            # Process the result
-            if result["success"]:
-                # If the output exceeds 4k characters, use a large model to summarize
-                if get_context_token_count(output) > self.max_token_count:
-                    PrettyOutput.section("输出过长，正在总结...", OutputType.SYSTEM)
-                    try:
-
-                        model = PlatformRegistry.get_global_platform_registry().get_normal_platform()
-                        model.set_suppress_output(False)
-                        # If the output exceeds the maximum context length, only take the last part
-                        max_count = self.max_token_count
-                        if get_context_token_count(output) > max_count:
-                            output_to_summarize = output[-max_count:]
-                            truncation_notice = f"\n(注意：由于输出过长，仅总结最后 {max_count} 个字符)"
-                        else:
-                            output_to_summarize = output
-                            truncation_notice = ""
-
-                        prompt = f"""请总结以下工具的执行结果，提取关键信息和重要结果。注意：
-1. 保留所有重要的数值、路径、错误信息等
-2. 保持结果的准确性
-3. 用简洁的语言描述主要内容
-4. 如果有错误信息，确保包含在总结中
-
-工具名称: {name}
-执行结果:
-{output_to_summarize}
-
-请提供总结:"""
-
-                        summary = model.chat_until_success(prompt)
-                        output = f"""--- 原始输出过长，以下是总结 ---{truncation_notice}
-
-{summary}
-
---- 总结结束 ---"""
-                    except Exception as e:
-                        PrettyOutput.print(f"总结失败: {str(e)}", OutputType.ERROR)
-                        output = f"输出过长 ({len(output)} 字符)，建议查看原始输出。\n前300字符预览:\n{output[:300]}..."
+            # 格式化输出
+            output = self._format_tool_output(result["stdout"], result.get("stderr", ""))
+
+            # 处理结果
+            if get_context_token_count(output) > self.max_input_token_count:
+                output_file = os.path.join(tempfile.gettempdir(), f"jarvis_output_{os.getpid()}.txt")
+                with open(output_file, "w", encoding="utf-8") as f:
+                    f.write(output)
+                model = PlatformRegistry().get_normal_platform()
+                model.set_suppress_output(False)
+                model.upload_files([output_file])
+                prompt = f"该文件为工具执行结果，请阅读文件内容，并根据文件提取出以下信息：{want}"
+                return f"""工具调用原始输出过长，以下是根据输出提出的信息：
+
+{model.chat_until_success(prompt)}"""
+                
             return output
 
         except Exception as e:

jarvis/jarvis_tools/search_web.py

@@ -1,6 +1,3 @@
-
-
-
 import os
 import statistics
 from typing import Any, Dict
@@ -10,6 +7,7 @@ from jarvis.jarvis_platform.registry import PlatformRegistry
 class SearchWebTool:
     name = "search_web"
     description = "搜索互联网上的信息"
+    labels = ['web', 'search', 'information']
     parameters = {
         "type": "object",
         "properties": {
@@ -17,16 +15,26 @@ class SearchWebTool:
         }
     }
 
+    def __init__(self):
+        if os.getenv("YUANBAO_COOKIES", "") != "" and os.getenv("YUANBAO_AGENT_ID", "") != "":
+            self.platform = "yuanbao"
+            self.model = "deep_seek"
+        elif os.getenv("KIMI_API_KEY", "") != "":
+            self.platform = "kimi"
+            self.model = "k1"
+        else:
+            self.platform = ""
+
 
     @staticmethod
     def check() -> bool:
-        return os.getenv("YUANBAO_COOKIES", "") != "" and os.getenv("YUANBAO_AGENT_ID", "") != ""
+        return os.getenv("YUANBAO_COOKIES", "") != "" and os.getenv("YUANBAO_AGENT_ID", "") != "" or os.getenv("KIMI_API_KEY", "") != ""
 
     def execute(self, args: Dict[str, Any]) -> Dict[str, Any]: # type: ignore
         query = args.get("query")
-        model = PlatformRegistry().create_platform("yuanbao")
+        model = PlatformRegistry().create_platform(self.platform)
         model.set_suppress_output(False) # type: ignore
-        model.set_model_name("deep_seek") # type: ignore
+        model.set_model_name(self.model) # type: ignore
         return {
             "stdout": model.chat_until_success(query), # type: ignore
             "stderr": "",