jarvis-ai-assistant 0.1.89__py3-none-any.whl → 0.1.91__py3-none-any.whl

jarvis/models/registry.py

@@ -11,7 +11,11 @@ REQUIRED_METHODS = [
     ('name', []),
     ('delete_chat', []),
     ('reset', []),
-    ('set_system_message', ['message'])
+    ('set_system_message', ['message']),
+    ('set_model_name', ['model_name']),
+    ('get_model_list', []),
+    ('set_suppress_output', ['suppress']),
+    ('upload_files', ['file_list'])
 ]
 
 class PlatformRegistry:
@@ -174,6 +178,13 @@ class PlatformRegistry:
         platform = self.create_platform(platform_name)
         platform.set_model_name(model_name)
         return platform
+    
+    def get_thinking_platform(self) -> BasePlatform:
+        platform_name = os.environ.get("JARVIS_THINKING_PLATFORM", os.environ.get("JARVIS_PLATFORM", "kimi"))
+        model_name = os.environ.get("JARVIS_THINKING_MODEL", os.environ.get("JARVIS_MODEL", "kimi"))
+        platform = self.create_platform(platform_name)
+        platform.set_model_name(model_name)
+        return platform
 
     def register_platform(self, name: str, platform_class: Type[BasePlatform]):
         """注册平台类

jarvis/tools/ask_user.py

@@ -3,18 +3,8 @@ from jarvis.tools.base import Tool
 from jarvis.utils import get_multiline_input, PrettyOutput, OutputType
 
 class AskUserTool:
-    name="ask_user",
-    description="""当缺少完成任务的信息或有关键决策信息缺失时，询问用户。
-用户可以输入多行文本，空行结束输入。
-
-使用场景：
-1. 需要用户提供更多信息来完成任务
-2. 需要用户做出关键决策
-3. 需要用户确认某些重要操作
-4. 需要用户提供额外信息
-
-参数说明：
-- question: 要询问用户的问题，应该清晰明确""",
+    name="ask_user"
+    description="""当缺少完成任务的信息或有关键决策信息缺失时，询问用户。用户可以输入多行文本，空行结束输入。使用场景：1. 需要用户提供更多信息来完成任务；2. 需要用户做出关键决策；3. 需要用户确认某些重要操作；4. 需要用户提供额外信息"""
     parameters={
         "type": "object",
         "properties": {
@@ -40,8 +30,8 @@ class AskUserTool:
             question = args["question"]
             
             # 显示问题
-            PrettyOutput.print("\n问题:", OutputType.QUESTION)
-            PrettyOutput.print(question, OutputType.QUESTION)
+            PrettyOutput.print("\n问题:", OutputType.SYSTEM)
+            PrettyOutput.print(question, OutputType.SYSTEM)
             
             # 获取用户输入
             PrettyOutput.print("\n请输入您的回答（输入空行结束）:", OutputType.INPUT)

jarvis/tools/codebase_qa.py

@@ -50,7 +50,7 @@ class CodebaseQATool:
             codebase = CodeBase(root_dir)
             # 生成索引
 
-            codebase.generate_codebase(force=True)
+            codebase.generate_codebase()
             # 执行问答
             response = codebase.ask_codebase(question, top_k)
             os.chdir(current_dir)

jarvis/tools/rag.py

@@ -0,0 +1,134 @@
+from typing import Dict, Any
+import os
+from jarvis.utils import OutputType, PrettyOutput
+from jarvis.jarvis_rag.main import RAGTool as RAGCore
+
+class RAGTool:
+    name = "rag"
+    description = "基于文档目录进行问答，支持多种文档格式（txt、pdf、docx等）"
+    parameters = {
+        "type": "object",
+        "properties": {
+            "dir": {
+                "type": "string",
+                "description": "文档目录路径，支持相对路径和绝对路径"
+            },
+            "question": {
+                "type": "string",
+                "description": "要询问的问题"
+            },
+            "rebuild_index": {
+                "type": "boolean",
+                "description": "是否重建索引",
+                "default": False
+            }
+        },
+        "required": ["dir", "question"]
+    }
+
+    def __init__(self):
+        """初始化 RAG 工具"""
+        self.rag_instances = {}  # 缓存不同目录的 RAG 实例
+
+    def _get_rag_instance(self, dir_path: str) -> RAGCore:
+        """获取或创建 RAG 实例
+        
+        Args:
+            dir_path: 文档目录的绝对路径
+            
+        Returns:
+            RAGCore: RAG 实例
+        """
+        if dir_path not in self.rag_instances:
+            self.rag_instances[dir_path] = RAGCore(dir_path)
+        return self.rag_instances[dir_path]
+
+    def execute(self, args: Dict[str, Any]) -> Dict[str, Any]:
+        """执行文档问答
+        
+        Args:
+            args: 包含参数的字典
+                - dir: 文档目录路径
+                - question: 要询问的问题
+                - rebuild_index: 是否重建索引
+                
+        Returns:
+            Dict[str, Any]: 执行结果
+        """
+        try:
+            # 获取参数
+            dir_path = os.path.expanduser(args["dir"])  # 展开 ~ 等路径
+            dir_path = os.path.abspath(dir_path)  # 转换为绝对路径
+            question = args["question"]
+            rebuild_index = args.get("rebuild_index", False)
+            
+            # 检查目录是否存在
+            if not os.path.exists(dir_path):
+                return {
+                    "success": False,
+                    "error": f"目录不存在: {dir_path}"
+                }
+                
+            # 检查是否是目录
+            if not os.path.isdir(dir_path):
+                return {
+                    "success": False,
+                    "error": f"路径不是目录: {dir_path}"
+                }
+                
+            # 获取 RAG 实例
+            rag = self._get_rag_instance(dir_path)
+            
+            # 如果需要重建索引或索引不存在
+            if rebuild_index or not rag.is_index_built():
+                PrettyOutput.print("正在构建文档索引...", OutputType.INFO)
+                rag.build_index(dir_path)
+            
+            # 执行问答
+            PrettyOutput.print(f"问题: {question}", OutputType.INFO)
+            response = rag.ask(question)
+            
+            if response is None:
+                return {
+                    "success": False,
+                    "error": "未能获取答案，可能是没有找到相关文档"
+                }
+                
+            return {
+                "success": True,
+                "stdout": response,
+                "stderr": ""
+            }
+            
+        except Exception as e:
+            PrettyOutput.print(f"文档问答失败: {str(e)}", OutputType.ERROR)
+            return {
+                "success": False,
+                "error": f"执行失败: {str(e)}"
+            }
+
+def main():
+    """命令行直接运行工具"""
+    import argparse
+    
+    parser = argparse.ArgumentParser(description='文档问答工具')
+    parser.add_argument('--dir', required=True, help='文档目录路径')
+    parser.add_argument('--question', required=True, help='要询问的问题')
+    parser.add_argument('--rebuild', action='store_true', help='重建索引')
+    args = parser.parse_args()
+    
+    tool = RAGTool()
+    result = tool.execute({
+        "dir": args.dir,
+        "question": args.question,
+        "rebuild_index": args.rebuild
+    })
+    
+    if result["success"]:
+        PrettyOutput.print("\n回答:", OutputType.INFO)
+        PrettyOutput.print(result["stdout"], OutputType.INFO)
+    else:
+        PrettyOutput.print(result["error"], OutputType.ERROR)
+
+if __name__ == "__main__":
+    main() 

jarvis/tools/thinker.py

@@ -0,0 +1,203 @@
+from typing import Dict, Any
+from jarvis.utils import OutputType, PrettyOutput, load_env_from_file
+from jarvis.models.registry import PlatformRegistry
+
+class ThinkerTool:
+    name = "thinker"
+    description = "使用思维链推理方式分析复杂问题，适用于需要多步推理、逻辑分析或创造性思考的场景"
+    parameters = {
+        "type": "object",
+        "properties": {
+            "question": {
+                "type": "string",
+                "description": "需要分析的问题或任务"
+            },
+            "context": {
+                "type": "string",
+                "description": "问题相关的上下文信息或背景知识",
+                "default": ""
+            },
+            "approach": {
+                "type": "string",
+                "enum": ["chain_of_thought", "tree_of_thought", "step_by_step"],
+                "description": "思考方式：chain_of_thought(思维链)、tree_of_thought(思维树)、step_by_step(步骤分解)",
+                "default": "chain_of_thought"
+            }
+        },
+        "required": ["question"]
+    }
+
+    def __init__(self):
+        """初始化思考工具"""
+        self.model = PlatformRegistry.get_global_platform_registry().get_thinking_platform()
+
+    def _generate_prompt(self, question: str, context: str, approach: str) -> str:
+        """生成提示词
+        
+        Args:
+            question: 问题
+            context: 上下文
+            approach: 思考方式
+            
+        Returns:
+            str: 完整的提示词
+        """
+        # 基础提示词
+        base_prompt = "你是一个擅长深度思考和逻辑推理的助手。"
+        
+        # 根据不同的思考方式添加具体指导
+        approach_prompts = {
+            "chain_of_thought": """请使用思维链方式分析问题：
+1. 仔细阅读问题和上下文
+2. 逐步推理，每一步都要说明推理依据
+3. 考虑多个可能的角度
+4. 得出最终结论
+
+请按以下格式输出：
+思考过程：
+1. [第一步推理]
+2. [第二步推理]
+...
+
+结论：
+[最终结论]""",
+
+            "tree_of_thought": """请使用思维树方式分析问题：
+1. 将问题分解为多个子问题
+2. 对每个子问题进行分支探索
+3. 评估每个分支的可行性
+4. 整合最优路径
+
+请按以下格式输出：
+问题分解：
+- 子问题1
+  - 分支1.1
+  - 分支1.2
+- 子问题2
+  - 分支2.1
+  - 分支2.2
+
+分析过程：
+[详细分析每个分支]
+
+最优路径：
+[说明选择原因]
+
+结论：
+[最终结论]""",
+
+            "step_by_step": """请使用步骤分解方式分析问题：
+1. 将问题分解为具体步骤
+2. 详细说明每个步骤的执行方法
+3. 考虑每个步骤可能的问题
+4. 提供完整的解决方案
+
+请按以下格式输出：
+步骤分解：
+步骤1: [具体内容]
+步骤2: [具体内容]
+...
+
+执行分析：
+[详细分析每个步骤]
+
+解决方案：
+[完整方案]"""
+        }
+
+        # 构建完整提示词
+        prompt = f"""{base_prompt}
+
+{approach_prompts[approach]}
+
+问题：
+{question}
+
+"""
+        # 如果有上下文，添加到提示词中
+        if context:
+            prompt += f"""
+相关上下文：
+{context}
+"""
+
+        prompt += "\n请开始分析："
+        return prompt
+
+    def execute(self, args: Dict[str, Any]) -> Dict[str, Any]:
+        """执行思考分析
+        
+        Args:
+            args: 包含参数的字典
+                - question: 问题
+                - context: 上下文（可选）
+                - approach: 思考方式（可选）
+                
+        Returns:
+            Dict[str, Any]: 执行结果
+        """
+        try:
+            # 获取参数
+            question = args["question"]
+            context = args.get("context", "")
+            approach = args.get("approach", "chain_of_thought")
+            
+            # 生成提示词
+            prompt = self._generate_prompt(question, context, approach)
+            
+            # 记录开始分析
+            PrettyOutput.print(f"开始分析问题: {question}", OutputType.INFO)
+            if context:
+                PrettyOutput.print("包含上下文信息", OutputType.INFO)
+            PrettyOutput.print(f"使用{approach}方式思考", OutputType.INFO)
+            
+            # 调用模型进行分析
+            response = self.model.chat(prompt)
+            
+            if not response:
+                return {
+                    "success": False,
+                    "error": "未能获得有效的分析结果"
+                }
+                
+            return {
+                "success": True,
+                "stdout": response,
+                "stderr": ""
+            }
+            
+        except Exception as e:
+            PrettyOutput.print(f"思考分析失败: {str(e)}", OutputType.ERROR)
+            return {
+                "success": False,
+                "error": f"执行失败: {str(e)}"
+            }
+
+def main():
+    """命令行直接运行工具"""
+    import argparse
+
+    load_env_from_file()
+    
+    parser = argparse.ArgumentParser(description='深度思考分析工具')
+    parser.add_argument('--question', required=True, help='需要分析的问题')
+    parser.add_argument('--context', help='问题相关的上下文信息')
+    parser.add_argument('--approach', choices=['chain_of_thought', 'tree_of_thought', 'step_by_step'],
+                      default='chain_of_thought', help='思考方式')
+    args = parser.parse_args()
+    
+    tool = ThinkerTool()
+    result = tool.execute({
+        "question": args.question,
+        "context": args.context,
+        "approach": args.approach
+    })
+    
+    if result["success"]:
+        PrettyOutput.print("\n分析结果:", OutputType.INFO)
+        PrettyOutput.print(result["stdout"], OutputType.INFO)
+    else:
+        PrettyOutput.print(result["error"], OutputType.ERROR)
+
+if __name__ == "__main__":
+    main() 

jarvis/utils.py

@@ -97,7 +97,7 @@ class PrettyOutput:
         return formatted_text
 
     @staticmethod
-    def print(text: str, output_type: OutputType, timestamp: bool = False):
+    def print(text: str, output_type: OutputType, timestamp: bool = True):
         """打印格式化的输出"""
         print(PrettyOutput.format(text, output_type, timestamp))
         if output_type == OutputType.ERROR:
@@ -110,7 +110,7 @@ class PrettyOutput:
         width = 60
         color = PrettyOutput.COLORS.get(output_type, "")
         print(f"\n{color}" + "=" * width + f"{ColoramaStyle.RESET_ALL}")
-        PrettyOutput.print(title.center(width - 10), output_type, timestamp=False)
+        PrettyOutput.print(title.center(width - 10), output_type, timestamp=True)
         print(f"{color}" + "=" * width + f"{ColoramaStyle.RESET_ALL}\n")
 
     @staticmethod
@@ -128,7 +128,7 @@ class PrettyOutput:
 
 def get_multiline_input(tip: str) -> str:
     """获取多行输入，支持方向键、历史记录等功能"""
-    PrettyOutput.print(tip + "\n", OutputType.INFO)
+    print(f"{Fore.GREEN}{tip}{ColoramaStyle.RESET_ALL}")
     
     # 创建输入会话，启用历史记录
     session = PromptSession(history=None)  # 使用默认历史记录
@@ -219,7 +219,6 @@ def load_embedding_model():
         # 首先尝试离线加载
         embedding_model = SentenceTransformer(
             model_name,
-            device="cpu",
             cache_folder=os.path.expanduser("~/.cache/huggingface/hub"),
             local_files_only=True
         )
@@ -229,7 +228,6 @@ def load_embedding_model():
         # 如果离线加载失败，尝试在线下载
         embedding_model = SentenceTransformer(
             model_name,
-            device="cpu",
             cache_folder=os.path.expanduser("~/.cache/huggingface/hub")
         )
         PrettyOutput.print("模型下载并加载成功", OutputType.SUCCESS)

{jarvis_ai_assistant-0.1.89.dist-info → jarvis_ai_assistant-0.1.91.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: jarvis-ai-assistant
-Version: 0.1.89
+Version: 0.1.91
 Summary: Jarvis: An AI assistant that uses tools to interact with the system
 Home-page: https://github.com/skyfireitdiy/Jarvis
 Author: skyfire
@@ -129,11 +129,13 @@ Jarvis supports configuration through environment variables that can be set in t
 |---------|------|--------|------|
 | JARVIS_PLATFORM | AI platform to use, supports kimi/openai/ai8 etc | kimi | Yes |
 | JARVIS_MODEL | Model name to use | - | No |
-| JARVIS_THREAD_COUNT | Number of threads for parallel processing | 10 | No |
 | JARVIS_CODEGEN_PLATFORM | AI platform for code generation | Same as JARVIS_PLATFORM | No |
 | JARVIS_CODEGEN_MODEL | Model name for code generation | Same as JARVIS_MODEL | No |
 | JARVIS_CHEAP_PLATFORM | AI platform for cheap operations | Same as JARVIS_PLATFORM | No |
 | JARVIS_CHEAP_MODEL | Model name for cheap operations | Same as JARVIS_MODEL | No |
+| JARVIS_THINKING_PLATFORM | AI platform for thinking | Same as JARVIS_PLATFORM | No |
+| JARVIS_THINKING_MODEL | Model name for thinking | Same as JARVIS_MODEL | No |
+| JARVIS_THREAD_COUNT | Number of threads for parallel processing | 10 | No |
 | OPENAI_API_KEY | API key for OpenAI platform | - | Required for OpenAI |
 | OPENAI_API_BASE | Base URL for OpenAI API | https://api.deepseek.com | No |
 | OPENAI_MODEL_NAME | Model name for OpenAI | deepseek-chat | No |
@@ -150,11 +152,6 @@ Jarvis supports configuration through environment variables that can be set in t
 jarvis
 ```
 
-### Code Generation
-```bash
-jarvis-coder
-```
-
 ### Codebase Search
 ```bash
 # Generate codebase index

jarvis_ai_assistant-0.1.91.dist-info/RECORD

@@ -0,0 +1,41 @@
+jarvis/__init__.py,sha256=3Ab7ipGCmdagjBEJ81C-I_xiKdsal6hFcitNVQKYwt4,50
+jarvis/agent.py,sha256=82F6n8YKOaLaHp40eljZLfV-Ad4RktB8s-1c8X8-LGU,19128
+jarvis/main.py,sha256=72od8757A3bhe0ncE38S7u-YZsAh0y50w9ozMhgqIU8,5423
+jarvis/utils.py,sha256=Maqu93-rixR-_rjvoBCocCH6XiB1ol1Xozj1eMjNjE0,10308
+jarvis/jarvis_codebase/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+jarvis/jarvis_codebase/main.py,sha256=hbFYetHZOKSU0HBbKUU8Vu4o3m25nufvBnHGI8OxT6g,26503
+jarvis/jarvis_platform/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+jarvis/jarvis_platform/main.py,sha256=V-gJQdfHihtrZD6WIYbVAQSDmtrf0PFVndZ7jPrlGZg,4277
+jarvis/jarvis_rag/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+jarvis/jarvis_rag/main.py,sha256=ZzNnU0KhAiz64kMmHd5BSBhImieXYbcfgRT-s-PiP2Y,26415
+jarvis/jarvis_smart_shell/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+jarvis/jarvis_smart_shell/main.py,sha256=4BKXMIz7zGATC_4K47UQxHh7sHPH4K2dqj4RSIH9t4Y,3552
+jarvis/models/__init__.py,sha256=mrOt67nselz_H1gX9wdAO4y2DY5WPXzABqJbr5Des8k,63
+jarvis/models/ai8.py,sha256=ejMUAf614wBgmsYPkcWrCzONBcIAP4pUTy4b9hrolJY,11830
+jarvis/models/base.py,sha256=qIpO32lejPhaZk5T3VIua3td4gEguXMaeFER5kXYwLY,1782
+jarvis/models/kimi.py,sha256=cB1tC_ifdTAZUEqndzNTIcgHWi4w4tM8zdzSOcDhOrQ,16348
+jarvis/models/ollama.py,sha256=3Mb9ukwKk_HFfYp5dU0Zao5aa9AH_cTJoYhwHJt3MOo,5626
+jarvis/models/openai.py,sha256=W1FmaNnFSxeu-4qFQcuop5xeb-zh8-x6E-D56hMaVuA,4370
+jarvis/models/oyi.py,sha256=IWsrReLmphm-UMzxXcbAoiWP6w1fJRP3iQTlyDpSyg8,14491
+jarvis/models/registry.py,sha256=SDnrNzf0wQi8mb7qbu4ZNfYIy5vhz1CxmVL3Ru-VATk,8797
+jarvis/tools/__init__.py,sha256=7Rqyj5hBAv5cWDVr5T9ZTZASO7ssBHeQNm2_4ZARdkA,72
+jarvis/tools/ask_user.py,sha256=KNRaiBlAqqAZ0uUkq_l7AKi-y9ZCLE83uLtTdd-UM0o,1919
+jarvis/tools/base.py,sha256=EGRGbdfbLXDLwtyoWdvp9rlxNX7bzc20t0Vc2VkwIEY,652
+jarvis/tools/chdir.py,sha256=TjfPbX8yvNKgUNJEMXh3ZlVDEIse_Fo8xMoVsiK7_dA,2688
+jarvis/tools/codebase_qa.py,sha256=Lk8EXkin45RKUsp5C_UHm6jADBhyCT-xsYubwJ1hSiE,2403
+jarvis/tools/file_ops.py,sha256=h8g0eT9UvlJf4kt0DLXvdSsjcPj7x19lxWdDApeDfpg,3842
+jarvis/tools/generator.py,sha256=TB1zcw_JmRL2W9w6L4IxtrLF3gjnNw5Jj2Zrowj0eSg,5763
+jarvis/tools/methodology.py,sha256=UG6s5VYRcd9wrKX4cg6f7zJhet5AIcthFGMOAdevBiw,5175
+jarvis/tools/rag.py,sha256=tJzuhHaTrujVXCBgmyMqSynVF0UFsIJeRTG0Y_Tt964,4483
+jarvis/tools/registry.py,sha256=AbADf8pcjHqfNoQNJkWqEuVg6zHRdryhJyDQ5w4O2sc,9177
+jarvis/tools/search.py,sha256=c9dXtyICdl8Lm8shNPNyIx9k67uY0rMF8xnIKu2RsnE,8787
+jarvis/tools/shell.py,sha256=UPKshPyOaUwTngresUw-ot1jHjQIb4wCY5nkJqa38lU,2520
+jarvis/tools/sub_agent.py,sha256=rEtAmSVY2ZjFOZEKr5m5wpACOQIiM9Zr_3dT92FhXYU,2621
+jarvis/tools/thinker.py,sha256=Zv1v4Mp7R08w5p3e2iv5X4UWQstgAMyRD9huE-b-ZS0,5938
+jarvis/tools/webpage.py,sha256=d3w3Jcjcu1ESciezTkz3n3Zf-rp_l91PrVoDEZnckOo,2391
+jarvis_ai_assistant-0.1.91.dist-info/LICENSE,sha256=AGgVgQmTqFvaztRtCAXsAMryUymB18gZif7_l2e1XOg,1063
+jarvis_ai_assistant-0.1.91.dist-info/METADATA,sha256=ahfOtRplwS5KpK_gh3_YP2WzfbRsc3wi5D8GaSvAUps,12710
+jarvis_ai_assistant-0.1.91.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+jarvis_ai_assistant-0.1.91.dist-info/entry_points.txt,sha256=47WUorU2y8wAKrGEVh5n6o3NaOB7A-4-PfSWp-xuFSg,286
+jarvis_ai_assistant-0.1.91.dist-info/top_level.txt,sha256=1BOxyWfzOP_ZXj8rVTDnNCJ92bBGB0rwq8N1PCpoMIs,7
+jarvis_ai_assistant-0.1.91.dist-info/RECORD,,

{jarvis_ai_assistant-0.1.89.dist-info → jarvis_ai_assistant-0.1.91.dist-info}/entry_points.txt

@@ -1,7 +1,7 @@
 [console_scripts]
 jarvis = jarvis.main:main
 jarvis-codebase = jarvis.jarvis_codebase.main:main
-jarvis-coder = jarvis.jarvis_coder.main:main
+jarvis-platform = jarvis.jarvis_platform.main:main
 jarvis-rag = jarvis.jarvis_rag.main:main
 jarvis-smart-shell = jarvis.jarvis_smart_shell.main:main
 jss = jarvis.jarvis_smart_shell.main:main

jarvis/jarvis_coder/git_utils.py

@@ -1,67 +0,0 @@
-import os
-from typing import List
-import yaml
-import time
-from jarvis.utils import OutputType, PrettyOutput
-from jarvis.models.registry import PlatformRegistry
-from .model_utils import call_model_with_retry
-
-def has_uncommitted_files() -> bool:
-    """判断代码库是否有未提交的文件"""
-    # 获取未暂存的修改
-    unstaged = os.popen("git diff --name-only").read()
-    # 获取已暂存但未提交的修改
-    staged = os.popen("git diff --cached --name-only").read()
-    # 获取未跟踪的文件
-    untracked = os.popen("git ls-files --others --exclude-standard").read()
-    
-    return bool(unstaged or staged or untracked)
-
-def generate_commit_message(git_diff: str, feature: str) -> str:
-    """根据git diff和功能描述生成commit信息"""
-    prompt = f"""你是一个经验丰富的程序员，请根据以下代码变更和功能描述生成简洁明了的commit信息：
-
-功能描述：
-{feature}
-
-代码变更：
-Git Diff:
-{git_diff}
-
-请遵循以下规则：
-1. 使用英文编写
-2. 采用常规的commit message格式：<type>(<scope>): <subject>
-3. 保持简洁，不超过50个字符
-4. 准确描述代码变更的主要内容
-5. 优先考虑功能描述和git diff中的变更内容
-"""
-    
-    model = PlatformRegistry().get_global_platform_registry().get_codegen_platform()
-    model.set_suppress_output(True)
-    success, response = call_model_with_retry(model, prompt)
-    if not success:
-        return "Update code changes"
-        
-    return response.strip().split("\n")[0]
-
-def save_edit_record(record_dir: str, commit_message: str, git_diff: str) -> None:
-    """保存代码修改记录"""
-    # 获取下一个序号
-    existing_records = [f for f in os.listdir(record_dir) if f.endswith('.yaml')]
-    next_num = 1
-    if existing_records:
-        last_num = max(int(f[:4]) for f in existing_records)
-        next_num = last_num + 1
-    
-    # 创建记录文件
-    record = {
-        "timestamp": time.strftime("%Y-%m-%d %H:%M:%S"),
-        "commit_message": commit_message,
-        "git_diff": git_diff
-    }
-    
-    record_path = os.path.join(record_dir, f"{next_num:04d}.yaml")
-    with open(record_path, "w", encoding="utf-8") as f:
-        yaml.safe_dump(record, f, allow_unicode=True)
-    
-    PrettyOutput.print(f"已保存修改记录: {record_path}", OutputType.SUCCESS)