auto-coder 0.1.233__py3-none-any.whl → 0.1.235__py3-none-any.whl

{auto_coder-0.1.233.dist-info → auto_coder-0.1.235.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.233
+Version: 0.1.235
 Summary: AutoCoder: AutoCoder
 Author: allwefantasy
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
@@ -39,7 +39,6 @@ Requires-Dist: real-agent
 Requires-Dist: python-docx
 Requires-Dist: docx2txt
 Requires-Dist: pdf2image
-Requires-Dist: Spire.Doc
 Requires-Dist: docx2pdf
 Requires-Dist: pyperclip
 Requires-Dist: colorama

{auto_coder-0.1.233.dist-info → auto_coder-0.1.235.dist-info}/RECORD

@@ -1,17 +1,17 @@
 autocoder/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/auto_coder.py,sha256=r2uHo-xPDbel6OhZwGzl8Iz1GP4d9KUWcDReEol9LOQ,59541
+autocoder/auto_coder.py,sha256=XOZGAJiy6CWnn7dfbGAmZwVI8KnIaSJ-FgZm8t4d0YE,61069
 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=PzrbhpwTgJEMuPOT7vBd4uPi58mwqLc59l2fVV6rJc8,102049
+autocoder/chat_auto_coder.py,sha256=US-HXJtBkj_7QifJxQNcWgUTYe-ZB2sf0aZI8gbLN9w,102931
 autocoder/chat_auto_coder_lang.py,sha256=YJsFi8an0Kjbo9X7xKZfpdbHS3rbhrvChZNjWqEQ5Sw,11032
 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=VphyhuzLuUXUhi7WwvwaGVGg2OSz77iY97Prah1F5g8,24
+autocoder/version.py,sha256=Ha3i0TzVJUOaC3CSX9IQT4bWoilQPZfSI4LPVJpLuuQ,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
@@ -24,10 +24,11 @@ autocoder/agent/project_reader.py,sha256=tWLaPoLw1gI6kO_NzivQj28KbobU2ceOLuppHMb
 autocoder/chat/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/common/JupyterClient.py,sha256=O-wi6pXeAEYhAY24kDa0BINrLYvKS6rKyWe98pDClS0,2816
 autocoder/common/ShellClient.py,sha256=fM1q8t_XMSbLBl2zkCNC2J9xuyKN3eXzGm6hHhqL2WY,2286
-autocoder/common/__init__.py,sha256=JDIZ_1JP7Nm74OL_aCFwvhiwiynukaE5oPCfh5GPQWU,11695
+autocoder/common/__init__.py,sha256=vpfo3RctksezDoraVSpHtfnxpspVNiYh8OmQhqQvcfE,11729
 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=2xyc_qqClEiqtkwccH5gs3MquYwMM16DMq-kIx7Hwcc,5834
 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
@@ -54,6 +55,7 @@ autocoder/common/mcp_hub.py,sha256=2ZyJv3Aiv4Y97UHut49oYhIFcu7ICR-mptDEBSgT3uE,1
 autocoder/common/mcp_server.py,sha256=QCFa-15kx7rbNsinwdGFFX2y47pww0fVdI-ldKFSSWI,12267
 autocoder/common/mcp_tools.py,sha256=KsLvRrB6pvmebqd-lDaSH6IBJR0AIxWRE-dtCEG_w9k,12485
 autocoder/common/memory_manager.py,sha256=2ZjYG7BPyvbYalZBF6AM_G5e10Qkw_zrqtD4Zd7GSsQ,3663
+autocoder/common/printer.py,sha256=dZQ5JXLy1OxA9uoLE7dNGuiOH0xbt5xN_fsMkEJ6syw,1304
 autocoder/common/recall_validation.py,sha256=Avt9Q9dX3kG6Pf2zsdlOHmsjd-OeSj7U1PFBDp_Cve0,1700
 autocoder/common/screenshots.py,sha256=_gA-z1HxGjPShBrtgkdideq58MG6rqFB2qMUJKjrycs,3769
 autocoder/common/search.py,sha256=245iPFgWhMldoUK3CqCP89ltaxZiNPK73evoG6Fp1h8,16518
@@ -69,20 +71,20 @@ autocoder/db/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/db/store.py,sha256=tFT66bP2ZKIqZip-uhLkHRSLaaOAUUDZfozJwcqix3c,1908
 autocoder/dispacher/__init__.py,sha256=YoA64dIxnx4jcE1pwSfg81sjkQtjDkhddkfac1-cMWo,1230
 autocoder/dispacher/actions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/dispacher/actions/action.py,sha256=NjJGLek8H0FlIOreBnl2KEC-jJ5Jq-V8D1RuI6ifUjc,19299
+autocoder/dispacher/actions/action.py,sha256=KUTpbkIQaIOhdnAaGbUj2ltj12IbzssxqEVoT9YM11U,19972
 autocoder/dispacher/actions/copilot.py,sha256=iMh4ckj9hO5Q-iemF3CStXd7DatWai7Eci5zOlKxK9c,13072
 autocoder/dispacher/actions/plugins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/dispacher/actions/plugins/action_regex_project.py,sha256=-ly-NRgQ8LfDQDoH0QmD_LP-G932Kt08WXy9oIvLy10,5325
+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=1T41clV3GXwiRbowubQ1iZM5k3_2ECS-DtcnbA9QxAk,10081
+autocoder/index/entry.py,sha256=f6_lrJLAfYslzAUY7JUCsJMhoMCIKy9ZHGBWHhmCtr8,11310
 autocoder/index/for_command.py,sha256=LGnz-OWogT8rd24m4Zcan7doLaijxqorAuiMk7WuRq0,3125
-autocoder/index/index.py,sha256=lwaobSHvOnzhTMf8SQXzw3nIJQUS4lyo6nLdtv0Ebc0,19223
+autocoder/index/index.py,sha256=HfWN0Mbk22S81-fnm9wrYNJO9Bcnws4G_PxJ8IQuH5A,20175
 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=BxOiZOlK2v6EnX0yV28R3ielXboTmrMvVwielCrqKpE,3678
+autocoder/index/filter/quick_filter.py,sha256=1cRPAd8uUFkbPg4JlLaYCd3a8CMifVYEIpqBBaJ9GXo,3874
 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
@@ -137,9 +139,9 @@ autocoder/utils/tests.py,sha256=BqphrwyycGAvs-5mhH8pKtMZdObwhFtJ5MC_ZAOiLq8,1340
 autocoder/utils/auto_coder_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/utils/auto_coder_utils/chat_stream_out.py,sha256=6D_SIa5hHSwIHC1poO_ztK7IVugAqNHu-jQySd7EnfQ,4181
 autocoder/utils/chat_auto_coder_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-auto_coder-0.1.233.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-auto_coder-0.1.233.dist-info/METADATA,sha256=w___EO9VOCh1T58HsPM3Qi5OYm2ktUVVD6I5KucBO3w,2641
-auto_coder-0.1.233.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-auto_coder-0.1.233.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
-auto_coder-0.1.233.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
-auto_coder-0.1.233.dist-info/RECORD,,
+auto_coder-0.1.235.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+auto_coder-0.1.235.dist-info/METADATA,sha256=JiMGcJmUn-bzkT0jg8tVMxu-D04DnaXV4On1qBk3iDc,2616
+auto_coder-0.1.235.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+auto_coder-0.1.235.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
+auto_coder-0.1.235.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
+auto_coder-0.1.235.dist-info/RECORD,,

autocoder/auto_coder.py

@@ -39,7 +39,7 @@ from rich.console import Console
 from rich.panel import Panel
 from rich.markdown import Markdown
 from rich.live import Live
-from autocoder.auto_coder_lang import get_message
+from autocoder.common.auto_coder_lang import get_message
 from autocoder.common.memory_manager import save_to_memory_file
 from autocoder import models as models_module
 from autocoder.common.utils_code_auto_generate import stream_chat_with_continue
@@ -69,8 +69,7 @@ def load_include_files(config, base_path, max_depth=10, current_depth=0):
             include_files = [include_files]
 
         for include_file in include_files:
-            abs_include_path = resolve_include_path(base_path, include_file)
-            logger.info(f"Loading include file: {abs_include_path}")
+            abs_include_path = resolve_include_path(base_path, include_file)            
             with open(abs_include_path, "r") as f:
                 include_config = yaml.safe_load(f)
                 if not include_config:
@@ -131,15 +130,15 @@ def main(input_args: Optional[List[str]] = None):
     if not os.path.isabs(args.source_dir):
         args.source_dir = os.path.abspath(args.source_dir)
 
-    if not args.silence:
-        print("Command Line Arguments:")
-        print("-" * 50)
-        for arg, value in vars(args).items():
-            if arg == "context" and value:
-                print(f"{arg:20}: {value[:30]}...")
-            else:
-                print(f"{arg:20}: {value}")
-        print("-" * 50)
+    # if not args.silence:
+    #     print("Command Line Arguments:")
+    #     print("-" * 50)
+    #     for arg, value in vars(args).items():
+    #         if arg == "context" and value:
+    #             print(f"{arg:20}: {value[:30]}...")
+    #         else:
+    #             print(f"{arg:20}: {value}")
+    #     print("-" * 50)
 
     # init store
     store = Store(os.path.join(args.source_dir, ".auto-coder", "metadata.db"))
@@ -244,10 +243,7 @@ def main(input_args: Optional[List[str]] = None):
                 new_file = os.path.join(
                     actions_dir, f"{new_seq}_{raw_args.name}.yml")
                 with open(new_file, "w") as f:
-                    f.write(content)
-
-        print(f"Successfully created new action file: {new_file}")
-
+                    f.write(content)        
         # open_yaml_file_in_editor(new_file)
         return
 
@@ -281,6 +277,22 @@ def main(input_args: Optional[List[str]] = None):
                 byzerllm.connect_cluster(address=args.ray_address)
 
             llm = byzerllm.ByzerLLM(verbose=args.print_request)
+
+            code_model = byzerllm.ByzerLLM()
+            code_model.setup_default_model_name("deepseek_chat")
+            llm.setup_sub_client("code_model", code_model)
+
+            index_filter_model = byzerllm.ByzerLLM()
+            index_filter_model.setup_default_model_name("deepseek_r1_chat")
+            llm.setup_sub_client("index_filter_model", index_filter_model)
+
+            generate_rerank_model = byzerllm.ByzerLLM()
+            generate_rerank_model.setup_default_model_name("deepseek_r1_chat")
+            llm.setup_sub_client("generate_rerank_model", generate_rerank_model)
+
+            chat_model = byzerllm.ByzerLLM()
+            chat_model.setup_default_model_name("deepseek_r1_chat")
+            llm.setup_sub_client("chat_model", chat_model)
         
         if args.product_mode == "lite":
             llm = byzerllm.SimpleByzerLLM(default_model_name="deepseek_chat")
@@ -344,9 +356,23 @@ def main(input_args: Optional[List[str]] = None):
                 }
             )
 
