auto-coder 0.1.256__py3-none-any.whl → 0.1.258__py3-none-any.whl

{auto_coder-0.1.256.dist-info → auto_coder-0.1.258.dist-info}/METADATA

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

{auto_coder-0.1.256.dist-info → auto_coder-0.1.258.dist-info}/RECORD

@@ -1,17 +1,17 @@
 autocoder/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/auto_coder.py,sha256=obRx5lFu8P4M5mmLA717lwMso7Ei3Kx3i9kG7I_xrRY,65654
+autocoder/auto_coder.py,sha256=whonXfuZrcveFliNXLglxo6CXtwzIqo-7QppjJqag0g,64811
 autocoder/auto_coder_lang.py,sha256=Rtupq6N3_HT7JRhDKdgCBcwRaiAnyCOR_Gsp4jUomrI,3229
 autocoder/auto_coder_rag.py,sha256=DDAmqw36CO6phtdQuN8LYIbIR3YGdoZw5-pG0LjVxMc,29063
 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=a1YEp6OPMzpLbRpr2hrbzF6pRnhVPTVxyZfBiQHFPIw,109283
-autocoder/chat_auto_coder_lang.py,sha256=1cJrjFGrcOQnuP2LdZpgGDSX4CNaIYI7KZGvEEtj6_Q,18242
+autocoder/chat_auto_coder.py,sha256=tfr0qu5yMHZFetL4ef3FvB7d29kjY8qNxW6YKBLXCAA,109659
+autocoder/chat_auto_coder_lang.py,sha256=WxylHYFHqBxM_6YvoqIrPdNQFlobYT1t07xlCISfWJw,18241
 autocoder/command_args.py,sha256=9aYJ-AmPxP1sQh6ciw04FWHjSn31f2W9afXFwo8wgx4,30441
 autocoder/lang.py,sha256=U6AjVV8Rs1uLyjFCZ8sT6WWuNUxMBqkXXIOs4S120uk,14511
-autocoder/models.py,sha256=xwWPcegwx945g433UZXna-7HBdnHWCq8oEfHm-HKIDQ,8651
-autocoder/version.py,sha256=Mlq4zYTZeRq2mquyM-8m1qr6sjxAHZpSDVyjrKqhayc,23
+autocoder/models.py,sha256=rG7ckiKlers-XoO1gWxNK-Y-IbqD82WS3qFMPHqvFsc,9072
+autocoder/version.py,sha256=ZqpPzT_HdggyvazgM4cfgBLNgZaYhDcBMd4HUgyfqGc,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
@@ -25,19 +25,19 @@ 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=6maackdzrYnUPvpgVPl92JdMOnw7X4n3EnEQA9OnLGE,11984
+autocoder/common/__init__.py,sha256=wJIAB6EOeCmo9UXCGC7E9u_GGWO5DjXW78wUi3hz1lI,12272
 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=x9Zjwvu9OZJjTmswwyimlMb1pvngUAF9_3oNQQut2i4,17634
+autocoder/common/auto_coder_lang.py,sha256=xxmO4Htil4S2KvWa3RRi9rMgrod_io1MbctJkRXBfDg,20760
 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
 autocoder/common/code_auto_execute.py,sha256=4KXGmiGObr_B1d6tzV9dwS6MifCSc3Gm4j2d6ildBXQ,6867
-autocoder/common/code_auto_generate.py,sha256=E8r3VI88hPBPhU5t56qnmeL_fWtCWX1CJvaOachGa2Y,12014
-autocoder/common/code_auto_generate_diff.py,sha256=dmMgN1yIOjJfiYFnzXZuktVFdj4_XR_Tavwx_ysm53U,17846
-autocoder/common/code_auto_generate_editblock.py,sha256=NI_dFwy1VhvdjvARb04-B1AGfgW9z4P1BfWMm-blnaU,19447
-autocoder/common/code_auto_generate_strict_diff.py,sha256=uf5P5B8ly0MP3jCK2PaYJiPLktd1cRRPouwkkaf-DfY,16457
+autocoder/common/code_auto_generate.py,sha256=74wCscxVEnY_VDkHcr-QA3b79RhDR_OeVPOI7UKRJwA,13040
+autocoder/common/code_auto_generate_diff.py,sha256=bns5KZq9ozvUtyqIUWsDNUtah-TTOsE7yRXHYGlrtT4,18872
+autocoder/common/code_auto_generate_editblock.py,sha256=LcGfG4bJVCVsWehex7MYWDF4NX0B2Rp2ALSh-27MclA,20472
+autocoder/common/code_auto_generate_strict_diff.py,sha256=JvKnD5Ph3JtAiVIO_k_XKUnVBeUxwLw_AHF_xWWtX7c,17488
 autocoder/common/code_auto_merge.py,sha256=-ksBjj4ZVcbY_tVH4JLXAMSRtsgaSxrSZ5-MOl9cAgE,7354
 autocoder/common/code_auto_merge_diff.py,sha256=qpEuHJEgX6sWK7EDFEKqcYkyI28wOyM4pytyl8BLohY,15350
 autocoder/common/code_auto_merge_editblock.py,sha256=sxgYMLMACRwJvw-bABkdDHezPelsDFrOCpGuhtT5Dzs,17504
