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

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/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/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/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.dist-info → idun_agent_engine-0.4.1.dist-info}/METADATA

@@ -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.dist-info → idun_agent_engine-0.4.1.dist-info}/RECORD

@@ -1,10 +1,10 @@
 idun_agent_engine/__init__.py,sha256=PhOL6foq5V0eXaoXw7xKUeCWXIWrOHrAFB8OuJnBqyM,550
-idun_agent_engine/_version.py,sha256=IhNtSwHLVf0OkJXf7kUXlUOI7HCuAsywgZymi_DTmac,72
+idun_agent_engine/_version.py,sha256=46Sg_0NATQ79huYXlK2uZI6BAN4q9gzxYyxZAuU0Bh0,72
 idun_agent_engine/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 idun_agent_engine/agent/__init__.py,sha256=foyOoRdI_04q1b6f2A5EXEpWSCKjZxpgWMWrKcsHNl8,220
 idun_agent_engine/agent/base.py,sha256=c-3gljSHQpm6aY0JNDmXkaPtcW55rXdsI8Cgv9l4bCs,3294
 idun_agent_engine/agent/adk/__init__.py,sha256=jLV2Z9qQGZtBpF_0pQI1FRCPJ_J1G3Z6cEAzHnQyuu4,83
-idun_agent_engine/agent/adk/adk.py,sha256=29YC5rgLjdod-CU3a2JqgAvzYcD5u2EY7pylozGNi9c,11375
+idun_agent_engine/agent/adk/adk.py,sha256=0lfubR5ylfmfuSTOryULxnlJS3gAxpP3xtRaDypGqvE,11415
 idun_agent_engine/agent/haystack/__init__.py,sha256=y5ADrD8gWBeYIvV7tmu6OpPdJ8POHt-tyraIL7RkkWI,179
 idun_agent_engine/agent/haystack/haystack.py,sha256=k4NAx_zTBO9uiExM9NtuDAN94H1lrCWtHf1GEWEN16g,10966
 idun_agent_engine/agent/haystack/haystack_model.py,sha256=EtOYnsWRufcrQufTRMeB3V-rZVQqfnmwKwPsYGfZdCs,362
@@ -13,7 +13,7 @@ idun_agent_engine/agent/langgraph/__init__.py,sha256=CoBdkp9P4livdy5B0bvj9o7ftoq
 idun_agent_engine/agent/langgraph/langgraph.py,sha256=cbVrUs74a0p6omypMHWO3-1uqGLtJdiyAMrCheQHF5w,22966
 idun_agent_engine/core/__init__.py,sha256=F0DMDlWcSWS_1dvh3xMbrdcVvZRHVnoAFFgREuSJfBI,408
 idun_agent_engine/core/app_factory.py,sha256=2w4o7wV8FQCZdBTk3W7cS0vnmpUM_TdAmphFKHbL31w,2597
-idun_agent_engine/core/config_builder.py,sha256=oTWzdaYARZUGvVKVznMZT7pPRKKbCKQuXkCTno0Xn-8,25757
+idun_agent_engine/core/config_builder.py,sha256=MosVcPjn7qXUAlmaYvnPquJ0jZ55pDOLfLv0i-D-A28,26049
 idun_agent_engine/core/engine_config.py,sha256=IR8WhbenDstNSL7ORrUW8AnzgS3exFQxtwip66pFhcM,545
 idun_agent_engine/core/server_runner.py,sha256=DwN5kHiKVvUJLw5iSkaxs2rKRaMgOwSUZmeUPSuDbf8,4899
 idun_agent_engine/guardrails/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -22,7 +22,7 @@ idun_agent_engine/guardrails/guardrails_hub/__init__.py,sha256=MwWdgrM1dNzKwuS9Z
 idun_agent_engine/guardrails/guardrails_hub/guardrails_hub.py,sha256=DgcUIDy9_bcVcpYVvDJowFe_QVlCpEkr8hntPRgRmS0,3456
 idun_agent_engine/guardrails/guardrails_hub/utils.py,sha256=bC6-MsCVF7xKTr48z7OzJJUeWvqAB7BiHeNTiKsme70,20
 idun_agent_engine/mcp/__init__.py,sha256=ygChpEQ4MzMpEifW3szi5kNiLIMhOhJEtqxy0nmewDU,501
