idun-agent-engine 0.2.5__tar.gz → 0.2.7__tar.gz

{idun_agent_engine-0.2.5 → idun_agent_engine-0.2.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: idun-agent-engine
-Version: 0.2.5
+Version: 0.2.7
 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
@@ -19,15 +19,17 @@ Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Software Development :: Libraries
 Classifier: Typing :: Typed
 Requires-Python: <3.14,>=3.12
+Requires-Dist: ag-ui-langgraph<0.1.0,>=0.0.20
 Requires-Dist: ag-ui-protocol<0.2.0,>=0.1.8
 Requires-Dist: aiosqlite<0.22.0,>=0.21.0
 Requires-Dist: arize-phoenix-otel<1.0.0,>=0.2.0
 Requires-Dist: arize-phoenix<12.0.0,>=11.22.0
 Requires-Dist: click>=8.2.1
-Requires-Dist: fastapi<0.117.0,>=0.116.1
+Requires-Dist: copilotkit<0.2.0,>=0.1.72
+Requires-Dist: fastapi<0.116.0,>=0.115.0
 Requires-Dist: google-adk<2.0.0,>=1.9.0
 Requires-Dist: httpx<0.29.0,>=0.28.1
-Requires-Dist: idun-agent-schema<0.3.0,>=0.2.5
+Requires-Dist: idun-agent-schema<0.3.0,>=0.2.7
 Requires-Dist: langchain-core<0.4.0,>=0.3.72
 Requires-Dist: langchain-google-vertexai<3.0.0,>=2.0.27
 Requires-Dist: langchain<0.4,>=0.3.27

{idun_agent_engine-0.2.5 → idun_agent_engine-0.2.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "idun-agent-engine"
-version = "0.2.5"
+version = "0.2.7"
 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"
@@ -26,7 +26,7 @@ classifiers = [
     "Typing :: Typed",
 ]
 dependencies = [
-    "fastapi>=0.116.1,<0.117.0",
+    "fastapi>=0.115.0,<0.116.0",
     "uvicorn>=0.35.0,<0.36.0",
     "langgraph>=0.6.3,<0.7.0",
     "langgraph-checkpoint-sqlite>=2.0.11,<3.0.0",
@@ -44,12 +44,14 @@ dependencies = [
     "langchain>=0.3.27,<0.4",
     # Pin <12: 12.x currently depends on placeholder jmespath==99.99.99 and fails in Docker
     "arize-phoenix>=11.22.0,<12.0.0",
-    "idun-agent-schema>=0.2.5,<0.3.0",
+    "idun-agent-schema>=0.2.7,<0.3.0",
     "langfuse-haystack>=2.3.0",
     "python-dotenv>=1.1.1",
     "click>=8.2.1",
     # Pin SQLAlchemy for Python 3.13 compatibility (required by Phoenix)
     "sqlalchemy>=2.0.36,<3.0.0",
+    "ag-ui-langgraph>=0.0.20,<0.1.0",
+    "copilotkit>=0.1.72,<0.2.0",
 ]
 
 

{idun_agent_engine-0.2.5 → idun_agent_engine-0.2.7}/src/idun_agent_engine/_version.py

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

{idun_agent_engine-0.2.5 → idun_agent_engine-0.2.7}/src/idun_agent_engine/agent/base.py

@@ -39,6 +39,15 @@ class BaseAgent[ConfigType: BaseAgentConfig](ABC):
         """
         pass
 
+    @property
+    @abstractmethod
+    def copilotkit_agent_instance(self) -> Any:
+        """Get the CopilotKit agent instance.
+
+        This might be set after initialization.
+        """
+        pass
+
     @property
     def configuration(self) -> ConfigType:
         """Return current configuration settings for the agent.

{idun_agent_engine-0.2.5 → idun_agent_engine-0.2.7}/src/idun_agent_engine/agent/haystack/haystack.py

@@ -70,6 +70,15 @@ class HaystackAgent(BaseAgent):
             raise RuntimeError("Agent not initialized. Call initialize() first.")
         return self._agent_instance
 
+    @property
+    def copilotkit_agent_instance(self) -> Any:
+        """Return the CopilotKit agent instance.
+
+        Raises:
+            RuntimeError: If the CopilotKit agent is not yet initialized.
+        """
+        raise NotImplementedError("CopilotKit agent instance not supported yet for Haystack agent.")
+
     @property
     def configuration(self) -> HaystackAgentConfig:
         """Return validated configuration.

{idun_agent_engine-0.2.5 → idun_agent_engine-0.2.7}/src/idun_agent_engine/agent/langgraph/langgraph.py

@@ -17,6 +17,7 @@ from langgraph.graph import StateGraph
 
 from idun_agent_engine import observability
 from idun_agent_engine.agent import base as agent_base
+from copilotkit import LangGraphAGUIAgent
 
 
 class LanggraphAgent(agent_base.BaseAgent):
@@ -29,6 +30,7 @@ class LanggraphAgent(agent_base.BaseAgent):
         self._input_schema: Any = None
         self._output_schema: Any = None
         self._agent_instance: Any = None
+        self._copilotkit_agent_instance: LangGraphAGUIAgent | None = None
         self._checkpointer: Any = None
         self._store: Any = None
         self._connection: Any = None
@@ -79,6 +81,17 @@ class LanggraphAgent(agent_base.BaseAgent):
             raise RuntimeError("Agent not initialized. Call initialize() first.")
         return self._agent_instance
 
+    @property
+    def copilotkit_agent_instance(self) -> LangGraphAGUIAgent:
+        """Return the CopilotKit agent instance.
+
+        Raises:
+            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.")
+        return self._copilotkit_agent_instance
+
     @property
     def configuration(self) -> LangGraphAgentConfig:
         """Return validated configuration.
@@ -158,6 +171,12 @@ class LanggraphAgent(agent_base.BaseAgent):
             checkpointer=self._checkpointer, store=self._store
         )
 
+        self._copilotkit_agent_instance = LangGraphAGUIAgent(
+            name=self._name,
+            description="Agent description", # TODO: add agent description
+            graph=self._agent_instance,
+        )
+
         if self._agent_instance:
             self._input_schema = self._agent_instance.input_schema
             self._output_schema = self._agent_instance.output_schema

{idun_agent_engine-0.2.5 → idun_agent_engine-0.2.7}/src/idun_agent_engine/core/app_factory.py

@@ -14,6 +14,7 @@ from ..server.routers.agent import agent_router
 from ..server.routers.base import base_router
 from .config_builder import ConfigBuilder
 from .engine_config import EngineConfig
+from .._version import __version__
 
 
 def create_app(
@@ -48,7 +49,7 @@ def create_app(
         lifespan=lifespan,
         title="Idun Agent Engine Server",
         description="A production-ready server for conversational AI agents",
-        version="0.1.0",
+        version=__version__,
         docs_url="/docs",
         redoc_url="/redoc",
     )

{idun_agent_engine-0.2.5 → idun_agent_engine-0.2.7}/src/idun_agent_engine/core/config_builder.py

@@ -139,7 +139,7 @@ class ConfigBuilder:
         langgraph_config = LangGraphAgentConfig.model_validate(agent_config_dict)
 
         # Create the agent config (store as strongly-typed model, not dict)
-        self._agent_config = AgentConfig(type="langgraph", config=langgraph_config)
+        self._agent_config = AgentConfig(type=AgentFramework.LANGGRAPH, config=langgraph_config)
         return self
 
     def with_custom_agent(
@@ -160,12 +160,12 @@ class ConfigBuilder:
         """
         if agent_type == AgentFramework.LANGGRAPH:
             self._agent_config = AgentConfig(
-                type="langgraph", config=LangGraphAgentConfig.model_validate(config)
+                type=AgentFramework.LANGGRAPH, config=LangGraphAgentConfig.model_validate(config)
             )
 
         elif agent_type == AgentFramework.HAYSTACK:
             self._agent_config = AgentConfig(
-                type="haystack", config=HaystackAgentConfig.model_validate(config)
+                type=AgentFramework.HAYSTACK, config=HaystackAgentConfig.model_validate(config)
             )
         else:
             raise ValueError(f"Unsupported agent type: {agent_type}")

{idun_agent_engine-0.2.5 → idun_agent_engine-0.2.7}/src/idun_agent_engine/server/dependencies.py

@@ -21,3 +21,20 @@ async def get_agent(request: Request):
         app_config = ConfigBuilder.load_from_file()
         agent = await ConfigBuilder.initialize_agent_from_config(app_config)
         return agent
+
+async def get_copilotkit_agent(request: Request):
+    """Return the pre-initialized agent instance from the app state.
+
+    Falls back to loading from the default config if not present (e.g., tests).
+    """
+    if hasattr(request.app.state, "copilotkit_agent"):
+        return request.app.state.copilotkit_agent
+    else:
+        # This is a fallback for cases where the lifespan event did not run,
+        # like in some testing scenarios.
+        # Consider logging a warning here.
+        print("⚠️  CopilotKit agent not found in app state, initializing fallback agent...")
+
+        app_config = ConfigBuilder.load_from_file()
+        copilotkit_agent = await ConfigBuilder.initialize_agent_from_config(app_config)
+        return copilotkit_agent

{idun_agent_engine-0.2.5 → idun_agent_engine-0.2.7}/src/idun_agent_engine/server/routers/agent.py

@@ -8,7 +8,11 @@ from fastapi.responses import StreamingResponse
 from idun_agent_schema.engine.api import ChatRequest, ChatResponse
 
 from idun_agent_engine.agent.base import BaseAgent
-from idun_agent_engine.server.dependencies import get_agent
+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
 
 logging.basicConfig(
     format="%(asctime)s %(levelname)-8s %(message)s",
@@ -57,7 +61,6 @@ async def stream(
 ):
     """Process a message with the agent, streaming ag-ui events."""
     try:
-
         async def event_stream():
             message = {"query": request.query, "session_id": request.session_id}
             async for event in agent.stream(message):
@@ -66,3 +69,28 @@ async def stream(
         return StreamingResponse(event_stream(), media_type="text/event-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,
+    request: Request,
+    copilotkit_agent: Annotated[LangGraphAGUIAgent, Depends(get_copilotkit_agent)],
+):
+    """Process a message with the agent, streaming ag-ui events."""
+    try:
+        # Get the accept header from the request
+        accept_header = request.headers.get("accept")
+
+        # Create an event encoder to properly format SSE events
+        encoder = EventEncoder(accept=accept_header or "") # type: ignore[arg-type]
+
+        async def event_generator():
+            async for event in copilotkit_agent.run(input_data):
+                yield encoder.encode(event)
+
+        return StreamingResponse(
+            event_generator(),  # type: ignore[arg-type]
+            media_type=encoder.get_content_type()
+        )
+    except Exception as e:  # noqa: BLE001
+        raise HTTPException(status_code=500, detail=str(e)) from e

{idun_agent_engine-0.2.5 → idun_agent_engine-0.2.7}/src/idun_platform_cli/groups/agent/package.py

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