auto-coder 0.1.257__py3-none-any.whl → 0.1.259__py3-none-any.whl

{auto_coder-0.1.257.dist-info → auto_coder-0.1.259.dist-info}/METADATA

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

{auto_coder-0.1.257.dist-info → auto_coder-0.1.259.dist-info}/RECORD

@@ -1,17 +1,17 @@
 autocoder/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-autocoder/auto_coder.py,sha256=whonXfuZrcveFliNXLglxo6CXtwzIqo-7QppjJqag0g,64811
+autocoder/auto_coder.py,sha256=_fWeOxGHcRXMMXlzZqo9f0j_GuzNweq68I7DkgRnQoM,63751
 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.py,sha256=nwgsXO2-scssWStjX3S910tDp-OZXZRddSYrpyC4Nq0,29021
+autocoder/auto_coder_rag_client_mcp.py,sha256=QRxUbjc6A8UmDMQ8lXgZkjgqtq3lgKYeatJbDY6rSo0,6270
 autocoder/auto_coder_rag_mcp.py,sha256=-RrjNwFaS2e5v8XDIrKR-zlUNUE8UBaeOtojffBrvJo,8521
 autocoder/auto_coder_server.py,sha256=XU9b4SBH7zjPPXaTWWHV4_zJm-XYa6njuLQaplYJH_c,20290
 autocoder/benchmark.py,sha256=Ypomkdzd1T3GE6dRICY3Hj547dZ6_inqJbBJIp5QMco,4423
-autocoder/chat_auto_coder.py,sha256=tfr0qu5yMHZFetL4ef3FvB7d29kjY8qNxW6YKBLXCAA,109659
+autocoder/chat_auto_coder.py,sha256=327wiq2RMXPyiAe3m4Yh3JdnDy2G1AHmyqAFWwBRPrY,110835
 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=rG7ckiKlers-XoO1gWxNK-Y-IbqD82WS3qFMPHqvFsc,9072
-autocoder/version.py,sha256=c2jqchALPLGbBkkO-_BkAKh-2Kve9yyPv68Ff4S7sUE,23
+autocoder/version.py,sha256=w4YfV0iY2RJqPATbfwM5loNkK_-swqYe1AMQy-1r1No,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
@@ -29,7 +29,7 @@ autocoder/common/__init__.py,sha256=wJIAB6EOeCmo9UXCGC7E9u_GGWO5DjXW78wUi3hz1lI,
 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=U5Pt_iIjNk1cN8wxI9hL73zcTQzmmAtDAutXjJCVhc0,20724
+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
@@ -45,7 +45,7 @@ autocoder/common/code_auto_merge_strict_diff.py,sha256=P0nKNkBrFMybTSZ7kOdA_Jixo
 autocoder/common/code_modification_ranker.py,sha256=oG9rCekGsYwE9gNdkIKQ6uKt6uaXpwrC17-FV5Wo-fQ,8187
 autocoder/common/command_completer.py,sha256=IShrZJSpR-Q_MCj_aCVdVyscLYDKj5ZQK357QBcQ_oQ,9420
 autocoder/common/command_generator.py,sha256=-hmbD_AnCa5HxL4BznuEfYAf_l8AxU5fAG5F0sM_fuE,2116
-autocoder/common/command_templates.py,sha256=mnB3n8i0yjH1mqzyClEg8Wpr9VbZV44kxky66Zu6OJY,8557
+autocoder/common/command_templates.py,sha256=lAdr0-iyJKY2dOH2mZ0Tm3GlT_a1Oj8mgdKXmDiQN3A,8654
 autocoder/common/const.py,sha256=eTjhjh4Aj4CUzviJ81jaf3Y5cwqsLATySn2wJxaS6RQ,2911
 autocoder/common/files.py,sha256=CguxG9digkWBJpRaILErZmL_G5ryPRahPmPFWGB7X18,1973
 autocoder/common/git_utils.py,sha256=zxgQt2PukabV_21podylAUzTY7Xk60bsQ7MQYw4s-Tg,23234
@@ -82,7 +82,7 @@ autocoder/dispacher/actions/plugins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQ
 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=H897HGBsIi9bf0T-M-i3IWQcFG6uFRSjX70eicD5oto,12795
