auto-coder 0.1.244__py3-none-any.whl → 0.1.246__py3-none-any.whl

{auto_coder-0.1.244.dist-info → auto_coder-0.1.246.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.244
+Version: 0.1.246
 Summary: AutoCoder: AutoCoder
 Author: allwefantasy
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
@@ -26,7 +26,7 @@ Requires-Dist: tabulate
 Requires-Dist: jupyter-client
 Requires-Dist: prompt-toolkit
 Requires-Dist: tokenizers
-Requires-Dist: byzerllm[saas] >=0.1.160
+Requires-Dist: byzerllm[saas] >=0.1.161
 Requires-Dist: patch
 Requires-Dist: diff-match-patch
 Requires-Dist: GitPython

{auto_coder-0.1.244.dist-info → auto_coder-0.1.246.dist-info}/RECORD

@@ -1,21 +1,22 @@
 autocoder/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/auto_coder.py,sha256=ghIEJ8NzcFr77fosIZhKxI5F55EZoU6DMHFFpw5OFso,62671
+autocoder/auto_coder.py,sha256=VoSRpvXRLIL3BKGudqFVeO8qolFAjKKWbUbbNQUikDE,63074
 autocoder/auto_coder_lang.py,sha256=Rtupq6N3_HT7JRhDKdgCBcwRaiAnyCOR_Gsp4jUomrI,3229
 autocoder/auto_coder_rag.py,sha256=illKgzP2bv-Tq50ujsofJnOHdI4pzr0ALtfR8NHHWdQ,22351
 autocoder/auto_coder_rag_client_mcp.py,sha256=WV7j5JUiQge0x4-B7Hp5-pSAFXLbvLpzQMcCovbauIM,6276
 autocoder/auto_coder_rag_mcp.py,sha256=-RrjNwFaS2e5v8XDIrKR-zlUNUE8UBaeOtojffBrvJo,8521
 autocoder/auto_coder_server.py,sha256=XU9b4SBH7zjPPXaTWWHV4_zJm-XYa6njuLQaplYJH_c,20290
 autocoder/benchmark.py,sha256=Ypomkdzd1T3GE6dRICY3Hj547dZ6_inqJbBJIp5QMco,4423
-autocoder/chat_auto_coder.py,sha256=RJC3Fh_9ywkllETWFgxapHK3aY0M9ORhRON9PNCiKZs,105338
-autocoder/chat_auto_coder_lang.py,sha256=pdGkEMR96TvjYIzgWWB51LYmBYSrS_pO3sjGH-muuDI,15052
+autocoder/chat_auto_coder.py,sha256=wdFPshtAWiFrDkCKHSxKKDhcnHQnd9mW0vQZcaRQlC4,105965
+autocoder/chat_auto_coder_lang.py,sha256=gbpjfMd1wYiIrOlLDc-G7eI497mMwjM_ud9GvO-wo9k,15261
 autocoder/command_args.py,sha256=9aYJ-AmPxP1sQh6ciw04FWHjSn31f2W9afXFwo8wgx4,30441
 autocoder/lang.py,sha256=U6AjVV8Rs1uLyjFCZ8sT6WWuNUxMBqkXXIOs4S120uk,14511
-autocoder/models.py,sha256=8Tabnz3wlVswaeTpfPxnBhN55enP_5sdAJIx4Z79nOo,5342
-autocoder/version.py,sha256=nyjL74IBCp8tNAtI88ddAEba5PkcN70aC5Kdv_PKe7w,23
+autocoder/models.py,sha256=_9Kc8oS_tnnqKzRGnybgOfn1NOey7OXZ8y9qhBmgiB4,5517
+autocoder/version.py,sha256=5HxDu_oJkWkJDjnFdCKsIoI4evQ0gJJY_wccUUoUJTU,23
 autocoder/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/agent/auto_demand_organizer.py,sha256=NWSAEsEk94vT3lGjfo25kKLMwYdPcpy9e-i21txPasQ,6942
 autocoder/agent/auto_filegroup.py,sha256=CW7bqp0FW1GIEMnl-blyAc2UGT7O9Mom0q66ITz1ckM,6635
 autocoder/agent/auto_guess_query.py,sha256=rDSdhpPHcOGE5MuDXvIrhCXAPR4ARS1LqpyoLsx2Jhw,11374
+autocoder/agent/auto_review_commit.py,sha256=3_XgcOGNbpOnwq9WMj66rceZFEa4lEQvs9OZdKQi7co,7663
 autocoder/agent/auto_tool.py,sha256=DBzip-P_T6ZtT2eHexPcusmKYD0h7ufzp7TLwXAY10E,11554
 autocoder/agent/coder.py,sha256=x6bdJwDuETGg9ebQnYlUWCxCtQcDGg73LtI6McpWslQ,72034
 autocoder/agent/designer.py,sha256=EpRbzO58Xym3GrnppIT1Z8ZFAlnNfgzHbIzZ3PX-Yv8,27037
@@ -28,7 +29,7 @@ autocoder/common/__init__.py,sha256=2isE_u4VgfogwmcUCnFcussVFlzeNOLHDMFm5z_axbU,
 autocoder/common/anything2images.py,sha256=0ILBbWzY02M-CiWB-vzuomb_J1hVdxRcenAfIrAXq9M,25283
 autocoder/common/anything2img.py,sha256=4TREa-sOA-iargieUy7MpyCYVUE-9Mmq0wJtwomPqnE,7662
 autocoder/common/audio.py,sha256=Kn9nWKQddWnUrAz0a_ZUgjcu4VUU_IcZBigT7n3N3qc,7439
-autocoder/common/auto_coder_lang.py,sha256=rIr4WJCUHgwkUkGwAaAiEd5kNx1nYRvVArbISbCfXp0,13248
+autocoder/common/auto_coder_lang.py,sha256=Slwyy0LNqsgHbyWIImYz8l4ke9cT9GagZA4HtbTXIX0,14065
 autocoder/common/buildin_tokenizer.py,sha256=L7d5t39ZFvUd6EoMPXUhYK1toD0FHlRH1jtjKRGokWU,1236
 autocoder/common/chunk_validation.py,sha256=BrR_ZWavW8IANuueEE7hS8NFAwEvm8TX34WnPx_1hs8,3030
 autocoder/common/cleaner.py,sha256=NU72i8C6o9m0vXExab7nao5bstBUsfJFcj11cXa9l4U,1089
@@ -77,14 +78,14 @@ autocoder/dispacher/actions/plugins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQ
 autocoder/dispacher/actions/plugins/action_regex_project.py,sha256=ht_HWzZt84IEogoFMggnXI6aFFerrsuksVflAkcodfU,5545
 autocoder/dispacher/actions/plugins/action_translate.py,sha256=nVAtRSQpdGNmZxg1R_9zXG3AuTv3CHf2v7ODgj8u65c,7727
 autocoder/index/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/index/entry.py,sha256=KJaxqtaKgL27w8-j7OiAqI0anPpmrJSl7PkfeVF2ipE,11713
+autocoder/index/entry.py,sha256=hXSOi-jPgRBKQB55eqWkT95vxvWrbVHDuIMtDkqjNNw,12232
 autocoder/index/for_command.py,sha256=BFvljE4t6VaMBGboZAuhUCzVK0EitCy_n5D_7FEnihw,3204
-autocoder/index/index.py,sha256=VjfcBYHywU4tjQTA7mpHfzRM8nBPhPHrUnkuBbsj6do,20409
+autocoder/index/index.py,sha256=xwh22nY0TtEJMJwYjOUd6xdRZozYXBt47YSwheZSP-4,20679
 autocoder/index/symbols_utils.py,sha256=CjcjUVajmJZB75Ty3a7kMv1BZphrm-tIBAdOJv6uo-0,2037
 autocoder/index/types.py,sha256=a2s_KV5FJlq7jqA2ELSo9E1sjuLwDB-JJYMhSpzBAhU,596
 autocoder/index/filter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/index/filter/normal_filter.py,sha256=pE5QwcBq6NYHFtYhwhfMJmYQYJwErNs-Q7iZmVBAh-k,7964
-autocoder/index/filter/quick_filter.py,sha256=Po32nJFAbAwb7kMaNkCrL5-ZjVE-Pobm5wzXyw9Y8iE,3882
+autocoder/index/filter/normal_filter.py,sha256=APu34iSvWhtlLtWgkj8N3Vo4oW1TegtZQq2bwDX_cs4,8031
+autocoder/index/filter/quick_filter.py,sha256=Omvsz9O1xQEH4xP-wNuCZhxn69P7Y59SiLPUIDuGFiA,3851
 autocoder/pyproject/__init__.py,sha256=dQ2_7YZ7guybT9BhfxSGn43eLQJGQN2zgeKa6--JlaQ,14403
 autocoder/rag/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/rag/api_server.py,sha256=dRbhAZVRAOlZ64Cnxf4_rKb4iJwHnrWS9Zr67IVORw0,7288
@@ -120,7 +121,7 @@ autocoder/rag/stream_event/types.py,sha256=rtLwOE8rShmi1dJdxyBpAV5ZjLBGG9vptMiSz
 autocoder/regex_project/__init__.py,sha256=EBZeCL5ORyD_9_5u_UuG4s7XtpXOu0y1sWDmxWFtufE,6781
 autocoder/regexproject/__init__.py,sha256=cEr-ZOaQjLD5sx7T7F2DhD5ips03HcJ02rded9EpSXc,9693
 autocoder/suffixproject/__init__.py,sha256=VcXjUbGf3uQrpoqVCItDvGG9DoeHJ_qEmghKwrVNw9w,11058
-autocoder/tsproject/__init__.py,sha256=yloVzkGLnbTd4Hcj9fMO-rcjNTTx4wI3Ga41LWOSYrY,11747
+autocoder/tsproject/__init__.py,sha256=boNuRCHi94xI_y4tvL5LKgSZ4gYxcPqUUQTw9MU_STI,11751
 autocoder/utils/__init__.py,sha256=KtcGElFNBgZPF7dEL8zF9JpXkCAjoyDrzaREJBhJrcs,994
 autocoder/utils/_markitdown.py,sha256=RU88qn4eZfYIy0GDrPxlI8oYXIypbi63VRJjdlnE0VU,47431
 autocoder/utils/coder.py,sha256=rK8e0svQBe0NOP26dIGToUXgha_hUDgxlWoC_p_r7oc,5698
@@ -138,11 +139,11 @@ autocoder/utils/rest.py,sha256=hLBhr78y-WVnV0oQf9Rxc22EwqF78KINkScvYa1MuYA,6435
 autocoder/utils/tests.py,sha256=BqphrwyycGAvs-5mhH8pKtMZdObwhFtJ5MC_ZAOiLq8,1340
 autocoder/utils/types.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/utils/auto_coder_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/utils/auto_coder_utils/chat_stream_out.py,sha256=yzh3QYJdYC1CPcdhSTn0fmFHtqV8iSXeH0drV3HjN5A,9053
+autocoder/utils/auto_coder_utils/chat_stream_out.py,sha256=64WfP5rFKYwI9OXrpRTQjUQa9n6ULTscubKziot-rAU,9218
 autocoder/utils/chat_auto_coder_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-auto_coder-0.1.244.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-auto_coder-0.1.244.dist-info/METADATA,sha256=EVj3424BKEVPU6GAnoHC-NQ4J2VQFT7kVt3ANVby28A,2616
-auto_coder-0.1.244.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-auto_coder-0.1.244.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
-auto_coder-0.1.244.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
-auto_coder-0.1.244.dist-info/RECORD,,
+auto_coder-0.1.246.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+auto_coder-0.1.246.dist-info/METADATA,sha256=zENdZdSfMKFmZWd-N1GIOXfm8xWnBv0nG-s9I3UpZqY,2616
+auto_coder-0.1.246.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+auto_coder-0.1.246.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
+auto_coder-0.1.246.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
+auto_coder-0.1.246.dist-info/RECORD,,

autocoder/agent/auto_review_commit.py

@@ -0,0 +1,207 @@
+from typing import Generator, List, Dict, Union, Tuple, Optional
+import os
+import yaml
+import byzerllm
+import pydantic
+import git
+from rich.console import Console
+from autocoder.utils.auto_coder_utils.chat_stream_out import stream_out
+from autocoder.common.printer import Printer
+from autocoder.common import AutoCoderArgs
+from autocoder.common.utils_code_auto_generate import stream_chat_with_continue
+
+
+def load_yaml_config(yaml_file: str) -> Dict:
+    """加载YAML配置文件"""
+    try:
+        with open(yaml_file, 'r', encoding='utf-8') as f:
+            return yaml.safe_load(f)
+    except Exception as e:
+        printer = Printer()
+        printer.print_in_terminal("yaml_load_error", style="red", yaml_file=yaml_file, error=str(e))
+        return {}
+
+
+class AutoReviewCommit:
+    def __init__(self, llm: Union[byzerllm.ByzerLLM,byzerllm.SimpleByzerLLM],
+                 args:AutoCoderArgs,                 
+                 skip_diff: bool = False,                 
+                 console: Optional[Console] = None):
+        """
+        初始化 AutoReviewCommit
+
+        Args:
+            llm: ByzerLLM 实例，用于代码审查
+            project_dir: 项目根目录
+            skip_diff: 是否跳过获取 diff 信息            
+        """
+        self.project_dir = args.source_dir
+        self.actions_dir = os.path.join(args.source_dir, "actions")
+        self.llm = llm        
+        self.skip_diff = skip_diff
+        self.console = console or Console()
+
+    @byzerllm.prompt()
+    def review(self, querie_with_urls_and_diffs: List[Tuple[str, List[str], str]], query: str) -> Generator[str,None,None]:
+        """
+        如果前面我们对话提供了文档，请参考上面的文档对提交的代码变更进行审查，提供改进建议。
+
+        下面包含最新一次提交的信息：        
+        <commit>
+        {% for query,urls,diff in querie_with_urls_and_diffs %}
+        ## 任务需求
+        {{ query }}
+
+        修改的文件:
+        {% for url in urls %}
+        - {{ url }}
+        {% endfor %}
+
+        代码变更:
+        ```diff
+        {{ diff }}
+        ```
+        {% endfor %}
+        </commit>
+
+        审查要求：
+        1. 代码质量评估
+           - 代码可读性：命名、注释、代码结构是否清晰
+           - 代码风格：是否符合项目规范
+           - 实现逻辑：算法和数据结构的选择是否合适
+           
+        2. 潜在问题检查
+           - 常见错误：是否存在空指针，数组越界，类型转换，未声明变量,typo等低级错误
+           - 安全性：是否存在安全隐患
+           - 性能：是否有性能问题
+           - 并发：是否有并发安全问题
+           - 异常处理：错误处理是否完善
+           - 资源管理：是否有资源泄露风险
+           
+        3. 架构合理性
+           - 模块化：职责划分是否合理
+           - 可扩展性：是否方便未来扩展
+           - 依赖关系：组件耦合是否合理
+           - 复用性：是否有重复代码
+
+        返回格式说明：
+        返回 markdown 格式，包含以下内容：
+        1. issues: 发现的具体问题列表
+        2. suggestions: 对应的改进建议列表
+        3. severity: 问题的严重程度(low/medium/high)
+        4. affected_files: 受影响的文件列表
+        5. summary: 总体评价
+
+        {% if query %}
+        用户额外 review 需求：
+        <user_review_requirement>
+        {{ query }}
+        </user_review_requirement>
+        {% endif %}
+
+        注意：
+        1. 评审意见应该具体且可操作，而不是泛泛而谈
+        2. 对于每个问题都应该提供明确的改进建议
+        3. 严重程度的判断要考虑问题对系统的潜在影响
+        4. 建议应该符合项目的技术栈和开发规范        
+        """
+        pass
+
+
+    def parse_history_tasks(self) -> List[Dict]:
+        """
+        解析历史任务信息
+
+        Returns:
+            List[Dict]: 每个字典包含一个历史任务的信息
+        """
+        # 获取所有YAML文件
+        action_files = [
+            f for f in os.listdir(self.actions_dir)
+            if f[:3].isdigit() and "_" in f and f.endswith('.yml')
+        ]
+
+        # 按序号排序
+        def get_seq(name):
+            return int(name.split("_")[0])
+
+        # 获取最新的action文件列表
+        action_files = sorted(action_files, key=get_seq)
+        action_files.reverse()        
+
+        action_file = action_files[0]
+
+        querie_with_urls_and_diffs = []
+        repo = git.Repo(self.project_dir)
+
+        # 收集所有query、urls和对应的commit diff
+        for yaml_file in [action_file]:
+            yaml_path = os.path.join(self.actions_dir, yaml_file)
+            config = load_yaml_config(yaml_path)
+
+            if not config:
+                continue
+
+            query = config.get('query', '')
+            urls = config.get('urls', [])
+
+            if query and urls:
+                commit_diff = ""
+                if not self.skip_diff:
+                    # 计算文件的MD5用于匹配commit
+                    import hashlib
+                    file_md5 = hashlib.md5(open(yaml_path, 'rb').read()).hexdigest()
+                    response_id = f"auto_coder_{yaml_file}_{file_md5}"
+                    # 查找对应的commit                   
+                    try:
+                        for commit in repo.iter_commits():
+                            if response_id in commit.message:
+                                if commit.parents:
+                                    parent = commit.parents[0]
+                                    commit_diff = repo.git.diff(
+                                        parent.hexsha, commit.hexsha)
+                                else:
+                                    commit_diff = repo.git.show(commit.hexsha)
+                                break
+                    except git.exc.GitCommandError as e:
+                        printer = Printer()
+                        printer.print_in_terminal("git_command_error", style="red", error=str(e))
+                    except Exception as e:
+                        printer = Printer()
+                        printer.print_in_terminal("get_commit_diff_error", style="red", error=str(e))
+
+                querie_with_urls_and_diffs.append((query, urls, commit_diff))
+
+        return querie_with_urls_and_diffs
+    
+
+    def review_commit(self,query: str, conversations: List[Dict]) -> Generator[str,None,None]:
+        """
+        审查最新的代码提交
+
+        Returns:
+            Optional[ReviewResult]: 审查结果，如果出错则返回None
+        """
+        printer = Printer()
+        # 获取最新的提交信息
+        commits = self.parse_history_tasks()
+        if not commits:            
+            printer.print_in_terminal("no_latest_commit", style="red")
+            return None
+
+        # 调用LLM进行代码审查
+        try:
+            # 获取 prompt 内容            
+            query = self.review.prompt(commits, query)
+            new_conversations = conversations.copy()[0:-1]
+            new_conversations.append({"role": "user", "content": query})
+            # 构造对话消息            
+            v = stream_chat_with_continue(
+                    llm=self.llm,
+                    conversations=new_conversations,
+                    llm_config={}
+            )
+            return v
+        except Exception as e:            
+            printer.print_in_terminal("code_review_error", style="red", error=str(e))
+            return None

autocoder/auto_coder.py

@@ -278,8 +278,13 @@ def main(input_args: Optional[List[str]] = None):
 
             llm = byzerllm.ByzerLLM(verbose=args.print_request)
 
+            # code_model,index_filter_model,generate_rerank_model,chat_model
+            # 这四个模型如果用户没有设置，就会使用默认的
+            # 如果用户随便填写 deepseek 官方key,就会导致 Authentic(No User) 的错误
+            # 或者 Insuffient Balance 之类的错误
+
             code_model = byzerllm.ByzerLLM()
-            code_model.setup_default_model_name("deepseek_chat")
+            code_model.setup_default_model_name(args.model)
             llm.setup_sub_client("code_model", code_model)
 
             index_filter_model = byzerllm.ByzerLLM()
@@ -295,84 +300,66 @@ def main(input_args: Optional[List[str]] = None):
             llm.setup_sub_client("chat_model", chat_model)
         
         if args.product_mode == "lite":
-            llm = byzerllm.SimpleByzerLLM(default_model_name="deepseek_chat")
-            api_key_dir = os.path.expanduser("~/.auto-coder/keys")
-            api_key_file = os.path.join(api_key_dir, "api.deepseek.com")
-            
-            if not os.path.exists(api_key_file):                
-                raise Exception(f"API key file not found: {api_key_file}")
-            
-            with open(api_key_file, "r") as f:
-                api_key = f.read()
-            
+            default_model = args.model        
+            model_info = models_module.get_model_by_name(default_model)                
+            llm = byzerllm.SimpleByzerLLM(default_model_name=default_model)
             llm.deploy(
                 model_path="",
-                pretrained_model_type="saas/openai",
-                udf_name="deepseek_chat",
+                pretrained_model_type=model_info["model_type"],
+                udf_name=args.model,
                 infer_params={
-                    "saas.base_url": "https://api.deepseek.com/v1",
-                    "saas.api_key": api_key,
-                    "saas.model": "deepseek-chat",
-                    "saas.is_reasoning": False
+                    "saas.base_url": model_info["base_url"],
+                    "saas.api_key": model_info["api_key"],
+                    "saas.model": model_info["model_name"],
+                    "saas.is_reasoning": model_info["is_reasoning"]
                 }
-            )
-            
-            code_llm = byzerllm.SimpleByzerLLM(default_model_name="deepseek_chat")
-            code_llm.deploy(
-                model_path="",
-                pretrained_model_type="saas/openai",
-                udf_name="deepseek_chat",
-                infer_params={
-                    "saas.base_url": "https://api.deepseek.com/v1",
-                    "saas.api_key": api_key,
-                    "saas.model": "deepseek-chat",
-                    "saas.is_reasoning": False
-                }
-            )
-
-            chat_llm = byzerllm.SimpleByzerLLM(default_model_name="deepseek_r1_chat")
-            chat_llm.deploy(
-                model_path="",
-                pretrained_model_type="saas/openai",
-                udf_name="deepseek_r1_chat",
-                infer_params={
-                    "saas.base_url": "https://api.deepseek.com/v1",
-                    "saas.api_key": api_key,
-                    "saas.model": "deepseek-reasoner",
-                    "saas.is_reasoning": True
-                }
-            )
-
-            generate_rerank_llm = byzerllm.SimpleByzerLLM(default_model_name="deepseek_r1_chat")
-            generate_rerank_llm.deploy(
-                model_path="",
-                pretrained_model_type="saas/openai",
-                udf_name="deepseek_r1_chat",
-                infer_params={
-                    "saas.base_url": "https://api.deepseek.com/v1",
-                    "saas.api_key": api_key,
-                    "saas.model": "deepseek-reasoner",
-                    "saas.is_reasoning": True
-                }
-            )
+            )                
+                        
+            if models_module.check_model_exists("deepseek_r1_chat"):
+                r1_model_info = models_module.get_model_by_name("deepseek_r1_chat")  
+                api_key = r1_model_info["api_key"]
+                chat_llm = byzerllm.SimpleByzerLLM(default_model_name="deepseek_r1_chat")
+                chat_llm.deploy(
+                    model_path="",
+                    pretrained_model_type="saas/openai",
+                    udf_name="deepseek_r1_chat",
+                    infer_params={
+                        "saas.base_url": "https://api.deepseek.com/v1",
+                        "saas.api_key": api_key,
+                        "saas.model": "deepseek-reasoner",
+                        "saas.is_reasoning": True
+                    }
+                )
 
