idun-agent-engine 0.3.0__tar.gz → 0.3.1__tar.gz

{idun_agent_engine-0.3.0 → idun_agent_engine-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: idun-agent-engine
-Version: 0.3.0
+Version: 0.3.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

{idun_agent_engine-0.3.0 → idun_agent_engine-0.3.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "idun-agent-engine"
-version = "0.3.0"
+version = "0.3.1"
 description = "Python SDK and runtime to serve AI agents with FastAPI, LangGraph, and observability."
 authors = [{ name = "Geoffrey HARRAZI", email = "geoffreyharrazi@gmail.com" }]
 requires-python = ">=3.12,<3.14"

{idun_agent_engine-0.3.0 → idun_agent_engine-0.3.1}/src/idun_agent_engine/_version.py

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

{idun_agent_engine-0.3.0 → idun_agent_engine-0.3.1}/src/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-0.3.0 → idun_agent_engine-0.3.1}/src/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-0.3.0 → idun_agent_engine-0.3.1}/src/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-0.3.0 → idun_agent_engine-0.3.1}/src/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 → idun_agent_engine-0.3.1}/src/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