jarvis-ai-assistant 0.3.11__py3-none-any.whl → 0.3.13__py3-none-any.whl

jarvis/__init__.py

@@ -1,4 +1,4 @@
 # -*- coding: utf-8 -*-
 """Jarvis AI Assistant"""
 
-__version__ = "0.3.11"
+__version__ = "0.3.13"

jarvis/jarvis_agent/__init__.py

@@ -180,11 +180,11 @@ origin_agent_system_prompt = f"""
 
 
 class Agent:
-    def clear(self):
+    def clear_history(self):
         """
         Clears the current conversation history by delegating to the session manager.
         """
-        self.session.clear()
+        self.session.clear_history()
 
     def __del__(self):
         # 只有在记录启动时才停止记录
@@ -444,9 +444,10 @@ class Agent:
             4. 会检查并处理上下文长度限制
         """
         # 处理输入
-        message = self._process_input(message, run_input_handlers)
-        if run_input_handlers and self._should_return_early(message):
-            return message
+        if run_input_handlers:
+            message = self._process_input(message)
+            if not message:
+                return ""
 
         # 添加附加提示
         message = self._add_addon_prompt(message, need_complete)
@@ -459,21 +460,16 @@ class Agent:
 
         return response
 
-    def _process_input(self, message: str, run_input_handlers: bool) -> str:
+    def _process_input(self, message: str) -> str:
         """处理输入消息"""
-        if run_input_handlers:
-            for handler in self.input_handler:
-                message, need_return = handler(message, self)
-                if need_return:
-                    self._last_handler_returned = True
-                    return message
+        for handler in self.input_handler:
+            message, need_return = handler(message, self)
+            if need_return:
+                self._last_handler_returned = True
+                return message
         self._last_handler_returned = False
         return message
 
-    def _should_return_early(self, message: str) -> bool:
-        """检查是否需要提前返回"""
-        return hasattr(self, "_last_handler_returned") and self._last_handler_returned
-
     def _add_addon_prompt(self, message: str, need_complete: bool) -> str:
         """添加附加提示到消息"""
         if self.session.addon_prompt:
@@ -561,12 +557,19 @@ class Agent:
     def _handle_history_with_summary(self) -> str:
         """使用摘要方式处理历史"""
         summary = self.generate_summary()
-        self.clear_history()
 
-        if not summary:
-            return ""
+        # 先获取格式化的摘要消息
+        formatted_summary = ""
+        if summary:
+            formatted_summary = self._format_summary_message(summary)
+
+        # 清理历史（但不清理prompt，因为prompt会在builtin_input_handler中设置）
+        if self.model:
+            self.model.reset()
+        # 重置会话
+        self.session.clear_history()
 
-        return self._format_summary_message(summary)
+        return formatted_summary
 
     def _handle_history_with_file_upload(self) -> str:
         """使用文件上传方式处理历史"""
@@ -704,6 +707,7 @@ class Agent:
                 current_response = self._call_model(
                     self.session.prompt, True, run_input_handlers
                 )
+
                 self.session.prompt = ""
                 run_input_handlers = False
 
@@ -713,12 +717,18 @@ class Agent:
                     if isinstance(interrupt_result, tuple):
                         run_input_handlers, should_continue = interrupt_result
                         if should_continue:
+                            self.run_input_handlers_next_turn = True
                             continue
                     else:
                         return interrupt_result
 
                 # 处理工具调用
-                need_return, self.session.prompt = self._call_tools(current_response)
+                need_return, prompt = self._call_tools(current_response)
+                if self.session.prompt and prompt:
+                    self.session.prompt += "\n\n" + prompt
+                else:
+                    self.session.prompt = prompt
+
                 if need_return:
                     return self.session.prompt
 
@@ -871,9 +881,3 @@ class Agent:
             organizer.organize_memories(memory_type, min_overlap=3)
         else:
             PrettyOutput.print(f"已取消 '{scope_name}' 记忆库整理。", OutputType.INFO)
-
-    def clear_history(self):
-        """
-        Clears conversation history by delegating to the session manager.
-        """
-        self.session.clear_history()

