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

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

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

{auto_coder-0.1.258.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=ZqpPzT_HdggyvazgM4cfgBLNgZaYhDcBMd4HUgyfqGc,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
@@ -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
@@ -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.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,,
+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/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/version.py

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