@copilotkit/sdk-js 1.76.0 → 1.77.1

package/dist/langgraph/middleware.cjs

@@ -10,20 +10,42 @@ const a2uiToolsByThread = /* @__PURE__ */ new Map();
 const A2UI_DEFAULT_THREAD_KEY = "__copilotkit_a2ui_default__";
 const a2uiThreadKey = (state) => state?.thread_id || A2UI_DEFAULT_THREAD_KEY;
 /**
+* Merge AG-UI defaults with CopilotKit overrides. Dictionaries recurse; every
+* conflicting leaf uses CopilotKit, including false, null and empty values.
+* Arrays and serialized strings are atomic: no positional pairing or decoding.
+*/
+const isPropertyBag = (value) => value !== null && typeof value === "object" && !Array.isArray(value);
+const mergeProperties = (base, overrides) => Object.fromEntries(Array.from(new Set([...Object.keys(base), ...Object.keys(overrides)])).map((key) => {
+	return [key, Object.prototype.hasOwnProperty.call(overrides, key) ? isPropertyBag(base[key]) && isPropertyBag(overrides[key]) ? mergeProperties(base[key], overrides[key]) : overrides[key] : base[key]];
+}));
+const effectiveProperties = (state) => {
+	const source = isPropertyBag(state) ? state : {};
+	return mergeProperties(isPropertyBag(source["ag-ui"]) ? source["ag-ui"] : {}, isPropertyBag(source.copilotkit) ? source.copilotkit : {});
+};
+const decodeCatalogValue = (value) => {
+	if (typeof value !== "string") return value;
+	try {
+		return JSON.parse(value);
+	} catch {
+		return value;
+	}
+};
+/**
 * Find the frontend-registered A2UI catalog wherever it was passed. Returns
 * `{ compositionGuide?, catalogId? }` when a catalog is present, else `null`
-* (so the tool is never advertised when the client can't render A2UI). Two
+* independently of the explicit tool-injection decision. Two
 * delivery paths, depending on how the agent is served:
 *  - AG-UI native endpoint → `state["ag-ui"].a2ui_schema` (JSON
 *    `{ catalogId, components }`); the toolkit reads it from state itself.
-*  - CopilotKit runtime proxy → a `state.copilotkit.context` entry describing
+*  - Runtime proxy → an effective `context` entry describing
 *    the A2UI catalog (catalog id + component schemas as text), passed to the
 *    subagent via `compositionGuide`.
 * `catalogId` binds generated surfaces to the frontend's catalog so BYOC
 * custom catalogs render their own components (not the basic one).
 */
 const resolveA2uiCatalog = (state) => {
-	const a2uiSchema = state?.["ag-ui"]?.a2ui_schema;
+	const properties = effectiveProperties(state);
+	const a2uiSchema = properties.a2ui_schema;
 	if (a2uiSchema) {
 		let catalogId;
 		try {
@@ -31,11 +53,11 @@ const resolveA2uiCatalog = (state) => {
 		} catch {}
 		return { catalogId };
 	}
-	const context = state?.copilotkit?.context;
+	const context = decodeCatalogValue(properties.context);
 	for (const entry of Array.isArray(context) ? context : []) {
-		const description = entry?.description ?? "";
-		const value = entry?.value ?? "";
-		if (!description.includes("A2UI catalog") || !value) continue;
+		const description = entry?.description;
+		const value = decodeCatalogValue(entry?.value);
+		if (typeof description !== "string" || typeof value !== "string" || !description.includes("A2UI catalog") || !value) continue;
 		return {
 			compositionGuide: value,
 			catalogId: /^\s*-\s+(\S+)/m.exec(value)?.[1]
@@ -50,7 +72,7 @@ const resolveA2uiCatalog = (state) => {
 * runtime A2UI tool on (truthy or a custom tool-name string). `undefined` means
 * no signal (off, or no A2UI middleware in the pipeline) → no auto-injection.
 */
-const a2uiInjectDecision = (state) => state?.["ag-ui"]?.inject_a2ui_tool;
+const a2uiInjectDecision = (state) => effectiveProperties(state).inject_a2ui_tool;
 /**
 * Augment a Standard-Schema–compatible schema (e.g. Zod) with a
 * `~standard.jsonSchema.input` hook so LangGraph's
@@ -150,7 +172,8 @@ const applyStateNote = (request, expose) => {
 const createAppContextBeforeAgent = (state, runtime) => {
 	const messages = state.messages;
 	if (!messages || messages.length === 0) return;
-	const appContext = state["copilotkit"]?.context ?? runtime?.context;
+	const properties = effectiveProperties(state);
+	const appContext = Object.prototype.hasOwnProperty.call(properties, "context") ? properties.context : runtime?.context;
 	if (!appContext || typeof appContext === "string" && appContext.trim() === "" || typeof appContext === "object" && Object.keys(appContext).length === 0) return;
 	const contextMessageContent = `App Context:\n${typeof appContext === "string" ? appContext : JSON.stringify(appContext, null, 2)}`;
 	const contextMessagePrefix = "App Context:\n";
@@ -219,12 +242,15 @@ const createAppContextBeforeAgent = (state, runtime) => {
 * });
 * ```
 */
-const copilotKitStateSchema = zod.object({ copilotkit: zodState(zod.object({
-	actions: zod.array(zod.any()),
-	context: zod.any().optional(),
-	interceptedToolCalls: zod.array(zod.any()).optional(),
-	originalAIMessageId: zod.string().optional()
-}).optional()) }).passthrough();
+const copilotKitStateSchema = zod.object({
+	"ag-ui": zodState(zod.record(zod.string(), zod.unknown()).optional()),
+	copilotkit: zodState(zod.object({
+		actions: zod.union([zod.array(zod.any()), zod.literal(false)]).nullable().optional(),
+		context: zod.any().optional(),
+		interceptedToolCalls: zod.array(zod.any()).optional(),
+		originalAIMessageId: zod.string().optional()
+	}).passthrough().optional())
+}).passthrough();
 const isToolCallContentBlock = (block) => typeof block === "object" && block !== null && "type" in block && (block.type === "tool_call" || block.type === "tool_call_chunk");
 const usesV1ContentBlocks = (responseMetadata) => typeof responseMetadata === "object" && responseMetadata !== null && "output_version" in responseMetadata && responseMetadata.output_version === "v1";
 /**
@@ -283,11 +309,30 @@ const buildMiddlewareInput = (exposeState, a2uiParams) => ({
 			}
 			const candidate = (0, _ag_ui_langgraph.getA2UITools)(params);
 			if (!new Set((request.tools || []).map((t) => t?.name)).has(candidate.name)) {
+				const invoke = candidate.invoke.bind(candidate);
+				candidate.invoke = (input, config) => {
+					const state = config && "state" in config && isPropertyBag(config.state) ? config.state : request.state;
+					const properties = effectiveProperties(state);
+					const schema = properties.a2ui_schema;
+					const context = decodeCatalogValue(properties.context);
+					return invoke(input, {
+						...config,
+						state: {
+							...state,
+							"ag-ui": {
+								...properties,
+								context: Array.isArray(context) ? context : [],
+								a2ui_schema: isPropertyBag(schema) ? JSON.stringify(schema) : schema
+							}
+						}
+					});
+				};
 				a2uiTool = candidate;
 				a2uiToolsByThread.set(a2uiThreadKey(request.state), a2uiTool);
 			}
 		}
-		let frontendTools = request.state["copilotkit"]?.actions ?? [];
+		const actions = effectiveProperties(request.state).actions;
+		let frontendTools = Array.isArray(actions) ? actions : [];
 		if (a2uiTool) {
 			const drop = typeof decision === "string" ? decision : "render_a2ui";
 			frontendTools = frontendTools.filter((t) => (t?.function?.name ?? t?.name) !== drop);
@@ -339,7 +384,8 @@ const buildMiddlewareInput = (exposeState, a2uiParams) => ({
 		};
 	},
 	afterModel: (state) => {
-		const frontendTools = state["copilotkit"]?.actions ?? [];
+		const actions = effectiveProperties(state).actions;
+		const frontendTools = Array.isArray(actions) ? actions : [];
 		if (frontendTools.length === 0) return;
 		const frontendToolNames = new Set(frontendTools.map((t) => t.function?.name || t.name));
 		const lastMessage = state.messages[state.messages.length - 1];

package/dist/langgraph/middleware.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"middleware.cjs","names":["z","SystemMessage","AIMessage","getForwardedHeaders","getA2UITools"],"sources":["../../src/langgraph/middleware.ts"],"sourcesContent":["import { createMiddleware, AIMessage, SystemMessage } from \"langchain\";\nimport type { InteropZodObject } from \"@langchain/core/utils/types\";\nimport type {\n  StandardJSONSchemaV1,\n  StandardSchemaV1,\n} from \"@standard-schema/spec\";\nimport * as z from \"zod\";\nimport { getA2UITools } from \"@ag-ui/langgraph\";\nimport type { A2UIToolParams } from \"@ag-ui/langgraph\";\nimport { getForwardedHeaders } from \"../header-propagation\";\n\n// ---------------------------------------------------------------------------\n// Auto-A2UI: bridge the inferred model's generate_a2ui tool from wrapModelCall\n// (the only hook that exposes the bound model) to wrapToolCall (where the tool\n// actually executes but the model is absent). Keyed by the run's thread id so\n// concurrent runs don't clobber each other.\n// ---------------------------------------------------------------------------\nconst a2uiToolsByThread = new Map<string, any>();\nconst A2UI_DEFAULT_THREAD_KEY = \"__copilotkit_a2ui_default__\";\nconst a2uiThreadKey = (state: any): string =>\n  (state?.thread_id as string) || A2UI_DEFAULT_THREAD_KEY;\n\n/**\n * Find the frontend-registered A2UI catalog wherever it was passed. Returns\n * `{ compositionGuide?, catalogId? }` when a catalog is present, else `null`\n * (so the tool is never advertised when the client can't render A2UI). Two\n * delivery paths, depending on how the agent is served:\n *  - AG-UI native endpoint → `state[\"ag-ui\"].a2ui_schema` (JSON\n *    `{ catalogId, components }`); the toolkit reads it from state itself.\n *  - CopilotKit runtime proxy → a `state.copilotkit.context` entry describing\n *    the A2UI catalog (catalog id + component schemas as text), passed to the\n *    subagent via `compositionGuide`.\n * `catalogId` binds generated surfaces to the frontend's catalog so BYOC\n * custom catalogs render their own components (not the basic one).\n */\nconst resolveA2uiCatalog = (\n  state: any,\n): { compositionGuide?: string; catalogId?: string } | null => {\n  const a2uiSchema = state?.[\"ag-ui\"]?.a2ui_schema;\n  if (a2uiSchema) {\n    let catalogId: string | undefined;\n    try {\n      const parsed =\n        typeof a2uiSchema === \"string\" ? JSON.parse(a2uiSchema) : a2uiSchema;\n      catalogId = parsed?.catalogId;\n    } catch {\n      // non-JSON schema — fall back to the toolkit's basic catalog\n    }\n    return { catalogId };\n  }\n  const context = state?.copilotkit?.context;\n  for (const entry of Array.isArray(context) ? context : []) {\n    const description = entry?.description ?? \"\";\n    const value = entry?.value ?? \"\";\n    if (!description.includes(\"A2UI catalog\") || !value) continue;\n    const match = /^\\s*-\\s+(\\S+)/m.exec(value);\n    return { compositionGuide: value, catalogId: match?.[1] };\n  }\n  return null;\n};\n\n/**\n * The A2UI `injectA2UITool` decision. The `@ag-ui/a2ui-middleware` forwards it on\n * `forwardedProps`, which `ag-ui-langgraph` surfaces into agent state at\n * `state[\"ag-ui\"].inject_a2ui_tool` — present only when the host turned the\n * runtime A2UI tool on (truthy or a custom tool-name string). `undefined` means\n * no signal (off, or no A2UI middleware in the pipeline) → no auto-injection.\n */\nconst a2uiInjectDecision = (state: any): boolean | string | undefined =>\n  state?.[\"ag-ui\"]?.inject_a2ui_tool;\n\ntype WithJsonSchema<T> = T extends { \"~standard\": infer S }\n  ? Omit<T, \"~standard\"> & {\n      \"~standard\": S &\n        StandardJSONSchemaV1.Props<\n          S extends StandardSchemaV1.Props<infer I, any> ? I : unknown,\n          S extends StandardSchemaV1.Props<any, infer O> ? O : unknown\n        >;\n    }\n  : T;\n\n/**\n * Augment a Standard-Schema–compatible schema (e.g. Zod) with a\n * `~standard.jsonSchema.input` hook so LangGraph's\n * `getJsonSchemaFromSchema` (called from `StateSchema.getJsonSchema`)\n * can serialize the field.\n *\n * Without this, Zod v4 fields carry `~standard.validate` + `vendor` only,\n * and `isStandardJSONSchema()` returns false, so the field is silently\n * dropped from the graph's `output_schema`. That makes AG-UI\n * `STATE_SNAPSHOT` events filter the field out of the payload sent to\n * the frontend even though the underlying thread state has the value.\n *\n * Use this on any custom state field you want visible to the frontend\n * via `useAgent().state.*`.\n *\n * @example\n * ```ts\n * import { zodState } from \"@copilotkit/sdk-js/langgraph\";\n *\n * const stateSchema = z.object({\n *   todos: zodState(z.array(TodoSchema).default(() => [])),\n * });\n * ```\n */\nexport function zodState<T extends object>(schema: T): WithJsonSchema<T> {\n  const std = (schema as { \"~standard\"?: { jsonSchema?: unknown } })[\n    \"~standard\"\n  ];\n  if (std && typeof std === \"object\" && !(\"jsonSchema\" in std)) {\n    let cached: Record<string, unknown> | undefined;\n    std.jsonSchema = {\n      input: () => {\n        if (cached) return cached;\n        // Prefer zod-v4's native `toJSONSchema` when available. Falls back to\n        // an empty object, which is sufficient for the field to appear in the\n        // graph's output_schema (langgraph-api treats it as an opaque field).\n        try {\n          const maybeV4ToJsonSchema = (\n            z as unknown as {\n              toJSONSchema?: (s: unknown) => Record<string, unknown>;\n            }\n          ).toJSONSchema;\n          cached =\n            typeof maybeV4ToJsonSchema === \"function\"\n              ? maybeV4ToJsonSchema(schema)\n              : {};\n        } catch {\n          cached = {};\n        }\n        return cached;\n      },\n    };\n  }\n  return schema as WithJsonSchema<T>;\n}\n\n/**\n * Internal/framework state keys that should never be auto-surfaced to the\n * LLM as user-facing state. These are reducer-managed message buckets,\n * CopilotKit/AG-UI plumbing, or graph-internal scaffolding.\n */\nconst RESERVED_STATE_KEYS: ReadonlySet<string> = new Set([\n  \"messages\",\n  \"copilotkit\",\n  \"ag-ui\",\n  \"tools\",\n  \"structured_response\",\n  \"thread_id\",\n  \"remaining_steps\",\n]);\n\n/**\n * Controls how user-defined state keys are surfaced into the LLM prompt\n * on every model call. Off by default to avoid leaking arbitrary state\n * into prompts; opt in explicitly.\n *\n * - `false` (default) — never surface state.\n * - `true` — every state key not in the reserved internal set and not\n *   prefixed with `_` is JSON-serialized into a \"Current agent state:\"\n *   note appended to the system prompt.\n * - `string[]` — only surface the named keys (use this when you want\n *   explicit control over what the LLM sees, e.g. `[\"liked\", \"todos\"]`).\n */\nexport type ExposeStateOption = boolean | readonly string[];\n\nconst buildStateNote = (\n  state: Record<string, unknown>,\n  expose: ExposeStateOption,\n): string | null => {\n  if (expose === false) return null;\n\n  const allow: ReadonlySet<string> | null = Array.isArray(expose)\n    ? new Set(expose)\n    : null;\n\n  const snapshot: Record<string, unknown> = {};\n  for (const key of Object.keys(state)) {\n    if (\n      allow\n        ? !allow.has(key)\n        : RESERVED_STATE_KEYS.has(key) || key.startsWith(\"_\")\n    ) {\n      continue;\n    }\n    const value = state[key];\n    if (\n      value === undefined ||\n      value === null ||\n      value === \"\" ||\n      (Array.isArray(value) && value.length === 0) ||\n      (typeof value === \"object\" &&\n        !Array.isArray(value) &&\n        Object.keys(value as Record<string, unknown>).length === 0)\n    ) {\n      continue;\n    }\n    snapshot[key] = value;\n  }\n\n  if (Object.keys(snapshot).length === 0) return null;\n\n  let body: string;\n  try {\n    body = JSON.stringify(snapshot, null, 2);\n  } catch {\n    body = String(snapshot);\n  }\n  return `Current agent state:\\n${body}`;\n};\n\nconst applyStateNote = (request: any, expose: ExposeStateOption): any => {\n  const note = buildStateNote(\n    (request.state ?? {}) as Record<string, unknown>,\n    expose,\n  );\n  if (!note) return request;\n\n  const existingMessage = request.systemMessage;\n  if (existingMessage != null) {\n    const separator = existingMessage.text === \"\" ? \"\" : \"\\n\\n\";\n    return {\n      ...request,\n      systemMessage: existingMessage.concat(`${separator}${note}`),\n    };\n  }\n\n  const existing = request.systemPrompt;\n  if (existing == null) {\n    return { ...request, systemPrompt: note };\n  }\n  // existing may be a string OR a SystemMessage\n  const baseText =\n    typeof existing === \"string\"\n      ? existing\n      : typeof existing.content === \"string\"\n        ? existing.content\n        : String(existing.content);\n  return {\n    ...request,\n    systemPrompt: `${baseText}\\n\\n${note}`,\n  };\n};\n\nconst createAppContextBeforeAgent = (state, runtime) => {\n  const messages = state.messages;\n\n  if (!messages || messages.length === 0) {\n    return;\n  }\n\n  // Get app context from runtime\n  const appContext = state[\"copilotkit\"]?.context ?? runtime?.context;\n\n  // Check if appContext is missing or empty\n  const isEmptyContext =\n    !appContext ||\n    (typeof appContext === \"string\" && appContext.trim() === \"\") ||\n    (typeof appContext === \"object\" && Object.keys(appContext).length === 0);\n\n  if (isEmptyContext) {\n    return;\n  }\n\n  // Create the context content\n  const contextContent =\n    typeof appContext === \"string\"\n      ? appContext\n      : JSON.stringify(appContext, null, 2);\n  const contextMessageContent = `App Context:\\n${contextContent}`;\n  const contextMessagePrefix = \"App Context:\\n\";\n\n  // Helper to get message content as string\n  const getContentString = (msg: any): string | null => {\n    if (typeof msg.content === \"string\") return msg.content;\n    if (Array.isArray(msg.content) && msg.content[0]?.text)\n      return msg.content[0].text;\n    return null;\n  };\n\n  // Find the first system/developer message (not our context message) to determine\n  // where to insert our context message (right after it)\n  let firstSystemIndex = -1;\n\n  for (let i = 0; i < messages.length; i++) {\n    const msg = messages[i];\n    const type = msg._getType?.();\n    if (type === \"system\" || type === \"developer\") {\n      const content = getContentString(msg);\n      // Skip if this is our own context message\n      if (content?.startsWith(contextMessagePrefix)) {\n        continue;\n      }\n      firstSystemIndex = i;\n      break;\n    }\n  }\n\n  // Check if our context message already exists\n  let existingContextIndex = -1;\n  for (let i = 0; i < messages.length; i++) {\n    const msg = messages[i];\n    const type = msg._getType?.();\n    if (type === \"system\" || type === \"developer\") {\n      const content = getContentString(msg);\n      if (content?.startsWith(contextMessagePrefix)) {\n        existingContextIndex = i;\n        break;\n      }\n    }\n  }\n\n  // Create the context message\n  const contextMessage = new SystemMessage({ content: contextMessageContent });\n\n  let updatedMessages;\n\n  if (existingContextIndex !== -1) {\n    // Replace existing context message\n    updatedMessages = [...messages];\n    updatedMessages[existingContextIndex] = contextMessage;\n  } else {\n    // Insert after the first system message, or at position 0 if no system message\n    const insertIndex = firstSystemIndex !== -1 ? firstSystemIndex + 1 : 0;\n    updatedMessages = [\n      ...messages.slice(0, insertIndex),\n      contextMessage,\n      ...messages.slice(insertIndex),\n    ];\n  }\n\n  return {\n    ...state,\n    messages: updatedMessages,\n  };\n};\n\n/**\n * CopilotKit Middleware for LangGraph agents.\n *\n * Enables:\n * - Dynamic frontend tools from state.tools\n * - Context provided from CopilotKit useCopilotReadable\n *\n * Works with any agent (prebuilt or custom).\n *\n * @example\n * ```typescript\n * import { createAgent } from \"langchain\";\n * import { copilotkitMiddleware } from \"@copilotkit/sdk-js/langgraph\";\n *\n * const agent = createAgent({\n *   model: \"gpt-4o\",\n *   tools: [backendTool],\n *   middleware: [copilotkitMiddleware],\n * });\n * ```\n */\nconst copilotKitStateSchema = z\n  .object({\n    copilotkit: zodState(\n      z\n        .object({\n          actions: z.array(z.any()),\n          context: z.any().optional(),\n          interceptedToolCalls: z.array(z.any()).optional(),\n          originalAIMessageId: z.string().optional(),\n        })\n        .optional(),\n    ),\n  })\n  .passthrough();\n\nconst isToolCallContentBlock = (block: unknown) =>\n  typeof block === \"object\" &&\n  block !== null &&\n  \"type\" in block &&\n  (block.type === \"tool_call\" || block.type === \"tool_call_chunk\");\n\nconst usesV1ContentBlocks = (responseMetadata: unknown) =>\n  typeof responseMetadata === \"object\" &&\n  responseMetadata !== null &&\n  \"output_version\" in responseMetadata &&\n  responseMetadata.output_version === \"v1\";\n\n/**\n * Rebuilds an AIMessage with `toolCalls` as the source of truth while\n * preserving its non-tool content and metadata. For v1 content blocks, old\n * tool blocks must be removed before construction so they cannot duplicate or\n * override the supplied tool calls when AIMessage synchronizes both fields.\n */\nconst rebuildAIMessageWithToolCalls = (\n  message: AIMessage,\n  toolCalls: AIMessage[\"tool_calls\"],\n) => {\n  let content = message.content;\n  if (\n    usesV1ContentBlocks(message.response_metadata) &&\n    Array.isArray(content)\n  ) {\n    content = content.filter((block) => !isToolCallContentBlock(block));\n  }\n\n  return new AIMessage({\n    content,\n    additional_kwargs: message.additional_kwargs,\n    response_metadata: message.response_metadata,\n    tool_calls: toolCalls,\n    invalid_tool_calls: message.invalid_tool_calls,\n    usage_metadata: message.usage_metadata,\n    id: message.id,\n    name: message.name,\n  });\n};\n\nconst buildMiddlewareInput = (\n  exposeState: ExposeStateOption,\n  a2uiParams?: Omit<A2UIToolParams, \"model\">,\n) => ({\n  name: \"CopilotKitMiddleware\",\n\n  stateSchema: copilotKitStateSchema as unknown as InteropZodObject,\n\n  // Inject frontend tools, surface user state, and forward x-aimock-* headers\n  wrapModelCall: async (request: any, handler: (req: any) => Promise<any>) => {\n    request = applyStateNote(request, exposeState);\n\n    // Forward x-aimock-* headers from the incoming AG-UI request\n    const forwardedHeaders = getForwardedHeaders();\n    if (Object.keys(forwardedHeaders).length > 0) {\n      const existingSettings = request.modelSettings ?? {};\n      const existingHeaders =\n        (existingSettings.headers as Record<string, string>) ?? {};\n      request = {\n        ...request,\n        modelSettings: {\n          ...existingSettings,\n          headers: { ...existingHeaders, ...forwardedHeaders },\n        },\n      };\n    }\n\n    // Opt-in auto-injection of generate_a2ui:\n    // (1) only inject when the A2UI injectA2UITool flag is truthy (forwarded by\n    //     @ag-ui/a2ui-middleware and surfaced at state[\"ag-ui\"].inject_a2ui_tool);\n    // (2) don't double-inject if the agent already defines this tool.\n    // The catalog (when present) only binds surfaces to the FE's catalog; it is\n    // not the gate. The model is inferred from request.model; the built tool is\n    // stashed for wrapToolCall to execute.\n    let a2uiTool: any = null;\n    const decision = a2uiInjectDecision(request.state);\n    if (typeof getA2UITools === \"function\" && decision) {\n      const catalog = resolveA2uiCatalog(request.state);\n      // Shared A2UIToolParams: a single params object owned by the toolkit.\n      // Start from the host overrides (guidelines / catalog id / tool name /\n      // recovery) so a host can steer the subagent, then layer in only what the\n      // host cannot know — the bound model, and the registered catalog id +\n      // compositionGuide — without clobbering any host-set value.\n      const params: A2UIToolParams = {\n        ...a2uiParams,\n        model: request.model,\n      };\n      if (catalog?.catalogId && params.defaultCatalogId == null)\n        params.defaultCatalogId = catalog.catalogId;\n      // Merge the registered catalog schema into any host `guidelines` bag; a\n      // host-set compositionGuide wins, host generation/design overrides stay.\n      if (catalog?.compositionGuide) {\n        const guidelines = { ...params.guidelines };\n        if (guidelines.compositionGuide == null)\n          guidelines.compositionGuide = catalog.compositionGuide;\n        params.guidelines = guidelines;\n      }\n      const candidate = getA2UITools(params);\n      const existingNames = new Set(\n        (request.tools || []).map((t: any) => t?.name),\n      );\n      if (!existingNames.has(candidate.name)) {\n        a2uiTool = candidate;\n        a2uiToolsByThread.set(a2uiThreadKey(request.state), a2uiTool);\n      }\n    }\n\n    let frontendTools = request.state[\"copilotkit\"]?.actions ?? [];\n    if (a2uiTool) {\n      // Our generate_a2ui replaces the runtime's render tool — don't advertise\n      // both. Drop the render tool the A2UI middleware injected.\n      const drop = typeof decision === \"string\" ? decision : \"render_a2ui\";\n      frontendTools = frontendTools.filter(\n        (t: any) => (t?.function?.name ?? t?.name) !== drop,\n      );\n    }\n\n    if (frontendTools.length === 0 && !a2uiTool) {\n      return handler(request);\n    }\n\n    const existingTools = request.tools || [];\n    const mergedTools = [\n      ...existingTools,\n      ...(a2uiTool ? [a2uiTool] : []),\n      ...frontendTools,\n    ];\n\n    return handler({\n      ...request,\n      tools: mergedTools,\n    });\n  },\n\n  // Execute the dynamically-advertised generate_a2ui tool. It is not in the\n  // agent's static tool registry, so the tool node cannot run it on its own;\n  // we supply the implementation (built with the inferred model) for that one\n  // tool. This hook's presence also disables createAgent's \"unknown tool\"\n  // guard for dynamically-advertised tools.\n  wrapToolCall: async (request: any, handler: (req: any) => Promise<any>) => {\n    const tool = a2uiToolsByThread.get(a2uiThreadKey(request.state));\n    if (tool && !request.tool && request.toolCall?.name === tool.name) {\n      return handler({ ...request, tool });\n    }\n    return handler(request);\n  },\n\n  beforeAgent: createAppContextBeforeAgent,\n\n  // Restore frontend tool calls to AIMessage before agent exits\n  afterAgent: (state) => {\n    // Drop the bridged A2UI tool for this run — all tool calls for the turn\n    // have executed by now; the next model call re-stashes if needed.\n    a2uiToolsByThread.delete(a2uiThreadKey(state));\n\n    const interceptedToolCalls = state[\"copilotkit\"]?.interceptedToolCalls;\n    const originalMessageId = state[\"copilotkit\"]?.originalAIMessageId;\n\n    if (!interceptedToolCalls?.length || !originalMessageId) {\n      return;\n    }\n\n    let messageFound = false;\n    const updatedMessages = state.messages.map((msg: any) => {\n      if (AIMessage.isInstance(msg) && msg.id === originalMessageId) {\n        messageFound = true;\n        const existingToolCalls = msg.tool_calls || [];\n        return rebuildAIMessageWithToolCalls(msg, [\n          ...existingToolCalls,\n          ...interceptedToolCalls,\n        ]);\n      }\n      return msg;\n    });\n\n    // Only clear intercepted state if we successfully restored the tool calls\n    if (!messageFound) {\n      console.warn(\n        `CopilotKit: Could not find message with id ${originalMessageId} to restore tool calls`,\n      );\n      return;\n    }\n\n    return {\n      messages: updatedMessages,\n      copilotkit: {\n        ...state[\"copilotkit\"],\n        interceptedToolCalls: undefined,\n        originalAIMessageId: undefined,\n      },\n    };\n  },\n\n  // Intercept frontend tool calls after model returns, before ToolNode executes\n  afterModel: (state) => {\n    const frontendTools = state[\"copilotkit\"]?.actions ?? [];\n    if (frontendTools.length === 0) return;\n\n    const frontendToolNames = new Set(\n      frontendTools.map((t: any) => t.function?.name || t.name),\n    );\n\n    const lastMessage = state.messages[state.messages.length - 1];\n    if (!AIMessage.isInstance(lastMessage) || !lastMessage.tool_calls?.length) {\n      return;\n    }\n\n    const backendToolCalls: any[] = [];\n    const frontendToolCalls: any[] = [];\n\n    for (const call of lastMessage.tool_calls) {\n      if (frontendToolNames.has(call.name)) {\n        frontendToolCalls.push(call);\n      } else {\n        backendToolCalls.push(call);\n      }\n    }\n\n    if (frontendToolCalls.length === 0) return;\n\n    const updatedAIMessage = rebuildAIMessageWithToolCalls(\n      lastMessage,\n      backendToolCalls,\n    );\n\n    return {\n      messages: [...state.messages.slice(0, -1), updatedAIMessage],\n      copilotkit: {\n        ...state[\"copilotkit\"],\n        interceptedToolCalls: frontendToolCalls,\n        originalAIMessageId: lastMessage.id,\n      },\n    };\n  },\n});\n\n/**\n * Build a CopilotKit middleware instance with custom options.\n *\n * Use this when you want to override the default state-exposure behavior\n * (for example to hide a sensitive key, or to use an explicit allowlist), or\n * to steer the auto-injected `generate_a2ui` subagent via `a2uiParams`.\n *\n * `a2uiParams` is an `A2UIToolParams` without `model` (the middleware always\n * injects the bound model). Use it to override the subagent guidelines\n * (`generationGuidelines` / `designGuidelines` / `compositionGuide`),\n * `defaultCatalogId`, `toolName`, `recovery`, etc. on the auto-inject path —\n * which otherwise only ever uses the toolkit defaults. The registered catalog\n * is still folded in, but host-set values win.\n *\n * @example\n * ```typescript\n * import { createCopilotkitMiddleware } from \"@copilotkit/sdk-js/langgraph\";\n *\n * const middleware = createCopilotkitMiddleware({\n *   exposeState: [\"liked\", \"todos\"],\n *   a2uiParams: { guidelines: { designGuidelines: \"...repeating-card layout...\" } },\n * });\n * ```\n */\nexport const createCopilotkitMiddleware = (\n  options: {\n    exposeState?: ExposeStateOption;\n    a2uiParams?: Omit<A2UIToolParams, \"model\">;\n  } = {},\n) => {\n  const exposeState = options.exposeState ?? false;\n  return createMiddleware(\n    buildMiddlewareInput(exposeState, options.a2uiParams) as any,\n  );\n};\n\n/**\n * Default CopilotKit middleware singleton — does NOT surface user state\n * to the LLM. Pass `exposeState: true` (or an allowlist) to\n * {@link createCopilotkitMiddleware} to opt in.\n */\nexport const copilotkitMiddleware = createCopilotkitMiddleware();\n"],"mappings":";;;;;;;;AAiBA,MAAM,oCAAoB,IAAI,KAAkB;AAChD,MAAM,0BAA0B;AAChC,MAAM,iBAAiB,UACpB,OAAO,aAAwB;;;;;;;;;;;;;;AAelC,MAAM,sBACJ,UAC6D;CAC7D,MAAM,aAAa,QAAQ,UAAU;AACrC,KAAI,YAAY;EACd,IAAI;AACJ,MAAI;AAGF,gBADE,OAAO,eAAe,WAAW,KAAK,MAAM,WAAW,GAAG,aACxC;UACd;AAGR,SAAO,EAAE,WAAW;;CAEtB,MAAM,UAAU,OAAO,YAAY;AACnC,MAAK,MAAM,SAAS,MAAM,QAAQ,QAAQ,GAAG,UAAU,EAAE,EAAE;EACzD,MAAM,cAAc,OAAO,eAAe;EAC1C,MAAM,QAAQ,OAAO,SAAS;AAC9B,MAAI,CAAC,YAAY,SAAS,eAAe,IAAI,CAAC,MAAO;AAErD,SAAO;GAAE,kBAAkB;GAAO,WADpB,iBAAiB,KAAK,MAAM,GACW;GAAI;;AAE3D,QAAO;;;;;;;;;AAUT,MAAM,sBAAsB,UAC1B,QAAQ,UAAU;;;;;;;;;;;;;;;;;;;;;;;;;AAoCpB,SAAgB,SAA2B,QAA8B;CACvE,MAAM,MAAO,OACX;AAEF,KAAI,OAAO,OAAO,QAAQ,YAAY,EAAE,gBAAgB,MAAM;EAC5D,IAAI;AACJ,MAAI,aAAa,EACf,aAAa;AACX,OAAI,OAAQ,QAAO;AAInB,OAAI;IACF,MAAM,sBACJA,IAGA;AACF,aACE,OAAO,wBAAwB,aAC3B,oBAAoB,OAAO,GAC3B,EAAE;WACF;AACN,aAAS,EAAE;;AAEb,UAAO;KAEV;;AAEH,QAAO;;;;;;;AAQT,MAAM,sBAA2C,IAAI,IAAI;CACvD;CACA;CACA;CACA;CACA;CACA;CACA;CACD,CAAC;AAgBF,MAAM,kBACJ,OACA,WACkB;AAClB,KAAI,WAAW,MAAO,QAAO;CAE7B,MAAM,QAAoC,MAAM,QAAQ,OAAO,GAC3D,IAAI,IAAI,OAAO,GACf;CAEJ,MAAM,WAAoC,EAAE;AAC5C,MAAK,MAAM,OAAO,OAAO,KAAK,MAAM,EAAE;AACpC,MACE,QACI,CAAC,MAAM,IAAI,IAAI,GACf,oBAAoB,IAAI,IAAI,IAAI,IAAI,WAAW,IAAI,CAEvD;EAEF,MAAM,QAAQ,MAAM;AACpB,MACE,UAAU,UACV,UAAU,QACV,UAAU,MACT,MAAM,QAAQ,MAAM,IAAI,MAAM,WAAW,KACzC,OAAO,UAAU,YAChB,CAAC,MAAM,QAAQ,MAAM,IACrB,OAAO,KAAK,MAAiC,CAAC,WAAW,EAE3D;AAEF,WAAS,OAAO;;AAGlB,KAAI,OAAO,KAAK,SAAS,CAAC,WAAW,EAAG,QAAO;CAE/C,IAAI;AACJ,KAAI;AACF,SAAO,KAAK,UAAU,UAAU,MAAM,EAAE;SAClC;AACN,SAAO,OAAO,SAAS;;AAEzB,QAAO,yBAAyB;;AAGlC,MAAM,kBAAkB,SAAc,WAAmC;CACvE,MAAM,OAAO,eACV,QAAQ,SAAS,EAAE,EACpB,OACD;AACD,KAAI,CAAC,KAAM,QAAO;CAElB,MAAM,kBAAkB,QAAQ;AAChC,KAAI,mBAAmB,MAAM;EAC3B,MAAM,YAAY,gBAAgB,SAAS,KAAK,KAAK;AACrD,SAAO;GACL,GAAG;GACH,eAAe,gBAAgB,OAAO,GAAG,YAAY,OAAO;GAC7D;;CAGH,MAAM,WAAW,QAAQ;AACzB,KAAI,YAAY,KACd,QAAO;EAAE,GAAG;EAAS,cAAc;EAAM;CAG3C,MAAM,WACJ,OAAO,aAAa,WAChB,WACA,OAAO,SAAS,YAAY,WAC1B,SAAS,UACT,OAAO,SAAS,QAAQ;AAChC,QAAO;EACL,GAAG;EACH,cAAc,GAAG,SAAS,MAAM;EACjC;;AAGH,MAAM,+BAA+B,OAAO,YAAY;CACtD,MAAM,WAAW,MAAM;AAEvB,KAAI,CAAC,YAAY,SAAS,WAAW,EACnC;CAIF,MAAM,aAAa,MAAM,eAAe,WAAW,SAAS;AAQ5D,KAJE,CAAC,cACA,OAAO,eAAe,YAAY,WAAW,MAAM,KAAK,MACxD,OAAO,eAAe,YAAY,OAAO,KAAK,WAAW,CAAC,WAAW,EAGtE;CAQF,MAAM,wBAAwB,iBAH5B,OAAO,eAAe,WAClB,aACA,KAAK,UAAU,YAAY,MAAM,EAAE;CAEzC,MAAM,uBAAuB;CAG7B,MAAM,oBAAoB,QAA4B;AACpD,MAAI,OAAO,IAAI,YAAY,SAAU,QAAO,IAAI;AAChD,MAAI,MAAM,QAAQ,IAAI,QAAQ,IAAI,IAAI,QAAQ,IAAI,KAChD,QAAO,IAAI,QAAQ,GAAG;AACxB,SAAO;;CAKT,IAAI,mBAAmB;AAEvB,MAAK,IAAI,IAAI,GAAG,IAAI,SAAS,QAAQ,KAAK;EACxC,MAAM,MAAM,SAAS;EACrB,MAAM,OAAO,IAAI,YAAY;AAC7B,MAAI,SAAS,YAAY,SAAS,aAAa;AAG7C,OAFgB,iBAAiB,IAAI,EAExB,WAAW,qBAAqB,CAC3C;AAEF,sBAAmB;AACnB;;;CAKJ,IAAI,uBAAuB;AAC3B,MAAK,IAAI,IAAI,GAAG,IAAI,SAAS,QAAQ,KAAK;EACxC,MAAM,MAAM,SAAS;EACrB,MAAM,OAAO,IAAI,YAAY;AAC7B,MAAI,SAAS,YAAY,SAAS,aAEhC;OADgB,iBAAiB,IAAI,EACxB,WAAW,qBAAqB,EAAE;AAC7C,2BAAuB;AACvB;;;;CAMN,MAAM,iBAAiB,IAAIC,wBAAc,EAAE,SAAS,uBAAuB,CAAC;CAE5E,IAAI;AAEJ,KAAI,yBAAyB,IAAI;AAE/B,oBAAkB,CAAC,GAAG,SAAS;AAC/B,kBAAgB,wBAAwB;QACnC;EAEL,MAAM,cAAc,qBAAqB,KAAK,mBAAmB,IAAI;AACrE,oBAAkB;GAChB,GAAG,SAAS,MAAM,GAAG,YAAY;GACjC;GACA,GAAG,SAAS,MAAM,YAAY;GAC/B;;AAGH,QAAO;EACL,GAAG;EACH,UAAU;EACX;;;;;;;;;;;;;;;;;;;;;;;AAwBH,MAAM,wBAAwBD,IAC3B,OAAO,EACN,YAAY,SACVA,IACG,OAAO;CACN,SAASA,IAAE,MAAMA,IAAE,KAAK,CAAC;CACzB,SAASA,IAAE,KAAK,CAAC,UAAU;CAC3B,sBAAsBA,IAAE,MAAMA,IAAE,KAAK,CAAC,CAAC,UAAU;CACjD,qBAAqBA,IAAE,QAAQ,CAAC,UAAU;CAC3C,CAAC,CACD,UAAU,CACd,EACF,CAAC,CACD,aAAa;AAEhB,MAAM,0BAA0B,UAC9B,OAAO,UAAU,YACjB,UAAU,QACV,UAAU,UACT,MAAM,SAAS,eAAe,MAAM,SAAS;AAEhD,MAAM,uBAAuB,qBAC3B,OAAO,qBAAqB,YAC5B,qBAAqB,QACrB,oBAAoB,oBACpB,iBAAiB,mBAAmB;;;;;;;AAQtC,MAAM,iCACJ,SACA,cACG;CACH,IAAI,UAAU,QAAQ;AACtB,KACE,oBAAoB,QAAQ,kBAAkB,IAC9C,MAAM,QAAQ,QAAQ,CAEtB,WAAU,QAAQ,QAAQ,UAAU,CAAC,uBAAuB,MAAM,CAAC;AAGrE,QAAO,IAAIE,oBAAU;EACnB;EACA,mBAAmB,QAAQ;EAC3B,mBAAmB,QAAQ;EAC3B,YAAY;EACZ,oBAAoB,QAAQ;EAC5B,gBAAgB,QAAQ;EACxB,IAAI,QAAQ;EACZ,MAAM,QAAQ;EACf,CAAC;;AAGJ,MAAM,wBACJ,aACA,gBACI;CACJ,MAAM;CAEN,aAAa;CAGb,eAAe,OAAO,SAAc,YAAwC;AAC1E,YAAU,eAAe,SAAS,YAAY;EAG9C,MAAM,mBAAmBC,gDAAqB;AAC9C,MAAI,OAAO,KAAK,iBAAiB,CAAC,SAAS,GAAG;GAC5C,MAAM,mBAAmB,QAAQ,iBAAiB,EAAE;GACpD,MAAM,kBACH,iBAAiB,WAAsC,EAAE;AAC5D,aAAU;IACR,GAAG;IACH,eAAe;KACb,GAAG;KACH,SAAS;MAAE,GAAG;MAAiB,GAAG;MAAkB;KACrD;IACF;;EAUH,IAAI,WAAgB;EACpB,MAAM,WAAW,mBAAmB,QAAQ,MAAM;AAClD,MAAI,OAAOC,kCAAiB,cAAc,UAAU;GAClD,MAAM,UAAU,mBAAmB,QAAQ,MAAM;GAMjD,MAAM,SAAyB;IAC7B,GAAG;IACH,OAAO,QAAQ;IAChB;AACD,OAAI,SAAS,aAAa,OAAO,oBAAoB,KACnD,QAAO,mBAAmB,QAAQ;AAGpC,OAAI,SAAS,kBAAkB;IAC7B,MAAM,aAAa,EAAE,GAAG,OAAO,YAAY;AAC3C,QAAI,WAAW,oBAAoB,KACjC,YAAW,mBAAmB,QAAQ;AACxC,WAAO,aAAa;;GAEtB,MAAM,+CAAyB,OAAO;AAItC,OAAI,CAHkB,IAAI,KACvB,QAAQ,SAAS,EAAE,EAAE,KAAK,MAAW,GAAG,KAAK,CAC/C,CACkB,IAAI,UAAU,KAAK,EAAE;AACtC,eAAW;AACX,sBAAkB,IAAI,cAAc,QAAQ,MAAM,EAAE,SAAS;;;EAIjE,IAAI,gBAAgB,QAAQ,MAAM,eAAe,WAAW,EAAE;AAC9D,MAAI,UAAU;GAGZ,MAAM,OAAO,OAAO,aAAa,WAAW,WAAW;AACvD,mBAAgB,cAAc,QAC3B,OAAY,GAAG,UAAU,QAAQ,GAAG,UAAU,KAChD;;AAGH,MAAI,cAAc,WAAW,KAAK,CAAC,SACjC,QAAO,QAAQ,QAAQ;EAIzB,MAAM,cAAc;GAClB,GAFoB,QAAQ,SAAS,EAAE;GAGvC,GAAI,WAAW,CAAC,SAAS,GAAG,EAAE;GAC9B,GAAG;GACJ;AAED,SAAO,QAAQ;GACb,GAAG;GACH,OAAO;GACR,CAAC;;CAQJ,cAAc,OAAO,SAAc,YAAwC;EACzE,MAAM,OAAO,kBAAkB,IAAI,cAAc,QAAQ,MAAM,CAAC;AAChE,MAAI,QAAQ,CAAC,QAAQ,QAAQ,QAAQ,UAAU,SAAS,KAAK,KAC3D,QAAO,QAAQ;GAAE,GAAG;GAAS;GAAM,CAAC;AAEtC,SAAO,QAAQ,QAAQ;;CAGzB,aAAa;CAGb,aAAa,UAAU;AAGrB,oBAAkB,OAAO,cAAc,MAAM,CAAC;EAE9C,MAAM,uBAAuB,MAAM,eAAe;EAClD,MAAM,oBAAoB,MAAM,eAAe;AAE/C,MAAI,CAAC,sBAAsB,UAAU,CAAC,kBACpC;EAGF,IAAI,eAAe;EACnB,MAAM,kBAAkB,MAAM,SAAS,KAAK,QAAa;AACvD,OAAIF,oBAAU,WAAW,IAAI,IAAI,IAAI,OAAO,mBAAmB;AAC7D,mBAAe;AAEf,WAAO,8BAA8B,KAAK,CACxC,GAFwB,IAAI,cAAc,EAAE,EAG5C,GAAG,qBACJ,CAAC;;AAEJ,UAAO;IACP;AAGF,MAAI,CAAC,cAAc;AACjB,WAAQ,KACN,8CAA8C,kBAAkB,wBACjE;AACD;;AAGF,SAAO;GACL,UAAU;GACV,YAAY;IACV,GAAG,MAAM;IACT,sBAAsB;IACtB,qBAAqB;IACtB;GACF;;CAIH,aAAa,UAAU;EACrB,MAAM,gBAAgB,MAAM,eAAe,WAAW,EAAE;AACxD,MAAI,cAAc,WAAW,EAAG;EAEhC,MAAM,oBAAoB,IAAI,IAC5B,cAAc,KAAK,MAAW,EAAE,UAAU,QAAQ,EAAE,KAAK,CAC1D;EAED,MAAM,cAAc,MAAM,SAAS,MAAM,SAAS,SAAS;AAC3D,MAAI,CAACA,oBAAU,WAAW,YAAY,IAAI,CAAC,YAAY,YAAY,OACjE;EAGF,MAAM,mBAA0B,EAAE;EAClC,MAAM,oBAA2B,EAAE;AAEnC,OAAK,MAAM,QAAQ,YAAY,WAC7B,KAAI,kBAAkB,IAAI,KAAK,KAAK,CAClC,mBAAkB,KAAK,KAAK;MAE5B,kBAAiB,KAAK,KAAK;AAI/B,MAAI,kBAAkB,WAAW,EAAG;EAEpC,MAAM,mBAAmB,8BACvB,aACA,iBACD;AAED,SAAO;GACL,UAAU,CAAC,GAAG,MAAM,SAAS,MAAM,GAAG,GAAG,EAAE,iBAAiB;GAC5D,YAAY;IACV,GAAG,MAAM;IACT,sBAAsB;IACtB,qBAAqB,YAAY;IAClC;GACF;;CAEJ;;;;;;;;;;;;;;;;;;;;;;;;;AA0BD,MAAa,8BACX,UAGI,EAAE,KACH;AAEH,wCACE,qBAFkB,QAAQ,eAAe,OAEP,QAAQ,WAAW,CACtD;;;;;;;AAQH,MAAa,uBAAuB,4BAA4B"}
+{"version":3,"file":"middleware.cjs","names":["z","SystemMessage","AIMessage","getForwardedHeaders","getA2UITools"],"sources":["../../src/langgraph/middleware.ts"],"sourcesContent":["import { createMiddleware, AIMessage, SystemMessage } from \"langchain\";\nimport type { InteropZodObject } from \"@langchain/core/utils/types\";\nimport type {\n  StandardJSONSchemaV1,\n  StandardSchemaV1,\n} from \"@standard-schema/spec\";\nimport * as z from \"zod\";\nimport { getA2UITools } from \"@ag-ui/langgraph\";\nimport type { A2UIToolParams } from \"@ag-ui/langgraph\";\nimport { getForwardedHeaders } from \"../header-propagation\";\n\n// ---------------------------------------------------------------------------\n// Auto-A2UI: bridge the inferred model's generate_a2ui tool from wrapModelCall\n// (the only hook that exposes the bound model) to wrapToolCall (where the tool\n// actually executes but the model is absent). Keyed by the run's thread id so\n// concurrent runs don't clobber each other.\n// ---------------------------------------------------------------------------\nconst a2uiToolsByThread = new Map<string, any>();\nconst A2UI_DEFAULT_THREAD_KEY = \"__copilotkit_a2ui_default__\";\nconst a2uiThreadKey = (state: any): string =>\n  (state?.thread_id as string) || A2UI_DEFAULT_THREAD_KEY;\n\n/**\n * Merge AG-UI defaults with CopilotKit overrides. Dictionaries recurse; every\n * conflicting leaf uses CopilotKit, including false, null and empty values.\n * Arrays and serialized strings are atomic: no positional pairing or decoding.\n */\nconst isPropertyBag = (value: unknown): value is Record<string, unknown> =>\n  value !== null && typeof value === \"object\" && !Array.isArray(value);\n\nconst mergeProperties = (\n  base: Record<string, unknown>,\n  overrides: Record<string, unknown>,\n): Record<string, unknown> =>\n  Object.fromEntries(\n    Array.from(new Set([...Object.keys(base), ...Object.keys(overrides)])).map(\n      (key) => {\n        const value = Object.prototype.hasOwnProperty.call(overrides, key)\n          ? isPropertyBag(base[key]) && isPropertyBag(overrides[key])\n            ? mergeProperties(base[key], overrides[key])\n            : overrides[key]\n          : base[key];\n        return [key, value];\n      },\n    ),\n  );\n\nconst effectiveProperties = (state: unknown): Record<string, unknown> => {\n  const source = isPropertyBag(state) ? state : {};\n  return mergeProperties(\n    isPropertyBag(source[\"ag-ui\"]) ? source[\"ag-ui\"] : {},\n    isPropertyBag(source.copilotkit) ? source.copilotkit : {},\n  );\n};\n\nconst decodeCatalogValue = (value: unknown): unknown => {\n  if (typeof value !== \"string\") return value;\n  try {\n    return JSON.parse(value);\n  } catch {\n    return value;\n  }\n};\n\n/**\n * Find the frontend-registered A2UI catalog wherever it was passed. Returns\n * `{ compositionGuide?, catalogId? }` when a catalog is present, else `null`\n * independently of the explicit tool-injection decision. Two\n * delivery paths, depending on how the agent is served:\n *  - AG-UI native endpoint → `state[\"ag-ui\"].a2ui_schema` (JSON\n *    `{ catalogId, components }`); the toolkit reads it from state itself.\n *  - Runtime proxy → an effective `context` entry describing\n *    the A2UI catalog (catalog id + component schemas as text), passed to the\n *    subagent via `compositionGuide`.\n * `catalogId` binds generated surfaces to the frontend's catalog so BYOC\n * custom catalogs render their own components (not the basic one).\n */\nconst resolveA2uiCatalog = (\n  state: any,\n): { compositionGuide?: string; catalogId?: string } | null => {\n  const properties = effectiveProperties(state);\n  const a2uiSchema = properties.a2ui_schema;\n  if (a2uiSchema) {\n    let catalogId: string | undefined;\n    try {\n      const parsed =\n        typeof a2uiSchema === \"string\" ? JSON.parse(a2uiSchema) : a2uiSchema;\n      catalogId = parsed?.catalogId;\n    } catch {\n      // non-JSON schema — fall back to the toolkit's basic catalog\n    }\n    return { catalogId };\n  }\n  const context = decodeCatalogValue(properties.context);\n  for (const entry of Array.isArray(context) ? context : []) {\n    const description = entry?.description;\n    const value = decodeCatalogValue(entry?.value);\n    if (\n      typeof description !== \"string\" ||\n      typeof value !== \"string\" ||\n      !description.includes(\"A2UI catalog\") ||\n      !value\n    )\n      continue;\n    const match = /^\\s*-\\s+(\\S+)/m.exec(value);\n    return { compositionGuide: value, catalogId: match?.[1] };\n  }\n  return null;\n};\n\n/**\n * The A2UI `injectA2UITool` decision. The `@ag-ui/a2ui-middleware` forwards it on\n * `forwardedProps`, which `ag-ui-langgraph` surfaces into agent state at\n * `state[\"ag-ui\"].inject_a2ui_tool` — present only when the host turned the\n * runtime A2UI tool on (truthy or a custom tool-name string). `undefined` means\n * no signal (off, or no A2UI middleware in the pipeline) → no auto-injection.\n */\nconst a2uiInjectDecision = (state: any): boolean | string | undefined =>\n  effectiveProperties(state).inject_a2ui_tool as boolean | string | undefined;\n\ntype WithJsonSchema<T> = T extends { \"~standard\": infer S }\n  ? Omit<T, \"~standard\"> & {\n      \"~standard\": S &\n        StandardJSONSchemaV1.Props<\n          S extends StandardSchemaV1.Props<infer I, any> ? I : unknown,\n          S extends StandardSchemaV1.Props<any, infer O> ? O : unknown\n        >;\n    }\n  : T;\n\n/**\n * Augment a Standard-Schema–compatible schema (e.g. Zod) with a\n * `~standard.jsonSchema.input` hook so LangGraph's\n * `getJsonSchemaFromSchema` (called from `StateSchema.getJsonSchema`)\n * can serialize the field.\n *\n * Without this, Zod v4 fields carry `~standard.validate` + `vendor` only,\n * and `isStandardJSONSchema()` returns false, so the field is silently\n * dropped from the graph's `output_schema`. That makes AG-UI\n * `STATE_SNAPSHOT` events filter the field out of the payload sent to\n * the frontend even though the underlying thread state has the value.\n *\n * Use this on any custom state field you want visible to the frontend\n * via `useAgent().state.*`.\n *\n * @example\n * ```ts\n * import { zodState } from \"@copilotkit/sdk-js/langgraph\";\n *\n * const stateSchema = z.object({\n *   todos: zodState(z.array(TodoSchema).default(() => [])),\n * });\n * ```\n */\nexport function zodState<T extends object>(schema: T): WithJsonSchema<T> {\n  const std = (schema as { \"~standard\"?: { jsonSchema?: unknown } })[\n    \"~standard\"\n  ];\n  if (std && typeof std === \"object\" && !(\"jsonSchema\" in std)) {\n    let cached: Record<string, unknown> | undefined;\n    std.jsonSchema = {\n      input: () => {\n        if (cached) return cached;\n        // Prefer zod-v4's native `toJSONSchema` when available. Falls back to\n        // an empty object, which is sufficient for the field to appear in the\n        // graph's output_schema (langgraph-api treats it as an opaque field).\n        try {\n          const maybeV4ToJsonSchema = (\n            z as unknown as {\n              toJSONSchema?: (s: unknown) => Record<string, unknown>;\n            }\n          ).toJSONSchema;\n          cached =\n            typeof maybeV4ToJsonSchema === \"function\"\n              ? maybeV4ToJsonSchema(schema)\n              : {};\n        } catch {\n          cached = {};\n        }\n        return cached;\n      },\n    };\n  }\n  return schema as WithJsonSchema<T>;\n}\n\n/**\n * Internal/framework state keys that should never be auto-surfaced to the\n * LLM as user-facing state. These are reducer-managed message buckets,\n * CopilotKit/AG-UI plumbing, or graph-internal scaffolding.\n */\nconst RESERVED_STATE_KEYS: ReadonlySet<string> = new Set([\n  \"messages\",\n  \"copilotkit\",\n  \"ag-ui\",\n  \"tools\",\n  \"structured_response\",\n  \"thread_id\",\n  \"remaining_steps\",\n]);\n\n/**\n * Controls how user-defined state keys are surfaced into the LLM prompt\n * on every model call. Off by default to avoid leaking arbitrary state\n * into prompts; opt in explicitly.\n *\n * - `false` (default) — never surface state.\n * - `true` — every state key not in the reserved internal set and not\n *   prefixed with `_` is JSON-serialized into a \"Current agent state:\"\n *   note appended to the system prompt.\n * - `string[]` — only surface the named keys (use this when you want\n *   explicit control over what the LLM sees, e.g. `[\"liked\", \"todos\"]`).\n */\nexport type ExposeStateOption = boolean | readonly string[];\n\nconst buildStateNote = (\n  state: Record<string, unknown>,\n  expose: ExposeStateOption,\n): string | null => {\n  if (expose === false) return null;\n\n  const allow: ReadonlySet<string> | null = Array.isArray(expose)\n    ? new Set(expose)\n    : null;\n\n  const snapshot: Record<string, unknown> = {};\n  for (const key of Object.keys(state)) {\n    if (\n      allow\n        ? !allow.has(key)\n        : RESERVED_STATE_KEYS.has(key) || key.startsWith(\"_\")\n    ) {\n      continue;\n    }\n    const value = state[key];\n    if (\n      value === undefined ||\n      value === null ||\n      value === \"\" ||\n      (Array.isArray(value) && value.length === 0) ||\n      (typeof value === \"object\" &&\n        !Array.isArray(value) &&\n        Object.keys(value as Record<string, unknown>).length === 0)\n    ) {\n      continue;\n    }\n    snapshot[key] = value;\n  }\n\n  if (Object.keys(snapshot).length === 0) return null;\n\n  let body: string;\n  try {\n    body = JSON.stringify(snapshot, null, 2);\n  } catch {\n    body = String(snapshot);\n  }\n  return `Current agent state:\\n${body}`;\n};\n\nconst applyStateNote = (request: any, expose: ExposeStateOption): any => {\n  const note = buildStateNote(\n    (request.state ?? {}) as Record<string, unknown>,\n    expose,\n  );\n  if (!note) return request;\n\n  const existingMessage = request.systemMessage;\n  if (existingMessage != null) {\n    const separator = existingMessage.text === \"\" ? \"\" : \"\\n\\n\";\n    return {\n      ...request,\n      systemMessage: existingMessage.concat(`${separator}${note}`),\n    };\n  }\n\n  const existing = request.systemPrompt;\n  if (existing == null) {\n    return { ...request, systemPrompt: note };\n  }\n  // existing may be a string OR a SystemMessage\n  const baseText =\n    typeof existing === \"string\"\n      ? existing\n      : typeof existing.content === \"string\"\n        ? existing.content\n        : String(existing.content);\n  return {\n    ...request,\n    systemPrompt: `${baseText}\\n\\n${note}`,\n  };\n};\n\nconst createAppContextBeforeAgent = (state, runtime) => {\n  const messages = state.messages;\n\n  if (!messages || messages.length === 0) {\n    return;\n  }\n\n  // Get app context from runtime\n  const properties = effectiveProperties(state);\n  const appContext = Object.prototype.hasOwnProperty.call(properties, \"context\")\n    ? properties.context\n    : runtime?.context;\n\n  // Check if appContext is missing or empty\n  const isEmptyContext =\n    !appContext ||\n    (typeof appContext === \"string\" && appContext.trim() === \"\") ||\n    (typeof appContext === \"object\" && Object.keys(appContext).length === 0);\n\n  if (isEmptyContext) {\n    return;\n  }\n\n  // Create the context content\n  const contextContent =\n    typeof appContext === \"string\"\n      ? appContext\n      : JSON.stringify(appContext, null, 2);\n  const contextMessageContent = `App Context:\\n${contextContent}`;\n  const contextMessagePrefix = \"App Context:\\n\";\n\n  // Helper to get message content as string\n  const getContentString = (msg: any): string | null => {\n    if (typeof msg.content === \"string\") return msg.content;\n    if (Array.isArray(msg.content) && msg.content[0]?.text)\n      return msg.content[0].text;\n    return null;\n  };\n\n  // Find the first system/developer message (not our context message) to determine\n  // where to insert our context message (right after it)\n  let firstSystemIndex = -1;\n\n  for (let i = 0; i < messages.length; i++) {\n    const msg = messages[i];\n    const type = msg._getType?.();\n    if (type === \"system\" || type === \"developer\") {\n      const content = getContentString(msg);\n      // Skip if this is our own context message\n      if (content?.startsWith(contextMessagePrefix)) {\n        continue;\n      }\n      firstSystemIndex = i;\n      break;\n    }\n  }\n\n  // Check if our context message already exists\n  let existingContextIndex = -1;\n  for (let i = 0; i < messages.length; i++) {\n    const msg = messages[i];\n    const type = msg._getType?.();\n    if (type === \"system\" || type === \"developer\") {\n      const content = getContentString(msg);\n      if (content?.startsWith(contextMessagePrefix)) {\n        existingContextIndex = i;\n        break;\n      }\n    }\n  }\n\n  // Create the context message\n  const contextMessage = new SystemMessage({ content: contextMessageContent });\n\n  let updatedMessages;\n\n  if (existingContextIndex !== -1) {\n    // Replace existing context message\n    updatedMessages = [...messages];\n    updatedMessages[existingContextIndex] = contextMessage;\n  } else {\n    // Insert after the first system message, or at position 0 if no system message\n    const insertIndex = firstSystemIndex !== -1 ? firstSystemIndex + 1 : 0;\n    updatedMessages = [\n      ...messages.slice(0, insertIndex),\n      contextMessage,\n      ...messages.slice(insertIndex),\n    ];\n  }\n\n  return {\n    ...state,\n    messages: updatedMessages,\n  };\n};\n\n/**\n * CopilotKit Middleware for LangGraph agents.\n *\n * Enables:\n * - Dynamic frontend tools from state.tools\n * - Context provided from CopilotKit useCopilotReadable\n *\n * Works with any agent (prebuilt or custom).\n *\n * @example\n * ```typescript\n * import { createAgent } from \"langchain\";\n * import { copilotkitMiddleware } from \"@copilotkit/sdk-js/langgraph\";\n *\n * const agent = createAgent({\n *   model: \"gpt-4o\",\n *   tools: [backendTool],\n *   middleware: [copilotkitMiddleware],\n * });\n * ```\n */\nconst copilotKitStateSchema = z\n  .object({\n    \"ag-ui\": zodState(z.record(z.string(), z.unknown()).optional()),\n    copilotkit: zodState(\n      z\n        .object({\n          actions: z\n            .union([z.array(z.any()), z.literal(false)])\n            .nullable()\n            .optional(),\n          context: z.any().optional(),\n          interceptedToolCalls: z.array(z.any()).optional(),\n          originalAIMessageId: z.string().optional(),\n        })\n        .passthrough()\n        .optional(),\n    ),\n  })\n  .passthrough();\n\nconst isToolCallContentBlock = (block: unknown) =>\n  typeof block === \"object\" &&\n  block !== null &&\n  \"type\" in block &&\n  (block.type === \"tool_call\" || block.type === \"tool_call_chunk\");\n\nconst usesV1ContentBlocks = (responseMetadata: unknown) =>\n  typeof responseMetadata === \"object\" &&\n  responseMetadata !== null &&\n  \"output_version\" in responseMetadata &&\n  responseMetadata.output_version === \"v1\";\n\n/**\n * Rebuilds an AIMessage with `toolCalls` as the source of truth while\n * preserving its non-tool content and metadata. For v1 content blocks, old\n * tool blocks must be removed before construction so they cannot duplicate or\n * override the supplied tool calls when AIMessage synchronizes both fields.\n */\nconst rebuildAIMessageWithToolCalls = (\n  message: AIMessage,\n  toolCalls: AIMessage[\"tool_calls\"],\n) => {\n  let content = message.content;\n  if (\n    usesV1ContentBlocks(message.response_metadata) &&\n    Array.isArray(content)\n  ) {\n    content = content.filter((block) => !isToolCallContentBlock(block));\n  }\n\n  return new AIMessage({\n    content,\n    additional_kwargs: message.additional_kwargs,\n    response_metadata: message.response_metadata,\n    tool_calls: toolCalls,\n    invalid_tool_calls: message.invalid_tool_calls,\n    usage_metadata: message.usage_metadata,\n    id: message.id,\n    name: message.name,\n  });\n};\n\nconst buildMiddlewareInput = (\n  exposeState: ExposeStateOption,\n  a2uiParams?: Omit<A2UIToolParams, \"model\">,\n) => ({\n  name: \"CopilotKitMiddleware\",\n\n  stateSchema: copilotKitStateSchema as unknown as InteropZodObject,\n\n  // Inject frontend tools, surface user state, and forward x-aimock-* headers\n  wrapModelCall: async (request: any, handler: (req: any) => Promise<any>) => {\n    request = applyStateNote(request, exposeState);\n\n    // Forward x-aimock-* headers from the incoming AG-UI request\n    const forwardedHeaders = getForwardedHeaders();\n    if (Object.keys(forwardedHeaders).length > 0) {\n      const existingSettings = request.modelSettings ?? {};\n      const existingHeaders =\n        (existingSettings.headers as Record<string, string>) ?? {};\n      request = {\n        ...request,\n        modelSettings: {\n          ...existingSettings,\n          headers: { ...existingHeaders, ...forwardedHeaders },\n        },\n      };\n    }\n\n    // Opt-in auto-injection of generate_a2ui:\n    // (1) only inject when the A2UI injectA2UITool flag is truthy (forwarded by\n    //     @ag-ui/a2ui-middleware and surfaced at state[\"ag-ui\"].inject_a2ui_tool);\n    // (2) don't double-inject if the agent already defines this tool.\n    // The catalog (when present) only binds surfaces to the FE's catalog; it is\n    // not the gate. The model is inferred from request.model; the built tool is\n    // stashed for wrapToolCall to execute.\n    let a2uiTool: any = null;\n    const decision = a2uiInjectDecision(request.state);\n    if (typeof getA2UITools === \"function\" && decision) {\n      const catalog = resolveA2uiCatalog(request.state);\n      // Shared A2UIToolParams: a single params object owned by the toolkit.\n      // Start from the host overrides (guidelines / catalog id / tool name /\n      // recovery) so a host can steer the subagent, then layer in only what the\n      // host cannot know — the bound model, and the registered catalog id +\n      // compositionGuide — without clobbering any host-set value.\n      const params: A2UIToolParams = {\n        ...a2uiParams,\n        model: request.model,\n      };\n      if (catalog?.catalogId && params.defaultCatalogId == null)\n        params.defaultCatalogId = catalog.catalogId;\n      // Merge the registered catalog schema into any host `guidelines` bag; a\n      // host-set compositionGuide wins, host generation/design overrides stay.\n      if (catalog?.compositionGuide) {\n        const guidelines = { ...params.guidelines };\n        if (guidelines.compositionGuide == null)\n          guidelines.compositionGuide = catalog.compositionGuide;\n        params.guidelines = guidelines;\n      }\n      const candidate = getA2UITools(params);\n      const existingNames = new Set(\n        (request.tools || []).map((t: any) => t?.name),\n      );\n      if (!existingNames.has(candidate.name)) {\n        // ToolNode derives tool runtime state from its config, ignoring an\n        // overridden middleware request.state. Adapt the tool's invocation\n        // config so the toolkit sees the same effective properties.\n        const invoke = candidate.invoke.bind(candidate);\n        candidate.invoke = (input, config) => {\n          const state =\n            config && \"state\" in config && isPropertyBag(config.state)\n              ? config.state\n              : request.state;\n          const properties = effectiveProperties(state);\n          const schema = properties.a2ui_schema;\n          const context = decodeCatalogValue(properties.context);\n          return invoke(input, {\n            ...config,\n            state: {\n              ...state,\n              \"ag-ui\": {\n                ...properties,\n                context: Array.isArray(context) ? context : [],\n                // The toolkit interpolates the native schema into its prompt.\n                a2ui_schema: isPropertyBag(schema)\n                  ? JSON.stringify(schema)\n                  : schema,\n              },\n            },\n          });\n        };\n        a2uiTool = candidate;\n        a2uiToolsByThread.set(a2uiThreadKey(request.state), a2uiTool);\n      }\n    }\n\n    const actions = effectiveProperties(request.state).actions;\n    let frontendTools = Array.isArray(actions) ? actions : [];\n    if (a2uiTool) {\n      // Our generate_a2ui replaces the runtime's render tool — don't advertise\n      // both. Drop the render tool the A2UI middleware injected.\n      const drop = typeof decision === \"string\" ? decision : \"render_a2ui\";\n      frontendTools = frontendTools.filter(\n        (t: any) => (t?.function?.name ?? t?.name) !== drop,\n      );\n    }\n\n    if (frontendTools.length === 0 && !a2uiTool) {\n      return handler(request);\n    }\n\n    const existingTools = request.tools || [];\n    const mergedTools = [\n      ...existingTools,\n      ...(a2uiTool ? [a2uiTool] : []),\n      ...frontendTools,\n    ];\n\n    return handler({\n      ...request,\n      tools: mergedTools,\n    });\n  },\n\n  // Execute the dynamically-advertised generate_a2ui tool. It is not in the\n  // agent's static tool registry, so the tool node cannot run it on its own;\n  // we supply the implementation (built with the inferred model) for that one\n  // tool. This hook's presence also disables createAgent's \"unknown tool\"\n  // guard for dynamically-advertised tools.\n  wrapToolCall: async (request: any, handler: (req: any) => Promise<any>) => {\n    const tool = a2uiToolsByThread.get(a2uiThreadKey(request.state));\n    if (tool && !request.tool && request.toolCall?.name === tool.name) {\n      return handler({ ...request, tool });\n    }\n    return handler(request);\n  },\n\n  beforeAgent: createAppContextBeforeAgent,\n\n  // Restore frontend tool calls to AIMessage before agent exits\n  afterAgent: (state) => {\n    // Drop the bridged A2UI tool for this run — all tool calls for the turn\n    // have executed by now; the next model call re-stashes if needed.\n    a2uiToolsByThread.delete(a2uiThreadKey(state));\n\n    const interceptedToolCalls = state[\"copilotkit\"]?.interceptedToolCalls;\n    const originalMessageId = state[\"copilotkit\"]?.originalAIMessageId;\n\n    if (!interceptedToolCalls?.length || !originalMessageId) {\n      return;\n    }\n\n    let messageFound = false;\n    const updatedMessages = state.messages.map((msg: any) => {\n      if (AIMessage.isInstance(msg) && msg.id === originalMessageId) {\n        messageFound = true;\n        const existingToolCalls = msg.tool_calls || [];\n        return rebuildAIMessageWithToolCalls(msg, [\n          ...existingToolCalls,\n          ...interceptedToolCalls,\n        ]);\n      }\n      return msg;\n    });\n\n    // Only clear intercepted state if we successfully restored the tool calls\n    if (!messageFound) {\n      console.warn(\n        `CopilotKit: Could not find message with id ${originalMessageId} to restore tool calls`,\n      );\n      return;\n    }\n\n    return {\n      messages: updatedMessages,\n      copilotkit: {\n        ...state[\"copilotkit\"],\n        interceptedToolCalls: undefined,\n        originalAIMessageId: undefined,\n      },\n    };\n  },\n\n  // Intercept frontend tool calls after model returns, before ToolNode executes\n  afterModel: (state) => {\n    const actions = effectiveProperties(state).actions;\n    const frontendTools = Array.isArray(actions) ? actions : [];\n    if (frontendTools.length === 0) return;\n\n    const frontendToolNames = new Set(\n      frontendTools.map((t: any) => t.function?.name || t.name),\n    );\n\n    const lastMessage = state.messages[state.messages.length - 1];\n    if (!AIMessage.isInstance(lastMessage) || !lastMessage.tool_calls?.length) {\n      return;\n    }\n\n    const backendToolCalls: any[] = [];\n    const frontendToolCalls: any[] = [];\n\n    for (const call of lastMessage.tool_calls) {\n      if (frontendToolNames.has(call.name)) {\n        frontendToolCalls.push(call);\n      } else {\n        backendToolCalls.push(call);\n      }\n    }\n\n    if (frontendToolCalls.length === 0) return;\n\n    const updatedAIMessage = rebuildAIMessageWithToolCalls(\n      lastMessage,\n      backendToolCalls,\n    );\n\n    return {\n      messages: [...state.messages.slice(0, -1), updatedAIMessage],\n      copilotkit: {\n        ...state[\"copilotkit\"],\n        interceptedToolCalls: frontendToolCalls,\n        originalAIMessageId: lastMessage.id,\n      },\n    };\n  },\n});\n\n/**\n * Build a CopilotKit middleware instance with custom options.\n *\n * Use this when you want to override the default state-exposure behavior\n * (for example to hide a sensitive key, or to use an explicit allowlist), or\n * to steer the auto-injected `generate_a2ui` subagent via `a2uiParams`.\n *\n * `a2uiParams` is an `A2UIToolParams` without `model` (the middleware always\n * injects the bound model). Use it to override the subagent guidelines\n * (`generationGuidelines` / `designGuidelines` / `compositionGuide`),\n * `defaultCatalogId`, `toolName`, `recovery`, etc. on the auto-inject path —\n * which otherwise only ever uses the toolkit defaults. The registered catalog\n * is still folded in, but host-set values win.\n *\n * @example\n * ```typescript\n * import { createCopilotkitMiddleware } from \"@copilotkit/sdk-js/langgraph\";\n *\n * const middleware = createCopilotkitMiddleware({\n *   exposeState: [\"liked\", \"todos\"],\n *   a2uiParams: { guidelines: { designGuidelines: \"...repeating-card layout...\" } },\n * });\n * ```\n */\nexport const createCopilotkitMiddleware = (\n  options: {\n    exposeState?: ExposeStateOption;\n    a2uiParams?: Omit<A2UIToolParams, \"model\">;\n  } = {},\n) => {\n  const exposeState = options.exposeState ?? false;\n  return createMiddleware(\n    buildMiddlewareInput(exposeState, options.a2uiParams) as any,\n  );\n};\n\n/**\n * Default CopilotKit middleware singleton — does NOT surface user state\n * to the LLM. Pass `exposeState: true` (or an allowlist) to\n * {@link createCopilotkitMiddleware} to opt in.\n */\nexport const copilotkitMiddleware = createCopilotkitMiddleware();\n"],"mappings":";;;;;;;;AAiBA,MAAM,oCAAoB,IAAI,KAAkB;AAChD,MAAM,0BAA0B;AAChC,MAAM,iBAAiB,UACpB,OAAO,aAAwB;;;;;;AAOlC,MAAM,iBAAiB,UACrB,UAAU,QAAQ,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,MAAM;AAEtE,MAAM,mBACJ,MACA,cAEA,OAAO,YACL,MAAM,KAAK,IAAI,IAAI,CAAC,GAAG,OAAO,KAAK,KAAK,EAAE,GAAG,OAAO,KAAK,UAAU,CAAC,CAAC,CAAC,CAAC,KACpE,QAAQ;AAMP,QAAO,CAAC,KALM,OAAO,UAAU,eAAe,KAAK,WAAW,IAAI,GAC9D,cAAc,KAAK,KAAK,IAAI,cAAc,UAAU,KAAK,GACvD,gBAAgB,KAAK,MAAM,UAAU,KAAK,GAC1C,UAAU,OACZ,KAAK,KACU;EAEtB,CACF;AAEH,MAAM,uBAAuB,UAA4C;CACvE,MAAM,SAAS,cAAc,MAAM,GAAG,QAAQ,EAAE;AAChD,QAAO,gBACL,cAAc,OAAO,SAAS,GAAG,OAAO,WAAW,EAAE,EACrD,cAAc,OAAO,WAAW,GAAG,OAAO,aAAa,EAAE,CAC1D;;AAGH,MAAM,sBAAsB,UAA4B;AACtD,KAAI,OAAO,UAAU,SAAU,QAAO;AACtC,KAAI;AACF,SAAO,KAAK,MAAM,MAAM;SAClB;AACN,SAAO;;;;;;;;;;;;;;;;AAiBX,MAAM,sBACJ,UAC6D;CAC7D,MAAM,aAAa,oBAAoB,MAAM;CAC7C,MAAM,aAAa,WAAW;AAC9B,KAAI,YAAY;EACd,IAAI;AACJ,MAAI;AAGF,gBADE,OAAO,eAAe,WAAW,KAAK,MAAM,WAAW,GAAG,aACxC;UACd;AAGR,SAAO,EAAE,WAAW;;CAEtB,MAAM,UAAU,mBAAmB,WAAW,QAAQ;AACtD,MAAK,MAAM,SAAS,MAAM,QAAQ,QAAQ,GAAG,UAAU,EAAE,EAAE;EACzD,MAAM,cAAc,OAAO;EAC3B,MAAM,QAAQ,mBAAmB,OAAO,MAAM;AAC9C,MACE,OAAO,gBAAgB,YACvB,OAAO,UAAU,YACjB,CAAC,YAAY,SAAS,eAAe,IACrC,CAAC,MAED;AAEF,SAAO;GAAE,kBAAkB;GAAO,WADpB,iBAAiB,KAAK,MAAM,GACW;GAAI;;AAE3D,QAAO;;;;;;;;;AAUT,MAAM,sBAAsB,UAC1B,oBAAoB,MAAM,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;AAoC7B,SAAgB,SAA2B,QAA8B;CACvE,MAAM,MAAO,OACX;AAEF,KAAI,OAAO,OAAO,QAAQ,YAAY,EAAE,gBAAgB,MAAM;EAC5D,IAAI;AACJ,MAAI,aAAa,EACf,aAAa;AACX,OAAI,OAAQ,QAAO;AAInB,OAAI;IACF,MAAM,sBACJA,IAGA;AACF,aACE,OAAO,wBAAwB,aAC3B,oBAAoB,OAAO,GAC3B,EAAE;WACF;AACN,aAAS,EAAE;;AAEb,UAAO;KAEV;;AAEH,QAAO;;;;;;;AAQT,MAAM,sBAA2C,IAAI,IAAI;CACvD;CACA;CACA;CACA;CACA;CACA;CACA;CACD,CAAC;AAgBF,MAAM,kBACJ,OACA,WACkB;AAClB,KAAI,WAAW,MAAO,QAAO;CAE7B,MAAM,QAAoC,MAAM,QAAQ,OAAO,GAC3D,IAAI,IAAI,OAAO,GACf;CAEJ,MAAM,WAAoC,EAAE;AAC5C,MAAK,MAAM,OAAO,OAAO,KAAK,MAAM,EAAE;AACpC,MACE,QACI,CAAC,MAAM,IAAI,IAAI,GACf,oBAAoB,IAAI,IAAI,IAAI,IAAI,WAAW,IAAI,CAEvD;EAEF,MAAM,QAAQ,MAAM;AACpB,MACE,UAAU,UACV,UAAU,QACV,UAAU,MACT,MAAM,QAAQ,MAAM,IAAI,MAAM,WAAW,KACzC,OAAO,UAAU,YAChB,CAAC,MAAM,QAAQ,MAAM,IACrB,OAAO,KAAK,MAAiC,CAAC,WAAW,EAE3D;AAEF,WAAS,OAAO;;AAGlB,KAAI,OAAO,KAAK,SAAS,CAAC,WAAW,EAAG,QAAO;CAE/C,IAAI;AACJ,KAAI;AACF,SAAO,KAAK,UAAU,UAAU,MAAM,EAAE;SAClC;AACN,SAAO,OAAO,SAAS;;AAEzB,QAAO,yBAAyB;;AAGlC,MAAM,kBAAkB,SAAc,WAAmC;CACvE,MAAM,OAAO,eACV,QAAQ,SAAS,EAAE,EACpB,OACD;AACD,KAAI,CAAC,KAAM,QAAO;CAElB,MAAM,kBAAkB,QAAQ;AAChC,KAAI,mBAAmB,MAAM;EAC3B,MAAM,YAAY,gBAAgB,SAAS,KAAK,KAAK;AACrD,SAAO;GACL,GAAG;GACH,eAAe,gBAAgB,OAAO,GAAG,YAAY,OAAO;GAC7D;;CAGH,MAAM,WAAW,QAAQ;AACzB,KAAI,YAAY,KACd,QAAO;EAAE,GAAG;EAAS,cAAc;EAAM;CAG3C,MAAM,WACJ,OAAO,aAAa,WAChB,WACA,OAAO,SAAS,YAAY,WAC1B,SAAS,UACT,OAAO,SAAS,QAAQ;AAChC,QAAO;EACL,GAAG;EACH,cAAc,GAAG,SAAS,MAAM;EACjC;;AAGH,MAAM,+BAA+B,OAAO,YAAY;CACtD,MAAM,WAAW,MAAM;AAEvB,KAAI,CAAC,YAAY,SAAS,WAAW,EACnC;CAIF,MAAM,aAAa,oBAAoB,MAAM;CAC7C,MAAM,aAAa,OAAO,UAAU,eAAe,KAAK,YAAY,UAAU,GAC1E,WAAW,UACX,SAAS;AAQb,KAJE,CAAC,cACA,OAAO,eAAe,YAAY,WAAW,MAAM,KAAK,MACxD,OAAO,eAAe,YAAY,OAAO,KAAK,WAAW,CAAC,WAAW,EAGtE;CAQF,MAAM,wBAAwB,iBAH5B,OAAO,eAAe,WAClB,aACA,KAAK,UAAU,YAAY,MAAM,EAAE;CAEzC,MAAM,uBAAuB;CAG7B,MAAM,oBAAoB,QAA4B;AACpD,MAAI,OAAO,IAAI,YAAY,SAAU,QAAO,IAAI;AAChD,MAAI,MAAM,QAAQ,IAAI,QAAQ,IAAI,IAAI,QAAQ,IAAI,KAChD,QAAO,IAAI,QAAQ,GAAG;AACxB,SAAO;;CAKT,IAAI,mBAAmB;AAEvB,MAAK,IAAI,IAAI,GAAG,IAAI,SAAS,QAAQ,KAAK;EACxC,MAAM,MAAM,SAAS;EACrB,MAAM,OAAO,IAAI,YAAY;AAC7B,MAAI,SAAS,YAAY,SAAS,aAAa;AAG7C,OAFgB,iBAAiB,IAAI,EAExB,WAAW,qBAAqB,CAC3C;AAEF,sBAAmB;AACnB;;;CAKJ,IAAI,uBAAuB;AAC3B,MAAK,IAAI,IAAI,GAAG,IAAI,SAAS,QAAQ,KAAK;EACxC,MAAM,MAAM,SAAS;EACrB,MAAM,OAAO,IAAI,YAAY;AAC7B,MAAI,SAAS,YAAY,SAAS,aAEhC;OADgB,iBAAiB,IAAI,EACxB,WAAW,qBAAqB,EAAE;AAC7C,2BAAuB;AACvB;;;;CAMN,MAAM,iBAAiB,IAAIC,wBAAc,EAAE,SAAS,uBAAuB,CAAC;CAE5E,IAAI;AAEJ,KAAI,yBAAyB,IAAI;AAE/B,oBAAkB,CAAC,GAAG,SAAS;AAC/B,kBAAgB,wBAAwB;QACnC;EAEL,MAAM,cAAc,qBAAqB,KAAK,mBAAmB,IAAI;AACrE,oBAAkB;GAChB,GAAG,SAAS,MAAM,GAAG,YAAY;GACjC;GACA,GAAG,SAAS,MAAM,YAAY;GAC/B;;AAGH,QAAO;EACL,GAAG;EACH,UAAU;EACX;;;;;;;;;;;;;;;;;;;;;;;AAwBH,MAAM,wBAAwBD,IAC3B,OAAO;CACN,SAAS,SAASA,IAAE,OAAOA,IAAE,QAAQ,EAAEA,IAAE,SAAS,CAAC,CAAC,UAAU,CAAC;CAC/D,YAAY,SACVA,IACG,OAAO;EACN,SAASA,IACN,MAAM,CAACA,IAAE,MAAMA,IAAE,KAAK,CAAC,EAAEA,IAAE,QAAQ,MAAM,CAAC,CAAC,CAC3C,UAAU,CACV,UAAU;EACb,SAASA,IAAE,KAAK,CAAC,UAAU;EAC3B,sBAAsBA,IAAE,MAAMA,IAAE,KAAK,CAAC,CAAC,UAAU;EACjD,qBAAqBA,IAAE,QAAQ,CAAC,UAAU;EAC3C,CAAC,CACD,aAAa,CACb,UAAU,CACd;CACF,CAAC,CACD,aAAa;AAEhB,MAAM,0BAA0B,UAC9B,OAAO,UAAU,YACjB,UAAU,QACV,UAAU,UACT,MAAM,SAAS,eAAe,MAAM,SAAS;AAEhD,MAAM,uBAAuB,qBAC3B,OAAO,qBAAqB,YAC5B,qBAAqB,QACrB,oBAAoB,oBACpB,iBAAiB,mBAAmB;;;;;;;AAQtC,MAAM,iCACJ,SACA,cACG;CACH,IAAI,UAAU,QAAQ;AACtB,KACE,oBAAoB,QAAQ,kBAAkB,IAC9C,MAAM,QAAQ,QAAQ,CAEtB,WAAU,QAAQ,QAAQ,UAAU,CAAC,uBAAuB,MAAM,CAAC;AAGrE,QAAO,IAAIE,oBAAU;EACnB;EACA,mBAAmB,QAAQ;EAC3B,mBAAmB,QAAQ;EAC3B,YAAY;EACZ,oBAAoB,QAAQ;EAC5B,gBAAgB,QAAQ;EACxB,IAAI,QAAQ;EACZ,MAAM,QAAQ;EACf,CAAC;;AAGJ,MAAM,wBACJ,aACA,gBACI;CACJ,MAAM;CAEN,aAAa;CAGb,eAAe,OAAO,SAAc,YAAwC;AAC1E,YAAU,eAAe,SAAS,YAAY;EAG9C,MAAM,mBAAmBC,gDAAqB;AAC9C,MAAI,OAAO,KAAK,iBAAiB,CAAC,SAAS,GAAG;GAC5C,MAAM,mBAAmB,QAAQ,iBAAiB,EAAE;GACpD,MAAM,kBACH,iBAAiB,WAAsC,EAAE;AAC5D,aAAU;IACR,GAAG;IACH,eAAe;KACb,GAAG;KACH,SAAS;MAAE,GAAG;MAAiB,GAAG;MAAkB;KACrD;IACF;;EAUH,IAAI,WAAgB;EACpB,MAAM,WAAW,mBAAmB,QAAQ,MAAM;AAClD,MAAI,OAAOC,kCAAiB,cAAc,UAAU;GAClD,MAAM,UAAU,mBAAmB,QAAQ,MAAM;GAMjD,MAAM,SAAyB;IAC7B,GAAG;IACH,OAAO,QAAQ;IAChB;AACD,OAAI,SAAS,aAAa,OAAO,oBAAoB,KACnD,QAAO,mBAAmB,QAAQ;AAGpC,OAAI,SAAS,kBAAkB;IAC7B,MAAM,aAAa,EAAE,GAAG,OAAO,YAAY;AAC3C,QAAI,WAAW,oBAAoB,KACjC,YAAW,mBAAmB,QAAQ;AACxC,WAAO,aAAa;;GAEtB,MAAM,+CAAyB,OAAO;AAItC,OAAI,CAHkB,IAAI,KACvB,QAAQ,SAAS,EAAE,EAAE,KAAK,MAAW,GAAG,KAAK,CAC/C,CACkB,IAAI,UAAU,KAAK,EAAE;IAItC,MAAM,SAAS,UAAU,OAAO,KAAK,UAAU;AAC/C,cAAU,UAAU,OAAO,WAAW;KACpC,MAAM,QACJ,UAAU,WAAW,UAAU,cAAc,OAAO,MAAM,GACtD,OAAO,QACP,QAAQ;KACd,MAAM,aAAa,oBAAoB,MAAM;KAC7C,MAAM,SAAS,WAAW;KAC1B,MAAM,UAAU,mBAAmB,WAAW,QAAQ;AACtD,YAAO,OAAO,OAAO;MACnB,GAAG;MACH,OAAO;OACL,GAAG;OACH,SAAS;QACP,GAAG;QACH,SAAS,MAAM,QAAQ,QAAQ,GAAG,UAAU,EAAE;QAE9C,aAAa,cAAc,OAAO,GAC9B,KAAK,UAAU,OAAO,GACtB;QACL;OACF;MACF,CAAC;;AAEJ,eAAW;AACX,sBAAkB,IAAI,cAAc,QAAQ,MAAM,EAAE,SAAS;;;EAIjE,MAAM,UAAU,oBAAoB,QAAQ,MAAM,CAAC;EACnD,IAAI,gBAAgB,MAAM,QAAQ,QAAQ,GAAG,UAAU,EAAE;AACzD,MAAI,UAAU;GAGZ,MAAM,OAAO,OAAO,aAAa,WAAW,WAAW;AACvD,mBAAgB,cAAc,QAC3B,OAAY,GAAG,UAAU,QAAQ,GAAG,UAAU,KAChD;;AAGH,MAAI,cAAc,WAAW,KAAK,CAAC,SACjC,QAAO,QAAQ,QAAQ;EAIzB,MAAM,cAAc;GAClB,GAFoB,QAAQ,SAAS,EAAE;GAGvC,GAAI,WAAW,CAAC,SAAS,GAAG,EAAE;GAC9B,GAAG;GACJ;AAED,SAAO,QAAQ;GACb,GAAG;GACH,OAAO;GACR,CAAC;;CAQJ,cAAc,OAAO,SAAc,YAAwC;EACzE,MAAM,OAAO,kBAAkB,IAAI,cAAc,QAAQ,MAAM,CAAC;AAChE,MAAI,QAAQ,CAAC,QAAQ,QAAQ,QAAQ,UAAU,SAAS,KAAK,KAC3D,QAAO,QAAQ;GAAE,GAAG;GAAS;GAAM,CAAC;AAEtC,SAAO,QAAQ,QAAQ;;CAGzB,aAAa;CAGb,aAAa,UAAU;AAGrB,oBAAkB,OAAO,cAAc,MAAM,CAAC;EAE9C,MAAM,uBAAuB,MAAM,eAAe;EAClD,MAAM,oBAAoB,MAAM,eAAe;AAE/C,MAAI,CAAC,sBAAsB,UAAU,CAAC,kBACpC;EAGF,IAAI,eAAe;EACnB,MAAM,kBAAkB,MAAM,SAAS,KAAK,QAAa;AACvD,OAAIF,oBAAU,WAAW,IAAI,IAAI,IAAI,OAAO,mBAAmB;AAC7D,mBAAe;AAEf,WAAO,8BAA8B,KAAK,CACxC,GAFwB,IAAI,cAAc,EAAE,EAG5C,GAAG,qBACJ,CAAC;;AAEJ,UAAO;IACP;AAGF,MAAI,CAAC,cAAc;AACjB,WAAQ,KACN,8CAA8C,kBAAkB,wBACjE;AACD;;AAGF,SAAO;GACL,UAAU;GACV,YAAY;IACV,GAAG,MAAM;IACT,sBAAsB;IACtB,qBAAqB;IACtB;GACF;;CAIH,aAAa,UAAU;EACrB,MAAM,UAAU,oBAAoB,MAAM,CAAC;EAC3C,MAAM,gBAAgB,MAAM,QAAQ,QAAQ,GAAG,UAAU,EAAE;AAC3D,MAAI,cAAc,WAAW,EAAG;EAEhC,MAAM,oBAAoB,IAAI,IAC5B,cAAc,KAAK,MAAW,EAAE,UAAU,QAAQ,EAAE,KAAK,CAC1D;EAED,MAAM,cAAc,MAAM,SAAS,MAAM,SAAS,SAAS;AAC3D,MAAI,CAACA,oBAAU,WAAW,YAAY,IAAI,CAAC,YAAY,YAAY,OACjE;EAGF,MAAM,mBAA0B,EAAE;EAClC,MAAM,oBAA2B,EAAE;AAEnC,OAAK,MAAM,QAAQ,YAAY,WAC7B,KAAI,kBAAkB,IAAI,KAAK,KAAK,CAClC,mBAAkB,KAAK,KAAK;MAE5B,kBAAiB,KAAK,KAAK;AAI/B,MAAI,kBAAkB,WAAW,EAAG;EAEpC,MAAM,mBAAmB,8BACvB,aACA,iBACD;AAED,SAAO;GACL,UAAU,CAAC,GAAG,MAAM,SAAS,MAAM,GAAG,GAAG,EAAE,iBAAiB;GAC5D,YAAY;IACV,GAAG,MAAM;IACT,sBAAsB;IACtB,qBAAqB,YAAY;IAClC;GACF;;CAEJ;;;;;;;;;;;;;;;;;;;;;;;;;AA0BD,MAAa,8BACX,UAGI,EAAE,KACH;AAEH,wCACE,qBAFkB,QAAQ,eAAe,OAEP,QAAQ,WAAW,CACtD;;;;;;;AAQH,MAAa,uBAAuB,4BAA4B"}

package/dist/langgraph/middleware.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"middleware.d.cts","names":[],"sources":["../../src/langgraph/middleware.ts"],"mappings":";;;;;;KAuEK,cAAA,MAAoB,CAAA;EAAY,WAAA;AAAA,IACjC,IAAA,CAAK,CAAA;EACH,WAAA,EAAa,CAAA,GACX,oBAAA,CAAqB,KAAA,CACnB,CAAA,SAAU,gBAAA,CAAiB,KAAA,iBAAsB,CAAA,YACjD,CAAA,SAAU,gBAAA,CAAiB,KAAA,iBAAsB,CAAA;AAAA,IAGvD,CAAA;;;;;;;;;;;;;;;;;;;;;;;;;iBA0BY,QAAA,kBAAA,CAA2B,MAAA,EAAQ,CAAA,GAAI,cAAA,CAAe,CAAA;;;;;;;;;;;AAAtE;;KA2DY,iBAAA;;;;;;;;;;;;;AAAZ;;;;;AAudA;;;;;;;cAAa,0BAAA,GACX,OAAA;EACE,WAAA,GAAc,iBAAA;EACd,UAAA,GAAa,IAAA,CAAK,cAAA;AAAA,MACd,SAAA,CAAA,eAAA,0CADa,sBAAA,CACb,UAAA,GAAA,sBAAA,CAAA,UAAA;;;;;;cAaK,oBAAA,EAAoB,SAAA,CAAA,eAAA,0CAA+B,sBAAA,CAA/B,UAAA,GAAA,sBAAA,CAAA,UAAA"}
+{"version":3,"file":"middleware.d.cts","names":[],"sources":["../../src/langgraph/middleware.ts"],"mappings":";;;;;;KAwHK,cAAA,MAAoB,CAAA;EAAY,WAAA;AAAA,IACjC,IAAA,CAAK,CAAA;EACH,WAAA,EAAa,CAAA,GACX,oBAAA,CAAqB,KAAA,CACnB,CAAA,SAAU,gBAAA,CAAiB,KAAA,iBAAsB,CAAA,YACjD,CAAA,SAAU,gBAAA,CAAiB,KAAA,iBAAsB,CAAA;AAAA,IAGvD,CAAA;;;;;;;;;;;;;;;;;;;;;;;;;iBA0BY,QAAA,kBAAA,CAA2B,MAAA,EAAQ,CAAA,GAAI,cAAA,CAAe,CAAA;;;;;;;;;;;AAAtE;;KA2DY,iBAAA;;;;;;;;;;;;;AAAZ;;;;;AA4fA;;;;;;;cAAa,0BAAA,GACX,OAAA;EACE,WAAA,GAAc,iBAAA;EACd,UAAA,GAAa,IAAA,CAAK,cAAA;AAAA,MACd,SAAA,CAAA,eAAA,0CADa,sBAAA,CACb,UAAA,GAAA,sBAAA,CAAA,UAAA;;;;;;cAaK,oBAAA,EAAoB,SAAA,CAAA,eAAA,0CAA+B,sBAAA,CAA/B,UAAA,GAAA,sBAAA,CAAA,UAAA"}

package/dist/langgraph/middleware.d.mts.map

@@ -1 +1 @@
-{"version":3,"file":"middleware.d.mts","names":[],"sources":["../../src/langgraph/middleware.ts"],"mappings":";;;;;;KAuEK,cAAA,MAAoB,CAAA;EAAY,WAAA;AAAA,IACjC,IAAA,CAAK,CAAA;EACH,WAAA,EAAa,CAAA,GACX,oBAAA,CAAqB,KAAA,CACnB,CAAA,SAAU,gBAAA,CAAiB,KAAA,iBAAsB,CAAA,YACjD,CAAA,SAAU,gBAAA,CAAiB,KAAA,iBAAsB,CAAA;AAAA,IAGvD,CAAA;;;;;;;;;;;;;;;;;;;;;;;;;iBA0BY,QAAA,kBAAA,CAA2B,MAAA,EAAQ,CAAA,GAAI,cAAA,CAAe,CAAA;;;;;;;;;;;AAAtE;;KA2DY,iBAAA;;;;;;;;;;;;;AAAZ;;;;;AAudA;;;;;;;cAAa,0BAAA,GACX,OAAA;EACE,WAAA,GAAc,iBAAA;EACd,UAAA,GAAa,IAAA,CAAK,cAAA;AAAA,MACd,SAAA,CAAA,eAAA,0CADa,sBAAA,CACb,UAAA,GAAA,sBAAA,CAAA,UAAA;;;;;;cAaK,oBAAA,EAAoB,SAAA,CAAA,eAAA,0CAA+B,sBAAA,CAA/B,UAAA,GAAA,sBAAA,CAAA,UAAA"}
+{"version":3,"file":"middleware.d.mts","names":[],"sources":["../../src/langgraph/middleware.ts"],"mappings":";;;;;;KAwHK,cAAA,MAAoB,CAAA;EAAY,WAAA;AAAA,IACjC,IAAA,CAAK,CAAA;EACH,WAAA,EAAa,CAAA,GACX,oBAAA,CAAqB,KAAA,CACnB,CAAA,SAAU,gBAAA,CAAiB,KAAA,iBAAsB,CAAA,YACjD,CAAA,SAAU,gBAAA,CAAiB,KAAA,iBAAsB,CAAA;AAAA,IAGvD,CAAA;;;;;;;;;;;;;;;;;;;;;;;;;iBA0BY,QAAA,kBAAA,CAA2B,MAAA,EAAQ,CAAA,GAAI,cAAA,CAAe,CAAA;;;;;;;;;;;AAAtE;;KA2DY,iBAAA;;;;;;;;;;;;;AAAZ;;;;;AA4fA;;;;;;;cAAa,0BAAA,GACX,OAAA;EACE,WAAA,GAAc,iBAAA;EACd,UAAA,GAAa,IAAA,CAAK,cAAA;AAAA,MACd,SAAA,CAAA,eAAA,0CADa,sBAAA,CACb,UAAA,GAAA,sBAAA,CAAA,UAAA;;;;;;cAaK,oBAAA,EAAoB,SAAA,CAAA,eAAA,0CAA+B,sBAAA,CAA/B,UAAA,GAAA,sBAAA,CAAA,UAAA"}

package/dist/langgraph/middleware.mjs

@@ -8,20 +8,42 @@ const a2uiToolsByThread = /* @__PURE__ */ new Map();
 const A2UI_DEFAULT_THREAD_KEY = "__copilotkit_a2ui_default__";
 const a2uiThreadKey = (state) => state?.thread_id || A2UI_DEFAULT_THREAD_KEY;
 /**
+* Merge AG-UI defaults with CopilotKit overrides. Dictionaries recurse; every
+* conflicting leaf uses CopilotKit, including false, null and empty values.
+* Arrays and serialized strings are atomic: no positional pairing or decoding.
+*/
+const isPropertyBag = (value) => value !== null && typeof value === "object" && !Array.isArray(value);
+const mergeProperties = (base, overrides) => Object.fromEntries(Array.from(new Set([...Object.keys(base), ...Object.keys(overrides)])).map((key) => {
+	return [key, Object.prototype.hasOwnProperty.call(overrides, key) ? isPropertyBag(base[key]) && isPropertyBag(overrides[key]) ? mergeProperties(base[key], overrides[key]) : overrides[key] : base[key]];
+}));
+const effectiveProperties = (state) => {
+	const source = isPropertyBag(state) ? state : {};
+	return mergeProperties(isPropertyBag(source["ag-ui"]) ? source["ag-ui"] : {}, isPropertyBag(source.copilotkit) ? source.copilotkit : {});
+};
+const decodeCatalogValue = (value) => {
+	if (typeof value !== "string") return value;
+	try {
+		return JSON.parse(value);
+	} catch {
+		return value;
+	}
+};
+/**
 * Find the frontend-registered A2UI catalog wherever it was passed. Returns
 * `{ compositionGuide?, catalogId? }` when a catalog is present, else `null`
-* (so the tool is never advertised when the client can't render A2UI). Two
+* independently of the explicit tool-injection decision. Two
 * delivery paths, depending on how the agent is served:
 *  - AG-UI native endpoint → `state["ag-ui"].a2ui_schema` (JSON
 *    `{ catalogId, components }`); the toolkit reads it from state itself.
-*  - CopilotKit runtime proxy → a `state.copilotkit.context` entry describing
+*  - Runtime proxy → an effective `context` entry describing
 *    the A2UI catalog (catalog id + component schemas as text), passed to the
 *    subagent via `compositionGuide`.
 * `catalogId` binds generated surfaces to the frontend's catalog so BYOC
 * custom catalogs render their own components (not the basic one).
 */
 const resolveA2uiCatalog = (state) => {
-	const a2uiSchema = state?.["ag-ui"]?.a2ui_schema;
+	const properties = effectiveProperties(state);
+	const a2uiSchema = properties.a2ui_schema;
 	if (a2uiSchema) {
 		let catalogId;
 		try {
@@ -29,11 +51,11 @@ const resolveA2uiCatalog = (state) => {
 		} catch {}
 		return { catalogId };
 	}
-	const context = state?.copilotkit?.context;
+	const context = decodeCatalogValue(properties.context);
 	for (const entry of Array.isArray(context) ? context : []) {
-		const description = entry?.description ?? "";
-		const value = entry?.value ?? "";
-		if (!description.includes("A2UI catalog") || !value) continue;
+		const description = entry?.description;
+		const value = decodeCatalogValue(entry?.value);
+		if (typeof description !== "string" || typeof value !== "string" || !description.includes("A2UI catalog") || !value) continue;
 		return {
 			compositionGuide: value,
 			catalogId: /^\s*-\s+(\S+)/m.exec(value)?.[1]
@@ -48,7 +70,7 @@ const resolveA2uiCatalog = (state) => {
 * runtime A2UI tool on (truthy or a custom tool-name string). `undefined` means
 * no signal (off, or no A2UI middleware in the pipeline) → no auto-injection.
 */
-const a2uiInjectDecision = (state) => state?.["ag-ui"]?.inject_a2ui_tool;
+const a2uiInjectDecision = (state) => effectiveProperties(state).inject_a2ui_tool;
 /**
 * Augment a Standard-Schema–compatible schema (e.g. Zod) with a
 * `~standard.jsonSchema.input` hook so LangGraph's
@@ -148,7 +170,8 @@ const applyStateNote = (request, expose) => {
 const createAppContextBeforeAgent = (state, runtime) => {
 	const messages = state.messages;
 	if (!messages || messages.length === 0) return;
-	const appContext = state["copilotkit"]?.context ?? runtime?.context;
+	const properties = effectiveProperties(state);
+	const appContext = Object.prototype.hasOwnProperty.call(properties, "context") ? properties.context : runtime?.context;
 	if (!appContext || typeof appContext === "string" && appContext.trim() === "" || typeof appContext === "object" && Object.keys(appContext).length === 0) return;
 	const contextMessageContent = `App Context:\n${typeof appContext === "string" ? appContext : JSON.stringify(appContext, null, 2)}`;
 	const contextMessagePrefix = "App Context:\n";
@@ -217,12 +240,15 @@ const createAppContextBeforeAgent = (state, runtime) => {
 * });
 * ```
 */
-const copilotKitStateSchema = z.object({ copilotkit: zodState(z.object({
-	actions: z.array(z.any()),
-	context: z.any().optional(),
-	interceptedToolCalls: z.array(z.any()).optional(),
-	originalAIMessageId: z.string().optional()
-}).optional()) }).passthrough();
+const copilotKitStateSchema = z.object({
+	"ag-ui": zodState(z.record(z.string(), z.unknown()).optional()),
+	copilotkit: zodState(z.object({
+		actions: z.union([z.array(z.any()), z.literal(false)]).nullable().optional(),
+		context: z.any().optional(),
+		interceptedToolCalls: z.array(z.any()).optional(),
+		originalAIMessageId: z.string().optional()
+	}).passthrough().optional())
+}).passthrough();
 const isToolCallContentBlock = (block) => typeof block === "object" && block !== null && "type" in block && (block.type === "tool_call" || block.type === "tool_call_chunk");
 const usesV1ContentBlocks = (responseMetadata) => typeof responseMetadata === "object" && responseMetadata !== null && "output_version" in responseMetadata && responseMetadata.output_version === "v1";
 /**
@@ -281,11 +307,30 @@ const buildMiddlewareInput = (exposeState, a2uiParams) => ({
 			}
 			const candidate = getA2UITools(params);
 			if (!new Set((request.tools || []).map((t) => t?.name)).has(candidate.name)) {
+				const invoke = candidate.invoke.bind(candidate);
+				candidate.invoke = (input, config) => {
+					const state = config && "state" in config && isPropertyBag(config.state) ? config.state : request.state;
+					const properties = effectiveProperties(state);
+					const schema = properties.a2ui_schema;
+					const context = decodeCatalogValue(properties.context);
+					return invoke(input, {
+						...config,
+						state: {
+							...state,
+							"ag-ui": {
+								...properties,
+								context: Array.isArray(context) ? context : [],
+								a2ui_schema: isPropertyBag(schema) ? JSON.stringify(schema) : schema
+							}
+						}
+					});
+				};
 				a2uiTool = candidate;
 				a2uiToolsByThread.set(a2uiThreadKey(request.state), a2uiTool);
 			}
 		}
-		let frontendTools = request.state["copilotkit"]?.actions ?? [];
+		const actions = effectiveProperties(request.state).actions;
+		let frontendTools = Array.isArray(actions) ? actions : [];
 		if (a2uiTool) {
 			const drop = typeof decision === "string" ? decision : "render_a2ui";
 			frontendTools = frontendTools.filter((t) => (t?.function?.name ?? t?.name) !== drop);
@@ -337,7 +382,8 @@ const buildMiddlewareInput = (exposeState, a2uiParams) => ({
 		};
 	},
 	afterModel: (state) => {
-		const frontendTools = state["copilotkit"]?.actions ?? [];
+		const actions = effectiveProperties(state).actions;
+		const frontendTools = Array.isArray(actions) ? actions : [];
 		if (frontendTools.length === 0) return;
 		const frontendToolNames = new Set(frontendTools.map((t) => t.function?.name || t.name));
 		const lastMessage = state.messages[state.messages.length - 1];

package/dist/langgraph/middleware.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"middleware.mjs","names":[],"sources":["../../src/langgraph/middleware.ts"],"sourcesContent":["import { createMiddleware, AIMessage, SystemMessage } from \"langchain\";\nimport type { InteropZodObject } from \"@langchain/core/utils/types\";\nimport type {\n  StandardJSONSchemaV1,\n  StandardSchemaV1,\n} from \"@standard-schema/spec\";\nimport * as z from \"zod\";\nimport { getA2UITools } from \"@ag-ui/langgraph\";\nimport type { A2UIToolParams } from \"@ag-ui/langgraph\";\nimport { getForwardedHeaders } from \"../header-propagation\";\n\n// ---------------------------------------------------------------------------\n// Auto-A2UI: bridge the inferred model's generate_a2ui tool from wrapModelCall\n// (the only hook that exposes the bound model) to wrapToolCall (where the tool\n// actually executes but the model is absent). Keyed by the run's thread id so\n// concurrent runs don't clobber each other.\n// ---------------------------------------------------------------------------\nconst a2uiToolsByThread = new Map<string, any>();\nconst A2UI_DEFAULT_THREAD_KEY = \"__copilotkit_a2ui_default__\";\nconst a2uiThreadKey = (state: any): string =>\n  (state?.thread_id as string) || A2UI_DEFAULT_THREAD_KEY;\n\n/**\n * Find the frontend-registered A2UI catalog wherever it was passed. Returns\n * `{ compositionGuide?, catalogId? }` when a catalog is present, else `null`\n * (so the tool is never advertised when the client can't render A2UI). Two\n * delivery paths, depending on how the agent is served:\n *  - AG-UI native endpoint → `state[\"ag-ui\"].a2ui_schema` (JSON\n *    `{ catalogId, components }`); the toolkit reads it from state itself.\n *  - CopilotKit runtime proxy → a `state.copilotkit.context` entry describing\n *    the A2UI catalog (catalog id + component schemas as text), passed to the\n *    subagent via `compositionGuide`.\n * `catalogId` binds generated surfaces to the frontend's catalog so BYOC\n * custom catalogs render their own components (not the basic one).\n */\nconst resolveA2uiCatalog = (\n  state: any,\n): { compositionGuide?: string; catalogId?: string } | null => {\n  const a2uiSchema = state?.[\"ag-ui\"]?.a2ui_schema;\n  if (a2uiSchema) {\n    let catalogId: string | undefined;\n    try {\n      const parsed =\n        typeof a2uiSchema === \"string\" ? JSON.parse(a2uiSchema) : a2uiSchema;\n      catalogId = parsed?.catalogId;\n    } catch {\n      // non-JSON schema — fall back to the toolkit's basic catalog\n    }\n    return { catalogId };\n  }\n  const context = state?.copilotkit?.context;\n  for (const entry of Array.isArray(context) ? context : []) {\n    const description = entry?.description ?? \"\";\n    const value = entry?.value ?? \"\";\n    if (!description.includes(\"A2UI catalog\") || !value) continue;\n    const match = /^\\s*-\\s+(\\S+)/m.exec(value);\n    return { compositionGuide: value, catalogId: match?.[1] };\n  }\n  return null;\n};\n\n/**\n * The A2UI `injectA2UITool` decision. The `@ag-ui/a2ui-middleware` forwards it on\n * `forwardedProps`, which `ag-ui-langgraph` surfaces into agent state at\n * `state[\"ag-ui\"].inject_a2ui_tool` — present only when the host turned the\n * runtime A2UI tool on (truthy or a custom tool-name string). `undefined` means\n * no signal (off, or no A2UI middleware in the pipeline) → no auto-injection.\n */\nconst a2uiInjectDecision = (state: any): boolean | string | undefined =>\n  state?.[\"ag-ui\"]?.inject_a2ui_tool;\n\ntype WithJsonSchema<T> = T extends { \"~standard\": infer S }\n  ? Omit<T, \"~standard\"> & {\n      \"~standard\": S &\n        StandardJSONSchemaV1.Props<\n          S extends StandardSchemaV1.Props<infer I, any> ? I : unknown,\n          S extends StandardSchemaV1.Props<any, infer O> ? O : unknown\n        >;\n    }\n  : T;\n\n/**\n * Augment a Standard-Schema–compatible schema (e.g. Zod) with a\n * `~standard.jsonSchema.input` hook so LangGraph's\n * `getJsonSchemaFromSchema` (called from `StateSchema.getJsonSchema`)\n * can serialize the field.\n *\n * Without this, Zod v4 fields carry `~standard.validate` + `vendor` only,\n * and `isStandardJSONSchema()` returns false, so the field is silently\n * dropped from the graph's `output_schema`. That makes AG-UI\n * `STATE_SNAPSHOT` events filter the field out of the payload sent to\n * the frontend even though the underlying thread state has the value.\n *\n * Use this on any custom state field you want visible to the frontend\n * via `useAgent().state.*`.\n *\n * @example\n * ```ts\n * import { zodState } from \"@copilotkit/sdk-js/langgraph\";\n *\n * const stateSchema = z.object({\n *   todos: zodState(z.array(TodoSchema).default(() => [])),\n * });\n * ```\n */\nexport function zodState<T extends object>(schema: T): WithJsonSchema<T> {\n  const std = (schema as { \"~standard\"?: { jsonSchema?: unknown } })[\n    \"~standard\"\n  ];\n  if (std && typeof std === \"object\" && !(\"jsonSchema\" in std)) {\n    let cached: Record<string, unknown> | undefined;\n    std.jsonSchema = {\n      input: () => {\n        if (cached) return cached;\n        // Prefer zod-v4's native `toJSONSchema` when available. Falls back to\n        // an empty object, which is sufficient for the field to appear in the\n        // graph's output_schema (langgraph-api treats it as an opaque field).\n        try {\n          const maybeV4ToJsonSchema = (\n            z as unknown as {\n              toJSONSchema?: (s: unknown) => Record<string, unknown>;\n            }\n          ).toJSONSchema;\n          cached =\n            typeof maybeV4ToJsonSchema === \"function\"\n              ? maybeV4ToJsonSchema(schema)\n              : {};\n        } catch {\n          cached = {};\n        }\n        return cached;\n      },\n    };\n  }\n  return schema as WithJsonSchema<T>;\n}\n\n/**\n * Internal/framework state keys that should never be auto-surfaced to the\n * LLM as user-facing state. These are reducer-managed message buckets,\n * CopilotKit/AG-UI plumbing, or graph-internal scaffolding.\n */\nconst RESERVED_STATE_KEYS: ReadonlySet<string> = new Set([\n  \"messages\",\n  \"copilotkit\",\n  \"ag-ui\",\n  \"tools\",\n  \"structured_response\",\n  \"thread_id\",\n  \"remaining_steps\",\n]);\n\n/**\n * Controls how user-defined state keys are surfaced into the LLM prompt\n * on every model call. Off by default to avoid leaking arbitrary state\n * into prompts; opt in explicitly.\n *\n * - `false` (default) — never surface state.\n * - `true` — every state key not in the reserved internal set and not\n *   prefixed with `_` is JSON-serialized into a \"Current agent state:\"\n *   note appended to the system prompt.\n * - `string[]` — only surface the named keys (use this when you want\n *   explicit control over what the LLM sees, e.g. `[\"liked\", \"todos\"]`).\n */\nexport type ExposeStateOption = boolean | readonly string[];\n\nconst buildStateNote = (\n  state: Record<string, unknown>,\n  expose: ExposeStateOption,\n): string | null => {\n  if (expose === false) return null;\n\n  const allow: ReadonlySet<string> | null = Array.isArray(expose)\n    ? new Set(expose)\n    : null;\n\n  const snapshot: Record<string, unknown> = {};\n  for (const key of Object.keys(state)) {\n    if (\n      allow\n        ? !allow.has(key)\n        : RESERVED_STATE_KEYS.has(key) || key.startsWith(\"_\")\n    ) {\n      continue;\n    }\n    const value = state[key];\n    if (\n      value === undefined ||\n      value === null ||\n      value === \"\" ||\n      (Array.isArray(value) && value.length === 0) ||\n      (typeof value === \"object\" &&\n        !Array.isArray(value) &&\n        Object.keys(value as Record<string, unknown>).length === 0)\n    ) {\n      continue;\n    }\n    snapshot[key] = value;\n  }\n\n  if (Object.keys(snapshot).length === 0) return null;\n\n  let body: string;\n  try {\n    body = JSON.stringify(snapshot, null, 2);\n  } catch {\n    body = String(snapshot);\n  }\n  return `Current agent state:\\n${body}`;\n};\n\nconst applyStateNote = (request: any, expose: ExposeStateOption): any => {\n  const note = buildStateNote(\n    (request.state ?? {}) as Record<string, unknown>,\n    expose,\n  );\n  if (!note) return request;\n\n  const existingMessage = request.systemMessage;\n  if (existingMessage != null) {\n    const separator = existingMessage.text === \"\" ? \"\" : \"\\n\\n\";\n    return {\n      ...request,\n      systemMessage: existingMessage.concat(`${separator}${note}`),\n    };\n  }\n\n  const existing = request.systemPrompt;\n  if (existing == null) {\n    return { ...request, systemPrompt: note };\n  }\n  // existing may be a string OR a SystemMessage\n  const baseText =\n    typeof existing === \"string\"\n      ? existing\n      : typeof existing.content === \"string\"\n        ? existing.content\n        : String(existing.content);\n  return {\n    ...request,\n    systemPrompt: `${baseText}\\n\\n${note}`,\n  };\n};\n\nconst createAppContextBeforeAgent = (state, runtime) => {\n  const messages = state.messages;\n\n  if (!messages || messages.length === 0) {\n    return;\n  }\n\n  // Get app context from runtime\n  const appContext = state[\"copilotkit\"]?.context ?? runtime?.context;\n\n  // Check if appContext is missing or empty\n  const isEmptyContext =\n    !appContext ||\n    (typeof appContext === \"string\" && appContext.trim() === \"\") ||\n    (typeof appContext === \"object\" && Object.keys(appContext).length === 0);\n\n  if (isEmptyContext) {\n    return;\n  }\n\n  // Create the context content\n  const contextContent =\n    typeof appContext === \"string\"\n      ? appContext\n      : JSON.stringify(appContext, null, 2);\n  const contextMessageContent = `App Context:\\n${contextContent}`;\n  const contextMessagePrefix = \"App Context:\\n\";\n\n  // Helper to get message content as string\n  const getContentString = (msg: any): string | null => {\n    if (typeof msg.content === \"string\") return msg.content;\n    if (Array.isArray(msg.content) && msg.content[0]?.text)\n      return msg.content[0].text;\n    return null;\n  };\n\n  // Find the first system/developer message (not our context message) to determine\n  // where to insert our context message (right after it)\n  let firstSystemIndex = -1;\n\n  for (let i = 0; i < messages.length; i++) {\n    const msg = messages[i];\n    const type = msg._getType?.();\n    if (type === \"system\" || type === \"developer\") {\n      const content = getContentString(msg);\n      // Skip if this is our own context message\n      if (content?.startsWith(contextMessagePrefix)) {\n        continue;\n      }\n      firstSystemIndex = i;\n      break;\n    }\n  }\n\n  // Check if our context message already exists\n  let existingContextIndex = -1;\n  for (let i = 0; i < messages.length; i++) {\n    const msg = messages[i];\n    const type = msg._getType?.();\n    if (type === \"system\" || type === \"developer\") {\n      const content = getContentString(msg);\n      if (content?.startsWith(contextMessagePrefix)) {\n        existingContextIndex = i;\n        break;\n      }\n    }\n  }\n\n  // Create the context message\n  const contextMessage = new SystemMessage({ content: contextMessageContent });\n\n  let updatedMessages;\n\n  if (existingContextIndex !== -1) {\n    // Replace existing context message\n    updatedMessages = [...messages];\n    updatedMessages[existingContextIndex] = contextMessage;\n  } else {\n    // Insert after the first system message, or at position 0 if no system message\n    const insertIndex = firstSystemIndex !== -1 ? firstSystemIndex + 1 : 0;\n    updatedMessages = [\n      ...messages.slice(0, insertIndex),\n      contextMessage,\n      ...messages.slice(insertIndex),\n    ];\n  }\n\n  return {\n    ...state,\n    messages: updatedMessages,\n  };\n};\n\n/**\n * CopilotKit Middleware for LangGraph agents.\n *\n * Enables:\n * - Dynamic frontend tools from state.tools\n * - Context provided from CopilotKit useCopilotReadable\n *\n * Works with any agent (prebuilt or custom).\n *\n * @example\n * ```typescript\n * import { createAgent } from \"langchain\";\n * import { copilotkitMiddleware } from \"@copilotkit/sdk-js/langgraph\";\n *\n * const agent = createAgent({\n *   model: \"gpt-4o\",\n *   tools: [backendTool],\n *   middleware: [copilotkitMiddleware],\n * });\n * ```\n */\nconst copilotKitStateSchema = z\n  .object({\n    copilotkit: zodState(\n      z\n        .object({\n          actions: z.array(z.any()),\n          context: z.any().optional(),\n          interceptedToolCalls: z.array(z.any()).optional(),\n          originalAIMessageId: z.string().optional(),\n        })\n        .optional(),\n    ),\n  })\n  .passthrough();\n\nconst isToolCallContentBlock = (block: unknown) =>\n  typeof block === \"object\" &&\n  block !== null &&\n  \"type\" in block &&\n  (block.type === \"tool_call\" || block.type === \"tool_call_chunk\");\n\nconst usesV1ContentBlocks = (responseMetadata: unknown) =>\n  typeof responseMetadata === \"object\" &&\n  responseMetadata !== null &&\n  \"output_version\" in responseMetadata &&\n  responseMetadata.output_version === \"v1\";\n\n/**\n * Rebuilds an AIMessage with `toolCalls` as the source of truth while\n * preserving its non-tool content and metadata. For v1 content blocks, old\n * tool blocks must be removed before construction so they cannot duplicate or\n * override the supplied tool calls when AIMessage synchronizes both fields.\n */\nconst rebuildAIMessageWithToolCalls = (\n  message: AIMessage,\n  toolCalls: AIMessage[\"tool_calls\"],\n) => {\n  let content = message.content;\n  if (\n    usesV1ContentBlocks(message.response_metadata) &&\n    Array.isArray(content)\n  ) {\n    content = content.filter((block) => !isToolCallContentBlock(block));\n  }\n\n  return new AIMessage({\n    content,\n    additional_kwargs: message.additional_kwargs,\n    response_metadata: message.response_metadata,\n    tool_calls: toolCalls,\n    invalid_tool_calls: message.invalid_tool_calls,\n    usage_metadata: message.usage_metadata,\n    id: message.id,\n    name: message.name,\n  });\n};\n\nconst buildMiddlewareInput = (\n  exposeState: ExposeStateOption,\n  a2uiParams?: Omit<A2UIToolParams, \"model\">,\n) => ({\n  name: \"CopilotKitMiddleware\",\n\n  stateSchema: copilotKitStateSchema as unknown as InteropZodObject,\n\n  // Inject frontend tools, surface user state, and forward x-aimock-* headers\n  wrapModelCall: async (request: any, handler: (req: any) => Promise<any>) => {\n    request = applyStateNote(request, exposeState);\n\n    // Forward x-aimock-* headers from the incoming AG-UI request\n    const forwardedHeaders = getForwardedHeaders();\n    if (Object.keys(forwardedHeaders).length > 0) {\n      const existingSettings = request.modelSettings ?? {};\n      const existingHeaders =\n        (existingSettings.headers as Record<string, string>) ?? {};\n      request = {\n        ...request,\n        modelSettings: {\n          ...existingSettings,\n          headers: { ...existingHeaders, ...forwardedHeaders },\n        },\n      };\n    }\n\n    // Opt-in auto-injection of generate_a2ui:\n    // (1) only inject when the A2UI injectA2UITool flag is truthy (forwarded by\n    //     @ag-ui/a2ui-middleware and surfaced at state[\"ag-ui\"].inject_a2ui_tool);\n    // (2) don't double-inject if the agent already defines this tool.\n    // The catalog (when present) only binds surfaces to the FE's catalog; it is\n    // not the gate. The model is inferred from request.model; the built tool is\n    // stashed for wrapToolCall to execute.\n    let a2uiTool: any = null;\n    const decision = a2uiInjectDecision(request.state);\n    if (typeof getA2UITools === \"function\" && decision) {\n      const catalog = resolveA2uiCatalog(request.state);\n      // Shared A2UIToolParams: a single params object owned by the toolkit.\n      // Start from the host overrides (guidelines / catalog id / tool name /\n      // recovery) so a host can steer the subagent, then layer in only what the\n      // host cannot know — the bound model, and the registered catalog id +\n      // compositionGuide — without clobbering any host-set value.\n      const params: A2UIToolParams = {\n        ...a2uiParams,\n        model: request.model,\n      };\n      if (catalog?.catalogId && params.defaultCatalogId == null)\n        params.defaultCatalogId = catalog.catalogId;\n      // Merge the registered catalog schema into any host `guidelines` bag; a\n      // host-set compositionGuide wins, host generation/design overrides stay.\n      if (catalog?.compositionGuide) {\n        const guidelines = { ...params.guidelines };\n        if (guidelines.compositionGuide == null)\n          guidelines.compositionGuide = catalog.compositionGuide;\n        params.guidelines = guidelines;\n      }\n      const candidate = getA2UITools(params);\n      const existingNames = new Set(\n        (request.tools || []).map((t: any) => t?.name),\n      );\n      if (!existingNames.has(candidate.name)) {\n        a2uiTool = candidate;\n        a2uiToolsByThread.set(a2uiThreadKey(request.state), a2uiTool);\n      }\n    }\n\n    let frontendTools = request.state[\"copilotkit\"]?.actions ?? [];\n    if (a2uiTool) {\n      // Our generate_a2ui replaces the runtime's render tool — don't advertise\n      // both. Drop the render tool the A2UI middleware injected.\n      const drop = typeof decision === \"string\" ? decision : \"render_a2ui\";\n      frontendTools = frontendTools.filter(\n        (t: any) => (t?.function?.name ?? t?.name) !== drop,\n      );\n    }\n\n    if (frontendTools.length === 0 && !a2uiTool) {\n      return handler(request);\n    }\n\n    const existingTools = request.tools || [];\n    const mergedTools = [\n      ...existingTools,\n      ...(a2uiTool ? [a2uiTool] : []),\n      ...frontendTools,\n    ];\n\n    return handler({\n      ...request,\n      tools: mergedTools,\n    });\n  },\n\n  // Execute the dynamically-advertised generate_a2ui tool. It is not in the\n  // agent's static tool registry, so the tool node cannot run it on its own;\n  // we supply the implementation (built with the inferred model) for that one\n  // tool. This hook's presence also disables createAgent's \"unknown tool\"\n  // guard for dynamically-advertised tools.\n  wrapToolCall: async (request: any, handler: (req: any) => Promise<any>) => {\n    const tool = a2uiToolsByThread.get(a2uiThreadKey(request.state));\n    if (tool && !request.tool && request.toolCall?.name === tool.name) {\n      return handler({ ...request, tool });\n    }\n    return handler(request);\n  },\n\n  beforeAgent: createAppContextBeforeAgent,\n\n  // Restore frontend tool calls to AIMessage before agent exits\n  afterAgent: (state) => {\n    // Drop the bridged A2UI tool for this run — all tool calls for the turn\n    // have executed by now; the next model call re-stashes if needed.\n    a2uiToolsByThread.delete(a2uiThreadKey(state));\n\n    const interceptedToolCalls = state[\"copilotkit\"]?.interceptedToolCalls;\n    const originalMessageId = state[\"copilotkit\"]?.originalAIMessageId;\n\n    if (!interceptedToolCalls?.length || !originalMessageId) {\n      return;\n    }\n\n    let messageFound = false;\n    const updatedMessages = state.messages.map((msg: any) => {\n      if (AIMessage.isInstance(msg) && msg.id === originalMessageId) {\n        messageFound = true;\n        const existingToolCalls = msg.tool_calls || [];\n        return rebuildAIMessageWithToolCalls(msg, [\n          ...existingToolCalls,\n          ...interceptedToolCalls,\n        ]);\n      }\n      return msg;\n    });\n\n    // Only clear intercepted state if we successfully restored the tool calls\n    if (!messageFound) {\n      console.warn(\n        `CopilotKit: Could not find message with id ${originalMessageId} to restore tool calls`,\n      );\n      return;\n    }\n\n    return {\n      messages: updatedMessages,\n      copilotkit: {\n        ...state[\"copilotkit\"],\n        interceptedToolCalls: undefined,\n        originalAIMessageId: undefined,\n      },\n    };\n  },\n\n  // Intercept frontend tool calls after model returns, before ToolNode executes\n  afterModel: (state) => {\n    const frontendTools = state[\"copilotkit\"]?.actions ?? [];\n    if (frontendTools.length === 0) return;\n\n    const frontendToolNames = new Set(\n      frontendTools.map((t: any) => t.function?.name || t.name),\n    );\n\n    const lastMessage = state.messages[state.messages.length - 1];\n    if (!AIMessage.isInstance(lastMessage) || !lastMessage.tool_calls?.length) {\n      return;\n    }\n\n    const backendToolCalls: any[] = [];\n    const frontendToolCalls: any[] = [];\n\n    for (const call of lastMessage.tool_calls) {\n      if (frontendToolNames.has(call.name)) {\n        frontendToolCalls.push(call);\n      } else {\n        backendToolCalls.push(call);\n      }\n    }\n\n    if (frontendToolCalls.length === 0) return;\n\n    const updatedAIMessage = rebuildAIMessageWithToolCalls(\n      lastMessage,\n      backendToolCalls,\n    );\n\n    return {\n      messages: [...state.messages.slice(0, -1), updatedAIMessage],\n      copilotkit: {\n        ...state[\"copilotkit\"],\n        interceptedToolCalls: frontendToolCalls,\n        originalAIMessageId: lastMessage.id,\n      },\n    };\n  },\n});\n\n/**\n * Build a CopilotKit middleware instance with custom options.\n *\n * Use this when you want to override the default state-exposure behavior\n * (for example to hide a sensitive key, or to use an explicit allowlist), or\n * to steer the auto-injected `generate_a2ui` subagent via `a2uiParams`.\n *\n * `a2uiParams` is an `A2UIToolParams` without `model` (the middleware always\n * injects the bound model). Use it to override the subagent guidelines\n * (`generationGuidelines` / `designGuidelines` / `compositionGuide`),\n * `defaultCatalogId`, `toolName`, `recovery`, etc. on the auto-inject path —\n * which otherwise only ever uses the toolkit defaults. The registered catalog\n * is still folded in, but host-set values win.\n *\n * @example\n * ```typescript\n * import { createCopilotkitMiddleware } from \"@copilotkit/sdk-js/langgraph\";\n *\n * const middleware = createCopilotkitMiddleware({\n *   exposeState: [\"liked\", \"todos\"],\n *   a2uiParams: { guidelines: { designGuidelines: \"...repeating-card layout...\" } },\n * });\n * ```\n */\nexport const createCopilotkitMiddleware = (\n  options: {\n    exposeState?: ExposeStateOption;\n    a2uiParams?: Omit<A2UIToolParams, \"model\">;\n  } = {},\n) => {\n  const exposeState = options.exposeState ?? false;\n  return createMiddleware(\n    buildMiddlewareInput(exposeState, options.a2uiParams) as any,\n  );\n};\n\n/**\n * Default CopilotKit middleware singleton — does NOT surface user state\n * to the LLM. Pass `exposeState: true` (or an allowlist) to\n * {@link createCopilotkitMiddleware} to opt in.\n */\nexport const copilotkitMiddleware = createCopilotkitMiddleware();\n"],"mappings":";;;;;;AAiBA,MAAM,oCAAoB,IAAI,KAAkB;AAChD,MAAM,0BAA0B;AAChC,MAAM,iBAAiB,UACpB,OAAO,aAAwB;;;;;;;;;;;;;;AAelC,MAAM,sBACJ,UAC6D;CAC7D,MAAM,aAAa,QAAQ,UAAU;AACrC,KAAI,YAAY;EACd,IAAI;AACJ,MAAI;AAGF,gBADE,OAAO,eAAe,WAAW,KAAK,MAAM,WAAW,GAAG,aACxC;UACd;AAGR,SAAO,EAAE,WAAW;;CAEtB,MAAM,UAAU,OAAO,YAAY;AACnC,MAAK,MAAM,SAAS,MAAM,QAAQ,QAAQ,GAAG,UAAU,EAAE,EAAE;EACzD,MAAM,cAAc,OAAO,eAAe;EAC1C,MAAM,QAAQ,OAAO,SAAS;AAC9B,MAAI,CAAC,YAAY,SAAS,eAAe,IAAI,CAAC,MAAO;AAErD,SAAO;GAAE,kBAAkB;GAAO,WADpB,iBAAiB,KAAK,MAAM,GACW;GAAI;;AAE3D,QAAO;;;;;;;;;AAUT,MAAM,sBAAsB,UAC1B,QAAQ,UAAU;;;;;;;;;;;;;;;;;;;;;;;;;AAoCpB,SAAgB,SAA2B,QAA8B;CACvE,MAAM,MAAO,OACX;AAEF,KAAI,OAAO,OAAO,QAAQ,YAAY,EAAE,gBAAgB,MAAM;EAC5D,IAAI;AACJ,MAAI,aAAa,EACf,aAAa;AACX,OAAI,OAAQ,QAAO;AAInB,OAAI;IACF,MAAM,sBACJ,EAGA;AACF,aACE,OAAO,wBAAwB,aAC3B,oBAAoB,OAAO,GAC3B,EAAE;WACF;AACN,aAAS,EAAE;;AAEb,UAAO;KAEV;;AAEH,QAAO;;;;;;;AAQT,MAAM,sBAA2C,IAAI,IAAI;CACvD;CACA;CACA;CACA;CACA;CACA;CACA;CACD,CAAC;AAgBF,MAAM,kBACJ,OACA,WACkB;AAClB,KAAI,WAAW,MAAO,QAAO;CAE7B,MAAM,QAAoC,MAAM,QAAQ,OAAO,GAC3D,IAAI,IAAI,OAAO,GACf;CAEJ,MAAM,WAAoC,EAAE;AAC5C,MAAK,MAAM,OAAO,OAAO,KAAK,MAAM,EAAE;AACpC,MACE,QACI,CAAC,MAAM,IAAI,IAAI,GACf,oBAAoB,IAAI,IAAI,IAAI,IAAI,WAAW,IAAI,CAEvD;EAEF,MAAM,QAAQ,MAAM;AACpB,MACE,UAAU,UACV,UAAU,QACV,UAAU,MACT,MAAM,QAAQ,MAAM,IAAI,MAAM,WAAW,KACzC,OAAO,UAAU,YAChB,CAAC,MAAM,QAAQ,MAAM,IACrB,OAAO,KAAK,MAAiC,CAAC,WAAW,EAE3D;AAEF,WAAS,OAAO;;AAGlB,KAAI,OAAO,KAAK,SAAS,CAAC,WAAW,EAAG,QAAO;CAE/C,IAAI;AACJ,KAAI;AACF,SAAO,KAAK,UAAU,UAAU,MAAM,EAAE;SAClC;AACN,SAAO,OAAO,SAAS;;AAEzB,QAAO,yBAAyB;;AAGlC,MAAM,kBAAkB,SAAc,WAAmC;CACvE,MAAM,OAAO,eACV,QAAQ,SAAS,EAAE,EACpB,OACD;AACD,KAAI,CAAC,KAAM,QAAO;CAElB,MAAM,kBAAkB,QAAQ;AAChC,KAAI,mBAAmB,MAAM;EAC3B,MAAM,YAAY,gBAAgB,SAAS,KAAK,KAAK;AACrD,SAAO;GACL,GAAG;GACH,eAAe,gBAAgB,OAAO,GAAG,YAAY,OAAO;GAC7D;;CAGH,MAAM,WAAW,QAAQ;AACzB,KAAI,YAAY,KACd,QAAO;EAAE,GAAG;EAAS,cAAc;EAAM;CAG3C,MAAM,WACJ,OAAO,aAAa,WAChB,WACA,OAAO,SAAS,YAAY,WAC1B,SAAS,UACT,OAAO,SAAS,QAAQ;AAChC,QAAO;EACL,GAAG;EACH,cAAc,GAAG,SAAS,MAAM;EACjC;;AAGH,MAAM,+BAA+B,OAAO,YAAY;CACtD,MAAM,WAAW,MAAM;AAEvB,KAAI,CAAC,YAAY,SAAS,WAAW,EACnC;CAIF,MAAM,aAAa,MAAM,eAAe,WAAW,SAAS;AAQ5D,KAJE,CAAC,cACA,OAAO,eAAe,YAAY,WAAW,MAAM,KAAK,MACxD,OAAO,eAAe,YAAY,OAAO,KAAK,WAAW,CAAC,WAAW,EAGtE;CAQF,MAAM,wBAAwB,iBAH5B,OAAO,eAAe,WAClB,aACA,KAAK,UAAU,YAAY,MAAM,EAAE;CAEzC,MAAM,uBAAuB;CAG7B,MAAM,oBAAoB,QAA4B;AACpD,MAAI,OAAO,IAAI,YAAY,SAAU,QAAO,IAAI;AAChD,MAAI,MAAM,QAAQ,IAAI,QAAQ,IAAI,IAAI,QAAQ,IAAI,KAChD,QAAO,IAAI,QAAQ,GAAG;AACxB,SAAO;;CAKT,IAAI,mBAAmB;AAEvB,MAAK,IAAI,IAAI,GAAG,IAAI,SAAS,QAAQ,KAAK;EACxC,MAAM,MAAM,SAAS;EACrB,MAAM,OAAO,IAAI,YAAY;AAC7B,MAAI,SAAS,YAAY,SAAS,aAAa;AAG7C,OAFgB,iBAAiB,IAAI,EAExB,WAAW,qBAAqB,CAC3C;AAEF,sBAAmB;AACnB;;;CAKJ,IAAI,uBAAuB;AAC3B,MAAK,IAAI,IAAI,GAAG,IAAI,SAAS,QAAQ,KAAK;EACxC,MAAM,MAAM,SAAS;EACrB,MAAM,OAAO,IAAI,YAAY;AAC7B,MAAI,SAAS,YAAY,SAAS,aAEhC;OADgB,iBAAiB,IAAI,EACxB,WAAW,qBAAqB,EAAE;AAC7C,2BAAuB;AACvB;;;;CAMN,MAAM,iBAAiB,IAAI,cAAc,EAAE,SAAS,uBAAuB,CAAC;CAE5E,IAAI;AAEJ,KAAI,yBAAyB,IAAI;AAE/B,oBAAkB,CAAC,GAAG,SAAS;AAC/B,kBAAgB,wBAAwB;QACnC;EAEL,MAAM,cAAc,qBAAqB,KAAK,mBAAmB,IAAI;AACrE,oBAAkB;GAChB,GAAG,SAAS,MAAM,GAAG,YAAY;GACjC;GACA,GAAG,SAAS,MAAM,YAAY;GAC/B;;AAGH,QAAO;EACL,GAAG;EACH,UAAU;EACX;;;;;;;;;;;;;;;;;;;;;;;AAwBH,MAAM,wBAAwB,EAC3B,OAAO,EACN,YAAY,SACV,EACG,OAAO;CACN,SAAS,EAAE,MAAM,EAAE,KAAK,CAAC;CACzB,SAAS,EAAE,KAAK,CAAC,UAAU;CAC3B,sBAAsB,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC,UAAU;CACjD,qBAAqB,EAAE,QAAQ,CAAC,UAAU;CAC3C,CAAC,CACD,UAAU,CACd,EACF,CAAC,CACD,aAAa;AAEhB,MAAM,0BAA0B,UAC9B,OAAO,UAAU,YACjB,UAAU,QACV,UAAU,UACT,MAAM,SAAS,eAAe,MAAM,SAAS;AAEhD,MAAM,uBAAuB,qBAC3B,OAAO,qBAAqB,YAC5B,qBAAqB,QACrB,oBAAoB,oBACpB,iBAAiB,mBAAmB;;;;;;;AAQtC,MAAM,iCACJ,SACA,cACG;CACH,IAAI,UAAU,QAAQ;AACtB,KACE,oBAAoB,QAAQ,kBAAkB,IAC9C,MAAM,QAAQ,QAAQ,CAEtB,WAAU,QAAQ,QAAQ,UAAU,CAAC,uBAAuB,MAAM,CAAC;AAGrE,QAAO,IAAI,UAAU;EACnB;EACA,mBAAmB,QAAQ;EAC3B,mBAAmB,QAAQ;EAC3B,YAAY;EACZ,oBAAoB,QAAQ;EAC5B,gBAAgB,QAAQ;EACxB,IAAI,QAAQ;EACZ,MAAM,QAAQ;EACf,CAAC;;AAGJ,MAAM,wBACJ,aACA,gBACI;CACJ,MAAM;CAEN,aAAa;CAGb,eAAe,OAAO,SAAc,YAAwC;AAC1E,YAAU,eAAe,SAAS,YAAY;EAG9C,MAAM,mBAAmB,qBAAqB;AAC9C,MAAI,OAAO,KAAK,iBAAiB,CAAC,SAAS,GAAG;GAC5C,MAAM,mBAAmB,QAAQ,iBAAiB,EAAE;GACpD,MAAM,kBACH,iBAAiB,WAAsC,EAAE;AAC5D,aAAU;IACR,GAAG;IACH,eAAe;KACb,GAAG;KACH,SAAS;MAAE,GAAG;MAAiB,GAAG;MAAkB;KACrD;IACF;;EAUH,IAAI,WAAgB;EACpB,MAAM,WAAW,mBAAmB,QAAQ,MAAM;AAClD,MAAI,OAAO,iBAAiB,cAAc,UAAU;GAClD,MAAM,UAAU,mBAAmB,QAAQ,MAAM;GAMjD,MAAM,SAAyB;IAC7B,GAAG;IACH,OAAO,QAAQ;IAChB;AACD,OAAI,SAAS,aAAa,OAAO,oBAAoB,KACnD,QAAO,mBAAmB,QAAQ;AAGpC,OAAI,SAAS,kBAAkB;IAC7B,MAAM,aAAa,EAAE,GAAG,OAAO,YAAY;AAC3C,QAAI,WAAW,oBAAoB,KACjC,YAAW,mBAAmB,QAAQ;AACxC,WAAO,aAAa;;GAEtB,MAAM,YAAY,aAAa,OAAO;AAItC,OAAI,CAHkB,IAAI,KACvB,QAAQ,SAAS,EAAE,EAAE,KAAK,MAAW,GAAG,KAAK,CAC/C,CACkB,IAAI,UAAU,KAAK,EAAE;AACtC,eAAW;AACX,sBAAkB,IAAI,cAAc,QAAQ,MAAM,EAAE,SAAS;;;EAIjE,IAAI,gBAAgB,QAAQ,MAAM,eAAe,WAAW,EAAE;AAC9D,MAAI,UAAU;GAGZ,MAAM,OAAO,OAAO,aAAa,WAAW,WAAW;AACvD,mBAAgB,cAAc,QAC3B,OAAY,GAAG,UAAU,QAAQ,GAAG,UAAU,KAChD;;AAGH,MAAI,cAAc,WAAW,KAAK,CAAC,SACjC,QAAO,QAAQ,QAAQ;EAIzB,MAAM,cAAc;GAClB,GAFoB,QAAQ,SAAS,EAAE;GAGvC,GAAI,WAAW,CAAC,SAAS,GAAG,EAAE;GAC9B,GAAG;GACJ;AAED,SAAO,QAAQ;GACb,GAAG;GACH,OAAO;GACR,CAAC;;CAQJ,cAAc,OAAO,SAAc,YAAwC;EACzE,MAAM,OAAO,kBAAkB,IAAI,cAAc,QAAQ,MAAM,CAAC;AAChE,MAAI,QAAQ,CAAC,QAAQ,QAAQ,QAAQ,UAAU,SAAS,KAAK,KAC3D,QAAO,QAAQ;GAAE,GAAG;GAAS;GAAM,CAAC;AAEtC,SAAO,QAAQ,QAAQ;;CAGzB,aAAa;CAGb,aAAa,UAAU;AAGrB,oBAAkB,OAAO,cAAc,MAAM,CAAC;EAE9C,MAAM,uBAAuB,MAAM,eAAe;EAClD,MAAM,oBAAoB,MAAM,eAAe;AAE/C,MAAI,CAAC,sBAAsB,UAAU,CAAC,kBACpC;EAGF,IAAI,eAAe;EACnB,MAAM,kBAAkB,MAAM,SAAS,KAAK,QAAa;AACvD,OAAI,UAAU,WAAW,IAAI,IAAI,IAAI,OAAO,mBAAmB;AAC7D,mBAAe;AAEf,WAAO,8BAA8B,KAAK,CACxC,GAFwB,IAAI,cAAc,EAAE,EAG5C,GAAG,qBACJ,CAAC;;AAEJ,UAAO;IACP;AAGF,MAAI,CAAC,cAAc;AACjB,WAAQ,KACN,8CAA8C,kBAAkB,wBACjE;AACD;;AAGF,SAAO;GACL,UAAU;GACV,YAAY;IACV,GAAG,MAAM;IACT,sBAAsB;IACtB,qBAAqB;IACtB;GACF;;CAIH,aAAa,UAAU;EACrB,MAAM,gBAAgB,MAAM,eAAe,WAAW,EAAE;AACxD,MAAI,cAAc,WAAW,EAAG;EAEhC,MAAM,oBAAoB,IAAI,IAC5B,cAAc,KAAK,MAAW,EAAE,UAAU,QAAQ,EAAE,KAAK,CAC1D;EAED,MAAM,cAAc,MAAM,SAAS,MAAM,SAAS,SAAS;AAC3D,MAAI,CAAC,UAAU,WAAW,YAAY,IAAI,CAAC,YAAY,YAAY,OACjE;EAGF,MAAM,mBAA0B,EAAE;EAClC,MAAM,oBAA2B,EAAE;AAEnC,OAAK,MAAM,QAAQ,YAAY,WAC7B,KAAI,kBAAkB,IAAI,KAAK,KAAK,CAClC,mBAAkB,KAAK,KAAK;MAE5B,kBAAiB,KAAK,KAAK;AAI/B,MAAI,kBAAkB,WAAW,EAAG;EAEpC,MAAM,mBAAmB,8BACvB,aACA,iBACD;AAED,SAAO;GACL,UAAU,CAAC,GAAG,MAAM,SAAS,MAAM,GAAG,GAAG,EAAE,iBAAiB;GAC5D,YAAY;IACV,GAAG,MAAM;IACT,sBAAsB;IACtB,qBAAqB,YAAY;IAClC;GACF;;CAEJ;;;;;;;;;;;;;;;;;;;;;;;;;AA0BD,MAAa,8BACX,UAGI,EAAE,KACH;AAEH,QAAO,iBACL,qBAFkB,QAAQ,eAAe,OAEP,QAAQ,WAAW,CACtD;;;;;;;AAQH,MAAa,uBAAuB,4BAA4B"}
+{"version":3,"file":"middleware.mjs","names":[],"sources":["../../src/langgraph/middleware.ts"],"sourcesContent":["import { createMiddleware, AIMessage, SystemMessage } from \"langchain\";\nimport type { InteropZodObject } from \"@langchain/core/utils/types\";\nimport type {\n  StandardJSONSchemaV1,\n  StandardSchemaV1,\n} from \"@standard-schema/spec\";\nimport * as z from \"zod\";\nimport { getA2UITools } from \"@ag-ui/langgraph\";\nimport type { A2UIToolParams } from \"@ag-ui/langgraph\";\nimport { getForwardedHeaders } from \"../header-propagation\";\n\n// ---------------------------------------------------------------------------\n// Auto-A2UI: bridge the inferred model's generate_a2ui tool from wrapModelCall\n// (the only hook that exposes the bound model) to wrapToolCall (where the tool\n// actually executes but the model is absent). Keyed by the run's thread id so\n// concurrent runs don't clobber each other.\n// ---------------------------------------------------------------------------\nconst a2uiToolsByThread = new Map<string, any>();\nconst A2UI_DEFAULT_THREAD_KEY = \"__copilotkit_a2ui_default__\";\nconst a2uiThreadKey = (state: any): string =>\n  (state?.thread_id as string) || A2UI_DEFAULT_THREAD_KEY;\n\n/**\n * Merge AG-UI defaults with CopilotKit overrides. Dictionaries recurse; every\n * conflicting leaf uses CopilotKit, including false, null and empty values.\n * Arrays and serialized strings are atomic: no positional pairing or decoding.\n */\nconst isPropertyBag = (value: unknown): value is Record<string, unknown> =>\n  value !== null && typeof value === \"object\" && !Array.isArray(value);\n\nconst mergeProperties = (\n  base: Record<string, unknown>,\n  overrides: Record<string, unknown>,\n): Record<string, unknown> =>\n  Object.fromEntries(\n    Array.from(new Set([...Object.keys(base), ...Object.keys(overrides)])).map(\n      (key) => {\n        const value = Object.prototype.hasOwnProperty.call(overrides, key)\n          ? isPropertyBag(base[key]) && isPropertyBag(overrides[key])\n            ? mergeProperties(base[key], overrides[key])\n            : overrides[key]\n          : base[key];\n        return [key, value];\n      },\n    ),\n  );\n\nconst effectiveProperties = (state: unknown): Record<string, unknown> => {\n  const source = isPropertyBag(state) ? state : {};\n  return mergeProperties(\n    isPropertyBag(source[\"ag-ui\"]) ? source[\"ag-ui\"] : {},\n    isPropertyBag(source.copilotkit) ? source.copilotkit : {},\n  );\n};\n\nconst decodeCatalogValue = (value: unknown): unknown => {\n  if (typeof value !== \"string\") return value;\n  try {\n    return JSON.parse(value);\n  } catch {\n    return value;\n  }\n};\n\n/**\n * Find the frontend-registered A2UI catalog wherever it was passed. Returns\n * `{ compositionGuide?, catalogId? }` when a catalog is present, else `null`\n * independently of the explicit tool-injection decision. Two\n * delivery paths, depending on how the agent is served:\n *  - AG-UI native endpoint → `state[\"ag-ui\"].a2ui_schema` (JSON\n *    `{ catalogId, components }`); the toolkit reads it from state itself.\n *  - Runtime proxy → an effective `context` entry describing\n *    the A2UI catalog (catalog id + component schemas as text), passed to the\n *    subagent via `compositionGuide`.\n * `catalogId` binds generated surfaces to the frontend's catalog so BYOC\n * custom catalogs render their own components (not the basic one).\n */\nconst resolveA2uiCatalog = (\n  state: any,\n): { compositionGuide?: string; catalogId?: string } | null => {\n  const properties = effectiveProperties(state);\n  const a2uiSchema = properties.a2ui_schema;\n  if (a2uiSchema) {\n    let catalogId: string | undefined;\n    try {\n      const parsed =\n        typeof a2uiSchema === \"string\" ? JSON.parse(a2uiSchema) : a2uiSchema;\n      catalogId = parsed?.catalogId;\n    } catch {\n      // non-JSON schema — fall back to the toolkit's basic catalog\n    }\n    return { catalogId };\n  }\n  const context = decodeCatalogValue(properties.context);\n  for (const entry of Array.isArray(context) ? context : []) {\n    const description = entry?.description;\n    const value = decodeCatalogValue(entry?.value);\n    if (\n      typeof description !== \"string\" ||\n      typeof value !== \"string\" ||\n      !description.includes(\"A2UI catalog\") ||\n      !value\n    )\n      continue;\n    const match = /^\\s*-\\s+(\\S+)/m.exec(value);\n    return { compositionGuide: value, catalogId: match?.[1] };\n  }\n  return null;\n};\n\n/**\n * The A2UI `injectA2UITool` decision. The `@ag-ui/a2ui-middleware` forwards it on\n * `forwardedProps`, which `ag-ui-langgraph` surfaces into agent state at\n * `state[\"ag-ui\"].inject_a2ui_tool` — present only when the host turned the\n * runtime A2UI tool on (truthy or a custom tool-name string). `undefined` means\n * no signal (off, or no A2UI middleware in the pipeline) → no auto-injection.\n */\nconst a2uiInjectDecision = (state: any): boolean | string | undefined =>\n  effectiveProperties(state).inject_a2ui_tool as boolean | string | undefined;\n\ntype WithJsonSchema<T> = T extends { \"~standard\": infer S }\n  ? Omit<T, \"~standard\"> & {\n      \"~standard\": S &\n        StandardJSONSchemaV1.Props<\n          S extends StandardSchemaV1.Props<infer I, any> ? I : unknown,\n          S extends StandardSchemaV1.Props<any, infer O> ? O : unknown\n        >;\n    }\n  : T;\n\n/**\n * Augment a Standard-Schema–compatible schema (e.g. Zod) with a\n * `~standard.jsonSchema.input` hook so LangGraph's\n * `getJsonSchemaFromSchema` (called from `StateSchema.getJsonSchema`)\n * can serialize the field.\n *\n * Without this, Zod v4 fields carry `~standard.validate` + `vendor` only,\n * and `isStandardJSONSchema()` returns false, so the field is silently\n * dropped from the graph's `output_schema`. That makes AG-UI\n * `STATE_SNAPSHOT` events filter the field out of the payload sent to\n * the frontend even though the underlying thread state has the value.\n *\n * Use this on any custom state field you want visible to the frontend\n * via `useAgent().state.*`.\n *\n * @example\n * ```ts\n * import { zodState } from \"@copilotkit/sdk-js/langgraph\";\n *\n * const stateSchema = z.object({\n *   todos: zodState(z.array(TodoSchema).default(() => [])),\n * });\n * ```\n */\nexport function zodState<T extends object>(schema: T): WithJsonSchema<T> {\n  const std = (schema as { \"~standard\"?: { jsonSchema?: unknown } })[\n    \"~standard\"\n  ];\n  if (std && typeof std === \"object\" && !(\"jsonSchema\" in std)) {\n    let cached: Record<string, unknown> | undefined;\n    std.jsonSchema = {\n      input: () => {\n        if (cached) return cached;\n        // Prefer zod-v4's native `toJSONSchema` when available. Falls back to\n        // an empty object, which is sufficient for the field to appear in the\n        // graph's output_schema (langgraph-api treats it as an opaque field).\n        try {\n          const maybeV4ToJsonSchema = (\n            z as unknown as {\n              toJSONSchema?: (s: unknown) => Record<string, unknown>;\n            }\n          ).toJSONSchema;\n          cached =\n            typeof maybeV4ToJsonSchema === \"function\"\n              ? maybeV4ToJsonSchema(schema)\n              : {};\n        } catch {\n          cached = {};\n        }\n        return cached;\n      },\n    };\n  }\n  return schema as WithJsonSchema<T>;\n}\n\n/**\n * Internal/framework state keys that should never be auto-surfaced to the\n * LLM as user-facing state. These are reducer-managed message buckets,\n * CopilotKit/AG-UI plumbing, or graph-internal scaffolding.\n */\nconst RESERVED_STATE_KEYS: ReadonlySet<string> = new Set([\n  \"messages\",\n  \"copilotkit\",\n  \"ag-ui\",\n  \"tools\",\n  \"structured_response\",\n  \"thread_id\",\n  \"remaining_steps\",\n]);\n\n/**\n * Controls how user-defined state keys are surfaced into the LLM prompt\n * on every model call. Off by default to avoid leaking arbitrary state\n * into prompts; opt in explicitly.\n *\n * - `false` (default) — never surface state.\n * - `true` — every state key not in the reserved internal set and not\n *   prefixed with `_` is JSON-serialized into a \"Current agent state:\"\n *   note appended to the system prompt.\n * - `string[]` — only surface the named keys (use this when you want\n *   explicit control over what the LLM sees, e.g. `[\"liked\", \"todos\"]`).\n */\nexport type ExposeStateOption = boolean | readonly string[];\n\nconst buildStateNote = (\n  state: Record<string, unknown>,\n  expose: ExposeStateOption,\n): string | null => {\n  if (expose === false) return null;\n\n  const allow: ReadonlySet<string> | null = Array.isArray(expose)\n    ? new Set(expose)\n    : null;\n\n  const snapshot: Record<string, unknown> = {};\n  for (const key of Object.keys(state)) {\n    if (\n      allow\n        ? !allow.has(key)\n        : RESERVED_STATE_KEYS.has(key) || key.startsWith(\"_\")\n    ) {\n      continue;\n    }\n    const value = state[key];\n    if (\n      value === undefined ||\n      value === null ||\n      value === \"\" ||\n      (Array.isArray(value) && value.length === 0) ||\n      (typeof value === \"object\" &&\n        !Array.isArray(value) &&\n        Object.keys(value as Record<string, unknown>).length === 0)\n    ) {\n      continue;\n    }\n    snapshot[key] = value;\n  }\n\n  if (Object.keys(snapshot).length === 0) return null;\n\n  let body: string;\n  try {\n    body = JSON.stringify(snapshot, null, 2);\n  } catch {\n    body = String(snapshot);\n  }\n  return `Current agent state:\\n${body}`;\n};\n\nconst applyStateNote = (request: any, expose: ExposeStateOption): any => {\n  const note = buildStateNote(\n    (request.state ?? {}) as Record<string, unknown>,\n    expose,\n  );\n  if (!note) return request;\n\n  const existingMessage = request.systemMessage;\n  if (existingMessage != null) {\n    const separator = existingMessage.text === \"\" ? \"\" : \"\\n\\n\";\n    return {\n      ...request,\n      systemMessage: existingMessage.concat(`${separator}${note}`),\n    };\n  }\n\n  const existing = request.systemPrompt;\n  if (existing == null) {\n    return { ...request, systemPrompt: note };\n  }\n  // existing may be a string OR a SystemMessage\n  const baseText =\n    typeof existing === \"string\"\n      ? existing\n      : typeof existing.content === \"string\"\n        ? existing.content\n        : String(existing.content);\n  return {\n    ...request,\n    systemPrompt: `${baseText}\\n\\n${note}`,\n  };\n};\n\nconst createAppContextBeforeAgent = (state, runtime) => {\n  const messages = state.messages;\n\n  if (!messages || messages.length === 0) {\n    return;\n  }\n\n  // Get app context from runtime\n  const properties = effectiveProperties(state);\n  const appContext = Object.prototype.hasOwnProperty.call(properties, \"context\")\n    ? properties.context\n    : runtime?.context;\n\n  // Check if appContext is missing or empty\n  const isEmptyContext =\n    !appContext ||\n    (typeof appContext === \"string\" && appContext.trim() === \"\") ||\n    (typeof appContext === \"object\" && Object.keys(appContext).length === 0);\n\n  if (isEmptyContext) {\n    return;\n  }\n\n  // Create the context content\n  const contextContent =\n    typeof appContext === \"string\"\n      ? appContext\n      : JSON.stringify(appContext, null, 2);\n  const contextMessageContent = `App Context:\\n${contextContent}`;\n  const contextMessagePrefix = \"App Context:\\n\";\n\n  // Helper to get message content as string\n  const getContentString = (msg: any): string | null => {\n    if (typeof msg.content === \"string\") return msg.content;\n    if (Array.isArray(msg.content) && msg.content[0]?.text)\n      return msg.content[0].text;\n    return null;\n  };\n\n  // Find the first system/developer message (not our context message) to determine\n  // where to insert our context message (right after it)\n  let firstSystemIndex = -1;\n\n  for (let i = 0; i < messages.length; i++) {\n    const msg = messages[i];\n    const type = msg._getType?.();\n    if (type === \"system\" || type === \"developer\") {\n      const content = getContentString(msg);\n      // Skip if this is our own context message\n      if (content?.startsWith(contextMessagePrefix)) {\n        continue;\n      }\n      firstSystemIndex = i;\n      break;\n    }\n  }\n\n  // Check if our context message already exists\n  let existingContextIndex = -1;\n  for (let i = 0; i < messages.length; i++) {\n    const msg = messages[i];\n    const type = msg._getType?.();\n    if (type === \"system\" || type === \"developer\") {\n      const content = getContentString(msg);\n      if (content?.startsWith(contextMessagePrefix)) {\n        existingContextIndex = i;\n        break;\n      }\n    }\n  }\n\n  // Create the context message\n  const contextMessage = new SystemMessage({ content: contextMessageContent });\n\n  let updatedMessages;\n\n  if (existingContextIndex !== -1) {\n    // Replace existing context message\n    updatedMessages = [...messages];\n    updatedMessages[existingContextIndex] = contextMessage;\n  } else {\n    // Insert after the first system message, or at position 0 if no system message\n    const insertIndex = firstSystemIndex !== -1 ? firstSystemIndex + 1 : 0;\n    updatedMessages = [\n      ...messages.slice(0, insertIndex),\n      contextMessage,\n      ...messages.slice(insertIndex),\n    ];\n  }\n\n  return {\n    ...state,\n    messages: updatedMessages,\n  };\n};\n\n/**\n * CopilotKit Middleware for LangGraph agents.\n *\n * Enables:\n * - Dynamic frontend tools from state.tools\n * - Context provided from CopilotKit useCopilotReadable\n *\n * Works with any agent (prebuilt or custom).\n *\n * @example\n * ```typescript\n * import { createAgent } from \"langchain\";\n * import { copilotkitMiddleware } from \"@copilotkit/sdk-js/langgraph\";\n *\n * const agent = createAgent({\n *   model: \"gpt-4o\",\n *   tools: [backendTool],\n *   middleware: [copilotkitMiddleware],\n * });\n * ```\n */\nconst copilotKitStateSchema = z\n  .object({\n    \"ag-ui\": zodState(z.record(z.string(), z.unknown()).optional()),\n    copilotkit: zodState(\n      z\n        .object({\n          actions: z\n            .union([z.array(z.any()), z.literal(false)])\n            .nullable()\n            .optional(),\n          context: z.any().optional(),\n          interceptedToolCalls: z.array(z.any()).optional(),\n          originalAIMessageId: z.string().optional(),\n        })\n        .passthrough()\n        .optional(),\n    ),\n  })\n  .passthrough();\n\nconst isToolCallContentBlock = (block: unknown) =>\n  typeof block === \"object\" &&\n  block !== null &&\n  \"type\" in block &&\n  (block.type === \"tool_call\" || block.type === \"tool_call_chunk\");\n\nconst usesV1ContentBlocks = (responseMetadata: unknown) =>\n  typeof responseMetadata === \"object\" &&\n  responseMetadata !== null &&\n  \"output_version\" in responseMetadata &&\n  responseMetadata.output_version === \"v1\";\n\n/**\n * Rebuilds an AIMessage with `toolCalls` as the source of truth while\n * preserving its non-tool content and metadata. For v1 content blocks, old\n * tool blocks must be removed before construction so they cannot duplicate or\n * override the supplied tool calls when AIMessage synchronizes both fields.\n */\nconst rebuildAIMessageWithToolCalls = (\n  message: AIMessage,\n  toolCalls: AIMessage[\"tool_calls\"],\n) => {\n  let content = message.content;\n  if (\n    usesV1ContentBlocks(message.response_metadata) &&\n    Array.isArray(content)\n  ) {\n    content = content.filter((block) => !isToolCallContentBlock(block));\n  }\n\n  return new AIMessage({\n    content,\n    additional_kwargs: message.additional_kwargs,\n    response_metadata: message.response_metadata,\n    tool_calls: toolCalls,\n    invalid_tool_calls: message.invalid_tool_calls,\n    usage_metadata: message.usage_metadata,\n    id: message.id,\n    name: message.name,\n  });\n};\n\nconst buildMiddlewareInput = (\n  exposeState: ExposeStateOption,\n  a2uiParams?: Omit<A2UIToolParams, \"model\">,\n) => ({\n  name: \"CopilotKitMiddleware\",\n\n  stateSchema: copilotKitStateSchema as unknown as InteropZodObject,\n\n  // Inject frontend tools, surface user state, and forward x-aimock-* headers\n  wrapModelCall: async (request: any, handler: (req: any) => Promise<any>) => {\n    request = applyStateNote(request, exposeState);\n\n    // Forward x-aimock-* headers from the incoming AG-UI request\n    const forwardedHeaders = getForwardedHeaders();\n    if (Object.keys(forwardedHeaders).length > 0) {\n      const existingSettings = request.modelSettings ?? {};\n      const existingHeaders =\n        (existingSettings.headers as Record<string, string>) ?? {};\n      request = {\n        ...request,\n        modelSettings: {\n          ...existingSettings,\n          headers: { ...existingHeaders, ...forwardedHeaders },\n        },\n      };\n    }\n\n    // Opt-in auto-injection of generate_a2ui:\n    // (1) only inject when the A2UI injectA2UITool flag is truthy (forwarded by\n    //     @ag-ui/a2ui-middleware and surfaced at state[\"ag-ui\"].inject_a2ui_tool);\n    // (2) don't double-inject if the agent already defines this tool.\n    // The catalog (when present) only binds surfaces to the FE's catalog; it is\n    // not the gate. The model is inferred from request.model; the built tool is\n    // stashed for wrapToolCall to execute.\n    let a2uiTool: any = null;\n    const decision = a2uiInjectDecision(request.state);\n    if (typeof getA2UITools === \"function\" && decision) {\n      const catalog = resolveA2uiCatalog(request.state);\n      // Shared A2UIToolParams: a single params object owned by the toolkit.\n      // Start from the host overrides (guidelines / catalog id / tool name /\n      // recovery) so a host can steer the subagent, then layer in only what the\n      // host cannot know — the bound model, and the registered catalog id +\n      // compositionGuide — without clobbering any host-set value.\n      const params: A2UIToolParams = {\n        ...a2uiParams,\n        model: request.model,\n      };\n      if (catalog?.catalogId && params.defaultCatalogId == null)\n        params.defaultCatalogId = catalog.catalogId;\n      // Merge the registered catalog schema into any host `guidelines` bag; a\n      // host-set compositionGuide wins, host generation/design overrides stay.\n      if (catalog?.compositionGuide) {\n        const guidelines = { ...params.guidelines };\n        if (guidelines.compositionGuide == null)\n          guidelines.compositionGuide = catalog.compositionGuide;\n        params.guidelines = guidelines;\n      }\n      const candidate = getA2UITools(params);\n      const existingNames = new Set(\n        (request.tools || []).map((t: any) => t?.name),\n      );\n      if (!existingNames.has(candidate.name)) {\n        // ToolNode derives tool runtime state from its config, ignoring an\n        // overridden middleware request.state. Adapt the tool's invocation\n        // config so the toolkit sees the same effective properties.\n        const invoke = candidate.invoke.bind(candidate);\n        candidate.invoke = (input, config) => {\n          const state =\n            config && \"state\" in config && isPropertyBag(config.state)\n              ? config.state\n              : request.state;\n          const properties = effectiveProperties(state);\n          const schema = properties.a2ui_schema;\n          const context = decodeCatalogValue(properties.context);\n          return invoke(input, {\n            ...config,\n            state: {\n              ...state,\n              \"ag-ui\": {\n                ...properties,\n                context: Array.isArray(context) ? context : [],\n                // The toolkit interpolates the native schema into its prompt.\n                a2ui_schema: isPropertyBag(schema)\n                  ? JSON.stringify(schema)\n                  : schema,\n              },\n            },\n          });\n        };\n        a2uiTool = candidate;\n        a2uiToolsByThread.set(a2uiThreadKey(request.state), a2uiTool);\n      }\n    }\n\n    const actions = effectiveProperties(request.state).actions;\n    let frontendTools = Array.isArray(actions) ? actions : [];\n    if (a2uiTool) {\n      // Our generate_a2ui replaces the runtime's render tool — don't advertise\n      // both. Drop the render tool the A2UI middleware injected.\n      const drop = typeof decision === \"string\" ? decision : \"render_a2ui\";\n      frontendTools = frontendTools.filter(\n        (t: any) => (t?.function?.name ?? t?.name) !== drop,\n      );\n    }\n\n    if (frontendTools.length === 0 && !a2uiTool) {\n      return handler(request);\n    }\n\n    const existingTools = request.tools || [];\n    const mergedTools = [\n      ...existingTools,\n      ...(a2uiTool ? [a2uiTool] : []),\n      ...frontendTools,\n    ];\n\n    return handler({\n      ...request,\n      tools: mergedTools,\n    });\n  },\n\n  // Execute the dynamically-advertised generate_a2ui tool. It is not in the\n  // agent's static tool registry, so the tool node cannot run it on its own;\n  // we supply the implementation (built with the inferred model) for that one\n  // tool. This hook's presence also disables createAgent's \"unknown tool\"\n  // guard for dynamically-advertised tools.\n  wrapToolCall: async (request: any, handler: (req: any) => Promise<any>) => {\n    const tool = a2uiToolsByThread.get(a2uiThreadKey(request.state));\n    if (tool && !request.tool && request.toolCall?.name === tool.name) {\n      return handler({ ...request, tool });\n    }\n    return handler(request);\n  },\n\n  beforeAgent: createAppContextBeforeAgent,\n\n  // Restore frontend tool calls to AIMessage before agent exits\n  afterAgent: (state) => {\n    // Drop the bridged A2UI tool for this run — all tool calls for the turn\n    // have executed by now; the next model call re-stashes if needed.\n    a2uiToolsByThread.delete(a2uiThreadKey(state));\n\n    const interceptedToolCalls = state[\"copilotkit\"]?.interceptedToolCalls;\n    const originalMessageId = state[\"copilotkit\"]?.originalAIMessageId;\n\n    if (!interceptedToolCalls?.length || !originalMessageId) {\n      return;\n    }\n\n    let messageFound = false;\n    const updatedMessages = state.messages.map((msg: any) => {\n      if (AIMessage.isInstance(msg) && msg.id === originalMessageId) {\n        messageFound = true;\n        const existingToolCalls = msg.tool_calls || [];\n        return rebuildAIMessageWithToolCalls(msg, [\n          ...existingToolCalls,\n          ...interceptedToolCalls,\n        ]);\n      }\n      return msg;\n    });\n\n    // Only clear intercepted state if we successfully restored the tool calls\n    if (!messageFound) {\n      console.warn(\n        `CopilotKit: Could not find message with id ${originalMessageId} to restore tool calls`,\n      );\n      return;\n    }\n\n    return {\n      messages: updatedMessages,\n      copilotkit: {\n        ...state[\"copilotkit\"],\n        interceptedToolCalls: undefined,\n        originalAIMessageId: undefined,\n      },\n    };\n  },\n\n  // Intercept frontend tool calls after model returns, before ToolNode executes\n  afterModel: (state) => {\n    const actions = effectiveProperties(state).actions;\n    const frontendTools = Array.isArray(actions) ? actions : [];\n    if (frontendTools.length === 0) return;\n\n    const frontendToolNames = new Set(\n      frontendTools.map((t: any) => t.function?.name || t.name),\n    );\n\n    const lastMessage = state.messages[state.messages.length - 1];\n    if (!AIMessage.isInstance(lastMessage) || !lastMessage.tool_calls?.length) {\n      return;\n    }\n\n    const backendToolCalls: any[] = [];\n    const frontendToolCalls: any[] = [];\n\n    for (const call of lastMessage.tool_calls) {\n      if (frontendToolNames.has(call.name)) {\n        frontendToolCalls.push(call);\n      } else {\n        backendToolCalls.push(call);\n      }\n    }\n\n    if (frontendToolCalls.length === 0) return;\n\n    const updatedAIMessage = rebuildAIMessageWithToolCalls(\n      lastMessage,\n      backendToolCalls,\n    );\n\n    return {\n      messages: [...state.messages.slice(0, -1), updatedAIMessage],\n      copilotkit: {\n        ...state[\"copilotkit\"],\n        interceptedToolCalls: frontendToolCalls,\n        originalAIMessageId: lastMessage.id,\n      },\n    };\n  },\n});\n\n/**\n * Build a CopilotKit middleware instance with custom options.\n *\n * Use this when you want to override the default state-exposure behavior\n * (for example to hide a sensitive key, or to use an explicit allowlist), or\n * to steer the auto-injected `generate_a2ui` subagent via `a2uiParams`.\n *\n * `a2uiParams` is an `A2UIToolParams` without `model` (the middleware always\n * injects the bound model). Use it to override the subagent guidelines\n * (`generationGuidelines` / `designGuidelines` / `compositionGuide`),\n * `defaultCatalogId`, `toolName`, `recovery`, etc. on the auto-inject path —\n * which otherwise only ever uses the toolkit defaults. The registered catalog\n * is still folded in, but host-set values win.\n *\n * @example\n * ```typescript\n * import { createCopilotkitMiddleware } from \"@copilotkit/sdk-js/langgraph\";\n *\n * const middleware = createCopilotkitMiddleware({\n *   exposeState: [\"liked\", \"todos\"],\n *   a2uiParams: { guidelines: { designGuidelines: \"...repeating-card layout...\" } },\n * });\n * ```\n */\nexport const createCopilotkitMiddleware = (\n  options: {\n    exposeState?: ExposeStateOption;\n    a2uiParams?: Omit<A2UIToolParams, \"model\">;\n  } = {},\n) => {\n  const exposeState = options.exposeState ?? false;\n  return createMiddleware(\n    buildMiddlewareInput(exposeState, options.a2uiParams) as any,\n  );\n};\n\n/**\n * Default CopilotKit middleware singleton — does NOT surface user state\n * to the LLM. Pass `exposeState: true` (or an allowlist) to\n * {@link createCopilotkitMiddleware} to opt in.\n */\nexport const copilotkitMiddleware = createCopilotkitMiddleware();\n"],"mappings":";;;;;;AAiBA,MAAM,oCAAoB,IAAI,KAAkB;AAChD,MAAM,0BAA0B;AAChC,MAAM,iBAAiB,UACpB,OAAO,aAAwB;;;;;;AAOlC,MAAM,iBAAiB,UACrB,UAAU,QAAQ,OAAO,UAAU,YAAY,CAAC,MAAM,QAAQ,MAAM;AAEtE,MAAM,mBACJ,MACA,cAEA,OAAO,YACL,MAAM,KAAK,IAAI,IAAI,CAAC,GAAG,OAAO,KAAK,KAAK,EAAE,GAAG,OAAO,KAAK,UAAU,CAAC,CAAC,CAAC,CAAC,KACpE,QAAQ;AAMP,QAAO,CAAC,KALM,OAAO,UAAU,eAAe,KAAK,WAAW,IAAI,GAC9D,cAAc,KAAK,KAAK,IAAI,cAAc,UAAU,KAAK,GACvD,gBAAgB,KAAK,MAAM,UAAU,KAAK,GAC1C,UAAU,OACZ,KAAK,KACU;EAEtB,CACF;AAEH,MAAM,uBAAuB,UAA4C;CACvE,MAAM,SAAS,cAAc,MAAM,GAAG,QAAQ,EAAE;AAChD,QAAO,gBACL,cAAc,OAAO,SAAS,GAAG,OAAO,WAAW,EAAE,EACrD,cAAc,OAAO,WAAW,GAAG,OAAO,aAAa,EAAE,CAC1D;;AAGH,MAAM,sBAAsB,UAA4B;AACtD,KAAI,OAAO,UAAU,SAAU,QAAO;AACtC,KAAI;AACF,SAAO,KAAK,MAAM,MAAM;SAClB;AACN,SAAO;;;;;;;;;;;;;;;;AAiBX,MAAM,sBACJ,UAC6D;CAC7D,MAAM,aAAa,oBAAoB,MAAM;CAC7C,MAAM,aAAa,WAAW;AAC9B,KAAI,YAAY;EACd,IAAI;AACJ,MAAI;AAGF,gBADE,OAAO,eAAe,WAAW,KAAK,MAAM,WAAW,GAAG,aACxC;UACd;AAGR,SAAO,EAAE,WAAW;;CAEtB,MAAM,UAAU,mBAAmB,WAAW,QAAQ;AACtD,MAAK,MAAM,SAAS,MAAM,QAAQ,QAAQ,GAAG,UAAU,EAAE,EAAE;EACzD,MAAM,cAAc,OAAO;EAC3B,MAAM,QAAQ,mBAAmB,OAAO,MAAM;AAC9C,MACE,OAAO,gBAAgB,YACvB,OAAO,UAAU,YACjB,CAAC,YAAY,SAAS,eAAe,IACrC,CAAC,MAED;AAEF,SAAO;GAAE,kBAAkB;GAAO,WADpB,iBAAiB,KAAK,MAAM,GACW;GAAI;;AAE3D,QAAO;;;;;;;;;AAUT,MAAM,sBAAsB,UAC1B,oBAAoB,MAAM,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;AAoC7B,SAAgB,SAA2B,QAA8B;CACvE,MAAM,MAAO,OACX;AAEF,KAAI,OAAO,OAAO,QAAQ,YAAY,EAAE,gBAAgB,MAAM;EAC5D,IAAI;AACJ,MAAI,aAAa,EACf,aAAa;AACX,OAAI,OAAQ,QAAO;AAInB,OAAI;IACF,MAAM,sBACJ,EAGA;AACF,aACE,OAAO,wBAAwB,aAC3B,oBAAoB,OAAO,GAC3B,EAAE;WACF;AACN,aAAS,EAAE;;AAEb,UAAO;KAEV;;AAEH,QAAO;;;;;;;AAQT,MAAM,sBAA2C,IAAI,IAAI;CACvD;CACA;CACA;CACA;CACA;CACA;CACA;CACD,CAAC;AAgBF,MAAM,kBACJ,OACA,WACkB;AAClB,KAAI,WAAW,MAAO,QAAO;CAE7B,MAAM,QAAoC,MAAM,QAAQ,OAAO,GAC3D,IAAI,IAAI,OAAO,GACf;CAEJ,MAAM,WAAoC,EAAE;AAC5C,MAAK,MAAM,OAAO,OAAO,KAAK,MAAM,EAAE;AACpC,MACE,QACI,CAAC,MAAM,IAAI,IAAI,GACf,oBAAoB,IAAI,IAAI,IAAI,IAAI,WAAW,IAAI,CAEvD;EAEF,MAAM,QAAQ,MAAM;AACpB,MACE,UAAU,UACV,UAAU,QACV,UAAU,MACT,MAAM,QAAQ,MAAM,IAAI,MAAM,WAAW,KACzC,OAAO,UAAU,YAChB,CAAC,MAAM,QAAQ,MAAM,IACrB,OAAO,KAAK,MAAiC,CAAC,WAAW,EAE3D;AAEF,WAAS,OAAO;;AAGlB,KAAI,OAAO,KAAK,SAAS,CAAC,WAAW,EAAG,QAAO;CAE/C,IAAI;AACJ,KAAI;AACF,SAAO,KAAK,UAAU,UAAU,MAAM,EAAE;SAClC;AACN,SAAO,OAAO,SAAS;;AAEzB,QAAO,yBAAyB;;AAGlC,MAAM,kBAAkB,SAAc,WAAmC;CACvE,MAAM,OAAO,eACV,QAAQ,SAAS,EAAE,EACpB,OACD;AACD,KAAI,CAAC,KAAM,QAAO;CAElB,MAAM,kBAAkB,QAAQ;AAChC,KAAI,mBAAmB,MAAM;EAC3B,MAAM,YAAY,gBAAgB,SAAS,KAAK,KAAK;AACrD,SAAO;GACL,GAAG;GACH,eAAe,gBAAgB,OAAO,GAAG,YAAY,OAAO;GAC7D;;CAGH,MAAM,WAAW,QAAQ;AACzB,KAAI,YAAY,KACd,QAAO;EAAE,GAAG;EAAS,cAAc;EAAM;CAG3C,MAAM,WACJ,OAAO,aAAa,WAChB,WACA,OAAO,SAAS,YAAY,WAC1B,SAAS,UACT,OAAO,SAAS,QAAQ;AAChC,QAAO;EACL,GAAG;EACH,cAAc,GAAG,SAAS,MAAM;EACjC;;AAGH,MAAM,+BAA+B,OAAO,YAAY;CACtD,MAAM,WAAW,MAAM;AAEvB,KAAI,CAAC,YAAY,SAAS,WAAW,EACnC;CAIF,MAAM,aAAa,oBAAoB,MAAM;CAC7C,MAAM,aAAa,OAAO,UAAU,eAAe,KAAK,YAAY,UAAU,GAC1E,WAAW,UACX,SAAS;AAQb,KAJE,CAAC,cACA,OAAO,eAAe,YAAY,WAAW,MAAM,KAAK,MACxD,OAAO,eAAe,YAAY,OAAO,KAAK,WAAW,CAAC,WAAW,EAGtE;CAQF,MAAM,wBAAwB,iBAH5B,OAAO,eAAe,WAClB,aACA,KAAK,UAAU,YAAY,MAAM,EAAE;CAEzC,MAAM,uBAAuB;CAG7B,MAAM,oBAAoB,QAA4B;AACpD,MAAI,OAAO,IAAI,YAAY,SAAU,QAAO,IAAI;AAChD,MAAI,MAAM,QAAQ,IAAI,QAAQ,IAAI,IAAI,QAAQ,IAAI,KAChD,QAAO,IAAI,QAAQ,GAAG;AACxB,SAAO;;CAKT,IAAI,mBAAmB;AAEvB,MAAK,IAAI,IAAI,GAAG,IAAI,SAAS,QAAQ,KAAK;EACxC,MAAM,MAAM,SAAS;EACrB,MAAM,OAAO,IAAI,YAAY;AAC7B,MAAI,SAAS,YAAY,SAAS,aAAa;AAG7C,OAFgB,iBAAiB,IAAI,EAExB,WAAW,qBAAqB,CAC3C;AAEF,sBAAmB;AACnB;;;CAKJ,IAAI,uBAAuB;AAC3B,MAAK,IAAI,IAAI,GAAG,IAAI,SAAS,QAAQ,KAAK;EACxC,MAAM,MAAM,SAAS;EACrB,MAAM,OAAO,IAAI,YAAY;AAC7B,MAAI,SAAS,YAAY,SAAS,aAEhC;OADgB,iBAAiB,IAAI,EACxB,WAAW,qBAAqB,EAAE;AAC7C,2BAAuB;AACvB;;;;CAMN,MAAM,iBAAiB,IAAI,cAAc,EAAE,SAAS,uBAAuB,CAAC;CAE5E,IAAI;AAEJ,KAAI,yBAAyB,IAAI;AAE/B,oBAAkB,CAAC,GAAG,SAAS;AAC/B,kBAAgB,wBAAwB;QACnC;EAEL,MAAM,cAAc,qBAAqB,KAAK,mBAAmB,IAAI;AACrE,oBAAkB;GAChB,GAAG,SAAS,MAAM,GAAG,YAAY;GACjC;GACA,GAAG,SAAS,MAAM,YAAY;GAC/B;;AAGH,QAAO;EACL,GAAG;EACH,UAAU;EACX;;;;;;;;;;;;;;;;;;;;;;;AAwBH,MAAM,wBAAwB,EAC3B,OAAO;CACN,SAAS,SAAS,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,SAAS,CAAC,CAAC,UAAU,CAAC;CAC/D,YAAY,SACV,EACG,OAAO;EACN,SAAS,EACN,MAAM,CAAC,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,EAAE,QAAQ,MAAM,CAAC,CAAC,CAC3C,UAAU,CACV,UAAU;EACb,SAAS,EAAE,KAAK,CAAC,UAAU;EAC3B,sBAAsB,EAAE,MAAM,EAAE,KAAK,CAAC,CAAC,UAAU;EACjD,qBAAqB,EAAE,QAAQ,CAAC,UAAU;EAC3C,CAAC,CACD,aAAa,CACb,UAAU,CACd;CACF,CAAC,CACD,aAAa;AAEhB,MAAM,0BAA0B,UAC9B,OAAO,UAAU,YACjB,UAAU,QACV,UAAU,UACT,MAAM,SAAS,eAAe,MAAM,SAAS;AAEhD,MAAM,uBAAuB,qBAC3B,OAAO,qBAAqB,YAC5B,qBAAqB,QACrB,oBAAoB,oBACpB,iBAAiB,mBAAmB;;;;;;;AAQtC,MAAM,iCACJ,SACA,cACG;CACH,IAAI,UAAU,QAAQ;AACtB,KACE,oBAAoB,QAAQ,kBAAkB,IAC9C,MAAM,QAAQ,QAAQ,CAEtB,WAAU,QAAQ,QAAQ,UAAU,CAAC,uBAAuB,MAAM,CAAC;AAGrE,QAAO,IAAI,UAAU;EACnB;EACA,mBAAmB,QAAQ;EAC3B,mBAAmB,QAAQ;EAC3B,YAAY;EACZ,oBAAoB,QAAQ;EAC5B,gBAAgB,QAAQ;EACxB,IAAI,QAAQ;EACZ,MAAM,QAAQ;EACf,CAAC;;AAGJ,MAAM,wBACJ,aACA,gBACI;CACJ,MAAM;CAEN,aAAa;CAGb,eAAe,OAAO,SAAc,YAAwC;AAC1E,YAAU,eAAe,SAAS,YAAY;EAG9C,MAAM,mBAAmB,qBAAqB;AAC9C,MAAI,OAAO,KAAK,iBAAiB,CAAC,SAAS,GAAG;GAC5C,MAAM,mBAAmB,QAAQ,iBAAiB,EAAE;GACpD,MAAM,kBACH,iBAAiB,WAAsC,EAAE;AAC5D,aAAU;IACR,GAAG;IACH,eAAe;KACb,GAAG;KACH,SAAS;MAAE,GAAG;MAAiB,GAAG;MAAkB;KACrD;IACF;;EAUH,IAAI,WAAgB;EACpB,MAAM,WAAW,mBAAmB,QAAQ,MAAM;AAClD,MAAI,OAAO,iBAAiB,cAAc,UAAU;GAClD,MAAM,UAAU,mBAAmB,QAAQ,MAAM;GAMjD,MAAM,SAAyB;IAC7B,GAAG;IACH,OAAO,QAAQ;IAChB;AACD,OAAI,SAAS,aAAa,OAAO,oBAAoB,KACnD,QAAO,mBAAmB,QAAQ;AAGpC,OAAI,SAAS,kBAAkB;IAC7B,MAAM,aAAa,EAAE,GAAG,OAAO,YAAY;AAC3C,QAAI,WAAW,oBAAoB,KACjC,YAAW,mBAAmB,QAAQ;AACxC,WAAO,aAAa;;GAEtB,MAAM,YAAY,aAAa,OAAO;AAItC,OAAI,CAHkB,IAAI,KACvB,QAAQ,SAAS,EAAE,EAAE,KAAK,MAAW,GAAG,KAAK,CAC/C,CACkB,IAAI,UAAU,KAAK,EAAE;IAItC,MAAM,SAAS,UAAU,OAAO,KAAK,UAAU;AAC/C,cAAU,UAAU,OAAO,WAAW;KACpC,MAAM,QACJ,UAAU,WAAW,UAAU,cAAc,OAAO,MAAM,GACtD,OAAO,QACP,QAAQ;KACd,MAAM,aAAa,oBAAoB,MAAM;KAC7C,MAAM,SAAS,WAAW;KAC1B,MAAM,UAAU,mBAAmB,WAAW,QAAQ;AACtD,YAAO,OAAO,OAAO;MACnB,GAAG;MACH,OAAO;OACL,GAAG;OACH,SAAS;QACP,GAAG;QACH,SAAS,MAAM,QAAQ,QAAQ,GAAG,UAAU,EAAE;QAE9C,aAAa,cAAc,OAAO,GAC9B,KAAK,UAAU,OAAO,GACtB;QACL;OACF;MACF,CAAC;;AAEJ,eAAW;AACX,sBAAkB,IAAI,cAAc,QAAQ,MAAM,EAAE,SAAS;;;EAIjE,MAAM,UAAU,oBAAoB,QAAQ,MAAM,CAAC;EACnD,IAAI,gBAAgB,MAAM,QAAQ,QAAQ,GAAG,UAAU,EAAE;AACzD,MAAI,UAAU;GAGZ,MAAM,OAAO,OAAO,aAAa,WAAW,WAAW;AACvD,mBAAgB,cAAc,QAC3B,OAAY,GAAG,UAAU,QAAQ,GAAG,UAAU,KAChD;;AAGH,MAAI,cAAc,WAAW,KAAK,CAAC,SACjC,QAAO,QAAQ,QAAQ;EAIzB,MAAM,cAAc;GAClB,GAFoB,QAAQ,SAAS,EAAE;GAGvC,GAAI,WAAW,CAAC,SAAS,GAAG,EAAE;GAC9B,GAAG;GACJ;AAED,SAAO,QAAQ;GACb,GAAG;GACH,OAAO;GACR,CAAC;;CAQJ,cAAc,OAAO,SAAc,YAAwC;EACzE,MAAM,OAAO,kBAAkB,IAAI,cAAc,QAAQ,MAAM,CAAC;AAChE,MAAI,QAAQ,CAAC,QAAQ,QAAQ,QAAQ,UAAU,SAAS,KAAK,KAC3D,QAAO,QAAQ;GAAE,GAAG;GAAS;GAAM,CAAC;AAEtC,SAAO,QAAQ,QAAQ;;CAGzB,aAAa;CAGb,aAAa,UAAU;AAGrB,oBAAkB,OAAO,cAAc,MAAM,CAAC;EAE9C,MAAM,uBAAuB,MAAM,eAAe;EAClD,MAAM,oBAAoB,MAAM,eAAe;AAE/C,MAAI,CAAC,sBAAsB,UAAU,CAAC,kBACpC;EAGF,IAAI,eAAe;EACnB,MAAM,kBAAkB,MAAM,SAAS,KAAK,QAAa;AACvD,OAAI,UAAU,WAAW,IAAI,IAAI,IAAI,OAAO,mBAAmB;AAC7D,mBAAe;AAEf,WAAO,8BAA8B,KAAK,CACxC,GAFwB,IAAI,cAAc,EAAE,EAG5C,GAAG,qBACJ,CAAC;;AAEJ,UAAO;IACP;AAGF,MAAI,CAAC,cAAc;AACjB,WAAQ,KACN,8CAA8C,kBAAkB,wBACjE;AACD;;AAGF,SAAO;GACL,UAAU;GACV,YAAY;IACV,GAAG,MAAM;IACT,sBAAsB;IACtB,qBAAqB;IACtB;GACF;;CAIH,aAAa,UAAU;EACrB,MAAM,UAAU,oBAAoB,MAAM,CAAC;EAC3C,MAAM,gBAAgB,MAAM,QAAQ,QAAQ,GAAG,UAAU,EAAE;AAC3D,MAAI,cAAc,WAAW,EAAG;EAEhC,MAAM,oBAAoB,IAAI,IAC5B,cAAc,KAAK,MAAW,EAAE,UAAU,QAAQ,EAAE,KAAK,CAC1D;EAED,MAAM,cAAc,MAAM,SAAS,MAAM,SAAS,SAAS;AAC3D,MAAI,CAAC,UAAU,WAAW,YAAY,IAAI,CAAC,YAAY,YAAY,OACjE;EAGF,MAAM,mBAA0B,EAAE;EAClC,MAAM,oBAA2B,EAAE;AAEnC,OAAK,MAAM,QAAQ,YAAY,WAC7B,KAAI,kBAAkB,IAAI,KAAK,KAAK,CAClC,mBAAkB,KAAK,KAAK;MAE5B,kBAAiB,KAAK,KAAK;AAI/B,MAAI,kBAAkB,WAAW,EAAG;EAEpC,MAAM,mBAAmB,8BACvB,aACA,iBACD;AAED,SAAO;GACL,UAAU,CAAC,GAAG,MAAM,SAAS,MAAM,GAAG,GAAG,EAAE,iBAAiB;GAC5D,YAAY;IACV,GAAG,MAAM;IACT,sBAAsB;IACtB,qBAAqB,YAAY;IAClC;GACF;;CAEJ;;;;;;;;;;;;;;;;;;;;;;;;;AA0BD,MAAa,8BACX,UAGI,EAAE,KACH;AAEH,QAAO,iBACL,qBAFkB,QAAQ,eAAe,OAEP,QAAQ,WAAW,CACtD;;;;;;;AAQH,MAAa,uBAAuB,4BAA4B"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@copilotkit/sdk-js",
-  "version": "1.76.0",
+  "version": "1.77.1",
   "private": false,
   "keywords": [
     "ai",
@@ -48,9 +48,9 @@
     "access": "public"
   },
   "dependencies": {
-    "@ag-ui/langgraph": "0.0.42",
+    "@ag-ui/langgraph": "0.0.44",
     "@standard-schema/spec": "^1.0.0",
-    "@copilotkit/shared": "1.76.0"
+    "@copilotkit/shared": "1.77.1"
   },
   "devDependencies": {
     "@langchain/core": "^1.1.41",

package/src/langgraph/__tests__/middleware.a2ui-params.test.ts

@@ -17,12 +17,17 @@
 
 import { describe, it, expect, beforeEach, vi } from "vitest";
 
-const { captured } = vi.hoisted(() => ({ captured: [] as any[] }));
+import type { A2UIToolParams } from "@ag-ui/langgraph";
+
+const { captured } = vi.hoisted(() => {
+  const params: A2UIToolParams[] = [];
+  return { captured: params };
+});
 
 vi.mock("@ag-ui/langgraph", () => ({
-  getA2UITools: (params: any) => {
+  getA2UITools: (params: A2UIToolParams) => {
     captured.push(params);
-    return { name: "generate_a2ui" };
+    return { name: "generate_a2ui", invoke: async () => "ok" };
   },
 }));
 

package/src/langgraph/__tests__/middleware.native-a2ui.test.ts

@@ -0,0 +1,118 @@
+import { describe, expect, it } from "vitest";
+import {
+  AIMessage,
+  AIMessageChunk,
+  HumanMessage,
+} from "@langchain/core/messages";
+import type { BaseMessage } from "@langchain/core/messages";
+import type { getA2UITools } from "@ag-ui/langgraph";
+import { createCopilotkitMiddleware } from "../middleware";
+
+const components = [{ id: "root", component: "Text", text: "Sales Dashboard" }];
+
+// A deterministic in-process model drives the actual toolkit; no provider
+// requests or SDK/factory mocks are involved in this adapter contract test.
+async function generate(properties: object) {
+  const prompts: string[] = [];
+  const model = {
+    bindTools: () => ({
+      async *stream(messages: BaseMessage[]) {
+        prompts.push(String(messages[0].content));
+        yield new AIMessageChunk({
+          content: "",
+          tool_calls: [
+            {
+              id: "render",
+              name: "render_a2ui",
+              args: { surfaceId: "dashboard", components },
+            },
+          ],
+        });
+      },
+    }),
+  };
+  const state = {
+    messages: [
+      new HumanMessage("Dashboard"),
+      new AIMessage({
+        content: "",
+        tool_calls: [{ id: "generate", name: "generate_a2ui", args: {} }],
+      }),
+    ],
+    thread_id: "native-tool-contract",
+    ...properties,
+  };
+  const before = JSON.stringify(state);
+  const middleware = createCopilotkitMiddleware();
+  let tool: ReturnType<typeof getA2UITools> | undefined;
+  await middleware.wrapModelCall(
+    { model, state, tools: [], messages: [], runtime: {} },
+    async (request) => {
+      tool = request.tools.find(
+        (candidate: { name: string }) => candidate.name === "generate_a2ui",
+      );
+      return new AIMessage("ok");
+    },
+  );
+  if (!tool) throw new Error("Expected the native A2UI tool");
+  const result = await tool.invoke({ intent: "create" }, { state });
+  if (typeof result !== "string")
+    throw new Error("Expected a native operations envelope");
+  expect(JSON.stringify(state)).toBe(before);
+  expect(JSON.parse(result)).toEqual({
+    a2ui_operations: [
+      {
+        version: "v0.9",
+        createSurface: {
+          surfaceId: "dashboard",
+          catalogId: "copilotkit://app-dashboard-catalog",
+        },
+      },
+      {
+        version: "v0.9",
+        updateComponents: { surfaceId: "dashboard", components },
+      },
+    ],
+  });
+  middleware.afterAgent(state);
+  return prompts[0];
+}
+
+describe("native A2UI tool namespace contract", () => {
+  it("uses the merged schema in the real toolkit prompt and CopilotKit catalog in native operations", async () => {
+    const prompt = await generate({
+      "ag-ui": {
+        inject_a2ui_tool: true,
+        a2ui_schema: {
+          catalogId: "agui://base",
+          components: { Base: { description: "RETAINED_NATIVE_SCHEMA" } },
+        },
+      },
+      copilotkit: {
+        a2ui_schema: {
+          catalogId: "copilotkit://app-dashboard-catalog",
+          components: { Text: { description: "COPILOTKIT_SCHEMA" } },
+        },
+      },
+    });
+    expect(prompt).toContain("RETAINED_NATIVE_SCHEMA");
+    expect(prompt).toContain("COPILOTKIT_SCHEMA");
+  });
+
+  it("uses an AG-UI-only serialized context catalog in native operations", async () => {
+    const prompt = await generate({
+      "ag-ui": {
+        inject_a2ui_tool: true,
+        context: JSON.stringify([
+          {
+            description: "A2UI catalog capabilities",
+            value: JSON.stringify(
+              "Available A2UI catalog:\n- copilotkit://app-dashboard-catalog\n  - Text: {...}",
+            ),
+          },
+        ]),
+      },
+    });
+    expect(prompt).toContain("copilotkit://app-dashboard-catalog");
+  });
+});

package/src/langgraph/__tests__/middleware.properties.test.ts

@@ -0,0 +1,288 @@
+import { beforeEach, describe, expect, it, vi } from "vitest";
+import { AIMessage, HumanMessage } from "@langchain/core/messages";
+import { StateGraph, START, END } from "@langchain/langgraph";
+import type { A2UIToolParams } from "@ag-ui/langgraph";
+
+// Capture the adapter boundary; no model or LLM responses are mocked.
+const { captured, invoked } = vi.hoisted(() => {
+  const params: A2UIToolParams[] = [];
+  return {
+    captured: params,
+    invoked: vi.fn(
+      async (_input: unknown, config: { state?: unknown }) => config.state,
+    ),
+  };
+});
+vi.mock("@ag-ui/langgraph", () => ({
+  getA2UITools: (params: A2UIToolParams) => {
+    captured.push(params);
+    return { name: "generate_a2ui", invoke: invoked };
+  },
+}));
+import { createCopilotkitMiddleware } from "../middleware";
+
+const catalog = (id: string) => [
+  {
+    description: "A2UI catalog capabilities",
+    value: "Available A2UI catalog:\n- " + id + "\n  - Card: {}",
+  },
+];
+const middleware = createCopilotkitMiddleware();
+async function wrap(state: object) {
+  const tools: Array<{
+    name: string;
+    invoke: (input: unknown, config: { state: object }) => Promise<unknown>;
+  }> = [];
+  const request = { state, model: {}, tools, messages: [], runtime: {} };
+  let received = request;
+  await middleware.wrapModelCall(request, async (req) => {
+    received = req;
+    return new AIMessage("ok");
+  });
+  return received;
+}
+function context(state: object, runtime = {}) {
+  return middleware.beforeAgent(
+    { messages: [new HumanMessage("hello")], ...state },
+    runtime,
+  );
+}
+
+beforeEach(() => {
+  captured.length = 0;
+  invoked.mockClear();
+});
+
+describe("effective AG-UI and CopilotKit properties", () => {
+  it.each([null, false, []])(
+    "treats explicit %j actions as empty while injection is enabled",
+    async (actions) => {
+      const result = await wrap({
+        "ag-ui": { inject_a2ui_tool: true, actions: [{ name: "base" }] },
+        copilotkit: { actions },
+      });
+      expect(result.tools.map((tool) => tool.name)).toEqual(["generate_a2ui"]);
+    },
+  );
+
+  it("replaces serialized schemas atomically and adapts object context only for the toolkit", async () => {
+    const schema = JSON.stringify({ catalogId: "override" });
+    const state = {
+      "ag-ui": {
+        inject_a2ui_tool: true,
+        a2ui_schema: JSON.stringify({
+          catalogId: "base",
+          components: { Card: {} },
+        }),
+      },
+      copilotkit: { a2ui_schema: schema, context: { user: "alice" } },
+    };
+    const result = await wrap(state);
+    await result.tools[0].invoke({}, { state });
+    expect(captured[0].defaultCatalogId).toBe("override");
+    expect(invoked.mock.calls[0][1].state).toEqual(
+      expect.objectContaining({
+        "ag-ui": expect.objectContaining({ a2ui_schema: schema, context: [] }),
+      }),
+    );
+    expect(state.copilotkit.context).toEqual({ user: "alice" });
+  });
+
+  it("retains both namespace fields through the graph state schema", async () => {
+    const graph = new StateGraph(middleware.stateSchema)
+      .addNode("echo", () => ({}))
+      .addEdge(START, "echo")
+      .addEdge("echo", END)
+      .compile();
+    const state = {
+      "ag-ui": { context: catalog("native") },
+      copilotkit: { a2ui_schema: "schema", inject_a2ui_tool: false },
+    };
+    expect(await graph.invoke(state)).toEqual(state);
+  });
+
+  it("decodes serialized catalog carriers only at the catalog consumer", async () => {
+    const entries = catalog("serialized");
+    entries[0].value = JSON.stringify(entries[0].value);
+    const state = {
+      "ag-ui": { inject_a2ui_tool: true, context: JSON.stringify(entries) },
+    };
+    await wrap(state);
+    expect(captured[0].defaultCatalogId).toBe("serialized");
+    expect(context(state).messages[0].content).toBe(
+      "App Context:\n" + state["ag-ui"].context,
+    );
+  });
+
+  it("ignores malformed catalog entries", async () => {
+    await wrap({
+      "ag-ui": {
+        inject_a2ui_tool: true,
+        context: [
+          null,
+          1,
+          { description: 4, value: true },
+          { description: "A2UI catalog", value: {} },
+        ],
+      },
+    });
+    expect(captured[0].defaultCatalogId).toBeUndefined();
+  });
+
+  it("merges nested context while preserving conflicting scalar and array leaves", () => {
+    const state = {
+      "ag-ui": {
+        context: {
+          nested: { retained: 1, conflict: "old", items: [{ base: 1 }] },
+          enabled: true,
+        },
+      },
+      copilotkit: {
+        context: {
+          nested: { added: 2, conflict: null, items: [{ override: 2 }] },
+          enabled: false,
+        },
+      },
+    };
+    const before = structuredClone(state);
+    const result = context(state);
+    expect(result.messages[0].content).toBe(
+      "App Context:\n" +
+        JSON.stringify(
+          {
+            nested: {
+              retained: 1,
+              conflict: null,
+              items: [{ override: 2 }],
+              added: 2,
+            },
+            enabled: false,
+          },
+          null,
+          2,
+        ),
+    );
+    expect(state).toEqual(before);
+  });
+
+  it.each([null, false, "", []])(
+    "does not resurrect AG-UI or runtime context after an explicit %j override",
+    (value) => {
+      expect(
+        context(
+          { "ag-ui": { context: "base" }, copilotkit: { context: value } },
+          { context: "runtime" },
+        ),
+      ).toBeUndefined();
+    },
+  );
+
+  it("keeps serialized context atomic and preserves the runtime fallback", () => {
+    expect(
+      context({
+        "ag-ui": { context: { base: 1 } },
+        copilotkit: { context: '{"override":2}' },
+      }).messages[0].content,
+    ).toBe('App Context:\n{"override":2}');
+    expect(context({}, { context: "runtime" }).messages[0].content).toBe(
+      "App Context:\nruntime",
+    );
+  });
+
+  it.each(["ag-ui", "copilotkit"])(
+    "uses a catalog and frontend action supplied only through %s",
+    async (namespace) => {
+      const state = {
+        [namespace]: {
+          context: catalog("catalog-one"),
+          actions: [{ name: "frontend" }],
+          inject_a2ui_tool: true,
+        },
+      };
+      const result = await wrap(state);
+      expect(result.tools.map((tool) => tool.name)).toEqual([
+        "generate_a2ui",
+        "frontend",
+      ]);
+      expect(captured[0].defaultCatalogId).toBe("catalog-one");
+      expect(captured[0].guidelines.compositionGuide).toContain("catalog-one");
+    },
+  );
+
+  it("replaces context and action lists as wholes, including empty lists", async () => {
+    const state = {
+      "ag-ui": {
+        context: catalog("base"),
+        actions: [{ name: "base" }],
+        inject_a2ui_tool: true,
+      },
+      copilotkit: { context: [], actions: [] },
+    };
+    const result = await wrap(state);
+    expect(result.tools.map((tool) => tool.name)).toEqual(["generate_a2ui"]);
+    expect(captured[0].defaultCatalogId).toBeUndefined();
+    expect(captured[0].guidelines).toBeUndefined();
+  });
+
+  it.each([false, null, ""])(
+    "honors an explicit %j injection override",
+    async (value) => {
+      const result = await wrap({
+        "ag-ui": { inject_a2ui_tool: true },
+        copilotkit: { inject_a2ui_tool: value },
+      });
+      expect(result.tools).toEqual([]);
+      expect(captured).toEqual([]);
+    },
+  );
+
+  it("intercepts and restores AG-UI frontend calls without losing the namespace", () => {
+    const message = new AIMessage({
+      id: "assistant",
+      content: "",
+      tool_calls: [{ id: "call", name: "frontend", args: {} }],
+    });
+    const state = {
+      "ag-ui": { actions: [{ name: "frontend" }] },
+      messages: [message],
+    };
+    const intercepted = middleware.afterModel(state);
+    expect(intercepted.messages[0].tool_calls).toEqual([]);
+    const restored = middleware.afterAgent({ ...state, ...intercepted });
+    expect(restored.messages[0].tool_calls).toEqual(message.tool_calls);
+    expect(state.messages[0]).toBe(message);
+  });
+
+  it("delivers the merged native schema to the tool invocation without mutating state", async () => {
+    const state = {
+      "ag-ui": {
+        inject_a2ui_tool: true,
+        a2ui_schema: {
+          catalogId: "base",
+          components: { Card: { base: true, shared: "old" } },
+        },
+      },
+      copilotkit: {
+        a2ui_schema: {
+          catalogId: "override",
+          components: { Card: { shared: "new" }, Button: {} },
+        },
+      },
+    };
+    const before = structuredClone(state);
+    const result = await wrap(state);
+    await result.tools[0].invoke({}, { state });
+    expect(captured[0].defaultCatalogId).toBe("override");
+    expect(invoked.mock.calls[0][1].state).toEqual(
+      expect.objectContaining({
+        "ag-ui": expect.objectContaining({
+          a2ui_schema: JSON.stringify({
+            catalogId: "override",
+            components: { Card: { base: true, shared: "new" }, Button: {} },
+          }),
+        }),
+      }),
+    );
+    expect(state).toEqual(before);
+  });
+});

package/src/langgraph/middleware.ts

@@ -20,14 +20,56 @@ const A2UI_DEFAULT_THREAD_KEY = "__copilotkit_a2ui_default__";
 const a2uiThreadKey = (state: any): string =>
   (state?.thread_id as string) || A2UI_DEFAULT_THREAD_KEY;
 
+/**
+ * Merge AG-UI defaults with CopilotKit overrides. Dictionaries recurse; every
+ * conflicting leaf uses CopilotKit, including false, null and empty values.
+ * Arrays and serialized strings are atomic: no positional pairing or decoding.
+ */
+const isPropertyBag = (value: unknown): value is Record<string, unknown> =>
+  value !== null && typeof value === "object" && !Array.isArray(value);
+
+const mergeProperties = (
+  base: Record<string, unknown>,
+  overrides: Record<string, unknown>,
+): Record<string, unknown> =>
+  Object.fromEntries(
+    Array.from(new Set([...Object.keys(base), ...Object.keys(overrides)])).map(
+      (key) => {
+        const value = Object.prototype.hasOwnProperty.call(overrides, key)
+          ? isPropertyBag(base[key]) && isPropertyBag(overrides[key])
+            ? mergeProperties(base[key], overrides[key])
+            : overrides[key]
+          : base[key];
+        return [key, value];
+      },
+    ),
+  );
+
+const effectiveProperties = (state: unknown): Record<string, unknown> => {
+  const source = isPropertyBag(state) ? state : {};
+  return mergeProperties(
+    isPropertyBag(source["ag-ui"]) ? source["ag-ui"] : {},
+    isPropertyBag(source.copilotkit) ? source.copilotkit : {},
+  );
+};
+
+const decodeCatalogValue = (value: unknown): unknown => {
+  if (typeof value !== "string") return value;
+  try {
+    return JSON.parse(value);
+  } catch {
+    return value;
+  }
+};
+
 /**
  * Find the frontend-registered A2UI catalog wherever it was passed. Returns
  * `{ compositionGuide?, catalogId? }` when a catalog is present, else `null`
- * (so the tool is never advertised when the client can't render A2UI). Two
+ * independently of the explicit tool-injection decision. Two
  * delivery paths, depending on how the agent is served:
  *  - AG-UI native endpoint → `state["ag-ui"].a2ui_schema` (JSON
  *    `{ catalogId, components }`); the toolkit reads it from state itself.
- *  - CopilotKit runtime proxy → a `state.copilotkit.context` entry describing
+ *  - Runtime proxy → an effective `context` entry describing
  *    the A2UI catalog (catalog id + component schemas as text), passed to the
  *    subagent via `compositionGuide`.
  * `catalogId` binds generated surfaces to the frontend's catalog so BYOC
@@ -36,7 +78,8 @@ const a2uiThreadKey = (state: any): string =>
 const resolveA2uiCatalog = (
   state: any,
 ): { compositionGuide?: string; catalogId?: string } | null => {
-  const a2uiSchema = state?.["ag-ui"]?.a2ui_schema;
+  const properties = effectiveProperties(state);
+  const a2uiSchema = properties.a2ui_schema;
   if (a2uiSchema) {
     let catalogId: string | undefined;
     try {
@@ -48,11 +91,17 @@ const resolveA2uiCatalog = (
     }
     return { catalogId };
   }
-  const context = state?.copilotkit?.context;
+  const context = decodeCatalogValue(properties.context);
   for (const entry of Array.isArray(context) ? context : []) {
-    const description = entry?.description ?? "";
-    const value = entry?.value ?? "";
-    if (!description.includes("A2UI catalog") || !value) continue;
+    const description = entry?.description;
+    const value = decodeCatalogValue(entry?.value);
+    if (
+      typeof description !== "string" ||
+      typeof value !== "string" ||
+      !description.includes("A2UI catalog") ||
+      !value
+    )
+      continue;
     const match = /^\s*-\s+(\S+)/m.exec(value);
     return { compositionGuide: value, catalogId: match?.[1] };
   }
@@ -67,7 +116,7 @@ const resolveA2uiCatalog = (
  * no signal (off, or no A2UI middleware in the pipeline) → no auto-injection.
  */
 const a2uiInjectDecision = (state: any): boolean | string | undefined =>
-  state?.["ag-ui"]?.inject_a2ui_tool;
+  effectiveProperties(state).inject_a2ui_tool as boolean | string | undefined;
 
 type WithJsonSchema<T> = T extends { "~standard": infer S }
   ? Omit<T, "~standard"> & {
@@ -250,7 +299,10 @@ const createAppContextBeforeAgent = (state, runtime) => {
   }
 
   // Get app context from runtime
-  const appContext = state["copilotkit"]?.context ?? runtime?.context;
+  const properties = effectiveProperties(state);
+  const appContext = Object.prototype.hasOwnProperty.call(properties, "context")
+    ? properties.context
+    : runtime?.context;
 
   // Check if appContext is missing or empty
   const isEmptyContext =
@@ -358,14 +410,19 @@ const createAppContextBeforeAgent = (state, runtime) => {
  */
 const copilotKitStateSchema = z
   .object({
+    "ag-ui": zodState(z.record(z.string(), z.unknown()).optional()),
     copilotkit: zodState(
       z
         .object({
-          actions: z.array(z.any()),
+          actions: z
+            .union([z.array(z.any()), z.literal(false)])
+            .nullable()
+            .optional(),
           context: z.any().optional(),
           interceptedToolCalls: z.array(z.any()).optional(),
           originalAIMessageId: z.string().optional(),
         })
+        .passthrough()
         .optional(),
     ),
   })
@@ -475,12 +532,40 @@ const buildMiddlewareInput = (
         (request.tools || []).map((t: any) => t?.name),
       );
       if (!existingNames.has(candidate.name)) {
+        // ToolNode derives tool runtime state from its config, ignoring an
+        // overridden middleware request.state. Adapt the tool's invocation
+        // config so the toolkit sees the same effective properties.
+        const invoke = candidate.invoke.bind(candidate);
+        candidate.invoke = (input, config) => {
+          const state =
+            config && "state" in config && isPropertyBag(config.state)
+              ? config.state
+              : request.state;
+          const properties = effectiveProperties(state);
+          const schema = properties.a2ui_schema;
+          const context = decodeCatalogValue(properties.context);
+          return invoke(input, {
+            ...config,
+            state: {
+              ...state,
+              "ag-ui": {
+                ...properties,
+                context: Array.isArray(context) ? context : [],
+                // The toolkit interpolates the native schema into its prompt.
+                a2ui_schema: isPropertyBag(schema)
+                  ? JSON.stringify(schema)
+                  : schema,
+              },
+            },
+          });
+        };
         a2uiTool = candidate;
         a2uiToolsByThread.set(a2uiThreadKey(request.state), a2uiTool);
       }
     }
 
-    let frontendTools = request.state["copilotkit"]?.actions ?? [];
+    const actions = effectiveProperties(request.state).actions;
+    let frontendTools = Array.isArray(actions) ? actions : [];
     if (a2uiTool) {
       // Our generate_a2ui replaces the runtime's render tool — don't advertise
       // both. Drop the render tool the A2UI middleware injected.
@@ -568,7 +653,8 @@ const buildMiddlewareInput = (
 
   // Intercept frontend tool calls after model returns, before ToolNode executes
   afterModel: (state) => {
-    const frontendTools = state["copilotkit"]?.actions ?? [];
+    const actions = effectiveProperties(state).actions;
+    const frontendTools = Array.isArray(actions) ? actions : [];
     if (frontendTools.length === 0) return;
 
     const frontendToolNames = new Set(