jarvis/jarvis_agent/builtin_input_handler.py

@@ -32,10 +32,18 @@ def builtin_input_handler(user_input: str, agent_: Any) -> Tuple[str, bool]:
     for tag in special_tags:
         # 优先处理特殊标记
         if tag == "Summary":
-            agent._summarize_and_clear_history()
-            return "", True
+            summary = agent._summarize_and_clear_history()
+            memory_tags_prompt = agent.memory_manager.prepare_memory_tags_prompt()
+            prompt = ""
+            if summary:
+                # 将摘要和记忆标签设置为新会话的初始提示
+                prompt = summary + "\n" + memory_tags_prompt
+            else:
+                # 即使没有摘要，也确保设置记忆标签作为新会话的初始提示
+                prompt = memory_tags_prompt
+            return prompt, True
         elif tag == "Clear":
-            agent.clear()
+            agent.clear_history()
             return "", True
         elif tag == "ToolUsage":
             agent.set_addon_prompt(agent.get_tool_usage_prompt())

jarvis/jarvis_agent/file_methodology_manager.py

@@ -48,13 +48,10 @@ class FileMethodologyManager:
             # 上传成功
             from jarvis.jarvis_agent.memory_manager import MemoryManager
 
-            memory_manager = MemoryManager(self.agent)
-            memory_tags_prompt = memory_manager.prepare_memory_tags_prompt()
-
             if self.agent.files:
-                self.agent.session.prompt = f"{self.agent.session.prompt}\n\n上传的文件包含历史对话信息和方法论文件，可以从中获取一些经验信息。{memory_tags_prompt}"
+                self.agent.session.prompt = f"{self.agent.session.prompt}\n\n上传的文件包含历史对话信息和方法论文件，可以从中获取一些经验信息。"
             else:
-                self.agent.session.prompt = f"{self.agent.session.prompt}\n\n上传的文件包含历史对话信息，可以从中获取一些经验信息。{memory_tags_prompt}"
+                self.agent.session.prompt = f"{self.agent.session.prompt}\n\n上传的文件包含历史对话信息，可以从中获取一些经验信息。"
 
     def _handle_files_upload(self):
         """处理普通文件上传"""
@@ -79,14 +76,13 @@ class FileMethodologyManager:
         from jarvis.jarvis_agent.memory_manager import MemoryManager
 
         memory_manager = MemoryManager(self.agent)
-        memory_tags_prompt = memory_manager.prepare_memory_tags_prompt()
         methodology = load_methodology(
             msg,
             self.agent.get_tool_registry(),
             platform_name=self.agent.model.platform_name(),
             model_name=self.agent.model.name(),
         )
-        self.agent.session.prompt = f"{self.agent.session.prompt}\n\n以下是历史类似问题的执行经验，可参考：\n{methodology}{memory_tags_prompt}"
+        self.agent.session.prompt = f"{self.agent.session.prompt}\n\n以下是历史类似问题的执行经验，可参考：\n{methodology}"
 
     def handle_history_with_file_upload(self) -> str:
         """使用文件上传方式处理历史"""

jarvis/jarvis_agent/jarvis.py

