auto-coder 0.1.243__py3-none-any.whl → 0.1.245__py3-none-any.whl

{auto_coder-0.1.243.dist-info → auto_coder-0.1.245.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.243
+Version: 0.1.245
 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.159
+Requires-Dist: byzerllm[saas] >=0.1.161
 Requires-Dist: patch
 Requires-Dist: diff-match-patch
 Requires-Dist: GitPython

{auto_coder-0.1.243.dist-info → auto_coder-0.1.245.dist-info}/RECORD

@@ -1,21 +1,22 @@
 autocoder/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/auto_coder.py,sha256=iQeyrg5L87IgBLFJLZLIJetHFeMuT2uslRlzo_Vjq9s,61829
+autocoder/auto_coder.py,sha256=832zmnXmZ7RlqsPnuEj-2IuhsJI0QkCGHJICH6QRCAY,63333
 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=s9uMjDQQawXEsq171GO7SVMO4fDjWd4xWT0KYO7nRp4,105295
-autocoder/chat_auto_coder_lang.py,sha256=V-VIieyKF5cwlK448B1V2LUbTdrU03tfgDrOk2aBvFk,14891
+autocoder/chat_auto_coder.py,sha256=SLJQzXuQoj_2mcdbANG93ZM2-wIQsptA_h3VRv5xZAQ,105926
+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=FlBrF6HhGao_RiCSgYhCmP7vs0KlG4hI_BI6dyZiL9s,5292
-autocoder/version.py,sha256=u0hWeuFclX3Z9nFe5oFsCdX854VeUHOs69Ggv1pvBvk,23
+autocoder/models.py,sha256=7Z97Hzc_26dZG_wm6M2f9TL1ZxzzIN649U_Z0-m28EU,5342
+autocoder/version.py,sha256=sncGdxYQvG5ZX4oQL4xUEZ96_LS3u1YIOVwACJUeMF4,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=9FBNhcl6Do4ICh-klevYsCTsDuy5kD99r8EE5Gs1QoM,12592
+autocoder/common/auto_coder_lang.py,sha256=7sr3Dz43ASeWYLtMvkE6tMd8dWDPZBJOxNVQ8rC54Js,13963
 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
@@ -46,7 +47,7 @@ autocoder/common/command_completer.py,sha256=SSeb8MDH0JPvfdyW-S2uaHnui4VBDfSQvQP
 autocoder/common/command_generator.py,sha256=v4LmU7sO-P7jEZIXCWHUC6P-vT7AvBi_x_PTwCqBAE8,1323
 autocoder/common/command_templates.py,sha256=mnB3n8i0yjH1mqzyClEg8Wpr9VbZV44kxky66Zu6OJY,8557
 autocoder/common/const.py,sha256=eTjhjh4Aj4CUzviJ81jaf3Y5cwqsLATySn2wJxaS6RQ,2911
-autocoder/common/files.py,sha256=uGpfKASYwIncK_Vt_e_FOjFlO5VyAQOnRJe2SFdSWrg,877
+autocoder/common/files.py,sha256=CguxG9digkWBJpRaILErZmL_G5ryPRahPmPFWGB7X18,1973
 autocoder/common/git_utils.py,sha256=btK45sxvfm4tX3fBRNUPRZoGQuZuOEQrWSAwLy1yoLw,23095
 autocoder/common/image_to_page.py,sha256=O0cNO_vHHUP-fP4GXiVojShmNqkPnZXeIyiY1MRLpKg,13936
 autocoder/common/interpreter.py,sha256=62-dIakOunYB4yjmX8SHC0Gdy2h8NtxdgbpdqRZJ5vk,2833
@@ -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=oFdSJW4ypc3_mhE3fk2O9UB34XMcq3a1Sp2MeTlfQ2o,11820
 autocoder/index/for_command.py,sha256=BFvljE4t6VaMBGboZAuhUCzVK0EitCy_n5D_7FEnihw,3204
-autocoder/index/index.py,sha256=8AcaELR1FS___7VlNyxPnJsDVQ4wjORbqXvcA6TifCE,20337
+autocoder/index/index.py,sha256=VjfcBYHywU4tjQTA7mpHfzRM8nBPhPHrUnkuBbsj6do,20409
 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=V0MAUKgEG9vVTwZK5lMfpZCjU57S6cBioeHDjog0kLs,7992
+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
@@ -119,7 +120,7 @@ autocoder/rag/stream_event/event_writer.py,sha256=l7kq_LnDDE8E5dZ-73C7J2MgzSL7Wr
 autocoder/rag/stream_event/types.py,sha256=rtLwOE8rShmi1dJdxyBpAV5ZjLBGG9vptMiSzMxGuIA,318
 autocoder/regex_project/__init__.py,sha256=EBZeCL5ORyD_9_5u_UuG4s7XtpXOu0y1sWDmxWFtufE,6781
 autocoder/regexproject/__init__.py,sha256=cEr-ZOaQjLD5sx7T7F2DhD5ips03HcJ02rded9EpSXc,9693
-autocoder/suffixproject/__init__.py,sha256=cmP54Y01ditZ83tiJqw5wle0I-uJBC0aZbZ7lYNSVO8,11080
+autocoder/suffixproject/__init__.py,sha256=VcXjUbGf3uQrpoqVCItDvGG9DoeHJ_qEmghKwrVNw9w,11058
 autocoder/tsproject/__init__.py,sha256=yloVzkGLnbTd4Hcj9fMO-rcjNTTx4wI3Ga41LWOSYrY,11747
 autocoder/utils/__init__.py,sha256=KtcGElFNBgZPF7dEL8zF9JpXkCAjoyDrzaREJBhJrcs,994
 autocoder/utils/_markitdown.py,sha256=RU88qn4eZfYIy0GDrPxlI8oYXIypbi63VRJjdlnE0VU,47431
@@ -134,15 +135,15 @@ autocoder/utils/print_table.py,sha256=ZMRhCA9DD0FUfKyJBWd5bDdj1RrtPtgOMWSJwtvZcL
 autocoder/utils/queue_communicate.py,sha256=buyEzdvab1QA4i2QKbq35rG5v_9x9PWVLWWMTznWcYM,6832
 autocoder/utils/request_event_queue.py,sha256=r3lo5qGsB1dIjzVQ05dnr0z_9Z3zOkBdP1vmRciKdi4,2095
 autocoder/utils/request_queue.py,sha256=nwp6PMtgTCiuwJI24p8OLNZjUiprC-TsefQrhMI-yPE,3889
-autocoder/utils/rest.py,sha256=opE_kBEdNQdxh350M5lUTMk5TViRfpuKP_qWc0B1lks,8861
+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=fcXusKEUKMu8WY9Y1_JL5aPkC-soKFxQcFAKThrNZoQ,13338
+autocoder/utils/auto_coder_utils/chat_stream_out.py,sha256=8OV1VOdbj8O7JXDrBUhXVPmMkoCd9n-hvshtR2XXYxk,9112
 autocoder/utils/chat_auto_coder_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-auto_coder-0.1.243.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-auto_coder-0.1.243.dist-info/METADATA,sha256=zq_UlYzkagreMYmIrWVkCmf43Zr3_jIX1w577MlSXQE,2616
-auto_coder-0.1.243.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-auto_coder-0.1.243.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
-auto_coder-0.1.243.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
-auto_coder-0.1.243.dist-info/RECORD,,
+auto_coder-0.1.245.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+auto_coder-0.1.245.dist-info/METADATA,sha256=OtK6YNCP5q11TSBcEWYhfDeljzTN7OqNYW4ze_mggmk,2616
+auto_coder-0.1.245.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+auto_coder-0.1.245.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
+auto_coder-0.1.245.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
+auto_coder-0.1.245.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,7 +300,8 @@ 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")
+            default_model = args.model
+            llm = byzerllm.SimpleByzerLLM(default_model_name=default_model)
             api_key_dir = os.path.expanduser("~/.auto-coder/keys")
             api_key_file = os.path.join(api_key_dir, "api.deepseek.com")
             
@@ -308,7 +314,7 @@ def main(input_args: Optional[List[str]] = None):
             llm.deploy(
                 model_path="",
                 pretrained_model_type="saas/openai",
-                udf_name="deepseek_chat",
+                udf_name=default_model,
                 infer_params={
                     "saas.base_url": "https://api.deepseek.com/v1",
                     "saas.api_key": api_key,
@@ -317,11 +323,11 @@ def main(input_args: Optional[List[str]] = None):
                 }
             )
             
-            code_llm = byzerllm.SimpleByzerLLM(default_model_name="deepseek_chat")
+            code_llm = byzerllm.SimpleByzerLLM(default_model)
             code_llm.deploy(
                 model_path="",
                 pretrained_model_type="saas/openai",
-                udf_name="deepseek_chat",
+                udf_name=default_model,
                 infer_params={
                     "saas.base_url": "https://api.deepseek.com/v1",
                     "saas.api_key": api_key,
@@ -368,7 +374,8 @@ def main(input_args: Optional[List[str]] = None):
                     "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)
@@ -692,6 +699,23 @@ def main(input_args: Optional[List[str]] = None):
                 )
                 llm.setup_sub_client("vl_model", vl_model)
 
+            if args.index_model:   
+                model_name = args.index_model.strip()
+                model_info = models_module.get_model_by_name(model_name)                
+                index_model = byzerllm.SimpleByzerLLM(default_model_name=model_name)
+                index_model.deploy(
+                    model_path="",
+                    pretrained_model_type="saas/openai",
+                    udf_name=model_name,
+                    infer_params={
+                        "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"]
+                    }
+                )
+                llm.setup_sub_client("index_model", index_model)    
+
             if args.sd_model:
                 model_name = args.sd_model.strip()
                 model_info = models_module.get_model_by_name(model_name)
@@ -1332,6 +1356,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,

autocoder/chat_auto_coder.py

@@ -1605,8 +1605,8 @@ def code_next(query: str):
         if os.path.exists(temp_yaml):
             os.remove(temp_yaml)
 
-    llm = byzerllm.ByzerLLM.from_default_model(
-        args.inference_model or args.model)
+    product_mode = conf.get("product_mode", "lite")
+    llm = get_single_llm(args.chat_model or args.model, product_mode=product_mode)
 
     auto_guesser = AutoGuessQuery(
         llm=llm, project_dir=os.getcwd(), skip_diff=True)
@@ -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,11 @@ def main():
             if "mode" not in memory:
                 memory["mode"] = "normal"
 
+            # 处理 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

@@ -1,4 +1,5 @@
 import locale
+from byzerllm.utils import format_str_jinja2
 
 MESSAGES = {
     "en": {
@@ -110,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}",
@@ -221,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。",
     }
 }
 
