copilotkit 0.1.75a6__tar.gz → 0.1.75a8__tar.gz

{copilotkit-0.1.75a6 → copilotkit-0.1.75a8}/PKG-INFO

@@ -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 → copilotkit-0.1.75a8}/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-0.1.75a6 → copilotkit-0.1.75a8}/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 → copilotkit-0.1.75a8}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "copilotkit"
-version = "0.1.75-alpha.6"
+version = "0.1.75-alpha.8"
 description = "CopilotKit python SDK"
 authors = ["Markus Ecker <markus.ecker@gmail.com>"]
 license = "MIT"