jarvis-ai-assistant 0.1.129__py3-none-any.whl → 0.1.131__py3-none-any.whl

jarvis/__init__.py

@@ -1,3 +1,3 @@
 """Jarvis AI Assistant"""
 
-__version__ = "0.1.129" 
+__version__ = "0.1.131" 

jarvis/jarvis_agent/__init__.py

@@ -6,6 +6,10 @@ import yaml
 from yaspin import yaspin
 
 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
@@ -23,13 +27,25 @@ import os
 class Agent:
 
     def set_summary_prompt(self, summary_prompt: str):
-        """Set the summary prompt for task completion.
+        """设置任务完成时的总结提示模板。
         
-        Args:
-            summary_prompt: The prompt template for generating task summaries
+        参数:
+            summary_prompt: 用于生成任务总结的提示模板
         """
         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)
 
@@ -126,7 +142,7 @@ class Agent:
         # 添加工具使用总结
         action_prompt += """
 # ❗ 重要操作使用规则
-1. 一次只使用一个操作
+1. 一次对话只能使用一个操作，否则会出错
 2. 严格按照每个操作的格式执行
 3. 等待操作结果后再进行下一个操作
 4. 处理完结果后再调用新的操作
@@ -152,23 +168,23 @@ class Agent:
 
 
     
-    def _call_model(self, message: str) -> str: 
-        """Call the AI model with retry logic.
+    def _call_model(self, message: str) -> str:
+        """调用AI模型并实现重试逻辑。
         
-        Args:
-            message: The input message for the model
+        参数:
+            message: 输入给模型的消息
             
-        Returns:
-            str: Model's response
+        返回:
+            str: 模型的响应
             
-        Note:
-            Will retry with exponential backoff up to 30 seconds between retries
+        注意:
+            将使用指数退避重试，最多重试30秒
         """
         for handler in self.input_handler:
             message, need_return = handler(message, self)
             if need_return:
                 return message
-        PrettyOutput.section("模型输出", OutputType.SYSTEM)
+        print("🤖 模型思考：")
         return self.model.chat_until_success(message)   # type: ignore
 
 
@@ -203,6 +219,8 @@ class Agent:
             try:
                 with spinner.hidden():
                     summary = self._call_model(self.prompt + "\n" + prompt)
+
+                self.model.delete_chat() # type: ignore
                 
                 # 清空当前对话历史，但保留系统消息
                 self.conversation_length = 0  # Reset conversation length
@@ -352,16 +370,16 @@ class Agent:
             return f"Task failed: {str(e)}"
 
     def _clear_history(self):
-        """Clear conversation history while preserving system prompt.
+        """清空对话历史但保留系统提示。
         
-        This will:
-        1. Clear the prompt
-        2. Reset the model
-        3. Reset conversation length counter
+        该方法将：
+        1. 清空当前提示
+        2. 重置模型状态
+        3. 重置对话长度计数器
         """
         self.prompt = "" 
         self.model.reset() # type: ignore
-        self.conversation_length = 0  # Reset conversation length
+        self.conversation_length = 0  # 重置对话长度
 
 
 
@@ -375,7 +393,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 +411,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):
@@ -467,10 +485,8 @@ origin_agent_system_prompt = """
 
 # 🔥 绝对行动要求
 1. 每个响应必须包含且仅包含一个工具调用
-2. 唯一例外：使用 <!!!COMPLETE!!!> 命令
+2. 唯一例外：任务结束
 3. 空响应会触发致命错误
-4. 不能处于"等待用户输入"状态
-5. 任何行动都不能使用完成命令
 
 # 🚫 违规示例
 - 没有工具调用的分析 → 永久挂起
@@ -500,8 +516,6 @@ origin_agent_system_prompt = """
 4. 任务完成
    - 验证目标完成情况
    - 如有价值则记录方法论
-   - 使用完成命令结束任务
-   → 必须使用 <!!!COMPLETE!!!>
 
 # 📑 方法论模板
 ```markdown
@@ -553,7 +567,7 @@ def main():
 
     try:
         # 获取全局模型实例
-        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()

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 {}