auto-coder 0.1.269__py3-none-any.whl → 0.1.271__py3-none-any.whl

{auto_coder-0.1.269.dist-info → auto_coder-0.1.271.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.269
+Version: 0.1.271
 Summary: AutoCoder: AutoCoder
 Author: allwefantasy
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence

{auto_coder-0.1.269.dist-info → auto_coder-0.1.271.dist-info}/RECORD

@@ -1,10 +1,10 @@
 autocoder/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/auto_coder.py,sha256=wAPeIawQCOxl8MfT_-PMaDI9ZNB8fNoI6FSSMnHRJeI,65206
+autocoder/auto_coder.py,sha256=SgAAoSpUaPZiiNnEWRYIULEsKBJaKuQK_b7pEl_mKY8,65516
 autocoder/auto_coder_lang.py,sha256=Rtupq6N3_HT7JRhDKdgCBcwRaiAnyCOR_Gsp4jUomrI,3229
 autocoder/auto_coder_rag.py,sha256=nwgsXO2-scssWStjX3S910tDp-OZXZRddSYrpyC4Nq0,29021
 autocoder/auto_coder_rag_client_mcp.py,sha256=QRxUbjc6A8UmDMQ8lXgZkjgqtq3lgKYeatJbDY6rSo0,6270
 autocoder/auto_coder_rag_mcp.py,sha256=-RrjNwFaS2e5v8XDIrKR-zlUNUE8UBaeOtojffBrvJo,8521
-autocoder/auto_coder_runner.py,sha256=o5-Q7hdv-tnDz-GQuWOINRZGCG2xK06J3iBVIfxFuHY,100316
+autocoder/auto_coder_runner.py,sha256=JGyLT231zu2cZgnf98Y-W5wDQ0xORZJef1GfL7SZQto,100725
 autocoder/auto_coder_server.py,sha256=6YQweNEKUrGAZ3yPvw8_qlNZJYLVSVUXGrn1K6udLts,20413
 autocoder/benchmark.py,sha256=Ypomkdzd1T3GE6dRICY3Hj547dZ6_inqJbBJIp5QMco,4423
 autocoder/chat_auto_coder.py,sha256=skujpqYqf4EvBLQARJELxj7Xwq9KQj2FGefUAiutF7c,16711
@@ -12,11 +12,12 @@ autocoder/chat_auto_coder_lang.py,sha256=ShOQVOnMA-WlT-fB9OrOer-xQkbcWxJGl-WMPuZ
 autocoder/command_args.py,sha256=9aYJ-AmPxP1sQh6ciw04FWHjSn31f2W9afXFwo8wgx4,30441
 autocoder/lang.py,sha256=U6AjVV8Rs1uLyjFCZ8sT6WWuNUxMBqkXXIOs4S120uk,14511
 autocoder/models.py,sha256=PlG1tKHSHwB57cKLOl5gTl5yTzFUDzCgeHPJU3N9F6Q,9106
-autocoder/version.py,sha256=S9yKKEg96OjIjypcL7XoP28mKfzSNjaNLrgAS0igsBw,23
+autocoder/version.py,sha256=ZgM5wgwos-e45Kw9MewMvDsa_W_7THMB-EhDfus_ybw,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_learn_from_commit.py,sha256=H_7i-73DZIOOKC9rfh1Uwb8SbWJ9XOJLwem3mhWSBEk,8240
 autocoder/agent/auto_review_commit.py,sha256=Ks5oCDp-82YXiRfcDCgYabKq0pcMJkMrP4CaWIgw_JE,9143
 autocoder/agent/auto_tool.py,sha256=DBzip-P_T6ZtT2eHexPcusmKYD0h7ufzp7TLwXAY10E,11554
 autocoder/agent/coder.py,sha256=x6bdJwDuETGg9ebQnYlUWCxCtQcDGg73LtI6McpWslQ,72034
@@ -25,15 +26,15 @@ autocoder/agent/planner.py,sha256=SZTSZHxHzDmuWZo3K5fs79RwvJLWurg-nbJRRNbX65o,91
 autocoder/agent/project_reader.py,sha256=tWLaPoLw1gI6kO_NzivQj28KbobU2ceOLuppHMbfGl8,18234
 autocoder/chat/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/commands/auto_command.py,sha256=BbQoSmkzAv6EtVbJDYCgWqMrN8w3A6BdhlMTtEp1dXQ,51384
-autocoder/commands/tools.py,sha256=xDhGD1jRN67fGfeck33pM74TxXlAMeo49S3Q9K-VKco,20107
+autocoder/commands/auto_command.py,sha256=ymAYnaxiKfYFegKxE5FhdD8XqXMMf4mbDHJsIOxPaUU,52396
+autocoder/commands/tools.py,sha256=lanjoBGR6H8HDJSY3KrM6ibrtHZbgKX6mKJHSSE66dg,20493
 autocoder/common/JupyterClient.py,sha256=O-wi6pXeAEYhAY24kDa0BINrLYvKS6rKyWe98pDClS0,2816
 autocoder/common/ShellClient.py,sha256=fM1q8t_XMSbLBl2zkCNC2J9xuyKN3eXzGm6hHhqL2WY,2286
 autocoder/common/__init__.py,sha256=d1AmAAYhm4b17dVhAJFwV3Vv2r1lUhMejpjr32Poyg8,12948
 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=AnWlvw3wCmfozxql5uwzMK1_drrPtzyBo6TpAINHq9Y,29919
+autocoder/common/auto_coder_lang.py,sha256=WZm5XQaYeECiTiFiHjXWHRZKeV0ux2v64hwZqY8ZDDU,31709
 autocoder/common/auto_configure.py,sha256=L0wjvR-6wFNpP3c9ZxwDOunTtGrzwzjUwsS6BUnJ3W8,12470
 autocoder/common/buildin_tokenizer.py,sha256=L7d5t39ZFvUd6EoMPXUhYK1toD0FHlRH1jtjKRGokWU,1236
 autocoder/common/chunk_validation.py,sha256=BrR_ZWavW8IANuueEE7hS8NFAwEvm8TX34WnPx_1hs8,3030
@@ -47,16 +48,16 @@ autocoder/common/code_auto_merge.py,sha256=cMEX44QT59iMPEo5B8OQKU9fAzn4iIahEPk2n
 autocoder/common/code_auto_merge_diff.py,sha256=jl6rYvioAuN5CWaYRPY4uBFKsDB0_yW6g6KeVlIxPJA,17948
 autocoder/common/code_auto_merge_editblock.py,sha256=D8petP0X74lScsVktFXEQ4_9u5UDVBRDFF16nkjWGZg,20310
 autocoder/common/code_auto_merge_strict_diff.py,sha256=VxasaD8IcxOEEilXmxT0WeLLA16JDlvvI7cV5GyyHAI,12086
-autocoder/common/code_modification_ranker.py,sha256=sRNIPIsQP8FucxSuBeAHKJ7RWB_iB5Pb_cd0BAhvgGc,10926
+autocoder/common/code_modification_ranker.py,sha256=e1i8oNPN_PfG4O8HQnQWxh9tyOd-ur6jFTXfkHPVXYo,13152
 autocoder/common/command_completer.py,sha256=Nw_EFXLDuVXbOKGlmzTODNRJUV9mSUEWV73GUmO_WLw,35166
-autocoder/common/command_generator.py,sha256=Je7OKJvyzeb4krdEspUJx2tyM2tjk-PkNpkc6bZ7PbA,2563
+autocoder/common/command_generator.py,sha256=kxdkRtnPl4O51ChUmvLTXJqJlto5G02eG6xyx8fcLdA,2669
 autocoder/common/command_templates.py,sha256=WAixVjue5QmCFAD13K4ElfcOEjdeGr8tFb0atDAbEoo,8658
 autocoder/common/conf_import_export.py,sha256=w__WsIobe6nmsGns2pV-laU7R5ZvtQNuIbXebxhbY7A,3967
 autocoder/common/conf_validator.py,sha256=EzSmadpZ22D9e8iWmfeWodUeYJt0IgMoaAOmCleXliI,8795
 autocoder/common/const.py,sha256=eTjhjh4Aj4CUzviJ81jaf3Y5cwqsLATySn2wJxaS6RQ,2911
-autocoder/common/context_pruner.py,sha256=i3DeVG3WABnTS30XyOGFSoJpQDquLDugDkOoT8zkZl0,20517
+autocoder/common/context_pruner.py,sha256=_JUpN8naoGNF2JupirM-5xn8Bl9uQgIVmZP3dXH4F1g,21725
 autocoder/common/conversation_pruner.py,sha256=pzmrQEa7pFzA66eYSS_h7VqP6ZwUABeooDQzm0PGu0A,5770
-autocoder/common/files.py,sha256=2-9CJwOZtyWkk2TQM4gPSkpJ3_cwb-l_3sdsCd1H5GQ,3380
+autocoder/common/files.py,sha256=nPiKcnUcYZbSUn3TskKeTVnAxCJRtuehPuB_5d2imX8,4618
 autocoder/common/git_utils.py,sha256=qeuF_IB3G3M72asHxWokROU3hINCuFA1nar-UtF9wIU,26022
 autocoder/common/global_cancel.py,sha256=hT7J7J5ChThIhk2x11_v4v9ASIn4HtwyPD26t2s-fwc,418
 autocoder/common/image_to_page.py,sha256=yWiTJQ49Lm3j0FngiJhQ9u7qayqE_bOGb8Rk0TmSWx0,14123
@@ -74,7 +75,7 @@ autocoder/common/result_manager.py,sha256=nBcFRj5reBC7vp13M91f4B8iPW8B8OehayHlUd
 autocoder/common/screenshots.py,sha256=_gA-z1HxGjPShBrtgkdideq58MG6rqFB2qMUJKjrycs,3769
 autocoder/common/search.py,sha256=245iPFgWhMldoUK3CqCP89ltaxZiNPK73evoG6Fp1h8,16518
 autocoder/common/search_replace.py,sha256=GphFkc57Hb673CAwmbiocqTbw8vrV7TrZxtOhD0332g,22147
-autocoder/common/shells.py,sha256=xSyQChybfClMgp8o07oS5CipME-XQgU564sjrRYK6fM,15977
+autocoder/common/shells.py,sha256=NOgsKbAJVBti_5lSp7kEKAzjH0AJWufWpgXQdJYQF7E,18312
 autocoder/common/stats_panel.py,sha256=wGl9O45pjVVDxhNumLv4_NfLYSlUP_18Tw4hcJSjw50,4596
 autocoder/common/sys_prompt.py,sha256=JlexfjZt554faqbgkCmzOJqYUzDHfbnxly5ugFfHfEE,26403
 autocoder/common/text.py,sha256=KGRQq314GHBmY4MWG8ossRoQi1_DTotvhxchpn78c-k,1003
@@ -94,9 +95,9 @@ autocoder/dispacher/actions/plugins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQ
 autocoder/dispacher/actions/plugins/action_regex_project.py,sha256=AqGIkjbqV1eOS3vBoZUTSOpyOlkv1p5h35mI2Kcvekw,6906
 autocoder/dispacher/actions/plugins/action_translate.py,sha256=GEn7dZA22jy5WyzINomjmzzB795p2Olg-CJla97lRF8,7744
 autocoder/index/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/index/entry.py,sha256=nRKV_T_u3_0HL3xURjd3zisrm-WR4an6_vbUFHaPKJE,14396
+autocoder/index/entry.py,sha256=l-RVlKSLDZkWaWi3jDzMbyWpZ97ZHIEZufsF5gae4Lo,14430
 autocoder/index/for_command.py,sha256=BFvljE4t6VaMBGboZAuhUCzVK0EitCy_n5D_7FEnihw,3204
-autocoder/index/index.py,sha256=1ZNw4s_Fq_ARqjnfU0Vmo1mvdzM6sjaLspjcfinTEh4,27350
+autocoder/index/index.py,sha256=1wUYfK9s0IHmisQL0bzd42PKdFg6WhYgglyMFEcrx6Y,27362
 autocoder/index/symbols_utils.py,sha256=_EP7E_qWXxluAxq3FGZLlLfdrfwx3FmxCdulI8VGuac,2244
 autocoder/index/types.py,sha256=a2s_KV5FJlq7jqA2ELSo9E1sjuLwDB-JJYMhSpzBAhU,596
 autocoder/index/filter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -163,9 +164,9 @@ 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=lkJ_A-sYU36JMzjFWkk3pR6uos8oZHYt9GPsPe_CPAo,11766
 autocoder/utils/chat_auto_coder_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-auto_coder-0.1.269.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-auto_coder-0.1.269.dist-info/METADATA,sha256=nsZmASYIQdIDIfvDFGnIIS8iRDCt5Ex0Q4YHteRoZAU,2643
-auto_coder-0.1.269.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-auto_coder-0.1.269.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
-auto_coder-0.1.269.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
-auto_coder-0.1.269.dist-info/RECORD,,
+auto_coder-0.1.271.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+auto_coder-0.1.271.dist-info/METADATA,sha256=eqiFY5uN-v89P1pH5YtnxnghXFABqjgmm4pPOqRUOpE,2643
+auto_coder-0.1.271.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+auto_coder-0.1.271.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
+auto_coder-0.1.271.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
+auto_coder-0.1.271.dist-info/RECORD,,

autocoder/agent/auto_learn_from_commit.py

@@ -0,0 +1,209 @@
+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.common.printer import Printer
+from autocoder.common import AutoCoderArgs
+from autocoder.common.utils_code_auto_generate import stream_chat_with_continue
+import hashlib
+
+
+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 AutoLearnFromCommit:
+    def __init__(self, llm: Union[byzerllm.ByzerLLM,byzerllm.SimpleByzerLLM],
+                 args:AutoCoderArgs,                 
+                 skip_diff: bool = False,                 
+                 console: Optional[Console] = None):
+        """
+        初始化 AutoLearnFromCommit
+
+        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 learn(self, querie_with_urls_and_changes: List[Tuple[str, List[str], Dict[str, Tuple[str, str]]]], query: str) -> Generator[str,None,None]:
+        """
+        请根据下面的代码变更，总结出通用的代码调整模式，以帮助实现类似的需求。
+
+        下面用户对本次的目标以及你需要总结的要求：
+        <goal>
+        {{ query }}
+        </goal>
+
+        下面是本次提交的代码变更：
+        <changes>
+        {% for query,urls,changes in querie_with_urls_and_changes %}
+        ## 原始的任务需求
+        {{ query }}
+
+        修改的文件:
+        {% for url in urls %}
+        - {{ url }}
+        {% endfor %}
+
+        代码变更:
+        {% for file_path, (before, after) in changes.items() %}
+        ##File: {{ file_path }}
+        ##修改前:
+        
+        {{ before or "New file" }}        
+        
+        ##File: {{ file_path }}
+        ##修改后:
+        
+        {{ after or "File deleted" }}
+        
+        {% endfor %}
+        {% endfor %}
+        </changes>
+
+        请总结以下内容：
+        1. 代码调整模式：描述为了实现这个目标，通常需要对代码做出哪些通用调整
+        2. 关键修改点：指出本次提交中最关键或最具代表性的修改点
+        3. 潜在扩展：基于这些修改，未来类似需求可能需要进行哪些扩展
+        4. 注意事项：在实现类似功能时需要注意哪些问题
+
+        总结要求：
+        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_changes = []
+        repo = git.Repo(self.project_dir)
+
+        # 收集所有query、urls和对应的文件变化
+        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:
+                changes = {}
+                if not self.skip_diff:
+                    # 计算文件的MD5用于匹配commit   
+                    with open(yaml_path, 'r', encoding='utf-8') as f:
+                        yaml_content = f.read()                 
+                        file_md5 = hashlib.md5(yaml_content.encode("utf-8")).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]
+                                    # 获取所有文件的前后内容
+                                    for diff_item in parent.diff(commit):
+                                        file_path = diff_item.a_path if diff_item.a_path else diff_item.b_path
+                                        
+                                        # 获取变更前内容
+                                        before_content = None
+                                        try:
+                                            if diff_item.a_blob:
+                                                before_content = repo.git.show(f"{parent.hexsha}:{file_path}")
+                                        except git.exc.GitCommandError:
+                                            pass  # 文件可能是新增的
+
+                                        # 获取变更后内容
+                                        after_content = None
+                                        try:
+                                            if diff_item.b_blob:
+                                                after_content = repo.git.show(f"{commit.hexsha}:{file_path}")
+                                        except git.exc.GitCommandError:
+                                            pass  # 文件可能被删除
+
+                                        changes[file_path] = (before_content, after_content)
+                                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_changes_error", style="red", error=str(e))
+
+                querie_with_urls_and_changes.append((query, urls, changes))
+
+        return querie_with_urls_and_changes
+    
+
+    def learn_from_commit(self,query: str, conversations: List[Dict]) -> Generator[str,None,None]:
+        """
+        从最新的代码提交中学习通用模式
+
+        Returns:
+            Optional[Generator]: 学习结果生成器，如果出错则返回None
+        """
+        printer = Printer()
+        # 获取最新的提交信息
+        changes = self.parse_history_tasks()
+        if not changes:            
+            printer.print_in_terminal("no_latest_commit", style="red")
+            return None
+
+        # 调用LLM进行代码学习
+        try:
+            # 获取 prompt 内容            
+            query = self.learn.prompt(changes, 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_learn_error", style="red", error=str(e))
+            return None

autocoder/auto_coder.py

@@ -1339,6 +1339,10 @@ def main(input_args: Optional[List[str]] = None):
                 from autocoder.agent.auto_review_commit import AutoReviewCommit
                 reviewer = AutoReviewCommit(llm=chat_llm, args=args)
                 v = reviewer.review_commit(query=args.query,conversations=loaded_conversations)
+            elif "learn_from_commit" in args.action:
+                from autocoder.agent.auto_learn_from_commit import AutoLearnFromCommit
+                learner = AutoLearnFromCommit(llm=chat_llm, args=args)
+                v = learner.learn_from_commit(query=args.query,conversations=loaded_conversations)
             else:                
                 # 预估token数量
                 estimated_input_tokens = count_tokens(json.dumps(loaded_conversations, ensure_ascii=False))

autocoder/auto_coder_runner.py

@@ -1,4 +1,6 @@
 from itertools import product
+from rich.console import Console
+from rich.panel import Panel
 from prompt_toolkit.formatted_text import HTML
 from prompt_toolkit.shortcuts import radiolist_dialog
 from prompt_toolkit import prompt
@@ -646,21 +648,28 @@ def revert():
         with redirect_stdout() as output:
             auto_coder_main(["revert", "--file", file_path])
         s = output.getvalue()
-        print(s, flush=True)
+        
+        console = Console()
+        panel = Panel(
+            Markdown(s),
+            title="Revert Result",
+            border_style="green" if "Successfully reverted changes" in s else "red",
+            padding=(1, 2),
+            expand=False
+        )
+        console.print(panel)
+        
         if "Successfully reverted changes" in s:
             result_manager.append(content=s, meta={"action": "revert","success":False, "input":{                
             }})
-            print(
-                "Reverted the last chat action successfully. Remove the yaml file {file_path}"
-            )
+                        
             os.remove(file_path)
         else:
             result_manager.append(content=s, meta={"action": "revert","success":False, "input":{                
             }})
     else:
         result_manager.append(content="No previous chat action found to revert.", meta={"action": "revert","success":False, "input":{                
-            }})
-        print("No previous chat action found to revert.")
+            }})        
 
 
 def add_files(args: List[str]):
