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

autocoder/chat_auto_coder.py

@@ -53,6 +53,7 @@ from autocoder import models
 import shlex
 from autocoder.utils.llms import get_single_llm
 import pkg_resources
+from autocoder.common.printer import Printer
 
 class SymbolItem(BaseModel):
     symbol_name: str
@@ -370,6 +371,33 @@ def initialize_system(args):
             "deepseek_chat",
         ]
 
+        try:
+            subprocess.run(deploy_cmd, check=True)
+            print_status(get_message("deploy_complete"), "success")
+        except subprocess.CalledProcessError:
+            print_status(get_message("deploy_fail"), "error")
+            return
+        
+
+        deploy_cmd = [
+            "byzerllm",
+            "deploy",
+            "--pretrained_model_type",
+            "saas/reasoning_openai",
+            "--cpus_per_worker",
+            "0.001",
+            "--gpus_per_worker",
+            "0",
+            "--worker_concurrency",
+            "1000",
+            "--num_workers",
+            "1",
+            "--infer_params",
+            f"saas.base_url=https://api.deepseek.com/v1 saas.api_key={api_key} saas.model=deepseek-reasoner",
+            "--model",
+            "deepseek_r1_chat",
+        ]
+
         try:
             subprocess.run(deploy_cmd, check=True)
             print_status(get_message("deploy_complete"), "success")
@@ -510,32 +538,31 @@ def redirect_stdout():
 
 
 def configure(conf: str, skip_print=False):
+    printer = Printer()
     parts = conf.split(None, 1)
     if len(parts) == 2 and parts[0] in ["/drop", "/unset", "/remove"]:
         key = parts[1].strip()
         if key in memory["conf"]:
             del memory["conf"][key]
             save_memory()
-            print(f"\033[92mDeleted configuration: {key}\033[0m")
+            printer.print_in_terminal("config_delete_success", style="green", key=key)
         else:
-            print(f"\033[93mConfiguration not found: {key}\033[0m")
+            printer.print_in_terminal("config_not_found", style="yellow", key=key)
     else:
         parts = conf.split(":", 1)
         if len(parts) != 2:
-            print(
-                "\033[91mError: Invalid configuration format. Use 'key:value' or '/drop key'.\033[0m"
-            )
+            printer.print_in_terminal("config_invalid_format", style="red")
             return
         key, value = parts
         key = key.strip()
         value = value.strip()
         if not value:
-            print("\033[91mError: Value cannot be empty. Use 'key:value'.\033[0m")
+            printer.print_in_terminal("config_value_empty", style="red")
             return
         memory["conf"][key] = value
         save_memory()
         if not skip_print:
-            print(f"\033[92mSet {key} to {value}\033[0m")
+            printer.print_in_terminal("config_set_success", style="green", key=key, value=value)
 
 # word_completer = WordCompleter(commands)
 
@@ -1004,15 +1031,10 @@ def add_files(args: List[str]):
     groups_info = memory["current_files"]["groups_info"]
 
     console = Console()
+    printer = Printer()
 
     if not args:
-        console.print(
-            Panel(
-                "Please provide arguments for the /add_files command.",
-                title="Error",
-                border_style="red",
-            )
-        )
+        printer.print_in_terminal("add_files_no_args", style="red")
         return
 
     if args[0] == "/refresh":
@@ -1226,7 +1248,7 @@ def add_files(args: List[str]):
         if files_to_add:
             memory["current_files"]["files"].extend(files_to_add)
             table = Table(
-                title="Added Files",
+                title=get_message("add_files_added_files"),
                 show_header=True,
                 header_style="bold magenta",
                 show_lines=True,  # 这会在每行之间添加分割线
@@ -1241,29 +1263,20 @@ def add_files(args: List[str]):
                 )
             console.print(Panel(table, border_style="green"))
         else:
-            console.print(
-                Panel(
-                    "All specified files are already in the current session or no matches found.",
-                    title="No Files Added",
-                    border_style="yellow",
-                )
-            )
+            printer.print_in_terminal("add_files_matched", style="yellow")
 
     completer.update_current_files(memory["current_files"]["files"])
     save_memory()
 
 
 def remove_files(file_names: List[str]):
-    console = Console()
     project_root = os.getcwd()
+    printer = Printer()
 
     if "/all" in file_names:
         memory["current_files"]["files"] = []
         memory["current_files"]["current_groups"] = []
-        console.print(
-            Panel("Removed all files.",
-                  title="Files Removed", border_style="green")
-        )
+        printer.print_in_terminal("remove_files_all", style="green")
     else:
         removed_files = []
         for file in memory["current_files"]["files"]:
@@ -1275,21 +1288,20 @@ def remove_files(file_names: List[str]):
             memory["current_files"]["files"].remove(file)
 
         if removed_files:
-            table = Table(
-                title="Removed Files", show_header=True, header_style="bold magenta"
+            table = Table(                
+                show_header=True, 
+                header_style="bold magenta"
             )
             table.add_column("File", style="green")
             for f in removed_files:
                 table.add_row(os.path.relpath(f, project_root))
-            console.print(Panel(table, border_style="green"))
-        else:
+
+            console = Console()
             console.print(
-                Panel(
-                    "No files were removed.",
-                    title="No Files Removed",
-                    border_style="yellow",
-                )
-            )
+                Panel(table, border_style="green",
+                      title=printer.get_message_from_key("files_removed")))            
+        else:
+            printer.print_in_terminal("remove_files_none", style="yellow")
 
     completer.update_current_files(memory["current_files"]["files"])
     save_memory()
@@ -1399,6 +1411,7 @@ def convert_yaml_to_config(yaml_file: str):
 def mcp(query: str):
     query = query.strip()
     mcp_server = get_mcp_server()
+    printer = Printer()
 
     # Handle remove command
     if query.startswith("/remove"):
@@ -1406,29 +1419,29 @@ def mcp(query: str):
         response = mcp_server.send_request(
             McpRemoveRequest(server_name=server_name))
         if response.error:
-            print(f"Error removing MCP server: {response.error}")
+            printer.print_in_terminal("mcp_remove_error", style="red", error=response.error)
         else:
-            print(f"Successfully removed MCP server: {response.result}")
+            printer.print_in_terminal("mcp_remove_success", style="green", result=response.result)
         return
 
     # Handle list command
     if query.startswith("/list_running"):
         response = mcp_server.send_request(McpListRunningRequest())
         if response.error:
-            print(f"Error listing running MCP servers: {response.error}")
+            printer.print_in_terminal("mcp_list_running_error", style="red", error=response.error)
         else:
-            print("Running MCP servers:")
-            print(response.result)
+            printer.print_in_terminal("mcp_list_running_title")
+            printer.print_str_in_terminal(response.result)
         return
 
     # Handle list command
     if query.startswith("/list"):
         response = mcp_server.send_request(McpListRequest())
         if response.error:
-            print(f"Error listing builtin MCP servers: {response.error}")
+            printer.print_in_terminal("mcp_list_builtin_error", style="red", error=response.error)
         else:
-            print("Available builtin MCP servers:")
-            print(response.result)
+            printer.print_in_terminal("mcp_list_builtin_title")
+            printer.print_str_in_terminal(response.result)
         return
 
     # Handle refresh command
@@ -1436,9 +1449,9 @@ def mcp(query: str):
         server_name = query.replace("/refresh", "", 1).strip()    
         response = mcp_server.send_request(McpRefreshRequest(name=server_name or None))
         if response.error:
-            print(f"Error refreshing MCP servers: {response.error}")
+            printer.print_in_terminal("mcp_refresh_error", style="red", error=response.error)
         else:
-            print("Successfully refreshed MCP servers")
+            printer.print_in_terminal("mcp_refresh_success", style="green")
         return
 
     # Handle add command