+            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("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)
 
         if args.product_mode == "lite":                                    
             # Set up default models based on configuration
@@ -444,7 +470,7 @@ def main(input_args: Optional[List[str]] = None):
                         "saas.is_reasoning": model_info["is_reasoning"]
                     }
                 )
-                llm.setup_sub_client("inference_model", inference_model) 
+                llm.setup_sub_client("inference_model", inference_model)                 
 
             if args.index_filter_model:
                 model_info = models_module.get_model_by_name(args.index_filter_model)
@@ -1114,20 +1140,25 @@ def main(input_args: Optional[List[str]] = None):
                 chat_llm = llm
 
             source_count = 0
-            pre_conversations = []
-            if args.context:
-                context = json.loads(args.context)
-                if "file_content" in context:
-                    file_content = context["file_content"]
-                    pre_conversations.append(
-                        {
-                            "role": "user",
-                            "content": f"请阅读下面的代码和文档：\n\n <files>\n{file_content}\n</files>",
-                        },
-                    )
-                    pre_conversations.append(
-                        {"role": "assistant", "content": "read"})
-                    source_count += 1
+            pre_conversations = []            
+            context_content = args.context if args.context else ""
+            if args.context:                
+                try:
+                    context = json.loads(args.context)
+                    if "file_content" in context:
+                        context_content = context["file_content"]
+                except:
+                    pass                     
+
+                pre_conversations.append(
+                    {
+                        "role": "user",
+                        "content": f"请阅读下面的代码和文档：\n\n <files>\n{context_content}\n</files>",
+                    },
+                )
+                pre_conversations.append(
+                    {"role": "assistant", "content": "read"})
+                source_count += 1
 
             from autocoder.index.index import IndexManager
             from autocoder.index.entry import build_index_and_filter_files