@@ -1155,6 +1164,7 @@ def mcp(query: str):
         "silence": conf.get("silence", "true") == "true",
         "include_project_structure": conf.get("include_project_structure", "true")
         == "true",
+        "exclude_files": memory.get("exclude_files", []),
     }
     for key, value in conf.items():
         converted_value = convert_config_value(key, value)
@@ -1224,6 +1234,7 @@ def code_next(query: str):
         "silence": conf.get("silence", "true") == "true",
         "include_project_structure": conf.get("include_project_structure", "true")
         == "true",
+        "exclude_files": memory.get("exclude_files", []),
     }
     for key, value in conf.items():
         converted_value = convert_config_value(key, value)
@@ -1595,6 +1606,9 @@ def chat(query: str):
     if "/review" in query and "/commit" in query:
         yaml_config["action"].append("review_commit")
         query = query.replace("/review", "", 1).replace("/commit", "", 1).strip()
+    elif "/learn" in query:
+        yaml_config["action"].append("learn_from_commit")
+        query = query.replace("/learn", "", 1).strip()
     else:    
         is_review = query.strip().startswith("/review")
         if is_review:
@@ -2283,6 +2297,7 @@ def index_build():
     conf = memory.get("conf", {})
     yaml_config = {
         "include_file": ["./base/base.yml"],
+        "exclude_files": memory.get("exclude_files", []),
     }
 
     for key, value in conf.items():