@@ -1448,9 +1461,9 @@ def mcp(query: str):
         response = mcp_server.send_request(request)
 
         if response.error:
-            print(f"Error installing MCP server: {response.error}")
+            printer.print_in_terminal("mcp_install_error", style="red", error=response.error)
         else:
-            print(f"Successfully installed MCP server: {response.result}")
+            printer.print_in_terminal("mcp_install_success", style="green", result=response.result)
         return
 
     # Handle default query
@@ -1486,13 +1499,16 @@ def mcp(query: str):
             model=args.inference_model or args.model
         )
     )
-    console = Console()
+
     if response.error:
-        console.print(Panel(
+        printer.print_panel(
             f"Error from MCP server: {response.error}",
-            title="Error",
-            border_style="red"
-        ))
+            text_options={"justify": "left"},
+            panel_options={
+                "title": printer.get_message_from_key("mcp_error_title"),
+                "border_style": "red"
+            }
+        )
     else:
         # Save conversation
         mcp_dir = os.path.join(".auto-coder", "mcp", "conversations")
@@ -1501,17 +1517,20 @@ def mcp(query: str):
         file_path = os.path.join(mcp_dir, f"{timestamp}.md")
         
         # Format response as markdown
-        markdown_content = f"# MCP Response\n\n{response.result}"
+        markdown_content = f"# {printer.get_message_from_key('mcp_response_title')}\n\n{response.result}"
         
         # Save to file
         with open(file_path, "w", encoding="utf-8") as f:
             f.write(markdown_content)
             
         # Print with markdown formatting
-        console.print(Panel(
-            Markdown(markdown_content),            
-            border_style="green"
-        ))
+        printer.print_panel(
+            Markdown(markdown_content),
+            text_options={"justify": "left"},
+            panel_options={
+                "border_style": "green"
+            }
+        )
 
 
 def code_next(query: str):
@@ -1715,7 +1734,8 @@ def coding(query: str):
             == "true",
         }
 
-        yaml_config["context"] = ""
+        yaml_config["context"] = ""        
+        yaml_config["in_code_apply"] = is_apply
 
         for key, value in conf.items():
             converted_value = convert_config_value(key, value)