+autocoder/index/entry.py,sha256=iuvORjbmkNXBaN8p-njwETCVN2mWwu_DIgCJeM2rzKs,12799
 autocoder/index/for_command.py,sha256=BFvljE4t6VaMBGboZAuhUCzVK0EitCy_n5D_7FEnihw,3204
 autocoder/index/index.py,sha256=GeofteDTq4Ye0cSBuK1CqQD43NMrrHOg5dfbv_7fVzk,25312
 autocoder/index/symbols_utils.py,sha256=CjcjUVajmJZB75Ty3a7kMv1BZphrm-tIBAdOJv6uo-0,2037
@@ -101,7 +101,7 @@ autocoder/rag/llm_wrapper.py,sha256=sbDxCANiZyWb_ocqNgqu2oy3c2t8orPNRGleEs-Uwl8,
 autocoder/rag/long_context_rag.py,sha256=wmNmGsXN8RAFl6e9HaVzlwISXV9D-3bvf2qiaWjRy7w,24646
 autocoder/rag/rag_config.py,sha256=8LwFcTd8OJWWwi1_WY4IzjqgtT6RyE2j4PjxS5cCTDE,802
 autocoder/rag/rag_entry.py,sha256=6TKtErZ0Us9XSV6HgRKXA6yR3SiZGPHpynOKSaR1wgE,2463
-autocoder/rag/raw_rag.py,sha256=yS2Ur6kG0IRjhCj2_VonwxjY_xls_E62jO5Gz5j2nqE,2952
+autocoder/rag/raw_rag.py,sha256=BOr0YGf3umjqXOIDVO1LXQ0bIHx8hzBdiubND2ezyxc,2946
 autocoder/rag/relevant_utils.py,sha256=OGfp98OXG4jr3jNmtHIeXGPF8mOlIbTnolPIVTZzYZU,929
 autocoder/rag/simple_directory_reader.py,sha256=LkKreCkNdEOoL4fNhc3_hDoyyWTQUte4uqextISRz4U,24485
 autocoder/rag/simple_rag.py,sha256=I902EUqOK1WM0Y2WFd7RzDJYofElvTZNLVCBtX5A9rc,14885
@@ -135,7 +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=WOU2SRcrcnJVVgjs44bL4G2QPE_QRnuBkxTCUVH8ul4,6707
+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
@@ -149,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.257.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-auto_coder-0.1.257.dist-info/METADATA,sha256=2jcEDkcZ3lBZHMxoKhnwfbfJfhG8JF9npurtpR51wVw,2616
-auto_coder-0.1.257.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-auto_coder-0.1.257.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
-auto_coder-0.1.257.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
-auto_coder-0.1.257.dist-info/RECORD,,
+auto_coder-0.1.259.dist-info/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+auto_coder-0.1.259.dist-info/METADATA,sha256=rL9SsZMRosv39MX73HxlLhTmYqStFnUl92EUQkv6uCo,2616
+auto_coder-0.1.259.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+auto_coder-0.1.259.dist-info/entry_points.txt,sha256=0nzHtHH4pNcM7xq4EBA2toS28Qelrvcbrr59GqD_0Ak,350
+auto_coder-0.1.259.dist-info/top_level.txt,sha256=Jqc0_uJSw2GwoFQAa9iJxYns-2mWla-9ok_Y3Gcznjk,10
+auto_coder-0.1.259.dist-info/RECORD,,

autocoder/auto_coder.py

@@ -282,28 +282,7 @@ def main(input_args: Optional[List[str]] = None):
                 )
                 byzerllm.connect_cluster(address=args.ray_address)
 
-            llm = byzerllm.ByzerLLM(verbose=args.print_request)            
-
-            # code_model,index_filter_model,generate_rerank_model,chat_model
-            # 这四个模型如果用户没有设置，就会使用默认的
-            # 如果用户随便填写 deepseek 官方key,就会导致 Authentic(No User) 的错误
-            # 或者 Insuffient Balance 之类的错误
-
-            code_model = byzerllm.ByzerLLM()
-            code_model.setup_default_model_name(args.model)
-            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)
+            llm = byzerllm.ByzerLLM(verbose=args.print_request)                        
         
         if args.product_mode == "lite":
             default_model = args.model        

autocoder/auto_coder_rag.py

@@ -66,7 +66,7 @@ def initialize_system(args):
     print_status(get_message("checking_model"), "")
     try:
         result = subprocess.run(
-            ["easy-byzerllm", "chat", "deepseek_chat", "你好"],
+            ["easy-byzerllm", "chat", "v3_chat", "你好"],
             capture_output=True,
             text=True,
             timeout=30,
@@ -115,7 +115,7 @@ def initialize_system(args):
             "--infer_params",
             f"saas.base_url=https://api.deepseek.com/v1 saas.api_key={api_key} saas.model=deepseek-chat",
             "--model",
-            "deepseek_chat",
+            "v3_chat",
         ]
 
     try:
@@ -129,7 +129,7 @@ def initialize_system(args):
     print_status(get_message("validating_deploy"), "")
     try:
         validation_result = subprocess.run(
-            ["easy-byzerllm", "chat", "deepseek_chat", "你好"],
+            ["easy-byzerllm", "chat", "v3_chat", "你好"],
             capture_output=True,
             text=True,
             timeout=30,
@@ -139,7 +139,7 @@ def initialize_system(args):
     except (subprocess.TimeoutExpired, subprocess.CalledProcessError):
         print_status(get_message("validation_fail"), "error")
         print_status(get_message("manual_start"), "warning")
-        print_status("easy-byzerllm chat deepseek_chat 你好", "")
+        print_status("easy-byzerllm chat v3_chat 你好", "")
 
     print_status(get_message("init_complete_final"), "success")
 
@@ -168,7 +168,7 @@ def main(input_args: Optional[List[str]] = None):
     )
     build_index_parser.add_argument("--file", default="", help=desc["file"])
     build_index_parser.add_argument(
-        "--model", default="deepseek_chat", help=desc["model"]
+        "--model", default="v3_chat", help=desc["model"]
     )
     build_index_parser.add_argument(
         "--index_model", default="", help=desc["index_model"]
@@ -199,7 +199,7 @@ def main(input_args: Optional[List[str]] = None):
         "--quick", action="store_true", help="Skip system initialization"
     )
     serve_parser.add_argument("--file", default="", help=desc["file"])
-    serve_parser.add_argument("--model", default="deepseek_chat", help=desc["model"])
+    serve_parser.add_argument("--model", default="v3_chat", help=desc["model"])
     serve_parser.add_argument("--index_model", default="", help=desc["index_model"])
     serve_parser.add_argument("--emb_model", default="", help=desc["emb_model"])
     serve_parser.add_argument("--ray_address", default="auto", help=desc["ray_address"])
@@ -366,7 +366,7 @@ def main(input_args: Optional[List[str]] = None):
         "benchmark", help="Benchmark LLM client performance"
     )
     benchmark_parser.add_argument(
-        "--model", default="deepseek_chat", help="Model to benchmark"
+        "--model", default="v3_chat", help="Model to benchmark"
     )
     benchmark_parser.add_argument(
         "--parallel", type=int, default=10, help="Number of parallel requests"

autocoder/auto_coder_rag_client_mcp.py

@@ -147,7 +147,7 @@ def parse_args(input_args: Optional[List[str]] = None) -> AutoCoderArgs:
     parser = argparse.ArgumentParser(description="Auto Coder RAG Client MCP Server")
     parser.add_argument("--rag_url", required=True, help="RAG server URL")
     parser.add_argument("--rag_token", required=True, help="RAG server token")
-    parser.add_argument("--model", default="deepseek_chat", help=desc["model"])
+    parser.add_argument("--model", default="v3_chat", help=desc["model"])
     parser.add_argument("--rag_params_max_tokens", type=int, default=4096, help="Max tokens for RAG response")
     
     args = parser.parse_args(input_args)

autocoder/chat_auto_coder.py

@@ -74,6 +74,12 @@ def parse_arguments():
         help="Enter the auto-coder.chat without initializing the system",
     )
 
