agent-chat-cli-python/src/agent_chat_cli/components/tool_permission_prompt.py at aa7010cf67eae0c8d311452f101f80ef6c02a222 · damassi/agent-chat-cli-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
from typing import Any, TYPE_CHECKING

from textual.widget import Widget
from textual.app import ComposeResult
from textual.widgets import Label, TextArea
from textual.reactive import reactive
from textual.binding import Binding

from agent_chat_cli.components.caret import Caret
from agent_chat_cli.components.flex import Flex
from agent_chat_cli.components.spacer import Spacer
from agent_chat_cli.utils import get_tool_info
from agent_chat_cli.utils.enums import Key
from agent_chat_cli.utils.logger import log_json

if TYPE_CHECKING:
    from agent_chat_cli.core.actions import Actions


class ToolPermissionPrompt(Widget):
    is_visible = reactive(False)
    tool_name = reactive("")
    tool_input: dict[str, Any] = reactive({}, init=False)  # type: ignore[assignment]

    BINDINGS = [
        Binding(Key.ENTER.value, "submit", "Submit", priority=True),
    ]

    def __init__(self, actions: "Actions") -> None:
        super().__init__()
        self.actions = actions

    def compose(self) -> ComposeResult:
        yield Label("", id="tool-display")
        yield Label("  [dim]Allow? (Enter=yes, ESC=no, or ask another question):[/dim]")

        yield Spacer()

        with Flex():
            yield Caret()
            yield TextArea(
                "",
                show_line_numbers=False,
                soft_wrap=True,
                placeholder="Yes",
                id="permission-input",
            )

    def watch_is_visible(self, is_visible: bool) -> None:
        self.display = is_visible

        if is_visible:
            input_widget = self.query_one("#permission-input", TextArea)
            input_widget.clear()
            input_widget.focus()

    def watch_tool_name(self, tool_name: str) -> None:
        if not tool_name:
            return

        tool_info = get_tool_info(tool_name)
        tool_display_label = self.query_one("#tool-display", Label)

        if tool_info["server_name"]:
            tool_display = (
                rf"[bold]Confirm Tool:[/] [cyan]\[{tool_info['server_name']}][/] "
                f"{tool_info['tool_name']}"
            )
        else:
            tool_display = f"[bold]Confirm Tool:[/] {tool_name}"

        tool_display_label.update(tool_display)

    async def action_submit(self) -> None:
        input_widget = self.query_one("#permission-input", TextArea)
        raw_value = input_widget.text
        response = raw_value.strip() or "yes"

        log_json(
            {
                "event": "permission_input_submitted",
                "raw_value": raw_value,
                "stripped_value": raw_value.strip(),
                "final_response": response,
            }
        )

        await self.actions.respond_to_tool_permission(response)

    def on_descendant_blur(self) -> None:
        if self.is_visible:
            input_widget = self.query_one("#permission-input", TextArea)
            input_widget.focus()

    async def on_key(self, event) -> None:
        if event.key == Key.ESCAPE.value:
            log_json({"event": "permission_escape_pressed"})

            event.stop()
            event.prevent_default()

            input_widget = self.query_one("#permission-input", TextArea)
            input_widget.clear()
            input_widget.insert("no")

            await self.action_submit()