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

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/auto_coder_lang.py

@@ -2,6 +2,7 @@ import locale
 
 MESSAGES = {
     "en": {
+        "memory_save_success": "✅ Saved to your memory",
         "index_file_too_large": "⚠️ File {{ file_path }} is too large ({{ file_size }} > {{ max_length }}), splitting into chunks...",
         "index_update_success": "✅ Successfully updated index for {{ file_path }} (md5: {{ md5 }}) in {{ duration }}s",
         "index_build_error": "❌ Error building index for {{ file_path }}: {{ error }}",
@@ -35,9 +36,32 @@ MESSAGES = {
             "Please use Web version model to get the answer.\n"
             "Or use /conf human_as_model:false to close this mode and get the answer in terminal directly."
             "Paste the answer to the input box below, use '/break' to exit, '/clear' to clear the screen, '/eof' to submit."
-        )
+        ),
+        "code_generation_start": "Auto generate the code...",
+        "code_generation_complete": "Code generation completed in {{ duration }} seconds, input_tokens_count: {{ input_tokens }}, generated_tokens_count: {{ output_tokens }}",
+        "code_merge_start": "Auto merge the code...",
+        "code_execution_warning": "Content(send to model) is {{ content_length }} tokens (you may collect too much files), which is larger than the maximum input length {{ max_length }}",
+        "quick_filter_start": "Starting filter context(quick_filter)...",
+        "normal_filter_start": "Starting filter context(normal_filter)...",
+        "pylint_check_failed": "⚠️ Pylint check failed: {{ error_message }}",
+        "pylint_error": "❌ Error running pylint: {{ error_message }}",
+        "unmerged_blocks_warning": "⚠️ Found {{ num_blocks }} unmerged blocks, the changes will not be applied. Please review them manually then try again.",
+        "pylint_file_check_failed": "⚠️ Pylint check failed for {{ file_path }}. Changes not applied. Error: {{ error_message }}",
+        "merge_success": "✅ Merged changes in {{ num_files }} files {{ num_changes }}/{{ total_blocks }} blocks.",
+        "no_changes_made": "⚠️ No changes were made to any files.",
+        "files_merged": "✅ Merged {{ total }} files into the project.",
+        "merge_failed": "❌ Merge file {{ path }} failed: {{ error }}",
+        "files_merged_total": "✅ Merged {{ total }} files into the project.",
+        "ranking_skip": "Only 1 candidate, skip ranking",
+        "ranking_start": "Start ranking {{ count }} candidates",
+        "ranking_failed_request": "Ranking request failed: {{ error }}",
+        "ranking_all_failed": "All ranking requests failed",
+        "ranking_complete": "Ranking completed in {{ elapsed }}s, total voters: {{ total_tasks }}, best candidate index: {{ best_candidate }}, scores: {{ scores }}, input_tokens: {{ input_tokens }}, output_tokens: {{ output_tokens }}",
+        "ranking_process_failed": "Ranking process failed: {{ error }}",
+        "ranking_failed": "Ranking failed in {{ elapsed }}s, using original order"
     },
     "zh": {
+        "memory_save_success": "✅ 已保存到您的记忆中",
         "index_file_too_large": "⚠️ 文件 {{ file_path }} 过大 ({{ file_size }} > {{ max_length }}), 正在分块处理...",
         "index_update_success": "✅ 成功更新 {{ file_path }} 的索引 (md5: {{ md5 }}), 耗时 {{ duration }} 秒",
         "index_build_error": "❌ 构建 {{ file_path }} 索引时出错: {{ error }}",
@@ -61,12 +85,10 @@ MESSAGES = {
             "系统正在等待您的输入。完成后，在新行输入'EOF'提交。\n"
             "使用'/break'退出此模式。如果复制粘贴有问题，使用'/clear'清理并重新粘贴。"
         ),
-
         "phase1_processing_sources": "阶段 1: 正在处理 REST/RAG/Search 源...",
         "phase2_building_index": "阶段 2: 正在为所有文件构建索引...",
         "phase6_file_selection": "阶段 6: 正在处理文件选择和限制...",
         "phase7_preparing_output": "阶段 7: 正在准备最终输出...",
-
         "chat_human_as_model_instructions": (
             "\n============= Chat 处于 Human as Model 模式 =============\n"
             "问题已复制到剪贴板\n"
@@ -74,7 +96,36 @@ MESSAGES = {
             "或者使用 /conf human_as_model:false 关闭该模式直接在终端获得答案。"
             "将获得答案黏贴到下面的输入框，换行后，使用 '/break' 退出，'/clear' 清屏，'/eof' 提交。"
         ),
-    }
+        "code_generation_start": "正在自动生成代码...",
+        "code_generation_complete": "代码生成完成，耗时 {{ duration }} 秒，输入token数: {{ input_tokens }}, 输出token数: {{ output_tokens }}",
+        "code_merge_start": "正在自动合并代码...",
+        "code_execution_warning": "发送给模型的内容长度为 {{ content_length }} tokens（您可能收集了太多文件），超过了最大输入长度 {{ max_length }}",
+        "quick_filter_start": "开始查找上下文(quick_filter)...",
+        "normal_filter_start": "开始查找上下文(normal_filter)...",
+        "pylint_check_failed": "⚠️ Pylint 检查失败: {{ error_message }}",
+        "pylint_error": "❌ 运行 Pylint 时出错: {{ error_message }}",
+        "unmerged_blocks_warning": "⚠️ 发现 {{ num_blocks }} 个未合并的代码块，更改将不会被应用。请手动检查后重试。",
+        "pylint_file_check_failed": "⚠️ {{ file_path }} 的 Pylint 检查失败。更改未应用。错误: {{ error_message }}",
+        "merge_success": "✅ 成功合并了 {{ num_files }} 个文件中的更改 {{ num_changes }}/{{ total_blocks }} 个代码块。",
+        "no_changes_made": "⚠️ 未对任何文件进行更改。",
+        "unmerged_blocks_title": "Unmerged Blocks",
+        "unmerged_file_path": "File: {file_path}",
+        "unmerged_search_block": "Search Block({similarity}):",
+        "unmerged_replace_block": "Replace Block:",
+        "unmerged_blocks_total": "Total unmerged blocks: {num_blocks}",
+        "git_init_required": "⚠️ auto_merge 仅适用于 git 仓库。\n\n请尝试在源目录中使用 git init：\n\n```shell\ncd {{ source_dir }}\ngit init .\n```\n\n然后再次运行 auto-coder。\n错误: {{ error }}",
+        "upsert_file": "✅ 更新文件: {{ file_path }}",
+        "files_merged": "✅ 成功合并了 {{ total }} 个文件到项目中。",
+        "merge_failed": "❌ 合并文件 {{ path }} 失败: {{ error }}",
+        "files_merged_total": "✅ 合并了 {{ total }} 个文件到项目中。",
+        "ranking_skip": "只有1个候选项，跳过排序",
+        "ranking_start": "开始对 {{ count }} 个候选项进行排序",
+        "ranking_failed_request": "排序请求失败: {{ error }}",
+        "ranking_all_failed": "所有排序请求都失败",
+        "ranking_complete": "排序完成，耗时 {{ elapsed }} 秒，总投票数: {{ total_tasks }}，最佳候选索引: {{ best_candidate }}，得分: {{ scores }}，输入token数: {{ input_tokens }}，输出token数: {{ output_tokens }}",
+        "ranking_process_failed": "排序过程失败: {{ error }}",
+        "ranking_failed": "排序失败，耗时 {{ elapsed }} 秒，使用原始顺序"
+    }    
 }
 
 

