auto-coder 0.1.235__tar.gz → 0.1.238__tar.gz

{auto_coder-0.1.235 → auto_coder-0.1.238}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.235
+Version: 0.1.238
 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.151
+Requires-Dist: byzerllm[saas]>=0.1.154
 Requires-Dist: patch
 Requires-Dist: diff_match_patch
 Requires-Dist: GitPython

{auto_coder-0.1.235 → auto_coder-0.1.238}/src/auto_coder.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: auto-coder
-Version: 0.1.235
+Version: 0.1.238
 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.151
+Requires-Dist: byzerllm[saas]>=0.1.154
 Requires-Dist: patch
 Requires-Dist: diff_match_patch
 Requires-Dist: GitPython

{auto_coder-0.1.235 → auto_coder-0.1.238}/src/auto_coder.egg-info/SOURCES.txt

@@ -142,6 +142,7 @@ src/autocoder/utils/request_event_queue.py
 src/autocoder/utils/request_queue.py
 src/autocoder/utils/rest.py
 src/autocoder/utils/tests.py
+src/autocoder/utils/types.py
 src/autocoder/utils/auto_coder_utils/__init__.py
 src/autocoder/utils/auto_coder_utils/chat_stream_out.py
 src/autocoder/utils/chat_auto_coder_utils/__init__.py

{auto_coder-0.1.235 → auto_coder-0.1.238}/src/auto_coder.egg-info/requires.txt

@@ -16,7 +16,7 @@ tabulate
 jupyter_client
 prompt-toolkit
 tokenizers
-byzerllm[saas]>=0.1.151
+byzerllm[saas]>=0.1.154
 patch
 diff_match_patch
 GitPython

{auto_coder-0.1.235 → auto_coder-0.1.238}/src/autocoder/auto_coder.py

@@ -44,7 +44,7 @@ from autocoder.common.memory_manager import save_to_memory_file
 from autocoder import models as models_module
 from autocoder.common.utils_code_auto_generate import stream_chat_with_continue
 from autocoder.utils.auto_coder_utils.chat_stream_out import stream_out
-
+from autocoder.common.printer import Printer
 console = Console()
 
 
@@ -1307,7 +1307,8 @@ def main(input_args: Optional[List[str]] = None):
                     save_to_memory_file(ask_conversation=chat_history["ask_conversation"],
                                         query=args.query,
                                         response=result)
-                    print("Saved to your memory")
+                    printer = Printer()
+                    printer.print_in_terminal("memory_save_success")
                 return {}
                         
             if "rag" in args.action:

{auto_coder-0.1.235 → auto_coder-0.1.238}/src/autocoder/chat_auto_coder.py

@@ -53,6 +53,8 @@ from autocoder import models
 import shlex
 from autocoder.utils.llms import get_single_llm
 import pkg_resources
+from autocoder.common.printer import Printer
+from byzerllm.utils.langutil import run_in_thread,run_in_thread_with_cancel
 
 class SymbolItem(BaseModel):
     symbol_name: str
@@ -537,32 +539,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)
 
@@ -1031,15 +1032,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":
@@ -1253,7 +1249,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,  # 这会在每行之间添加分割线
@@ -1268,29 +1264,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"]:
@@ -1302,26 +1289,25 @@ 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()
 
-
+@run_in_thread()
 def ask(query: str):
     conf = memory.get("conf", {})
     yaml_config = {
@@ -1371,7 +1357,6 @@ def get_llm_friendly_package_docs(
     docs = []
 
     if not os.path.exists(llm_friendly_packages_dir):
-        print("llm_friendly_packages directory not found.")
         return docs
 
     libs = list(memory.get("libs", {}).keys())
@@ -1422,10 +1407,11 @@ def convert_yaml_to_config(yaml_file: str):
                 setattr(args, key, value)
     return args
 
-
+@run_in_thread()
 def mcp(query: str):
     query = query.strip()
     mcp_server = get_mcp_server()
+    printer = Printer()
 
     # Handle remove command
     if query.startswith("/remove"):
@@ -1433,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
@@ -1463,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
@@ -1475,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
@@ -1513,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")
@@ -1528,19 +1517,23 @@ 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"
+            }
+        )
 
 
+@run_in_thread()
 def code_next(query: str):
     conf = memory.get("conf", {})
     yaml_config = {
@@ -1620,6 +1613,7 @@ def code_next(query: str):
     )
 
 
+@run_in_thread()
 def commit(query: str):
     conf = memory.get("conf", {})
     product_mode = conf.get("product_mode", "lite")
@@ -1701,6 +1695,7 @@ def commit(query: str):
                 os.remove(execute_file)
 
 
+@run_in_thread()
 def coding(query: str):
     console = Console()
     is_apply = query.strip().startswith("/apply")
@@ -1850,6 +1845,7 @@ def code_review(query: str) -> str:
     """
 
 
+@run_in_thread()
 def chat(query: str):
     conf = memory.get("conf", {})
 
@@ -1937,6 +1933,7 @@ def chat(query: str):
         os.remove(execute_file)
 
 
+@run_in_thread()
 def summon(query: str):
     conf = memory.get("conf", {})
     current_files = memory["current_files"]["files"]
@@ -1993,6 +1990,7 @@ def summon(query: str):
         os.remove(execute_file)
 
 
+@run_in_thread()
 def design(query: str):
 
     conf = memory.get("conf", {})
@@ -2070,6 +2068,7 @@ def voice_input():
         os.remove(execute_file)
 
 
+@run_in_thread()
 def generate_shell_command(input_text):
     conf = memory.get("conf", {})
     yaml_config = {
@@ -2104,11 +2103,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()
@@ -2128,17 +2127,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", "")
@@ -2146,17 +2149,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
@@ -2166,11 +2168,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 ...       
@@ -2179,19 +2181,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
@@ -2207,23 +2209,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
@@ -2239,7 +2241,7 @@ def exclude_dirs(dir_names: List[str]):
     save_memory()
     completer.refresh_files()
 
-
+@run_in_thread()
 def index_build():
     conf = memory.get("conf", {})
     yaml_config = {
@@ -2265,6 +2267,7 @@ def index_build():
         os.remove(yaml_file)
 
 
+@run_in_thread()
 def index_query(query: str):
     conf = memory.get("conf", {})
     yaml_config = {
@@ -2314,6 +2317,7 @@ def list_files():
         )
 
 
+@run_in_thread()
 def execute_shell_command(command: str):
     console = Console()
     try: