copilotkit 0.1.75a6__py3-none-any.whl → 0.1.75a8__py3-none-any.whl

copilotkit/copilotkit_lg_middleware.py

@@ -25,30 +25,19 @@ from langchain.agents.middleware import (
 )
 from langgraph.runtime import Runtime
 
-from .langgraph import CopilotContextItem
-
-class CopilotKitState:
-    """Extended state schema for CopilotKit middleware."""
-
-    # CopilotKit frontend tools passed via state
-    actions: List[Any]
-    context: List[CopilotContextItem]
-
-    # Private state for CopilotKit middleware
-    intercepted_tool_calls: Any
-    original_ai_message_id: Any
+from .langgraph import CopilotKitProperties
 
 class StateSchema(AgentState):
-    copilotkit: CopilotKitState
+    copilotkit: CopilotKitProperties
 
 
-class CopilotKitMiddleware(AgentMiddleware[CopilotKitState, Any]):
+class CopilotKitMiddleware(AgentMiddleware[StateSchema, Any]):
     """CopilotKit Middleware for LangGraph agents.
 
     Handles frontend tool injection and interception for CopilotKit.
     """
 
-    state_schema = CopilotKitState
+    state_schema = StateSchema
     tools: ClassVar[list] = []
 
     @property
@@ -77,7 +66,6 @@ class CopilotKitMiddleware(AgentMiddleware[CopilotKitState, Any]):
             handler: Callable[[ModelRequest], Awaitable[ModelResponse]],
     ) -> ModelResponse:
         frontend_tools = request.state.get("copilotkit", {}).get("actions", [])
-        print(request.state)
 
         if not frontend_tools:
             return await handler(request)
@@ -90,10 +78,10 @@ class CopilotKitMiddleware(AgentMiddleware[CopilotKitState, Any]):
     # Intercept frontend tool calls after model returns, before ToolNode executes
     def after_model(
             self,
-            state: CopilotKitState,
+            state: StateSchema,
             runtime: Runtime[Any],
     ) -> dict[str, Any] | None:
-        frontend_tools = state.get("copilotkit", {}).get("tools", [])
+        frontend_tools = state.get("copilotkit", {}).get("actions", [])
         if not frontend_tools:
             return None
 
@@ -144,7 +132,7 @@ class CopilotKitMiddleware(AgentMiddleware[CopilotKitState, Any]):
 
     async def aafter_model(
             self,
-            state: CopilotKitState,
+            state: StateSchema,
             runtime: Runtime[Any],
     ) -> dict[str, Any] | None:
         # Delegate to sync implementation
@@ -153,7 +141,7 @@ class CopilotKitMiddleware(AgentMiddleware[CopilotKitState, Any]):
     # Restore frontend tool calls to AIMessage before agent exits
     def after_agent(
             self,
-            state: CopilotKitState,
+            state: StateSchema,
             runtime: Runtime[Any],
     ) -> dict[str, Any] | None:
         copilotkit_state = state.get("copilotkit", {})
@@ -187,7 +175,7 @@ class CopilotKitMiddleware(AgentMiddleware[CopilotKitState, Any]):
 
     async def aafter_agent(
             self,
-            state: CopilotKitState,
+            state: StateSchema,
             runtime: Runtime[Any],
     ) -> dict[str, Any] | None:
         # Delegate to sync implementation

copilotkit/langgraph.py

@@ -37,6 +37,10 @@ class CopilotKitProperties(TypedDict):
     actions: List[Any]
     context: List[CopilotContextItem]
 
+    # Private state for CopilotKit middleware
+    intercepted_tool_calls: Any
+    original_ai_message_id: Any
+
 class CopilotKitState(MessagesState):
     """CopilotKit state"""
     copilotkit: CopilotKitProperties

{copilotkit-0.1.75a6.dist-info → copilotkit-0.1.75a8.dist-info}/METADATA

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

{copilotkit-0.1.75a6.dist-info → copilotkit-0.1.75a8.dist-info}/RECORD

@@ -1,7 +1,7 @@
 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=ya4ZDu7L8zKp53K-KxEhoWFcbFg8ZCKcvkQETS_-oMs,5866
+copilotkit/copilotkit_lg_middleware.py,sha256=7MIiMfdKp9TuwmL0k7dIvTYYob6InxQbOUqICowYRDs,5517
 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
@@ -11,7 +11,7 @@ copilotkit/html.py,sha256=IHnY4cx89LuYuPPf8Tady2DXfr3MyRWilMHpZ7d8bFk,4357
 copilotkit/integrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 copilotkit/integrations/fastapi.py,sha256=3DAAFbpFqL0cJ22sE6JDz3HbTjYerDqn4ZJhYhCvwZs,10487
 copilotkit/langchain.py,sha256=QFIHnhhAzCKO16F6aFYh_gsCEfm-eaCUPZm-yMTczxU,752
-copilotkit/langgraph.py,sha256=Y8VSZ-0vCijppf_0PrHtyT24wC6L16gETbsKwVX4F6g,14992
+copilotkit/langgraph.py,sha256=wfAqp1P2KfMQw5X1AL2BJtXrQBgT02U3KoJh54--uXA,15103
 copilotkit/langgraph_agent.py,sha256=WPg45s3ITOU2exgE_sW1u-rgnYY4WherAzTVLscBkBk,29668
 copilotkit/langgraph_agui_agent.py,sha256=t6Vpdiod6-iIGRI303-ipN6FfakpB5So2gaYKYZ6zlY,7798
 copilotkit/logging.py,sha256=lpPEZbI6LMaQLzSgiQMYV4qHrNk-MuO2qZ3zcaHl1nA,499
@@ -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.75a6.dist-info/METADATA,sha256=tPZS0ECIe5RrS5lZk4N8JfLjew6iEKX2vfWrpRfrQ2M,2606
-copilotkit-0.1.75a6.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-copilotkit-0.1.75a6.dist-info/RECORD,,
+copilotkit-0.1.75a8.dist-info/METADATA,sha256=3NGADg3OXLOLV_IcWmRzTaVrlNqI8wbq1UTqg3bEKq0,2606
+copilotkit-0.1.75a8.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+copilotkit-0.1.75a8.dist-info/RECORD,,