autocoder/common/code_auto_merge.py

@@ -1,24 +1,26 @@
 
 import os
 from byzerllm.utils.client import code_utils
-from autocoder.common import AutoCoderArgs,git_utils
-from typing import List,Union,Tuple
+from autocoder.common import AutoCoderArgs, git_utils
+from typing import List, Union, Tuple
 import pydantic
 import byzerllm
-from loguru import logger
 from autocoder.common.types import CodeGenerateResult, MergeCodeWithoutEffect
 from autocoder.common.code_modification_ranker import CodeModificationRanker
 import hashlib
 from autocoder.common import files as FileUtils
+from autocoder.common.printer import Printer
+from autocoder.common.auto_coder_lang import get_message
 
 class PathAndCode(pydantic.BaseModel):
     path: str
     content: str
 
 class CodeAutoMerge:
-    def __init__(self, llm:byzerllm.ByzerLLM,args:AutoCoderArgs):
+    def __init__(self, llm: byzerllm.ByzerLLM, args: AutoCoderArgs):
         self.llm = llm
-        self.args = args  
+        self.args = args
+        self.printer = Printer()
 
 
     def parse_whole_text_v2(self,text: str) -> List[PathAndCode]:
@@ -159,8 +161,9 @@ class CodeAutoMerge:
             try:
                 git_utils.commit_changes(self.args.source_dir, f"auto_coder_pre_{file_name}_{md5}")
             except Exception as e:            