@@ -2295,10 +2310,8 @@ def index_build():
 
     with open(yaml_file, "w",encoding="utf-8") as f:
         f.write(yaml_content)
-    try:
-        with redirect_stdout() as output:
-            auto_coder_main(["index", "--file", yaml_file])
-        print(output.getvalue(), flush=True)
+    try:        
+        auto_coder_main(["index", "--file", yaml_file])        
         completer.refresh_files()
     finally:
         os.remove(yaml_file)
@@ -2315,6 +2328,7 @@ def get_final_config()->AutoCoderArgs:
         "silence": conf.get("silence", "true") == "true",
         "include_project_structure": conf.get("include_project_structure", "true")
         == "true",
+        "exclude_files": memory.get("exclude_files", []),
     }
     for key, value in conf.items():
         converted_value = convert_config_value(key, value)
@@ -2380,10 +2394,8 @@ def index_query(query: str):
 
     with open(yaml_file, "w",encoding="utf-8") as f:
         f.write(yaml_content)
-    try:
-        with redirect_stdout() as output:
-            auto_coder_main(["index-query", "--file", yaml_file])
-        print(output.getvalue(), flush=True)
+    try:        
+        auto_coder_main(["index-query", "--file", yaml_file])        
     finally:
         os.remove(yaml_file)
 

