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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.232
+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.232.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=REnZJUy4Dx9b7Z8gM9D4SWMkbHXl0JjF-DFiod4aB-A,58357
+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=KtDAwIiBB1b2jBSY8BCoSj88iRSwtRACkzME9h91ido,101601
+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=pvV6K1v2HfuQywwyaRYD8uMohj-o0EfkRauHjMIrYtk,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=-ATCQp7EhFepyUsupHEZpUN3ouQ-OcM3PAoCnAXezhg,11652
+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,16 +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=JqghsBpgrM-5J79UYjGIVivNgz6YnMtibwBPP0Ww9xY,19299
-autocoder/dispacher/actions/copilot.py,sha256=BdD-L0LFb5YUH2C87IVCm446SJLyOltaaG9shmsQ25Q,13072
+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=oldAZeGaDcGQBiTdCigO6yiV_GJQTGRLy_B_hVHAQZw,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=f6_lrJLAfYslzAUY7JUCsJMhoMCIKy9ZHGBWHhmCtr8,11310
 autocoder/index/for_command.py,sha256=LGnz-OWogT8rd24m4Zcan7doLaijxqorAuiMk7WuRq0,3125
-autocoder/index/index.py,sha256=1HLwK-ylpibYjHgPommdaL1bRwGP1QGWONaNYExzqRc,34828
+autocoder/index/index.py,sha256=HfWN0Mbk22S81-fnm9wrYNJO9Bcnws4G_PxJ8IQuH5A,20175
 autocoder/index/symbols_utils.py,sha256=CjcjUVajmJZB75Ty3a7kMv1BZphrm-tIBAdOJv6uo-0,2037
-autocoder/index/types.py,sha256=tYoFicbS6k1Dx4EoMpuNq71-4pF6hhEbtej0VYCVlSo,524
+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=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
@@ -133,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.232.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-auto_coder-0.1.232.dist-info/METADATA,sha256=5XhWylx2N9N4DNDoJpF3GUWHJzpNVg2wwgBZ5m_rBDU,2641
-auto_coder-0.1.232.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-auto_coder-0.1.232.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
-auto_coder-0.1.232.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
-auto_coder-0.1.232.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,9 +39,10 @@ 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
 from autocoder.utils.auto_coder_utils.chat_stream_out import stream_out
 
 console = Console()
@@ -68,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:
@@ -130,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"))
@@ -243,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
 
@@ -280,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")
@@ -343,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
@@ -443,7 +470,24 @@ 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)
+                model_name = args.index_filter_model
+                index_filter_model = byzerllm.SimpleByzerLLM(default_model_name=model_name)
+                index_filter_model.deploy(
+                    model_path="",
+                    pretrained_model_type=model_info["model_type"],
+                    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_filter_model", index_filter_model)            
 
 
         if args.product_mode == "pro":
@@ -482,7 +526,12 @@ def main(input_args: Optional[List[str]] = None):
             if args.inference_model:
                 inference_model = byzerllm.ByzerLLM()
                 inference_model.setup_default_model_name(args.inference_model)
-                llm.setup_sub_client("inference_model", inference_model)        
+                llm.setup_sub_client("inference_model", inference_model)   
+
+            if args.index_filter_model:
+                index_filter_model = byzerllm.ByzerLLM()
+                index_filter_model.setup_default_model_name(args.index_filter_model)
+                llm.setup_sub_client("index_filter_model", index_filter_model)
             
 
         if args.human_as_model:
@@ -1091,22 +1140,28 @@ 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                     
 
-            from autocoder.index.index import IndexManager, build_index_and_filter_files
+                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
             from autocoder.pyproject import PyProject
             from autocoder.tsproject import TSProject
             from autocoder.suffixproject import SuffixProject
@@ -1118,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(
                     {
@@ -1272,8 +1328,7 @@ def main(input_args: Optional[List[str]] = None):
                     )
                 )
                 v = [[response.result,None]]
-            else:
-                from autocoder.common.utils_code_auto_generate import stream_chat_with_continue
+            else:                
                 v = stream_chat_with_continue(
                     llm=chat_llm,
                     conversations=loaded_conversations,

autocoder/chat_auto_coder.py

@@ -52,6 +52,7 @@ from autocoder.common.memory_manager import get_global_memory_file_paths
 from autocoder import models
 import shlex
 from autocoder.utils.llms import get_single_llm
+import pkg_resources
 
 class SymbolItem(BaseModel):
     symbol_name: str
@@ -369,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")
@@ -1714,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)
@@ -2478,6 +2507,17 @@ def lib_command(args: List[str]):
 
 
 def main():
+    from autocoder.rag.variable_holder import VariableHolder
+    from tokenizers import Tokenizer    
+    try:
+        tokenizer_path = pkg_resources.resource_filename(
+            "autocoder", "data/tokenizer.json"
+        )
+        VariableHolder.TOKENIZER_PATH = tokenizer_path
+        VariableHolder.TOKENIZER_MODEL = Tokenizer.from_file(tokenizer_path)
+    except FileNotFoundError:
+        tokenizer_path = None
+        
     ARGS = parse_arguments()
     
     if ARGS.lite:

autocoder/common/__init__.py

@@ -259,6 +259,7 @@ class AutoCoderArgs(pydantic.BaseModel):
     skip_filter_index: Optional[bool] = False
 
     index_model: Optional[str] = ""
+    index_filter_model: Optional[str] = ""
     index_model_max_length: Optional[int] = 0
     index_model_max_input_length: Optional[int] = 0
     index_model_anti_quota_limit: Optional[int] = 0
@@ -354,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

@@ -9,7 +9,7 @@ from autocoder.common.buildin_tokenizer import BuildinTokenizer
 from autocoder.pyproject import PyProject, Level1PyProject
 from autocoder.tsproject import TSProject
 from autocoder.suffixproject import SuffixProject
-from autocoder.index.index import build_index_and_filter_files
+from autocoder.index.entry import build_index_and_filter_files
 from autocoder.common.code_auto_merge import CodeAutoMerge
 from autocoder.common.code_auto_merge_diff import CodeAutoMergeDiff
 from autocoder.common.code_auto_merge_strict_diff import CodeAutoMergeStrictDiff
@@ -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/copilot.py

@@ -9,7 +9,7 @@ from autocoder.common.JupyterClient import JupyterNotebook
 from autocoder.common.ShellClient import ShellClient
 from autocoder.suffixproject import SuffixProject
 from autocoder.common.search import Search, SearchEngine
-from autocoder.index.index import build_index_and_filter_files
+from autocoder.index.entry import build_index_and_filter_files
 from autocoder.common.image_to_page import ImageToPage, ImageToPageDirectly
 from typing import Optional, Dict, Any, List
 import byzerllm

autocoder/dispacher/actions/plugins/action_regex_project.py

@@ -9,7 +9,7 @@ from autocoder.common.code_auto_generate import CodeAutoGenerate
 from autocoder.common.code_auto_generate_diff import CodeAutoGenerateDiff
 from autocoder.common.code_auto_generate_strict_diff import CodeAutoGenerateStrictDiff
 from autocoder.common.code_auto_generate_editblock import CodeAutoGenerateEditBlock
-from autocoder.index.index import build_index_and_filter_files
+from autocoder.index.entry import build_index_and_filter_files
 from autocoder.regexproject import RegexProject
 from autocoder.utils.conversation_store import store_code_model_conversation
 from loguru import logger
@@ -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):