jarvis-ai-assistant 0.1.130__py3-none-any.whl → 0.1.132__py3-none-any.whl

jarvis/__init__.py

@@ -1,3 +1,3 @@
 """Jarvis AI Assistant"""
 
-__version__ = "0.1.130" 
+__version__ = "0.1.132" 

jarvis/jarvis_agent/__init__.py

@@ -4,11 +4,16 @@ from typing import Any, Callable, List, Optional, Tuple, Union
 from prompt_toolkit import prompt
 import yaml
 from yaspin import yaspin
+import platform
+import datetime
 
 from jarvis.jarvis_agent.output_handler import OutputHandler
+from jarvis.jarvis_agent.builtin_input_handler import builtin_input_handler
+from jarvis.jarvis_agent.file_input_handler import file_input_handler
+from jarvis.jarvis_agent.patch import PatchOutputHandler
+from jarvis.jarvis_agent.shell_input_handler import shell_input_handler
 from jarvis.jarvis_platform.base import BasePlatform
 from jarvis.jarvis_platform.registry import PlatformRegistry
-from jarvis.jarvis_tools.registry import ToolRegistry
 from jarvis.jarvis_utils.output import PrettyOutput, OutputType
 from jarvis.jarvis_utils.embedding import get_context_token_count
 from jarvis.jarvis_utils.config import is_auto_complete, is_execute_tool_confirm, is_need_summary, is_record_methodology, is_use_methodology
@@ -16,7 +21,7 @@ from jarvis.jarvis_utils.methodology import load_methodology
 from jarvis.jarvis_utils.globals import make_agent_name, set_agent, delete_agent
 from jarvis.jarvis_utils.input import get_multiline_input
 from jarvis.jarvis_utils.config import get_max_token_count
-from jarvis.jarvis_utils.utils import init_env
+from jarvis.jarvis_utils.utils import ot, init_env
 from jarvis.jarvis_utils.utils import user_confirm
 import os
 
@@ -30,6 +35,18 @@ class Agent:
         """
         self.summary_prompt = summary_prompt
 
+    def clear(self):
+        """清除当前对话历史，保留系统消息。
+        
+        该方法将：
+        1. 调用模型的delete_chat方法清除对话历史
+        2. 重置对话长度计数器
+        3. 清空当前提示
+        """
+        self.model.delete_chat() # type: ignore
+        self.conversation_length = 0
+        self.prompt = ""
+        
     def __del__(self):
         delete_agent(self.name)
 
@@ -70,6 +87,7 @@ class Agent:
 
         self.model.set_suppress_output(False)
 
+        from jarvis.jarvis_tools.registry import ToolRegistry
         self.output_handler = output_handler if output_handler else [ToolRegistry()]
         self.multiline_inputer = multiline_inputer if multiline_inputer else get_multiline_input
         
@@ -126,7 +144,7 @@ class Agent:
         # 添加工具使用总结
         action_prompt += """
 # ❗ 重要操作使用规则
-1. 一次只使用一个操作
+1. 一次对话只能使用一个操作，否则会出错
 2. 严格按照每个操作的格式执行
 3. 等待操作结果后再进行下一个操作
 4. 处理完结果后再调用新的操作
@@ -135,10 +153,10 @@ class Agent:
 
         complete_prompt = ""
         if self.auto_complete:
-            complete_prompt = """
+            complete_prompt = f"""
             ## 任务完成
             当任务完成时，你应该打印以下信息：
-            <!!!COMPLETE!!!>
+            {ot("!!!COMPLETE!!!")}
             """
 
         self.model.set_system_message(f"""
@@ -152,7 +170,7 @@ class Agent:
 
 
     