-idun_agent_engine/mcp/helpers.py,sha256=KYYAi9VxX6mRvw70ZiN_6rUz2fMt_KMn3-vUFoa6GC0,5084
+idun_agent_engine/mcp/helpers.py,sha256=b345RHXgeZ15a4gfg8sUJ3Ke7Iwke6FR1RqDRfRUB0c,6589
 idun_agent_engine/mcp/registry.py,sha256=8zt9o8MZmYKx55xb0E7n-qOCR_3m2-XGf7ToQEwHkTU,4229
 idun_agent_engine/observability/__init__.py,sha256=DCtK6b3xiX4dh0_8GBDOcSXQdcIJz2wTqqPa_ZFAES4,430
 idun_agent_engine/observability/base.py,sha256=C6mY_ZafDEHkMz8_ZeaEh--pyYuzJYEUqhoJ0sVP2wY,5570
@@ -53,34 +53,37 @@ idun_agent_engine/templates/deep_research.py,sha256=j4WypBkBz7V3EG-U8vawfNXDNhcC
 idun_agent_engine/templates/translation.py,sha256=vjp1yEyhDMFO4nEbq5WG7LAA7nLrhVAjznOkV7ob_Ss,2149
 idun_platform_cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 idun_platform_cli/main.py,sha256=B7b1x0kgfuiAWGmSCFJeu_zkWiodJjOGu0nWuNAWM-k,299
+idun_platform_cli/telemetry.py,sha256=mLTO8717ckmglm_oSl1xFxomk-eqwqrCU9Nl1uq27Dg,1711
 idun_platform_cli/groups/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-idun_platform_cli/groups/init.py,sha256=gQo_khZf-arvsG51bdwFXQyfaKvIBFaK9v5KU21AeDo,692
+idun_platform_cli/groups/init.py,sha256=gBWU7tQ0v9aaPNSQXorKWrOkUc5YgJU06pzDC-qG97U,769
 idun_platform_cli/groups/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 idun_platform_cli/groups/agent/main.py,sha256=QMGQi3JZ76SeFI3miIjVWpMt0L-hGz5FwxtTPQX4-Uw,301
-idun_platform_cli/groups/agent/package.py,sha256=LdIFAcfrlUcCssmSliYQIi0NFMuVkpx6t8WhMHATan8,2525
-idun_platform_cli/groups/agent/serve.py,sha256=2AbL0G1WqR33jlyiGaNvAoPZ3G1o52KYUptz_HaAjIg,3863
+idun_platform_cli/groups/agent/package.py,sha256=JeoQBZ_egkeKhQhB-FuGGrY1T1VG4f4ixuhjt0mDygE,2615
+idun_platform_cli/groups/agent/serve.py,sha256=ujCmJ3JsrjHkXvNwyB7V4poAaTxMQAPcIbRPDVybRQo,3947
 idun_platform_cli/tui/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-idun_platform_cli/tui/main.py,sha256=QO63okOX7s91r5JDA5mVjvmHuDSY9IArO6GQRaDYxCk,3083
+idun_platform_cli/tui/main.py,sha256=1gFEdbynBfiBLrZrKvNJN16Uc_3DOJl1spexRz38Q-E,3080
 idun_platform_cli/tui/css/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-idun_platform_cli/tui/css/create_agent.py,sha256=SxqNhB-sEbBcq9B_X4fktyhi31DJgJj61ymR4s0i9PE,12264
-idun_platform_cli/tui/css/main.py,sha256=9ii_dobgiz2DNdFhOC6VZSRNrRybI1a6ZykXrVqn7A4,1185
+idun_platform_cli/tui/css/create_agent.py,sha256=ZyFNv6-EUTUaJ_817wh0KUvdu8rtuuPj9jTbigLfOR8,14050
+idun_platform_cli/tui/css/main.py,sha256=gJ_kNb2lC2l99lq8NqtYU2imXAjUpxfiOaf1mNJr-Po,1134
 idun_platform_cli/tui/schemas/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 idun_platform_cli/tui/schemas/create_agent.py,sha256=2jsZ5v1sZLKmypgimdVoeVjd-wOylTE3IUYPw63eb8Y,1512
 idun_platform_cli/tui/screens/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-idun_platform_cli/tui/screens/create_agent.py,sha256=mtuN7HtQvwloiDZVIp9olJnEA9qHBmQPy-6YdiFXkaE,18332
+idun_platform_cli/tui/screens/create_agent.py,sha256=3Yav2gvJhh0el4BTLuYzYELrHWuKDBnNnNHlRIBUEJE,23881
 idun_platform_cli/tui/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-idun_platform_cli/tui/utils/config.py,sha256=An-pfUiD-sYQ12QLiE_M0NHAcpBCYo4TnxUbAmqodIA,6316