@@ -19,21 +19,18 @@ def run_cli(
     ctx: typer.Context,
     llm_type: str = typer.Option(
         "normal",
-        "-t", "--llm-type",
+        "-t",
+        "--llm-type",
         help="使用的LLM类型，可选值：'normal'（普通）或 'thinking'（思考模式）",
     ),
-    task: Optional[str] = typer.Option(
-        None, "-T", "--task", help="从命令行直接输入任务内容"
-    ),
+    task: Optional[str] = typer.Option(None, "-T", "--task", help="从命令行直接输入任务内容"),
     model_group: Optional[str] = typer.Option(
         None, "-g", "--llm-group", help="使用的模型组，覆盖配置文件中的设置"
     ),
     tool_group: Optional[str] = typer.Option(
         None, "-G", "--tool-group", help="使用的工具组，覆盖配置文件中的设置"
     ),
-    config_file: Optional[str] = typer.Option(
-        None, "-f", "--config", help="自定义配置文件路径"
-    ),
+    config_file: Optional[str] = typer.Option(None, "-f", "--config", help="自定义配置文件路径"),
     restore_session: bool = typer.Option(
         False,
         "--restore-session",
@@ -43,9 +40,7 @@ def run_cli(
     share_methodology: bool = typer.Option(
         False, "--share-methodology", help="分享本地方法论到中心方法论仓库"
     ),
-    share_tool: bool = typer.Option(
-        False, "--share-tool", help="分享本地工具到中心工具仓库"
-    ),
+    share_tool: bool = typer.Option(False, "--share-tool", help="分享本地工具到中心工具仓库"),
 ) -> None:
     """Jarvis AI assistant command-line interface."""
     if ctx.invoked_subcommand is not None:
@@ -59,21 +54,19 @@ def run_cli(
     # 处理方法论分享
     if share_methodology:
         init_env("", config_file=config_file)  # 初始化配置但不显示欢迎信息
-        manager = MethodologyShareManager()
-        manager.run()
+        methodology_manager = MethodologyShareManager()
+        methodology_manager.run()
         return
 
     # 处理工具分享
     if share_tool:
         init_env("", config_file=config_file)  # 初始化配置但不显示欢迎信息
-        manager = ToolShareManager()
-        manager.run()
+        tool_manager = ToolShareManager()
+        tool_manager.run()
         return
 
     # 初始化环境
-    init_env(
-        "欢迎使用 Jarvis AI 助手，您的智能助理已准备就绪！", config_file=config_file
-    )
+    init_env("欢迎使用 Jarvis AI 助手，您的智能助理已准备就绪！", config_file=config_file)
 
     # 运行主流程
     try:

jarvis/jarvis_agent/session_manager.py

@@ -34,14 +34,6 @@ class SessionManager:
         """Sets the addon prompt for the next model call."""
         self.addon_prompt = addon_prompt
 
-    def clear(self):
-        """
-        Clears the current conversation history, prompt, and length counter.
-        """
-        self.model.reset()
-        self.conversation_length = 0
-        self.prompt = ""
-
     def save_session(self) -> bool:
         """Saves the current session state to a file."""
         session_dir = os.path.join(os.getcwd(), ".jarvis")

jarvis/jarvis_agent/share_manager.py

@@ -2,7 +2,7 @@
 """分享管理模块，负责工具和方法论的分享功能"""
 import os
 import subprocess
-from typing import List, Dict, Any
+from typing import List, Dict, Any, Set
 from abc import ABC, abstractmethod
 
 from prompt_toolkit import prompt
@@ -16,7 +16,7 @@ def parse_selection(selection_str: str, max_value: int) -> List[int]:
 
     例如: "1,2,3,4-9,20" -> [1, 2, 3, 4, 5, 6, 7, 8, 9, 20]
     """
-    selected: set[int] = set()
+    selected: Set[int] = set()
     parts = selection_str.split(",")
 
     for part in parts:
@@ -73,13 +73,15 @@ class ShareManager(ABC):
                     if "__pycache__" not in content:
                         f.write("\n__pycache__/\n")
                         modified = True
-            
+
             if modified:
-                subprocess.run(["git", "add", ".gitignore"], cwd=self.repo_path, check=True)
                 subprocess.run(
-                    ["git", "commit", "-m", "chore: add __pycache__ to .gitignore"], 
-                    cwd=self.repo_path, 
-                    check=True
+                    ["git", "add", ".gitignore"], cwd=self.repo_path, check=True
+                )
+                subprocess.run(
+                    ["git", "commit", "-m", "chore: add __pycache__ to .gitignore"],
+                    cwd=self.repo_path,
+                    check=True,
                 )
                 subprocess.run(["git", "push"], cwd=self.repo_path, check=True)
         else:
@@ -163,9 +165,7 @@ class ShareManager(ABC):
     def select_resources(self, resources: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
         """让用户选择要分享的资源"""
         # 显示可选的资源
-        resource_list = [
-            f"\n可分享的{self.get_resource_type()}（已排除中心仓库中已有的）："
-        ]
+        resource_list = [f"\n可分享的{self.get_resource_type()}（已排除中心仓库中已有的）："]
         for i, resource in enumerate(resources, 1):
             resource_list.append(f"[{i}] {self.format_resource_display(resource)}")
 

{jarvis_ai_assistant-0.3.11.dist-info → jarvis_ai_assistant-0.3.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: jarvis-ai-assistant
-Version: 0.3.11
+Version: 0.3.13
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire

{jarvis_ai_assistant-0.3.11.dist-info → jarvis_ai_assistant-0.3.13.dist-info}/RECORD

@@ -1,11 +1,11 @@
-jarvis/__init__.py,sha256=4eE4JJA0Fb7zUXTVi01_rPJMEyES6tv7MbD410hdOqw,74
-jarvis/jarvis_agent/__init__.py,sha256=NF_eUF85RRplGJyLmInIsTn-09aEPtjj4utYvEALvJw,31802
+jarvis/__init__.py,sha256=sNG-zIDnzHZonw0jMokUfOeIX1O9_pvk1H7gGrXcEUI,74
+jarvis/jarvis_agent/__init__.py,sha256=vpioyz2E-F4hTK0A2isCbdeH6_90kfZ3FM-I5_mOnaM,31879
 jarvis/jarvis_agent/agent_manager.py,sha256=YzpMiF0H2-eyk2kn2o24Bkj3bXsQx7Pv2vfD4gWepo0,2893
-jarvis/jarvis_agent/builtin_input_handler.py,sha256=Qs4LAr4xdKLBJpQE81YP4CkucAop86ms0iVoKa1nnso,2468
+jarvis/jarvis_agent/builtin_input_handler.py,sha256=wS-FqpT3pIXwHn1dfL3SpXonUKWgVThbQueUIeyRc2U,2917
 jarvis/jarvis_agent/config_editor.py,sha256=Ctk82sO6w2cNW0-_5L7Bomj-hgM4U7WwMc52fwhAJyg,1809
 jarvis/jarvis_agent/edit_file_handler.py,sha256=w-byNJ4TN_SlV3djjfFC7OksySOFGrM8ku49w662dzc,11854
-jarvis/jarvis_agent/file_methodology_manager.py,sha256=qCRh36LLiaF3cXwIRlHurCzQlQQCUvd9BteCdj3CxfQ,4526
-jarvis/jarvis_agent/jarvis.py,sha256=eBz0cKCaDHHA9JZvYSKu1igSCMclhRw6bEr0hOJuK-0,3252
+jarvis/jarvis_agent/file_methodology_manager.py,sha256=PwDUQwq7HVIyPInsN8fgWyMXLwi8heIXPrqfBZJhVHs,4260
+jarvis/jarvis_agent/jarvis.py,sha256=mE7sPQzuHtL9j8e_yDO91KfpbEqxrBrBG3px_WyWz50,3238
 jarvis/jarvis_agent/main.py,sha256=0xq-rjadcTedcB2HJSk2v0ihcm2-r2NXLj_Nq9xS9LY,3345
 jarvis/jarvis_agent/memory_manager.py,sha256=F7HTNzdN1_-cSygnz7zKSJRJvPLUOosqcXQeiW8zG4U,5266
 jarvis/jarvis_agent/methodology_share_manager.py,sha256=vwWNexluTXSI3qeNP3zJAemOjWW37o_1AlqDR1C8wCI,6910
@@ -13,8 +13,8 @@ jarvis/jarvis_agent/output_handler.py,sha256=P7oWpXBGFfOsWq7cIhS_z9crkQ19ES7qU5p
 jarvis/jarvis_agent/prompt_builder.py,sha256=PH1fPDVa8z_RXkoXHJFNDf8PQjUoLNLYwkh2lC__p40,1705
 jarvis/jarvis_agent/prompts.py,sha256=X6cXa-n0xqBQ8LDTgLsD0kqziAh1s0cNp89i4mxcvHg,9444
 jarvis/jarvis_agent/protocols.py,sha256=JWnJDikFEuwvFUv7uzXu0ggJ4O9K2FkMnfVCwIJ5REw,873
-jarvis/jarvis_agent/session_manager.py,sha256=DnvI9rWkVmkyO1XfKZyo9lTn4ajg4ccwzEkoRHFPOJM,2925
-jarvis/jarvis_agent/share_manager.py,sha256=h3H6PXSTVUBLablFys_h16mdI7a48RVALQhADL17rSs,8696
+jarvis/jarvis_agent/session_manager.py,sha256=xPeQcQRBmAlc-CGAzVz4MTOKhZKb27rTS-5De3Oauwg,2713
+jarvis/jarvis_agent/share_manager.py,sha256=wFcdULSog1mMxDyB94ofbqitFL8DCX8i1u6qVzSEuAk,8704
 jarvis/jarvis_agent/shell_input_handler.py,sha256=1IboqdxcJuoIqRpmDU10GugR9fWXUHyCEbVF4nIWbyo,1328
 jarvis/jarvis_agent/task_analyzer.py,sha256=-fQ9YBYFcc-Z1FSoDIPzRfAgkREFoIOXtU2TdBkB-e0,4656
 jarvis/jarvis_agent/task_manager.py,sha256=HJm4_SMpsFbQMUUsAZeHm7cZuhNbz28YW-DRLYgoarc,4422
@@ -119,9 +119,9 @@ jarvis/jarvis_utils/methodology.py,sha256=IIMU17WVSunsWXsnXROd4G77LxgYs4xEC_xm_0
 jarvis/jarvis_utils/output.py,sha256=QRLlKObQKT0KuRSeZRqYb7NlTQvsd1oZXZ41WxeWEuU,10894
 jarvis/jarvis_utils/tag.py,sha256=f211opbbbTcSyzCDwuIK_oCnKhXPNK-RknYyGzY1yD0,431
 jarvis/jarvis_utils/utils.py,sha256=LiVui9RMsbfUdzbvBBwbGNC4uniGnLp3LFsk7LXGrQE,47370
-jarvis_ai_assistant-0.3.11.dist-info/licenses/LICENSE,sha256=AGgVgQmTqFvaztRtCAXsAMryUymB18gZif7_l2e1XOg,1063
-jarvis_ai_assistant-0.3.11.dist-info/METADATA,sha256=oUuaxxRov1QSVuHlazfqRNpS5mZe3RU31xNJaD1kVTo,18216
-jarvis_ai_assistant-0.3.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-jarvis_ai_assistant-0.3.11.dist-info/entry_points.txt,sha256=4GcWKFxRJD-QU14gw_3ZaW4KuEVxOcZK9i270rwPdjA,1395
-jarvis_ai_assistant-0.3.11.dist-info/top_level.txt,sha256=1BOxyWfzOP_ZXj8rVTDnNCJ92bBGB0rwq8N1PCpoMIs,7
-jarvis_ai_assistant-0.3.11.dist-info/RECORD,,
+jarvis_ai_assistant-0.3.13.dist-info/licenses/LICENSE,sha256=AGgVgQmTqFvaztRtCAXsAMryUymB18gZif7_l2e1XOg,1063
+jarvis_ai_assistant-0.3.13.dist-info/METADATA,sha256=YJ01G5YB7gEX59rrdrPw6xb-0p_pdg-KT5318_yl3t4,18216
+jarvis_ai_assistant-0.3.13.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+jarvis_ai_assistant-0.3.13.dist-info/entry_points.txt,sha256=4GcWKFxRJD-QU14gw_3ZaW4KuEVxOcZK9i270rwPdjA,1395
+jarvis_ai_assistant-0.3.13.dist-info/top_level.txt,sha256=1BOxyWfzOP_ZXj8rVTDnNCJ92bBGB0rwq8N1PCpoMIs,7
+jarvis_ai_assistant-0.3.13.dist-info/RECORD,,