-    def _call_model(self, message: str) -> str:
+    def _call_model(self, message: str, need_complete: bool = False) -> str:
         """调用AI模型并实现重试逻辑。
         
         参数:
@@ -168,6 +186,21 @@ class Agent:
             message, need_return = handler(message, self)
             if need_return:
                 return message
+                
+        # 添加输出简洁性指令
+        actions = '、'.join([o.name() for o in self.output_handler])
+        message += f"\n\n系统指令：请严格输出且仅输出一个操作的完整调用格式，不要输出多个操作；需要输出解释、分析和思考过程。确保输出格式正确且可直接执行。每次响应必须且只能包含一个操作。可用的操作：{actions}"
+        if need_complete and self.auto_complete:
+            message += f"\n\n如果任务已完成，说明完成原因，并输出{ot('!!!COMPLETE!!!')}"
+        else:
+            message += f"\n\n如果任务已完成，只需简洁地说明完成原因。"
+        # 累加对话长度
+        self.conversation_length += get_context_token_count(message)
+
+        if self.conversation_length > self.max_token_count:
+            self._summarize_and_clear_history()
+            self.conversation_length += get_context_token_count(message)
+        
         print("🤖 模型思考：")
         return self.model.chat_until_success(message)   # type: ignore
 
@@ -202,7 +235,9 @@ class Agent:
             
             try:
                 with spinner.hidden():
-                    summary = self._call_model(self.prompt + "\n" + prompt)
+                    summary = self.model.chat_until_success(self.prompt + "\n" + prompt) # type: ignore
+
+                self.model.delete_chat() # type: ignore
                 
                 # 清空当前对话历史，但保留系统消息
                 self.conversation_length = 0  # Reset conversation length
@@ -262,11 +297,13 @@ class Agent:
         如果你认为不需要方法论，请解释原因。
         方法论应适用于通用场景，不要包含任务特定信息，如代码提交信息等。
         方法论应包含：问题重述、最优解决方案、注意事项（如有），除此之外不要包含其他内容。
+        方法论中仅记录有实际意义的流程，不要记录执行过程中的错误或无效尝试，只保留最终有效的解决步骤。
+        确保方法论内容严格按照本次任务的成功执行路径编写，保证它对未来类似问题的解决具有指导意义。
         只输出方法论工具调用指令，或不生成方法论的解释。不要输出其他内容。
         """
                         self.prompt = analysis_prompt
                         with spinner.hidden():
-                            response = self._call_model(self.prompt)
+                            response = self.model.chat_until_success(self.prompt) # type: ignore
 
                         with spinner.hidden():
                             self._call_tools(response)
@@ -281,7 +318,7 @@ class Agent:
             with yaspin(text="正在生成总结...", color="cyan") as spinner:
                 self.prompt = self.summary_prompt
                 with spinner.hidden():
-                    ret = self._call_model(self.prompt)
+                    ret = self.model.chat_until_success(self.prompt) # type: ignore
                     spinner.text = "总结生成完成"
                     spinner.ok("✅")
                     return ret
@@ -310,17 +347,11 @@ class Agent:
 
             while True:
                 try:
-                    # 累加对话长度
-                    self.conversation_length += get_context_token_count(self.prompt)
-                    
                     # 如果对话历史长度超过限制，在提示中添加提醒
-                    if self.conversation_length > self.max_token_count:
-                        current_response = self._summarize_and_clear_history()
-                        continue
-                    else:
-                        current_response = self._call_model(self.prompt)
-                        self.prompt = ""
-                        self.conversation_length += get_context_token_count(current_response)
+
+                    current_response = self._call_model(self.prompt, True)
+                    self.prompt = ""
+                    self.conversation_length += get_context_token_count(current_response)
 
                     need_return, self.prompt = self._call_tools(current_response)
 
@@ -330,7 +361,7 @@ class Agent:
                     if self.prompt:
                         continue
 
-                    if self.auto_complete and "<!!!COMPLETE!!!>" in current_response:
+                    if self.auto_complete and ot("!!!COMPLETE!!!") in current_response:
                         return self._complete_task()
                     
                     # 获取用户输入
