copilotkit 0.1.75a3__py3-none-any.whl → 0.1.75a5__py3-none-any.whl

copilotkit/__init__.py

@@ -6,7 +6,7 @@ from .parameter import Parameter
 from .agent import Agent
 from .langgraph_agent import LangGraphAgent
 from .langgraph_agui_agent import LangGraphAGUIAgent
-from .copilotkit_lg_middleware import copilotkit_middleware
+from .copilotkit_lg_middleware import CopilotKitMiddleware
 
 
 
@@ -22,5 +22,5 @@ __all__ = [
     'CrewAIAgent', # pyright: ignore[reportUnsupportedDunderAll] pylint: disable=undefined-all-variable
     'LangGraphAgent', # pyright: ignore[reportUnsupportedDunderAll] pylint: disable=undefined-all-variable
     "LangGraphAGUIAgent",
-    "copilotkit_middleware"
+    "CopilotKitMiddleware"
 ]

copilotkit/copilotkit_lg_middleware.py

@@ -1,5 +1,21 @@
-from typing import Any, Callable, Awaitable, Annotated
-from typing_extensions import TypedDict, NotRequired
+"""
+CopilotKit Middleware for LangGraph agents.
+
+Works with any agent (prebuilt or custom).
+
+Example:
+    from langchain.agents import create_agent
+    from copilotkit import CopilotKitMiddleware
+
+    agent = create_agent(
+        model="openai:gpt-4o",
+        tools=[backend_tool],
+        middleware=[CopilotKitMiddleware()],
+    )
+"""
+
+from typing import Any, Callable, Awaitable, ClassVar, List
+from typing_extensions import NotRequired
 
 from langchain_core.messages import AIMessage
 from langchain.agents.middleware import (
@@ -10,8 +26,9 @@ from langchain.agents.middleware import (
 )
 from langgraph.runtime import Runtime
 
+from .langgraph import CopilotContextItem
 
-class CopilotKitState(AgentState):
+class CopilotKitState:
     """Extended state schema for CopilotKit middleware."""
 
     # CopilotKit frontend tools passed via state
@@ -21,6 +38,9 @@ class CopilotKitState(AgentState):
     # Private state for CopilotKit middleware
     copilotkit: NotRequired[dict[str, Any]]
 
+class StateSchema(AgentState):
+    copilotkit: CopilotKitState
+
 
 class CopilotKitMiddleware(AgentMiddleware[CopilotKitState, Any]):
     """CopilotKit Middleware for LangGraph agents.
@@ -29,7 +49,7 @@ class CopilotKitMiddleware(AgentMiddleware[CopilotKitState, Any]):
     """
 
     state_schema = CopilotKitState
-    tools = []
+    tools: ClassVar[list] = []
 
     @property
     def name(self) -> str:
@@ -41,7 +61,7 @@ class CopilotKitMiddleware(AgentMiddleware[CopilotKitState, Any]):
             request: ModelRequest,
             handler: Callable[[ModelRequest], ModelResponse],
     ) -> ModelResponse:
-        frontend_tools = request.state.get("copilotkit", {}).get("tools", [])
+        frontend_tools = request.state.get("copilotkit", {}).get("actions", [])
 
         if not frontend_tools:
             return handler(request)
@@ -56,7 +76,7 @@ class CopilotKitMiddleware(AgentMiddleware[CopilotKitState, Any]):
             request: ModelRequest,
             handler: Callable[[ModelRequest], Awaitable[ModelResponse]],
     ) -> ModelResponse:
-        frontend_tools = request.state.get("copilotkit", {}).get("tools", [])
+        frontend_tools = request.state.get("copilotkit", {}).get("actions", [])
 
         if not frontend_tools:
             return await handler(request)
@@ -172,6 +192,3 @@ class CopilotKitMiddleware(AgentMiddleware[CopilotKitState, Any]):
         # Delegate to sync implementation
         return self.after_agent(state, runtime)
 
-
-# Pre-created instance for convenience
-copilotkit_middleware = CopilotKitMiddleware()

{copilotkit-0.1.75a3.dist-info → copilotkit-0.1.75a5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: copilotkit
-Version: 0.1.75a3
+Version: 0.1.75a5
 Summary: CopilotKit python SDK
 License: MIT
 Keywords: copilot,copilotkit,langgraph,langchain,ai,langsmith,langserve

{copilotkit-0.1.75a3.dist-info → copilotkit-0.1.75a5.dist-info}/RECORD

@@ -1,7 +1,7 @@
-copilotkit/__init__.py,sha256=pK1UmbqWsd9blnIZ0p3wYHjkLthAavI2-gCyI5ulJzE,861
+copilotkit/__init__.py,sha256=22bUXudklMop9r1DYkecCaTiHbj_m0Wf4txOU-yfSI4,859
 copilotkit/action.py,sha256=bg_bPusEfN8TMQV1B3lVyMLbHyLn1E7yVYAvfE2_PXA,1691
 copilotkit/agent.py,sha256=e-jPsUQo18FOYUPdE3YNmSLcb5FEfWL6ARScOntyvb8,1663
-copilotkit/copilotkit_lg_middleware.py,sha256=CHPtSJnc_wEA-pVsq-TwNexA3K8x87dCEORADGxCiSY,5497
+copilotkit/copilotkit_lg_middleware.py,sha256=LlLlRxX9ws8JDRJAALPnRJVi1-0nBEcRBeO-w0WuixE,5859
 copilotkit/crewai/__init__.py,sha256=RAkBTwDpFuvD5prULUnoWdl5d8W373cm1ZlqOMeSGbY,1034
 copilotkit/crewai/copilotkit_integration.py,sha256=vd1LVhXo-cpA4PQ_-cZHdKgm67MvrHpx4VJweUvCzPE,14550
 copilotkit/crewai/crewai_agent.py,sha256=maoMnl0WIHUnFAK5ALUKD9z0GvjtfNGi31zi37LyyaE,15323
@@ -21,6 +21,6 @@ copilotkit/runloop.py,sha256=9pBqWeu4QCRPOizz00se1d0_BvCX3ZIABzNeiXhdyNM,10641
 copilotkit/sdk.py,sha256=VEQqA9QAXJS5MDLxtJJdtfrApX9EawCHzp-IXnat5vw,11882
 copilotkit/types.py,sha256=_YQluiHpJGGXYB4eWLk0EocDZp5C8rs3ZsMNFhmXEYU,972
 copilotkit/utils.py,sha256=rOAaaFBXgb3iv_cLH9S2steyGwHmnTjaxQTA53qdtQo,203
-copilotkit-0.1.75a3.dist-info/METADATA,sha256=1wvJmalBXXfUWXb0rNTpsKng2yfVTC1b3wR_sm2vQbA,2606
-copilotkit-0.1.75a3.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-copilotkit-0.1.75a3.dist-info/RECORD,,
+copilotkit-0.1.75a5.dist-info/METADATA,sha256=mCEps29kaADNPa1eQwgUtfhn1UpO0L-eY4Mgly9J2VU,2606
+copilotkit-0.1.75a5.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+copilotkit-0.1.75a5.dist-info/RECORD,,