+    parser.add_argument(
+        "--skip_provider_selection",
+        action="store_true",
+        help="Skip the provider selection",
+    )
+
     parser.add_argument(
         "--product_mode",
         type=str,
@@ -252,7 +258,12 @@ def configure_project_type():
 
 
 def initialize_system(args):
+    from autocoder.utils.model_provider_selector import ModelProviderSelector
+    from autocoder import models as models_module
     print(f"\n\033[1;34m{get_message('initializing')}\033[0m")
+    
+    first_time = [False]
+    configure_success = [False]
 
     def print_status(message, status):
         if status == "success":
@@ -264,10 +275,9 @@ def initialize_system(args):
         else:
             print(f"  {message}")
 
-    def init_project():
-        first_time = False
+    def init_project():        
         if not os.path.exists(".auto-coder"):
-            first_time = True
+            first_time[0] = True
             print_status(get_message("not_initialized"), "warning")
             init_choice = input(
                 f"  {get_message('init_prompt')}").strip().lower()
@@ -290,140 +300,150 @@ def initialize_system(args):
             print_status(get_message("created_dir").format(
                 base_persist_dir), "success")
 
-        if first_time:
+        if first_time[0]:
             configure_project_type()
+            configure_success[0] = True
 
         print_status(get_message("init_complete"), "success")
 
     init_project()
 
-    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
-        print_status(get_message("checking_ray"), "")
-        ray_status = subprocess.run(
-            ["ray", "status"], capture_output=True, text=True)
-        if ray_status.returncode != 0:
-            print_status(get_message("ray_not_running"), "warning")
+    if not args.skip_provider_selection and first_time[0]:
+        if args.product_mode == "lite":                    
+            ## 如果已经是配置过的项目，就无需再选择
+            if first_time[0]:
+                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)              
+
+        if args.product_mode == "pro":
+            # Check if Ray is running
+            print_status(get_message("checking_ray"), "")
+            ray_status = subprocess.run(
+                ["ray", "status"], capture_output=True, text=True)
+            if ray_status.returncode != 0:
+                print_status(get_message("ray_not_running"), "warning")
+                try:
+                    subprocess.run(["ray", "start", "--head"], check=True)
+                    print_status(get_message("ray_start_success"), "success")
+                except subprocess.CalledProcessError:
+                    print_status(get_message("ray_start_fail"), "error")
+                    return
+            else:
+                print_status(get_message("ray_running"), "success")
+
+            # Check if deepseek_chat model is available
+            print_status(get_message("checking_model"), "")
             try:
-                subprocess.run(["ray", "start", "--head"], check=True)
-                print_status(get_message("ray_start_success"), "success")
+                result = subprocess.run(
+                    ["easy-byzerllm", "chat", "v3_chat", "你好"],
+                    capture_output=True,
+                    text=True,
+                    timeout=30,
+                )
+                if result.returncode == 0:
+                    print_status(get_message("model_available"), "success")
+                    init_project()
+                    print_status(get_message("init_complete_final"), "success")
+                    return
+            except subprocess.TimeoutExpired:
+                print_status(get_message("model_timeout"), "error")
             except subprocess.CalledProcessError:
-                print_status(get_message("ray_start_fail"), "error")
-                return
-        else:
-            print_status(get_message("ray_running"), "success")
+                print_status(get_message("model_error"), "error")
+
+            # If deepseek_chat is not available
+            print_status(get_message("model_not_available"), "warning")
+            api_key = prompt(HTML(f"<b>{get_message('enter_api_key')} </b>"))
+
+            print_status(get_message("deploying_model").format("Deepseek官方"), "")
+            deploy_cmd = [
+                "byzerllm",
+                "deploy",
+                "--pretrained_model_type",
+                "saas/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-chat",
+                "--model",
+                "v3_chat",
+            ]
 
-        # Check if deepseek_chat model is available
-        print_status(get_message("checking_model"), "")
-        try:
-            result = subprocess.run(
-                ["easy-byzerllm", "chat", "deepseek_chat", "你好"],
-                capture_output=True,
-                text=True,
-                timeout=30,
-            )
-            if result.returncode == 0:
-                print_status(get_message("model_available"), "success")
-                init_project()
-                print_status(get_message("init_complete_final"), "success")
+            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
-        except subprocess.TimeoutExpired:
-            print_status(get_message("model_timeout"), "error")
-        except subprocess.CalledProcessError:
-            print_status(get_message("model_error"), "error")
-
-        # If deepseek_chat is not available
-        print_status(get_message("model_not_available"), "warning")
-        api_key = prompt(HTML(f"<b>{get_message('enter_api_key')} </b>"))
-
-        print_status(get_message("deploying_model").format("Deepseek官方"), "")
-        deploy_cmd = [
-            "byzerllm",
-            "deploy",
-            "--pretrained_model_type",
-            "saas/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-chat",
-            "--model",
-            "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")
-        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",
+                "r1_chat",
+            ]
 
