alloy-runtime-cli 0.1.0__py3-none-any.whl

cli/tui/screens/tool_configs.py

@@ -0,0 +1,335 @@
+"""Tool Configs screen for browsing and managing tool configurations.
+
+This screen provides:
+- Search and browse tool configs
+- View tool config details in preview panel (including JSON config)
+- Create new tool configs
+- Edit existing tool configs
+- Delete tool configs (with in-use protection)
+"""
+
+from typing import TYPE_CHECKING
+
+from rich.markup import escape
+from textual import on, work
+from textual.app import ComposeResult
+from textual.binding import Binding
+from textual.containers import Horizontal, Vertical, VerticalScroll
+from textual.widgets import Button, DataTable, Input, Static
+
+from alloy_runtime_types.dtos.tool_configs import ToolConfigSummary
+
+from cli.infrastructure.scope_utils import format_scope_label
+from cli.infrastructure.tui.clipboard import copy_to_clipboard
+from cli.infrastructure.tui.formatters import format_datetime, truncate_text
+from cli.tui.screens.base import BrowserWidget
+from cli.tui.screens.nav_screen import NavScreen
+from cli.tui.widgets.confirm_modal import ConfirmModal
+from cli.tui.widgets.tool_config_create_modal import (
+    ToolConfigCreateModal,
+    ToolConfigCreateResult,
+)
+from cli.tui.widgets.tool_config_update_modal import ToolConfigUpdateModal
+
+if TYPE_CHECKING:
+    from cli.tui.app import AlloyRuntimeApp
+
+
+class ToolConfigsWidget(BrowserWidget[ToolConfigSummary]):
+    """Widget for browsing and managing tool configurations."""
+
+    app: "AlloyRuntimeApp"
+
+    TITLE = "Tool Configs"
+    TABLE_COLUMNS = ["Name", "Tool", "Scope", "Updated"]
+
+    def compose(self) -> ComposeResult:
+        """Compose the browser layout with a New button."""
+        with Vertical(id="browser-root"):
+            with Vertical(id="search-container"):
+                with Horizontal(id="search-row"):
+                    yield Input(
+                        placeholder=f"Search {self.TITLE.lower()}...",
+                        id="search-input",
+                    )
+                    yield Button("+ New", id="new-config-btn", variant="primary")
+                yield Static("Loading...", id="status-bar")
+
+            with Horizontal(id="main-container"):
+                with Vertical(id="list-container"):
+                    yield DataTable(id="data-table", cursor_type="row")
+
+                with Vertical(id="detail-container"):
+                    yield Static("Preview", id="preview-title")
+                    with VerticalScroll(id="preview-scroll"):
+                        yield Static(
+                            "Select an item to view details",
+                            id="preview-content",
+                        )
+
+    @on(Button.Pressed, "#new-config-btn")
+    def _on_new_config_btn_pressed(self, event: Button.Pressed) -> None:
+        """Handle New button press."""
+        self.open_new_modal()
+
+    async def _fetch_items(self, query: str) -> tuple[list[ToolConfigSummary], int]:
+        """Fetch tool configs from the API."""
+        client = await self._ensure_client()
+        response = await client.list_tool_configs(
+            search=query if query else None,
+            limit=50,
+        )
+        return list(response.tool_configs), response.total
+
+    def _format_row(self, item: ToolConfigSummary) -> tuple[str, ...]:
+        """Format a tool config as a table row."""
+        return (
+            truncate_text(item.name, max_length=30),
+            item.tool_id,
+            format_scope_label(
+                str(item.organization_id) if item.organization_id else None,
+                str(item.user_id) if item.user_id else None,
+            ),
+            format_datetime(item.updated_at),
+        )
+
+    def _update_preview(self, item: ToolConfigSummary) -> None:
+        """Update the preview pane with tool config details.
+
+        Note: ToolConfigSummary doesn't include the full config JSON.
+        We show the basic info here. For full config, user can edit.
+        """
+        preview = self.query_one("#preview-content", Static)
+
+        scope = format_scope_label(
+            str(item.organization_id) if item.organization_id else None,
+            str(item.user_id) if item.user_id else None,
+        )
+
+        created = format_datetime(item.created_at)
+        updated = format_datetime(item.updated_at)
+
+        preview_text = f"""[bold]Tool Config ID:[/] {escape(str(item.id))}
+[bold]Name:[/] {escape(item.name)}
+[bold]Slug:[/] {escape(item.slug)}
+[bold]Tool:[/] {escape(item.tool_id)}
+[bold]Scope:[/] {scope}
+
+[bold]Description:[/]
+{escape(item.description or "(no description)")}
+
+[dim]Note: Edit to view full configuration JSON[/]
+
+[bold]Created:[/] {created}
+[bold]Updated:[/] {updated}"""
+
+        preview.update(preview_text)
+
+    # =========================================================================
+    # Public methods for parent Screen to call
+    # =========================================================================
+
+    def copy_config_name(self) -> None:
+        """Copy the tool config name to clipboard."""
+        config = self.get_selected_item()
+        if not config:
+            self.app.notify("No tool config selected", severity="warning")
+            return
+
+        copy_to_clipboard(self.app, config.name, "Tool config name")
+
+    def open_new_modal(self) -> None:
+        """Open the tool config creation modal."""
+        self._open_create_modal()
+
+    def open_edit_modal(self) -> None:
+        """Open the tool config update modal for the selected config."""
+        config = self.get_selected_item()
+        if not config:
+            self.app.notify("No tool config selected", severity="warning")
+            return
+        self._open_update_modal(config)
+
+    def delete_selected(self) -> None:
+        """Delete the currently selected tool config."""
+        config = self.get_selected_item()
+        if not config:
+            self.app.notify("No tool config selected", severity="warning")
+            return
+
+        config_name = config.name
+        self.app.push_screen(
+            ConfirmModal(
+                title="Delete Tool Config",
+                message=f"Delete tool config '{config_name}'?\n\n"
+                "Note: Deletion will fail if this config is used by any agents.",
+            ),
+            lambda confirmed: self._on_delete_confirmed(confirmed, str(config.id)),
+        )
+
+    # =========================================================================
+    # Internal async operations
+    # =========================================================================
+
+    @work(exclusive=True)
+    async def _open_create_modal(self) -> None:
+        """Open the tool config creation modal."""
+        try:
+            client = await self._ensure_client()
+            self.app.push_screen(
+                ToolConfigCreateModal(client=client),
+                self._on_config_created,
+            )
+        except Exception as e:
+            self.app.notify(f"Failed to open config creator: {e}", severity="error")
+
+    def _on_config_created(self, result: ToolConfigCreateResult | None) -> None:
+        """Callback when tool config creation modal is dismissed."""
+        if result is not None:
+            self.app.notify(
+                f"Created tool config: {result.config_name}",
+                severity="information",
+            )
+            # Refresh the list
+            self._do_search(self._last_search)
+
+    @work(exclusive=True)
+    async def _open_update_modal(self, config_summary: ToolConfigSummary) -> None:
+        """Fetch full tool config details and open the update modal."""
+        try:
+            client = await self._ensure_client()
+            # Fetch full config (includes the config JSON)
+            config = await client.get_tool_config(str(config_summary.id))
+            self.app.push_screen(
+                ToolConfigUpdateModal(client=client, config=config),
+                self._on_config_updated,
+            )
+        except Exception as e:
+            self.app.notify(f"Failed to open config editor: {e}", severity="error")
+
+    def _on_config_updated(self, result: ToolConfigCreateResult | None) -> None:
+        """Callback when tool config update modal is dismissed."""
+        if result is not None:
+            self.app.notify(
+                f"Updated tool config: {result.config_name}",
+                severity="information",
+            )
+            # Refresh the list
+            self._do_search(self._last_search)
+
+    def _on_delete_confirmed(self, confirmed: bool | None, config_id: str) -> None:
+        """Callback when delete confirmation modal is dismissed."""
+        if confirmed:
+            self._execute_delete(config_id)
+
+    @work(exclusive=True)
+    async def _execute_delete(self, config_id: str) -> None:
+        """Execute the delete operation."""
+        try:
+            client = await self._ensure_client()
+            response = await client.delete_tool_config(config_id)
+
+            self.app.notify(
+                f"Tool config '{response.deleted_tool_config.name}' deleted",
+                severity="information",
+            )
+            # Refresh the list
+            self._do_search(self._last_search)
+
+        except Exception as e:
+            error_msg = str(e)
+            if "in use" in error_msg.lower() or "referenced" in error_msg.lower():
+                self.app.notify(
+                    "Cannot delete: config is used by one or more agents",
+                    severity="error",
+                )
+            else:
+                self.app.notify(f"Failed to delete tool config: {e}", severity="error")
+
+
+class ToolConfigsScreen(NavScreen):
+    """Screen for browsing and managing tool configurations.
+
+    This is a proper Textual Screen that wraps the ToolConfigsWidget
+    and provides keybindings that work regardless of focus.
+    """
+
+    app: "AlloyRuntimeApp"
+
+    SCREEN_ID = "tool_configs"
+
+    DEFAULT_CSS = """
+    ToolConfigsScreen #screen-root {
+        height: 1fr;
+    }
+
+    ToolConfigsScreen .browser-widget {
+        height: 100%;
+        width: 100%;
+    }
+    """
+
+    # Screen-specific bindings (Ctrl+ prefixed)
+    # NavScreen.BINDINGS provides global navigation (c, a, t, s, m, d, g, ?)
+    BINDINGS = NavScreen.BINDINGS + [
+        Binding("ctrl+f", "focus_search", "Search", show=True),
+        Binding("ctrl+r", "refresh", "Refresh", show=True),
+        Binding("ctrl+y", "copy_name", "Copy Name", show=True),
+        Binding("ctrl+n", "new_config", "New", show=True),
+        Binding("ctrl+e", "edit_config", "Edit", show=True),
+        Binding("ctrl+d", "delete_config", "Delete", show=True),
+        Binding("escape", "focus_table", "Focus List", show=False),
+        Binding("enter", "copy_name", "Copy Name", show=False),
+    ]
+
+    def compose_content(self) -> ComposeResult:
+        """Compose the screen content."""
+        yield ToolConfigsWidget(id="tool-configs-widget")
+
+    def on_screen_resume(self) -> None:
+        """Focus the data table when screen becomes active."""
+        self.call_after_refresh(self._focus_table)
+
+    def _focus_table(self) -> None:
+        """Focus the data table."""
+        try:
+            widget = self.query_one("#tool-configs-widget", ToolConfigsWidget)
+            widget.focus_table()
+        except Exception:
+            pass
+
+    def _get_widget(self) -> ToolConfigsWidget:
+        """Get the tool configs widget."""
+        return self.query_one("#tool-configs-widget", ToolConfigsWidget)
+
+    # =========================================================================
+    # Actions (keybinding handlers)
+    # =========================================================================
+
+    def action_focus_search(self) -> None:
+        """Focus the search input."""
+        self._get_widget().focus_search()
+
+    def action_focus_table(self) -> None:
+        """Focus the data table."""
+        self._get_widget().focus_table()
+
+    def action_refresh(self) -> None:
+        """Refresh the current search results."""
+        self._get_widget().refresh_data()
+
+    def action_copy_name(self) -> None:
+        """Copy the tool config name to clipboard."""
+        self._get_widget().copy_config_name()
+
+    def action_new_config(self) -> None:
+        """Open the tool config creation modal."""
+        self._get_widget().open_new_modal()
+
+    def action_edit_config(self) -> None:
+        """Open the tool config update modal."""
+        self._get_widget().open_edit_modal()
+
+    def action_delete_config(self) -> None:
+        """Delete the selected tool config."""
+        self._get_widget().delete_selected()

