idun-agent-engine 0.2.5__py3-none-any.whl → 0.2.7__py3-none-any.whl

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/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/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/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/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/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/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/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.dist-info → idun_agent_engine-0.2.7.dist-info}/METADATA

@@ -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.dist-info → idun_agent_engine-0.2.7.dist-info}/RECORD

@@ -1,18 +1,18 @@
 idun_agent_engine/__init__.py,sha256=PhOL6foq5V0eXaoXw7xKUeCWXIWrOHrAFB8OuJnBqyM,550
-idun_agent_engine/_version.py,sha256=I5-7-uhpm_19BECbYd0Vlz1Feg_ET0sWIZldIalT69c,72
+idun_agent_engine/_version.py,sha256=o2wtvcZFXHMC9HdGCpCMglFAwq-cYBJIAi_MTK4BIFo,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=xzuHIV_P7EwGyK7V2qbFUZccpm5xHaAjAkIxJL4tAwo,2856
+idun_agent_engine/agent/base.py,sha256=JT_U4tgC6yPi0IYbDjSQ2a6W2YwABiC9NSGjYPCe8Yw,3059
 idun_agent_engine/agent/haystack/__init__.py,sha256=y5ADrD8gWBeYIvV7tmu6OpPdJ8POHt-tyraIL7RkkWI,179
-idun_agent_engine/agent/haystack/haystack.py,sha256=6xq7tOlvQxyc2w9KxikpS5S6vmXjUlK9Giwhmzg2X0w,10556
+idun_agent_engine/agent/haystack/haystack.py,sha256=6WHW9z9LqMa5hHkfxUEaxyMMFw7y9nUu9wnXavxdsg8,10872
 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=Fm6t-TeG_7s0tJHev5sIbKcD7gLLXv22pJH0S14Hp2k,17412
+idun_agent_engine/agent/langgraph/langgraph.py,sha256=eawnV6XgC9gxplTogyKioeqnHvdBwCtwobA8tLYmaYk,18167
 idun_agent_engine/cli/__init__.py,sha256=5S_Oo7n7YwKk5VPyqborrRqKVz6lvwODQ5olj70wbnQ,490
 idun_agent_engine/core/__init__.py,sha256=F0DMDlWcSWS_1dvh3xMbrdcVvZRHVnoAFFgREuSJfBI,408
-idun_agent_engine/core/app_factory.py,sha256=fqxX6n4huCAsEY5lzPQPtBb7nWWaxzvhYaPsRUbZuEY,2333
-idun_agent_engine/core/config_builder.py,sha256=nqlFEASHN9zlZhUnu5NAhRo2DeFkFVQ0rpx-yfnBuIw,16283
+idun_agent_engine/core/app_factory.py,sha256=Af57kuEic6g4RA3JyLr9-gl_MQh41xcpHGA_wx92cO8,2372
+idun_agent_engine/core/config_builder.py,sha256=lLyAvnatPSWR5PqTj4sWPNxnMBKgXdtFuxzqAYzhxcY,16322
 idun_agent_engine/core/engine_config.py,sha256=pHa-qiYUS7mvzW2eb6sXJoxxJCxC5fwcNUTcqiO90c0,584
 idun_agent_engine/core/server_runner.py,sha256=01_aAJC-s0C5fdkEUQzwk4Vlf1SX4lsZwRTkVaJx0VE,4898
 idun_agent_engine/observability/__init__.py,sha256=Jei-E8SgYulxFu5HkJz9HfcorIfQ6O7WZWGzP41ZK-4,298
@@ -24,20 +24,20 @@ idun_agent_engine/observability/phoenix/phoenix_handler.py,sha256=lGqSq-L1vmoEhA
 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=7yGYWxqbL4xNafpj8g8-S-TQ_GmGlPnSB_wRbW_lfMA,824
+idun_agent_engine/server/dependencies.py,sha256=3b9ADxZWMqDsye2yOPsZ_uPlj1hHynN5vnv0sW2SLkI,1580
 idun_agent_engine/server/lifespan.py,sha256=BC914qErOmKLNZY0-XkMkYnA0aVhLZFY7nhpKXx2D_g,1301
 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=7vVzhnKXYv6yZEKP8o-ETMWhIa0Hd2qyMu63MhoiLL0,2361
+idun_agent_engine/server/routers/agent.py,sha256=PIdSZjrrskvaLRYksE7qWIyfrOwvRCSiPr07xTt4XiA,3456
 idun_agent_engine/server/routers/base.py,sha256=BWueBPN7ecdNWOyQaxpdpnd6GxOcN78N8bFWN_aNgmU,1899
 idun_platform_cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 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=5VuEHSkcAqTbNQjwRy3tBz1sbm9kaJHqIhshFhRc19Q,2525
+idun_platform_cli/groups/agent/package.py,sha256=gc7HoA2wtHlFC9f7rDXgGEBZUvibEwh9Vmlr-C1C1uA,2525
 idun_platform_cli/groups/agent/serve.py,sha256=IHC3vp_R45RW4enkNQCE-SP_J9GkmapAycktKcxT9Y8,3800
-idun_agent_engine-0.2.5.dist-info/METADATA,sha256=3xUtr_6v8iNIg2vSd5-TTEhAYIAes5sCkV03KD5nyNo,8660
-idun_agent_engine-0.2.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-idun_agent_engine-0.2.5.dist-info/entry_points.txt,sha256=XG3oxlSOaCrYKT1oyhKa0Ag1iJPMZ-WF6gaV_mzIJW4,52
-idun_agent_engine-0.2.5.dist-info/RECORD,,
+idun_agent_engine-0.2.7.dist-info/METADATA,sha256=Zg4GQA7qBy5Zl1FgrFHAbVYTrUXCL9GgwSK4zKtFnE0,8747
+idun_agent_engine-0.2.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+idun_agent_engine-0.2.7.dist-info/entry_points.txt,sha256=XG3oxlSOaCrYKT1oyhKa0Ag1iJPMZ-WF6gaV_mzIJW4,52
+idun_agent_engine-0.2.7.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.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