-            index_filter_llm = byzerllm.SimpleByzerLLM(default_model_name="deepseek_r1_chat")
-            index_filter_llm.deploy(
-                model_path="",
-                pretrained_model_type="saas/openai",
-                udf_name="deepseek_r1_chat",
-                infer_params={
-                    "saas.base_url": "https://api.deepseek.com/v1",
-                    "saas.api_key": api_key,
-                    "saas.model": "deepseek-reasoner",
-                    "saas.is_reasoning": True
-                }
-            )
+                generate_rerank_llm = byzerllm.SimpleByzerLLM(default_model_name="deepseek_r1_chat")
+                generate_rerank_llm.deploy(
+                    model_path="",
+                    pretrained_model_type="saas/openai",
+                    udf_name="deepseek_r1_chat",
+                    infer_params={
+                        "saas.base_url": "https://api.deepseek.com/v1",
+                        "saas.api_key": api_key,
+                        "saas.model": "deepseek-reasoner",
+                        "saas.is_reasoning": True
+                    }
+                )
 
-            llm.setup_sub_client("code_model", code_llm)
-            llm.setup_sub_client("chat_model", chat_llm)
-            llm.setup_sub_client("generate_rerank_model", generate_rerank_llm)
-            llm.setup_sub_client("index_filter_model", index_filter_llm)
+                index_filter_llm = byzerllm.SimpleByzerLLM(default_model_name="deepseek_r1_chat")
+                index_filter_llm.deploy(
+                    model_path="",
+                    pretrained_model_type="saas/openai",
+                    udf_name="deepseek_r1_chat",
+                    infer_params={
+                        "saas.base_url": "https://api.deepseek.com/v1",
+                        "saas.api_key": api_key,
+                        "saas.model": "deepseek-reasoner",
+                        "saas.is_reasoning": True
+                    }
+                )
+                                                
+                llm.setup_sub_client("chat_model", chat_llm)
+                llm.setup_sub_client("generate_rerank_model", generate_rerank_llm)
+                llm.setup_sub_client("index_filter_model", index_filter_llm)
 
         if args.product_mode == "lite":                                    
             # Set up default models based on configuration
