copilotkit 0.1.75a3__tar.gz → 0.1.75a5__tar.gz

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

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

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

@@ -1,5 +1,21 @@
-from typing import Any, Callable, Awaitable, Annotated
-from typing_extensions import TypedDict, NotRequired
+"""
+CopilotKit Middleware for LangGraph agents.
+
+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, ClassVar, List
+from typing_extensions import NotRequired
 
 from langchain_core.messages import AIMessage
 from langchain.agents.middleware import (
@@ -10,8 +26,9 @@ from langchain.agents.middleware import (
 )
 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
@@ -21,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.
@@ -29,7 +49,7 @@ class CopilotKitMiddleware(AgentMiddleware[CopilotKitState, Any]):
     """
 
     state_schema = CopilotKitState
-    tools = []
+    tools: ClassVar[list] = []
 
     @property
     def name(self) -> str:
@@ -41,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)
@@ -56,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)
@@ -172,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.75a3 → copilotkit-0.1.75a5}/pyproject.toml

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