autocoder/commands/auto_command.py

@@ -171,15 +171,21 @@ class CommandAutoTuner:
         当前用户环境信息如下:
         <os_info>
         操作系统: {{ env_info.os_name }} {{ env_info.os_version }}
+        操作系统发行版: {{ os_distribution }}
         Python版本: {{ env_info.python_version }}
         终端类型: {{ env_info.shell_type }}
         终端编码: {{ env_info.shell_encoding }}
+        
+        {%- if shell_type %}
+        脚本类型：{{ shell_type }}
+        {%- endif %}
+
         {%- if env_info.conda_env %}
         Conda环境: {{ env_info.conda_env }}
         {%- endif %}
         {%- if env_info.virtualenv %}
         虚拟环境: {{ env_info.virtualenv }}
-        {%- endif %}  
+        {%- endif %}   
         </os_info>
         
         我们的目标是根据用户输入和当前上下文，组合多个函数来完成用户的需求。
@@ -221,6 +227,7 @@ class CommandAutoTuner:
         每个文件的tokens数，你可以根据这个信息来决定如何读取这个文件。比如对于很小的文件，那么可以直接全部读取，
         而对于分析一个超大文件推荐组合 read_files 带上 line_ranges 参数来读取，或者组合 read_file_withread_file_with_keyword_ranges 等来读取，
         每个函数你还可以使用多次来获取更多信息。
+        9. 根据操作系统，终端类型，脚本类型等各种信息，在涉及到路径或者脚本的时候，需要考虑平台差异性。
         </function_combination_readme>
 
 
