idun-agent-engine 0.4.0__tar.gz → 0.4.1__tar.gz

{idun_agent_engine-0.4.0 → idun_agent_engine-0.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: idun-agent-engine
-Version: 0.4.0
+Version: 0.4.1
 Summary: Python SDK and runtime to serve AI agents with FastAPI, LangGraph, and observability.
 Project-URL: Homepage, https://github.com/geoffreyharrazi/idun-agent-platform
 Project-URL: Repository, https://github.com/geoffreyharrazi/idun-agent-platform
@@ -31,7 +31,7 @@ Requires-Dist: deepagents<1.0.0,>=0.2.8
 Requires-Dist: fastapi<0.116.0,>=0.115.0
 Requires-Dist: google-adk<2.0.0,>=1.19.0
 Requires-Dist: google-cloud-logging<4.0.0,>=3.10.0
-Requires-Dist: guardrails-ai<0.8.0,>=0.7.0
+Requires-Dist: guardrails-ai<0.8.0,>=0.7.2
 Requires-Dist: httpx<0.29.0,>=0.28.1
 Requires-Dist: idun-agent-schema<1.0.0,>=0.3.8
 Requires-Dist: langchain-core<2.0.0,>=1.0.0

{idun_agent_engine-0.4.0 → idun_agent_engine-0.4.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "idun-agent-engine"
-version = "0.4.0"
+version = "0.4.1"
 description = "Python SDK and runtime to serve AI agents with FastAPI, LangGraph, and observability."
 authors = [{ name = "Geoffrey HARRAZI", email = "geoffreyharrazi@gmail.com" }]
 requires-python = ">=3.12,<3.14"
@@ -32,7 +32,7 @@ classifiers = [
 dependencies = [
     "tavily-python>=0.7.9,<0.8.0",
     "deepagents>=0.2.8,<1.0.0",
-    "guardrails-ai>=0.7.0, <0.8.0",
+    "guardrails-ai>=0.7.2, <0.8.0",
     "fastapi>=0.115.0,<0.116.0",
     "uvicorn>=0.35.0,<0.36.0",
     "langgraph>=1.0.0,<2.0.0",
@@ -50,7 +50,7 @@ dependencies = [
     "arize-phoenix-otel>=0.2.0,<1.0.0",
     "openinference-instrumentation-langchain>=0.1.13,<1.0.0",
     "langchain>=1.0.0,<2.0.0",
-    "textual>=7.3.0,<7.4.0",  
+    "textual>=7.3.0,<7.4.0",
     # Pin <12: 12.x currently depends on placeholder jmespath==99.99.99 and fails in Docker
     "arize-phoenix>=11.22.0,<12.0.0",
     "idun-agent-schema>=0.3.8,<1.0.0",

{idun_agent_engine-0.4.0 → idun_agent_engine-0.4.1}/src/idun_agent_engine/_version.py

@@ -1,3 +1,3 @@
 """Version information for Idun Agent Engine."""
 
-__version__ = "0.4.0"
+__version__ = "0.4.1"

{idun_agent_engine-0.4.0 → idun_agent_engine-0.4.1}/src/idun_agent_engine/agent/adk/adk.py

@@ -123,7 +123,7 @@ class AdkAgent(agent_base.BaseAgent):
         # Observability (provider-agnostic)
         if observability_config:
             handlers, infos = observability.create_observability_handlers(
-                observability_config # type: ignore[arg-type]
+                observability_config  # type: ignore[arg-type]
             )
             self._obs_callbacks = []
             for handler in handlers:
@@ -153,6 +153,7 @@ class AdkAgent(agent_base.BaseAgent):
 
                 if is_langfuse_enabled:
                     import os
+
                     langfuse_pk = os.environ.get("LANGFUSE_PUBLIC_KEY")
                     langfuse_host = os.environ.get("LANGFUSE_BASE_URL")
                     print(f"LANGFUSE_PUBLIC_KEY: {langfuse_pk}")
@@ -171,7 +172,9 @@ class AdkAgent(agent_base.BaseAgent):
                     except Exception as e:
                         print(f"Failed to instrument Google ADK: {e}")
             except Exception as e:
-                print(f"Error checking observability config for ADK instrumentation: {e}")
+                print(
+                    f"Error checking observability config for ADK instrumentation: {e}"
+                )
 
         # Initialize Session Service
         await self._initialize_session_service()

{idun_agent_engine-0.4.0 → idun_agent_engine-0.4.1}/src/idun_agent_engine/core/config_builder.py

@@ -4,6 +4,7 @@ This module provides a fluent API for building configuration objects using Pydan
 This approach ensures type safety, validation, and consistency with the rest of the codebase.
 """
 
+import os
 from pathlib import Path
 from typing import Any
 
@@ -517,6 +518,9 @@ class ConfigBuilder:
     def load_from_file(config_path: str = "config.yaml") -> EngineConfig:
         """Load configuration from a YAML file and return a validated EngineConfig.
 
+        Sets IDUN_CONFIG_PATH environment variable to enable MCP helper functions
+        (get_adk_tools, get_langchain_tools) to automatically discover the config file.
+
         Args:
             config_path: Path to the configuration YAML file
 
@@ -532,6 +536,9 @@ class ConfigBuilder:
             # Resolve relative to the current working directory
             path = Path.cwd() / path
 
+        # Set IDUN_CONFIG_PATH for MCP helpers to discover
+        os.environ["IDUN_CONFIG_PATH"] = str(path)
+
         with open(path) as f:
             config_data = yaml.safe_load(f)
 

{idun_agent_engine-0.4.0 → idun_agent_engine-0.4.1}/src/idun_agent_engine/mcp/helpers.py

@@ -6,6 +6,7 @@ import os
 from idun_agent_engine.mcp.registry import MCPClientRegistry
 from idun_agent_schema.engine.mcp_server import MCPServer
 
+
 def _extract_mcp_configs(config_data: dict[str, Any]) -> list[MCPServer]:
     """Parse MCP server configs from a config dictionary."""
     mcp_configs_data = config_data.get("mcp_servers") or config_data.get("mcpServers")
@@ -86,6 +87,7 @@ def _fetch_config_from_api() -> dict[str, Any]:
     except yaml.YAMLError as e:
         raise ValueError(f"Failed to parse config YAML: {e}") from e
 
+
 def get_adk_tools_from_file(config_path: str | Path) -> list[Any]:
     """
     Loads MCP configurations from a YAML file and returns a list of ADK toolsets.
@@ -99,6 +101,7 @@ def get_adk_tools_from_file(config_path: str | Path) -> list[Any]:
     config_data = _load_config_from_file(config_path)
     return _get_toolsets_from_data(config_data)
 
+
 def get_adk_tools_from_api() -> list[Any]:
     """
     Fetches configuration from the Idun Manager API and returns a list of ADK toolsets.
@@ -112,13 +115,31 @@ def get_adk_tools_from_api() -> list[Any]:
 
 def get_adk_tools(config_path: str | Path | None = None) -> list[Any]:
     """
-    Returns ADK toolsets using config from file when provided, otherwise from API.
+    Returns ADK toolsets using config from file when provided, from IDUN_CONFIG_PATH env var, or from API.
+
+    The function resolves configuration in the following order:
+    1. Uses the provided config_path if specified
+    2. Uses IDUN_CONFIG_PATH environment variable if set
+    3. Falls back to fetching from Idun Manager API
+
+    Args:
+        config_path: Optional path to configuration YAML file. If provided, takes precedence.
 
     Returns:
         List of initialized ADK McpToolset instances.
+
+    Raises:
+        ValueError: If no config source is available or API credentials are missing.
+        FileNotFoundError: If specified config file doesn't exist.
     """
     if config_path:
         return get_adk_tools_from_file(config_path)
+
+    # Check for IDUN_CONFIG_PATH environment variable
+    env_config_path = os.environ.get("IDUN_CONFIG_PATH")
+    if env_config_path:
+        return get_adk_tools_from_file(env_config_path)
+
     return get_adk_tools_from_api()
 
 
@@ -140,8 +161,29 @@ async def get_langchain_tools_from_api() -> list[Any]:
 
 async def get_langchain_tools(config_path: str | Path | None = None) -> list[Any]:
     """
-    Returns LangChain tool instances using config from file when provided, otherwise from API.
+    Returns LangChain tool instances using config from file when provided, from IDUN_CONFIG_PATH env var, or from API.
+
+    The function resolves configuration in the following order:
+    1. Uses the provided config_path if specified
+    2. Uses IDUN_CONFIG_PATH environment variable if set
+    3. Falls back to fetching from Idun Manager API
+
+    Args:
+        config_path: Optional path to configuration YAML file. If provided, takes precedence.
+
+    Returns:
+        List of initialized LangChain tool instances.
+
+    Raises:
+        ValueError: If no config source is available or API credentials are missing.
+        FileNotFoundError: If specified config file doesn't exist.
     """
     if config_path:
         return await get_langchain_tools_from_file(config_path)
+
+    # Check for IDUN_CONFIG_PATH environment variable
+    env_config_path = os.environ.get("IDUN_CONFIG_PATH")
+    if env_config_path:
+        return await get_langchain_tools_from_file(env_config_path)
+
     return await get_langchain_tools_from_api()

{idun_agent_engine-0.4.0 → idun_agent_engine-0.4.1}/src/idun_platform_cli/groups/agent/package.py

@@ -4,6 +4,8 @@ from pathlib import Path
 
 import click
 
+from idun_platform_cli.telemetry import track_command
+
 
 class Dependency(StrEnum):
     """Dependency Enum."""
@@ -57,6 +59,7 @@ CMD ["idun", "agent", "serve", "--source=manager"]
 @click.command("package")
 @click.argument("path", default=".")
 @click.option("--target", required=False, default=".")
+@track_command("agent package")
 def package_command(path: str, target: str):
     """Packages the agent and it's dependencies into a Dockerfile. You can specifiy the input path and the destination. Defaults to current directory."""
     dependency = get_dependencies(path)

{idun_agent_engine-0.4.0 → idun_agent_engine-0.4.1}/src/idun_platform_cli/groups/agent/serve.py

@@ -8,6 +8,7 @@ from idun_agent_engine.core.app_factory import create_app
 from idun_agent_engine.core.config_builder import ConfigBuilder
 from idun_agent_engine.core.engine_config import EngineConfig
 from idun_agent_engine.core.server_runner import run_server
+from idun_platform_cli.telemetry import track_command
 
 
 class ServerSource(StrEnum):
@@ -84,6 +85,7 @@ class Serve:
 @click.command("serve")
 @click.option("--source", required=True)
 @click.option("--path")
+@track_command("agent serve")
 def serve_command(source: str, path: str | None):
     """Reads a config and exposes it's agent as an API. Config is either fetched from the manager, or from a path.
 

{idun_agent_engine-0.4.0 → idun_agent_engine-0.4.1}/src/idun_platform_cli/groups/init.py

@@ -2,10 +2,12 @@ import sys
 
 import click
 
+from idun_platform_cli.telemetry import track_command
 from idun_platform_cli.tui.main import IdunApp
 
 
 @click.command("init")
+@track_command("init")
 def init_command() -> None:
     """Starts a terminal user interface that guides you through configuring and managing Idun agents."""
     try:

idun_agent_engine-0.4.1/src/idun_platform_cli/telemetry.py

@@ -0,0 +1,54 @@
+"""Telemetry utilities for CLI command tracking."""
+
+from __future__ import annotations
+
+import functools
+import time
+from typing import Any, Callable, TypeVar
+
+from idun_agent_engine.telemetry import get_telemetry, sanitize_telemetry_config
+
+F = TypeVar("F", bound=Callable[..., Any])
+
+
+def track_command(command_name: str) -> Callable[[F], F]:
+    """Decorator to track CLI command invocations via telemetry.
+
+    Args:
+        command_name: The name of the command to track (e.g., "init", "agent serve").
+
+    Returns:
+        A decorator that wraps the command function with telemetry tracking.
+    """
+
+    def decorator(func: F) -> F:
+        @functools.wraps(func)
+        def wrapper(*args: Any, **kwargs: Any) -> Any:
+            telemetry = get_telemetry()
+            start_time = time.time()
+            success = True
+            error_message: str | None = None
+
+            try:
+                return func(*args, **kwargs)
+            except Exception as e:
+                success = False
+                error_message = str(e)
+                raise
+            finally:
+                duration_ms = (time.time() - start_time) * 1000
+                telemetry.capture(
+                    "cli command invoked",
+                    properties={
+                        "command": command_name,
+                        "arguments": sanitize_telemetry_config(kwargs),
+                        "success": success,
+                        "error": error_message,
+                        "duration_ms": round(duration_ms, 2),
+                    },
+                )
+                telemetry.shutdown(timeout_seconds=0.5)
+
+        return wrapper  # type: ignore[return-value]
+
+    return decorator

{idun_agent_engine-0.4.0 → idun_agent_engine-0.4.1}/src/idun_platform_cli/tui/css/create_agent.py

@@ -467,35 +467,38 @@ Footer > .footer--description {
     padding: 2;
 }
 
-.serve-summary {
+.serve-yaml-display {
     width: 100%;
+    height: auto;
+    max-height: 15;
     border: round white;
     border-title-color: yellow;
     border-title-style: bold;
-    padding: 2;
+    padding: 1;
     margin-bottom: 2;
+    overflow-y: auto;
 }
 
-.summary-box {
+#yaml_content {
     width: 100%;
-    border: round cyan;
-    border-title-color: cyan;
-    border-title-style: bold;
-    padding: 1;
-    margin-bottom: 1;
+    height: auto;
+    color: white;
 }
 
-.summary-box:last-child {
-    margin-bottom: 0;
+.serve-button-container {
+    width: 100%;
+    height: auto;
+    align: center middle;
 }
 
 .validate-run-btn {
-    width: 100%;
+    width: 1fr;
     height: 3;
     background: green;
     color: white;
     text-style: bold;
     border: round white;
+    margin: 0 1;
 }
 
 .validate-run-btn:hover {
@@ -509,16 +512,16 @@ Footer > .footer--description {
     border: round yellow;
 }
 
-Button#validate_run_button.kill-mode {
+Button#save_run_button.kill-mode {
     background: red;
 }
 
-Button#validate_run_button.kill-mode:hover {
+Button#save_run_button.kill-mode:hover {
     background: darkred;
     color: white;
 }
 
-Button#validate_run_button.kill-mode:focus {
+Button#save_run_button.kill-mode:focus {
     background: darkred;
     color: white;
     border: round yellow;
@@ -786,4 +789,124 @@ Switch {
     text-align: center;
     text-style: bold;
 }
+
+#widget-chat {
+    padding: 2;
+}
+
+.chat-history-container {
+    width: 100%;
+    height: 1fr;
+    border: round white;
+    border-title-color: yellow;
+    border-title-style: bold;
+    padding: 1;
+    margin-bottom: 2;
+}
+
+#chat_history {
+    width: 100%;
+    height: 100%;
+    background: black;
+    color: white;
+}
+
+.chat-thinking-container {
+    width: 100%;
+    height: 2;
+    align: left middle;
+    padding: 0 1;
+}
+
+#chat_spinner {
+    width: auto;
+    height: auto;
+    color: yellow;
+}
+
+#thinking_label {
+    width: auto;
+    height: auto;
+    color: yellow;
+    margin-left: 1;
+    text-style: italic;
+}
+
+.chat-input-container {
+    width: 100%;
+    height: 5;
+    align: left middle;
+}
+
+.chat-input {
+    width: 1fr;
+    height: 3;
+    color: white;
+    background: transparent;
+    border: round white;
+    padding: 0 1;
+    margin-right: 1;
+}
+
+.chat-input:focus {
+    border: round cyan;
+}
+
+.chat-input:disabled {
+    opacity: 0.5;
+}
+
+.send-btn {
+    width: 12;
+    height: 3;
+    background: green;
+    color: white;
+    text-style: bold;
+    border: round white;
+}
+
+.send-btn:hover {
+    background: cyan;
+    color: black;
+}
+
+.send-btn:focus {
+    background: cyan;
+    color: black;
+    border: round yellow;
+}
+
+.send-btn:disabled {
+    background: gray;
+    opacity: 0.5;
+}
+
+#widget-memory {
+    padding: 2;
+}
+
+.memory-main {
+    width: 100%;
+    height: auto;
+    border: round white;
+    border-title-color: yellow;
+    border-title-style: bold;
+    padding: 1;
+    margin-bottom: 1;
+}
+
+.checkpoint-config-container {
+    width: 100%;
+    height: auto;
+    margin-top: 1;
+}
+
+.checkpoint-fields-section {
+    width: 100%;
+    height: auto;
+    border: round white;
+    border-title-color: yellow;
+    border-title-style: bold;
+    padding: 1;
+}
 """

{idun_agent_engine-0.4.0 → idun_agent_engine-0.4.1}/src/idun_platform_cli/tui/css/main.py

@@ -30,10 +30,16 @@ Container {
     margin-top: 1;
 }
 
+.features-subtle {
+    text-align: center;
+    color: $text-muted;
+    margin-top: 1;
+}
+
 .built-by {
     text-align: center;
     color: gray;
-    margin-top: 1;
+    margin-top: 2;
 }
 
 .version {
@@ -79,14 +85,5 @@ Container {
     text-style: bold;
 }
 
-.question_prompt {
-    width: 100%;
-    height: auto;
-    text-align: center;
-    text-style: bold;
-    color: white;
-    margin-top: 2;
-}
-
 
 """

{idun_agent_engine-0.4.0 → idun_agent_engine-0.4.1}/src/idun_platform_cli/tui/main.py

@@ -59,14 +59,14 @@ class IdunApp(App):
             )
 
             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("Deploy, guard and monitor your agents.", 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 Link("🌐 Website", url=self.DOCS, classes="links")
 
-            yield Static("What do you want to do?", classes="question_prompt")
             yield MainPageActions()
         yield Footer()