jarvis-ai-assistant 0.1.213__py3-none-any.whl → 0.1.216__py3-none-any.whl

jarvis/__init__.py

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

jarvis/jarvis_agent/__init__.py

@@ -356,6 +356,43 @@ class Agent:
         """
         self.after_tool_call_cb = cb
 
+    def save_session(self) -> bool:
+        """保存当前会话状态到文件"""
+        if not self.model:
+            PrettyOutput.print("没有可用的模型实例来保存会话。", OutputType.ERROR)
+            return False
+        session_dir = os.path.join(os.getcwd(), ".jarvis")
+        os.makedirs(session_dir, exist_ok=True)
+        platform_name = self.model.platform_name()
+        model_name = self.model.name().replace("/", "_").replace("\\", "_")
+        session_file = os.path.join(
+            session_dir, f"saved_session_{self.name}_{platform_name}_{model_name}.json"
+        )
+        return self.model.save(session_file)
+
+    def restore_session(self) -> bool:
+        """从文件恢复会话状态"""
+        if not self.model:
+            return False  # No model, cannot restore
+        platform_name = self.model.platform_name()
+        model_name = self.model.name().replace("/", "_").replace("\\", "_")
+        session_file = os.path.join(
+            os.getcwd(),
+            ".jarvis",
+            f"saved_session_{self.name}_{platform_name}_{model_name}.json",
+        )
+        if not os.path.exists(session_file):
+            return False
+
+        if self.model.restore(session_file):
+            try:
+                os.remove(session_file)
+                PrettyOutput.print("会话已恢复，并已删除会话文件。", OutputType.SUCCESS)
+            except OSError as e:
+                PrettyOutput.print(f"删除会话文件失败: {e}", OutputType.ERROR)
+            return True
+        return False
+
     def get_tool_registry(self) -> Optional[Any]:
         """获取工具注册表实例"""
         from jarvis.jarvis_tools.registry import ToolRegistry

jarvis/jarvis_agent/builtin_input_handler.py

@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 import re
+import sys
 from typing import Any, Tuple
 
 from jarvis.jarvis_utils.config import get_replace_map
@@ -29,7 +30,6 @@ def builtin_input_handler(user_input: str, agent_: Any) -> Tuple[str, bool]:
     replace_map = get_replace_map()
     # 处理每个标记
     for tag in special_tags:
-
         # 优先处理特殊标记
         if tag == "Summary":
             agent._summarize_and_clear_history()
@@ -51,6 +51,17 @@ def builtin_input_handler(user_input: str, agent_: Any) -> Tuple[str, bool]:
 
             load_config()
             return "", True
+        elif tag == "SaveSession":
+            if agent.save_session():
+                from jarvis.jarvis_utils.output import OutputType, PrettyOutput
+
+                PrettyOutput.print("会话已成功保存。正在退出...", OutputType.SUCCESS)
+                sys.exit(0)
+            else:
+                from jarvis.jarvis_utils.output import OutputType, PrettyOutput
+
+                PrettyOutput.print("保存会话失败。", OutputType.ERROR)
+            return "", True
 
         processed_tag = set()
         add_on_prompt = ""

jarvis/jarvis_agent/edit_file_handler.py

@@ -206,48 +206,60 @@ class EditFileHandler(OutputHandler):
                 exact_search = search_text
 
                 if exact_search in modified_content:
-                    if modified_content.count(exact_search) > 1:
-                        PrettyOutput.print(
-                            f"搜索文本在文件中存在多处匹配：\n{exact_search}",
-                            output_type=OutputType.WARNING,
-                        )
-                        return False, f"搜索文本在文件中存在多处匹配：\n{exact_search}"
-
                     # 直接执行替换（保留所有原始格式）
                     modified_content = modified_content.replace(
                         exact_search, replace_text
                     )
                     print(f"✅ 补丁 #{patch_count} 应用成功")
                 else:
-                    # 尝试增加缩进重试
                     found = False
-                    for space_count in range(1, 17):
-                        indented_search = "\n".join(
-                            " " * space_count + line if line.strip() else line
-                            for line in search_text.split("\n")
-                        )
-                        indented_replace = "\n".join(
-                            " " * space_count + line if line.strip() else line
-                            for line in replace_text.split("\n")
-                        )
-                        if indented_search in modified_content:
-                            if modified_content.count(indented_search) > 1:
-                                PrettyOutput.print(
-                                    f"搜索文本在文件中存在多处匹配：\n{indented_search}",
-                                    output_type=OutputType.WARNING,
-                                )
-                                return (
-                                    False,
-                                    f"搜索文本在文件中存在多处匹配：\n{indented_search}",
-                                )
+                    # 如果匹配不到，并且search与replace块的首尾都是换行，尝试去掉第一个和最后一个换行
+                    if (
+                        search_text.startswith("\n")
+                        and search_text.endswith("\n")
+                        and replace_text.startswith("\n")
+                        and replace_text.endswith("\n")
+                    ):
+                        stripped_search = search_text[1:-1]
+                        stripped_replace = replace_text[1:-1]
+                        if stripped_search in modified_content:
                             modified_content = modified_content.replace(
-                                indented_search, indented_replace
-                            )
-                            print(
-                                f"✅ 补丁 #{patch_count} 应用成功 (自动增加 {space_count} 个空格缩进)"
+                                stripped_search, stripped_replace
                             )
+                            print(f"✅ 补丁 #{patch_count} 应用成功 (自动去除首尾换行)")
                             found = True
-                            break
+
+                    if not found:
+                        # 尝试增加缩进重试
+                        current_search = search_text
+                        current_replace = replace_text
+                        if (
+                            current_search.startswith("\n")
+                            and current_search.endswith("\n")
+                            and current_replace.startswith("\n")
+                            and current_replace.endswith("\n")
+                        ):
+                            current_search = current_search[1:-1]
+                            current_replace = current_replace[1:-1]
+
+                        for space_count in range(1, 17):
+                            indented_search = "\n".join(
+                                " " * space_count + line if line.strip() else line
+                                for line in current_search.split("\n")
+                            )
+                            indented_replace = "\n".join(
+                                " " * space_count + line if line.strip() else line
+                                for line in current_replace.split("\n")
+                            )
+                            if indented_search in modified_content:
+                                modified_content = modified_content.replace(
+                                    indented_search, indented_replace
+                                )
+                                print(
+                                    f"✅ 补丁 #{patch_count} 应用成功 (自动增加 {space_count} 个空格缩进)"
+                                )
+                                found = True
+                                break
 
                     if not found:
                         PrettyOutput.print(

jarvis/jarvis_agent/jarvis.py

@@ -112,9 +112,18 @@ def main() -> None:
     parser.add_argument("-p", "--platform", type=str, help="Platform to use")
     parser.add_argument("-m", "--model", type=str, help="Model to use")
     parser.add_argument(
-        "-t", "--task", type=str, help="Directly input task content from command line"
+        "-t",
+        "--task",
+        type=str,
+        help="Directly input task content from command line",
     )
     parser.add_argument("-f", "--config", type=str, help="Path to custom config file")
+    parser.add_argument(
+        "--restore-session",
+        action="store_true",
+        help="Restore session from .jarvis/saved_session.json",
+        default=False,
+    )
     args = parser.parse_args()
     init_env(
         "欢迎使用 Jarvis AI 助手，您的智能助理已准备就绪！", config_file=args.config
@@ -130,6 +139,13 @@ def main() -> None:
             need_summary=False,
         )
 
+        # 尝试恢复会话
+        if args.restore_session:
+            if agent.restore_session():
+                PrettyOutput.print("会话已成功恢复。", OutputType.SUCCESS)
+            else:
+                PrettyOutput.print("无法恢复会话。", OutputType.WARNING)
+
         # 优先处理命令行直接传入的任务
         if args.task:
             agent.run(args.task)

jarvis/jarvis_code_agent/code_agent.py

@@ -131,7 +131,7 @@ class CodeAgent:
             input_handler=[shell_input_handler, builtin_input_handler],
             need_summary=need_summary,
             use_methodology=False,  # 禁用方法论
-            use_analysis=False      # 禁用分析
+            use_analysis=False,  # 禁用分析
         )
 
         self.agent.set_after_tool_call_cb(self.after_tool_call_cb)
@@ -359,9 +359,7 @@ class CodeAgent:
 
                 # 添加提交信息到final_ret
                 if commits:
-                    final_ret += (
-                        f"\n\n代码已修改完成\n补丁内容:\n```diff\n{diff}\n```\n"
-                    )
+                    final_ret += f"\n\n代码已修改完成\n补丁内容:\n```diff\n{diff}\n```\n"
                     # 修改后的提示逻辑
                     lint_tools_info = "\n".join(
                         f"   - {file}: 使用 {'、'.join(get_lint_tools(file))}"
@@ -380,6 +378,7 @@ class CodeAgent:
 {file_list}
 {tool_info}
 如果本次修改引入了警告和错误，请根据警告和错误信息修复代码
+注意：如果要进行静态检查，需要在所有的修改都完成之后进行集中检查
                     """
                         agent.set_addon_prompt(addon_prompt)
                 else:
