| """
|
| Terminal display utilities with colors and formatting
|
| """
|
|
|
|
|
|
|
| class Colors:
|
| RED = "\033[91m"
|
| GREEN = "\033[92m"
|
| YELLOW = "\033[93m"
|
| BLUE = "\033[94m"
|
| MAGENTA = "\033[95m"
|
| CYAN = "\033[96m"
|
| BOLD = "\033[1m"
|
| UNDERLINE = "\033[4m"
|
| RESET = "\033[0m"
|
|
|
|
|
| def truncate_to_lines(text: str, max_lines: int = 6) -> str:
|
| """Truncate text to max_lines, adding '...' if truncated"""
|
| lines = text.split("\n")
|
| if len(lines) <= max_lines:
|
| return text
|
| return (
|
| "\n".join(lines[:max_lines])
|
| + f"\n{Colors.CYAN}... ({len(lines) - max_lines} more lines){Colors.RESET}"
|
| )
|
|
|
|
|
| def format_header(text: str, emoji: str = "") -> str:
|
| """Format a header with bold"""
|
| full_text = f"{emoji} {text}" if emoji else text
|
| return f"{Colors.BOLD}{full_text}{Colors.RESET}"
|
|
|
|
|
| def format_plan_display() -> str:
|
| """Format the current plan for display (no colors, full visibility)"""
|
| from agent.tools.plan_tool import get_current_plan
|
|
|
| plan = get_current_plan()
|
| if not plan:
|
| return ""
|
|
|
| lines = ["\n" + "=" * 60]
|
| lines.append("CURRENT PLAN")
|
| lines.append("=" * 60 + "\n")
|
|
|
|
|
| completed = [t for t in plan if t["status"] == "completed"]
|
| in_progress = [t for t in plan if t["status"] == "in_progress"]
|
| pending = [t for t in plan if t["status"] == "pending"]
|
|
|
| if completed:
|
| lines.append("Completed:")
|
| for todo in completed:
|
| lines.append(f" [x] {todo['id']}. {todo['content']}")
|
| lines.append("")
|
|
|
| if in_progress:
|
| lines.append("In Progress:")
|
| for todo in in_progress:
|
| lines.append(f" [~] {todo['id']}. {todo['content']}")
|
| lines.append("")
|
|
|
| if pending:
|
| lines.append("Pending:")
|
| for todo in pending:
|
| lines.append(f" [ ] {todo['id']}. {todo['content']}")
|
| lines.append("")
|
|
|
| lines.append(
|
| f"Total: {len(plan)} todos ({len(completed)} completed, {len(in_progress)} in progress, {len(pending)} pending)"
|
| )
|
| lines.append("=" * 60 + "\n")
|
|
|
| return "\n".join(lines)
|
|
|
|
|
| def format_error(message: str) -> str:
|
| """Format an error message in red"""
|
| return f"{Colors.RED}ERROR: {message}{Colors.RESET}"
|
|
|
|
|
| def format_success(message: str, emoji: str = "") -> str:
|
| """Format a success message in green"""
|
| prefix = f"{emoji} " if emoji else ""
|
| return f"{Colors.GREEN}{prefix}{message}{Colors.RESET}"
|
|
|
|
|
| def format_tool_call(tool_name: str, arguments: str) -> str:
|
| """Format a tool call message"""
|
| return f"{Colors.YELLOW}Calling tool: {Colors.BOLD}{tool_name}{Colors.RESET}{Colors.YELLOW} with arguments: {arguments}{Colors.RESET}"
|
|
|
|
|
| def format_tool_output(output: str, success: bool, truncate: bool = True) -> str:
|
| """Format tool output with color and optional truncation"""
|
| original_length = len(output)
|
| if truncate:
|
| output = truncate_to_lines(output, max_lines=6)
|
|
|
| if success:
|
| return (
|
| f"{Colors.YELLOW}Tool output ({original_length} tkns): {Colors.RESET}\n{output}"
|
| )
|
| else:
|
| return (
|
| f"{Colors.RED}Tool output ({original_length} tokens): {Colors.RESET}\n{output}"
|
| )
|
|
|
|
|
| def format_turn_complete() -> str:
|
| """Format turn complete message in green with hugging face emoji"""
|
| return f"{Colors.GREEN}{Colors.BOLD}\U0001f917 Turn complete{Colors.RESET}\n"
|
|
|
|
|
| def format_separator(char: str = "=", length: int = 60) -> str:
|
| """Format a separator line"""
|
| return char * length
|
|
|
|
|
| def format_plan_tool_output(todos: list) -> str:
|
| """Format the plan tool output (no colors, full visibility)"""
|
| if not todos:
|
| return "Plan is empty."
|
|
|
| lines = ["Plan updated successfully", ""]
|
|
|
|
|
| completed = [t for t in todos if t["status"] == "completed"]
|
| in_progress = [t for t in todos if t["status"] == "in_progress"]
|
| pending = [t for t in todos if t["status"] == "pending"]
|
|
|
| if completed:
|
| lines.append("Completed:")
|
| for todo in completed:
|
| lines.append(f" [x] {todo['id']}. {todo['content']}")
|
| lines.append("")
|
|
|
| if in_progress:
|
| lines.append("In Progress:")
|
| for todo in in_progress:
|
| lines.append(f" [~] {todo['id']}. {todo['content']}")
|
| lines.append("")
|
|
|
| if pending:
|
| lines.append("Pending:")
|
| for todo in pending:
|
| lines.append(f" [ ] {todo['id']}. {todo['content']}")
|
| lines.append("")
|
|
|
| lines.append(
|
| f"Total: {len(todos)} todos ({len(completed)} completed, {len(in_progress)} in progress, {len(pending)} pending)"
|
| )
|
|
|
| return "\n".join(lines)
|
|
|