@@ -1142,10 +1173,11 @@ def main(input_args: Optional[List[str]] = None):
             else:
                 pp = SuffixProject(args=args, llm=llm, file_filter=None)
             pp.run()
-            sources = pp.sources
+            sources = pp.sources                       
 
             s = build_index_and_filter_files(
-                llm=llm, args=args, sources=sources)
+                llm=llm, args=args, sources=sources)                        
+            
             if s:
                 pre_conversations.append(
                     {

autocoder/chat_auto_coder.py

@@ -370,6 +370,33 @@ def initialize_system(args):
             "deepseek_chat",
         ]
 
+        try:
+            subprocess.run(deploy_cmd, check=True)
+            print_status(get_message("deploy_complete"), "success")
+        except subprocess.CalledProcessError:
+            print_status(get_message("deploy_fail"), "error")
+            return
+        
+
+        deploy_cmd = [
+            "byzerllm",
+            "deploy",
+            "--pretrained_model_type",
+            "saas/reasoning_openai",
+            "--cpus_per_worker",
+            "0.001",
+            "--gpus_per_worker",
+            "0",
+            "--worker_concurrency",
+            "1000",
+            "--num_workers",
+            "1",
+            "--infer_params",
+            f"saas.base_url=https://api.deepseek.com/v1 saas.api_key={api_key} saas.model=deepseek-reasoner",
+            "--model",
+            "deepseek_r1_chat",
+        ]
+
         try:
             subprocess.run(deploy_cmd, check=True)
             print_status(get_message("deploy_complete"), "success")
@@ -1715,7 +1742,8 @@ def coding(query: str):
             == "true",
         }
 