@@ -420,6 +419,9 @@ def main() -> None:
     parser.add_argument(
         "-r", "--requirement", type=str, help="Requirement to process", default=None
     )
+    parser.add_argument(
+        "--restore-session", action="store_true", help="Restore session from .jarvis/saved_session.json", default=False
+    )
     args = parser.parse_args()
 
     curr_dir = os.getcwd()
@@ -428,6 +430,18 @@ def main() -> None:
 
     try:
         agent = CodeAgent(platform=args.platform, model=args.model, need_summary=False)
+
+        # 尝试恢复会话
+        if args.restore_session:
+            if agent.agent.restore_session():
+                PrettyOutput.print(
+                    "已从 .jarvis/saved_session.json 恢复会话。", OutputType.SUCCESS
+                )
+            else:
+                PrettyOutput.print(
+                    "无法从 .jarvis/saved_session.json 恢复会话。", OutputType.WARNING
+                )
+
         if args.requirement:
             agent.run(args.requirement)
         else:

jarvis/jarvis_platform/ai8.py

@@ -1,5 +1,6 @@
 import os
-from typing import Generator, List, Tuple
+from typing import Any, Dict, Generator, List, Tuple
+
 from jarvis.jarvis_platform.base import BasePlatform
 import json
 
@@ -11,7 +12,6 @@ from jarvis.jarvis_utils.utils import while_success
 class AI8Model(BasePlatform):
     """AI8 model implementation"""
 
