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

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.1.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.1
+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

{idun_agent_engine-0.4.1.dist-info → idun_agent_engine-0.4.2.dist-info}/RECORD

@@ -1,64 +1,61 @@
 idun_agent_engine/__init__.py,sha256=PhOL6foq5V0eXaoXw7xKUeCWXIWrOHrAFB8OuJnBqyM,550
-idun_agent_engine/_version.py,sha256=46Sg_0NATQ79huYXlK2uZI6BAN4q9gzxYyxZAuU0Bh0,72
+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=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
+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=cbVrUs74a0p6omypMHWO3-1uqGLtJdiyAMrCheQHF5w,22966
+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=MosVcPjn7qXUAlmaYvnPquJ0jZ55pDOLfLv0i-D-A28,26049
+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=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=b345RHXgeZ15a4gfg8sUJ3Ke7Iwke6FR1RqDRfRUB0c,6589
-idun_agent_engine/mcp/registry.py,sha256=8zt9o8MZmYKx55xb0E7n-qOCR_3m2-XGf7ToQEwHkTU,4229
+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=C6mY_ZafDEHkMz8_ZeaEh--pyYuzJYEUqhoJ0sVP2wY,5570
+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=YBlUqUwPuOPi0yDNEy9x_g52nJBlSe9a0EEHDIHIr68,3898
+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=RP9MXiYDRe7zWTmZmzQBn6AjRn0y8jrBnWOaUUaMxNA,2795
+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=MVH8STOQ8wu-iYE_28y5dvw5FGT5PX0uQl0ldkpxw6Y,2080
-idun_agent_engine/server/lifespan.py,sha256=pTPZ36R4jlx1jq249kpj4DEe-opIGYiRNUHncKlnuGs,4670
+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=qLaVRKgMIu-YfOi-tPEzLNRZKZ4DaZEQ8KQ-4HrWbbY,9194
-idun_agent_engine/server/routers/agui.py,sha256=Z1G4fuo57MazQWfp7ao8QZ1or2H9BXLS_JB1nFPXAkE,1891
-idun_agent_engine/server/routers/base.py,sha256=wICOXCokCIRjmHuDPDTWqeXqOVPes1CtDtiR_HvtsF0,3756
-idun_agent_engine/telemetry/__init__.py,sha256=oYmmgUYvuwM-eV1qBeTgy6HoG6H3LC_S4fMH04QiMWY,519
-idun_agent_engine/telemetry/config.py,sha256=APSmZWocwSqB1xSxoZkJrSTZnWlBiQdif4whdXi8uBE,845
-idun_agent_engine/telemetry/telemetry.py,sha256=g14ouZnSIV85A58HeAH5IlJ_IDKO_uYo7kFbWRo5PpI,7319
+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=qApH4W99K31maxNrGf0hPVe_Wd4xjYjJrEKlu7PbRpw,1823
-idun_agent_engine/templates/deep_research.py,sha256=j4WypBkBz7V3EG-U8vawfNXDNhcCf2QiEK0veI4M68s,1077
-idun_agent_engine/templates/translation.py,sha256=vjp1yEyhDMFO4nEbq5WG7LAA7nLrhVAjznOkV7ob_Ss,2149
+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=mLTO8717ckmglm_oSl1xFxomk-eqwqrCU9Nl1uq27Dg,1711
+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=JeoQBZ_egkeKhQhB-FuGGrY1T1VG4f4ixuhjt0mDygE,2615
+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
@@ -66,24 +63,24 @@ idun_platform_cli/tui/css/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJW
 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/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=3Yav2gvJhh0el4BTLuYzYELrHWuKDBnNnNHlRIBUEJE,23881
+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=AEgjgnNkCtQwuGxI8lclteunKZXvn_TMx0eBx7uJWto,7286
+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=l6TJKxjpr8Nktwndh6C7Juz4oY-bqpg295jtYTKHuQ8,3537
-idun_platform_cli/tui/validators/mcps.py,sha256=oDxbxt7qOzsW8yjd_fwBnlDPOSxPqCXqbF5q8eEYRVQ,2852
+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=8RCAY97asVxyy7VV7IJ4kEAX9edK46UGO90BJ0X4BL0,5731
-idun_platform_cli/tui/widgets/guardrails_widget.py,sha256=3EDrfTKXyN7N8qMuWVy-BjK6ia3CZIE0EVm1sh6FhMw,14552
+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=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/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

@@ -38,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

idun_platform_cli/telemetry.py

@@ -4,7 +4,8 @@ from __future__ import annotations
 
 import functools
 import time
-from typing import Any, Callable, TypeVar
+from collections.abc import Callable
+from typing import Any, TypeVar
 
 from idun_agent_engine.telemetry import get_telemetry, sanitize_telemetry_config
 

idun_platform_cli/tui/schemas/create_agent.py

@@ -21,7 +21,7 @@ agent:
 # 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",
+        "HAYSTACK": "component_definition",
         "LANGGRAPH": "graph_definition",
         "ADK": "agent",
     }
@@ -35,21 +35,25 @@ class TUIAgentConfig(BaseModel):
     graph_definition: str
 
     @field_validator("*", mode="after")
-    def validate_not_null(cls, value: str | Any | None) -> str:
+    def validate_not_null(cls, value: str | Any | None) -> str:  # noqa: N805
         if value is None or value == "":
             raise ValueError("Cannot have empty fields!")
         return value
 
     def to_engine_config(self) -> dict[str, Any]:
+        sanitized_name = self.name.replace("-", "_").replace(" ", "_")
+
         agent_config = {
-            "name": self.name,
+            "name": sanitized_name if self.framework == "ADK" else 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["app_name"] = sanitized_name
             agent_config["session_service"] = {"type": "in_memory"}
             agent_config["memory_service"] = {"type": "in_memory"}
+        elif self.framework == "HAYSTACK":
+            agent_config["component_type"] = "pipeline"
 
         return {
             "server": {"api": {"port": self.port}},

idun_platform_cli/tui/screens/create_agent.py

@@ -7,7 +7,7 @@ from textual.app import ComposeResult
 from textual.containers import Container, Horizontal, Vertical
 from textual.reactive import reactive
 from textual.screen import Screen
-from textual.widgets import Button, Footer, Label, RichLog, Static
+from textual.widgets import Button, Label, RichLog, Static
 
 from idun_platform_cli.tui.css.create_agent import CREATE_AGENT_CSS
 from idun_platform_cli.tui.schemas.create_agent import TUIAgentConfig
@@ -71,7 +71,11 @@ class CreateAgentScreen(Screen):
                 config = self.config_manager.load_config()
                 if config:
                     port = config.get("server", {}).get("api", {}).get("port", 8008)
-                    pids = subprocess.check_output(["lsof", "-ti", f":{port}"], text=True).strip().split("\n")
+                    pids = (
+                        subprocess.check_output(["lsof", "-ti", f":{port}"], text=True)
+                        .strip()
+                        .split("\n")
+                    )
                     for pid in pids:
                         if pid:
                             try:
@@ -254,7 +258,10 @@ class CreateAgentScreen(Screen):
                 yield serve
                 yield chat
 
-        footer = Static("💡 Press Next to save section | Press Ctrl+Q to exit", classes="custom-footer")
+        footer = Static(
+            "💡 Press Next to save section | Press Ctrl+Q to exit",
+            classes="custom-footer",
+        )
         yield footer
 
     def on_mount(self) -> None:
@@ -395,9 +402,7 @@ class CreateAgentScreen(Screen):
                 if data is not None:
                     success, msg = self.config_manager.save_partial("memory", data)
                     if not success:
-                        self.notify(
-                            "Memory configuration is invalid", severity="error"
-                        )
+                        self.notify("Memory configuration is invalid", severity="error")
                         return
                     self.validated_sections.add("memory")
                     self._update_nav_checkmark("memory")
@@ -447,14 +452,20 @@ class CreateAgentScreen(Screen):
 
                 validated_servers, msg = validate_mcp_servers(data)
                 if validated_servers is None:
-                    self.notify("Error validating MCPs: make sure all fields are correct.", severity="error")
+                    self.notify(
+                        "Error validating MCPs: make sure all fields are correct.",
+                        severity="error",
+                    )
                     return
 
                 success, save_msg = self.config_manager.save_partial(
                     "mcp_servers", validated_servers
                 )
                 if not success:
-                    self.notify("Error saving MCPs: make sure all fields are correct.", severity="error")
+                    self.notify(
+                        "Error saving MCPs: make sure all fields are correct.",
+                        severity="error",
+                    )
                     return
 
                 self.validated_sections.add("mcps")
@@ -510,9 +521,15 @@ class CreateAgentScreen(Screen):
                             ["lsof", "-ti", f":{port}"],
                             capture_output=True,
                             text=True,
-                            check=True
+                            check=True,
+                        )
+                        pids = (
+                            subprocess.check_output(
+                                ["lsof", "-ti", f":{port}"], text=True
+                            )
+                            .strip()
+                            .split("\n")
                         )
-                        pids = subprocess.check_output(["lsof", "-ti", f":{port}"], text=True).strip().split("\n")
                         for pid in pids:
                             if pid:
                                 try:
@@ -563,7 +580,13 @@ class CreateAgentScreen(Screen):
 
             try:
                 process = subprocess.Popen(
-                    ["idun", "agent", "serve", "--source=file", f"--path={config_path}"],
+                    [
+                        "idun",
+                        "agent",
+                        "serve",
+                        "--source=file",
+                        f"--path={config_path}",
+                    ],
                     stdout=subprocess.PIPE,
                     stderr=subprocess.STDOUT,
                     text=True,
@@ -585,7 +608,10 @@ class CreateAgentScreen(Screen):
                 self.run_worker(self._stream_logs(process), exclusive=True)
 
             except Exception:
-                self.notify("Failed to start server. Check your configuration.", severity="error")
+                self.notify(
+                    "Failed to start server. Check your configuration.",
+                    severity="error",
+                )
 
     async def _stream_logs(self, process) -> None:
         import asyncio

idun_platform_cli/tui/utils/config.py

@@ -178,5 +178,5 @@ class ConfigManager:
             with self.agent_path.open("r") as f:
                 return yaml.safe_load(f) or {}
 
-        except Exception as e:
+        except Exception:
             return {}

idun_platform_cli/tui/validators/guardrails.py

@@ -1,12 +1,12 @@
 """Guardrails validation logic."""
 
 from idun_agent_schema.engine.guardrails_v2 import (
-    GuardrailConfigId,
+    BanListConfig,
     BiasCheckConfig,
-    ToxicLanguageConfig,
     CompetitionCheckConfig,
-    BanListConfig,
     DetectPIIConfig,
+    GuardrailConfigId,
+    ToxicLanguageConfig,
 )
 
 
@@ -19,7 +19,7 @@ def validate_guardrail(guardrail_id: str, config: dict) -> tuple[any, str]:
                     config_id=GuardrailConfigId.BIAS_CHECK,
                     api_key=config.get("api_key", ""),
                     reject_message=config.get("reject_message", "Bias detected"),
-                    threshold=threshold
+                    threshold=threshold,
                 )
                 return validated, "ok"
 
@@ -28,8 +28,10 @@ def validate_guardrail(guardrail_id: str, config: dict) -> tuple[any, str]:
                 validated = ToxicLanguageConfig(
                     config_id=GuardrailConfigId.TOXIC_LANGUAGE,
                     api_key=config.get("api_key", ""),
-                    reject_message=config.get("reject_message", "Toxic language detected"),
-                    threshold=threshold
+                    reject_message=config.get(
+                        "reject_message", "Toxic language detected"
+                    ),
+                    threshold=threshold,
                 )
                 return validated, "ok"
 

idun_platform_cli/tui/validators/mcps.py

@@ -30,12 +30,15 @@ def validate_mcp_servers(
                         None,
                         f"Server '{name}': command is required for stdio transport",
                     )
-            elif transport in ["sse", "streamable_http", "websocket"]:
-                if not server_data.get("url"):
-                    return (
-                        None,
-                        f"Server '{name}': url is required for {transport} transport",
-                    )
+            elif transport in [
+                "sse",
+                "streamable_http",
+                "websocket",
+            ] and not server_data.get("url"):
+                return (
+                    None,
+                    f"Server '{name}': url is required for {transport} transport",
+                )
 
             args = server_data.get("args", [])
             if isinstance(args, str):

idun_platform_cli/tui/widgets/chat_widget.py

@@ -22,7 +22,9 @@ class ChatWidget(Widget):
         with chat_container:
             yield RichLog(id="chat_history", highlight=True, markup=True, wrap=True)
 
-        thinking_container = Horizontal(classes="chat-thinking-container", id="chat_thinking")
+        thinking_container = Horizontal(
+            classes="chat-thinking-container", id="chat_thinking"
+        )
         thinking_container.display = False
         with thinking_container:
             yield LoadingIndicator(id="chat_spinner")

idun_platform_cli/tui/widgets/guardrails_widget.py

@@ -1,11 +1,11 @@
 """Guardrails configuration widget."""
 
+from idun_agent_schema.engine.guardrails_v2 import GuardrailsV2
 from textual.app import ComposeResult
-from textual.containers import Horizontal, Vertical, Grid
-from textual.widgets import Static, Input, Switch, RadioSet, RadioButton, TextArea
+from textual.containers import Grid, Horizontal, Vertical
 from textual.widget import Widget
+from textual.widgets import Input, RadioButton, RadioSet, Static, Switch, TextArea
 
-from idun_agent_schema.engine.guardrails_v2 import GuardrailsV2
 from idun_platform_cli.tui.validators.guardrails import validate_guardrail
 
 
@@ -310,7 +310,7 @@ class GuardrailsWidget(Widget):
             self.app.notify(
                 "Error validating Guardrails: make sure all fields are correct.",
                 severity="error",
-                timeout=10
+                timeout=10,
             )
             return None