idun-agent-engine 0.3.0__py3-none-any.whl → 0.3.2__py3-none-any.whl

idun_agent_engine/_version.py

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

idun_agent_engine/agent/langgraph/langgraph.py

@@ -96,7 +96,9 @@ class LanggraphAgent(agent_base.BaseAgent):
             RuntimeError: If the CopilotKit agent is not yet initialized.
         """
         if self._copilotkit_agent_instance is None:
-            raise RuntimeError("CopilotKit agent not initialized. Call initialize() first.")
+            raise RuntimeError(
+                "CopilotKit agent not initialized. Call initialize() first."
+            )
         return self._copilotkit_agent_instance
 
     @property
@@ -134,7 +136,7 @@ class LanggraphAgent(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:
@@ -205,7 +207,7 @@ class LanggraphAgent(agent_base.BaseAgent):
 
         self._copilotkit_agent_instance = LangGraphAGUIAgent(
             name=self._name,
-            description="Agent description", # TODO: add agent description
+            description="Agent description",  # TODO: add agent description
             graph=self._agent_instance,
             config={"callbacks": self._obs_callbacks} if self._obs_callbacks else None,
         )
@@ -292,7 +294,8 @@ class LanggraphAgent(agent_base.BaseAgent):
         # Try loading as a file path first
         try:
             import os
-            print("Current directory: ", os.getcwd()) # TODO remove
+
+            print("Current directory: ", os.getcwd())  # TODO remove
             from pathlib import Path
 
             resolved_path = Path(module_path).resolve()
@@ -310,16 +313,23 @@ class LanggraphAgent(agent_base.BaseAgent):
             spec.loader.exec_module(module)
 
             graph_builder = getattr(module, graph_variable_name)
-            return self._validate_graph_builder(graph_builder, module_path, graph_variable_name)
+            return self._validate_graph_builder(
+                graph_builder, module_path, graph_variable_name
+            )
 
         except (FileNotFoundError, ImportError):
             # Fallback: try loading as a python module
             try:
-                module = importlib.import_module(module_path)
+                module_import_path = (
+                    module_path[:-3] if module_path.endswith(".py") else module_path
+                )
+                module = importlib.import_module(module_import_path)
                 graph_builder = getattr(module, graph_variable_name)
-                return self._validate_graph_builder(graph_builder, module_path, graph_variable_name)
+                return self._validate_graph_builder(
+                    graph_builder, module_path, graph_variable_name
+                )
             except ImportError as e:
-                 raise ValueError(
+                raise ValueError(
                     f"Failed to load agent from {graph_definition}. Checked file path and python module: {e}"
                 ) from e
             except AttributeError as e:
@@ -327,19 +337,21 @@ class LanggraphAgent(agent_base.BaseAgent):
                     f"Variable '{graph_variable_name}' not found in module {module_path}: {e}"
                 ) from e
         except Exception as e:
-             raise ValueError(
+            raise ValueError(
                 f"Failed to load agent from {graph_definition}: {e}"
             ) from e
 
-    def _validate_graph_builder(self, graph_builder: Any, module_path: str, graph_variable_name: str) -> StateGraph:
-         # TODO to remove, dirty fix for template deepagent langgraph
+    def _validate_graph_builder(
+        self, graph_builder: Any, module_path: str, graph_variable_name: str
+    ) -> StateGraph:
+        # TODO to remove, dirty fix for template deepagent langgraph
         if not isinstance(graph_builder, StateGraph) and not isinstance(
             graph_builder, CompiledStateGraph
         ):
             raise TypeError(
                 f"The variable '{graph_variable_name}' from {module_path} is not a StateGraph instance."
             )
-        return graph_builder # type: ignore[return-value]
+        return graph_builder  # type: ignore[return-value]
 
     async def invoke(self, message: Any) -> Any:
         """Process a single input to chat with the agent.

idun_agent_engine/core/config_builder.py

@@ -53,6 +53,7 @@ class ConfigBuilder:
         self._mcp_servers: list[MCPServer] | None = None
         self._observability: list[ObservabilityConfig] | None = None
         self._guardrails: Guardrails | None = None
