copilotkit 0.1.75a4__tar.gz → 0.1.75a6__tar.gz

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

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

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

@@ -15,7 +15,6 @@ Example:
 """
 
 from typing import Any, Callable, Awaitable, ClassVar, List
-from typing_extensions import NotRequired
 
 from langchain_core.messages import AIMessage
 from langchain.agents.middleware import (
@@ -28,8 +27,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
@@ -37,7 +35,11 @@ class CopilotKitState(AgentState):
     context: List[CopilotContextItem]
 
     # Private state for CopilotKit middleware
-    copilotkit: NotRequired[dict[str, Any]]
+    intercepted_tool_calls: Any
+    original_ai_message_id: Any
+
+class StateSchema(AgentState):
+    copilotkit: CopilotKitState
 
 
 class CopilotKitMiddleware(AgentMiddleware[CopilotKitState, Any]):
@@ -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,8 @@ 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", [])
+        print(request.state)
 
         if not frontend_tools:
             return await handler(request)
@@ -190,6 +193,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.75a6}/pyproject.toml

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