@@ -76,20 +76,22 @@ 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=sfh3pCasy622Jm0_AIKU7xtR-tqY2tS2_9YJsEd0FJY,22753
+autocoder/dispacher/actions/action.py,sha256=k5ank4nHYQ-Rh6gIeeVgWJL42aNVOtgkDZYSZ0jhhiQ,23502
 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=22EZL3mLFxgsEZ8ymPCGvaHCJFnrW6C_prp1ykYCuEY,6335
+autocoder/dispacher/actions/plugins/action_regex_project.py,sha256=rKQtRo2icVrBhzkn1HNhch1eozMvvBo8x-_G1sDMIBY,6495
 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=1KIGPCtxQN0OdErAco9OmGTd5hB8WJTpWGrxsGLsTcE,12634
+autocoder/index/entry.py,sha256=H897HGBsIi9bf0T-M-i3IWQcFG6uFRSjX70eicD5oto,12795
 autocoder/index/for_command.py,sha256=BFvljE4t6VaMBGboZAuhUCzVK0EitCy_n5D_7FEnihw,3204
-autocoder/index/index.py,sha256=hVAIyF10N9hxKMWHA_ibYygGRZYJQZfZxRRrhQhrTvk,21225
+autocoder/index/index.py,sha256=GeofteDTq4Ye0cSBuK1CqQD43NMrrHOg5dfbv_7fVzk,25312
 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=APu34iSvWhtlLtWgkj8N3Vo4oW1TegtZQq2bwDX_cs4,8031
 autocoder/index/filter/quick_filter.py,sha256=5toipv7XwLsmG_UaqrElpGNjKXq_0bcvFr8W80vT44g,15206
+autocoder/privacy/__init__.py,sha256=LnIVvGu_K66zCE-yhN_-dPO8R80pQyedCsXJ7wRqQaI,72
+autocoder/privacy/model_filter.py,sha256=-N9ZvxxDKpxU7hkn-tKv-QHyXjvkCopUaKgvJwTOGQs,3369
 autocoder/pyproject/__init__.py,sha256=bRuGxFV4QyE85xVjDzeMFmlLVqGbbcFs09FI15Uss4Q,14423
 autocoder/rag/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 autocoder/rag/api_server.py,sha256=dRbhAZVRAOlZ64Cnxf4_rKb4iJwHnrWS9Zr67IVORw0,7288
@@ -133,6 +135,7 @@ autocoder/utils/conversation_store.py,sha256=sz-hhY7sttPAUOAQU6Pze-5zJc3j0_Emj22
 autocoder/utils/llm_client_interceptors.py,sha256=FEHNXoFZlCjAHQcjPRyX8FOMjo6rPXpO2AJ2zn2KTTo,901
 autocoder/utils/llms.py,sha256=HM5K_v4AcuWo65lgcp66DEqaU9-fjoT7mcI1iv2Fopg,3839
 autocoder/utils/log_capture.py,sha256=I-bsJFLWoGUiX-GKoZsH9kWJCKSV7ZlUnRt7jh-fOL0,1548
+autocoder/utils/model_provider_selector.py,sha256=g5O9frBWkXR7iqjYDdTvhoxzTQx0NaPVPu9M2ItqhpE,7602
 autocoder/utils/multi_turn.py,sha256=unK9OpqVRbK6uIcTKXgggX2wNmyj7s5eyEAQ2xUwHoM,88
 autocoder/utils/operate_config_api.py,sha256=99YAKsuUFLPwrRvj0CJal_bAPgyiXWMma6ZKMU56thw,5790
 autocoder/utils/print_table.py,sha256=ZMRhCA9DD0FUfKyJBWd5bDdj1RrtPtgOMWSJwtvZcLs,403
@@ -146,9 +149,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=xWXqICANbDOovH4wcFW1eSI7lB7TjXbk1mSU4bTKEW4,11434
 autocoder/utils/chat_auto_coder_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-auto_coder-0.1.256.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-auto_coder-0.1.256.dist-info/METADATA,sha256=NrC0Y2oSS6lDTeKbXKk3F1QnZ7_3Kie7pho_i7JhwJk,2616
