idun-agent-engine 0.4.1__py3-none-any.whl → 0.4.2__py3-none-any.whl

idun_platform_cli/tui/widgets/mcps_widget.py

@@ -2,11 +2,18 @@
 
 from textual.app import ComposeResult
 from textual.containers import Horizontal, Vertical
-from textual.widgets import Static, Input, Button, RadioSet, RadioButton, TextArea, OptionList
 from textual.widget import Widget
+from textual.widgets import (
+    Button,
+    Input,
+    OptionList,
+    RadioButton,
+    RadioSet,
+    Static,
+    TextArea,
+)
 from textual.widgets.option_list import Option
 
-
 MCP_TEMPLATES = {
     "time": {
         "name": "time-reference",
@@ -34,14 +41,24 @@ class MCPsWidget(Widget):
         templates_row = Horizontal(classes="templates-row")
         templates_row.compose_add_child(Static("Select template:", classes="mcp-label"))
         option_list = OptionList(id="template_selector", classes="template-selector")
-        for template_name in MCP_TEMPLATES.keys():
+        for template_name in MCP_TEMPLATES:
             option_list.add_option(Option(template_name.title(), id=template_name))
         templates_row.compose_add_child(option_list)
-        templates_row.compose_add_child(Button("Add from Template", id="add_from_template_button", classes="add-template-btn"))
+        templates_row.compose_add_child(
+            Button(
+                "Add from Template",
+                id="add_from_template_button",
+                classes="add-template-btn",
+            )
+        )
         templates_section.compose_add_child(templates_row)
         yield templates_section
 
-        yield Button("+ Add Custom MCP Server", id="add_custom_mcp_button", classes="add-custom-btn")
+        yield Button(
+            "+ Add Custom MCP Server",
+            id="add_custom_mcp_button",
+            classes="add-custom-btn",
+        )
 
         yield Vertical(id="mcps_container", classes="mcps-container")
 
@@ -67,7 +84,9 @@ class MCPsWidget(Widget):
     def _add_from_template(self) -> None:
         template_selector = self.query_one("#template_selector", OptionList)
         if template_selector.highlighted is not None:
-            option_id = template_selector.get_option_at_index(template_selector.highlighted).id
+            option_id = template_selector.get_option_at_index(
+                template_selector.highlighted
+            ).id
             if option_id and str(option_id) in MCP_TEMPLATES:
                 template_data = MCP_TEMPLATES[str(option_id)].copy()
                 self._add_mcp_server(template_data)
@@ -95,52 +114,109 @@ class MCPsWidget(Widget):
 
         header = Horizontal(classes="mcp-header")
         name_value = template_data.get("name", "") if template_data else ""
-        header.compose_add_child(Static(name_value or f"Server {index + 1}", id=f"mcp_name_display_{index}", classes="mcp-name-display"))
-        header.compose_add_child(Button("Remove", id=f"remove_mcp_{index}", classes="remove-mcp-btn"))
+        header.compose_add_child(
+            Static(
+                name_value or f"Server {index + 1}",
+                id=f"mcp_name_display_{index}",
+                classes="mcp-name-display",
+            )
+        )
+        header.compose_add_child(
+            Button("Remove", id=f"remove_mcp_{index}", classes="remove-mcp-btn")
+        )
         card.compose_add_child(header)
 
         name_row = Horizontal(classes="mcp-field-row")
         name_row.compose_add_child(Static("Name:", classes="mcp-label"))
-        name_row.compose_add_child(Input(value=name_value, placeholder="server-name", id=f"mcp_name_{index}", classes="mcp-input"))
+        name_row.compose_add_child(
+            Input(
+                value=name_value,
+                placeholder="server-name",
+                id=f"mcp_name_{index}",
+                classes="mcp-input",
+            )
+        )
         card.compose_add_child(name_row)
 
         transport_row = Horizontal(classes="mcp-field-row")
         transport_row.compose_add_child(Static("Transport:", classes="mcp-label"))
         radio_set = RadioSet(id=f"mcp_transport_{index}")
 
-        transport_value = template_data.get("transport", "streamable_http") if template_data else "streamable_http"
-        radio_set.compose_add_child(RadioButton("stdio", id="stdio", value=(transport_value == "stdio")))
-        radio_set.compose_add_child(RadioButton("sse", id="sse", value=(transport_value == "sse")))
-        radio_set.compose_add_child(RadioButton("streamable_http", id="streamable_http", value=(transport_value == "streamable_http")))
-        radio_set.compose_add_child(RadioButton("websocket", id="websocket", value=(transport_value == "websocket")))
+        transport_value = (
+            template_data.get("transport", "streamable_http")
+            if template_data
+            else "streamable_http"
+        )
+        radio_set.compose_add_child(
+            RadioButton("stdio", id="stdio", value=(transport_value == "stdio"))
+        )
+        radio_set.compose_add_child(
+            RadioButton("sse", id="sse", value=(transport_value == "sse"))
+        )
+        radio_set.compose_add_child(
+            RadioButton(
+                "streamable_http",
+                id="streamable_http",
+                value=(transport_value == "streamable_http"),
+            )
+        )
+        radio_set.compose_add_child(
+            RadioButton(
+                "websocket", id="websocket", value=(transport_value == "websocket")
+            )
+        )
         transport_row.compose_add_child(radio_set)
         card.compose_add_child(transport_row)
 
-        http_fields = Vertical(id=f"mcp_http_fields_{index}", classes="http-fields-container")
+        http_fields = Vertical(
+            id=f"mcp_http_fields_{index}", classes="http-fields-container"
+        )
         http_fields.border_title = "HTTP Configuration"
 
         url_row = Horizontal(classes="mcp-field-row")
         url_row.compose_add_child(Static("URL:", classes="mcp-label"))
         url_value = template_data.get("url", "") if template_data else ""
-        url_row.compose_add_child(Input(value=url_value, placeholder="https://api.example.com/mcp", id=f"mcp_url_{index}", classes="mcp-input"))
+        url_row.compose_add_child(
+            Input(
+                value=url_value,
+                placeholder="https://api.example.com/mcp",
+                id=f"mcp_url_{index}",
+                classes="mcp-input",
+            )
+        )
         http_fields.compose_add_child(url_row)
 
         headers_row = Horizontal(classes="mcp-field-row")
         headers_row.compose_add_child(Static("Headers (JSON):", classes="mcp-label"))
         headers_value = template_data.get("headers", "") if template_data else ""
-        headers_row.compose_add_child(TextArea(text=str(headers_value) if headers_value else "", id=f"mcp_headers_{index}", classes="mcp-textarea"))
+        headers_row.compose_add_child(
+            TextArea(
+                text=str(headers_value) if headers_value else "",
+                id=f"mcp_headers_{index}",
+                classes="mcp-textarea",
+            )
+        )
         http_fields.compose_add_child(headers_row)
 
         http_fields.display = transport_value in ["sse", "streamable_http", "websocket"]
         card.compose_add_child(http_fields)
 
-        stdio_fields = Vertical(id=f"mcp_stdio_fields_{index}", classes="stdio-fields-container")
+        stdio_fields = Vertical(
+            id=f"mcp_stdio_fields_{index}", classes="stdio-fields-container"
+        )
         stdio_fields.border_title = "Stdio Configuration"
 
         command_row = Horizontal(classes="mcp-field-row")
         command_row.compose_add_child(Static("Command:", classes="mcp-label"))
         command_value = template_data.get("command", "") if template_data else ""
-        command_row.compose_add_child(Input(value=command_value, placeholder="npx", id=f"mcp_command_{index}", classes="mcp-input"))
+        command_row.compose_add_child(
+            Input(
+                value=command_value,
+                placeholder="npx",
+                id=f"mcp_command_{index}",
+                classes="mcp-input",
+            )
+        )
         stdio_fields.compose_add_child(command_row)
 
         args_row = Horizontal(classes="mcp-field-row")
@@ -150,7 +226,9 @@ class MCPsWidget(Widget):
             args_list = template_data["args"]
             if isinstance(args_list, list):
                 args_value = "\n".join(args_list)
-        args_textarea = TextArea(text=args_value, id=f"mcp_args_{index}", classes="mcp-textarea")
+        args_textarea = TextArea(
+            text=args_value, id=f"mcp_args_{index}", classes="mcp-textarea"
+        )
         args_textarea.placeholder = "run\n-i\n--rm"
         args_row.compose_add_child(args_textarea)
         stdio_fields.compose_add_child(args_row)
@@ -160,8 +238,11 @@ class MCPsWidget(Widget):
         env_value = ""
         if template_data and "env" in template_data:
             import json
+
             env_value = json.dumps(template_data["env"], indent=2)
-        env_row.compose_add_child(TextArea(text=env_value, id=f"mcp_env_{index}", classes="mcp-textarea"))
+        env_row.compose_add_child(
+            TextArea(text=env_value, id=f"mcp_env_{index}", classes="mcp-textarea")
+        )
         stdio_fields.compose_add_child(env_row)
 
         stdio_fields.display = transport_value == "stdio"
@@ -189,7 +270,9 @@ class MCPsWidget(Widget):
                 name = name_input.value
 
                 if not name:
-                    self.app.notify(f"Server {server_id + 1}: Name is required", severity="error")
+                    self.app.notify(
+                        f"Server {server_id + 1}: Name is required", severity="error"
+                    )
                     return None
 
                 radio_set = self.query_one(f"#mcp_transport_{server_id}", RadioSet)
@@ -206,7 +289,9 @@ class MCPsWidget(Widget):
                     url_input = self.query_one(f"#mcp_url_{server_id}", Input)
                     server_config["url"] = url_input.value
 
-                    headers_textarea = self.query_one(f"#mcp_headers_{server_id}", TextArea)
+                    headers_textarea = self.query_one(
+                        f"#mcp_headers_{server_id}", TextArea
+                    )
                     if headers_textarea.text.strip():
                         server_config["headers"] = headers_textarea.text.strip()
 
@@ -224,7 +309,10 @@ class MCPsWidget(Widget):
                 servers_data.append(server_config)
 
             except Exception:
-                self.app.notify(f"Error reading MCP server {server_id + 1}: check your configuration.", severity="error")
+                self.app.notify(
+                    f"Error reading MCP server {server_id + 1}: check your configuration.",
+                    severity="error",
+                )
                 return None
 
         return servers_data

idun_platform_cli/tui/widgets/memory_widget.py

@@ -92,7 +92,6 @@ class MemoryWidget(Widget):
 
     def get_data(self) -> CheckpointConfig | None:
         from idun_agent_schema.engine.langgraph import (
-            CheckpointConfig,
             InMemoryCheckpointConfig,
             PostgresCheckpointConfig,
             SqliteCheckpointConfig,

idun_platform_cli/tui/widgets/observability_widget.py

@@ -4,9 +4,9 @@ from idun_agent_schema.engine.observability_v2 import (
 )
 from textual.app import ComposeResult
 from textual.containers import Horizontal, Vertical
-from textual.widgets import Static, Input, RadioSet, RadioButton, Switch
-from textual.widget import Widget
 from textual.reactive import reactive
+from textual.widget import Widget
+from textual.widgets import Input, RadioButton, RadioSet, Static, Switch
 
 from idun_platform_cli.tui.validators.observability import validate_observability
 

idun_platform_cli/tui/widgets/serve_widget.py

@@ -1,6 +1,5 @@
 """Serve configuration widget."""
 
-from typing import Any
 
 from rich.syntax import Syntax
 from textual.app import ComposeResult
@@ -57,7 +56,7 @@ class ServeWidget(Widget):
                 self.config_data,
                 default_flow_style=False,
                 sort_keys=False,
-                allow_unicode=True
+                allow_unicode=True,
             )
 
             syntax = Syntax(
@@ -67,7 +66,7 @@ class ServeWidget(Widget):
                 line_numbers=True,
                 word_wrap=False,
                 indent_guides=True,
-                background_color="default"
+                background_color="default",
             )
 
             self.query_one("#yaml_content", Static).update(syntax)

idun_agent_engine/agent/haystack/haystack_model.py

@@ -1,13 +0,0 @@
-"""Configuration models for Haystack agents."""
-
-from typing import Literal
-
-from idun_agent_engine.core.engine_config import BaseAgentConfig
-
-
-class HaystackAgentConfig(BaseAgentConfig):
-    """Configuration model for Haystack Agents."""
-
-    type: Literal["haystack"] = "haystack"
-    component_type: Literal["pipeline", "agent"]
-    component_definition: str

idun_agent_engine/guardrails/guardrails_hub/utils.py

@@ -1 +0,0 @@
-"""Utils module."""

idun_agent_engine/server/routers/agui.py

@@ -1,47 +0,0 @@
-# """AGUI routes for CopilotKit integration with LangGraph agents."""
-
-# import logging
-# from typing import Annotated
-
-# from ag_ui_langgraph import add_langgraph_fastapi_endpoint
-# from copilotkit import LangGraphAGUIAgent
-# from ag_ui_adk import ADKAgent as ADKAGUIAgent
-# from ag_ui_adk import add_adk_fastapi_endpoint
-# from fastapi import APIRouter, Depends, HTTPException, Request
-
-# from idun_agent_engine.agent.langgraph.langgraph import LanggraphAgent
-# from idun_agent_engine.agent.adk.adk import AdkAgent
-# from idun_agent_engine.server.dependencies import get_agent
-
-# logging.basicConfig(
-#     format="%(asctime)s %(levelname)-8s %(message)s",
-#     level=logging.INFO,
-#     datefmt="%Y-%m-%d %H:%M:%S",
-# )
-
-# logger = logging.getLogger(__name__)
-
-
-# def setup_agui_router(app, agent: LanggraphAgent | AdkAgent) -> LangGraphAGUIAgent | ADKAGUIAgent:
-#     """Set up AGUI routes for CopilotKit integration.
-
-#     This function adds the LangGraph agent as a CopilotKit-compatible endpoint.
-
-#     Args:
-#         app: The FastAPI application instance
-#         agent: The initialized LangGraph agent instance
-#     """
-#     try:
-#         if isinstance(agent, LanggraphAgent):
-#             # Create the AGUI agent wrapper
-#             agui_agent = agent.copilotkit_agent_instance
-#         elif isinstance(agent, AdkAgent):
-#             # Create the AGUI agent wrapper
-#                 agui_agent = agent.copilotkit_agent_instance # TODO: duplicate in agent.adk.adk.py init
-#         else:
-#             raise ValueError(f"Unsupported agent type: {type(agent)}")
-#         return agui_agent
-#         logger.info(f"✅ AGUI endpoint configured at /agui for agent: {agent.name}")
-#     except Exception as e:
-#         logger.error(f"❌ Failed to setup AGUI router: {e}")
-#         raise HTTPException(status_code=500, detail=f"Failed to setup AGUI router: {e}") from e