copilotkit 0.1.75a0__py3-none-any.whl → 0.1.75a1__py3-none-any.whl

copilotkit/copilotkit_lg_middleware.py

@@ -1,23 +1,3 @@
-"""
-AG-UI Middleware for LangGraph agents.
-
-Enables:
-- Dynamic frontend tools from state['ag-ui']['tools']
-- Frontend tool routing (intercept calls, skip backend execution)
-
-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, Annotated
 from typing_extensions import TypedDict, NotRequired
 
@@ -32,9 +12,9 @@ from langgraph.runtime import Runtime
 
 
 class CopilotKitState(AgentState):
-    """Extended state schema for AG-UI middleware."""
+    """Extended state schema for CopilotKit middleware."""
 
-    # AG-UI frontend tools passed via state
+    # CopilotKit frontend tools passed via state
     actions: List[Any]
     context: List[CopilotContextItem]
 
@@ -42,13 +22,13 @@ class CopilotKitState(AgentState):
     copilotkit: NotRequired[dict[str, Any]]
 
 
-class CopilotKitMiddleware(AgentMiddleware[AGUIState, Any]):
-    """AG-UI Middleware for LangGraph agents.
+class CopilotKitMiddleware(AgentMiddleware[CopilotKitState, Any]):
+    """CopilotKit Middleware for LangGraph agents.
 
-    Handles frontend tool injection and interception for AG-UI protocol.
+    Handles frontend tool injection and interception for CopilotKit.
     """
 
-    state_schema = AGUIState
+    state_schema = CopilotKitState
     tools = []
 
     @property
@@ -61,7 +41,7 @@ class CopilotKitMiddleware(AgentMiddleware[AGUIState, Any]):
             request: ModelRequest,
             handler: Callable[[ModelRequest], ModelResponse],
     ) -> ModelResponse:
-        frontend_tools = request.state.get("ag-ui", {}).get("tools", [])
+        frontend_tools = request.state.get("copilotkit", {}).get("tools", [])
 
         if not frontend_tools:
             return handler(request)
@@ -76,7 +56,7 @@ class CopilotKitMiddleware(AgentMiddleware[AGUIState, Any]):
             request: ModelRequest,
             handler: Callable[[ModelRequest], Awaitable[ModelResponse]],
     ) -> ModelResponse:
-        frontend_tools = request.state.get("ag-ui", {}).get("tools", [])
+        frontend_tools = request.state.get("copilotkit", {}).get("tools", [])
 
         if not frontend_tools:
             return await handler(request)
@@ -89,10 +69,10 @@ class CopilotKitMiddleware(AgentMiddleware[AGUIState, Any]):
     # Intercept frontend tool calls after model returns, before ToolNode executes
     def after_model(
             self,
-            state: AGUIState,
+            state: CopilotKitState,
             runtime: Runtime[Any],
     ) -> dict[str, Any] | None:
-        frontend_tools = state.get("ag-ui", {}).get("tools", [])
+        frontend_tools = state.get("copilotkit", {}).get("tools", [])
         if not frontend_tools:
             return None
 
@@ -143,7 +123,7 @@ class CopilotKitMiddleware(AgentMiddleware[AGUIState, Any]):
 
     async def aafter_model(
             self,
-            state: AGUIState,
+            state: CopilotKitState,
             runtime: Runtime[Any],
     ) -> dict[str, Any] | None:
         # Delegate to sync implementation
@@ -152,7 +132,7 @@ class CopilotKitMiddleware(AgentMiddleware[AGUIState, Any]):
     # Restore frontend tool calls to AIMessage before agent exits
     def after_agent(
             self,
-            state: AGUIState,
+            state: CopilotKitState,
             runtime: Runtime[Any],
     ) -> dict[str, Any] | None:
         copilotkit_state = state.get("copilotkit", {})
@@ -186,7 +166,7 @@ class CopilotKitMiddleware(AgentMiddleware[AGUIState, Any]):
 
     async def aafter_agent(
             self,
-            state: AGUIState,
+            state: CopilotKitState,
             runtime: Runtime[Any],
     ) -> dict[str, Any] | None:
         # Delegate to sync implementation

{copilotkit-0.1.75a0.dist-info → copilotkit-0.1.75a1.dist-info}/METADATA

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

{copilotkit-0.1.75a0.dist-info → copilotkit-0.1.75a1.dist-info}/RECORD

@@ -1,7 +1,7 @@
 copilotkit/__init__.py,sha256=pK1UmbqWsd9blnIZ0p3wYHjkLthAavI2-gCyI5ulJzE,861
 copilotkit/action.py,sha256=bg_bPusEfN8TMQV1B3lVyMLbHyLn1E7yVYAvfE2_PXA,1691
 copilotkit/agent.py,sha256=e-jPsUQo18FOYUPdE3YNmSLcb5FEfWL6ARScOntyvb8,1663
-copilotkit/copilotkit_lg_middleware.py,sha256=_G71STMN6Kqi-b_VZIeY6XX189GQpD7GjhI5AnFMwv8,5900
+copilotkit/copilotkit_lg_middleware.py,sha256=CHPtSJnc_wEA-pVsq-TwNexA3K8x87dCEORADGxCiSY,5497
 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.75a0.dist-info/METADATA,sha256=8lYI9IdjJPpLdB3AWOtj5zPgYMEe89a2zp1-amB7l2c,2606
-copilotkit-0.1.75a0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-copilotkit-0.1.75a0.dist-info/RECORD,,
+copilotkit-0.1.75a1.dist-info/METADATA,sha256=N8BPWhy0GIjqbXa5JZF7nSEuJRTDHwDsuVGMrgex3dU,2606
+copilotkit-0.1.75a1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+copilotkit-0.1.75a1.dist-info/RECORD,,