cli/tui/widgets/agent_create_modal.py

@@ -0,0 +1,139 @@
+"""Agent creation modal for creating new agents.
+
+Provides a comprehensive modal screen that allows users to create a new agent
+with all required fields: name, models, tags, and optional configuration.
+"""
+
+from typing import Any
+
+from alloy_runtime_sdk.api_client.client import ApiClient
+from alloy_runtime_types.dtos.agents import CreateAgentRequest
+
+from cli.infrastructure.forms.agent_picker import AgentPicker
+from cli.infrastructure.forms.schema_picker import SchemaPicker
+from cli.infrastructure.forms.tag_picker import TagPicker
+from cli.infrastructure.forms.template_picker import TemplatePicker
+from cli.tui.widgets.agent_form_modal import AgentFormModal, AgentFormResult
+
+
+class AgentCreateModal(AgentFormModal):
+    """Modal for creating a new agent.
+
+    Provides a comprehensive form with:
+    - Required fields: name, models, tags
+    - Basic info: description, scope, base agent
+    - System instruction: template picker
+    - Schemas: input/output schema pickers or inline JSON schema builder
+    - Generation parameters: temperature, max_tokens, etc.
+    - Advanced: default headers with quick-add buttons
+
+    Usage:
+        def on_agent_created(result: AgentCreateResult | None):
+            if result is not None:
+                self.notify(f"Created agent: {result.agent_name}")
+                self._do_search("")  # Refresh list
+
+        client = await self.app.store.get_client()
+        self.app.push_screen(
+            AgentCreateModal(client=client),
+            on_agent_created
+        )
+    """
+
+    def __init__(
+        self,
+        client: ApiClient,
+        **kwargs: Any,
+    ) -> None:
+        """Initialize the agent creation modal.
+
+        Args:
+            client: ApiClient instance for API calls
+        """
+        super().__init__(client=client, **kwargs)
+
+    def _get_modal_title(self) -> str:
+        """Return the modal title."""
+        return "Create New Agent"
+
+    def _get_submit_button_text(self) -> str:
+        """Return the submit button text."""
+        return "Create Agent"
+
+    def _is_scope_editable(self) -> bool:
+        """Return whether the scope can be changed."""
+        return True
+
+    async def _submit_form(self) -> AgentFormResult:
+        """Submit the form to create a new agent."""
+        request = self._build_create_request()
+        response = await self._client.create_agent(request)
+
+        return AgentFormResult(
+            agent_id=str(response.id),
+            agent_name=response.name,
+            is_update=False,
+        )
+
+    def _build_create_request(self) -> CreateAgentRequest:
+        """Build the CreateAgentRequest from form fields."""
+        from textual.widgets import Input, TextArea
+
+        # Required: name
+        name = self.query_one("#name-input", Input).value.strip()
+
+        # Required: models
+        model_configs = self._build_model_configs()
+
+        # Required for non-global: tags
+        tag_picker = self.query_one("#tag-picker", TagPicker)
+        tags = tag_picker.selected_tags
+
+        # Basic info
+        description = self.query_one("#description-input", Input).value.strip() or None
+        scope = self._get_scope()
+
+        # Base agent
+        base_agent_picker = self.query_one("#base-agent-picker", AgentPicker)
+        base_agent_id = self._parse_uuid(
+            base_agent_picker.selected_agent_id, "Base Agent"
+        )
+
+        # System instruction template
+        template_picker = self.query_one("#system-template-picker", TemplatePicker)
+        system_instruction_version_id = self._parse_uuid(
+            template_picker.selected_id, "System Template"
+        )
+
+        # Schemas
+        output_schema_picker = self.query_one("#output-schema-picker", SchemaPicker)
+        output_schema_id = self._parse_uuid(
+            output_schema_picker.selected_id, "Output Schema"
+        )
+
+        # Output schema from JSON builder
+        output_schema = self._get_output_schema()
+
+        # Generation parameters
+        gen_params = self._build_generation_params()
+
+        # Advanced: default headers
+        default_headers = self._parse_json(
+            self.query_one("#default-headers-textarea", TextArea).text,
+            "Default Headers",
+        )
+
+        return CreateAgentRequest(
+            name=name,
+            models=model_configs,
+            tags=tags,
+            scope=scope,
+            base_agent_id=base_agent_id,
+            description=description,
+            system_instruction_version_id=system_instruction_version_id,
+            output_schema_id=output_schema_id,
+            output_schema=output_schema,
+            default_request_headers=default_headers,
+            generation_params=gen_params,
+            tools=self._build_tools(),
+        )