idun-agent-engine 0.3.8__py3-none-any.whl → 0.4.0__py3-none-any.whl

idun_platform_cli/tui/css/main.py

@@ -0,0 +1,92 @@
+CSS = """
+Screen {
+    align: center middle;
+}
+
+
+Container {
+    width: 90%;
+    max-width: 80;
+    height: 40;
+    border: heavy white;
+    padding: 2 4;
+}
+
+.ascii-logo {
+    text-align: center;
+    color: yellow;
+}
+
+.platform {
+    text-align: center;
+    text-style: bold;
+    color: cyan;
+    margin-top: 2;
+}
+
+.tagline {
+    text-align: center;
+    color: white;
+    margin-top: 1;
+}
+
+.built-by {
+    text-align: center;
+    color: gray;
+    margin-top: 1;
+}
+
+.version {
+    text-align: center;
+    color: gray;
+    margin-top: 0;
+}
+
+.link-container {
+    align: center middle;
+    width: 100%;
+    height: auto;
+    margin-top: 2;
+}
+
+.links {
+    color: green;
+    margin: 0 2;
+}
+
+.actions {
+    align: center middle;
+    width: 100%;
+    height: auto;
+    padding: 2;
+}
+
+.actions Button {
+    width: 32;
+    margin: 1 0;
+    background: transparent;
+    border: none;
+    color: white;
+}
+
+.actions Button:hover {
+    background: $boost;
+    color: $text;
+}
+
+.actions Button:focus {
+    background: $surface;
+    text-style: bold;
+}
+
+.question_prompt {
+    width: 100%;
+    height: auto;
+    text-align: center;
+    text-style: bold;
+    color: white;
+    margin-top: 2;
+}
+
+
+"""

idun_platform_cli/tui/main.py

@@ -0,0 +1,87 @@
+from enum import StrEnum
+from typing import Any
+
+from textual.app import App, ComposeResult
+from textual.containers import Container, Horizontal, VerticalScroll
+from textual.widget import Widget
+from textual.widgets import Button, Footer, Link, Static
+
+from idun_agent_engine._version import __version__
+from idun_platform_cli.tui.css.main import CSS
+from idun_platform_cli.tui.screens.create_agent import CreateAgentScreen
+from idun_platform_cli.tui.utils.config import ConfigManager
+
+
+class Actions(StrEnum):
+    NEW_AGENT = "new_agent"
+    UPDATE_AGENT = "update_agent"
+    TOUR = "tour"
+    EXIT = "exit"
+
+
+class MainPageActions(Widget):
+    CSS_PATH = "selection_list.tcss"
+
+    def compose(self) -> ComposeResult:
+        with VerticalScroll(classes="actions"):
+            yield Button("✨ Configure a new Agent", id="new_agent")
+            # yield Button("🔨 Modify an existing Agent", id="update_agent")
+            # yield Button("🔍 Tour Idun Agent Platform", id="tour")
+            yield Button("🚪 Exit", id="exit")
+
+    def on_mount(self):
+        self.query_one("#new_agent", Button).focus()
+
+
+class IdunApp(App):
+    CSS = CSS
+    REPO = "https://github.com/idun-group/idun-agent-platform"
+    DOCS = "https://idun-group.github.io/idun-agent-platform"
+
+    BINDINGS = [
+        ("up", "focus_previous", "Previous"),
+        ("down", "focus_next", "Next"),
+    ]
+
+    config_manager = ConfigManager()
+    config: dict[str, Any] = {}
+
+    def compose(self):
+        with Container():
+            yield Static(
+                """██╗██████╗ ██╗   ██╗███╗   ██╗
+██║██╔══██╗██║   ██║████╗  ██║
+██║██║  ██║██║   ██║██╔██╗ ██║
+██║██║  ██║██║   ██║██║╚██╗██║
+██║██████╔╝╚██████╔╝██║ ╚████║
+╚═╝╚═════╝  ╚═════╝ ╚═╝  ╚═══╝""",
+                classes="ascii-logo",
+            )
+
+            yield Static("IDUN AGENT PLATFORM", classes="platform")
+            yield Static("Deploy, guard and monitor any agent", classes="tagline")
+            yield Static("Built with ❤️ by Idun Group", classes="built-by")
+            yield Static(f"v{__version__}", classes="version")
+            with Horizontal(classes="link-container"):
+                yield Link("⭐️ Github", url=self.REPO, classes="links")
+                yield Link("📚 Docs", url=self.DOCS, classes="links")
+
+            yield Static("What do you want to do?", classes="question_prompt")
+            yield MainPageActions()
+        yield Footer()
+
+    def on_button_pressed(self, event: Button.Pressed):
+        import sys
+
+        button_id = event.button.id
+        match button_id:
+            case Actions.EXIT:
+                sys.exit(0)
+
+            case Actions.NEW_AGENT:
+                self.push_screen(CreateAgentScreen())
+
+
+if __name__ == "__main__":
+    app = IdunApp()
+    app.run()

idun_platform_cli/tui/schemas/create_agent.py

@@ -0,0 +1,60 @@
+from __future__ import annotations
+
+from typing import Any
+
+from pydantic import BaseModel, field_validator
+
+"""
+server:
+  api:
+    port: 8010
+
+agent:
+  type: "LANGGRAPH"
+  config:
+    name: "Guardrails"
+    graph_definition: "example_agent.py:app"
+
+"""
+
+
+# we need to map the graph_definition, pipeline_definition, agent_definition fields based on framework
+AGENT_SOURCE_KEY_MAPPING: dict[str, str] = dict(
+    {
+        "HAYSTACK": "pipeline_definition",
+        "LANGGRAPH": "graph_definition",
+        "ADK": "agent",
+    }
+)
+
+
+class TUIAgentConfig(BaseModel):
+    name: str
+    framework: str
+    port: int
+    graph_definition: str
+
+    @field_validator("*", mode="after")
+    def validate_not_null(cls, value: str | Any | None) -> str:
+        if value is None or value == "":
+            raise ValueError("Cannot have empty fields!")
+        return value
+
+    def to_engine_config(self) -> dict[str, Any]:
+        agent_config = {
+            "name": self.name,
+            AGENT_SOURCE_KEY_MAPPING[self.framework]: self.graph_definition,
+        }
+
+        if self.framework == "ADK":
+            agent_config["app_name"] = self.name.replace("-", "_").replace(" ", "_")
+            agent_config["session_service"] = {"type": "in_memory"}
+            agent_config["memory_service"] = {"type": "in_memory"}
+
+        return {
+            "server": {"api": {"port": self.port}},
+            "agent": {
+                "type": self.framework,
+                "config": agent_config,
+            },
+        }