copilotkit 0.1.95a0__tar.gz → 0.1.95a2__tar.gz

{copilotkit-0.1.95a0 → copilotkit-0.1.95a2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: copilotkit
-Version: 0.1.95a0
+Version: 0.1.95a2
 Summary: CopilotKit python SDK
 License: MIT
 Keywords: copilot,copilotkit,langgraph,langchain,ai,langsmith,langserve
@@ -15,7 +15,7 @@ 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.42a0)
+Requires-Dist: ag-ui-langgraph[fastapi] (==0.0.42-alpha.0)
 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)

{copilotkit-0.1.95a0 → copilotkit-0.1.95a2}/copilotkit/copilotkit_lg_middleware.py

@@ -35,13 +35,9 @@ from .langgraph import CopilotKitProperties
 # "no auto-A2UI" instead of breaking the whole middleware import.
 try:  # pragma: no cover - exercised indirectly via the a2ui injection path
     from ag_ui_langgraph import get_a2ui_tools, A2UIToolParams
-    # Framework-agnostic catalog resolution shared with every A2UI adapter
-    # (Strands, etc.) so the two-path lookup lives in one place.
-    from ag_ui_a2ui_toolkit import resolve_a2ui_catalog
 except Exception:  # noqa: BLE001 - any import failure means the feature is off
     get_a2ui_tools = None
     A2UIToolParams = None
-    resolve_a2ui_catalog = None
 
 # Track which httpx clients already have the header-propagation hook installed
 # (by object id) so we never double-install on repeated model calls.
@@ -325,20 +321,58 @@ class CopilotKitMiddleware(AgentMiddleware[StateSchema, Any]):
 
     @staticmethod
     def _resolve_a2ui_catalog(state: dict) -> "tuple[str | None, str | None] | None":
-        """Resolve ``(component_schema, catalog_id)`` from run state.
-
-        Delegates to the framework-agnostic, ``ag-ui``-keyed resolver in
-        ``ag-ui-a2ui-toolkit``. CopilotKit's LangGraph layer mirrors the SAME
-        ag-ui state under a ``copilotkit`` alias key, so if the canonical
-        ``ag-ui`` lookup comes up empty we retry against ``copilotkit`` (this
-        copilotkit-side read is fine here; the OSS toolkit stays
-        copilotkit-agnostic).
+        """Find the frontend-registered A2UI catalog wherever it was passed.
+
+        Returns ``(component_schema, catalog_id)`` when a catalog is present,
+        else ``None`` (so the tool is never advertised when the client can't
+        render A2UI). Two delivery paths are supported, because the catalog
+        lands in different places depending on how the agent is served:
+
+        - **AG-UI native endpoint** → ``state["ag-ui"]["a2ui_schema"]``, a JSON
+          string ``{"catalogId": ..., "components": [...]}``.
+        - **CopilotKit runtime proxy** → a ``state["copilotkit"]["context"]``
+          entry describing the A2UI catalog (catalog id + component schemas as
+          text).
+
+        ``component_schema`` is the text/JSON the subagent should compose from;
+        ``catalog_id`` binds generated surfaces to the frontend's catalog (so
+        BYOC custom catalogs render their own components, not the basic one).
         """
-        if resolve_a2ui_catalog is None:
-            return None
-        return resolve_a2ui_catalog(state) or resolve_a2ui_catalog(
-            {"ag-ui": state.get("copilotkit") or {}}
-        )
+        # AG-UI native path.
+        ag_ui = state.get("ag-ui") or {}
+        a2ui_schema = ag_ui.get("a2ui_schema")
+        if a2ui_schema:
+            catalog_id = None
+            try:
+                parsed = (
+                    json.loads(a2ui_schema)
+                    if isinstance(a2ui_schema, str)
+                    else a2ui_schema
+                )
+                if isinstance(parsed, dict):
+                    catalog_id = parsed.get("catalogId")
+            except (TypeError, ValueError):
+                pass
+            # Native path: the toolkit reads ``a2ui_schema`` from state itself,
+            # so no composition_guide is needed — just surface the catalog id.
+            return None, catalog_id
+
+        # CopilotKit runtime-proxy path: the catalog arrives as a context entry.
+        context = (state.get("copilotkit") or {}).get("context") or []
+        for entry in context:
+            if not isinstance(entry, dict):
+                continue
+            description = entry.get("description") or ""
+            value = entry.get("value") or ""
+            if "A2UI catalog" not in description or not value:
+                continue
+            # The value lists catalogs as "- <catalogId>" lines; the first is
+            # the custom catalog the client registered.
+            match = re.search(r"(?m)^\s*-\s+(\S+)", value)
+            catalog_id = match.group(1) if match else None
+            return value, catalog_id
+
+        return None
 
     @staticmethod
     def _a2ui_inject_decision(state: dict) -> "bool | str | None":

{copilotkit-0.1.95a0 → copilotkit-0.1.95a2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "copilotkit"
-version = "0.1.95-alpha.0"
+version = "0.1.95-alpha.2"
 description = "CopilotKit python SDK"
 authors = ["Markus Ecker <markus.ecker@gmail.com>"]
 license = "MIT"
@@ -31,7 +31,7 @@ python = ">=3.10,<3.15"
 langgraph = { version = ">=0.3.25,<2" }
 langchain = { version = ">=0.3.0" }
 crewai = { version = ">=0.118.0", optional = true, python = ">=3.10,<3.14" }
-ag-ui-langgraph = { version = ">=0.0.42a0", extras = ["fastapi"] }
+ag-ui-langgraph = { version = "0.0.42-alpha.0", extras = ["fastapi"] }
 ag-ui-protocol = ">=0.1.15"
 fastapi = ">=0.115.0,<1.0.0"
 partialjson = "^0.0.8"