+idun_platform_cli/tui/utils/config.py,sha256=AEgjgnNkCtQwuGxI8lclteunKZXvn_TMx0eBx7uJWto,7286
 idun_platform_cli/tui/validators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-idun_platform_cli/tui/validators/guardrails.py,sha256=XIDzouBtNo63ppyGFMx9lcRHquxOvhY67ouJpF01u50,2894
+idun_platform_cli/tui/validators/guardrails.py,sha256=l6TJKxjpr8Nktwndh6C7Juz4oY-bqpg295jtYTKHuQ8,3537
 idun_platform_cli/tui/validators/mcps.py,sha256=oDxbxt7qOzsW8yjd_fwBnlDPOSxPqCXqbF5q8eEYRVQ,2852
 idun_platform_cli/tui/validators/observability.py,sha256=IWNuJg42eZUxjV-7v12kms7DpJsAs3FSSD5zLU5eKUc,2436
-idun_platform_cli/tui/widgets/__init__.py,sha256=d0tv94kZiFO6u7Rm8qvAXp5njDZqeVa7WfdU3A1GRdo,407
-idun_platform_cli/tui/widgets/guardrails_widget.py,sha256=ZYnvdDiUkOcAAp8gcR8ThpL8wVeAlmQrhPj5j-gUdOw,14299
-idun_platform_cli/tui/widgets/identity_widget.py,sha256=JWwfE4Dedv7F6r40tGnscHh8oevUr-_a-k5hZlXKvVg,8840
-idun_platform_cli/tui/widgets/mcps_widget.py,sha256=x2IS0SKIX6sfM5yBXW2d_arxWnwuaTEKUfYMTK-hZHw,10958
-idun_platform_cli/tui/widgets/observability_widget.py,sha256=Q2_vP7oC5LXznFUjMpse_1dLorrfGtbL11rhpEQOY8c,15064
-idun_platform_cli/tui/widgets/serve_widget.py,sha256=B2wK16AHo8BP-igZ0MJWhdCzuILvTzT0zn8PSJ_Szvc,2950
-idun_agent_engine-0.4.0.dist-info/METADATA,sha256=8PJDB-q0o90J46WfpMOrp13U9W-uJOeOYga1z2ZBeKA,10870
-idun_agent_engine-0.4.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-idun_agent_engine-0.4.0.dist-info/entry_points.txt,sha256=XG3oxlSOaCrYKT1oyhKa0Ag1iJPMZ-WF6gaV_mzIJW4,52
-idun_agent_engine-0.4.0.dist-info/RECORD,,
+idun_platform_cli/tui/widgets/__init__.py,sha256=w8rQsUxqTtxuhpOPxCgl1pmv2vjCJseKNc1lhfMMkBM,521
+idun_platform_cli/tui/widgets/chat_widget.py,sha256=8RCAY97asVxyy7VV7IJ4kEAX9edK46UGO90BJ0X4BL0,5731
+idun_platform_cli/tui/widgets/guardrails_widget.py,sha256=3EDrfTKXyN7N8qMuWVy-BjK6ia3CZIE0EVm1sh6FhMw,14552
+idun_platform_cli/tui/widgets/identity_widget.py,sha256=LjpX-mgIinxJ5FzKBFz1HYbImwX7khpbetl86LsPnfw,9508
+idun_platform_cli/tui/widgets/mcps_widget.py,sha256=5AToZHU5PAUJEuq0HhL4Mi_VmWlFLyTeY5pohWY3rKw,10974
+idun_platform_cli/tui/widgets/memory_widget.py,sha256=QHCH0et_lJa9pheTpcZdbzTDUzw95Do2koi6hEdmrOc,6996
+idun_platform_cli/tui/widgets/observability_widget.py,sha256=9AO7-nqMEIx_9FCmqbtprxoBejq4w8bR9wOhxl2BbmA,14915
+idun_platform_cli/tui/widgets/serve_widget.py,sha256=0FinYhkyA0LJnX1g4JDrkcD5df-l-qgMxjRjgF0CleQ,2749
+idun_agent_engine-0.4.1.dist-info/METADATA,sha256=xGS6C3n58AGB9WezKja-6Oy2EigEcuO3v8M6JKnZFu0,10870
+idun_agent_engine-0.4.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+idun_agent_engine-0.4.1.dist-info/entry_points.txt,sha256=XG3oxlSOaCrYKT1oyhKa0Ag1iJPMZ-WF6gaV_mzIJW4,52
+idun_agent_engine-0.4.1.dist-info/RECORD,,

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_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_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_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_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_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_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()