@@ -2076,11 +2096,11 @@ def manage_models(params, query: str):
       /models /add_model name=xxx base_url=xxx ... - Add model with custom params
       /models /remove <name> - Remove model by name
     """
-    print("manage_models", params, query)
+    printer = Printer()
     console = Console()
     
     if params.product_mode != "lite":
-        console.print(f"[red]{get_message('models_lite_only')}[/red]")
+        printer.print_in_terminal("models_lite_only", style="red")
         return
         
     models_data = models.load_models()
@@ -2100,17 +2120,21 @@ def manage_models(params, query: str):
     if "/remove" in query:
         subcmd = "/remove"
         query = query.replace("/remove", "", 1).strip()
+
     if not subcmd:
-        console.print(get_message("models_usage"))
+        printer.print_in_terminal("models_usage")
         return    
 
     if subcmd == "/list":                    
         if models_data:
-            table = Table(title=get_message("models_title"))
-            table.add_column("Name", style="cyan")
-            table.add_column("Model Type", style="green")
-            table.add_column("Model Name", style="magenta")             
-            table.add_column("Description", style="white")
+            table = Table(
+                title=printer.get_message_from_key("models_title"),
+                expand=True,
+                show_lines=True
+            )
+            table.add_column("Name", style="cyan", width=40, no_wrap=False)
+            table.add_column("Model Name", style="magenta", width=30, overflow="fold")
+            table.add_column("Description", style="white", width=50, overflow="fold")
             for m in models_data:
                 # Check if api_key_path exists and file exists
                 api_key_path = m.get("api_key_path", "")
@@ -2118,17 +2142,16 @@ def manage_models(params, query: str):
                 if api_key_path:
                     api_key_file = os.path.expanduser(f"~/.auto-coder/keys/{api_key_path}")
                     if os.path.exists(api_key_file):
-                        name = f"{name}*"
-                
+                        name = f"{name}*"                        
+            
                 table.add_row(
-                    name,
-                    m.get("model_type", ""),
+                    name,                    
                     m.get("model_name", ""),                    
                     m.get("description", "")
                 )
             console.print(table)
         else:
-            console.print(f"[yellow]{get_message('models_no_models')}[/yellow]")
+            printer.print_in_terminal("models_no_models", style="yellow")
 
     elif subcmd == "/add":
         # Support both simplified and legacy formats
@@ -2138,11 +2161,11 @@ def manage_models(params, query: str):
             name, api_key = args[0], args[1]            
             result = models.update_model_with_api_key(name, api_key)
             if result:
-                console.print(f"[green]{get_message('models_added').format(name=name)}[/green]")
+                printer.print_in_terminal("models_added", style="green", name=name)
             else:
-                console.print(f"[red]{get_message('models_add_failed').format(name=name)}[/red]")
+                printer.print_in_terminal("models_add_failed", style="red", name=name)
         else:
-            console.print(f"[red]{get_message('models_add_usage')}[/red]")
+            printer.print_in_terminal("models_add_usage", style="red")
 
     elif subcmd == "/add_model":
         # Parse key=value pairs: /models /add_model name=abc base_url=http://xx ...       
@@ -2151,19 +2174,19 @@ def manage_models(params, query: str):
         data_dict = {}
         for pair in kv_pairs:
             if '=' not in pair:
-                console.print(f"[red]Invalid parameter: {pair}, should be key=value[/red]")
+                printer.print_in_terminal("models_add_model_params", style="red")
                 continue
             k, v = pair.split('=', 1)
             data_dict[k.strip()] = v.strip()
 
         # Name is required
         if "name" not in data_dict:
-            console.print(f"[red]{get_message('models_add_model_name_required')}[/red]")
+            printer.print_in_terminal("models_add_model_name_required", style="red")
             return
 
         # Check duplication
         if any(m["name"] == data_dict["name"] for m in models_data):
-            console.print(f"[yellow]{get_message('models_add_model_exists').format(name=data_dict['name'])}[/yellow]")
+            printer.print_in_terminal("models_add_model_exists", style="yellow", name=data_dict["name"])
             return
 
         # Create model with defaults
@@ -2179,23 +2202,23 @@ def manage_models(params, query: str):
 
         models_data.append(final_model)
         models.save_models(models_data)
-        console.print(f"[green]{get_message('models_add_model_success').format(name=data_dict['name'])}[/green]")
+        printer.print_in_terminal("models_add_model_success", style="green", name=data_dict["name"])
 
     elif subcmd == "/remove":
         args = query.strip().split(" ")
         if len(args) < 1:
-            console.print(f"[red]{get_message('models_add_usage')}[/red]")
+            printer.print_in_terminal("models_add_usage", style="red")
             return
         name = args[0]
         filtered_models = [m for m in models_data if m["name"] != name]
         if len(filtered_models) == len(models_data):
-            console.print(f"[yellow]{get_message('models_add_model_remove').format(name=name)}[/yellow]")
+            printer.print_in_terminal("models_add_model_remove", style="yellow", name=name)
             return
         models.save_models(filtered_models)
-        console.print(f"[green]{get_message('models_add_model_removed').format(name=name)}[/green]")
+        printer.print_in_terminal("models_add_model_removed", style="green", name=name)
 
     else:
-        console.print(f"[yellow]{get_message('models_unknown_subcmd').format(subcmd=subcmd)}[/yellow]")
+        printer.print_in_terminal("models_unknown_subcmd", style="yellow", subcmd=subcmd)
 
 def exclude_dirs(dir_names: List[str]):
     new_dirs = dir_names

autocoder/chat_auto_coder_lang.py

@@ -2,6 +2,19 @@ import locale
 
 MESSAGES = {
     "en": {
+        "mcp_remove_error": "Error removing MCP server: {error}",
+        "mcp_remove_success": "Successfully removed MCP server: {result}",
+        "mcp_list_running_error": "Error listing running MCP servers: {error}",
+        "mcp_list_running_title": "Running MCP servers:",
+        "mcp_list_builtin_error": "Error listing builtin MCP servers: {error}",
+        "mcp_list_builtin_title": "Available builtin MCP servers:",
+        "mcp_refresh_error": "Error refreshing MCP servers: {error}",
+        "mcp_refresh_success": "Successfully refreshed MCP servers",
+        "mcp_install_error": "Error installing MCP server: {error}",
+        "mcp_install_success": "Successfully installed MCP server: {result}",
+        "mcp_query_empty": "Please enter your query.",
+        "mcp_error_title": "Error",
+        "mcp_response_title": "MCP Response",
         "initializing": "🚀 Initializing system...",
         "not_initialized": "The current directory is not initialized as an auto-coder project.",
         "init_prompt": "Do you want to initialize the project now? (y/n): ",
@@ -68,21 +81,47 @@ MESSAGES = {
         "commit_desc": "Auto generate yaml file and commit changes based on user's manual changes",
         "models_desc": "Manage model configurations, only available in lite mode",
         "models_usage": "Usage: /models /list|/add|/add_model|/remove ...",
-        "models_added": "Added/Updated model '{name}' successfully.",
-        "models_add_failed": "Failed to add model '{name}'. Model not found in defaults.",
+        "models_added": "Added/Updated model '{{name}}' successfully.",
+        "models_add_failed": "Failed to add model '{{name}}'. Model not found in defaults.",
         "models_add_usage": "Usage: /models /add <name> <api_key> or\n/models /add <name> <model_type> <model_name> <base_url> <api_key_path> [description]",
         "models_add_model_params": "Please provide parameters in key=value format",
         "models_add_model_name_required": "'name' parameter is required",
-        "models_add_model_exists": "Model '{name}' already exists.",
-        "models_add_model_success": "Successfully added custom model: {name}",
-        "models_add_model_remove": "Model '{name}' not found.",
-        "models_add_model_removed": "Removed model: {name}",
-        "models_unknown_subcmd": "Unknown subcommand: {subcmd}",
+        "models_add_model_exists": "Model '{{name}}' already exists.",
+        "models_add_model_success": "Successfully added custom model: {{name}}",
+        "models_add_model_remove": "Model '{{name}}' not found.",
+        "models_add_model_removed": "Removed model: {{name}}",
+        "models_unknown_subcmd": "Unknown subcommand: {{subcmd}}",
         "models_title": "All Models (内置 + models.json)",
         "models_no_models": "No models found.",
-        "models_lite_only": "The /models command is only available in lite mode"
+        "models_lite_only": "The /models command is only available in lite mode",
+        "models_api_key_exists": "API key file exists: {{path}}",
+        "config_invalid_format": "Error: Invalid configuration format. Use 'key:value' or '/drop key'.",
+        "config_value_empty": "Error: Value cannot be empty. Use 'key:value'.",
+        "config_set_success": "Set {{key}} to {{value}}",
+        "config_delete_success": "Deleted configuration: {{key}}",
+        "config_not_found": "Configuration not found: {{key}}",
+        "add_files_matched": "All specified files are already in the current session or no matches found.",
+        "add_files_added_files": "Added Files",
+        "add_files_no_args": "Please provide arguments for the /add_files command.",
+        "remove_files_all": "Removed all files.",
+        "remove_files_removed": "Removed Files",
+        "remove_files_none": "No files were removed.",
+        "files_removed": "Files Removed"
     },
     "zh": {
+        "mcp_remove_error": "移除 MCP 服务器时出错：{error}",
+        "mcp_remove_success": "成功移除 MCP 服务器：{result}",
+        "mcp_list_running_error": "列出运行中的 MCP 服务器时出错：{error}",
+        "mcp_list_running_title": "正在运行的 MCP 服务器：",
+        "mcp_list_builtin_error": "列出内置 MCP 服务器时出错：{error}",
+        "mcp_list_builtin_title": "可用的内置 MCP 服务器：",
+        "mcp_refresh_error": "刷新 MCP 服务器时出错：{error}",
+        "mcp_refresh_success": "成功刷新 MCP 服务器",
+        "mcp_install_error": "安装 MCP 服务器时出错：{error}",
+        "mcp_install_success": "成功安装 MCP 服务器：{result}",
+        "mcp_query_empty": "请输入您的查询。",
+        "mcp_error_title": "错误",
+        "mcp_response_title": "MCP 响应",
         "initializing": "🚀 正在初始化系统...",
         "not_initialized": "当前目录未初始化为auto-coder项目。",
         "init_prompt": "是否现在初始化项目？(y/n): ",
@@ -147,21 +186,34 @@ MESSAGES = {
         "exit_desc": "退出程序",
         "design_desc": "根据需求设计SVG图片",
         "commit_desc": "根据用户人工修改的代码自动生成yaml文件并提交更改",
-        "models_desc": "管理模型配置,仅在lite模式下可用",
+        "models_desc": "管理模型配置，仅在lite模式下可用",
         "models_usage": "用法: /models /list|/add|/add_model|/remove ...",
-        "models_added": "成功添加/更新模型 '{name}'。",
-        "models_add_failed": "添加模型 '{name}' 失败。在默认模型中未找到该模型。",
+        "models_added": "成功添加/更新模型 '{{name}}'。",
+        "models_add_failed": "添加模型 '{{name}}' 失败。在默认模型中未找到该模型。",
         "models_add_usage": "用法: /models /add <name> <api_key> 或\n/models /add <name> <model_type> <model_name> <base_url> <api_key_path> [description]",
         "models_add_model_params": "请提供 key=value 格式的参数",
         "models_add_model_name_required": "缺少必需的 'name' 参数",
-        "models_add_model_exists": "模型 '{name}' 已存在。",
-        "models_add_model_success": "成功添加自定义模型: {name}",
-        "models_add_model_remove": "找不到模型 '{name}'。",
-        "models_add_model_removed": "已移除模型: {name}",
-        "models_unknown_subcmd": "未知的子命令: {subcmd}",
+        "models_add_model_exists": "模型 '{{name}}' 已存在。",
+        "models_add_model_success": "成功添加自定义模型: {{name}}",
+        "models_add_model_remove": "找不到模型 '{{name}}'。",
+        "models_add_model_removed": "已移除模型: {{name}}",
+        "models_unknown_subcmd": "未知的子命令: {{subcmd}}",
         "models_title": "所有模型 (内置 + models.json)",
         "models_no_models": "未找到任何模型。",
-        "models_lite_only": "/models 命令仅在 lite 模式下可用"
+        "models_lite_only": "/models 命令仅在 lite 模式下可用",
+        "models_api_key_exists": "API密钥文件存在: {{path}}",
+        "config_invalid_format": "错误：配置格式无效。请使用 'key:value' 或 '/drop key'。",
+        "config_value_empty": "错误：值不能为空。请使用 'key:value'。",
+        "config_set_success": "已设置 {{key}} 为 {{value}}",
+        "config_delete_success": "已删除配置：{{key}}",
+        "config_not_found": "未找到配置：{{key}}",
+        "add_files_matched": "所有指定的文件都已在当前会话中或未找到匹配项。",
+        "add_files_added_files": "已添加的文件",
+        "add_files_no_args": "请为 /add_files 命令提供参数。",
+        "remove_files_all": "已移除所有文件。",
+        "remove_files_removed": "已移除的文件",
+        "remove_files_none": "没有文件被移除。",
+        "files_removed": "移除的文件"
     }
 }
 

autocoder/common/__init__.py

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