-                logger.error(self.git_require_msg(source_dir=self.args.source_dir,error=str(e)))
-                return            
+                self.printer.print_in_terminal("git_init_required", 
+                    source_dir=self.args.source_dir, error=str(e))
+                return
 
         codes = self.parse_whole_text_v2(content)
         for block in codes:
@@ -168,11 +171,11 @@ class CodeAutoMerge:
             os.makedirs(os.path.dirname(file_path), exist_ok=True)
 
             with open(file_path, "w") as f:
-                logger.info(f"Upsert path: {file_path}")
+                self.printer.print_in_terminal("upsert_file", file_path=file_path)
                 total += 1
                 f.write(block.content)
 
-        logger.info(f"Merged {total} files into the project.")
+        self.printer.print_in_terminal("files_merged", total=total)
         if not force_skip_git:
             commit_result = git_utils.commit_changes(self.args.source_dir, f"auto_coder_{file_name}_{md5}")
             git_utils.print_commit_info(commit_result=commit_result)

autocoder/common/code_auto_merge_diff.py

@@ -4,7 +4,7 @@ from autocoder.common import AutoCoderArgs,git_utils
 from typing import List,Union,Tuple
 import pydantic
 import byzerllm
-from loguru import logger
+from autocoder.common.printer import Printer
 import hashlib
 from pathlib import Path
 from itertools import groupby
@@ -361,7 +361,8 @@ other_hunks_applied = (
 class CodeAutoMergeDiff:
     def __init__(self, llm:byzerllm.ByzerLLM,args:AutoCoderArgs):
         self.llm = llm
-        self.args = args  
+        self.args = args
+        self.printer = Printer()
 
     def get_edits(self,content:str):        
         # might raise ValueError for malformed ORIG/UPD blocks
@@ -514,13 +515,13 @@ class CodeAutoMergeDiff:
             try:
                 git_utils.commit_changes(self.args.source_dir, f"auto_coder_pre_{file_name}_{md5}")
             except Exception as e:            
-                logger.error(self.git_require_msg(source_dir=self.args.source_dir,error=str(e)))
+                self.printer.print_in_terminal("git_init_required", style="red", source_dir=self.args.source_dir, error=str(e))
                 return            
        
         edits = self.get_edits(content)        
         self.apply_edits(edits)
 
-        logger.info(f"Merged {total} files into the project.")
+        self.printer.print_in_terminal("files_merged_total", total=total)
         if not force_skip_git:
             commit_result = git_utils.commit_changes(self.args.source_dir, f"auto_coder_{file_name}_{md5}")
             git_utils.print_commit_info(commit_result=commit_result)

autocoder/common/code_auto_merge_editblock.py

@@ -9,7 +9,7 @@ from autocoder.utils.queue_communicate import (
 )
 import pydantic
 import byzerllm
-from loguru import logger
+
 import hashlib
 import subprocess
 import tempfile
@@ -21,6 +21,7 @@ from typing import Union, List, Tuple
 from autocoder.common.types import CodeGenerateResult, MergeCodeWithoutEffect
 from autocoder.common.code_modification_ranker import CodeModificationRanker
 from autocoder.common import files as FileUtils
+from autocoder.common.printer import Printer
 
 class PathAndCode(pydantic.BaseModel):
     path: str
@@ -39,6 +40,7 @@ class CodeAutoMergeEditBlock:
         self.args = args
         self.fence_0 = fence_0
         self.fence_1 = fence_1
+        self.printer = Printer()
 
     def run_pylint(self, code: str) -> tuple[bool, str]:
         with tempfile.NamedTemporaryFile(
@@ -62,12 +64,12 @@ class CodeAutoMergeEditBlock:
             os.unlink(temp_file_path)
             if result.returncode != 0:
                 error_message = result.stdout.strip() or result.stderr.strip()
-                logger.warning(f"Pylint check failed: {error_message}")
+                self.printer.print_in_terminal("pylint_check_failed", error_message=error_message)
                 return False, error_message
             return True, ""
         except subprocess.CalledProcessError as e:
             error_message = f"Error running pylint: {str(e)}"
-            logger.error(error_message)
+            self.printer.print_in_terminal("pylint_error", error_message=error_message)
             os.unlink(temp_file_path)
             return False, error_message
 
@@ -344,9 +346,8 @@ class CodeAutoMergeEditBlock:
                     ),
                 )
                 return
-
-            s = f"Found {len(unmerged_blocks)} unmerged blocks, the changes will not be applied. Please review them manually then try again."
-            logger.warning(s)
+            
+            self.printer.print_in_terminal("unmerged_blocks_warning", num_blocks=len(unmerged_blocks))
             self._print_unmerged_blocks(unmerged_blocks)
             return
 
@@ -355,9 +356,9 @@ class CodeAutoMergeEditBlock:
             if file_path.endswith(".py"):
                 pylint_passed, error_message = self.run_pylint(new_content)
                 if not pylint_passed:
-                    logger.warning(
-                        f"Pylint check failed for {file_path}. Changes not applied. Error: {error_message}"
-                    )
+                    self.printer.print_in_terminal("pylint_file_check_failed", 
+                                                  file_path=file_path, 
+                                                  error_message=error_message)
 
         if changes_made and not force_skip_git:
             try:
@@ -365,9 +366,9 @@ class CodeAutoMergeEditBlock:
                     self.args.source_dir, f"auto_coder_pre_{file_name}_{md5}"
                 )
             except Exception as e:
-                logger.error(
-                    self.git_require_msg(
-                        source_dir=self.args.source_dir, error=str(e))
+                self.printer.print_str_in_terminal(
+                    self.git_require_msg(source_dir=self.args.source_dir, error=str(e)),
+                    style="red"
                 )
                 return
         # Now, apply the changes
@@ -406,30 +407,31 @@ class CodeAutoMergeEditBlock:
                     )
                     git_utils.print_commit_info(commit_result=commit_result)
                 except Exception as e:
-                    logger.error(
-                        self.git_require_msg(
-                            source_dir=self.args.source_dir, error=str(e)
-                        )
+                    self.printer.print_str_in_terminal(
+                        self.git_require_msg(source_dir=self.args.source_dir, error=str(e)),
+                        style="red"
                     )
-            logger.info(
-                f"Merged changes in {len(file_content_mapping.keys())} files {len(changes_to_make)}/{len(codes)} blocks."
-            )
+            self.printer.print_in_terminal("merge_success", 
+                                         num_files=len(file_content_mapping.keys()),
+                                         num_changes=len(changes_to_make),
+                                         total_blocks=len(codes))
         else:
-            logger.warning("No changes were made to any files.")
+            self.printer.print_in_terminal("no_changes_made")
 
     def _print_unmerged_blocks(self, unmerged_blocks: List[tuple]):
-        console = Console()
-        console.print("\n[bold red]Unmerged Blocks:[/bold red]")
+        self.printer.print_in_terminal("unmerged_blocks_title", style="bold red")
         for file_path, head, update, similarity in unmerged_blocks:
-            console.print(f"\n[bold blue]File:[/bold blue] {file_path}")
-            console.print(
-                f"\n[bold green]Search Block({similarity}):[/bold green]")
+            self.printer.print_str_in_terminal(
+                f"\n{self.printer.get_message_from_key('unmerged_file_path').format(file_path=file_path)}",
+                style="bold blue"
+            )
+            self.printer.print_str_in_terminal(
+                f"\n{self.printer.get_message_from_key('unmerged_search_block').format(similarity=similarity)}",
+                style="bold green"
+            )
             syntax = Syntax(head, "python", theme="monokai", line_numbers=True)
-            console.print(Panel(syntax, expand=False))
-            console.print("\n[bold yellow]Replace Block:[/bold yellow]")
-            syntax = Syntax(update, "python", theme="monokai",
-                            line_numbers=True)
-            console.print(Panel(syntax, expand=False))
-        console.print(
-            f"\n[bold red]Total unmerged blocks: {len(unmerged_blocks)}[/bold red]"
-        )
+            self.printer.console.print(Panel(syntax, expand=False))
+            self.printer.print_in_terminal("unmerged_replace_block", style="bold yellow")
+            syntax = Syntax(update, "python", theme="monokai", line_numbers=True)
+            self.printer.console.print(Panel(syntax, expand=False))
+        self.printer.print_in_terminal("unmerged_blocks_total", num_blocks=len(unmerged_blocks), style="bold red")

autocoder/common/code_auto_merge_strict_diff.py

@@ -5,7 +5,7 @@ from autocoder.common import AutoCoderArgs, git_utils
 from typing import List,Tuple
 import pydantic
 import byzerllm
-from loguru import logger
+from autocoder.common.printer import Printer
 import hashlib
 from pathlib import Path
 from autocoder.common.types import CodeGenerateResult, MergeCodeWithoutEffect
@@ -84,6 +84,7 @@ class CodeAutoMergeStrictDiff:
     def __init__(self, llm: byzerllm.ByzerLLM, args: AutoCoderArgs):
         self.llm = llm
         self.args = args
+        self.printer = Printer()
 
 
     def parse_diff_block(self,text: str) -> List[PathAndCode]:
@@ -192,7 +193,7 @@ class CodeAutoMergeStrictDiff:
                 else:
                     failed_blocks.append((full_path, content))
             except Exception as e:
-                logger.warning(f"Failed to apply patch to {full_path}: {str(e)}")
+                self.printer.print_in_terminal("merge_failed", style="yellow", path=full_path, error=str(e))
                 failed_blocks.append((full_path, content))
                 
         return MergeCodeWithoutEffect(
@@ -212,7 +213,7 @@ class CodeAutoMergeStrictDiff:
             try:
                 git_utils.commit_changes(self.args.source_dir, f"auto_coder_pre_{file_name}_{md5}")
             except Exception as e:            
-                logger.error(self.git_require_msg(source_dir=self.args.source_dir, error=str(e)))
+                self.printer.print_in_terminal("git_init_required", style="red", source_dir=self.args.source_dir, error=str(e))
                 return            
        
         diff_blocks = self.parse_diff_block(content)
@@ -230,7 +231,7 @@ class CodeAutoMergeStrictDiff:
             if not success:
                 raise Exception("Error applying diff to file: " + path)
                             
-        logger.info(f"Merged {total} files into the project.")
+        self.printer.print_in_terminal("files_merged_total", total=total)
         if not force_skip_git:
             commit_result = git_utils.commit_changes(self.args.source_dir, f"auto_coder_{file_name}_{md5}")
             git_utils.print_commit_info(commit_result=commit_result)