-        yaml_config["context"] = ""
+        yaml_config["context"] = ""        
+        yaml_config["in_code_apply"] = is_apply
 
         for key, value in conf.items():
             converted_value = convert_config_value(key, value)

autocoder/common/__init__.py

@@ -355,5 +355,8 @@ class AutoCoderArgs(pydantic.BaseModel):
     enable_global_memory: Optional[bool] = True  
     product_mode: Optional[str] = "lite"
 
+    in_code_apply: bool = False
+
     class Config:
         protected_namespaces = ()
+

autocoder/common/auto_coder_lang.py

@@ -0,0 +1,90 @@
+import locale
+
+MESSAGES = {
+    "en": {
+        "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_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 }})",
+        "index_related_files_fail": "⚠️ Failed to find related files for chunk {{ chunk_count }}",
+        "index_threads_completed": "✅ Completed {{ completed_threads }}/{{ total_threads }} threads",
+        "index_related_files_fail": "⚠️ Failed to find related files for chunk {{ chunk_count }}",
+        "human_as_model_instructions": (
+            "You are now in Human as Model mode. The content has been copied to your clipboard.\n"
+            "The system is waiting for your input. When finished, enter 'EOF' on a new line to submit.\n"
+            "Use '/break' to exit this mode. If you have issues with copy-paste, use '/clear' to clean and paste again."
+        ),
+        "clipboard_not_supported": (
+            "pyperclip not installed or clipboard is not supported, instruction will not be copied to clipboard."
+        ),
+        "human_as_model_instructions_no_clipboard": (
+            "You are now in Human as Model mode. [bold red]The content could not be copied to your clipboard.[/bold red]\n"
+            "but you can copy prompt from output.txt file.\n"
+            "The system is waiting for your input. When finished, enter 'EOF' on a new line to submit.\n"
+            "Use '/break' to exit this mode. If you have issues with copy-paste, use '/clear' to clean and paste again."
+        ),
+        "phase1_processing_sources": "Phase 1: Processing REST/RAG/Search sources...",
+        "phase2_building_index": "Phase 2: Building index for all files...",
+        "phase6_file_selection": "Phase 6: Processing file selection and limits...",
+        "phase7_preparing_output": "Phase 7: Preparing final output...",
+        "chat_human_as_model_instructions": (
+            "Chat is now in Human as Model mode.\n"
+            "The question has been copied to your clipboard.\n"
+            "Please use Web version model to get the answer.\n"
+            "Or use /conf human_as_model:false to close this mode and get the answer in terminal directly."
+            "Paste the answer to the input box below, use '/break' to exit, '/clear' to clear the screen, '/eof' to submit."
+        )
+    },
+    "zh": {
+        "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_build_summary": "📊 总文件数: {{ total_files }}, 需要构建索引: {{ num_files }}",
+        "building_index_progress": "⏳ 正在构建索引: {{ counter }}/{{ num_files }}...",
+        "index_source_dir_mismatch": "⚠️ 源目录不匹配 (文件路径: {{ file_path }}, 源目录: {{ source_dir }})",
+        "index_related_files_fail": "⚠️ 无法为块 {{ chunk_count }} 找到相关文件",
+        "index_threads_completed": "✅ 已完成 {{ completed_threads }}/{{ total_threads }} 个线程",
+        "index_related_files_fail": "⚠️ 无法为块 {{ chunk_count }} 找到相关文件",
+        "human_as_model_instructions": (
+            "您现在处于人类作为模型模式。内容已复制到您的剪贴板。\n"
+            "系统正在等待您的输入。完成后，在新行输入'EOF'提交。\n"
+            "使用'/break'退出此模式。如果复制粘贴有问题，使用'/clear'清理并重新粘贴。"
+        ),
+        "clipboard_not_supported": (
+            "未安装pyperclip或不支持剪贴板，指令将不会被复制到剪贴板。"
+        ),
+        "human_as_model_instructions_no_clipboard": (
+            "您现在处于人类作为模型模式。[bold red]内容无法复制到您的剪贴板。[/bold red]\n"
+            "但您可以从output.txt文件复制提示。\n"
+            "系统正在等待您的输入。完成后，在新行输入'EOF'提交。\n"
+            "使用'/break'退出此模式。如果复制粘贴有问题，使用'/clear'清理并重新粘贴。"
+        ),
+
+        "phase1_processing_sources": "阶段 1: 正在处理 REST/RAG/Search 源...",
+        "phase2_building_index": "阶段 2: 正在为所有文件构建索引...",
+        "phase6_file_selection": "阶段 6: 正在处理文件选择和限制...",
+        "phase7_preparing_output": "阶段 7: 正在准备最终输出...",
+
+        "chat_human_as_model_instructions": (
+            "\n============= Chat 处于 Human as Model 模式 =============\n"
+            "问题已复制到剪贴板\n"
+            "请使用Web版本模型获取答案\n"
+            "或者使用 /conf human_as_model:false 关闭该模式直接在终端获得答案。"
+            "将获得答案黏贴到下面的输入框，换行后，使用 '/break' 退出，'/clear' 清屏，'/eof' 提交。"
+        ),
+    }
+}
+
+
+def get_system_language():
+    try:
+        return locale.getdefaultlocale()[0][:2]
+    except:
+        return 'en'
+
+
+def get_message(key):
+    lang = get_system_language()
+    return MESSAGES.get(lang, MESSAGES['en']).get(key, MESSAGES['en'][key])

autocoder/common/printer.py

@@ -0,0 +1,36 @@
+from rich.console import Console
+from typing import Optional
+from byzerllm.utils import format_str_jinja2
+from autocoder.common.auto_coder_lang import get_message
+from autocoder.chat_auto_coder_lang import get_message as get_chat_message
+class Printer:
+    def __init__(self,console:Optional[Console]=None):
+        if console is None:
+            self.console = Console()
+        else:
+            self.console = console
+
+    def get_message_from_key(self, key: str):
+        try:
+            return get_message(key)
+        except Exception as e:
+            return get_chat_message(key)
+
+    def print_in_terminal(self, key: str, style: str = None,**kwargs):     
+        try:
+            if style:
+                self.console.print(format_str_jinja2(self.get_message_from_key(key),**kwargs), style=style)
+            else:
+                self.console.print(format_str_jinja2(self.get_message_from_key(key),**kwargs))
+        except Exception as e:
+            print(self.get_message_from_key(key))
+
+    
+    def print_str_in_terminal(self, content: str, style: str = None):     
+        try:
+            if style:
+                self.console.print(content, style=style)
+            else:
+                self.console.print(content)
+        except Exception as e:
+            print(content)        

autocoder/dispacher/actions/action.py

@@ -54,9 +54,14 @@ class ActionTSProject(BaseAction):
 
         source_code = pp.output()
         if self.llm:
+            if args.in_code_apply:
+                old_query = args.query
+                args.query = (args.context or "") + "\n\n" + args.query
             source_code = build_index_and_filter_files(
                 llm=self.llm, args=args, sources=pp.sources
             )
+            if args.in_code_apply:
+                args.query = old_query
 
         if args.image_file:
             if args.image_mode == "iterative":
@@ -256,11 +261,17 @@ class ActionPyProject(BaseAction):
         pp = PyProject(args=self.args, llm=self.llm)
         self.pp = pp
         pp.run(packages=args.py_packages.split(",") if args.py_packages else [])            
-        source_code = pp.output()
+        source_code = pp.output()                
+
         if self.llm:
+            old_query = args.query
+            if args.in_code_apply:
+                args.query = (args.context or "") + "\n\n" + args.query
             source_code = build_index_and_filter_files(
                 llm=self.llm, args=args, sources=pp.sources
             )
+            if args.in_code_apply:
+                args.query = old_query
 
         self.process_content(source_code)
         return True
@@ -355,9 +366,14 @@ class ActionSuffixProject(BaseAction):
         pp.run()
         source_code = pp.output()
         if self.llm:
+            if args.in_code_apply:
+                old_query = args.query
+                args.query = (args.context or "") + "\n\n" + args.query
             source_code = build_index_and_filter_files(
                 llm=self.llm, args=args, sources=pp.sources
             )
+            if args.in_code_apply:
+                args.query = old_query
         self.process_content(source_code)
 
     def process_content(self, content: str):

autocoder/dispacher/actions/plugins/action_regex_project.py

@@ -36,9 +36,14 @@ class ActionRegexProject:
         pp.run()
         source_code = pp.output()
         if self.llm:
+            if args.in_code_apply:
+                old_query = args.query
+                args.query = (args.context or "") + "\n\n" + args.query
             source_code = build_index_and_filter_files(
                 llm=self.llm, args=args, sources=pp.sources
             )
+            if args.in_code_apply:
+                args.query = old_query
         self.process_content(source_code)
 
     def process_content(self, content: str):

autocoder/index/entry.py

@@ -9,7 +9,7 @@ from rich.console import Console
 from rich.table import Table
 from rich.panel import Panel
 