+
     def with_api_port(self, port: int) -> "ConfigBuilder":
         """Set the API port for the server.
 
@@ -126,15 +127,20 @@ class ConfigBuilder:
                 raise YAMLError(f"Failed to parse yaml file for Guardrails: {e}") from e
 
             try:
-                observability_list = yaml_config.get("engine_config", {}).get("observability")
+                observability_list = yaml_config.get("engine_config", {}).get(
+                    "observability"
+                )
                 if observability_list:
                     self._observability = [
-                        ObservabilityConfig.model_validate(obs) for obs in observability_list
+                        ObservabilityConfig.model_validate(obs)
+                        for obs in observability_list
                     ]
                 else:
                     self._observability = None
             except Exception as e:
-                raise YAMLError(f"Failed to parse yaml file for Observability: {e}") from e
+                raise YAMLError(
+                    f"Failed to parse yaml file for Observability: {e}"
+                ) from e
             # try:
             #     mcp_servers_list = yaml_config.get("engine_config", {}).get("mcp_servers") or yaml_config.get("engine_config", {}).get("mcpServers") # TODO to fix camelcase issues
             #     if mcp_servers_list:
@@ -310,7 +316,8 @@ class ConfigBuilder:
         if agent_type == AgentFramework.LANGGRAPH:
             from idun_agent_engine.agent.langgraph.langgraph import LanggraphAgent
             import os
-            print("Current directory: ", os.getcwd()) # TODO remove
+
+            print("Current directory: ", os.getcwd())  # TODO remove
             try:
                 validated_config = LangGraphAgentConfig.model_validate(agent_config_obj)
 
@@ -423,14 +430,17 @@ class ConfigBuilder:
             try:
                 validated_config = AdkAgentConfig.model_validate(agent_config_obj)
             except Exception as e:
-                raise ValueError(f"Cannot validate into a AdkAgentConfig model. Got {agent_config_obj}") from e
+                raise ValueError(
+                    f"Cannot validate into a AdkAgentConfig model. Got {agent_config_obj}"
+                ) from e
             agent_instance = AdkAgent()
         else:
             raise ValueError(f"Unsupported agent type: {agent_type}")
 
         # Initialize the agent with its configuration
         await agent_instance.initialize(
-            validated_config, observability_config#, mcp_registry=mcp_registry
+            validated_config,
+            observability_config,  # , mcp_registry=mcp_registry
         )  # type: ignore[arg-type]
         return agent_instance
 

idun_agent_engine/server/lifespan.py

@@ -54,9 +54,7 @@ async def configure_app(app: FastAPI, engine_config):
 
     # Use ConfigBuilder's centralized agent initialization, passing the registry
     try:
-        agent_instance = await ConfigBuilder.initialize_agent_from_config(
-            engine_config
-        )
+        agent_instance = await ConfigBuilder.initialize_agent_from_config(engine_config)
     except Exception as e:
         raise ValueError(
             f"Error retrieving agent instance from ConfigBuilder: {e}"
@@ -66,7 +64,7 @@ async def configure_app(app: FastAPI, engine_config):
     app.state.config = engine_config
     app.state.engine_config = engine_config
 
-    app.state.guardrails = guardrails  # TODO: to reactivate
+    app.state.guardrails = guardrails
     agent_name = getattr(agent_instance, "name", "Unknown")
     print(f"✅ Agent '{agent_name}' initialized and ready to serve!")
 
@@ -89,7 +87,6 @@ async def configure_app(app: FastAPI, engine_config):
     #     print(f"🔌 MCP servers ready: {servers}")
 
 
-
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     """FastAPI lifespan context to initialize and teardown the agent."""

idun_agent_engine/server/routers/agent.py

@@ -3,6 +3,10 @@
 import logging
 from typing import Annotated
 
+from ag_ui.core.types import RunAgentInput
+from ag_ui.encoder import EventEncoder
+from ag_ui_adk import ADKAgent as ADKAGUIAgent
+from copilotkit import LangGraphAGUIAgent
 from fastapi import APIRouter, Depends, HTTPException, Request, status
 from fastapi.responses import StreamingResponse
 from idun_agent_schema.engine.api import ChatRequest, ChatResponse
@@ -11,11 +15,6 @@ from idun_agent_schema.engine.guardrails import Guardrail
 from idun_agent_engine.agent.base import BaseAgent
 from idun_agent_engine.server.dependencies import get_agent, get_copilotkit_agent
 
-from ag_ui.core.types import RunAgentInput
-from ag_ui.encoder import EventEncoder
-from copilotkit import LangGraphAGUIAgent
-from ag_ui_adk import ADKAgent as ADKAGUIAgent
-
 logging.basicConfig(
     format="%(asctime)s %(levelname)-8s %(message)s",
     level=logging.INFO,
@@ -26,14 +25,23 @@ logger = logging.getLogger(__name__)
 agent_router = APIRouter()
 
 
+def _format_deep_agent_response(response_content: list[dict[str, str]]) -> str:
+    """Deep Research Agent responds with a list contaning a single dict: {'type': 'text', 'text': 'Your text'}."""
+    try:
+        response = response_content[0]["text"]
+        return response
+    except KeyError as k:
+        raise ValueError("Cannot parse Deep Research Agent's response") from k
+
+
 def _run_guardrails(
     guardrails: list[Guardrail], message: dict[str, str] | str, position: str
 ) -> None:
     """Validates the request's message, by running it on given guardrails. If input is a dict -> input, else its an output guardrails."""
     text = message["query"] if isinstance(message, dict) else message
     for guard in guardrails:
-        if guard.position == position and not guard.validate(text): # type: ignore[attr-defined]
-            raise HTTPException(status_code=429, detail=guard.reject_message) # type: ignore[attr-defined]
+        if guard.position == position and not guard.validate(text):  # type: ignore[attr-defined]
+            raise HTTPException(status_code=429, detail=guard.reject_message)  # type: ignore[attr-defined]
 
 
 @agent_router.get("/config")
@@ -60,7 +68,7 @@ async def invoke(
     """Process a chat message with the agent without streaming."""
     try:
         message = {"query": chat_request.query, "session_id": chat_request.session_id}
-        guardrails = getattr(request.app.state, 'guardrails', [])
+        guardrails = getattr(request.app.state, "guardrails", [])
         if guardrails:
             _run_guardrails(guardrails, message, position="input")
         response_content = await agent.invoke(
@@ -68,7 +76,14 @@ async def invoke(
         )
         if guardrails:
             _run_guardrails(guardrails, response_content, position="output")
+
+        if agent.name == "Deep Research Agent":
+            return ChatResponse(
+                session_id=message["session_id"],
+                response=_format_deep_agent_response(response_content),
+            )
         return ChatResponse(session_id=message["session_id"], response=response_content)
+
     except Exception as e:  # noqa: BLE001
         raise HTTPException(status_code=500, detail=str(e)) from e
 
@@ -89,6 +104,7 @@ async def stream(
     except Exception as e:  # noqa: BLE001
         raise HTTPException(status_code=500, detail=str(e)) from e
 
+
 @agent_router.post("/copilotkit/stream")
 async def copilotkit_stream(
     input_data: RunAgentInput,
@@ -140,7 +156,7 @@ async def copilotkit_stream(
                                 exc_info=True,
                             )
                             # Create a RunErrorEvent for encoding failures
-                            from ag_ui.core import RunErrorEvent, EventType
+                            from ag_ui.core import EventType, RunErrorEvent
 
                             error_event = RunErrorEvent(
                                 type=EventType.RUN_ERROR,
@@ -163,7 +179,7 @@ async def copilotkit_stream(
                     # ADKAgent should have yielded a RunErrorEvent, but if something went wrong
                     # in the async generator itself, we need to handle it
                     try:
-                        from ag_ui.core import RunErrorEvent, EventType
+                        from ag_ui.core import EventType, RunErrorEvent
 
                         error_event = RunErrorEvent(
                             type=EventType.RUN_ERROR,

{idun_agent_engine-0.3.0.dist-info → idun_agent_engine-0.3.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: idun-agent-engine
-Version: 0.3.0
+Version: 0.3.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
@@ -33,7 +33,7 @@ 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: httpx<0.29.0,>=0.28.1
-Requires-Dist: idun-agent-schema<1.0.0,>=0.2.7
+Requires-Dist: idun-agent-schema<1.0.0,>=0.3.1
 Requires-Dist: langchain-core<2.0.0,>=1.0.0
 Requires-Dist: langchain-google-vertexai<4.0.0,>=2.0.27
 Requires-Dist: langchain-mcp-adapters<0.2.0,>=0.1.0

{idun_agent_engine-0.3.0.dist-info → idun_agent_engine-0.3.2.dist-info}/RECORD

@@ -1,5 +1,5 @@
 idun_agent_engine/__init__.py,sha256=PhOL6foq5V0eXaoXw7xKUeCWXIWrOHrAFB8OuJnBqyM,550
-idun_agent_engine/_version.py,sha256=0EcilWMIT4cjN03dXuEtQDhswH9tkXZPcYqYHIbsc-o,72
+idun_agent_engine/_version.py,sha256=7BdLKK9bjmk2rdX4sIoYIOBM826NN6_v_3RnCieDd0s,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
@@ -10,10 +10,10 @@ idun_agent_engine/agent/haystack/haystack.py,sha256=k4NAx_zTBO9uiExM9NtuDAN94H1l
 idun_agent_engine/agent/haystack/haystack_model.py,sha256=EtOYnsWRufcrQufTRMeB3V-rZVQqfnmwKwPsYGfZdCs,362
 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=ozc0ClaAl4B-_dSxiRt8KO2oxGIwNI_m-Hyk2qyJV_c,22641
+idun_agent_engine/agent/langgraph/langgraph.py,sha256=vM5ppt0s_2izzZVbVXXGXGCVUGSN-GZhvQ5-6YbOgVM,22904
 idun_agent_engine/core/__init__.py,sha256=F0DMDlWcSWS_1dvh3xMbrdcVvZRHVnoAFFgREuSJfBI,408
 idun_agent_engine/core/app_factory.py,sha256=XiLrctFT_n8LP3flKFwJoJDbiWPiw98N9lbkpR8P1O0,2597
-idun_agent_engine/core/config_builder.py,sha256=j_gia6z66Xun2ZOtaO4TWwRMfAIA6lnk8IGaRMJKEZM,25484
+idun_agent_engine/core/config_builder.py,sha256=Q6f7lITqIHFlAfXEIfQvz_6s_GVFc4zVotOFF_YTFho,25641
 idun_agent_engine/core/engine_config.py,sha256=IR8WhbenDstNSL7ORrUW8AnzgS3exFQxtwip66pFhcM,545
 idun_agent_engine/core/server_runner.py,sha256=vLlgLQM-xyvFgJMgFW2eWZoN1oc0x9FGL6bH8WsF6O8,4897
 idun_agent_engine/guardrails/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -37,10 +37,10 @@ idun_agent_engine/observability/phoenix_local/__init__.py,sha256=m9dIw1GWGKAW4wP
 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=REB93NTAivgYtZ9-7xFKb5RdDZSy3IrMAF5LM39-SAo,3886
+idun_agent_engine/server/lifespan.py,sha256=whWv7Gsl3CI_C13vWxl4wODOzlxIPRKSe58-_8B4zI8,3840
 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=OtviTU3nClyEHHUR5mJOMJvaNgcGR5CAkD5dGMA6Clk,8392
+idun_agent_engine/server/routers/agent.py,sha256=HQkPRYFE5CadpzEQ3La8kK7lt5QUVuZTk-rzhZg9EsU,8995
 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/templates/__init__.py,sha256=xxbJZXaX6VEm_UrqzAOQcuujpEji5yqYzwQfwiqig8o,31
@@ -52,9 +52,9 @@ idun_platform_cli/main.py,sha256=jWL7Ob0p4KdRUqgPTP_EB68n7z2LyMKC2DeUsfWlBO4,200
 idun_platform_cli/groups/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 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=CdZCbqmDG_PFoI1EFPn6Ibl9Bexel_8eO7XgmzYVCRg,2525
+idun_platform_cli/groups/agent/package.py,sha256=LdIFAcfrlUcCssmSliYQIi0NFMuVkpx6t8WhMHATan8,2525
 idun_platform_cli/groups/agent/serve.py,sha256=2AbL0G1WqR33jlyiGaNvAoPZ3G1o52KYUptz_HaAjIg,3863
-idun_agent_engine-0.3.0.dist-info/METADATA,sha256=prVc5VpdJJCfRNPVUTZyUxZmF4DCCC9wxpTC9TA7X-c,10648
-idun_agent_engine-0.3.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-idun_agent_engine-0.3.0.dist-info/entry_points.txt,sha256=XG3oxlSOaCrYKT1oyhKa0Ag1iJPMZ-WF6gaV_mzIJW4,52
-idun_agent_engine-0.3.0.dist-info/RECORD,,
+idun_agent_engine-0.3.2.dist-info/METADATA,sha256=3mHURTBewd-c5G3EwzKxEANkXnD6VUMO2N7zmoeobF0,10648
+idun_agent_engine-0.3.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+idun_agent_engine-0.3.2.dist-info/entry_points.txt,sha256=XG3oxlSOaCrYKT1oyhKa0Ag1iJPMZ-WF6gaV_mzIJW4,52
+idun_agent_engine-0.3.2.dist-info/RECORD,,

idun_platform_cli/groups/agent/package.py

@@ -39,8 +39,8 @@ def generate_dockerfile(dependency: Dependency) -> str:
         requirements_dockerfile = f"""FROM python:3.12-slim
 RUN apt-get update && pip install uv
 
-RUN uv pip install idun-agent-schema==0.3.0 --system
-RUN uv pip install idun-agent-engine==0.3.0 --system
+RUN uv pip install idun-agent-schema==0.3.1 --system
+RUN uv pip install idun-agent-engine==0.3.1 --system
 
 COPY . .
 RUN uv pip install -r requirements.txt --system