-auto_coder-0.1.256.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-auto_coder-0.1.256.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
-auto_coder-0.1.256.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
-auto_coder-0.1.256.dist-info/RECORD,,
+auto_coder-0.1.258.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+auto_coder-0.1.258.dist-info/METADATA,sha256=t48Rt7ffsIuQKceJzcS7igvP4sDu_40I3DkSG2vYhBE,2616
+auto_coder-0.1.258.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+auto_coder-0.1.258.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
+auto_coder-0.1.258.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
+auto_coder-0.1.258.dist-info/RECORD,,

autocoder/auto_coder.py

@@ -47,6 +47,8 @@ from autocoder.common.utils_code_auto_generate import stream_chat_with_continue
 from autocoder.utils.auto_coder_utils.chat_stream_out import stream_out
 from autocoder.common.printer import Printer
 from autocoder.rag.token_counter import count_tokens
+from autocoder.privacy.model_filter import ModelPathFilter
+
 console = Console()
 
 
@@ -317,53 +319,7 @@ def main(input_args: Optional[List[str]] = None):
                     "saas.model": model_info["model_name"],
                     "saas.is_reasoning": model_info["is_reasoning"]
                 }
-            )                
-                        
-            if models_module.check_model_exists("deepseek_r1_chat"):
-                r1_model_info = models_module.get_model_by_name("deepseek_r1_chat")  
-                api_key = r1_model_info["api_key"]
-                chat_llm = byzerllm.SimpleByzerLLM(default_model_name="deepseek_r1_chat")
-                chat_llm.deploy(
-                    model_path="",
-                    pretrained_model_type="saas/openai",
-                    udf_name="deepseek_r1_chat",
-                    infer_params={
-                        "saas.base_url": "https://api.deepseek.com/v1",
-                        "saas.api_key": api_key,
-                        "saas.model": "deepseek-reasoner",
-                        "saas.is_reasoning": True
-                    }
-                )
-
-                generate_rerank_llm = byzerllm.SimpleByzerLLM(default_model_name="deepseek_r1_chat")
-                generate_rerank_llm.deploy(
-                    model_path="",
-                    pretrained_model_type="saas/openai",
-                    udf_name="deepseek_r1_chat",
-                    infer_params={
-                        "saas.base_url": "https://api.deepseek.com/v1",
-                        "saas.api_key": api_key,
-                        "saas.model": "deepseek-reasoner",
-                        "saas.is_reasoning": True
-                    }
-                )
-
-                index_filter_llm = byzerllm.SimpleByzerLLM(default_model_name="deepseek_r1_chat")
-                index_filter_llm.deploy(
-                    model_path="",
-                    pretrained_model_type="saas/openai",
-                    udf_name="deepseek_r1_chat",
-                    infer_params={
-                        "saas.base_url": "https://api.deepseek.com/v1",
-                        "saas.api_key": api_key,
-                        "saas.model": "deepseek-reasoner",
-                        "saas.is_reasoning": True
-                    }
-                )
-                                                
-                llm.setup_sub_client("chat_model", chat_llm)
-                llm.setup_sub_client("generate_rerank_model", generate_rerank_llm)
-                llm.setup_sub_client("index_filter_model", index_filter_llm)
+            )                                                    
 
         if args.product_mode == "lite":                                    
             # Set up default models based on configuration
@@ -947,6 +903,17 @@ def main(input_args: Optional[List[str]] = None):
             # )
             return
         elif raw_args.agent_command == "project_reader":
+                        
+            target_llm = llm.get_sub_client("planner_model")
+            if not target_llm:
+                target_llm = llm
+            model_filter = ModelPathFilter.from_model_object(target_llm, args)
+            if model_filter.has_rules():
+                printer = Printer()
+                msg = printer.get_message_from_key_with_format("model_has_access_restrictions",                                            
+                                           model_name=",".join(get_llm_names(target_llm)))
+                raise ValueError(msg)
+
             from autocoder.agent.project_reader import ProjectReader
 
             project_reader = ProjectReader(args, llm)