-from loguru import logger
+from autocoder.common.printer import Printer
 from autocoder.utils.queue_communicate import (
     queue_communicate,
     CommunicateEvent,
@@ -58,7 +58,8 @@ def build_index_and_filter_files(
     final_files: Dict[str, TargetFile] = {}
 
     # Phase 1: Process REST/RAG/Search sources
-    logger.info("Phase 1: Processing REST/RAG/Search sources...")
+    printer = Printer()
+    printer.print_in_terminal("phase1_processing_sources")
     phase_start = time.monotonic()
     for source in sources:
         if source.tag in ["REST", "RAG", "SEARCH"]:
@@ -79,7 +80,7 @@ def build_index_and_filter_files(
                 )
             )
 
-        logger.info("Phase 2: Building index for all files...")
+        printer.print_in_terminal("phase2_building_index")
         phase_start = time.monotonic()
         index_manager = IndexManager(llm=llm, sources=sources, args=args)
         index_data = index_manager.build_index()
@@ -133,19 +134,48 @@ def build_index_and_filter_files(
 
         return [file for file in result] if result else []
 
+    def shorten_path(path: str, keep_levels: int = 3) -> str:
+        """
+        优化长路径显示，保留最后指定层级
+        示例：/a/b/c/d/e/f.py -> .../c/d/e/f.py
+        """
+        parts = path.split(os.sep)
+        if len(parts) > keep_levels:
+            return ".../" + os.sep.join(parts[-keep_levels:])
+        return path
+
     def print_selected(data):
         console = Console()
+        
+        # 获取终端宽度
+        console_width = console.width
 
         table = Table(
             title="Files Used as Context",
             show_header=True,
             header_style="bold magenta",
+            # 设置表格最大宽度为终端宽度（留 10 字符边距）
+            width=min(console_width - 10, 120),
+            expand=True
         )
-        table.add_column("File Path", style="cyan", no_wrap=True)
-        table.add_column("Reason", style="green")
+        
+        # 优化列配置
+        table.add_column("File Path", 
+                        style="cyan", 
+                        width=int((console_width - 10) * 0.6),  # 分配 60% 宽度给文件路径
+                        overflow="fold",  # 自动折叠过长的路径
+                        no_wrap=False)  # 允许换行
+        
+        table.add_column("Reason", 
+                        style="green", 
+                        width=int((console_width - 10) * 0.4),  # 分配 40% 宽度给原因
+                        no_wrap=False)
 
+        # 添加处理过的文件路径
         for file, reason in data:
-            table.add_row(file, reason)
+            # 路径截取优化：保留最后 3 级路径
+            processed_path = shorten_path(file, keep_levels=3)
+            table.add_row(processed_path, reason)
 
         panel = Panel(
             table,
@@ -157,7 +187,7 @@ def build_index_and_filter_files(
         console.print(panel)
 
     # Phase 6: File selection and limitation
-    logger.info("Phase 6: Processing file selection and limits...")
+    printer.print_in_terminal("phase6_file_selection")
     phase_start = time.monotonic()
 
     if args.index_filter_file_num > 0:
@@ -188,7 +218,7 @@ def build_index_and_filter_files(
     stats["timings"]["file_selection"] = phase_end - phase_start
 
     # Phase 7: Display results and prepare output
-    logger.info("Phase 7: Preparing final output...")
+    printer.print_in_terminal("phase7_preparing_output")
     phase_start = time.monotonic()    
     try:
         print_selected(
@@ -269,7 +299,7 @@ def build_index_and_filter_files(
 • Total time: {total_time:.2f}s
 ====================================
 """
-    logger.info(summary)
+    printer.print_str_in_terminal(summary)
 
     if args.request_id and not args.skip_events:
         queue_communicate.send_event(

autocoder/index/filter/quick_filter.py

@@ -55,7 +55,10 @@ class QuickFilter():
         }
         ```
 
-        特别注意,如果用户的query里 @文件 或者 @@符号，那么被@的文件或者@@的符号必须要返回，并且查看他们依赖的文件是否相关。        
+        特别注意
+        1. 如果用户的query里 @文件 或者 @@符号，那么被@的文件或者@@的符号必须要返回，并且查看他们依赖的文件是否相关。        
+        2. 如果 query 里是一段历史对话，那么对话里的内容提及的文件路径必须要返回。
+        3. json格式数据不允许有注释
         '''
         file_meta_str = "\n".join([f"##[{index}]{item.module_name}\n{item.symbols}" for index,item in enumerate(file_meta_list)])
         context = {
@@ -66,15 +69,13 @@ 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.args.index_filter_model:
+        if not self.args.skip_filter_index and self.index_manager.llm.get_sub_client("index_filter_model"):
             start_time = time.monotonic()
             index_items = self.index_manager.read_index()
 
-            prompt_str = self.quick_filter_files.prompt(index_items,query)
-
-            print(prompt_str)
+            prompt_str = self.quick_filter_files.prompt(index_items,query)            
             
-            tokens_len = count_tokens(prompt_str)
+            tokens_len = count_tokens(prompt_str)            
             
             if tokens_len > 55*1024:
                 logger.warning(f"Quick filter prompt is too long, tokens_len: {tokens_len}/{55*1024} fallback to normal filter")

autocoder/index/index.py

@@ -15,7 +15,8 @@ import threading
 import byzerllm
 import hashlib
 
-from loguru import logger
+from autocoder.common.printer import Printer
+from autocoder.common.auto_coder_lang import get_message
 from autocoder.index.types import (
     IndexItem,
     TargetFile,
@@ -48,6 +49,7 @@ class IndexManager:
         self.max_input_length = (
             args.index_model_max_input_length or args.model_max_input_length
         )
+        self.printer = Printer()
 
         # 如果索引目录不存在,则创建它
         if not os.path.exists(self.index_dir):
@@ -206,8 +208,12 @@ class IndexManager:
             start_time = time.monotonic()
             source_code = source.source_code
             if len(source.source_code) > self.max_input_length:
-                logger.warning(
-                    f"Warning[Build Index]: The length of source code({source.module_name}) is too long ({len(source.source_code)}) > model_max_input_length({self.max_input_length}), splitting into chunks..."
+                self.printer.print_in_terminal(
+                    "index_file_too_large",
+                    style="yellow",
+                    file_path=source.module_name,
+                    file_size=len(source.source_code),
+                    max_length=self.max_input_length
                 )
                 chunks = self.split_text_into_chunks(
                     source_code, self.max_input_length - 1000
@@ -224,12 +230,23 @@ class IndexManager:
                     self.index_llm).run(source.module_name, source_code)
                 time.sleep(self.anti_quota_limit)
 
-            logger.info(
-                f"Parse and update index for {file_path} md5: {md5} took {time.monotonic() - start_time:.2f}s"
+            self.printer.print_in_terminal(
+                "index_update_success",
+                style="green",
+                file_path=file_path,
+                md5=md5,
+                duration=time.monotonic() - start_time
             )
 
         except Exception as e:
-            logger.warning(f"Error: {e}")
+            # import traceback
+            # traceback.print_exc()
+            self.printer.print_in_terminal(
+                "index_build_error",
+                style="red",
+                file_path=file_path,
+                error=str(e)
+            )
             return None
 
         return {
@@ -255,8 +272,11 @@ class IndexManager:
 
         for item in index_data.keys():
             if not item.startswith(self.source_dir):
-                logger.warning(
-                    error_message(source_dir=self.source_dir, file_path=item)
+                self.printer.print_in_terminal(
+                    "index_source_dir_mismatch",
+                    style="yellow",
+                    source_dir=self.source_dir,
+                    file_path=item
                 )
                 break
 
@@ -291,8 +311,12 @@ class IndexManager:
             counter = 0
             num_files = len(wait_to_build_files)
             total_files = len(self.sources)
-            logger.info(
-                f"Total Files: {total_files}, Need to Build Index: {num_files}")
+            self.printer.print_in_terminal(
+                "index_build_summary",
+                style="bold blue",
+                total_files=total_files,
+                num_files=num_files
+            )
 
             futures = [
                 executor.submit(self.build_index_for_single_source, source)
@@ -302,7 +326,12 @@ class IndexManager:
                 result = future.result()
                 if result is not None:
                     counter += 1
-                    logger.info(f"Building Index:{counter}/{num_files}...")
+                    self.printer.print_in_terminal(
+                        "building_index_progress",
+                        style="blue",
+                        counter=counter,
+                        num_files=num_files
+                    )
                     module_name = result["module_name"]
                     index_data[module_name] = result
                     updated_sources.append(module_name)
@@ -404,8 +433,10 @@ class IndexManager:
                 with lock:
                     all_results.extend(result.file_list)
             else:
-                logger.warning(
-                    f"Fail to find related files for chunk {chunk_count}. This may be caused by the model limit or the query not being suitable for the files."
+                self.printer.print_in_terminal(
+                    "index_related_files_fail",
+                    style="yellow",
+                    chunk_count=chunk_count
                 )
             time.sleep(self.args.anti_quota_limit)
 
@@ -442,8 +473,10 @@ class IndexManager:
                     all_results.extend(result.file_list)
                     completed_threads += 1
             else:
-                logger.warning(
-                    f"Fail to find target files for chunk. This is caused by the model response not being in JSON format or the JSON being empty."
+                self.printer.print_in_terminal(
+                    "index_related_files_fail",
+                    style="yellow",
+                    chunk_count="unknown"
                 )
             time.sleep(self.args.anti_quota_limit)
 
@@ -457,7 +490,12 @@ class IndexManager:
             for future in as_completed(futures):
                 future.result()
 
-        logger.info(f"Completed {completed_threads}/{total_threads} threads")
+        self.printer.print_in_terminal(
+            "index_threads_completed",
+            style="green",
+            completed_threads=completed_threads,
+            total_threads=total_threads
+        )
         return all_results, total_threads, completed_threads
 
     def get_target_files_by_query(self, query: str) -> FileList:

autocoder/version.py

@@ -1 +1 @@
-__version__ = "0.1.233"
+__version__ = "0.1.235"