mirror of
https://github.com/mistralai/mistral-vibe
synced 2026-04-28 18:37:53 +02:00
d8dbeeb31e
Co-Authored-By: Quentin Torroba <quentin.torroba@mistral.ai> Co-Authored-By: Michel Thomazo <michel.thomazo@mistral.ai> Co-Authored-By: Kracekumar <kracethekingmaker@gmail.com>
318 lines
11 KiB
Python
318 lines
11 KiB
Python
from __future__ import annotations
|
|
|
|
from textual.app import ComposeResult
|
|
from textual.containers import Vertical
|
|
from textual.widgets import Markdown, Static
|
|
|
|
from vibe.cli.textual_ui.widgets.utils import DEFAULT_TOOL_SHORTCUT, TOOL_SHORTCUTS
|
|
|
|
|
|
class ToolApprovalWidget(Vertical):
|
|
def __init__(self, data: dict) -> None:
|
|
super().__init__()
|
|
self.data = data
|
|
self.add_class("tool-approval-widget")
|
|
|
|
def compose(self) -> ComposeResult:
|
|
MAX_APPROVAL_MSG_SIZE = 150
|
|
|
|
for key, value in self.data.items():
|
|
value_str = str(value)
|
|
if len(value_str) > MAX_APPROVAL_MSG_SIZE:
|
|
hidden = len(value_str) - MAX_APPROVAL_MSG_SIZE
|
|
value_str = (
|
|
value_str[:MAX_APPROVAL_MSG_SIZE] + f"… ({hidden} more characters)"
|
|
)
|
|
yield Static(
|
|
f"{key}: {value_str}", markup=False, classes="approval-description"
|
|
)
|
|
|
|
|
|
class ToolResultWidget(Static):
|
|
SHORTCUT = DEFAULT_TOOL_SHORTCUT
|
|
|
|
def __init__(self, data: dict, collapsed: bool = True) -> None:
|
|
super().__init__()
|
|
self.data = data
|
|
self.collapsed = collapsed
|
|
self.add_class("tool-result-widget")
|
|
|
|
def _hint(self) -> str:
|
|
action = "expand" if self.collapsed else "collapse"
|
|
return f"({self.SHORTCUT} to {action})"
|
|
|
|
def compose(self) -> ComposeResult:
|
|
message = self.data.get("message", "")
|
|
|
|
if self.collapsed:
|
|
yield Static(f"{message} {self._hint()}", markup=False)
|
|
else:
|
|
yield Static(message, markup=False)
|
|
|
|
if not self.collapsed and (details := self.data.get("details")):
|
|
for key, value in details.items():
|
|
if value:
|
|
yield Static(
|
|
f"{key}: {value}", markup=False, classes="tool-result-detail"
|
|
)
|
|
|
|
|
|
class BashApprovalWidget(ToolApprovalWidget):
|
|
def compose(self) -> ComposeResult:
|
|
command = self.data.get("command", "")
|
|
description = self.data.get("description", "")
|
|
|
|
if description:
|
|
yield Static(description, markup=False, classes="approval-description")
|
|
yield Static("")
|
|
|
|
yield Markdown(f"```bash\n{command}\n```")
|
|
|
|
|
|
class BashResultWidget(ToolResultWidget):
|
|
def compose(self) -> ComposeResult:
|
|
message = self.data.get("message", "")
|
|
|
|
if self.collapsed:
|
|
yield Static(f"{message} {self._hint()}", markup=False)
|
|
else:
|
|
yield Static(message, markup=False)
|
|
|
|
if not self.collapsed and (details := self.data.get("details")):
|
|
for key, value in details.items():
|
|
if value:
|
|
yield Static(
|
|
f"{key}: {value}", markup=False, classes="tool-result-detail"
|
|
)
|
|
|
|
|
|
class WriteFileApprovalWidget(ToolApprovalWidget):
|
|
def compose(self) -> ComposeResult:
|
|
path = self.data.get("path", "")
|
|
content = self.data.get("content", "")
|
|
file_extension = self.data.get("file_extension", "text")
|
|
|
|
yield Static(f"File: {path}", markup=False, classes="approval-description")
|
|
yield Static("")
|
|
|
|
yield Markdown(f"```{file_extension}\n{content}\n```")
|
|
|
|
|
|
class WriteFileResultWidget(ToolResultWidget):
|
|
def compose(self) -> ComposeResult:
|
|
MAX_LINES = 10
|
|
message = self.data.get("message", "")
|
|
|
|
if self.collapsed:
|
|
yield Static(f"{message} {self._hint()}", markup=False)
|
|
else:
|
|
yield Static(message, markup=False)
|
|
|
|
if not self.collapsed:
|
|
if path := self.data.get("path"):
|
|
yield Static(
|
|
f"Path: {path}", markup=False, classes="tool-result-detail"
|
|
)
|
|
|
|
if bytes_written := self.data.get("bytes_written"):
|
|
yield Static(
|
|
f"Bytes: {bytes_written}",
|
|
markup=False,
|
|
classes="tool-result-detail",
|
|
)
|
|
|
|
if content := self.data.get("content"):
|
|
yield Static("")
|
|
file_extension = self.data.get("file_extension", "text")
|
|
|
|
lines = content.split("\n")
|
|
total_lines = len(lines)
|
|
|
|
if total_lines > MAX_LINES:
|
|
shown_lines = lines[:MAX_LINES]
|
|
remaining = total_lines - MAX_LINES
|
|
truncated_content = "\n".join(
|
|
shown_lines + [f"… ({remaining} more lines)"]
|
|
)
|
|
yield Markdown(f"```{file_extension}\n{truncated_content}\n```")
|
|
else:
|
|
yield Markdown(f"```{file_extension}\n{content}\n```")
|
|
|
|
|
|
class SearchReplaceApprovalWidget(ToolApprovalWidget):
|
|
def compose(self) -> ComposeResult:
|
|
file_path = self.data.get("file_path", "")
|
|
diff_lines = self.data.get("diff_lines", [])
|
|
|
|
yield Static(f"File: {file_path}", markup=False, classes="approval-description")
|
|
yield Static("")
|
|
|
|
if diff_lines:
|
|
for line in diff_lines:
|
|
if line.startswith("---") or line.startswith("+++"):
|
|
yield Static(line, markup=False, classes="diff-header")
|
|
elif line.startswith("-"):
|
|
yield Static(line, markup=False, classes="diff-removed")
|
|
elif line.startswith("+"):
|
|
yield Static(line, markup=False, classes="diff-added")
|
|
elif line.startswith("@@"):
|
|
yield Static(line, markup=False, classes="diff-range")
|
|
else:
|
|
yield Static(line, markup=False, classes="diff-context")
|
|
|
|
|
|
class SearchReplaceResultWidget(ToolResultWidget):
|
|
def compose(self) -> ComposeResult:
|
|
message = self.data.get("message", "")
|
|
|
|
if self.collapsed:
|
|
yield Static(f"{message} {self._hint()}", markup=False)
|
|
else:
|
|
yield Static(message, markup=False)
|
|
|
|
if not self.collapsed and (diff_lines := self.data.get("diff_lines")):
|
|
yield Static("")
|
|
for line in diff_lines:
|
|
if line.startswith("---") or line.startswith("+++"):
|
|
yield Static(line, markup=False, classes="diff-header")
|
|
elif line.startswith("-"):
|
|
yield Static(line, markup=False, classes="diff-removed")
|
|
elif line.startswith("+"):
|
|
yield Static(line, markup=False, classes="diff-added")
|
|
elif line.startswith("@@"):
|
|
yield Static(line, markup=False, classes="diff-range")
|
|
else:
|
|
yield Static(line, markup=False, classes="diff-context")
|
|
|
|
|
|
class TodoApprovalWidget(ToolApprovalWidget):
|
|
def compose(self) -> ComposeResult:
|
|
description = self.data.get("description", "")
|
|
if description:
|
|
yield Static(description, markup=False, classes="approval-description")
|
|
|
|
|
|
class TodoResultWidget(ToolResultWidget):
|
|
SHORTCUT = TOOL_SHORTCUTS["todo"]
|
|
|
|
def compose(self) -> ComposeResult:
|
|
message = self.data.get("message", "")
|
|
|
|
if self.collapsed:
|
|
yield Static(f"{message} {self._hint()}", markup=False)
|
|
else:
|
|
yield Static(f"{message} {self._hint()}", markup=False)
|
|
yield Static("")
|
|
|
|
by_status = self.data.get("todos_by_status", {})
|
|
if not any(by_status.values()):
|
|
yield Static("No todos", markup=False, classes="todo-empty")
|
|
return
|
|
|
|
for status in ["in_progress", "pending", "completed", "cancelled"]:
|
|
todos = by_status.get(status, [])
|
|
for todo in todos:
|
|
content = todo.get("content", "")
|
|
icon = self._get_status_icon(status)
|
|
yield Static(
|
|
f"{icon} {content}", markup=False, classes=f"todo-{status}"
|
|
)
|
|
|
|
def _get_status_icon(self, status: str) -> str:
|
|
icons = {"pending": "☐", "in_progress": "☐", "completed": "☑", "cancelled": "☒"}
|
|
return icons.get(status, "☐")
|
|
|
|
|
|
class ReadFileApprovalWidget(ToolApprovalWidget):
|
|
def compose(self) -> ComposeResult:
|
|
for key, value in self.data.items():
|
|
if value:
|
|
yield Static(
|
|
f"{key}: {value}", markup=False, classes="approval-description"
|
|
)
|
|
|
|
|
|
class ReadFileResultWidget(ToolResultWidget):
|
|
def compose(self) -> ComposeResult:
|
|
MAX_LINES = 10
|
|
message = self.data.get("message", "")
|
|
|
|
if self.collapsed:
|
|
yield Static(f"{message} {self._hint()}", markup=False)
|
|
else:
|
|
yield Static(message, markup=False)
|
|
|
|
if self.collapsed:
|
|
return
|
|
|
|
if path := self.data.get("path"):
|
|
yield Static(f"Path: {path}", markup=False, classes="tool-result-detail")
|
|
|
|
if warnings := self.data.get("warnings"):
|
|
for warning in warnings:
|
|
yield Static(
|
|
f"⚠ {warning}", markup=False, classes="tool-result-warning"
|
|
)
|
|
|
|
if content := self.data.get("content"):
|
|
yield Static("")
|
|
file_extension = self.data.get("file_extension", "text")
|
|
|
|
lines = content.split("\n")
|
|
total_lines = len(lines)
|
|
|
|
if total_lines > MAX_LINES:
|
|
shown_lines = lines[:MAX_LINES]
|
|
remaining = total_lines - MAX_LINES
|
|
truncated_content = "\n".join(
|
|
shown_lines + [f"… ({remaining} more lines)"]
|
|
)
|
|
yield Markdown(f"```{file_extension}\n{truncated_content}\n```")
|
|
else:
|
|
yield Markdown(f"```{file_extension}\n{content}\n```")
|
|
|
|
|
|
class GrepApprovalWidget(ToolApprovalWidget):
|
|
def compose(self) -> ComposeResult:
|
|
for key, value in self.data.items():
|
|
if value:
|
|
yield Static(
|
|
f"{key}: {value!s}", classes="approval-description", markup=False
|
|
)
|
|
|
|
|
|
class GrepResultWidget(ToolResultWidget):
|
|
def compose(self) -> ComposeResult:
|
|
MAX_LINES = 30
|
|
message = self.data.get("message", "")
|
|
|
|
if self.collapsed:
|
|
yield Static(f"{message} {self._hint()}", markup=False)
|
|
else:
|
|
yield Static(message, markup=False)
|
|
|
|
if self.collapsed:
|
|
return
|
|
|
|
if warnings := self.data.get("warnings"):
|
|
for warning in warnings:
|
|
yield Static(
|
|
f"⚠ {warning}", classes="tool-result-warning", markup=False
|
|
)
|
|
|
|
if matches := self.data.get("matches"):
|
|
yield Static("")
|
|
|
|
lines = matches.split("\n")
|
|
total_lines = len(lines)
|
|
|
|
if total_lines > MAX_LINES:
|
|
shown_lines = lines[:MAX_LINES]
|
|
remaining = total_lines - MAX_LINES
|
|
truncated_content = "\n".join(
|
|
shown_lines + [f"… ({remaining} more lines)"]
|
|
)
|
|
yield Markdown(f"```\n{truncated_content}\n```")
|
|
else:
|
|
yield Markdown(f"```\n{matches}\n```")
|