auto-coder 0.1.334__py3-none-any.whl → 0.1.340__py3-none-any.whl

autocoder/common/v2/agent/agentic_tool_display.py

@@ -0,0 +1,184 @@
+import json
+from typing import Dict, Callable, Type
+from autocoder.common.auto_coder_lang import get_system_language, format_str_jinja2
+from autocoder.common.v2.agent.agentic_edit_types import (
+    BaseTool,
+    ReadFileTool, WriteToFileTool, ReplaceInFileTool, ExecuteCommandTool,
+    ListFilesTool, SearchFilesTool, ListCodeDefinitionNamesTool,
+    AskFollowupQuestionTool, UseMcpTool, AttemptCompletionTool
+)
+
+# Define message templates for each tool in English and Chinese
+TOOL_DISPLAY_MESSAGES: Dict[Type[BaseTool], Dict[str, str]] = {
+    ReadFileTool: {
+        "en": "AutoCoder wants to read this file:\n[bold cyan]{{ path }}[/]",
+        "zh": "AutoCoder 想要读取此文件：\n[bold cyan]{{ path }}[/]"
+    },
+    WriteToFileTool: {
+        "en": (
+            "AutoCoder wants to write to this file:\n[bold cyan]{{ path }}[/]\n\n"
+            "[dim]Content Snippet:[/dim]\n{{ content_snippet }}{{ ellipsis }}"
+        ),
+        "zh": (
+            "AutoCoder 想要写入此文件：\n[bold cyan]{{ path }}[/]\n\n"
+            "[dim]内容片段：[/dim]\n{{ content_snippet }}{{ ellipsis }}"
+        )
+    },
+    ReplaceInFileTool: {
+        "en": (
+            "AutoCoder wants to replace content in this file:\n[bold cyan]{{ path }}[/]\n\n"
+            "[dim]Diff Snippet:[/dim]\n{{ diff_snippet }}{{ ellipsis }}"
+        ),
+        "zh": (
+            "AutoCoder 想要替换此文件中的内容：\n[bold cyan]{{ path }}[/]\n\n"
+            "[dim]差异片段：[/dim]\n{{ diff_snippet }}{{ ellipsis }}"
+        )
+    },
+    ExecuteCommandTool: {
+        "en": (
+            "AutoCoder wants to execute this command:\n[bold yellow]{{ command }}[/]\n"
+            "[dim](Requires Approval: {{ requires_approval }})[/]"
+        ),
+        "zh": (
+            "AutoCoder 想要执行此命令：\n[bold yellow]{{ command }}[/]\n"
+            "[dim](需要批准：{{ requires_approval }})[/]"
+        )
+    },
+    ListFilesTool: {
+        "en": (
+            "AutoCoder wants to list files in:\n[bold green]{{ path }}[/] "
+            "{{ recursive_text }}"
+        ),
+        "zh": (
+            "AutoCoder 想要列出此目录中的文件：\n[bold green]{{ path }}[/] "
+            "{{ recursive_text }}"
+        )
+    },
+    SearchFilesTool: {
+        "en": (
+            "AutoCoder wants to search files in:\n[bold green]{{ path }}[/]\n"
+            "[dim]File Pattern:[/dim] [yellow]{{ file_pattern }}[/]\n"
+            "[dim]Regex:[/dim] [yellow]{{ regex }}[/]"
+        ),
+        "zh": (
+            "AutoCoder 想要在此目录中搜索文件：\n[bold green]{{ path }}[/]\n"
+            "[dim]文件模式：[/dim] [yellow]{{ file_pattern }}[/]\n"
+            "[dim]正则表达式：[/dim] [yellow]{{ regex }}[/]"
+        )
+    },
+    ListCodeDefinitionNamesTool: {
+        "en": "AutoCoder wants to list definitions in:\n[bold green]{{ path }}[/]",
+        "zh": "AutoCoder 想要列出此路径中的定义：\n[bold green]{{ path }}[/]"
+    },
+    AskFollowupQuestionTool: {
+        "en": (
+            "AutoCoder is asking a question:\n[bold magenta]{{ question }}[/]\n"
+            "{{ options_text }}"
+        ),
+        "zh": (
+            "AutoCoder 正在提问：\n[bold magenta]{{ question }}[/]\n"
+            "{{ options_text }}"
+        )
+    },
+    UseMcpTool: {
+        "en": (
+            "AutoCoder wants to use an MCP tool:\n"
+            "[dim]Server:[/dim] [blue]{{ server_name }}[/]\n"
+            "[dim]Tool:[/dim] [blue]{{ tool_name }}[/]\n"
+            "[dim]Args:[/dim] {{ arguments_snippet }}{{ ellipsis }}"
+        ),
+        "zh": (
+            "AutoCoder 想要使用 MCP 工具：\n"
+            "[dim]服务器：[/dim] [blue]{{ server_name }}[/]\n"
+            "[dim]工具：[/dim] [blue]{{ tool_name }}[/]\n"
+            "[dim]参数：[/dim] {{ arguments_snippet }}{{ ellipsis }}"
+        )
+    }    
+    # AttemptCompletionTool is handled separately in the display loop
+}
+
+def get_tool_display_message(tool: BaseTool) -> str:
+    """
+    Generates a user-friendly, internationalized string representation for a tool call.
+
+    Args:
+        tool: The tool instance (Pydantic model).
+
+    Returns:
+        A formatted string for display in the terminal.
+    """
+    lang = get_system_language()
+    tool_type = type(tool)
+
+    if tool_type not in TOOL_DISPLAY_MESSAGES:
+        # Fallback for unknown tools
+        return f"Unknown tool type: {tool_type.__name__}\nData: {tool.model_dump_json(indent=2)}"
+
+    templates = TOOL_DISPLAY_MESSAGES[tool_type]
+    template = templates.get(lang, templates.get("en", "Tool display template not found")) # Fallback to English
+
+    # Prepare context specific to each tool type
+    context = {}
+    if isinstance(tool, ReadFileTool):
+        context = {"path": tool.path}
+    elif isinstance(tool, WriteToFileTool):
+        snippet = tool.content[:150]
+        context = {
+            "path": tool.path,
+            "content_snippet": snippet,
+            "ellipsis": '...' if len(tool.content) > 150 else ''
+        }
+    elif isinstance(tool, ReplaceInFileTool):
+        snippet = tool.diff
+        context = {
+            "path": tool.path,
+            "diff_snippet": snippet,
+            "ellipsis": ''
+        }
+    elif isinstance(tool, ExecuteCommandTool):
+        context = {"command": tool.command, "requires_approval": tool.requires_approval}
+    elif isinstance(tool, ListFilesTool):
+        rec_text_en = '(Recursively)' if tool.recursive else '(Top Level)'
+        rec_text_zh = '（递归）' if tool.recursive else '（顶层）'
+        context = {
+            "path": tool.path,
+            "recursive_text": rec_text_zh if lang == 'zh' else rec_text_en
+        }
+    elif isinstance(tool, SearchFilesTool):
+        context = {
+            "path": tool.path,
+            "file_pattern": tool.file_pattern or '*',
+            "regex": tool.regex
+        }
+    elif isinstance(tool, ListCodeDefinitionNamesTool):
+        context = {"path": tool.path}
+    elif isinstance(tool, AskFollowupQuestionTool):
+        options_text_en = ""
+        options_text_zh = ""
+        if tool.options:
+            options_list_en = "\n".join([f"- {opt}" for opt in tool.options])
+            options_list_zh = "\n".join([f"- {opt}" for opt in tool.options]) # Assuming options are simple enough not to need translation
+            options_text_en = f"[dim]Options:[/dim]\n{options_list_en}"
+            options_text_zh = f"[dim]选项：[/dim]\n{options_list_zh}"
+        context = {
+            "question": tool.question,
+            "options_text": options_text_zh if lang == 'zh' else options_text_en
+        }
+    elif isinstance(tool, UseMcpTool):
+        args_str = tool.query
+        snippet = args_str[:100]
+        context = {
+            "server_name": tool.server_name,
+            "tool_name": tool.tool_name,
+            "arguments_snippet": snippet,
+            "ellipsis": '...' if len(args_str) > 100 else ''
+        }
+    else:
+        # Generic context for tools not specifically handled above
+        context = tool.model_dump()
+
+    try:
+        return format_str_jinja2(template, **context)
+    except Exception as e:
+        # Fallback in case of formatting errors
+        return f"Error formatting display for {tool_type.__name__}: {e}\nTemplate: {template}\nContext: {context}"