copilotkit 0.1.75a4__tar.gz → 0.1.75a5__tar.gz

{copilotkit-0.1.75a4 → copilotkit-0.1.75a5}/PKG-INFO

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

{copilotkit-0.1.75a4 → copilotkit-0.1.75a5}/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-0.1.75a4 → copilotkit-0.1.75a5}/copilotkit/copilotkit_lg_middleware.py

@@ -28,8 +28,7 @@ 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
@@ -39,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.
@@ -59,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)
@@ -74,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)
@@ -190,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.75a4 → copilotkit-0.1.75a5}/pyproject.toml

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