@@ -1194,10 +1161,23 @@ 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
+            
+            # Apply model filter for chat_llm            
+            model_filter = ModelPathFilter.from_model_object(chat_llm, args)
+            filtered_sources = []
+            printer = Printer()            
+            for source in sources:
+                if model_filter.is_accessible(source.module_name):
+                    filtered_sources.append(source)
+                else:                    
+                    printer.print_in_terminal("index_file_filtered", 
+                                               style="yellow",
+                                               file_path=source.module_name, 
+                                               model_name=",".join(get_llm_names(chat_llm)))
+            
             s = build_index_and_filter_files(
-                llm=llm, args=args, sources=sources)                        
+                llm=llm, args=args, sources=filtered_sources).to_str()                        
             
             if s:
                 pre_conversations.append(

autocoder/chat_auto_coder.py

@@ -297,23 +297,22 @@ def initialize_system(args):
 
     init_project()
 
-    if args.product_mode == "lite":
-        # Setup deepseek api key
-        api_key_dir = os.path.expanduser("~/.auto-coder/keys")
-        api_key_file = os.path.join(api_key_dir, "api.deepseek.com")
-        
-        if not os.path.exists(api_key_file):
-            print_status(get_message("model_not_available"), "warning")
-            api_key = prompt(HTML(f"<b>{get_message('enter_api_key')} </b>"))
-            
-            # Create directory if it doesn't exist
-            os.makedirs(api_key_dir, exist_ok=True)
-            
-            # Save the API key
-            with open(api_key_file, "w") as f:
-                f.write(api_key)
-            
-            print_status(f"API key saved successfully: {api_key_file}", "success")                    
+    if args.product_mode == "lite":        
+        from autocoder.utils.model_provider_selector import ModelProviderSelector
+        from autocoder import models as models_module
+        if not models_module.check_model_exists("v3_chat") or not models_module.check_model_exists("r1_chat"):
+            model_provider_selector = ModelProviderSelector()
+            model_provider_info = model_provider_selector.select_provider()
+            if model_provider_info is not None:
+                models_json_list = model_provider_selector.to_models_json(model_provider_info)
+                models_module.add_and_activate_models(models_json_list)
+                r1_model = models_json_list[0]['name']
+                v3_model = models_json_list[1]['name']            
+                configure(f"model:{v3_model}", skip_print=True)
+                configure(f"chat_model:{r1_model}", skip_print=True)
+                configure(f"generate_rerank_model:{r1_model}", skip_print=True)
+                configure(f"code_model:{v3_model}", skip_print=True)
+                configure(f"index_filter_model:{r1_model}", skip_print=True)            
 
     if args.product_mode == "pro":
         # Check if Ray is running

autocoder/chat_auto_coder_lang.py

@@ -128,7 +128,7 @@ MESSAGES = {
         "official_doc": "Official Documentation: https://uelng8wukz.feishu.cn/wiki/NhPNwSRcWimKFIkQINIckloBncI",
     },
     "zh": {
-        "commit_generating": "{{ model_name }} 正在生成提交信息...", 
+        "commit_generating": "{{ model_name }} 正在生成提交信息...",
         "commit_message": "{{ model_name }} 生成的提交信息: {{ message }}",
         "commit_failed": "{{ model_name }} 生成提交信息失败: {{ error }}",
         "mcp_remove_error": "移除 MCP 服务器时出错:{error}",

autocoder/common/__init__.py

@@ -14,7 +14,13 @@ class SourceCode(pydantic.BaseModel):
     tokens: int = -1
     metadata: Dict[str, Any] = {}
 
+class SourceCodeList():
+    def __init__(self, sources: List[SourceCode]):
+        self.sources = sources
 
+    def to_str(self):
+        return "\n".join([f"##File: {source.module_name}\n{source.source_code}\n" for source in self.sources])
+    
 class TranslateReadme(pydantic.BaseModel):
     filename: str = pydantic.Field(..., description="需要翻译的文件路径")
     content: str = pydantic.Field(..., description="翻译后的内容")
@@ -362,6 +368,7 @@ class AutoCoderArgs(pydantic.BaseModel):
     keep_only_reasoning_content: Optional[bool] = False
 
     in_code_apply: bool = False
+    model_filter_path: Optional[str] = None
 
     class Config:
         protected_namespaces = ()

autocoder/common/auto_coder_lang.py

@@ -3,6 +3,20 @@ from byzerllm.utils import format_str_jinja2
 
 MESSAGES = {
     "en": {
+        "model_provider_select_title": "Select Model Provider",
+        "model_provider_select_text": "Please select your model provider:",
+        "model_provider_volcano": "Volcano Engine",
+        "model_provider_siliconflow": "SiliconFlow AI",
+        "model_provider_deepseek": "DeepSeek Official",        
+        "model_provider_api_key_title": "API Key",
+        "model_provider_volcano_api_key_text": "Please enter your Volcano Engine API key:",
+        "model_provider_volcano_r1_text": "Please enter your Volcano Engine R1 endpoint (format: ep-20250204215011-vzbsg):",
+        "model_provider_volcano_v3_text": "Please enter your Volcano Engine V3 endpoint (format: ep-20250204215011-vzbsg):",
+        "model_provider_siliconflow_api_key_text": "Please enter your SiliconFlow AI API key:",
+        "model_provider_deepseek_api_key_text": "Please enter your DeepSeek API key:",
+        "model_provider_selected": "Provider configuration completed successfully! You can use /models command to view, add and modify all models later.",
+        "model_provider_success_title": "Success",
+        "index_file_filtered": "File {{file_path}} is filtered by model {{model_name}} restrictions",
         "models_no_active": "No active models found",
         "models_speed_test_results": "Model Speed Test Results",
         "models_testing": "Testing model: {{name}}...",
@@ -11,7 +25,7 @@ MESSAGES = {
         "generation_cancelled": "[Interrupted] Generation cancelled",
         "model_not_found": "Model {{model_name}} not found",
         "generating_shell_script": "Generating Shell Script",
-        "new_session_started": "New session started. Previous chat history has been archived.",    
+        "new_session_started": "New session started. Previous chat history has been archived.",
         "memory_save_success": "✅ Saved to your memory",
         "file_decode_error": "Failed to decode file: {{file_path}}. Tried encodings: {{encodings}}",
         "file_write_error": "Failed to write file: {{file_path}}. Error: {{error}}",
@@ -21,7 +35,7 @@ MESSAGES = {
         "no_latest_commit": "Unable to get latest commit information",
         "code_review_error": "Code review process error: {{error}}",
         "index_file_too_large": "⚠️ File {{ file_path }} is too large ({{ file_size }} > {{ max_length }}), splitting into chunks...",
-        "index_update_success": "✅ {{ model_name }} Successfully updated index for {{ file_path }} (md5: {{ md5 }}) in {{ duration }}s",
+        "index_update_success": "✅ {{ model_name }} Successfully updated index for {{ file_path }} (md5: {{ md5 }}) in {{ duration }}s, input_tokens: {{ input_tokens }}, output_tokens: {{ output_tokens }}, input_cost: {{ input_cost }}, output_cost: {{ output_cost }}",
         "index_build_error": "❌ {{ model_name }} Error building index for {{ file_path }}: {{ error }}",
         "index_build_summary": "📊 Total Files: {{ total_files }}, Need to Build Index: {{ num_files }}",
         "building_index_progress": "⏳ Building Index: {{ counter }}/{{ num_files }}...",
@@ -30,7 +44,7 @@ MESSAGES = {
         "index_threads_completed": "✅ Completed {{ completed_threads }}/{{ total_threads }} threads",
         "index_related_files_fail": "⚠️ Failed to find related files for chunk {{ chunk_count }}",
         "index_file_removed": "🗑️ Removed non-existent file index: {{ file_path }}",
-        "index_file_saved": "💾 Saved index file, updated {{ updated_files }} files, removed {{ removed_files }} files",
+        "index_file_saved": "💾 Saved index file, updated {{ updated_files }} files, removed {{ removed_files }} files, input_tokens: {{ input_tokens }}, output_tokens: {{ output_tokens }}, input_cost: {{ input_cost }}, output_cost: {{ output_cost }}",
         "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"
@@ -81,7 +95,7 @@ MESSAGES = {
         "begin_index_source_code": "🚀 Begin to index source code in {{ source_dir }}",
         "stream_out_stats": "Model: {{ model_name }}, Total time: {{ elapsed_time }} seconds, First token time: {{ first_token_time }} seconds, Speed: {{ speed }} tokens/s, Input tokens: {{ input_tokens }}, Output tokens: {{ output_tokens }}, Input cost: {{ input_cost }}, Output cost: {{ output_cost }}",
         "quick_filter_stats": "{{ model_names }} 快速过滤器完成，耗时 {{ elapsed_time }} 秒，输入token数: {{ input_tokens }}, 输出token数: {{ output_tokens }}, 输入成本: {{ input_cost }}, 输出成本: {{ output_cost }}",
-        "upsert_file": "✅ Updated file: {{ file_path }}",  
+        "upsert_file": "✅ Updated file: {{ file_path }}",
         "unmerged_blocks_title": "Unmerged Blocks",
         "quick_filter_title": "{{ model_name }} is analyzing how to filter context...",
         "quick_filter_failed": "❌ Quick filter failed: {{ error }}. ",
@@ -95,8 +109,23 @@ MESSAGES = {
         "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 }}",
+        "model_has_access_restrictions": "{{model_name}} has access restrictions, cannot use the current function",
     },
     "zh": {
+        "model_provider_select_title": "选择模型供应商",
+        "model_provider_select_text": "请选择您的模型供应商：",
+        "model_provider_volcano": "火山方舟",
+        "model_provider_siliconflow": "硅基流动",
+        "model_provider_deepseek": "DeepSeek官方",        
+        "model_provider_api_key_title": "API密钥",
+        "model_provider_volcano_api_key_text": "请输入您的火山方舟API密钥：",
+        "model_provider_volcano_r1_text": "请输入您的火山方舟 R1 推理点(格式如: ep-20250204215011-vzbsg)：",
+        "model_provider_volcano_v3_text": "请输入您的火山方舟 V3 推理点(格式如: ep-20250204215011-vzbsg)：",
+        "model_provider_siliconflow_api_key_text": "请输入您的硅基流动API密钥：",
+        "model_provider_deepseek_api_key_text": "请输入您的DeepSeek API密钥：",
+        "model_provider_selected": "供应商配置已成功完成！后续你可以使用 /models 命令，查看，新增和修改所有模型",
+        "model_provider_success_title": "成功",
+        "index_file_filtered": "文件 {{file_path}} 被模型 {{model_name}} 的访问限制过滤",
         "models_no_active": "未找到激活的模型",
         "models_speed_test_results": "模型速度测试结果",
         "models_testing": "正在测试模型: {{name}}...",
@@ -114,7 +143,7 @@ MESSAGES = {
         "no_latest_commit": "无法获取最新的提交信息",
         "code_review_error": "代码审查过程出错: {{error}}",
         "index_file_too_large": "⚠️ 文件 {{ file_path }} 过大 ({{ file_size }} > {{ max_length }}), 正在分块处理...",
-        "index_update_success": "✅ {{ model_name }} 成功更新 {{ file_path }} 的索引 (md5: {{ md5 }}), 耗时 {{ duration }} 秒",
+        "index_update_success": "✅ {{ model_name }} 成功更新 {{ file_path }} 的索引 (md5: {{ md5 }}), 耗时 {{ duration }} 秒, 输入token数: {{ input_tokens }}, 输出token数: {{ output_tokens }}, 输入成本: {{ input_cost }}, 输出成本: {{ output_cost }}",
         "index_build_error": "❌ {{ model_name }} 构建 {{ file_path }} 索引时出错: {{ error }}",
         "index_build_summary": "📊 总文件数: {{ total_files }}, 需要构建索引: {{ num_files }}",
         "building_index_progress": "⏳ 正在构建索引: {{ counter }}/{{ num_files }}...",
@@ -123,7 +152,7 @@ MESSAGES = {
         "index_threads_completed": "✅ 已完成 {{ completed_threads }}/{{ total_threads }} 个线程",
         "index_related_files_fail": "⚠️ 无法为块 {{ chunk_count }} 找到相关文件",
         "index_file_removed": "🗑️ 已移除不存在的文件索引：{{ file_path }}",
-        "index_file_saved": "💾 已保存索引文件，更新了 {{ updated_files }} 个文件，移除了 {{ removed_files }} 个文件",
+        "index_file_saved": "💾 已保存索引文件，更新了 {{ updated_files }} 个文件，移除了 {{ removed_files }} 个文件，输入token数: {{ input_tokens }}, 输出token数: {{ output_tokens }}, 输入成本: {{ input_cost }}, 输出成本: {{ output_cost }}",
         "human_as_model_instructions": (
             "您现在处于人类作为模型模式。内容已复制到您的剪贴板。\n"
             "系统正在等待您的输入。完成后，在新行输入'EOF'提交。\n"
@@ -188,8 +217,8 @@ MESSAGES = {
         "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 }}",
-    },
-}
+        "model_has_access_restrictions": "{{model_name}} 有访问限制，无法使用当前功能",
+    }}
 
 
 def get_system_language():
@@ -203,5 +232,6 @@ def get_message(key):
     lang = get_system_language()
     return MESSAGES.get(lang, MESSAGES['en']).get(key, MESSAGES['en'][key])
 
+
 def get_message_with_format(msg_key: str, **kwargs):
     return format_str_jinja2(get_message(msg_key), **kwargs)

autocoder/common/code_auto_generate.py

@@ -11,6 +11,8 @@ import json
 from autocoder.common.printer import Printer
 from autocoder.rag.token_counter import count_tokens
 from autocoder.utils import llms as llm_utils
+from autocoder.common import SourceCodeList
+from autocoder.privacy.model_filter import ModelPathFilter
 
 
 class CodeAutoGenerate:
@@ -156,10 +158,27 @@ class CodeAutoGenerate:
         }
 
     def single_round_run(
-        self, query: str, source_content: str
+        self, query: str, source_code_list: SourceCodeList
     ) -> Tuple[List[str], Dict[str, str]]:
         llm_config = {"human_as_model": self.args.human_as_model}
 
+        # Apply model filter for code_llm  
+        printer = Printer()      
+        for llm in self.llms:
+            model_filter = ModelPathFilter.from_model_object(llm, self.args)
+            filtered_sources = []
+            for source in source_code_list.sources:
+                if model_filter.is_accessible(source.module_name):
+                    filtered_sources.append(source)
+                else:
+                    printer.print_in_terminal("index_file_filtered", 
+                                               style="yellow",
+                                               file_path=source.module_name, 
+                                               model_name=",".join(llm_utils.get_llm_names(llm)))
+            
+        source_code_list = SourceCodeList(filtered_sources)
+        source_content = source_code_list.to_str()
+
         if self.args.request_id and not self.args.skip_events:
             queue_communicate.send_event_no_wait(
                 request_id=self.args.request_id,
@@ -262,10 +281,11 @@ class CodeAutoGenerate:
         return CodeGenerateResult(contents=results, conversations=conversations_list, metadata=statistics)
 
     def multi_round_run(
-        self, query: str, source_content: str, max_steps: int = 10
+        self, query: str, source_code_list: SourceCodeList, max_steps: int = 10
     ) -> Tuple[List[str], List[Dict[str, str]]]:
         llm_config = {"human_as_model": self.args.human_as_model}
         result = []
+        source_content = source_code_list.to_str()
 
         if self.args.template == "common":
             init_prompt = self.multi_round_instruction.prompt(

autocoder/common/code_auto_generate_diff.py

@@ -2,6 +2,7 @@ from typing import List, Dict, Tuple
 from autocoder.common.types import Mode, CodeGenerateResult
 from autocoder.common import AutoCoderArgs
 import byzerllm
+from autocoder.privacy.model_filter import ModelPathFilter
 from autocoder.utils.queue_communicate import queue_communicate, CommunicateEvent, CommunicateEventType
 from autocoder.common import sys_prompt
 from concurrent.futures import ThreadPoolExecutor
@@ -10,7 +11,7 @@ from autocoder.common.utils_code_auto_generate import chat_with_continue
 from autocoder.common.printer import Printer
 from autocoder.rag.token_counter import count_tokens
 from autocoder.utils import llms as llm_utils
-
+from autocoder.common import SourceCodeList
 
 class CodeAutoGenerateDiff:
     def __init__(
@@ -302,9 +303,10 @@ class CodeAutoGenerateDiff:
         }
 
     def single_round_run(
-        self, query: str, source_content: str
+        self, query: str, source_code_list: SourceCodeList
     ) -> CodeGenerateResult:
         llm_config = {"human_as_model": self.args.human_as_model}
+        source_content = source_code_list.to_str()
 
         if self.args.template == "common":
             init_prompt = self.single_round_instruction.prompt(
@@ -410,11 +412,28 @@ class CodeAutoGenerateDiff:
         return CodeGenerateResult(contents=results, conversations=conversations_list, metadata=statistics)
 
     def multi_round_run(
-        self, query: str, source_content: str, max_steps: int = 10
+        self, query: str, source_code_list: SourceCodeList, max_steps: int = 10
     ) -> CodeGenerateResult:
+        
+        # Apply model filter for code_llm
+        printer = Printer()
+        for llm in self.llms:
+            model_filter = ModelPathFilter.from_model_object(llm, self.args)
+            filtered_sources = []
+            for source in source_code_list.sources:
+                if model_filter.is_accessible(source.module_name):
+                    filtered_sources.append(source)
+                else:
+                    printer.print_in_terminal("index_file_filtered", 
+                                               style="yellow",
+                                               file_path=source.path, 
+                                               model_name=",".join(llm_utils.get_llm_names(llm)))
+            
+        source_code_list = SourceCodeList(filtered_sources)
+        
         llm_config = {"human_as_model": self.args.human_as_model}
         result = []
-
+        source_content = source_code_list.to_str()
         if self.args.template == "common":
             init_prompt = self.multi_round_instruction.prompt(
                 instruction=query, content=source_content, context=self.args.context

autocoder/common/code_auto_generate_editblock.py

@@ -3,6 +3,7 @@ from autocoder.common.types import Mode, CodeGenerateResult
 from autocoder.common import AutoCoderArgs
 import byzerllm
 from autocoder.common import sys_prompt
+from autocoder.privacy.model_filter import ModelPathFilter
 from autocoder.utils.queue_communicate import (
     queue_communicate,
     CommunicateEvent,
@@ -14,6 +15,7 @@ from autocoder.common.utils_code_auto_generate import chat_with_continue
 from autocoder.common.printer import Printer
 from autocoder.rag.token_counter import count_tokens
 from autocoder.utils import llms as llm_utils
+from autocoder.common import SourceCodeList
 
 
 class CodeAutoGenerateEditBlock:
@@ -384,10 +386,29 @@ class CodeAutoGenerateEditBlock:
         }
 
     def single_round_run(
-        self, query: str, source_content: str
+        self, query: str, source_code_list: SourceCodeList
     ) -> CodeGenerateResult:
+        
+        # Apply model filter for code_llm
+        printer = Printer()
+        for llm in self.llms:
+            model_filter = ModelPathFilter.from_model_object(llm, self.args)
+            filtered_sources = []
+            for source in source_code_list.sources:
+                if model_filter.is_accessible(source.module_name): 
+                    filtered_sources.append(source)
+                else:
+                    printer.print_in_terminal("index_file_filtered", 
+                                               style="yellow",
+                                               file_path=source.module_name, 
+                                               model_name=",".join(llm_utils.get_llm_names(llm)))
+
+        source_code_list = SourceCodeList(filtered_sources)
+        
         llm_config = {"human_as_model": self.args.human_as_model}
 
+        source_content = source_code_list.to_str()
+
         if self.args.template == "common":
             init_prompt = self.single_round_instruction.prompt(
                 instruction=query, content=source_content, context=self.args.context
@@ -498,10 +519,11 @@ class CodeAutoGenerateEditBlock:
         return CodeGenerateResult(contents=results, conversations=conversations_list, metadata=statistics)
 
     def multi_round_run(
-        self, query: str, source_content: str, max_steps: int = 10
+        self, query: str, source_code_list: SourceCodeList, max_steps: int = 10
     ) -> CodeGenerateResult:
         llm_config = {"human_as_model": self.args.human_as_model}
         result = []
+        source_content = source_code_list.to_str()
 
         if self.args.template == "common":
             init_prompt = self.multi_round_instruction.prompt(

autocoder/common/code_auto_generate_strict_diff.py

@@ -10,7 +10,8 @@ from autocoder.common.utils_code_auto_generate import chat_with_continue
 from autocoder.common.printer import Printer
 from autocoder.rag.token_counter import count_tokens
 from autocoder.utils import llms as llm_utils
-
+from autocoder.common import SourceCodeList
+from autocoder.privacy.model_filter import ModelPathFilter
 class CodeAutoGenerateStrictDiff:
     def __init__(
         self, llm: byzerllm.ByzerLLM, args: AutoCoderArgs, action=None
@@ -272,9 +273,10 @@ class CodeAutoGenerateStrictDiff:
         }
 
     def single_round_run(
-        self, query: str, source_content: str
+        self, query: str, source_code_list: SourceCodeList
     ) -> CodeGenerateResult:
         llm_config = {"human_as_model": self.args.human_as_model}
+        source_content = source_code_list.to_str()
 
         if self.args.template == "common":
             init_prompt = self.single_round_instruction.prompt(
@@ -379,10 +381,28 @@ class CodeAutoGenerateStrictDiff:
         return CodeGenerateResult(contents=results, conversations=conversations_list, metadata=statistics)
 
     def multi_round_run(
-        self, query: str, source_content: str, max_steps: int = 10
+        self, query: str, source_code_list: SourceCodeList, max_steps: int = 10
     ) -> CodeGenerateResult:
+        
+        # Apply model filter for code_llm
+        printer = Printer()
+        for llm in self.llms:
+            model_filter = ModelPathFilter.from_model_object(llm, self.args)
+            filtered_sources = []
+            for source in source_code_list.sources:
+                if model_filter.is_accessible(source.module_name): 
+                    filtered_sources.append(source)
+                else:
+                    printer.print_in_terminal("index_file_filtered", 
+                                               style="yellow",
+                                               file_path=source.module_name, 
+                                               model_name=",".join(llm_utils.get_llm_names(llm)))
+        
+        source_code_list = SourceCodeList(filtered_sources)
+        
         llm_config = {"human_as_model": self.args.human_as_model}
         result = []
+        source_content = source_code_list.to_str()
 
         if self.args.template == "common":
             init_prompt = self.multi_round_instruction.prompt(