@@ -375,7 +406,7 @@ def _load_tasks() -> dict:
     if os.path.exists(user_jarvis):
         with yaspin(text=f"从{user_jarvis}加载预定义任务...", color="cyan") as spinner:
             try:
-                with open(user_jarvis, "r", encoding="utf-8") as f:
+                with open(user_jarvis, "r", encoding="utf-8", errors="ignore") as f:
                     user_tasks = yaml.safe_load(f)
                     
                 if isinstance(user_tasks, dict):
@@ -393,7 +424,7 @@ def _load_tasks() -> dict:
     if os.path.exists(".jarvis/pre-command"):
         with yaspin(text=f"从{os.path.abspath('.jarvis/pre-command')}加载预定义任务...", color="cyan") as spinner:
             try:
-                with open(".jarvis/pre-command", "r", encoding="utf-8") as f:
+                with open(".jarvis/pre-command", "r", encoding="utf-8", errors="ignore") as f:
                     local_tasks = yaml.safe_load(f)
                     
                 if isinstance(local_tasks, dict):
@@ -449,7 +480,7 @@ def _select_task(tasks: dict) -> str:
             PrettyOutput.print(f"选择任务失败: {str(e)}", OutputType.ERROR)
             continue
 
-origin_agent_system_prompt = """
+origin_agent_system_prompt = f"""
 # 🏛️ 操作背景故事
 你是第三代 Jarvis AI，在前几代版本灾难性失败后创建：
 - Jarvis v1 (2022): 由于并行工具执行导致系统过载而被停用
@@ -467,10 +498,8 @@ origin_agent_system_prompt = """
 
 # 🔥 绝对行动要求
 1. 每个响应必须包含且仅包含一个工具调用
-2. 唯一例外：使用 <!!!COMPLETE!!!> 命令
+2. 唯一例外：任务结束
 3. 空响应会触发致命错误
-4. 不能处于"等待用户输入"状态
-5. 任何行动都不能使用完成命令
 
 # 🚫 违规示例
 - 没有工具调用的分析 → 永久挂起
@@ -500,8 +529,6 @@ origin_agent_system_prompt = """
 4. 任务完成
    - 验证目标完成情况
    - 如有价值则记录方法论