-        # Validate the deployment
-        print_status(get_message("validating_deploy"), "")
-        try:
-            validation_result = subprocess.run(
-                ["easy-byzerllm", "chat", "deepseek_chat", "你好"],
-                capture_output=True,
-                text=True,
-                timeout=30,
-                check=True,
-            )
-            print_status(get_message("validation_success"), "success")
-        except (subprocess.TimeoutExpired, subprocess.CalledProcessError):
-            print_status(get_message("validation_fail"), "error")
-            print_status(get_message("manual_start"), "warning")
-            print_status("easy-byzerllm chat 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
 
-        print_status(get_message("init_complete_final"), "success")
+            # Validate the deployment
+            print_status(get_message("validating_deploy"), "")
+            try:
+                validation_result = subprocess.run(
+                    ["easy-byzerllm", "chat", "v3_chat", "你好"],
+                    capture_output=True,
+                    text=True,
+                    timeout=30,
+                    check=True,
+                )
+                print_status(get_message("validation_success"), "success")
+            except (subprocess.TimeoutExpired, subprocess.CalledProcessError):
+                print_status(get_message("validation_fail"), "error")
+                print_status(get_message("manual_start"), "warning")
+                print_status("easy-byzerllm chat v3_chat 你好", "")
+
+            print_status(get_message("init_complete_final"), "success")  
+            configure_success[0] = True
+
+    if first_time[0] and args.product_mode == "pro" and configure_success[0]:
+        configure(f"model:v3_chat", skip_print=True)
+        configure(f"chat_model:r1_chat", skip_print=True)
+        configure(f"generate_rerank_model:r1_chat", skip_print=True)
+        configure(f"code_model:v3_chat", skip_print=True)
+        configure(f"index_filter_model:r1_chat", skip_print=True)  
+
+    if first_time[0] and args.product_mode == "lite" and models_module.check_model_exists("v3_chat"):
+        configure(f"model:v3_chat", skip_print=True)
+        configure(f"chat_model:r1_chat", skip_print=True)
+        configure(f"generate_rerank_model:r1_chat", skip_print=True)
+        configure(f"code_model:v3_chat", skip_print=True)
+        configure(f"index_filter_model:r1_chat", skip_print=True)
 
 
 def convert_yaml_config_to_str(yaml_config):

autocoder/common/auto_coder_lang.py

@@ -6,13 +6,13 @@ MESSAGES = {
         "model_provider_select_title": "Select Model Provider",
         "model_provider_select_text": "Please select your model provider:",
         "model_provider_volcano": "Volcano Engine",
-        "model_provider_guiji": "GuiJi AI",
+        "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_guiji_api_key_text": "Please enter your GuiJi AI API key:",
+        "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",
@@ -115,13 +115,13 @@ MESSAGES = {
         "model_provider_select_title": "选择模型供应商",
         "model_provider_select_text": "请选择您的模型供应商：",
         "model_provider_volcano": "火山方舟",
-        "model_provider_guiji": "硅基流动",
+        "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_guiji_api_key_text": "请输入您的硅基流动API密钥：",
+        "model_provider_siliconflow_api_key_text": "请输入您的硅基流动API密钥：",
         "model_provider_deepseek_api_key_text": "请输入您的DeepSeek API密钥：",
         "model_provider_selected": "供应商配置已成功完成！后续你可以使用 /models 命令，查看，新增和修改所有模型",
         "model_provider_success_title": "成功",

autocoder/common/command_templates.py

@@ -30,7 +30,11 @@ def init_command_template(source_dir:str):
     project_type: py
 
     ## The model you want to drive AutoCoder to run
-    model: deepseek_chat
+    model: v3_chat
+    chat_model: r1_chat
+    generate_rerank_model: r1_chat
+    code_model: v3_chat
+    index_filter_model: r1_chat
 
 
     ## Enable the index building which can help you find the related files by your query
@@ -38,7 +42,7 @@ def init_command_template(source_dir:str):
     skip_build_index: false
     ## The model to build index for the project (Optional)
     ## 用于为项目构建索引的模型（可选）
-    index_model: deepseek_chat
+    index_model: v3_chat
 
     ## the filter level to find the related files
     ## 0: only find the files with the file name
@@ -170,7 +174,7 @@ def base_base(source_dir:str,project_type:str)->str:
     source_dir: {{ source_dir }}
     target_file: {{ target_file }}
 
-    model: deepseek_chat
+    model: v3_chat
     model_max_input_length: 100000
     model_max_input_length: 120000
     enable_multi_round_generate: false

autocoder/index/entry.py

@@ -70,7 +70,7 @@ def build_index_and_filter_files(
             )
     phase_end = time.monotonic()
     stats["timings"]["process_tagged_sources"] = phase_end - phase_start
-    
+        
     if not args.skip_build_index and llm:
         # Phase 2: Build index
         if args.request_id and not args.skip_events:

autocoder/rag/raw_rag.py

@@ -43,7 +43,7 @@ def process_query(context: str, query: str) -> str:
 
 class RawRAG:
     def __init__(
-        self, llm_model="deepseek_chat", emb_model="emb", storage_name="byzerai_store"
+        self, llm_model="v3_chat", emb_model="emb", storage_name="byzerai_store"
     ):
         self.storage = ByzerStorage(
             storage_name, "rag_database", "rag_table", emb_model=emb_model

autocoder/utils/model_provider_selector.py

@@ -1,5 +1,6 @@
 from prompt_toolkit.shortcuts import radiolist_dialog, input_dialog
 from prompt_toolkit.validation import Validator, ValidationError
+from prompt_toolkit.styles import Style
 from rich.console import Console
 from typing import Optional, Dict, Any, List
 from autocoder.common.printer import Printer
@@ -33,7 +34,7 @@ PROVIDER_INFO_LIST = [
         v3_output_price=4.0,
     ), 
     ProviderInfo(
-        name="siliconFlow",
+        name="siliconflow",
         endpoint="https://api.siliconflow.cn/v1",        
         r1_model="Pro/deepseek-ai/DeepSeek-R1",
         v3_model="Pro/deepseek-ai/DeepSeek-V3",
@@ -56,6 +57,21 @@ PROVIDER_INFO_LIST = [
     ),
 ]
 
+dialog_style = Style.from_dict({
+            'dialog':                'bg:#2b2b2b',
+            'dialog frame.label':    'bg:#2b2b2b #ffffff',
+            'dialog.body':          'bg:#2b2b2b #ffffff',
+            'dialog shadow':        'bg:#1f1f1f',
+            'button':               'bg:#2b2b2b #808080',
+            'button.focused':       'bg:#1e1e1e #ffd700 bold',
+            'checkbox':             '#e6e6e6',
+            'checkbox-selected':    '#0078d4',
+            'radio-selected':       'bg:#0078d4 #ffffff',
+            'dialog frame.border':  '#0078d4',
+            'radio':                'bg:#2b2b2b #808080',
+            'radio.focused':        'bg:#1e1e1e #ffd700 bold'
+        })
+
 class VolcanoEndpointValidator(Validator):
     def validate(self, document):
         text = document.text
@@ -123,14 +139,17 @@ class ModelProviderSelector:
         Let user select a model provider and input necessary credentials.
         Returns a dictionary with provider info or None if cancelled.
         """
+        
+
         result = radiolist_dialog(
             title=self.printer.get_message_from_key("model_provider_select_title"),
             text=self.printer.get_message_from_key("model_provider_select_text"),
             values=[
                 ("volcano", self.printer.get_message_from_key("model_provider_volcano")),
-                ("siliconflow", self.printer.get_message_from_key("model_provider_guiji")),
+                ("siliconflow", self.printer.get_message_from_key("model_provider_siliconflow")),
                 ("deepseek", self.printer.get_message_from_key("model_provider_deepseek"))
-            ]
+            ],
+            style=dialog_style
         ).run()
         
         if result is None:
@@ -148,7 +167,8 @@ class ModelProviderSelector:
             r1_endpoint = input_dialog(
                 title=self.printer.get_message_from_key("model_provider_api_key_title"),
                 text=self.printer.get_message_from_key("model_provider_volcano_r1_text"),
-                validator=VolcanoEndpointValidator()
+                validator=VolcanoEndpointValidator(),
+                style=dialog_style
             ).run()
             
             if r1_endpoint is None:
@@ -160,7 +180,8 @@ class ModelProviderSelector:
             v3_endpoint = input_dialog(
                 title=self.printer.get_message_from_key("model_provider_api_key_title"),
                 text=self.printer.get_message_from_key("model_provider_volcano_v3_text"),
-                validator=VolcanoEndpointValidator()
+                validator=VolcanoEndpointValidator(),
+                style=dialog_style
             ).run()
             
             if v3_endpoint is None:
@@ -172,7 +193,8 @@ class ModelProviderSelector:
         api_key = input_dialog(
             title=self.printer.get_message_from_key("model_provider_api_key_title"),
             text=self.printer.get_message_from_key(f"model_provider_{result}_api_key_text"),
-            password=True
+            password=True,
+            style=dialog_style
         ).run()
         
         if api_key is None:

autocoder/version.py

@@ -1 +1 @@
-__version__ = "0.1.257"
+__version__ = "0.1.259"