copilotkit 0.1.87__tar.gz → 0.1.88__tar.gz

{copilotkit-0.1.87 → copilotkit-0.1.88}/PKG-INFO

@@ -1,25 +1,28 @@
 Metadata-Version: 2.3
 Name: copilotkit
-Version: 0.1.87
+Version: 0.1.88
 Summary: CopilotKit python SDK
 License: MIT
 Keywords: copilot,copilotkit,langgraph,langchain,ai,langsmith,langserve
 Author: Markus Ecker
 Author-email: markus.ecker@gmail.com
-Requires-Python: >=3.10,<3.13
+Requires-Python: >=3.10,<3.15
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Provides-Extra: crewai
-Requires-Dist: ag-ui-langgraph[fastapi] (>=0.0.29)
-Requires-Dist: ag-ui-protocol (>=0.1.11)
-Requires-Dist: crewai (==0.118.0) ; extra == "crewai"
+Requires-Dist: ag-ui-langgraph[fastapi] (>=0.0.35)
+Requires-Dist: ag-ui-protocol (>=0.1.15)
+Requires-Dist: crewai (>=0.118.0) ; (python_version >= "3.10" and python_version < "3.14") and (extra == "crewai")
 Requires-Dist: fastapi (>=0.115.0,<1.0.0)
 Requires-Dist: langchain (>=0.3.0)
 Requires-Dist: langgraph (>=0.3.25,<2)
 Requires-Dist: partialjson (>=0.0.8,<0.0.9)
+Requires-Dist: pydantic-core (>=2.35.0)
 Requires-Dist: toml (>=0.10.2,<0.11.0)
 Project-URL: Homepage, https://copilotkit.ai
 Project-URL: Repository, https://github.com/CopilotKit/CopilotKit/tree/main/sdk-python

{copilotkit-0.1.87 → copilotkit-0.1.88}/copilotkit/__init__.py

@@ -4,7 +4,6 @@ from .action import Action
 from .langgraph import CopilotKitState
 from .parameter import Parameter
 from .agent import Agent
-from .langgraph_agent import LangGraphAgent
 from .langgraph_agui_agent import LangGraphAGUIAgent
 from .copilotkit_lg_middleware import CopilotKitMiddleware
 from ag_ui_langgraph.middlewares.state_streaming import StateStreamingMiddleware, StateItem