@@ -260,7 +267,13 @@ class CommandAutoTuner:
         满足需求。
         """
 
-        env_info = detect_env()            
+        env_info = detect_env() 
+        shell_type = "bash"
+        if shells.is_running_in_cmd():
+            shell_type = "cmd"
+        elif shells.is_running_in_powershell():
+            shell_type = "powershell"
+
         return {
             "user_input": request.user_input,
             "current_files": self.memory_config.memory["current_files"]["files"],            
@@ -268,9 +281,10 @@ class CommandAutoTuner:
             "available_commands": self._command_readme.prompt(),
             "current_conf": json.dumps(self.memory_config.memory["conf"], indent=2),        
             "env_info": env_info,
-            "shell_type": shells.get_terminal_name(),
+            "shell_type": shell_type,
             "shell_encoding": shells.get_terminal_encoding(),
-            "conversation_safe_zone_tokens": self.args.conversation_prune_safe_zone_tokens
+            "conversation_safe_zone_tokens": self.args.conversation_prune_safe_zone_tokens,
+            "os_distribution": shells.get_os_distribution()
         }
     
     @byzerllm.prompt()
@@ -1179,6 +1193,18 @@ class CommandAutoTuner:
          此时会返回诸如 "ts,py,java,go,js,ts" 这样的字符串，表示项目类型。
         </usage>
         </command>
+
+        <command>
+        <name>response_user</name>
+        <description>响应用户。</description>
+        <usage>
+         如果你需要直接发送信息给用户，那么可以通过 response_user 函数来直接回复用户。
+         
+         比如用户问你是谁？
+         你可以通过如下方式来回答：
+         response_user(response="你好，我是 auto-coder")
+        </usage>
+        </command>
         </commands>
         
         
@@ -1224,7 +1250,9 @@ class CommandAutoTuner:
             "get_project_related_files": self.tools.get_project_related_files,
             "ask_user":self.tools.ask_user,
             "read_file_with_keyword_ranges": self.tools.read_file_with_keyword_ranges,
-            "get_project_type": self.project_type_analyzer.analyze,                                    
+            "get_project_type": self.project_type_analyzer.analyze,
+            "response_user": self.tools.response_user,
+                                    
         }
 
         if command not in command_map:

autocoder/commands/tools.py

@@ -28,6 +28,10 @@ from autocoder.utils.queue_communicate import (
 import sys
 import io
 from autocoder.common import files as files_utils
+from autocoder.common.printer import Printer
+from prompt_toolkit import PromptSession
+from prompt_toolkit.styles import Style
+
 
 @byzerllm.prompt()
 def detect_rm_command(command: str) -> Bool:
@@ -56,6 +60,7 @@ class AutoCommandTools:
         self.args = args
         self.llm = llm
         self.result_manager = ResultManager()
+        self.printer = Printer()
 
     def ask_user(self,question:str) -> str:
         '''
@@ -91,35 +96,43 @@ class AutoCommandTools:
         )
 
         # 显示问题面板
-        console.print(question_panel)
+        console.print(question_panel)        
 
-        # 创建一个自定义提示符
-        prompt = Prompt.ask(
-            "\n[bold green]Your Answer[/bold green]",
-            console=console
-        )
+        session = PromptSession( message=self.printer.get_message_from_key('tool_ask_user'))
+        try:
+            answer = session.prompt()
+        except KeyboardInterrupt:
+            answer = ""        
 
-        # 获取用户的回答
-        answer = prompt
+        self.result_manager.append(content=answer, meta = {
+            "action": "ask_user",
+            "input": {
+                "question": question
+            }
+        })
 
-        # 显示用户的回答
-        answer_text = Text(answer, style="italic")
+        return answer
+    
+    def response_user(self, response: str):
+        console = Console()
+        answer_text = Text(response, style="italic")
         answer_panel = Panel(
             answer_text,
-            title="[bold yellow]Your Response[/bold yellow]",
+            title="",
             border_style="green",
             expand=False
         )
         console.print(answer_panel)
 
-        self.result_manager.append(content=answer, meta = {
-            "action": "ask_user",
+        self.result_manager.append(content=response, meta = {
+            "action": "response_user",
             "input": {
-                "question": question
+                "response": response
             }
         })
 
-        return answer
+        return response
+        
 
     def run_python_code(self, code: str) -> str:
         """

autocoder/common/auto_coder_lang.py