@@ -1349,6 +1336,10 @@ def main(input_args: Optional[List[str]] = None):
                     )
                 )
                 v = [[response.result,None]]
+            elif "review_commit" in args.action:
+                from autocoder.agent.auto_review_commit import AutoReviewCommit
+                reviewer = AutoReviewCommit(llm=chat_llm, args=args)
+                v = reviewer.review_commit(args.query)
             else:                
                 v = stream_chat_with_continue(
                     llm=chat_llm,
@@ -1356,12 +1347,16 @@ def main(input_args: Optional[List[str]] = None):
                     llm_config={}
                 )              
 
-                
+
+            model_name = getattr(chat_llm, 'default_model_name', None)
+            if not model_name:
+                model_name = "unknown(without default model name)"    
             
             assistant_response, last_meta = stream_out(
                     v, 
                     request_id=args.request_id,                    
-                    console=console
+                    console=console,
+                    model_name=model_name
                 )
                                                                        
             # 打印耗时和token统计            

autocoder/chat_auto_coder.py

@@ -1937,14 +1937,18 @@ def chat(query: str):
     if "/save" in query:
         yaml_config["action"].append("save")
         query = query.replace("/save", "", 1).strip()        
-
-    is_review = query.strip().startswith("/review")
-    if is_review:
-        query = query.replace("/review", "", 1).strip()
-        if "prompt_review" in conf:
-            query = format_str_jinja2(conf["prompt_review"], query=query)
-        else:
-            query = code_review.prompt(query)
+    
+    if "/review" in query and "/commit" in query:
+        yaml_config["action"].append("review_commit")
+        query = query.replace("/review", "", 1).replace("/commit", "", 1).strip()
+    else:    
+        is_review = query.strip().startswith("/review")
+        if is_review:
+            query = query.replace("/review", "", 1).strip()
+            if "prompt_review" in conf:
+                query = format_str_jinja2(conf["prompt_review"], query=query)
+            else:
+                query = code_review.prompt(query)
 
     is_no_context = query.strip().startswith("/no_context")
     if is_no_context:
@@ -2194,13 +2198,14 @@ def manage_models(params, query: str):
             table.add_column("Base URL", style="white", width=50, overflow="fold")
             for m in models_data:
                 # Check if api_key_path exists and file exists
-                api_key_path = m.get("api_key_path", "")
-                name = m.get("name", "")
-                if api_key_path:
-                    api_key_file = os.path.expanduser(f"~/.auto-coder/keys/{api_key_path}")
-                    if os.path.exists(api_key_file):
-                        name = f"{name}*"                        
-            
+                is_api_key_set = "api_key" in m  
+                name = m.get("name", "")              
+                if is_api_key_set:
+                    api_key = m.get("api_key", "").strip()                    
+                    if not api_key:                                                
+                        printer.print_in_terminal("models_api_key_empty", style="yellow", name=name)                                           
+                    name = f"{name} *"
+                
                 table.add_row(
                     name,                    
                     m.get("model_name", ""),                    
@@ -2700,6 +2705,12 @@ def main():
             if "mode" not in memory:
                 memory["mode"] = "normal"
 
+            # 处理 user_input 的空格
+            if user_input:
+                temp_user_input = user_input.lstrip()  # 去掉左侧空格
+                if temp_user_input.startswith('/'):
+                    user_input = temp_user_input
+
             if (
                 memory["mode"] == "auto_detect"
                 and user_input

autocoder/chat_auto_coder_lang.py

@@ -111,7 +111,8 @@ MESSAGES = {
         "remove_files_all": "Removed all files.",
         "remove_files_removed": "Removed Files",
         "remove_files_none": "No files were removed.",
-        "files_removed": "Files Removed"
+        "files_removed": "Files Removed",
+        "models_api_key_empty": "Warning : {{name}} API key is empty. Please set a valid API key.",
     },
     "zh": {
         "mcp_remove_error": "移除 MCP 服务器时出错：{error}",
@@ -222,7 +223,8 @@ MESSAGES = {
         "remove_files_all": "已移除所有文件。",
         "remove_files_removed": "已移除的文件",
         "remove_files_none": "没有文件被移除。",
-        "files_removed": "移除的文件"
+        "files_removed": "移除的文件",
+        "models_api_key_empty": "警告:  {{name}}  API key 为空。请设置一个有效的 API key。",
     }
 }
 

autocoder/common/auto_coder_lang.py

@@ -8,9 +8,14 @@ MESSAGES = {
         "memory_save_success": "✅ Saved to your memory",
         "file_decode_error": "Failed to decode file: {{file_path}}. Tried encodings: {{encodings}}",
         "file_write_error": "Failed to write file: {{file_path}}. Error: {{error}}",
+        "yaml_load_error": "Error loading yaml file {{yaml_file}}: {{error}}",
+        "git_command_error": "Git command execution error: {{error}}",
+        "get_commit_diff_error": "Error getting commit diff: {{error}}",
+        "no_latest_commit": "Unable to get latest commit information",
+        "code_review_error": "Code review process error: {{error}}",
         "index_file_too_large": "⚠️ File {{ file_path }} is too large ({{ file_size }} > {{ max_length }}), splitting into chunks...",
-        "index_update_success": "✅ Successfully updated index for {{ file_path }} (md5: {{ md5 }}) in {{ duration }}s",
-        "index_build_error": "❌ Error building index for {{ file_path }}: {{ error }}",
+        "index_update_success": "✅ {{ model_name }} Successfully updated index for {{ file_path }} (md5: {{ md5 }}) in {{ duration }}s",
+        "index_build_error": "❌ {{ model_name }} Error building index for {{ file_path }}: {{ error }}",
         "index_build_summary": "📊 Total Files: {{ total_files }}, Need to Build Index: {{ num_files }}",
         "building_index_progress": "⏳ Building Index: {{ counter }}/{{ num_files }}...",
         "index_source_dir_mismatch": "⚠️ Source directory mismatch (file_path: {{ file_path }}, source_dir: {{ source_dir }})",
@@ -49,7 +54,7 @@ MESSAGES = {
         "code_merge_start": "Auto merge the code...",
         "code_execution_warning": "Content(send to model) is {{ content_length }} tokens (you may collect too much files), which is larger than the maximum input length {{ max_length }}",
         "quick_filter_start": "Starting filter context(quick_filter)...",
-        "normal_filter_start": "Starting filter context(normal_filter)...",
+        "normal_filter_start": "{{ model_name }} Starting filter context(normal_filter)...",
         "pylint_check_failed": "⚠️ Pylint check failed: {{ error_message }}",
         "pylint_error": "❌ Error running pylint: {{ error_message }}",
         "unmerged_blocks_warning": "⚠️ Found {{ num_blocks }} unmerged blocks, the changes will not be applied. Please review them manually then try again.",
@@ -82,9 +87,14 @@ MESSAGES = {
         "memory_save_success": "✅ 已保存到您的记忆中",
         "file_decode_error": "无法解码文件: {{file_path}}。尝试的编码: {{encodings}}",
         "file_write_error": "无法写入文件: {{file_path}}. 错误: {{error}}",
+        "yaml_load_error": "加载YAML文件出错 {{yaml_file}}: {{error}}",
+        "git_command_error": "Git命令执行错误: {{error}}",
+        "get_commit_diff_error": "获取commit diff时出错: {{error}}",
+        "no_latest_commit": "无法获取最新的提交信息",
+        "code_review_error": "代码审查过程出错: {{error}}",
         "index_file_too_large": "⚠️ 文件 {{ file_path }} 过大 ({{ file_size }} > {{ max_length }}), 正在分块处理...",
-        "index_update_success": "✅ 成功更新 {{ file_path }} 的索引 (md5: {{ md5 }}), 耗时 {{ duration }} 秒",
-        "index_build_error": "❌ 构建 {{ file_path }} 索引时出错: {{ error }}",
+        "index_update_success": "✅ {{ model_name }} 成功更新 {{ file_path }} 的索引 (md5: {{ md5 }}), 耗时 {{ duration }} 秒",
+        "index_build_error": "❌ {{ model_name }} 构建 {{ file_path }} 索引时出错: {{ error }}",
         "index_build_summary": "📊 总文件数: {{ total_files }}, 需要构建索引: {{ num_files }}",
         "building_index_progress": "⏳ 正在构建索引: {{ counter }}/{{ num_files }}...",
         "index_source_dir_mismatch": "⚠️ 源目录不匹配 (文件路径: {{ file_path }}, 源目录: {{ source_dir }})",
@@ -123,7 +133,7 @@ MESSAGES = {
         "code_merge_start": "正在自动合并代码...",
         "code_execution_warning": "发送给模型的内容长度为 {{ content_length }} tokens（您可能收集了太多文件），超过了最大输入长度 {{ max_length }}",
         "quick_filter_start": "开始查找上下文(quick_filter)...",
-        "normal_filter_start": "开始查找上下文(normal_filter)...",
+        "normal_filter_start": "{{ model_name }} 开始查找上下文(normal_filter)...",
         "pylint_check_failed": "⚠️ Pylint 检查失败: {{ error_message }}",
         "pylint_error": "❌ 运行 Pylint 时出错: {{ error_message }}",
         "begin_index_source_code": "🚀 开始为 {{ source_dir }} 中的源代码建立索引",
@@ -132,7 +142,7 @@ MESSAGES = {
         "merge_success": "✅ 成功合并了 {{ num_files }} 个文件中的更改 {{ num_changes }}/{{ total_blocks }} 个代码块。",
         "no_changes_made": "⚠️ 未对任何文件进行更改。",
         "unmerged_blocks_title": "未合并代码块",
-        "unmerged_file_path": "文件: {file_path}",
+        "unmerged_file_path": "文件: {{file_path}}",
         "unmerged_search_block": "Search Block({{similarity}}):",
         "unmerged_replace_block": "Replace Block:",
         "unmerged_blocks_total": "未合并代码块数量: {{num_blocks}}",

autocoder/index/entry.py

@@ -100,14 +100,20 @@ def build_index_and_filter_files(
                     })
                 )
             )
+
+        if not args.skip_filter_index and args.index_filter_model:
+            model_name = getattr(index_manager.index_filter_llm, 'default_model_name', None)
+            if not model_name:
+                model_name = "unknown(without default model name)"    
+            printer.print_in_terminal("quick_filter_start", style="blue", model_name=model_name)
+            quick_filter = QuickFilter(index_manager,stats,sources)        
+            final_files = quick_filter.filter(index_manager.read_index(),args.query)
         
-        #MARK
-        printer.print_in_terminal("quick_filter_start", style="blue")
-        quick_filter = QuickFilter(index_manager,stats,sources)        
-        final_files = quick_filter.filter(index_manager.read_index(),args.query)
-        
-        if not final_files:
-            printer.print_in_terminal("normal_filter_start", style="blue")
+        if not args.skip_filter_index and not args.index_filter_model:
+            model_name = getattr(index_manager.llm, 'default_model_name', None)
+            if not model_name:
+                model_name = "unknown(without default model name)" 
+            printer.print_in_terminal("normal_filter_start", style="blue",model_name=model_name)
             normal_filter = NormalFilter(index_manager,stats,sources)
             final_files = normal_filter.filter(index_manager.read_index(),args.query)
         

autocoder/index/filter/normal_filter.py

@@ -34,6 +34,7 @@ class NormalFilter():
         self.sources = sources
 
     def filter(self, index_items: List[IndexItem], query: str) -> Dict[str, TargetFile]:
+        
         final_files: Dict[str, TargetFile] = {}
         if not self.args.skip_filter_index:
             if self.args.request_id and not self.args.skip_events:
@@ -59,7 +60,7 @@ class NormalFilter():
             phase_end = time.monotonic()
             self.stats["timings"]["normal_filter"]["level1_filter"] = phase_end - phase_start
 
-            # Phase 4: Level 2 filtering - Related files
+            # Phase 4: Level 2 filtering - Related files                        
             if target_files is not None and self.args.index_filter_level >= 2:
                 logger.info(
                     "Phase 4: Performing Level 2 filtering (related files)...")
@@ -83,13 +84,14 @@ class NormalFilter():
                 phase_end = time.monotonic()
                 self.stats["timings"]["normal_filter"]["level2_filter"] = phase_end - phase_start
 
-            if not final_files:
-                logger.warning("No related files found, using all files")
-                for source in self.sources:
-                    final_files[get_file_path(source.module_name)] = TargetFile(
-                        file_path=source.module_name,
-                        reason="No related files found, use all files",
-                    )
+            # if not final_files:
+            #     logger.warning("No related files found, using all files")
+            #     for source in self.sources:
+            #         final_files[get_file_path(source.module_name)] = TargetFile(
+            #             file_path=source.module_name,
+            #             reason="No related files found, use all files",
+            #         )
+
 
             # Phase 5: Relevance verification
             logger.info("Phase 5: Performing relevance verification...")
@@ -125,7 +127,7 @@ class NormalFilter():
                         if source.module_name == file.file_path:
                             file_content = source.source_code
                             try:
-                                result = self.index_manager.verify_file_relevance.with_llm(llm).with_return_type(VerifyFileRelevance).run(
+                                result = self.index_manager.verify_file_relevance.with_llm(self.index_manager.llm).with_return_type(VerifyFileRelevance).run(
                                     file_content=file_content,
                                     query=self.args.query
                                 )

autocoder/index/filter/quick_filter.py

@@ -69,7 +69,7 @@ class QuickFilter():
 
     def filter(self, index_items: List[IndexItem], query: str) -> Dict[str, TargetFile]:
         final_files: Dict[str, TargetFile] = {}
-        if not self.args.skip_filter_index and self.index_manager.llm.get_sub_client("index_filter_model"):
+        if not self.args.skip_filter_index and self.args.index_filter_model:
             start_time = time.monotonic()
             index_items = self.index_manager.read_index()
 

autocoder/index/index.py

@@ -195,7 +195,7 @@ class IndexManager:
             return True
         return False
 
-    def build_index_for_single_source(self, source: SourceCode):
+    def build_index_for_single_source(self, source: SourceCode):        
         file_path = source.module_name
         if not os.path.exists(file_path):
             return None
@@ -205,6 +205,10 @@ class IndexManager:
 
         md5 = hashlib.md5(source.source_code.encode("utf-8")).hexdigest()
 
+        model_name = getattr(self.index_llm, 'default_model_name', None)
+        if not model_name:
+            model_name = "unknown(without default model name)"
+
         try:
             start_time = time.monotonic()
             source_code = source.source_code
@@ -230,13 +234,14 @@ class IndexManager:
                 symbols = self.get_all_file_symbols.with_llm(
                     self.index_llm).run(source.module_name, source_code)
                 time.sleep(self.anti_quota_limit)
-
+            
             self.printer.print_in_terminal(
                 "index_update_success",
                 style="green",
                 file_path=file_path,
                 md5=md5,
-                duration=time.monotonic() - start_time
+                duration=time.monotonic() - start_time,
+                model_name=model_name
             )
 
         except Exception as e:
@@ -246,7 +251,8 @@ class IndexManager:
                 "index_build_error",
                 style="red",
                 file_path=file_path,
-                error=str(e)
+                error=str(e),
+                model_name=model_name
             )
             return None
 
@@ -464,7 +470,7 @@ class IndexManager:
             {file.file_path: file for file in all_results}.values())
         return FileList(file_list=all_results)
 
-    def _query_index_with_thread(self, query, func):
+    def _query_index_with_thread(self, query, func):        
         all_results = []
         lock = threading.Lock()
         completed_threads = 0

autocoder/models.py

@@ -2,6 +2,7 @@ import os
 import json
 from typing import List, Dict
 from urllib.parse import urlparse
+from autocoder.common.auto_coder_lang import get_message_with_format
 
 MODELS_JSON = os.path.expanduser("~/.auto-coder/keys/models.json")
 
@@ -108,11 +109,18 @@ def get_model_by_name(name: str) -> Dict:
     models = load_models()
     v = [m for m in models if m["name"] == name.strip()]
     
-    if len(v) == 0:
-        from autocoder.common.auto_coder_lang import get_message_with_format
+    if len(v) == 0:        
         raise Exception(get_message_with_format("model_not_found", model_name=name))
     return v[0]
 
+
+def check_model_exists(name: str) -> bool:
+    """
+    检查模型是否存在
+    """
+    models = load_models()
+    return any(m["name"] == name.strip() for m in models)
+
 def update_model_with_api_key(name: str, api_key: str) -> Dict:
     """
     根据模型名称查找并更新模型的 api_key_path。

autocoder/tsproject/__init__.py

@@ -1,5 +1,5 @@
 from autocoder.common import SourceCode, AutoCoderArgs
-from autocoder import common as FileUtils
+from autocoder import common as CommonUtils
 from autocoder.utils.rest import HttpDoc
 import os
 from typing import Optional, Generator, List, Dict, Any
@@ -164,7 +164,7 @@ class TSProject:
             logger.warning(f"Failed to read file: {file_path}. Error: {str(e)}")
             return None
 
-        if not FileUtils.has_sufficient_content(source_code, min_line_count=1):
+        if not CommonUtils.has_sufficient_content(source_code, min_line_count=1):
             return None
 
         return SourceCode(module_name=module_name, source_code=source_code)

autocoder/utils/auto_coder_utils/chat_stream_out.py

@@ -141,7 +141,8 @@ def multi_stream_out(
 def stream_out(
     stream_generator: Generator[Tuple[str, Dict[str, Any]], None, None],
     request_id: Optional[str] = None,    
-    console: Optional[Console] = None
+    console: Optional[Console] = None,
+    model_name: Optional[str] = None
 ) -> Tuple[str, Optional[Dict[str, Any]]]:
     """
     处理流式输出事件并在终端中展示
@@ -164,7 +165,7 @@ def stream_out(
     
     try:
         with Live(
-            Panel("", title="Response", border_style="green"),
+            Panel("", title=f"Response[ {model_name} ]", border_style="green"),
             refresh_per_second=4,
             console=console
         ) as live:
@@ -209,7 +210,7 @@ def stream_out(
                 live.update(
                     Panel(
                         Markdown(display_content),
-                        title="Response",
+                        title=f"Response[ {model_name} ]",
                         border_style="green",
                         height=min(50, live.console.height - 4)
                     )
@@ -223,7 +224,7 @@ def stream_out(
             live.update(
                 Panel(
                     Markdown(assistant_response),
-                    title="Final Response",
+                    title=f"Final Response[ {model_name} ]",
                     border_style="blue"
                 )
             )
@@ -231,9 +232,11 @@ def stream_out(
     except Exception as e:
         console.print(Panel(
             f"Error: {str(e)}",  
-            title="Error",
+            title=f"Error[ {model_name} ]",
             border_style="red"
         ))
+        # import traceback
+        # traceback.print_exc()
         
         if request_id and request_queue:
             request_queue.add_request(

autocoder/version.py

@@ -1 +1 @@
-__version__ = "0.1.244"
+__version__ = "0.1.246"