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

idun_agent_engine/server/lifespan.py

@@ -8,19 +8,15 @@ from collections.abc import Sequence
 from contextlib import asynccontextmanager
 
 from fastapi import FastAPI
+from idun_agent_schema.engine.guardrails import Guardrails
 
 from ..core.config_builder import ConfigBuilder
-from ..mcp import MCPClientRegistry
-
-from idun_agent_schema.engine.guardrails import Guardrails, Guardrail
-
 from ..guardrails.base import BaseGuardrail
 from ..telemetry import get_telemetry, sanitize_telemetry_config
 
 
 def _parse_guardrails(guardrails_obj: Guardrails) -> Sequence[BaseGuardrail]:
     """Adds the position of the guardrails (input/output) and returns the lift of updated guardrails."""
-
     from ..guardrails.guardrails_hub.guardrails_hub import GuardrailsHubGuard as GHGuard
 
     if not guardrails_obj:
@@ -70,8 +66,8 @@ async def configure_app(app: FastAPI, engine_config):
     print(f"✅ Agent '{agent_name}' initialized and ready to serve!")
 
     # Setup AGUI routes if the agent is a LangGraph agent
-    from ..agent.langgraph.langgraph import LanggraphAgent
     from ..agent.adk.adk import AdkAgent
+    from ..agent.langgraph.langgraph import LanggraphAgent
     # from ..server.routers.agui import setup_agui_router
 
     if isinstance(agent_instance, (LanggraphAgent, AdkAgent)):
@@ -91,7 +87,6 @@ async def configure_app(app: FastAPI, engine_config):
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     """FastAPI lifespan context to initialize and teardown the agent."""
-
     # Load config and initialize agent on startup
     print("Server starting up...")
     if not app.state.engine_config:

idun_agent_engine/server/routers/agent.py

@@ -84,6 +84,8 @@ async def invoke(
             )
         return ChatResponse(session_id=message["session_id"], response=response_content)
 
+    except HTTPException:
+        raise
     except Exception as e:  # noqa: BLE001
         raise HTTPException(status_code=500, detail=str(e)) from e
 
@@ -142,7 +144,6 @@ async def copilotkit_stream(
         try:
             # Get the accept header from the request
             accept_header = request.headers.get("accept")
-            agent_id = request.url.path.lstrip("/")
 
             # Create an event encoder to properly format SSE events
             encoder = EventEncoder(accept=accept_header or "")

idun_agent_engine/server/routers/base.py

@@ -1,8 +1,8 @@
 """Base routes for service health and landing info."""
 
 import os
-from typing import Optional
-from fastapi import APIRouter, Request, HTTPException
+
+from fastapi import APIRouter, HTTPException, Request
 from pydantic import BaseModel
 
 from ..._version import __version__
@@ -15,7 +15,7 @@ base_router = APIRouter()
 class ReloadRequest(BaseModel):
     """Request body for reload endpoint."""
 
-    path: Optional[str] = None
+    path: str | None = None
 
 
 @base_router.get("/health")
@@ -25,9 +25,8 @@ def health_check():
 
 
 @base_router.post("/reload")
-async def reload_config(request: Request, body: Optional[ReloadRequest] = None):
+async def reload_config(request: Request, body: ReloadRequest | None = None):
     """Reload the agent configuration from the manager or a file."""
-
     try:
         if body and body.path:
             print(f"🔄 Reloading configuration from file: {body.path}...")
@@ -60,6 +59,9 @@ async def reload_config(request: Request, body: Optional[ReloadRequest] = None):
             "message": "Agent configuration reloaded successfully",
         }
 
+    except HTTPException:
+        raise
+
     except Exception as e:
         print(f"❌ Error reloading configuration: {e}")
         raise HTTPException(

idun_agent_engine/telemetry/__init__.py

@@ -9,7 +9,6 @@ _telemetry_singleton: IdunTelemetry | None = None
 
 def get_telemetry() -> IdunTelemetry:
     """Return the process-wide telemetry singleton."""
-
     global _telemetry_singleton
     if _telemetry_singleton is None:
         _telemetry_singleton = IdunTelemetry()

idun_agent_engine/telemetry/config.py

@@ -13,7 +13,6 @@ def telemetry_enabled(environ: dict[str, str] | None = None) -> bool:
     Telemetry is ON by default. Users can disable it by setting the environment
     variable `IDUN_TELEMETRY_ENABLED` to a falsy value (e.g. "false", "0", "no").
     """
-
     env = os.environ if environ is None else environ
     raw = env.get(IDUN_TELEMETRY_ENABLED_ENV)
     if raw is None:

idun_agent_engine/telemetry/telemetry.py

@@ -113,7 +113,6 @@ def _get_or_create_distinct_id() -> str | None:
     - `IDUN_TELEMETRY_DISTINCT_ID` (if set)
     - A UUID persisted to `<cache_dir>/idun/telemetry_user_id`
     """
-
     raw = os.environ.get(_DISTINCT_ID_ENV)
     if raw and raw.strip():
         return raw.strip()
@@ -183,9 +182,10 @@ class IdunTelemetry:
             self._distinct_id = _get_or_create_distinct_id()
             return self._client
 
-    def capture(self, event: str, properties: dict[str, Any] | None = None) -> Future[None] | None:
+    def capture(
+        self, event: str, properties: dict[str, Any] | None = None
+    ) -> Future[None] | None:
         """Capture an event asynchronously (best-effort)."""
-
         if not self.enabled:
             return None
 
@@ -220,7 +220,6 @@ class IdunTelemetry:
 
     def shutdown(self, timeout_seconds: float = 1.0) -> None:
         """Best-effort flush/shutdown without blocking application shutdown."""
-
         executor = self._executor
         client = self._client
 

idun_agent_engine/templates/correction.py

@@ -1,7 +1,7 @@
 """Correction Agent Template."""
 
 import os
-from typing import TypedDict, Annotated, List, Any
+from typing import Annotated, Any, TypedDict
 
 try:
     from langchain.chat_models import init_chat_model
@@ -11,13 +11,13 @@ except ImportError:
     except ImportError:
         init_chat_model = None
 
-from langchain_core.messages import SystemMessage, BaseMessage
-from langgraph.graph import StateGraph, START, END
+from langchain_core.messages import BaseMessage, SystemMessage
+from langgraph.graph import END, START, StateGraph
 from langgraph.graph.message import add_messages
 
 
 class State(TypedDict):
-    messages: Annotated[List[BaseMessage], add_messages]
+    messages: Annotated[list[BaseMessage], add_messages]
 
 
 MODEL_NAME = os.getenv("CORRECTION_MODEL", "gemini-2.5-flash")
@@ -33,18 +33,15 @@ else:
     print("Warning: init_chat_model not found in langchain.")
 
 
-
 async def correct_text(state: State):
     """Correct the spelling, syntax, and grammar of the text."""
     if not llm:
-
         return {
             "messages": [
                 SystemMessage(content="Error: Model not initialized. Check logs.")
             ]
         }
 
-
     prompt = (
         f"You are a professional text corrector for {LANGUAGE}. "
         f"Correct the spelling, syntax, grammar, and conjugation of the following text. "

idun_agent_engine/templates/deep_research.py

@@ -1,6 +1,7 @@
 """Deep Research Agent Template."""
 
 import os
+
 from deepagents import create_deep_agent
 from tavily import TavilyClient
 

idun_agent_engine/templates/translation.py

@@ -1,7 +1,7 @@
 """Translation Agent Template."""
 
 import os
-from typing import TypedDict, Annotated, List
+from typing import Annotated, TypedDict
 
 # Try importing init_chat_model, fallback if necessary
 try:
@@ -12,14 +12,14 @@ except ImportError:
     except ImportError:
         init_chat_model = None
 
-from langchain_core.messages import SystemMessage, BaseMessage
-from langgraph.graph import StateGraph, START, END
+from langchain_core.messages import BaseMessage, SystemMessage
+from langgraph.graph import END, START, StateGraph
 from langgraph.graph.message import add_messages
 
 
 # Define the state
 class State(TypedDict):
-    messages: Annotated[List[BaseMessage], add_messages]
+    messages: Annotated[list[BaseMessage], add_messages]
 
 
 # Read configuration from environment variables

{idun_agent_engine-0.4.0.dist-info → idun_agent_engine-0.4.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: idun-agent-engine
-Version: 0.4.0
+Version: 0.4.2
 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.2.dist-info/RECORD

@@ -0,0 +1,86 @@
+idun_agent_engine/__init__.py,sha256=PhOL6foq5V0eXaoXw7xKUeCWXIWrOHrAFB8OuJnBqyM,550
+idun_agent_engine/_version.py,sha256=rHg5d6fgwIvzBZWjYtyruzXJK6jO7QljvyRBzdBlBBU,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=hljK87MJ5pfQWudbPhN4OpZxHXOXbZGLQA3mKaJ6dx4,11415
+idun_agent_engine/agent/haystack/__init__.py,sha256=mfYFciVPKRXBxPhnpIy5qPJ8mUSyywUAxiFXFPTzEV4,104
+idun_agent_engine/agent/haystack/haystack.py,sha256=nh86oGy7Q2mBbNdbvMfSYrtKLnNdXCI53S3SnzYp-J4,11039
+idun_agent_engine/agent/haystack/utils.py,sha256=sKRoPhzZWFw1NPsYwCockafzMBCCq3lGOrndbNE_C3M,609
+idun_agent_engine/agent/langgraph/__init__.py,sha256=CoBdkp9P4livdy5B0bvj9o7ftoqKmXEr9cZv4TZLncs,107
+idun_agent_engine/agent/langgraph/langgraph.py,sha256=s8mSm6exo69KBPIhCinRotS8v4DyMGQLG1nFL65rUbA,22871
+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=5wbFNKLomw7aWEnfZ6iyljTwpFO_rdgw4c3CUzwLsW4,26221
+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
+idun_agent_engine/guardrails/base.py,sha256=rvi7gcuKK8-3GtGicWzMWYsKKv0g3LjGBI-FycxgKVU,814
+idun_agent_engine/guardrails/guardrails_hub/__init__.py,sha256=MwWdgrM1dNzKwuS9ZEFw5Cin91JXh2PYlMma8yZFO7o,123
+idun_agent_engine/guardrails/guardrails_hub/guardrails_hub.py,sha256=3H3_1XLXfOB_hrv9cti6aX5CPUNBrV9eQPLwjGx8YN4,4918
+idun_agent_engine/mcp/__init__.py,sha256=PsBUpfFvvgzJhCFeIH8215VrHiHLnBCrP4jfHGw1gmQ,501
+idun_agent_engine/mcp/helpers.py,sha256=NMieZ6FSOaEpxeBLwUWvr-NYje1_AM3_h9LK7Sym6dk,6561
+idun_agent_engine/mcp/registry.py,sha256=ncur1Zp_2bSZBdJRTI8wIq2Lt-fP0oyPHPJH4RnEaKs,4241
+idun_agent_engine/observability/__init__.py,sha256=DCtK6b3xiX4dh0_8GBDOcSXQdcIJz2wTqqPa_ZFAES4,430
+idun_agent_engine/observability/base.py,sha256=Hst5dAxoJwPpiKw9p3MH79-AVZSNTciZ0hlq8K0rSFg,5686
+idun_agent_engine/observability/gcp_logging/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+idun_agent_engine/observability/gcp_logging/gcp_logging_handler.py,sha256=cMUKfukmSZhIdO2v5oWkqI2_c6umtYqlhzflnOeoq6A,1685
+idun_agent_engine/observability/gcp_trace/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+idun_agent_engine/observability/gcp_trace/gcp_trace_handler.py,sha256=WMg8u161P-rIS1k8jZiNcRyxvxNE7MwEkktKqcOFWHQ,3920
+idun_agent_engine/observability/langfuse/__init__.py,sha256=J8XcHV4aT1pF97k5EZiqrnYYPs9VjwfV5rUMihc5Pgk,128
+idun_agent_engine/observability/langfuse/langfuse_handler.py,sha256=c9Rzu7SCiyWOPVDimqwMY1Swvp57rKf-RePtHs-zhkw,2765
+idun_agent_engine/observability/phoenix/__init__.py,sha256=tEwJYijcvSGNhFW4QJmvBcTu1D0YVJkZRTmkNCGTteM,130
+idun_agent_engine/observability/phoenix/phoenix_handler.py,sha256=lGqSq-L1vmoEhAr9rbWO3KlNX5HSgBhCKESHMdZ-AfY,2539
+idun_agent_engine/observability/phoenix_local/__init__.py,sha256=m9dIw1GWGKAW4wP08jxA7j4yrOg0Nxq_08bwVh8YogE,146
+idun_agent_engine/observability/phoenix_local/phoenix_local_handler.py,sha256=wjOZuMpAxdD5D33rzxycNEzFMunETpPnYjiHjbjz5GA,4252
+idun_agent_engine/server/__init__.py,sha256=WaFektUsy37bNg2niAUy_TykzStukgWPnxC-t49CEwo,177
+idun_agent_engine/server/dependencies.py,sha256=hRviN2YEf8tvTek2F7IUIp-JH7_V-ru-HKbCqLAox2o,2117
+idun_agent_engine/server/lifespan.py,sha256=oXPDmCjNxCIQdrcZ_Q7bDx6mtndpLRI7VNeKEDIuQNw,4619
+idun_agent_engine/server/server_config.py,sha256=RYA7Y0c5aRw_WXaX8svFUIEtTPqzn3o-WQRm2p52C6g,213
+idun_agent_engine/server/routers/__init__.py,sha256=BgNzSVvHtGPGn5zhXhomwpKlDYBkeFi7xCbdcWVOgc8,102
+idun_agent_engine/server/routers/agent.py,sha256=QDqlR8P19ghLQrIFSd2w-c7iODXAv_P6bQ2AIYCWd7A,9182
+idun_agent_engine/server/routers/base.py,sha256=5kZtPE0QAUZGlvsry2ygzz95IqdjvE2cFqGepD9TA6I,3763
+idun_agent_engine/telemetry/__init__.py,sha256=jeKw2GMqqwoSh1sOxljVmVgP2BHObFYnZeNIF2NPwD8,518
+idun_agent_engine/telemetry/config.py,sha256=JJhw3hysT_0HdbdyuOevD9WeEo4Eu4x0Y3sH5NWZvEY,844
+idun_agent_engine/telemetry/telemetry.py,sha256=yOEExjFwhxvj5J6QW9cMeqFk6J6S0b1XlEb2Kni195o,7330
+idun_agent_engine/templates/__init__.py,sha256=xxbJZXaX6VEm_UrqzAOQcuujpEji5yqYzwQfwiqig8o,31
+idun_agent_engine/templates/correction.py,sha256=_NdgXLENpUhjT81CQI-gzH_yhGwbxSmBAo_VRFmBgQ0,1814
+idun_agent_engine/templates/deep_research.py,sha256=x9BJucpkHzocxmCQ3--HrxjU8A1-StwL-2SVPcslTTI,1078
+idun_agent_engine/templates/translation.py,sha256=bUEhhJuCpoON7PPi0yGE6f3SlZgXhN9WNwj1F9IyN9o,2143
+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=pSjnOXMbT2LuvNVPNCvBdYXRQG4EJbfrfiiC20NSIqk,1738
+idun_platform_cli/groups/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+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=DjCUSEbPLnHbYkxpkumLb7jZytMWr5WuPDxgLCTcy-8,2614
+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=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=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=7aXTyf8boBr5xyLAf1O8rXsPca5LUqHz1GqKhOYwPsY,1714
+idun_platform_cli/tui/screens/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+idun_platform_cli/tui/screens/create_agent.py,sha256=0j-euLfosun7syXAjZYuoexoXmIzmlnZjNiKcoMWKvc,24473
+idun_platform_cli/tui/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+idun_platform_cli/tui/utils/config.py,sha256=SIDJUfE7vzI3xGXnVPeD7CeTA1Bi6xtQStLvjDMwn50,7281
+idun_platform_cli/tui/validators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+idun_platform_cli/tui/validators/guardrails.py,sha256=E4KEVD0KV4gG9ZdlA591HA0seYY2xEDB3HeVPAxbH5E,3585
+idun_platform_cli/tui/validators/mcps.py,sha256=r_YOxIWLPXVDXN9lwO1UvhvbTns7WrwmunO1zydttu4,2883
+idun_platform_cli/tui/validators/observability.py,sha256=IWNuJg42eZUxjV-7v12kms7DpJsAs3FSSD5zLU5eKUc,2436
+idun_platform_cli/tui/widgets/__init__.py,sha256=w8rQsUxqTtxuhpOPxCgl1pmv2vjCJseKNc1lhfMMkBM,521
+idun_platform_cli/tui/widgets/chat_widget.py,sha256=hPWZSX2kwSGwFxWWW1TQ-HOvOa5Y3BjCJnj5WgOrVqY,5753
+idun_platform_cli/tui/widgets/guardrails_widget.py,sha256=EzrZEaAyx7ViW7Vk-QLyu235b1jqauCoyvM9i3qsuGY,14553
+idun_platform_cli/tui/widgets/identity_widget.py,sha256=LjpX-mgIinxJ5FzKBFz1HYbImwX7khpbetl86LsPnfw,9508
+idun_platform_cli/tui/widgets/mcps_widget.py,sha256=fXZzGbK7I4_yDFolvI3UbsR5yeKmW-sNz8STGKYk0E8,12125
+idun_platform_cli/tui/widgets/memory_widget.py,sha256=QCJF6bd21s-QWdQY-UqqPtWZSIl-9xil6ul-a2r6vL0,6966
+idun_platform_cli/tui/widgets/observability_widget.py,sha256=9ioS_BwOkW6NPtzJJa-S-mNUDOqP3QYYgTePcAun18s,14915
+idun_platform_cli/tui/widgets/serve_widget.py,sha256=fSr75DuOxiXr0KbbEDAl0t5iOlDqMQLRiHdMCrENZdI,2728
+idun_agent_engine-0.4.2.dist-info/METADATA,sha256=HrghAw_locsILEAp1BDj6D5SIq8xfFslUv6NndiLzx8,10870
+idun_agent_engine-0.4.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+idun_agent_engine-0.4.2.dist-info/entry_points.txt,sha256=XG3oxlSOaCrYKT1oyhKa0Ag1iJPMZ-WF6gaV_mzIJW4,52
+idun_agent_engine-0.4.2.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."""
@@ -36,7 +38,7 @@ def generate_dockerfile(dependency: Dependency) -> str:
         return ""  # Unreachable, but satisfies type checker
     if dependency == Dependency.REQUIREMENT:
         # TODO: use from file
-        requirements_dockerfile = f"""FROM python:3.12-slim
+        requirements_dockerfile = """FROM python:3.12-slim
 RUN apt-get update && pip install uv
 
 RUN uv pip install idun-agent-schema==0.3.1 --system
@@ -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,55 @@
+"""Telemetry utilities for CLI command tracking."""
+
+from __future__ import annotations
+
+import functools
+import time
+from collections.abc import Callable
+from typing import Any, 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;
-}
-
 
 """