-    platform_name = "ai8"
     BASE_URL = "https://ai8.rcouyi.com"
 
     def get_model_list(self) -> List[Tuple[str, str]]:
@@ -57,7 +57,6 @@ class AI8Model(BasePlatform):
     def create_conversation(self) -> bool:
         """Create a new conversation"""
         try:
-
             # 1. 创建会话
             response = while_success(
                 lambda: http.post(
@@ -117,13 +116,12 @@ class AI8Model(BasePlatform):
     def chat(self, message: str) -> Generator[str, None, None]:
         """Execute conversation"""
         try:
-
             # 确保有会话ID
             if not self.conversation:
                 if not self.create_conversation():
                     raise Exception("Failed to create conversation")
 
-            payload = {
+            payload: Dict[str, Any] = {
                 "text": message,
                 "sessionId": self.conversation["id"] if self.conversation else None,
                 "files": [],
@@ -168,6 +166,11 @@ class AI8Model(BasePlatform):
         """Return model name"""
         return self.model_name
 
+    @classmethod
+    def platform_name(cls) -> str:
+        """Return platform name"""
+        return "ai8"
+
     def delete_chat(self) -> bool:
         """Delete current chat session"""
         try:
@@ -184,7 +187,7 @@ class AI8Model(BasePlatform):
 
             data = response.json()
             if data["code"] == 0:
-                self.conversation = None
+                self.conversation = {}
                 return True
             else:
                 error_msg = f"删除会话失败: {data.get('msg', '未知错误')}"
@@ -195,6 +198,55 @@ class AI8Model(BasePlatform):
             PrettyOutput.print(f"删除会话失败: {str(e)}", OutputType.ERROR)
             return False
 
+    def save(self, file_path: str) -> bool:
+        """Save chat session to a file."""
+        if not self.conversation:
+            PrettyOutput.print("没有活动的会话可供保存", OutputType.WARNING)
+            return False
+
+        state = {
+            "conversation": self.conversation,
+            "model_name": self.model_name,
+            "system_prompt": self.system_prompt,
+        }
+
+        try:
+            with open(file_path, "w", encoding="utf-8") as f:
+                json.dump(state, f, ensure_ascii=False, indent=4)
+            self._saved = True
+            PrettyOutput.print(f"会话已成功保存到 {file_path}", OutputType.SUCCESS)
+            return True
+        except Exception as e:
+            PrettyOutput.print(f"保存会话失败: {str(e)}", OutputType.ERROR)
+            return False
+
+    def restore(self, file_path: str) -> bool:
+        """Restore chat session from a file."""
+        try:
+            with open(file_path, "r", encoding="utf-8") as f:
+                state = json.load(f)
+
+            self.conversation = state["conversation"]
+            self.model_name = state["model_name"]
+            self.system_prompt = state.get("system_prompt", "")
+
+            # A restored session should not be deleted on exit, as it's persistent.
+            self._saved = True
+
+            PrettyOutput.print(f"从 {file_path} 成功恢复会话", OutputType.SUCCESS)
+            return True
+        except FileNotFoundError:
+            PrettyOutput.print(f"会话文件未找到: {file_path}", OutputType.ERROR)
+            return False
+        except KeyError as e:
+            PrettyOutput.print(
+                f"恢复失败: 会话文件格式不正确，缺少键 {e}", OutputType.ERROR
+            )
+            return False
+        except Exception as e:
+            PrettyOutput.print(f"恢复会话失败: {str(e)}", OutputType.ERROR)
+            return False
+
     def get_available_models(self) -> List[str]:
         """Get available model list
 

jarvis/jarvis_platform/base.py

@@ -27,10 +27,12 @@ class BasePlatform(ABC):
         """Initialize model"""
         self.suppress_output = True  # 添加输出控制标志
         self.web = False  # 添加web属性，默认false
+        self._saved = False
 
     def __del__(self):
         """Destroy model"""
-        self.delete_chat()
+        if not self._saved:
+            self.delete_chat()
 
     @abstractmethod
     def set_model_name(self, model_name: str):
@@ -178,11 +180,45 @@ class BasePlatform(ABC):
         """Model name"""
         raise NotImplementedError("name is not implemented")
 
+    @classmethod
+    @abstractmethod
+    def platform_name(cls) -> str:
+        """Platform name"""
+        raise NotImplementedError("platform_name is not implemented")
+
     @abstractmethod
     def delete_chat(self) -> bool:
         """Delete chat"""
         raise NotImplementedError("delete_chat is not implemented")
 
+    @abstractmethod
+    def save(self, file_path: str) -> bool:
+        """Save chat session to a file.
+
+        Note:
+            Implementations of this method should set `self._saved = True` upon successful saving
+            to prevent the session from being deleted on object destruction.
+
+        Args:
+            file_path: The path to save the session file.
+
+        Returns:
+            True if saving is successful, False otherwise.
+        """
+        raise NotImplementedError("save is not implemented")
+
+    @abstractmethod
+    def restore(self, file_path: str) -> bool:
+        """Restore chat session from a file.
+
+        Args:
+            file_path: The path to restore the session file from.
+
+        Returns:
+            True if restoring is successful, False otherwise.
+        """
+        raise NotImplementedError("restore is not implemented")
+
     @abstractmethod
     def set_system_prompt(self, message: str):
         """Set system message"""

jarvis/jarvis_platform/human.py

@@ -4,6 +4,7 @@
 
 # 提供与真实人类交互的模拟接口
 
+import json
 import random
 import string
 from typing import Generator, List, Tuple
@@ -16,8 +17,6 @@ from jarvis.jarvis_utils.output import OutputType, PrettyOutput
 class HumanPlatform(BasePlatform):
     """人类交互平台实现，模拟大模型但实际上与人交互"""
 
-    platform_name = "human"
-
     def get_model_list(self) -> List[Tuple[str, str]]:
         """获取支持的模型列表"""
         return [("human", "Human Interaction")]
@@ -59,8 +58,9 @@ class HumanPlatform(BasePlatform):
 
         # 将prompt复制到剪贴板
         import subprocess
+
         try:
-            subprocess.run(['xsel', '-ib'], input=prompt.encode('utf-8'), check=True)
+            subprocess.run(["xsel", "-ib"], input=prompt.encode("utf-8"), check=True)
             PrettyOutput.print("提示已复制到剪贴板", OutputType.INFO)
         except subprocess.CalledProcessError as e:
             PrettyOutput.print(f"无法复制到剪贴板: {e}", OutputType.WARNING)
@@ -80,10 +80,55 @@ class HumanPlatform(BasePlatform):
         self.first_message = True
         return True
 
+    def save(self, file_path: str) -> bool:
+        """Save chat session to a file."""
+        state = {
+            "conversation_id": self.conversation_id,
+            "model_name": self.model_name,
+            "system_message": self.system_message,
+            "first_message": self.first_message,
+        }
+
+        try:
+            with open(file_path, "w", encoding="utf-8") as f:
+                json.dump(state, f, ensure_ascii=False, indent=4)
+            self._saved = True
+            PrettyOutput.print(f"会话已成功保存到 {file_path}", OutputType.SUCCESS)
+            return True
+        except Exception as e:
+            PrettyOutput.print(f"保存会话失败: {str(e)}", OutputType.ERROR)
+            return False
+
+    def restore(self, file_path: str) -> bool:
+        """Restore chat session from a file."""
+        try:
+            with open(file_path, "r", encoding="utf-8") as f:
+                state = json.load(f)
+
+            self.conversation_id = state.get("conversation_id", "")
+            self.model_name = state.get("model_name", "human")
+            self.system_message = state.get("system_message", "")
+            self.first_message = state.get("first_message", True)
+            self._saved = True
+
+            PrettyOutput.print(f"从 {file_path} 成功恢复会话", OutputType.SUCCESS)
+            return True
+        except FileNotFoundError:
+            PrettyOutput.print(f"会话文件未找到: {file_path}", OutputType.ERROR)
+            return False
+        except Exception as e:
+            PrettyOutput.print(f"恢复会话失败: {str(e)}", OutputType.ERROR)
+            return False
+
     def name(self) -> str:
         """平台名称"""
         return self.model_name
 
+    @classmethod
+    def platform_name(cls) -> str:
+        """平台名称"""
+        return "human"
+
     def support_web(self) -> bool:
         """是否支持网页浏览功能"""
         return False

jarvis/jarvis_platform/kimi.py

@@ -21,8 +21,6 @@ class KimiModel(BasePlatform):
     - 消息收发
     """
 
-    platform_name = "kimi"
-
     def get_model_list(self) -> List[Tuple[str, str]]:
         """Get model list"""
         return [
@@ -158,7 +156,7 @@ class KimiModel(BasePlatform):
             )
 
             response_data = b""
-            
+
             # 处理流式响应
             for chunk in response_stream:
                 response_data += chunk
@@ -298,9 +296,9 @@ class KimiModel(BasePlatform):
                 lambda: http.stream_post(url, headers=headers, json=payload),
                 sleep_time=5,
             )
-            
+
             response_data = b""
-            
+
             # 处理流式响应
             for chunk in response_stream:
                 response_data += chunk
@@ -367,10 +365,61 @@ class KimiModel(BasePlatform):
             PrettyOutput.print(f"删除会话时发生错误: {str(e)}", OutputType.ERROR)
             return False
 
+    def save(self, file_path: str) -> bool:
+        """Save chat session to a file."""
+        if not self.chat_id:
+            PrettyOutput.print("没有活动的会话可供保存", OutputType.WARNING)
+            return False
+
+        state = {
+            "chat_id": self.chat_id,
+            "model_name": self.model_name,
+            "system_message": self.system_message,
+            "first_chat": self.first_chat,
+            "uploaded_files": self.uploaded_files,
+        }
+
+        try:
+            with open(file_path, "w", encoding="utf-8") as f:
+                json.dump(state, f, ensure_ascii=False, indent=4)
+            self._saved = True
+            PrettyOutput.print(f"会话已成功保存到 {file_path}", OutputType.SUCCESS)
+            return True
+        except Exception as e:
+            PrettyOutput.print(f"保存会话失败: {str(e)}", OutputType.ERROR)
+            return False
+
+    def restore(self, file_path: str) -> bool:
+        """Restore chat session from a file."""
+        try:
+            with open(file_path, "r", encoding="utf-8") as f:
+                state = json.load(f)
+
+            self.chat_id = state.get("chat_id", "")
+            self.model_name = state.get("model_name", "kimi")
+            self.system_message = state.get("system_message", "")
+            self.first_chat = state.get("first_chat", True)
+            self.uploaded_files = state.get("uploaded_files", [])
+            self._saved = True
+
+            PrettyOutput.print(f"从 {file_path} 成功恢复会话", OutputType.SUCCESS)
+            return True
+        except FileNotFoundError:
+            PrettyOutput.print(f"会话文件未找到: {file_path}", OutputType.ERROR)
+            return False
+        except Exception as e:
+            PrettyOutput.print(f"恢复会话失败: {str(e)}", OutputType.ERROR)
+            return False
+
     def name(self) -> str:
         """Model name"""
         return self.model_name
 
+    @classmethod
+    def platform_name(cls) -> str:
+        """Platform name"""
+        return "kimi"
+
     def support_web(self) -> bool:
         """Kimi平台支持web功能"""
         return True