[pull] main from CopilotKit:main

.claude-plugin/marketplace.json

@@ -6,14 +6,14 @@
   },
   "metadata": {
     "description": "AI agent skills for CopilotKit — setup, develop, integrate, debug, upgrade, and contribute",
-    "version": "1.66.0"
+    "version": "1.66.1"
   },
   "plugins": [
     {
       "name": "copilotkit",
       "source": "./",
       "description": "AI agent skills for CopilotKit — setup, develop, integrate, debug, upgrade, and contribute to CopilotKit projects",
-      "version": "1.66.0",
+      "version": "1.66.1",
       "author": {
         "name": "CopilotKit",
         "url": "https://copilotkit.ai"

.claude-plugin/plugin.json

@@ -1,7 +1,7 @@
 {
   "name": "copilotkit",
   "description": "AI agent skills for CopilotKit — setup, develop, integrate, debug, upgrade, and contribute to CopilotKit projects",
-  "version": "1.66.0",
+  "version": "1.66.1",
   "author": {
     "name": "CopilotKit",
     "url": "https://copilotkit.ai"

examples/integrations/pydantic-ai/agent/pyproject.toml

@@ -6,8 +6,8 @@ readme = "README.md"
 requires-python = ">=3.12"
 dependencies = [
   "uvicorn",
-  "pydantic-ai-slim[ag-ui,openai]==1.0.18",
-  "ag-ui-protocol==0.1.18",
+  "pydantic-ai-slim[ag-ui,openai]==2.22.0",
+  "ag-ui-protocol==0.1.19",
   "python-dotenv",
   "logfire>=4.10.0",
 ]

examples/integrations/pydantic-ai/agent/src/agent.py

@@ -2,7 +2,7 @@
 from textwrap import dedent
 from pydantic import BaseModel, Field
 from pydantic_ai import Agent, RunContext
-from pydantic_ai.ag_ui import StateDeps
+from pydantic_ai.ui import StateDeps
 from ag_ui.core import EventType, StateSnapshotEvent
 from pydantic_ai.models.openai import OpenAIResponsesModel
 

examples/integrations/pydantic-ai/agent/src/main.py

@@ -1,9 +1,17 @@
 from agent import ProverbsState, StateDeps, agent
+from pydantic_ai.ui.ag_ui import AGUIAdapter
 from starlette.applications import Starlette
-from starlette.responses import JSONResponse
-from starlette.routing import Mount, Route
+from starlette.requests import Request
+from starlette.responses import JSONResponse, Response
+from starlette.routing import Route
 
-agui_app = agent.to_ag_ui(deps=StateDeps(ProverbsState()))
+
+async def run_agent(request: Request) -> Response:
+    # Fresh deps per request: `dispatch_request` writes the client's state into
+    # `deps.state`, so a shared instance leaks state between threads, channels and users.
+    return await AGUIAdapter.dispatch_request(
+        request, agent=agent, deps=StateDeps(ProverbsState())
+    )
 
 
 async def health(request):
@@ -13,7 +21,7 @@ async def health(request):
 app = Starlette(
     routes=[
         Route("/health", health),
-        Mount("/", app=agui_app),
+        Route("/", run_agent, methods=["POST"]),
     ]
 )
 

packages/a2ui-renderer/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@copilotkit/a2ui-renderer",
-  "version": "1.66.0",
+  "version": "1.66.1",
   "private": false,
   "description": "A2UI Renderer for CopilotKit - render A2UI surfaces in React applications",
   "keywords": [

packages/agentcore-runner/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@copilotkit/agentcore-runner",
-  "version": "1.66.0",
+  "version": "1.66.1",
   "description": "AWS Bedrock AgentCore-compatible agent runner for CopilotKit2",
   "repository": {
     "type": "git",

packages/channels-core/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@copilotkit/channels-core",
-  "version": "0.7.0",
+  "version": "0.7.1",
   "description": "Platform-agnostic JSX channel engine for CopilotKit (createChannel, Thread, PlatformAdapter, ActionStore).",
   "license": "MIT",
   "repository": {

packages/channels-discord/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@copilotkit/channels-discord",
-  "version": "0.7.0",
+  "version": "0.7.1",
   "description": "Discord PlatformAdapter for @copilotkit/channels.",
   "license": "MIT",
   "repository": {

packages/channels-intelligence/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@copilotkit/channels-intelligence",
-  "version": "0.7.0",
+  "version": "0.7.1",
   "description": "Intelligence-delivered Channel adapter for CopilotKit JSX channels (@copilotkit/channels).",
   "license": "MIT",
   "repository": {

packages/channels-intelligence/src/index.ts

@@ -16,6 +16,7 @@ export type {
 
 export {
   connectRealtimeGateway,
+  RealtimeGatewayJoinError,
   RealtimeGatewayUnreachableError,
 } from "./realtime-gateway.js";
 export type {

packages/channels-intelligence/src/realtime-gateway-retryability.test.ts

@@ -0,0 +1,181 @@
+import { expect, test, vi } from "vitest";
+import {
+  connectRealtimeGateway,
+  RealtimeGatewayUnreachableError,
+} from "./realtime-gateway.js";
+
+interface RetryabilitySetup {
+  connection: ReturnType<typeof connectRealtimeGateway>;
+  diagnosticFetch: ReturnType<typeof vi.fn<typeof fetch>>;
+}
+
+/** Build one socket that opens but rejects or never answers its join. */
+function setupJoinFailure(
+  response?: Record<string, unknown>,
+): ReturnType<typeof connectRealtimeGateway> {
+  class RejectedWebSocket {
+    static readonly CONNECTING = 0;
+    static readonly OPEN = 1;
+    static readonly CLOSING = 2;
+    static readonly CLOSED = 3;
+    readyState = RejectedWebSocket.CONNECTING;
+    binaryType = "arraybuffer";
+    onopen: (() => void) | null = null;
+    onmessage: ((event: { data: string }) => void) | null = null;
+    onerror: ((error: Error) => void) | null = null;
+    onclose: ((event: { code: number }) => void) | null = null;
+
+    constructor() {
+      queueMicrotask(() => {
+        this.readyState = RejectedWebSocket.OPEN;
+        this.onopen?.();
+      });
+    }
+
+    send(data: string): void {
+      const frame = JSON.parse(data) as [string, string, string, string];
+      const [joinRef, ref, topic, event] = frame;
+      if (event !== "phx_join" || response === undefined) return;
+      queueMicrotask(() =>
+        this.onmessage?.({
+          data: JSON.stringify([
+            joinRef,
+            ref,
+            topic,
+            "phx_reply",
+            { status: "error", response },
+          ]),
+        }),
+      );
+    }
+
+    close(): void {
+      this.readyState = RejectedWebSocket.CLOSED;
+      queueMicrotask(() => this.onclose?.({ code: 1000 }));
+    }
+  }
+
+  return connectRealtimeGateway({
+    wsUrl: "wss://gateway.example/channels",
+    apiKey: "cpk-test",
+    projectId: 7,
+    join: {
+      protocol: "channel_delivery_v1",
+      runtimeInstanceId: "rti_1",
+      channels: [{ channelName: "support", adapter: "slack" }],
+    },
+    timeoutMs: 10,
+    webSocket: RejectedWebSocket,
+  });
+}
+
+/** Build one socket that cannot upgrade and an HTTP diagnosis for its host. */
+function setupRetryability(
+  status: number,
+  transportError: Error = new Error("upgrade refused"),
+): RetryabilitySetup {
+  class RefusedWebSocket {
+    static readonly CONNECTING = 0;
+    static readonly OPEN = 1;
+    static readonly CLOSING = 2;
+    static readonly CLOSED = 3;
+    readyState = RefusedWebSocket.CONNECTING;
+    binaryType = "arraybuffer";
+    onopen: (() => void) | null = null;
+    onmessage: ((event: { data: string }) => void) | null = null;
+    onerror: ((error: Error) => void) | null = null;
+    onclose: ((event: { code: number }) => void) | null = null;
+
+    constructor() {
+      queueMicrotask(() => this.onerror?.(transportError));
+    }
+
+    send(): void {}
+
+    close(): void {
+      this.readyState = RefusedWebSocket.CLOSED;
+      queueMicrotask(() => this.onclose?.({ code: 1000 }));
+    }
+  }
+
+  const diagnosticFetch = vi.fn<typeof fetch>(async () =>
+    Promise.resolve(new Response(null, { status })),
+  );
+  const connection = connectRealtimeGateway({
+    wsUrl: "wss://gateway.example/channels",
+    apiKey: "cpk-test",
+    projectId: 7,
+    join: {
+      protocol: "channel_delivery_v1",
+      runtimeInstanceId: "rti_1",
+      channels: [{ channelName: "support", adapter: "slack" }],
+    },
+    connectTimeoutMs: 10,
+    diagnosticFetch,
+    webSocket: RefusedWebSocket,
+  });
+
+  return { connection, diagnosticFetch };
+}
+
+test("an HTTP 502 gateway diagnosis is retryable", async () => {
+  const { connection, diagnosticFetch } = setupRetryability(502);
+
+  const error = await connection.catch((cause: unknown) => cause);
+
+  expect(diagnosticFetch).toHaveBeenCalledOnce();
+  expect(error).toBeInstanceOf(RealtimeGatewayUnreachableError);
+  expect((error as RealtimeGatewayUnreachableError).retryable).toBe(true);
+});
+
+test("an HTTP 403 gateway diagnosis is terminal", async () => {
+  const { connection, diagnosticFetch } = setupRetryability(403);
+
+  const error = await connection.catch((cause: unknown) => cause);
+
+  expect(diagnosticFetch).toHaveBeenCalledOnce();
+  expect(error).toBeInstanceOf(RealtimeGatewayUnreachableError);
+  expect((error as RealtimeGatewayUnreachableError).retryable).toBe(false);
+});
+
+test("an ENOTFOUND transport failure is terminal without an HTTP probe", async () => {
+  const transportError = Object.assign(
+    new Error("getaddrinfo ENOTFOUND gateway.example"),
+    { code: "ENOTFOUND" },
+  );
+  const { connection, diagnosticFetch } = setupRetryability(
+    502,
+    transportError,
+  );
+
+  const error = await connection.catch((cause: unknown) => cause);
+
+  expect(diagnosticFetch).not.toHaveBeenCalled();
+  expect(error).toBeInstanceOf(RealtimeGatewayUnreachableError);
+  expect((error as RealtimeGatewayUnreachableError).retryable).toBe(false);
+});
+
+test("a gateway_draining join rejection preserves its retry classification", async () => {
+  const response = { reason: "gateway_draining", retryable: true };
+
+  const error = await setupJoinFailure(response).catch(
+    (cause: unknown) => cause,
+  );
+
+  expect(error).toMatchObject({
+    code: "GATEWAY_JOIN_FAILED",
+    reason: response,
+    retryable: true,
+  });
+  expect((error as Error).message).toMatch(/gateway_draining/);
+});
+
+test("an initial gateway join timeout is retryable", async () => {
+  const error = await setupJoinFailure().catch((cause: unknown) => cause);
+
+  expect(error).toMatchObject({
+    code: "GATEWAY_JOIN_FAILED",
+    retryable: true,
+  });
+  expect((error as Error).message).toMatch(/join timed out/);
+});