@@ -236,3 +239,7 @@ def get_system_language():
 def get_message(key):
     lang = get_system_language()
     return MESSAGES.get(lang, MESSAGES['en']).get(key, MESSAGES['en'][key])
+
+
+def get_message_with_format(msg_key: str, **kwargs):
+    return format_str_jinja2(get_message(msg_key), **kwargs)

autocoder/common/auto_coder_lang.py

@@ -1,9 +1,18 @@
 import locale
+from byzerllm.utils import format_str_jinja2
 
 MESSAGES = {
-"en": {
+    "en": {
+        "model_not_found": "Model {{model_name}} not found",
         "new_session_started": "New session started. Previous chat history has been archived.",    
         "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 }}",
@@ -73,8 +82,16 @@ MESSAGES = {
         "git_init_required": "⚠️ auto_merge only applies to git repositories.\n\nPlease try using git init in the source directory:\n\n```shell\ncd {{ source_dir }}\ngit init.\n```\n\nThen run auto - coder again.\nError: {{ error }}"
     },
     "zh": {
+        "model_not_found": "未找到模型: {{model_name}}",
         "new_session_started": "新会话已开始。之前的聊天历史已存档。",
         "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 }}",
@@ -125,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}}",
@@ -156,3 +173,6 @@ def get_system_language():
 def get_message(key):
     lang = get_system_language()
     return MESSAGES.get(lang, MESSAGES['en']).get(key, MESSAGES['en'][key])
+
+def get_message_with_format(msg_key: str, **kwargs):
+    return format_str_jinja2(get_message(msg_key), **kwargs)

autocoder/common/files.py

@@ -1,3 +1,6 @@
+from autocoder.common.auto_coder_lang import get_message_with_format
+from typing import List, Dict, Union
+
 def read_file(file_path):
     """Read a file with automatic encoding detection.
     
@@ -23,4 +26,33 @@ def read_file(file_path):
         except UnicodeDecodeError:
             continue
             
-    raise ValueError(f"无法解码文件: {file_path}。尝试的编码: {', '.join(encodings)}")
+    raise ValueError(get_message_with_format("file_decode_error", 
+        file_path=file_path, 
+        encodings=", ".join(encodings)))
+
+
+
+def save_file(file_path: str, content: Union[str, List[str]]) -> None:
+    """Save content to a file using UTF-8 encoding.
+    
+    Args:
+        file_path (str): Path to the file to write
+        content (Union[str, List[str]]): Content to write to the file. 
+            Can be a string or list of strings (will be joined with newlines)
+        
+    Raises:
+        IOError: If the file cannot be written
+        TypeError: If content is neither str nor List[str]
+    """
+    try:
+        with open(file_path, 'w', encoding='utf-8') as f:
+            if isinstance(content, str):
+                f.write(content)
+            elif isinstance(content, list):
+                f.write('\n'.join(content))
+            else:
+                raise TypeError("Content must be either str or List[str]")
+    except IOError as e:
+        raise IOError(get_message_with_format("file_write_error",
+            file_path=file_path,
+            error=str(e)))

autocoder/index/entry.py

@@ -100,13 +100,13 @@ def build_index_and_filter_files(
                     })
                 )
             )
+
+        if not args.skip_filter_index and args.index_filter_model:       
+            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)
         
-        #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:
+        if not args.skip_filter_index and not args.index_filter_model:
             printer.print_in_terminal("normal_filter_start", style="blue")
             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:
@@ -125,7 +126,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
                                 )