@@ -21,7 +20,6 @@ __all__ = [
     'CopilotKitContext',
     'CopilotKitSDKContext',
     'CrewAIAgent', # pyright: ignore[reportUnsupportedDunderAll] pylint: disable=undefined-all-variable
-    'LangGraphAgent', # pyright: ignore[reportUnsupportedDunderAll] pylint: disable=undefined-all-variable
     "LangGraphAGUIAgent",
     "CopilotKitMiddleware",
     "StateStreamingMiddleware",

{copilotkit-0.1.87 → copilotkit-0.1.88}/copilotkit/a2ui.py

@@ -117,6 +117,11 @@ CRITICAL: You MUST call the render_a2ui tool with ALL of these arguments:
 - every component must have the "component" field specifying the component type (e.g. "Text", "Image", "Row", "Column", "List", "Button", etc.)
 
 COMPONENT ID RULES:
+- Exactly one component MUST have id="root". This is the surface's entry
+  point — the renderer begins at "root" and walks the child/children tree
+  from there. Every other component must be reachable from "root". If no
+  component has id="root", the surface renders an empty loading placeholder
+  and none of your components will be shown.
 - Every component ID must be unique within the surface.
 - A component MUST NOT reference itself as child/children. This causes a
   circular dependency error. For example, if a component has id="avatar",

{copilotkit-0.1.87 → copilotkit-0.1.88}/copilotkit/copilotkit_lg_middleware.py

@@ -16,7 +16,7 @@ Example:
 
 import json
 import re
-from typing import Any, Callable, Awaitable, ClassVar, List
+from typing import Any, Callable, Awaitable, ClassVar, Iterable, List, Union
 
 from langchain_core.messages import AIMessage, SystemMessage, ToolMessage
 from langchain.agents.middleware import (
@@ -33,25 +33,118 @@ class StateSchema(AgentState):
     copilotkit: CopilotKitProperties
 
 
+# Internal/framework keys that should never be surfaced to the LLM as
+# user-facing state. These are either reducer-managed message buckets,
+# CopilotKit/AG-UI plumbing, or graph-internal scaffolding.
+_RESERVED_STATE_KEYS = frozenset({
+    "messages",
+    "copilotkit",
+    "ag-ui",
+    "tools",
+    "structured_response",
+    "thread_id",
+    "remaining_steps",
+})
+
+
 class CopilotKitMiddleware(AgentMiddleware[StateSchema, Any]):
     """CopilotKit Middleware for LangGraph agents.
 
-    Handles frontend tool injection and interception for CopilotKit.
+    Handles frontend tool injection, interception for CopilotKit, and
+    automatic exposure of agent state to the LLM so values written via
+    ``agent.setState`` on the frontend (or via ``Command(update=...)`` in a
+    tool) are visible in the next model call without needing a custom
+    ``get_state`` tool.
+
+    Args:
+        expose_state: Controls how user-defined state keys are surfaced into
+            ``request.system_message`` on every model call. Off by default
+            to avoid leaking arbitrary state into prompts; opt in explicitly.
+
+            - ``False`` (default) — never surface state.
+            - ``True`` — every state key that is not in the reserved
+              internal set and does not start with an underscore is
+              JSON-serialized into a "Current agent state:" note appended
+              to the system message.
+            - ``list``/``tuple``/``set[str]`` — only surface the named keys.
+              Use this when you want explicit control over what the LLM
+              sees (e.g. ``["liked", "todos"]``).
     """
 
     state_schema = StateSchema
     tools: ClassVar[list] = []
 
+    def __init__(
+        self,
+        *,
+        expose_state: Union[bool, Iterable[str]] = False,
+    ):
+        super().__init__()
+        if isinstance(expose_state, bool):
+            self._expose_state: Union[bool, frozenset[str]] = expose_state
+        else:
+            self._expose_state = frozenset(expose_state)
+
     @property
     def name(self) -> str:
         return "CopilotKitMiddleware"
 
-    # Inject frontend tools before model call
+    # ------------------------------------------------------------------
+    # State-to-prompt surfacing
+    # ------------------------------------------------------------------
+
+    def _build_state_note(self, state: dict) -> str | None:
+        """Serialize a snapshot of user state into a system-prompt note.
+
+        Returns ``None`` when nothing should be appended (feature disabled
+        or no non-empty user keys present).
+        """
+        if self._expose_state is False:
+            return None
+        if isinstance(self._expose_state, frozenset):
+            keys: list[str] = [k for k in self._expose_state if k in state]
+        else:
+            keys = [
+                k for k in state
+                if k not in _RESERVED_STATE_KEYS and not str(k).startswith("_")
+            ]
+
+        snapshot: dict[str, Any] = {}
+        for k in keys:
+            v = state.get(k)
+            # Skip empty / no-op values to keep the note tight.
+            if v in (None, "", [], {}):
+                continue
+            snapshot[k] = v
+
+        if not snapshot:
+            return None
+
+        try:
+            body = json.dumps(snapshot, default=str, ensure_ascii=False, indent=2)
+        except (TypeError, ValueError):
+            body = str(snapshot)
+        return f"Current agent state:\n{body}"
+
+    def _apply_state_note(self, request: ModelRequest) -> ModelRequest:
+        note = self._build_state_note(request.state or {})
+        if not note:
+            return request
+        existing = request.system_message
+        if existing is None:
+            return request.override(system_message=SystemMessage(content=note))
+        base = existing.content if isinstance(existing.content, str) else str(existing.content)
+        return request.override(
+            system_message=SystemMessage(content=f"{base}\n\n{note}")
+        )
+
+    # Inject frontend tools and surface user state before model call
     def wrap_model_call(
             self,
             request: ModelRequest,
             handler: Callable[[ModelRequest], ModelResponse],
     ) -> ModelResponse:
+        request = self._apply_state_note(request)
         frontend_tools = request.state.get("copilotkit", {}).get("actions", [])
 
         if not frontend_tools:
@@ -224,6 +317,7 @@ class CopilotKitMiddleware(AgentMiddleware[StateSchema, Any]):
             handler: Callable[[ModelRequest], Awaitable[ModelResponse]],
     ) -> ModelResponse:
         self._fix_messages_for_bedrock(request.messages)
+        request = self._apply_state_note(request)
 
         frontend_tools = request.state.get("copilotkit", {}).get("actions", [])
 

{copilotkit-0.1.87 → copilotkit-0.1.88}/copilotkit/crewai/crewai_sdk.py

@@ -16,13 +16,22 @@ from litellm.types.utils import (
 )
 from litellm.litellm_core_utils.streaming_handler import CustomStreamWrapper
 from crewai.flow.flow import FlowState, Flow
-from crewai.utilities.events.flow_events import (
-    FlowEvent as CrewAIFlowEvent,
-    FlowStartedEvent,
-    MethodExecutionStartedEvent,
-    MethodExecutionFinishedEvent,
-    FlowFinishedEvent,
-)
+try:
+    from crewai.utilities.events.flow_events import (
+        FlowEvent as CrewAIFlowEvent,
+        FlowStartedEvent,
+        MethodExecutionStartedEvent,
+        MethodExecutionFinishedEvent,
+        FlowFinishedEvent,
+    )
+except ImportError:
+    from crewai.events.types.flow_events import (  # type: ignore[no-redef]
+        FlowEvent as CrewAIFlowEvent,
+        FlowStartedEvent,
+        MethodExecutionStartedEvent,
+        MethodExecutionFinishedEvent,
+        FlowFinishedEvent,
+    )
 from crewai.utilities.events import crewai_event_bus as _crewai_event_bus
 
 from copilotkit.types import Message
@@ -552,7 +561,13 @@ def crewai_flow_messages_to_copilotkit(messages: List[Dict]) -> List[Message]: #
         if "content" in message and message.get("role") == "assistant":
             if message.get("tool_calls"):
                 for tool_call in message["tool_calls"]:
-                    tool_call_names[tool_call["id"]] = tool_call["function"]["name"]
+                    tc_id = tool_call.get("id")
+                    if tc_id is None:
+                        continue
+                    if tool_call.get("function"):
+                        tool_call_names[tc_id] = tool_call["function"].get("name", "")
+                    else:
+                        tool_call_names[tc_id] = tool_call.get("name", "")
 
     for message in messages:
         message_id = message_ids[id(message)]
@@ -565,19 +580,30 @@ def crewai_flow_messages_to_copilotkit(messages: List[Dict]) -> List[Message]: #
                 "id": message_id,
             })
         elif message.get("tool_calls"):