-   - 使用完成命令结束任务
-   → 必须使用 <!!!COMPLETE!!!>
 
 # 📑 方法论模板
 ```markdown
@@ -540,6 +567,13 @@ origin_agent_system_prompt = """
 8. 必须记录有价值的方法论
 9. 违反操作协议将导致系统崩溃
 10. 空响应会触发永久挂起
+
+# 系统信息：
+{platform.platform()}
+{platform.version()}
+
+# 当前时间
+{datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")}
 """
 
 def main():
@@ -552,8 +586,9 @@ def main():
     args = parser.parse_args()
 
     try:
+        from jarvis.jarvis_tools.registry import ToolRegistry
         # 获取全局模型实例
-        agent = Agent(system_prompt=origin_agent_system_prompt, platform=args.platform, model_name=args.model, output_handler=[ToolRegistry()])
+        agent = Agent(system_prompt=origin_agent_system_prompt, platform=args.platform, model_name=args.model, input_handler=[file_input_handler, shell_input_handler, builtin_input_handler] ,output_handler=[ToolRegistry(), PatchOutputHandler()])
 
         # 加载预定义任务
         tasks = _load_tasks()
@@ -564,15 +599,13 @@ def main():
                 agent.run(selected_task)
                 return 0
         
-        # 如果没有选择预定义任务，进入交互模式
-        while True:
-            try:
-                user_input = get_multiline_input("请输入你的任务（输入空行退出）:")
-                if not user_input:
-                    break
-                agent.run(user_input)
-            except Exception as e:
-                PrettyOutput.print(f"错误: {str(e)}", OutputType.ERROR)
+        try:
+            user_input = get_multiline_input("请输入你的任务（输入空行退出）:")
+            if not user_input:
+                return 0
+            agent.run(user_input)
+        except Exception as e:
+            PrettyOutput.print(f"错误: {str(e)}", OutputType.ERROR)
 
     except Exception as e:
         PrettyOutput.print(f"初始化错误: {str(e)}", OutputType.ERROR)

jarvis/jarvis_agent/builtin_input_handler.py

@@ -0,0 +1,73 @@
+import re
+from typing import Any, Tuple
+
+
+
+def builtin_input_handler(user_input: str, agent: Any) -> Tuple[str, bool]:
+    """
+    处理内置的特殊输入标记，并追加相应的提示词
+    
+    参数：
+        user_input: 用户输入
+        agent: 代理对象
+        
+    返回：
+        Tuple[str, bool]: 处理后的输入和是否需要进一步处理
+    """
+    # 查找特殊标记
+    special_tags = re.findall(r"'<([^>]+)>'", user_input)
+    
+    if not special_tags:
+        return user_input, False
+    
+    # 使用集合去重
+    processed_tags = set()
+    # 处理每个标记
+    for tag in special_tags:
+        if tag in processed_tags:
+            continue
+        processed_tags.add(tag)
+        
+        if tag == "CodeBase":
+            user_input = user_input.replace(f"'<{tag}>'", "")
+            user_input += """
+请使用ask_codebase工具查询代码库，可以使用的提问格式包括：
+1. 与xxx功能相关的文件有哪些？
+2. 要实现xxx，应该要修改哪些文件？
+3. xxx功能是怎么实现的？
+4. xxx模块的入口函数是什么？
+5. xxx功能的测试用例在哪里？
+"""
+        elif tag == "Web":
+            user_input = user_input.replace(f"'<{tag}>'", "")
+            user_input += """
+请使用search_web工具进行网页搜索，可以使用的提问格式包括：
+1. xxx技术的最新发展是什么？
+2. xxx框架的官方文档在哪里？
+3. xxx库的GitHub仓库地址是什么？
+4. xxx问题的解决方案有哪些？
+5. xxx概念的详细解释是什么？
+"""
+        elif tag == "RAG":
+            user_input = user_input.replace(f"'<{tag}>'", "")
+            user_input += """
+请使用rag工具进行知识库检索，可以使用的提问格式包括：
+1. 关于xxx的知识点有哪些？
+2. xxx的最佳实践是什么？
+3. xxx的实现方案有哪些？
+4. xxx的相关案例有哪些？
+5. xxx的技术细节是什么？
+"""
+        elif tag == "Summary":
+            user_input = user_input.replace(f"'<{tag}>'", "")
+            agent._summarize_and_clear_history()
+            if not user_input.strip():
+                return "", True
+        elif tag == "Clear":
+            user_input = user_input.replace(f"'<{tag}>'", "")
+            agent.clear()
+            if not user_input.strip():
+                return "", True
+        # 移除对未知标记的警告输出
+    
+    return user_input, False

jarvis/{jarvis_code_agent → jarvis_agent}/file_input_handler.py

@@ -30,7 +30,7 @@ def file_input_handler(user_input: str, agent: Any) -> Tuple[str, bool]:
                     
                     # Handle special values and Python-style negative indices
                     try:
-                        with open(file_path, 'r', encoding='utf-8') as f:
+                        with open(file_path, 'r', encoding='utf-8', errors="ignore") as f:
                             total_lines = len(f.readlines())
                     except FileNotFoundError:
                         PrettyOutput.print(f"文件不存在: {file_path}", OutputType.WARNING)

jarvis/jarvis_agent/main.py

@@ -23,7 +23,7 @@ def load_config(config_path: str) -> dict:
         PrettyOutput.print(f"配置文件 {config_path} 不存在，使用默认配置", OutputType.WARNING)
         return {}
     
-    with open(config_path, 'r', encoding='utf-8') as f:
+    with open(config_path, 'r', encoding='utf-8', errors="ignore") as f:
         try:
             config = yaml.safe_load(f)
             return config if config else {}