@@ -130,7 +130,7 @@ MESSAGES = {
         "quick_filter_too_long": "⚠️ index file is too large ({{ tokens_len }}/{{ max_tokens }}). The query will be split into {{ split_size }} chunks.",
         "quick_filter_tokens_len": "📊 Current index size: {{ tokens_len }} tokens",
         "estimated_chat_input_tokens": "Estimated chat input tokens: {{ estimated_input_tokens }}",
-        "estimated_input_tokens_in_generate": "Estimated input tokens in generate ({{ generate_mode }}): {{ estimated_input_tokens }}",
+        "estimated_input_tokens_in_generate": "Estimated input tokens in generate ({{ generate_mode }}): {{ estimated_input_tokens }}",        
         "model_has_access_restrictions": "{{model_name}} has access restrictions, cannot use the current function",
         "auto_command_not_found": "Auto command not found: {{command}}. Please check your input and try again.",
         "auto_command_failed": "Auto command failed: {{error}}. Please check your input and try again.",
@@ -163,11 +163,22 @@ MESSAGES = {
         "diff_blocks_title":"diff blocks",
         "index_exclude_files_error": "index filter exclude files fail: {{ error }}",
         "file_sliding_window_processing": "File {{ file_path }} is too large ({{ tokens }} tokens), processing with sliding window...",
-        "file_snippet_processing": "Processing file {{ file_path }} with code snippet extraction..."
+        "file_snippet_processing": "Processing file {{ file_path }} with code snippet extraction...",
+        "context_pruning_start": "⚠️ Context pruning started. Total tokens: {{ total_tokens }} (max allowed: {{ max_tokens }}). Applying strategy: {{ strategy }}.",
+        "context_pruning_reason": "Context length exceeds maximum limit ({{ total_tokens }} > {{ max_tokens }}). Pruning is required to fit within the model's context window.",
+        "rank_code_modification_title": "{{model_name}} ranking codes",
+        "sorted_files_message": "Reordered files:\n{% for file in files %}- {{ file }}\n{% endfor %}",
+        "estimated_input_tokens_in_ranking": "estimate input token {{ estimated_input_tokens }} when ranking",
+        "file_snippet_procesed": "{{ file_path }} processed with tokens: {{ tokens }} => {{ snippet_tokens }}. Current total tokens: {{ total_tokens }}",
+        "tool_ask_user": "Your Reply: ",
+        "tool_ask_user_accept":"Your Response received",
+        
     },
     "zh": {
         "file_sliding_window_processing": "文件 {{ file_path }} 过大 ({{ tokens }} tokens)，正在使用滑动窗口处理...",
         "file_snippet_processing": "正在对文件 {{ file_path }} 进行代码片段提取...",
+        "context_pruning_start": "⚠️ 开始上下文剪枝。总token数: {{ total_tokens }} (最大允许: {{ max_tokens }})。正在应用策略: {{ strategy }}。",
+        "context_pruning_reason": "上下文长度超过最大限制 ({{ total_tokens }} > {{ max_tokens }})。需要进行剪枝以适配模型的上下文窗口。",
         "file_scored_message": "文件评分: {{file_path}} - 分数: {{score}}",
         "invalid_file_pattern": "无效的文件模式: {{file_pattern}}. 例如: regex://.*/package-lock\\.json",
         "conf_not_found": "未找到配置文件: {{path}}",
@@ -308,7 +319,7 @@ MESSAGES = {
         "quick_filter_title": "{{ model_name }} 正在分析如何筛选上下文...",
         "quick_filter_failed": "❌ 快速过滤器失败: {{ error }}. ",
         "estimated_chat_input_tokens": "对话输入token预估为: {{ estimated_input_tokens }}",
-        "estimated_input_tokens_in_generate": "生成代码({{ generate_mode }})预计输入token数: {{ estimated_input_tokens_in_generate }}",
+        "estimated_input_tokens_in_generate": "生成代码({{ generate_mode }})预计输入token数: {{ estimated_input_tokens }}",        
         "model_has_access_restrictions": "{{model_name}} 有访问限制，无法使用当前功能",
         "auto_command_not_found": "未找到自动命令: {{command}}。请检查您的输入并重试。",
         "auto_command_failed": "自动命令执行失败: {{error}}。请检查您的输入并重试。",
@@ -324,7 +335,13 @@ MESSAGES = {
         "index_import_success": "索引导入成功: {{path}}",
         "edits_title": "编辑块",
         "diff_blocks_title": "差异块",
-        "index_exclude_files_error": "索引排除文件时出错: {{error}}"
+        "index_exclude_files_error": "索引排除文件时出错: {{error}}",
+        "rank_code_modification_title": "模型{{model_name}}对代码打分",
+        "sorted_files_message": "重新排序后的文件路径:\n{% for file in files %}- {{ file }}\n{% endfor %}",
+        "estimated_input_tokens_in_ranking": "排序预计输入token数: {{ estimated_input_tokens }}",
+        "file_snippet_procesed": "文件 {{ file_path }} 处理后token数: {{ tokens }} => {{ snippet_tokens }} 当前总token数: {{ total_tokens }}",
+        "tool_ask_user": "您的回复: ",
+        "tool_ask_user_accept":"收到您的回复",
     }}
 
 

autocoder/common/code_modification_ranker.py

@@ -6,11 +6,13 @@ from pydantic import BaseModel
 from autocoder.common.printer import Printer
 from concurrent.futures import ThreadPoolExecutor, as_completed
 import traceback
-from autocoder.common.utils_code_auto_generate import chat_with_continue
+from autocoder.common.utils_code_auto_generate import chat_with_continue,stream_chat_with_continue
 from byzerllm.utils.str2model import to_model
+from autocoder.utils.auto_coder_utils.chat_stream_out import stream_out
 from autocoder.utils.llms import get_llm_names, get_model_info
 from autocoder.common.types import CodeGenerateResult, MergeCodeWithoutEffect
 import os
+from autocoder.rag.token_counter import count_tokens
 
 class RankResult(BaseModel):
     rank_result: List[int]
@@ -133,6 +135,15 @@ class CodeModificationRanker:
         else:
             raise Exception(f"Invalid rank strategy: {self.args.rank_strategy}")
 
+        # 计算 query 的 token 数量                
+        token_count = count_tokens(query)
+        
+        # 打印 token 统计信息
+        self.printer.print_in_terminal(
+            "estimated_input_tokens_in_ranking",            
+            estimated_input_tokens=token_count
+        )
+
         input_tokens_count = 0
         generated_tokens_count = 0
         try:
@@ -145,16 +156,25 @@ class CodeModificationRanker:
                     self.printer.print_in_terminal(
                         "ranking_start", style="blue", count=len(generate_result.contents), model_name=model_name)
                     
-                    for _ in range(rank_times):
-                        
-                        futures.append(
-                            executor.submit(
-                                chat_with_continue,
-                                llm,
-                                [{"role": "user", "content": query}],
-                                {}
+                    for i in range(rank_times):
+                        if i == 0:
+                            futures.append(
+                                executor.submit(
+                                    stream_chat_with_continue,
+                                    llm,
+                                    [{"role": "user", "content": query}],
+                                    {}
+                                )
+                            )
+                        else:    
+                            futures.append(
+                                executor.submit(
+                                    chat_with_continue,
+                                    llm,
+                                    [{"role": "user", "content": query}],
+                                    {}
+                                )
                             )
-                        )
 
                 # Collect all results
                 results = []
@@ -180,7 +200,31 @@ class CodeModificationRanker:
                 total_input_cost = 0.0
                 total_output_cost = 0.0
 
-                for future, model_name in zip(futures, model_names):
+                # 第一个future使用流式输出
+                stream_future = futures[0]
+                model_name = model_names[0]
+                stream_generator = stream_future.result()
+                full_response, last_meta = stream_out(
+                        stream_generator,
+                        model_name=model_name,
+                        title=self.printer.get_message_from_key_with_format(
+                            "rank_code_modification_title", model_name=model_name),
+                        args=self.args
+                    )
+                
+                if last_meta:
+                    input_tokens_count += last_meta.input_tokens_count
+                    generated_tokens_count += last_meta.generated_tokens_count
+                    # 计算成本                        
+                    info = model_info_map.get(model_name, {})
+                    # 计算公式:token数 * 单价 / 1000000
+                    total_input_cost += (last_meta.input_tokens_count * info.get("input_cost", 0.0)) / 1000000
+                    total_output_cost += (last_meta.generated_tokens_count * info.get("output_cost", 0.0)) / 1000000
+                
+                v = to_model(full_response,RankResult)                        
+                results.append(v.rank_result)
+
+                for future, model_name in zip(futures[1:], model_names[1:]):
                     try:
                         result = future.result()
                         input_tokens_count += result.input_tokens_count

autocoder/common/command_generator.py

@@ -12,6 +12,7 @@ def _generate_shell_script(user_input: str) -> str:
     环境信息如下:
 
     操作系统: {{ env_info.os_name }} {{ env_info.os_version }}
+    操作系统发行版: {{ os_distribution }}
     Python版本: {{ env_info.python_version }}
     终端类型: {{ env_info.shell_type }}
     终端编码: {{ env_info.shell_encoding }}
@@ -49,7 +50,8 @@ def _generate_shell_script(user_input: str) -> str:
     return {
         "env_info": env_info,
         "shell_type": shell_type,
-        "shell_encoding": shells.get_terminal_encoding()
+        "shell_encoding": shells.get_terminal_encoding(),
+        "os_distribution": shells.get_os_distribution()
     }
 
 

autocoder/common/context_pruner.py

@@ -287,8 +287,12 @@ class PruneContext:
                         content_snippets = self._build_snippet_content(file_path, content, merged_snippets)
                         snippet_tokens = count_tokens(content_snippets)
                         if token_count + snippet_tokens <= self.max_tokens:
-                            selected_files.append(SourceCode(module_name=file_path,source_code=content_snippets,tokens=snippet_tokens))
+                            selected_files.append(SourceCode(module_name=file_path,source_code=content_snippets,tokens=snippet_tokens))                            
                             token_count += snippet_tokens
+                            self.printer.print_in_terminal("file_snippet_procesed", file_path=file_path, 
+                                                           total_tokens=token_count, 
+                                                           tokens=tokens, 
+                                                           snippet_tokens=snippet_tokens)
                             continue
                         else:
                             break
@@ -315,11 +319,15 @@ class PruneContext:
                         content_snippets = self._build_snippet_content(file_path, content, snippets)
 
                         snippet_tokens = count_tokens(content_snippets)
-                        if token_count + snippet_tokens <= self.max_tokens:
+                        if token_count + snippet_tokens <= self.max_tokens:                            
                             selected_files.append(SourceCode(module_name=file_path,
                                                              source_code=content_snippets,
                                                              tokens=snippet_tokens))
                             token_count += snippet_tokens
+                            self.printer.print_in_terminal("file_snippet_procesed", file_path=file_path, 
+                                                           total_tokens = token_count,
+                                                           tokens=tokens, 
+                                                           snippet_tokens=snippet_tokens)
                         else:
                             break
             except Exception as e:
@@ -379,7 +387,26 @@ class PruneContext:
         total_tokens,sources = self._count_tokens(file_paths)
         if total_tokens <= self.max_tokens:
             return sources
-        # print(f"total_tokens: {total_tokens} {self.max_tokens}, 进行策略: {strategy}")
+        
+        self.printer.print_in_terminal(
+            "context_pruning_reason",
+            total_tokens=total_tokens,
+            max_tokens=self.max_tokens,
+            style="yellow"
+        )
+
+        self.printer.print_in_terminal(
+                "sorted_files_message", 
+                files=file_paths
+            )
+
+        self.printer.print_in_terminal(
+            "context_pruning_start",
+            total_tokens=total_tokens,
+            max_tokens=self.max_tokens,
+            strategy=strategy
+        )
+
         if strategy == "score":            
             return self._score_and_filter_files(file_paths, conversations)
         if strategy == "delete":

autocoder/common/files.py

@@ -4,8 +4,7 @@ from typing import List, Dict, Union, Generator, Tuple
 def read_file(file_path):
     """Read a file with automatic encoding detection.
     
-    Tries common encodings in sequence (UTF-8 > GBK > UTF-16 > Latin-1) to handle
-    cross-platform encoding issues between Windows and Linux systems.
+    Tries common encodings in sequence to handle cross-platform encoding issues.
     
     Args:
         file_path (str): Path to the file to read
@@ -16,7 +15,10 @@ def read_file(file_path):
     Raises:
         ValueError: If the file cannot be decoded with any of the tried encodings
     """
-    encodings = ['utf-8', 'gbk', 'utf-16', 'latin-1']
+    # Expanded list of encodings to try
+    encodings = ['utf-8', 'gbk', 'utf-16', 'latin-1', 
+                'big5', 'shift_jis', 'euc-jp', 'iso-8859-1',
+                'cp1252', 'ascii']
     
     for encoding in encodings:
         try:
@@ -25,22 +27,54 @@ def read_file(file_path):
                 return content
         except UnicodeDecodeError:
             continue
+        except Exception as e:
+            # Log other errors but continue trying other encodings
+            continue
             
-    raise ValueError(get_message_with_format("file_decode_error", 
-        file_path=file_path, 
-        encodings=", ".join(encodings)))
+    # If all encodings fail, try reading as binary and decode with 'replace'
+    try:
+        with open(file_path, 'rb') as f:
+            return f.read().decode('utf-8', errors='replace')
+    except Exception as e:
+        raise ValueError(get_message_with_format("file_decode_error", 
+            file_path=file_path, 
+            encodings=", ".join(encodings)))
 
 def read_lines(file_path:str):
-    encodings = ['utf-8', 'gbk', 'utf-16', 'latin-1']
+    """Read a file line by line with automatic encoding detection.
+    
+    Args:
+        file_path (str): Path to the file to read
+        
+    Returns:
+        List[str]: List of lines in the file
+        
+    Raises:
+        ValueError: If the file cannot be decoded with any of the tried encodings
+    """
+    # Expanded list of encodings to try
+    encodings = ['utf-8', 'gbk', 'utf-16', 'latin-1', 
+                'big5', 'shift_jis', 'euc-jp', 'iso-8859-1',
+                'cp1252', 'ascii']
+    
     for encoding in encodings:
         try:
             with open(file_path, 'r', encoding=encoding) as f:
                 return f.readlines()
         except UnicodeDecodeError:
             continue
-    raise ValueError(get_message_with_format("file_decode_error", 
-        file_path=file_path, 
-        encodings=", ".join(encodings)))
+        except Exception as e:
+            # Log other errors but continue trying other encodings
+            continue
+            
+    # If all encodings fail, try reading as binary and decode with 'replace'
+    try:
+        with open(file_path, 'rb') as f:
+            return f.read().decode('utf-8', errors='replace').splitlines()
+    except Exception as e:
+        raise ValueError(get_message_with_format("file_decode_error", 
+            file_path=file_path, 
+            encodings=", ".join(encodings)))
 
 
 

autocoder/common/shells.py

@@ -5,6 +5,7 @@ import subprocess
 import platform
 import tempfile
 import uuid
+import re
 from rich.console import Console
 from rich.panel import Panel
 from rich.text import Text
@@ -187,6 +188,71 @@ def _get_windows_terminal_name() -> str:
     # 默认返回 cmd.exe
     return 'cmd'
 
+def get_os_distribution() -> str:
+    """
+    获取操作系统发行版名称
+    
+    Returns:
+        str: 操作系统发行版名称，如 'Ubuntu 22.04', 'CentOS 7', 'Windows 10' 等
+    """
+    system = platform.system()
+    
+    if system == 'Windows':
+        # Windows 系统版本
+        version = sys.getwindowsversion()
+        if version.major == 10 and version.build >= 22000:
+            return 'Windows 11'
+        elif version.major == 10:
+            return 'Windows 10'
+        elif version.major == 6:
+            if version.minor == 3:
+                return 'Windows 8.1'
+            elif version.minor == 2:
+                return 'Windows 8'
+            elif version.minor == 1:
+                return 'Windows 7'
+            elif version.minor == 0:
+                return 'Windows Vista'
+        elif version.major == 5:
+            if version.minor == 2:
+                return 'Windows Server 2003'
+            elif version.minor == 1:
+                return 'Windows XP'
+        return f'Windows {version.major}.{version.minor}'
+    
+    elif system == 'Linux':
+        # Linux 发行版检测
+        try:
+            # 尝试读取 /etc/os-release
+            with open('/etc/os-release', 'r') as f:
+                os_release = f.read()
+            
+            # 解析 NAME 和 VERSION_ID
+            name = re.search(r'NAME="([^"]+)"', os_release)
+            version = re.search(r'VERSION_ID="([^"]+)"', os_release)
+            
+            if name and version:
+                return f"{name.group(1)} {version.group(1)}"
+            
+            # 如果无法解析，尝试 lsb_release
+            try:
+                output = subprocess.check_output(['lsb_release', '-d'], stderr=subprocess.STDOUT)
+                return output.decode('utf-8').split(':')[1].strip()
+            except (subprocess.CalledProcessError, FileNotFoundError):
+                pass
+            
+            # 最后尝试 uname
+            return platform.platform()
+        except Exception:
+            return platform.platform()
+    
+    elif system == 'Darwin':
+        # macOS 版本
+        version = platform.mac_ver()[0]
+        return f'macOS {version}'
+    
+    return platform.platform()
+
 def _get_unix_terminal_name() -> str:
     """Linux/Mac 系统终端检测"""
     # 从环境变量获取终端名称
@@ -238,6 +304,8 @@ def get_terminal_encoding() -> str:
         return _get_unix_terminal_encoding()
 
 def _get_windows_terminal_encoding() -> str:
+    # 添加发行版信息到环境变量
+    # os.environ['OS_DISTRIBUTION'] = get_os_distribution()
     """Windows 系统编码检测"""
     try:
         # 通过 chcp 命令获取代码页

autocoder/index/entry.py

@@ -291,10 +291,12 @@ def build_index_and_filter_files(
             # 提取排序后的文件路径列表
             sorted_file_paths = [file_path for _, file_path in position_file_pairs]
             # 根据 sorted_file_paths 重新排序 temp_sources
-            temp_sources.sort(key=lambda x: sorted_file_paths.index(x.module_name) if x.module_name in sorted_file_paths else len(sorted_file_paths))           
-        pruned_files = context_pruner.handle_overflow([source.module_name for source in temp_sources], [{"role":"user","content":args.query}], args.context_prune_strategy)
+            temp_sources.sort(key=lambda x: sorted_file_paths.index(x.module_name) if x.module_name in sorted_file_paths else len(sorted_file_paths))                       
+            
+        pruned_files = context_pruner.handle_overflow([source.module_name for source in temp_sources], [{"role":"user","content":args.query}], args.context_prune_strategy)        
         source_code_list.sources = pruned_files
 
+
     if args.request_id and not args.skip_events:
         queue_communicate.send_event(
             request_id=args.request_id,

autocoder/index/index.py

@@ -400,7 +400,7 @@ class IndexManager:
 
         # 删除被排除的文件
         try:
-            exclude_patterns = self.parse_exclude_files(self.args.exclude_files)
+            exclude_patterns = self.parse_exclude_files(self.args.exclude_files)            
             for file_path in index_data:
                 if self.filter_exclude_files(file_path, exclude_patterns):
                     keys_to_remove.append(file_path)

autocoder/version.py

@@ -1 +1 @@
-__version__ = "0.1.269"
+__version__ = "0.1.271"