+            # Always emit the assistant message, even with empty content.
+            # Tool call entries reference it via parentMessageId; omitting it
+            # orphans tool calls and breaks frontend thread reconstruction.
+            result.append({
+                "role": message["role"],
+                "content": message.get("content") if message.get("content") is not None else "",
+                "id": message_id,
+            })
             for tool_call in message["tool_calls"]:
+                tc_id = tool_call.get("id")
+                if tc_id is None:
+                    continue
                 if tool_call.get("function"):
                     result.append({
-                        "id": tool_call["id"],
-                        "name": tool_call["function"]["name"],
+                        "id": tc_id,
+                        "name": tool_call["function"].get("name", ""),
                         "arguments": json.loads(tool_call["function"]["arguments"]),
                         "parentMessageId": message_id,
                     })
                 else:
                     result.append({
-                        "id": tool_call["id"],
-                        "name": tool_call["name"],
-                        "arguments": tool_call["arguments"],
+                        "id": tc_id,
+                        "name": tool_call.get("name", ""),
+                        "arguments": tool_call.get("arguments", {}),
                         "parentMessageId": message_id,
                     })
         elif message.get("content"):

{copilotkit-0.1.87 → copilotkit-0.1.88}/copilotkit/langchain.py

@@ -4,7 +4,6 @@ copilotkit.langchain is deprecated. Use copilotkit.langgraph instead.
 import warnings
 from copilotkit.langgraph import (
   langchain_messages_to_copilotkit,
-  copilotkit_messages_to_langchain,
   copilotkit_customize_config,
   copilotkit_exit,
   copilotkit_emit_state,
@@ -21,7 +20,6 @@ warnings.warn(
 
 __all__ = [
   "langchain_messages_to_copilotkit",
-  "copilotkit_messages_to_langchain",
   "copilotkit_customize_config",
   "copilotkit_exit",
   "copilotkit_emit_state",

{copilotkit-0.1.87 → copilotkit-0.1.88}/copilotkit/langgraph.py

@@ -6,7 +6,7 @@ import uuid
 import json
 import warnings
 import asyncio
-from typing import List, Optional, Any, Union, Dict, Callable, cast
+from typing import List, Optional, Any, Union, Dict
 from typing_extensions import TypedDict
 from langgraph.graph import MessagesState
 
@@ -46,82 +46,6 @@ class CopilotKitState(MessagesState):
     copilotkit: CopilotKitProperties
 
 
-def copilotkit_messages_to_langchain(
-        use_function_call: bool = False
-    ) -> Callable[[List[Message]], List[BaseMessage]]:
-    """
-    Convert CopilotKit messages to LangChain messages
-    """
-    def _copilotkit_messages_to_langchain(messages: List[Message]) -> List[BaseMessage]:
-        result = []
-        processed_action_executions = set()
-        for message in cast(Any, messages):
-            if message["type"] == "TextMessage":
-                if message["role"] == "user":
-                    result.append(HumanMessage(content=message["content"], id=message["id"]))
-                elif message["role"] == "system":
-                    result.append(SystemMessage(content=message["content"], id=message["id"]))
-                elif message["role"] == "assistant":
-                    result.append(AIMessage(content=message["content"], id=message["id"]))
-            elif message["type"] == "ActionExecutionMessage":
-                if use_function_call:
-                    result.append(AIMessage(
-                        id=message["id"],
-                        content="",
-                        additional_kwargs={
-                            'function_call':{
-                                'name': message["name"],
-                                'arguments': json.dumps(message["arguments"]),
-                            }
-                        } 
-                    ))
-                else:
-                    # convert multiple tool calls to a single message
-                    message_id = message.get("parentMessageId")
-                    if message_id is None:
-                        message_id = message["id"]
-
-                    if message_id in processed_action_executions:
-                        continue
-
-                    processed_action_executions.add(message_id)
-
-                    all_tool_calls = []
-
-                    # Find all tool calls for this message (only ActionExecutionMessage type)
-                    for msg in messages:
-                        if msg.get("type") != "ActionExecutionMessage":
-                            continue
-                        if msg.get("parentMessageId", None) == message_id or msg["id"] == message_id:
-                            all_tool_calls.append(msg)
-
-                    tool_calls = [{
-                        "name": t["name"],
-                        "args": t["arguments"],
-                        "id": t["id"],
-                    } for t in all_tool_calls]
-
-                    result.append(
-                        AIMessage(
-                            id=message_id,
-                            content="",
-                            tool_calls=tool_calls
-                        )
-                    )
-
-            elif message["type"] == "ResultMessage":
-                result.append(ToolMessage(
-                    id=message["id"],
-                    content=message["result"],
-                    name=message["actionName"],
-                    tool_call_id=message["actionExecutionId"]
-                ))
-
-        return result
-
-    return _copilotkit_messages_to_langchain
-
-
 def langchain_messages_to_copilotkit(
         messages: List[BaseMessage]
     ) -> List[Message]:
@@ -173,12 +97,14 @@ def langchain_messages_to_copilotkit(
                 "id": message.id,
             })
         elif isinstance(message, AIMessage):
-            if content:
-                result.append({
-                    "role": "assistant",
-                    "content": content,
-                    "id": message.id,
-                })
+            # Always emit the assistant message, even with empty content.
+            # Tool call entries reference it via parentMessageId; omitting it
+            # orphans tool calls and breaks frontend thread reconstruction.
+            result.append({
+                "role": "assistant",
+                "content": content if content is not None else "",
+                "id": message.id,
+            })
             if message.tool_calls:
                 for tool_call in message.tool_calls:
                     result.append({
@@ -507,6 +433,13 @@ def copilotkit_interrupt(
         "__copilotkit_interrupt_value__": interrupt_values,
         "__copilotkit_messages__": [interrupt_message]
     })
-    answer = response[-1].content
+    if isinstance(response, str):
+        answer = response
+    elif isinstance(response, dict):
+        answer = json.dumps(response)
+    elif isinstance(response, list):
+        answer = response[-1].content
+    else:
+        answer = str(response)
 
     return answer, response

{copilotkit-0.1.87 → copilotkit-0.1.88}/copilotkit/langgraph_agui_agent.py

@@ -198,8 +198,8 @@ class LangGraphAGUIAgent(LangGraphAgent):
         return {
             **merged_state,
             'copilotkit': {
-                'actions': agui_properties.get('tools', []),
-                'context': agui_properties.get('context', [])
+                'actions': [a.model_dump() if hasattr(a, 'model_dump') else a for a in agui_properties.get('tools', [])],
+                'context': [c.model_dump() if hasattr(c, 'model_dump') else c for c in agui_properties.get('context', [])]
             },
         }
 

{copilotkit-0.1.87 → copilotkit-0.1.88}/copilotkit/sdk.py

@@ -130,12 +130,12 @@ class CopilotKitRemoteEndpoint:
     Serving agents works in a similar way to serving actions:
 
     ```python
-    from copilotkit import CopilotKitRemoteEndpoint, LangGraphAgent
+    from copilotkit import CopilotKitRemoteEndpoint, LangGraphAGUIAgent
     from my_agent.agent import graph
 
     sdk = CopilotKitRemoteEndpoint(
         agents=[
-            LangGraphAgent(
+            LangGraphAGUIAgent(
                 name="email_agent",
                 description="This agent sends emails",
                 graph=graph,
@@ -147,12 +147,12 @@ class CopilotKitRemoteEndpoint:
     To dynamically build agents, provide a callable that returns a list of agents:
 
     ```python
-    from copilotkit import CopilotKitRemoteEndpoint, LangGraphAgent
+    from copilotkit import CopilotKitRemoteEndpoint, LangGraphAGUIAgent
     from my_agent.agent import graph
 
     sdk = CopilotKitRemoteEndpoint(
         agents=lambda context: [
-            LangGraphAgent(
+            LangGraphAGUIAgent(
                 name="email_agent",
                 description="This agent sends emails",
                 graph=graph,
@@ -226,13 +226,6 @@ class CopilotKitRemoteEndpoint:
         self.agents = agents or []
         self.actions = actions or []
 
-        if isinstance(agents, list):
-            from .langgraph_agent import LangGraphAgent
-            for agent in agents:
-                if isinstance(agent, LangGraphAgent):
-                    raise ValueError(
-                        "LangGraphAgent should be instantiated using LangGraphAGUIAgent. Refer to https://docs.copilotkit.ai/langgraph for more information.")
-        
 
     def info(
         self,

{copilotkit-0.1.87 → copilotkit-0.1.88}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "copilotkit"
-version = "0.1.87"
+version = "0.1.88"
 description = "CopilotKit python SDK"
 authors = ["Markus Ecker <markus.ecker@gmail.com>"]
 license = "MIT"
@@ -16,19 +16,29 @@ keywords = [
   "langsmith",
   "langserve",
 ]
+classifiers = [
+  "Programming Language :: Python :: 3.10",
+  "Programming Language :: Python :: 3.11",
+  "Programming Language :: Python :: 3.12",
+  "Programming Language :: Python :: 3.13",
+  "Programming Language :: Python :: 3.14",
+]
 packages = [{ include = "copilotkit" }]
-include = ["copilotkit/*.json"]
+include = ["copilotkit/*.json", "copilotkit/py.typed"]
 
 [tool.poetry.dependencies]
-python = ">=3.10,<3.13"
+python = ">=3.10,<3.15"
 langgraph = { version = ">=0.3.25,<2" }
 langchain = { version = ">=0.3.0" }
-crewai = { version = "0.118.0", optional = true }
-ag-ui-langgraph = { version = ">=0.0.29", extras = ["fastapi"] }
-ag-ui-protocol = ">=0.1.11"
+crewai = { version = ">=0.118.0", optional = true, python = ">=3.10,<3.14" }
+ag-ui-langgraph = { version = ">=0.0.35", extras = ["fastapi"] }
+ag-ui-protocol = ">=0.1.15"
 fastapi = ">=0.115.0,<1.0.0"
 partialjson = "^0.0.8"
 toml = "^0.10.2"
+# Pin transitive: cp314 wheels first appear in 2.35.0; older versions
+# would force a Rust source build that fails on PyO3 0.24 (capped at 3.13).
+pydantic-core = ">=2.35.0"
 
 [tool.poetry.extras]
 crewai = ["crewai"]