@copilotkit/runtime 2.0.0-next.1 → 2.0.0-rc.1

package/dist/runtime/endpoints/single-route-helpers.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"single-route-helpers.mjs","names":[],"sources":["../../../src/runtime/endpoints/single-route-helpers.ts"],"sourcesContent":["const METHOD_NAMES = [\n  \"agent/run\",\n  \"agent/suggest\",\n  \"agent/connect\",\n  \"trajectory/connect\",\n  \"agent/stop\",\n  \"info\",\n  \"inspector/metadata\",\n  \"inspector/learning\",\n  \"transcribe\",\n  \"resource/request\",\n] as const;\n\nexport type EndpointMethod = (typeof METHOD_NAMES)[number];\n\ninterface JsonEnvelope {\n  method?: string;\n  params?: Record<string, unknown>;\n  body?: unknown;\n}\n\nexport interface MethodCall {\n  method: EndpointMethod;\n  params?: Record<string, unknown>;\n  body?: unknown;\n}\n\n/**\n * Detect a single-route JSON envelope on a request the multi-route router\n * could not match.\n *\n * A single-endpoint client (`useSingleEndpoint` on the frontend provider)\n * POSTs `{ method, params, body }` at the base path. A multi-route runtime\n * matches no route for that path and would otherwise answer a bare 404, giving\n * the developer nothing to go on. Recognising the envelope here lets the\n * handler name the actual cause instead.\n *\n * Deliberately conservative: it reports a method only for a POST carrying a\n * JSON body whose `method` is one the single-route endpoint actually accepts.\n * Anything else is a genuine 404 and must stay one.\n *\n * @returns The envelope's method name, or `null` if this is not an envelope.\n */\nexport async function detectSingleRouteEnvelope(\n  request: Request,\n): Promise<EndpointMethod | null> {\n  if (request.method !== \"POST\") return null;\n\n  const contentType = request.headers.get(\"content-type\") || \"\";\n  if (!contentType.includes(\"application/json\")) return null;\n\n  let envelope: unknown;\n  try {\n    envelope = await request.clone().json();\n  } catch {\n    return null;\n  }\n\n  if (typeof envelope !== \"object\" || envelope === null) return null;\n\n  const method = (envelope as JsonEnvelope).method;\n  if (typeof method !== \"string\") return null;\n  if (!(METHOD_NAMES as readonly string[]).includes(method)) return null;\n\n  return method as EndpointMethod;\n}\n\nexport async function parseMethodCall(request: Request): Promise<MethodCall> {\n  const contentType = request.headers.get(\"content-type\") || \"\";\n\n  if (!contentType.includes(\"application/json\")) {\n    throw createResponseError(\n      \"Single-route endpoint expects JSON payloads\",\n      415,\n    );\n  }\n\n  let jsonEnvelope: JsonEnvelope;\n  try {\n    jsonEnvelope = (await request.clone().json()) as JsonEnvelope;\n  } catch {\n    throw createResponseError(\"Invalid JSON payload\", 400);\n  }\n\n  const method = validateMethod(jsonEnvelope.method);\n\n  return {\n    method,\n    params: jsonEnvelope.params,\n    body: jsonEnvelope.body,\n  };\n}\n\nexport function expectString(\n  params: Record<string, unknown> | undefined,\n  key: string,\n): string {\n  const value = params?.[key];\n  if (typeof value === \"string\" && value.trim().length > 0) {\n    return value;\n  }\n\n  throw createResponseError(`Missing or invalid parameter '${key}'`, 400);\n}\n\nexport function createJsonRequest(base: Request, body: unknown): Request {\n  if (body === undefined || body === null) {\n    throw createResponseError(\"Missing request body for JSON handler\", 400);\n  }\n\n  const headers = new Headers(base.headers);\n  headers.set(\"content-type\", \"application/json\");\n  headers.delete(\"content-length\");\n\n  const serializedBody = serializeJsonBody(body);\n\n  return new Request(base.url, {\n    method: \"POST\",\n    headers,\n    body: serializedBody,\n    signal: base.signal,\n  });\n}\n\n/**\n * Rebuild a resource request carried inside the single-route JSON envelope.\n *\n * The resource path must stay relative to the mounted Runtime. This prevents\n * the envelope from becoming an open HTTP proxy while preserving the method,\n * query string, headers, and cancellation signal used by the REST handler.\n */\nexport function createResourceRequest(\n  base: Request,\n  path: string,\n  httpMethod: string,\n  body: unknown,\n): Request {\n  if (!path.startsWith(\"/\") || path.startsWith(\"//\")) {\n    throw createResponseError(\"Resource path must be Runtime-relative\", 400);\n  }\n\n  const resourceUrl = new URL(path, \"http://copilotkit.resource\");\n  if (resourceUrl.origin !== \"http://copilotkit.resource\") {\n    throw createResponseError(\"Resource path must be Runtime-relative\", 400);\n  }\n\n  const targetUrl = new URL(base.url);\n  targetUrl.pathname = `${targetUrl.pathname.replace(/\\/$/, \"\")}${resourceUrl.pathname}`;\n  targetUrl.search = resourceUrl.search;\n\n  const method = httpMethod.toUpperCase();\n  const headers = new Headers(base.headers);\n  headers.delete(\"content-length\");\n  const hasBody = method !== \"GET\" && method !== \"HEAD\" && body != null;\n\n  return new Request(targetUrl, {\n    method,\n    headers,\n    ...(hasBody ? { body: serializeJsonBody(body) } : {}),\n    signal: base.signal,\n  });\n}\n\nexport function createResponseError(message: string, status: number): Response {\n  return new Response(\n    JSON.stringify({\n      error: \"invalid_request\",\n      message,\n    }),\n    {\n      status,\n      headers: {\n        \"Content-Type\": \"application/json\",\n      },\n    },\n  );\n}\n\nfunction validateMethod(method: string | undefined): EndpointMethod {\n  if (!method) {\n    throw createResponseError(\"Missing method field\", 400);\n  }\n\n  if ((METHOD_NAMES as readonly string[]).includes(method)) {\n    return method as EndpointMethod;\n  }\n\n  throw createResponseError(`Unsupported method '${method}'`, 400);\n}\n\nfunction serializeJsonBody(body: unknown): BodyInit {\n  if (typeof body === \"string\") {\n    return body;\n  }\n\n  if (\n    body instanceof Blob ||\n    body instanceof ArrayBuffer ||\n    body instanceof Uint8Array\n  ) {\n    // Uint8Array<ArrayBufferLike> is a valid fetch body at runtime, but the\n    // DOM lib's BodyInit only admits ArrayBuffer-backed views.\n    return body as BodyInit;\n  }\n\n  if (body instanceof FormData || body instanceof URLSearchParams) {\n    return body;\n  }\n\n  return JSON.stringify(body);\n}\n"],"mappings":";AAAA,MAAM,eAAe;CACnB;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACD;;;;;;;;;;;;;;;;;AAgCD,eAAsB,0BACpB,SACgC;AAChC,KAAI,QAAQ,WAAW,OAAQ,QAAO;AAGtC,KAAI,EADgB,QAAQ,QAAQ,IAAI,eAAe,IAAI,IAC1C,SAAS,mBAAmB,CAAE,QAAO;CAEtD,IAAI;AACJ,KAAI;AACF,aAAW,MAAM,QAAQ,OAAO,CAAC,MAAM;SACjC;AACN,SAAO;;AAGT,KAAI,OAAO,aAAa,YAAY,aAAa,KAAM,QAAO;CAE9D,MAAM,SAAU,SAA0B;AAC1C,KAAI,OAAO,WAAW,SAAU,QAAO;AACvC,KAAI,CAAE,aAAmC,SAAS,OAAO,CAAE,QAAO;AAElE,QAAO;;AAGT,eAAsB,gBAAgB,SAAuC;AAG3E,KAAI,EAFgB,QAAQ,QAAQ,IAAI,eAAe,IAAI,IAE1C,SAAS,mBAAmB,CAC3C,OAAM,oBACJ,+CACA,IACD;CAGH,IAAI;AACJ,KAAI;AACF,iBAAgB,MAAM,QAAQ,OAAO,CAAC,MAAM;SACtC;AACN,QAAM,oBAAoB,wBAAwB,IAAI;;AAKxD,QAAO;EACL,QAHa,eAAe,aAAa,OAAO;EAIhD,QAAQ,aAAa;EACrB,MAAM,aAAa;EACpB;;AAGH,SAAgB,aACd,QACA,KACQ;CACR,MAAM,QAAQ,SAAS;AACvB,KAAI,OAAO,UAAU,YAAY,MAAM,MAAM,CAAC,SAAS,EACrD,QAAO;AAGT,OAAM,oBAAoB,iCAAiC,IAAI,IAAI,IAAI;;AAGzE,SAAgB,kBAAkB,MAAe,MAAwB;AACvE,KAAI,SAAS,UAAa,SAAS,KACjC,OAAM,oBAAoB,yCAAyC,IAAI;CAGzE,MAAM,UAAU,IAAI,QAAQ,KAAK,QAAQ;AACzC,SAAQ,IAAI,gBAAgB,mBAAmB;AAC/C,SAAQ,OAAO,iBAAiB;CAEhC,MAAM,iBAAiB,kBAAkB,KAAK;AAE9C,QAAO,IAAI,QAAQ,KAAK,KAAK;EAC3B,QAAQ;EACR;EACA,MAAM;EACN,QAAQ,KAAK;EACd,CAAC;;;;;;;;;AAUJ,SAAgB,sBACd,MACA,MACA,YACA,MACS;AACT,KAAI,CAAC,KAAK,WAAW,IAAI,IAAI,KAAK,WAAW,KAAK,CAChD,OAAM,oBAAoB,0CAA0C,IAAI;CAG1E,MAAM,cAAc,IAAI,IAAI,MAAM,6BAA6B;AAC/D,KAAI,YAAY,WAAW,6BACzB,OAAM,oBAAoB,0CAA0C,IAAI;CAG1E,MAAM,YAAY,IAAI,IAAI,KAAK,IAAI;AACnC,WAAU,WAAW,GAAG,UAAU,SAAS,QAAQ,OAAO,GAAG,GAAG,YAAY;AAC5E,WAAU,SAAS,YAAY;CAE/B,MAAM,SAAS,WAAW,aAAa;CACvC,MAAM,UAAU,IAAI,QAAQ,KAAK,QAAQ;AACzC,SAAQ,OAAO,iBAAiB;CAChC,MAAM,UAAU,WAAW,SAAS,WAAW,UAAU,QAAQ;AAEjE,QAAO,IAAI,QAAQ,WAAW;EAC5B;EACA;EACA,GAAI,UAAU,EAAE,MAAM,kBAAkB,KAAK,EAAE,GAAG,EAAE;EACpD,QAAQ,KAAK;EACd,CAAC;;AAGJ,SAAgB,oBAAoB,SAAiB,QAA0B;AAC7E,QAAO,IAAI,SACT,KAAK,UAAU;EACb,OAAO;EACP;EACD,CAAC,EACF;EACE;EACA,SAAS,EACP,gBAAgB,oBACjB;EACF,CACF;;AAGH,SAAS,eAAe,QAA4C;AAClE,KAAI,CAAC,OACH,OAAM,oBAAoB,wBAAwB,IAAI;AAGxD,KAAK,aAAmC,SAAS,OAAO,CACtD,QAAO;AAGT,OAAM,oBAAoB,uBAAuB,OAAO,IAAI,IAAI;;AAGlE,SAAS,kBAAkB,MAAyB;AAClD,KAAI,OAAO,SAAS,SAClB,QAAO;AAGT,KACE,gBAAgB,QAChB,gBAAgB,eAChB,gBAAgB,WAIhB,QAAO;AAGT,KAAI,gBAAgB,YAAY,gBAAgB,gBAC9C,QAAO;AAGT,QAAO,KAAK,UAAU,KAAK"}

package/dist/runtime/generative-ui/generative-ui-middleware.cjs

@@ -0,0 +1,393 @@
+const require_runtime = require('../../_virtual/_rolldown/runtime.cjs');
+let _copilotkit_shared = require("@copilotkit/shared");
+let rxjs = require("rxjs");
+let _ag_ui_client = require("@ag-ui/client");
+
+//#region src/runtime/generative-ui/generative-ui-middleware.ts
+function isRecord(value) {
+	return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+/**
+* Builds the manifest from the generative UI envelope
+* (`forwardedProps.copilotkit.ui`). The client sends the envelope, so it is
+* not trusted: `readCopilotKitUIEnvelope` checks every field, and
+* `envelopeToManifest` turns it into the manifest. Returns
+* `undefined` (the run passes through) when:
+* - there is no valid envelope;
+* - the transport is "markdown" (the model writes fences, not render_ui);
+* - the catalog has no components (the server cannot validate operations).
+*/
+function readManifest(forwardedProps) {
+	const envelope = (0, _copilotkit_shared.readCopilotKitUIEnvelope)(forwardedProps);
+	if (!envelope) return void 0;
+	const manifest = (0, _copilotkit_shared.envelopeToManifest)(envelope);
+	return manifest.transport === "tool" && manifest.hasComponents ? manifest : void 0;
+}
+function prepareInput(input, manifest) {
+	const tools = (input.tools ?? []).filter((t) => t.name !== _copilotkit_shared.GENERATIVE_UI_RENDER_TOOL);
+	tools.push({
+		name: _copilotkit_shared.GENERATIVE_UI_RENDER_TOOL,
+		description: "Render or update generative UI surfaces, or call app functions, with A2UI v1.0 operations. Follow the generative UI guide in context.",
+		parameters: {
+			type: "object",
+			properties: {
+				target: {
+					type: "string",
+					enum: manifest.targets.map((t) => t.id),
+					description: "Where the surface is shown: \"chat\" or a named inline target."
+				},
+				operations: {
+					type: "array",
+					description: "A2UI v1.0 operations, applied in order.",
+					items: { type: "object" }
+				}
+			},
+			required: ["target", "operations"]
+		}
+	});
+	const context = (input.context ?? []).filter((c) => c.description !== _copilotkit_shared.GENERATIVE_UI_CONTEXT_DESCRIPTION);
+	context.push({
+		description: _copilotkit_shared.GENERATIVE_UI_CONTEXT_DESCRIPTION,
+		value: (0, _copilotkit_shared.buildGenerativeUIPrompt)(manifest)
+	});
+	return {
+		...input,
+		tools,
+		context
+	};
+}
+/**
+* Experimental. Lets an agent render CopilotKit generative UI surfaces via the
+* `render_ui` tool. It applies only to runs whose generative UI envelope uses
+* the "tool" transport. Each complete operation in the streamed arguments is
+* validated against the catalog in the envelope and committed immediately:
+* - `ACTIVITY_SNAPSHOT` for a new surface, `ACTIVITY_DELTA` for every further
+*   operation on it.
+* - One `ACTIVITY_SNAPSHOT` of type `generative-ui-function-call` for each
+*   `callRendererFunction`. The client runs the function.
+*/
+var GenerativeUIMiddleware = class extends _ag_ui_client.Middleware {
+	run(input, next) {
+		const manifest = readManifest(input.forwardedProps);
+		if (!manifest) return this.runNext(input, next);
+		return processStream(this.runNextWithState(prepareInput(input, manifest), next), manifest);
+	}
+};
+/**
+* The first time a render_ui call touches an existing surface, mark it
+* streaming and clear its rejection list: earlier calls' rejections were
+* already reported to the model in their own tool results, and a repaired
+* surface must not keep showing them. A surface created by this call starts
+* with an empty list in its snapshot.
+*/
+function beginSurfaceUpdate(state, surfaceId) {
+	if (state.touched.has(surfaceId)) return [];
+	state.touched.add(surfaceId);
+	return [{
+		op: "replace",
+		path: "/status",
+		value: "streaming"
+	}, {
+		op: "replace",
+		path: "/rejected",
+		value: []
+	}];
+}
+function processStream(source, manifest) {
+	return new rxjs.Observable((subscriber) => {
+		const entries = Object.entries(manifest.surfaces);
+		const surfaces = new Map(entries.filter(([, s]) => !s.status).map(([id, s]) => [id, s.target]));
+		const createdSurfaces = new Set(entries.filter(([, s]) => s.status !== "failed").map(([id]) => id));
+		const targetComponents = new Map(manifest.targets.filter((t) => t.components !== void 0).map((t) => [t.id, new Set(t.components)]));
+		const functionCallIds = /* @__PURE__ */ new Set();
+		const calls = /* @__PURE__ */ new Map();
+		let heldFinish = null;
+		const emit = (event) => subscriber.next(event);
+		const patch = (surfaceId, ops) => emit({
+			type: _ag_ui_client.EventType.ACTIVITY_DELTA,
+			messageId: (0, _copilotkit_shared.generativeUISurfaceMessageId)(surfaceId),
+			activityType: _copilotkit_shared.GENERATIVE_UI_ACTIVITY_TYPE,
+			patch: ops
+		});
+		const processOperation = (state, op) => {
+			const index = state.nextIndex++;
+			const reject = (reason, surfaceId) => {
+				const entry = {
+					index,
+					reason
+				};
+				state.rejected.push(entry);
+				if (surfaceId && surfaces.has(surfaceId)) patch(surfaceId, [...beginSurfaceUpdate(state, surfaceId), {
+					op: "add",
+					path: "/rejected/-",
+					value: entry
+				}]);
+			};
+			if (op === null) return reject("operation is not valid JSON");
+			let surfaceId;
+			try {
+				surfaceId = (0, _copilotkit_shared.getOperationSurfaceId)(op);
+				applyOperation(state, withCatalogId(op), surfaceId, reject);
+			} catch (error) {
+				reject(`operation could not be processed: ${error instanceof Error ? error.message : String(error)}`, surfaceId);
+			}
+		};
+		/**
+		* The model does not have to name the catalog. A createSurface keeps a
+		* catalogId that the envelope lists, and gets the catalog id of the run
+		* for any other value. The whole operation, with its inline components
+		* and data model, stays in the snapshot.
+		*/
+		const catalogIds = manifest.catalogIds ?? [manifest.catalogId];
+		const withCatalogId = (op) => {
+			const create = op.createSurface;
+			if (!isRecord(create)) return op;
+			const catalogId = catalogIds.includes(create.catalogId) ? create.catalogId : manifest.catalogId;
+			return {
+				...op,
+				createSurface: {
+					...create,
+					catalogId
+				}
+			};
+		};
+		/**
+		* The components that the target of the surface takes. A createSurface
+		* goes to the target of the render_ui call. Every other operation goes to
+		* the target that the surface was created on.
+		*/
+		const allowedComponentsFor = (state, op, surfaceId) => {
+			const target = (0, _copilotkit_shared.getOperationKind)(op) === "createSurface" ? state.target : surfaces.get(surfaceId);
+			const names = targetComponents.get(target);
+			return names && {
+				target,
+				names
+			};
+		};
+		const applyOperation = (state, op, surfaceId, reject) => {
+			const validation = (0, _copilotkit_shared.validateOperation)(op, {
+				manifest,
+				knownSurfaces: new Set(surfaces.keys()),
+				createdSurfaces,
+				functions: manifest.functions,
+				allowedComponents: allowedComponentsFor(state, op, surfaceId)
+			});
+			if (validation.ok === false) return reject(validation.reason, surfaceId);
+			const accepted = op;
+			if ("callRendererFunction" in accepted) return callRendererFunction(state, accepted.callRendererFunction, reject);
+			if ("createSurface" in accepted) return createSurface(state, surfaceId, accepted, reject);
+			patch(surfaceId, [...beginSurfaceUpdate(state, surfaceId), {
+				op: "add",
+				path: "/operations/-",
+				value: accepted
+			}]);
+			if ("deleteSurface" in accepted) surfaces.delete(surfaceId);
+			state.applied++;
+		};
+		const createSurface = (state, surfaceId, op, reject) => {
+			const { target } = state;
+			if (target === void 0) return reject("render_ui was called without a target");
+			if (!manifest.targets.some((t) => t.id === target)) return reject(`target "${target}" is not mounted`);
+			surfaces.set(surfaceId, target);
+			createdSurfaces.add(surfaceId);
+			const content = {
+				surfaceId,
+				target,
+				catalogId: "createSurface" in op && op.createSurface.catalogId || manifest.catalogId,
+				operations: [op],
+				status: "streaming",
+				rejected: []
+			};
+			emit({
+				type: _ag_ui_client.EventType.ACTIVITY_SNAPSHOT,
+				messageId: (0, _copilotkit_shared.generativeUISurfaceMessageId)(surfaceId),
+				activityType: _copilotkit_shared.GENERATIVE_UI_ACTIVITY_TYPE,
+				content,
+				replace: true
+			});
+			state.touched.add(surfaceId);
+			state.applied++;
+		};
+		/**
+		* Sends one function call to the client, which runs it. A function call
+		* needs no target and no surface. Each `functionCallId` runs once.
+		*/
+		const callRendererFunction = (state, content, reject) => {
+			const { functionCallId } = content;
+			if (functionCallIds.has(functionCallId)) return reject(`functionCallId "${functionCallId}" was used before; use a new functionCallId`);
+			functionCallIds.add(functionCallId);
+			emit({
+				type: _ag_ui_client.EventType.ACTIVITY_SNAPSHOT,
+				messageId: (0, _copilotkit_shared.generativeUIFunctionCallMessageId)(functionCallId),
+				activityType: _copilotkit_shared.GENERATIVE_UI_FUNCTION_CALL_ACTIVITY_TYPE,
+				content: {
+					functionCallId,
+					callFunction: content.callFunction
+				},
+				replace: true
+			});
+			state.calls++;
+		};
+		/** Same buffering rule as streamed ops: hold them until the target is known. */
+		const acceptOperation = (state, op) => {
+			if (state.target === void 0) state.buffered.push(op);
+			else processOperation(state, op);
+		};
+		/**
+		* The streaming parser found no operations at all. Models sometimes send
+		* `operations` as a JSON-encoded string; recover those from the complete
+		* arguments. Anything still unusable leaves the call with zero operations,
+		* which the tool result reports as an error.
+		*/
+		const recoverOperations = (state) => {
+			let args;
+			try {
+				args = JSON.parse(state.stream.text);
+			} catch {
+				return;
+			}
+			if (!isRecord(args)) return;
+			if (state.target === void 0 && typeof args.target === "string") state.target = args.target;
+			let operations = args.operations;
+			if (typeof operations === "string") try {
+				operations = JSON.parse(operations);
+			} catch {
+				return;
+			}
+			if (!Array.isArray(operations)) return;
+			for (const op of operations) acceptOperation(state, isRecord(op) ? op : null);
+		};
+		const finishCall = (state) => {
+			if (state.finished) return;
+			state.finished = true;
+			if (state.nextIndex === 0 && state.buffered.length === 0) recoverOperations(state);
+			if (state.target === void 0) {
+				const buffered = state.buffered;
+				state.buffered = [];
+				for (const op of buffered) {
+					if ((0, _copilotkit_shared.getOperationKind)(op) === "callRendererFunction") {
+						processOperation(state, op);
+						continue;
+					}
+					state.rejected.push({
+						index: state.nextIndex++,
+						reason: "render_ui was called without a target"
+					});
+				}
+			}
+			if (state.stream.hasIncompleteOperation) state.rejected.push({
+				index: state.nextIndex++,
+				reason: "operation was truncated"
+			});
+			for (const id of state.touched) patch(id, [{
+				op: "replace",
+				path: "/status",
+				value: "ready"
+			}]);
+		};
+		const flushFinish = () => {
+			if (heldFinish) {
+				emit(heldFinish.event);
+				heldFinish = null;
+			}
+		};
+		const handleEvent = ({ event, messages }) => {
+			flushFinish();
+			if (event.type === _ag_ui_client.EventType.RUN_FINISHED) {
+				heldFinish = {
+					event,
+					messages
+				};
+				return;
+			}
+			emit(event);
+			if (event.type === _ag_ui_client.EventType.TOOL_CALL_START) {
+				const start = event;
+				if (start.toolCallName === _copilotkit_shared.GENERATIVE_UI_RENDER_TOOL) calls.set(start.toolCallId, {
+					stream: new _copilotkit_shared.RenderUIArgsStream(),
+					buffered: [],
+					nextIndex: 0,
+					applied: 0,
+					calls: 0,
+					rejected: [],
+					touched: /* @__PURE__ */ new Set(),
+					finished: false
+				});
+			} else if (event.type === _ag_ui_client.EventType.TOOL_CALL_ARGS) {
+				const args = event;
+				const state = calls.get(args.toolCallId);
+				if (!state || state.finished) return;
+				const { target, newOperations } = state.stream.push(args.delta);
+				if (target !== void 0 && state.target === void 0) {
+					state.target = target;
+					const buffered = state.buffered;
+					state.buffered = [];
+					for (const op of buffered) processOperation(state, op);
+				}
+				for (const op of newOperations) acceptOperation(state, op);
+			} else if (event.type === _ag_ui_client.EventType.TOOL_CALL_END) {
+				const state = calls.get(event.toolCallId);
+				if (state) finishCall(state);
+			}
+		};
+		const completeRun = () => {
+			for (const state of calls.values()) finishCall(state);
+			if (heldFinish) {
+				const answered = new Set(heldFinish.messages.filter((m) => m.role === "tool").map((m) => m.toolCallId));
+				for (const [toolCallId, state] of calls) {
+					if (answered.has(toolCallId)) continue;
+					const nothingParsed = state.applied === 0 && state.calls === 0 && state.rejected.length === 0;
+					emit({
+						type: _ag_ui_client.EventType.TOOL_CALL_RESULT,
+						messageId: (0, _copilotkit_shared.randomUUID)(),
+						toolCallId,
+						content: JSON.stringify(nothingParsed ? {
+							status: "error",
+							applied: 0,
+							calls: 0,
+							rejected: [],
+							reason: "operations must be a JSON array of A2UI operation objects; nothing was rendered"
+						} : {
+							status: "rendered",
+							applied: state.applied,
+							calls: state.calls,
+							rejected: state.rejected
+						})
+					});
+				}
+				flushFinish();
+			}
+			subscriber.complete();
+		};
+		let failed = false;
+		const subscription = source.subscribe({
+			next: (item) => {
+				if (failed) return;
+				try {
+					handleEvent(item);
+				} catch (error) {
+					failed = true;
+					subscriber.error(error);
+				}
+			},
+			error: (error) => {
+				flushFinish();
+				subscriber.error(error);
+			},
+			complete: () => {
+				if (failed) return;
+				try {
+					completeRun();
+				} catch (error) {
+					failed = true;
+					subscriber.error(error);
+				}
+			}
+		});
+		return () => subscription.unsubscribe();
+	});
+}
+
+//#endregion
+exports.GenerativeUIMiddleware = GenerativeUIMiddleware;
+//# sourceMappingURL=generative-ui-middleware.cjs.map

package/dist/runtime/generative-ui/generative-ui-middleware.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"generative-ui-middleware.cjs","names":["GENERATIVE_UI_RENDER_TOOL","GENERATIVE_UI_CONTEXT_DESCRIPTION","Middleware","Observable","EventType","GENERATIVE_UI_ACTIVITY_TYPE","GENERATIVE_UI_FUNCTION_CALL_ACTIVITY_TYPE","RenderUIArgsStream"],"sources":["../../../src/runtime/generative-ui/generative-ui-middleware.ts"],"sourcesContent":["import { EventType, Middleware } from \"@ag-ui/client\";\nimport type {\n  AbstractAgent,\n  BaseEvent,\n  Message,\n  RunAgentInput,\n  ToolCallArgsEvent,\n  ToolCallEndEvent,\n  ToolCallStartEvent,\n} from \"@ag-ui/client\";\nimport { Observable } from \"rxjs\";\nimport {\n  GENERATIVE_UI_ACTIVITY_TYPE,\n  GENERATIVE_UI_CONTEXT_DESCRIPTION,\n  GENERATIVE_UI_FUNCTION_CALL_ACTIVITY_TYPE,\n  GENERATIVE_UI_RENDER_TOOL,\n  RenderUIArgsStream,\n  buildGenerativeUIPrompt,\n  envelopeToManifest,\n  generativeUIFunctionCallMessageId,\n  generativeUISurfaceMessageId,\n  getOperationKind,\n  getOperationSurfaceId,\n  randomUUID,\n  readCopilotKitUIEnvelope,\n  validateOperation,\n} from \"@copilotkit/shared\";\nimport type {\n  A2UIOperation,\n  GenerativeUIFunctionCallContent,\n  GenerativeUIManifest,\n  GenerativeUIRejectedOperation,\n  GenerativeUISurfaceContent,\n} from \"@copilotkit/shared\";\n\ntype EventWithState = { event: BaseEvent; messages: Message[] };\ntype JsonPatchOperation = {\n  op: \"add\" | \"replace\";\n  path: string;\n  value: unknown;\n};\n\ninterface RenderCallState {\n  stream: RenderUIArgsStream;\n  target?: string;\n  buffered: Array<Record<string, unknown> | null>;\n  nextIndex: number;\n  /** Accepted surface operations. */\n  applied: number;\n  /** Accepted `callRendererFunction` operations. */\n  calls: number;\n  rejected: GenerativeUIRejectedOperation[];\n  touched: Set<string>;\n  finished: boolean;\n}\n\nfunction isRecord(value: unknown): value is Record<string, unknown> {\n  return typeof value === \"object\" && value !== null && !Array.isArray(value);\n}\n\n/**\n * Builds the manifest from the generative UI envelope\n * (`forwardedProps.copilotkit.ui`). The client sends the envelope, so it is\n * not trusted: `readCopilotKitUIEnvelope` checks every field, and\n * `envelopeToManifest` turns it into the manifest. Returns\n * `undefined` (the run passes through) when:\n * - there is no valid envelope;\n * - the transport is \"markdown\" (the model writes fences, not render_ui);\n * - the catalog has no components (the server cannot validate operations).\n */\nfunction readManifest(forwardedProps: unknown) {\n  const envelope = readCopilotKitUIEnvelope(forwardedProps);\n  if (!envelope) return undefined;\n  const manifest = envelopeToManifest(envelope);\n  const isToolRun = manifest.transport === \"tool\" && manifest.hasComponents;\n  return isToolRun ? manifest : undefined;\n}\n\nfunction prepareInput(input: RunAgentInput, manifest: GenerativeUIManifest) {\n  const tools = (input.tools ?? []).filter(\n    (t) => t.name !== GENERATIVE_UI_RENDER_TOOL,\n  );\n  tools.push({\n    name: GENERATIVE_UI_RENDER_TOOL,\n    description:\n      \"Render or update generative UI surfaces, or call app functions, with A2UI v1.0 operations. Follow the generative UI guide in context.\",\n    parameters: {\n      type: \"object\",\n      properties: {\n        target: {\n          type: \"string\",\n          enum: manifest.targets.map((t) => t.id),\n          description:\n            'Where the surface is shown: \"chat\" or a named inline target.',\n        },\n        operations: {\n          type: \"array\",\n          description: \"A2UI v1.0 operations, applied in order.\",\n          items: { type: \"object\" },\n        },\n      },\n      required: [\"target\", \"operations\"],\n    },\n  });\n  const context = (input.context ?? []).filter(\n    (c) => c.description !== GENERATIVE_UI_CONTEXT_DESCRIPTION,\n  );\n  context.push({\n    description: GENERATIVE_UI_CONTEXT_DESCRIPTION,\n    value: buildGenerativeUIPrompt(manifest),\n  });\n  return { ...input, tools, context };\n}\n\n/**\n * Experimental. Lets an agent render CopilotKit generative UI surfaces via the\n * `render_ui` tool. It applies only to runs whose generative UI envelope uses\n * the \"tool\" transport. Each complete operation in the streamed arguments is\n * validated against the catalog in the envelope and committed immediately:\n * - `ACTIVITY_SNAPSHOT` for a new surface, `ACTIVITY_DELTA` for every further\n *   operation on it.\n * - One `ACTIVITY_SNAPSHOT` of type `generative-ui-function-call` for each\n *   `callRendererFunction`. The client runs the function.\n */\nexport class GenerativeUIMiddleware extends Middleware {\n  run(input: RunAgentInput, next: AbstractAgent): Observable<BaseEvent> {\n    const manifest = readManifest(input.forwardedProps);\n    if (!manifest) return this.runNext(input, next);\n    return processStream(\n      this.runNextWithState(\n        prepareInput(input, manifest),\n        next,\n      ) as Observable<EventWithState>,\n      manifest,\n    );\n  }\n}\n\n/**\n * The first time a render_ui call touches an existing surface, mark it\n * streaming and clear its rejection list: earlier calls' rejections were\n * already reported to the model in their own tool results, and a repaired\n * surface must not keep showing them. A surface created by this call starts\n * with an empty list in its snapshot.\n */\nfunction beginSurfaceUpdate(\n  state: RenderCallState,\n  surfaceId: string,\n): JsonPatchOperation[] {\n  if (state.touched.has(surfaceId)) return [];\n  state.touched.add(surfaceId);\n  return [\n    { op: \"replace\", path: \"/status\", value: \"streaming\" },\n    { op: \"replace\", path: \"/rejected\", value: [] },\n  ];\n}\n\nfunction processStream(\n  source: Observable<EventWithState>,\n  manifest: GenerativeUIManifest,\n) {\n  return new Observable<BaseEvent>((subscriber) => {\n    const entries = Object.entries(manifest.surfaces);\n    // Only live surfaces take updates. A surface that the client deleted, or\n    // that was never created, has a `status`.\n    const surfaces = new Map<string, string>(\n      entries.filter(([, s]) => !s.status).map(([id, s]) => [id, s.target]),\n    );\n    // A surface id is unique for the life of the page. Keep the ids that were\n    // created, also after deleteSurface (in this run or an earlier one), so a\n    // second createSurface is rejected. A failed create can be sent again.\n    const createdSurfaces = new Set(\n      entries.filter(([, s]) => s.status !== \"failed\").map(([id]) => id),\n    );\n    // The targets that limit their components, by target id.\n    const targetComponents = new Map(\n      manifest.targets\n        .filter((t) => t.components !== undefined)\n        .map((t) => [t.id, new Set(t.components)]),\n    );\n    const functionCallIds = new Set<string>();\n    const calls = new Map<string, RenderCallState>();\n    let heldFinish: EventWithState | null = null;\n\n    const emit = (event: BaseEvent) => subscriber.next(event);\n    const patch = (surfaceId: string, ops: JsonPatchOperation[]) =>\n      emit({\n        type: EventType.ACTIVITY_DELTA,\n        messageId: generativeUISurfaceMessageId(surfaceId),\n        activityType: GENERATIVE_UI_ACTIVITY_TYPE,\n        patch: ops,\n      });\n\n    const processOperation = (\n      state: RenderCallState,\n      op: Record<string, unknown> | null,\n    ) => {\n      const index = state.nextIndex++;\n      const reject = (reason: string, surfaceId?: string) => {\n        const entry = { index, reason };\n        state.rejected.push(entry);\n        if (surfaceId && surfaces.has(surfaceId)) {\n          patch(surfaceId, [\n            ...beginSurfaceUpdate(state, surfaceId),\n            { op: \"add\", path: \"/rejected/-\", value: entry },\n          ]);\n        }\n      };\n      if (op === null) return reject(\"operation is not valid JSON\");\n      let surfaceId: string | undefined;\n      try {\n        surfaceId = getOperationSurfaceId(op);\n        applyOperation(state, withCatalogId(op), surfaceId, reject);\n      } catch (error) {\n        // Never let one bad operation take down the run: report it to the\n        // model through the tool result instead.\n        reject(\n          `operation could not be processed: ${error instanceof Error ? error.message : String(error)}`,\n          surfaceId,\n        );\n      }\n    };\n\n    /**\n     * The model does not have to name the catalog. A createSurface keeps a\n     * catalogId that the envelope lists, and gets the catalog id of the run\n     * for any other value. The whole operation, with its inline components\n     * and data model, stays in the snapshot.\n     */\n    const catalogIds = manifest.catalogIds ?? [manifest.catalogId];\n    const withCatalogId = (op: Record<string, unknown>) => {\n      const create = op.createSurface;\n      if (!isRecord(create)) return op;\n      const catalogId = catalogIds.includes(create.catalogId as string)\n        ? create.catalogId\n        : manifest.catalogId;\n      return { ...op, createSurface: { ...create, catalogId } };\n    };\n\n    /**\n     * The components that the target of the surface takes. A createSurface\n     * goes to the target of the render_ui call. Every other operation goes to\n     * the target that the surface was created on.\n     */\n    const allowedComponentsFor = (\n      state: RenderCallState,\n      op: Record<string, unknown>,\n      surfaceId: string | undefined,\n    ) => {\n      const target =\n        getOperationKind(op) === \"createSurface\"\n          ? state.target\n          : surfaces.get(surfaceId);\n      const names = targetComponents.get(target);\n      return names && { target, names };\n    };\n\n    const applyOperation = (\n      state: RenderCallState,\n      op: Record<string, unknown>,\n      surfaceId: string | undefined,\n      reject: (reason: string, surfaceId?: string) => void,\n    ) => {\n      const validation = validateOperation(op, {\n        manifest,\n        knownSurfaces: new Set(surfaces.keys()),\n        createdSurfaces,\n        functions: manifest.functions,\n        allowedComponents: allowedComponentsFor(state, op, surfaceId),\n      });\n      // `=== false` (not `!ok`): runtime is compiled non-strict, where `!ok`\n      // does not narrow the discriminated union.\n      if (validation.ok === false) return reject(validation.reason, surfaceId);\n      // validateOperation checked the shape of the operation.\n      const accepted = op as A2UIOperation;\n      if (\"callRendererFunction\" in accepted) {\n        return callRendererFunction(\n          state,\n          accepted.callRendererFunction,\n          reject,\n        );\n      }\n      if (\"createSurface\" in accepted) {\n        return createSurface(state, surfaceId, accepted, reject);\n      }\n      patch(surfaceId, [\n        ...beginSurfaceUpdate(state, surfaceId),\n        { op: \"add\", path: \"/operations/-\", value: accepted },\n      ]);\n      if (\"deleteSurface\" in accepted) surfaces.delete(surfaceId);\n      state.applied++;\n    };\n\n    const createSurface = (\n      state: RenderCallState,\n      surfaceId: string,\n      op: A2UIOperation,\n      reject: (reason: string) => void,\n    ) => {\n      const { target } = state;\n      if (target === undefined) {\n        return reject(\"render_ui was called without a target\");\n      }\n      if (!manifest.targets.some((t) => t.id === target)) {\n        return reject(`target \"${target}\" is not mounted`);\n      }\n      surfaces.set(surfaceId, target);\n      createdSurfaces.add(surfaceId);\n      const content: GenerativeUISurfaceContent = {\n        surfaceId,\n        target,\n        catalogId:\n          (\"createSurface\" in op && op.createSurface.catalogId) ||\n          manifest.catalogId,\n        operations: [op],\n        status: \"streaming\",\n        rejected: [],\n      };\n      emit({\n        type: EventType.ACTIVITY_SNAPSHOT,\n        messageId: generativeUISurfaceMessageId(surfaceId),\n        activityType: GENERATIVE_UI_ACTIVITY_TYPE,\n        content,\n        replace: true,\n      });\n      state.touched.add(surfaceId);\n      state.applied++;\n    };\n\n    /**\n     * Sends one function call to the client, which runs it. A function call\n     * needs no target and no surface. Each `functionCallId` runs once.\n     */\n    const callRendererFunction = (\n      state: RenderCallState,\n      content: GenerativeUIFunctionCallContent,\n      reject: (reason: string) => void,\n    ) => {\n      const { functionCallId } = content;\n      if (functionCallIds.has(functionCallId)) {\n        return reject(\n          `functionCallId \"${functionCallId}\" was used before; use a new functionCallId`,\n        );\n      }\n      functionCallIds.add(functionCallId);\n      emit({\n        type: EventType.ACTIVITY_SNAPSHOT,\n        messageId: generativeUIFunctionCallMessageId(functionCallId),\n        activityType: GENERATIVE_UI_FUNCTION_CALL_ACTIVITY_TYPE,\n        content: {\n          functionCallId,\n          callFunction: content.callFunction,\n        } satisfies GenerativeUIFunctionCallContent,\n        replace: true,\n      });\n      state.calls++;\n    };\n\n    /** Same buffering rule as streamed ops: hold them until the target is known. */\n    const acceptOperation = (\n      state: RenderCallState,\n      op: Record<string, unknown> | null,\n    ) => {\n      if (state.target === undefined) state.buffered.push(op);\n      else processOperation(state, op);\n    };\n\n    /**\n     * The streaming parser found no operations at all. Models sometimes send\n     * `operations` as a JSON-encoded string; recover those from the complete\n     * arguments. Anything still unusable leaves the call with zero operations,\n     * which the tool result reports as an error.\n     */\n    const recoverOperations = (state: RenderCallState) => {\n      let args: unknown;\n      try {\n        args = JSON.parse(state.stream.text);\n      } catch {\n        return;\n      }\n      if (!isRecord(args)) return;\n      if (state.target === undefined && typeof args.target === \"string\") {\n        state.target = args.target;\n      }\n      let operations = args.operations;\n      if (typeof operations === \"string\") {\n        try {\n          operations = JSON.parse(operations);\n        } catch {\n          return;\n        }\n      }\n      if (!Array.isArray(operations)) return;\n      for (const op of operations)\n        acceptOperation(state, isRecord(op) ? op : null);\n    };\n\n    const finishCall = (state: RenderCallState) => {\n      if (state.finished) return;\n      state.finished = true;\n      if (state.nextIndex === 0 && state.buffered.length === 0)\n        recoverOperations(state);\n      if (state.target === undefined) {\n        // A function call needs no target, so it still runs. Every other\n        // operation is rejected.\n        const buffered = state.buffered;\n        state.buffered = [];\n        for (const op of buffered) {\n          if (getOperationKind(op) === \"callRendererFunction\") {\n            processOperation(state, op);\n            continue;\n          }\n          state.rejected.push({\n            index: state.nextIndex++,\n            reason: \"render_ui was called without a target\",\n          });\n        }\n      }\n      if (state.stream.hasIncompleteOperation) {\n        // The arguments ended inside an element (for example the model hit its\n        // output limit): report it instead of silently dropping it.\n        state.rejected.push({\n          index: state.nextIndex++,\n          reason: \"operation was truncated\",\n        });\n      }\n      for (const id of state.touched) {\n        patch(id, [{ op: \"replace\", path: \"/status\", value: \"ready\" }]);\n      }\n    };\n\n    const flushFinish = () => {\n      if (heldFinish) {\n        emit(heldFinish.event);\n        heldFinish = null;\n      }\n    };\n\n    const handleEvent = ({ event, messages }: EventWithState) => {\n      flushFinish();\n      if (event.type === EventType.RUN_FINISHED) {\n        heldFinish = { event, messages };\n        return;\n      }\n      emit(event);\n      if (event.type === EventType.TOOL_CALL_START) {\n        const start = event as ToolCallStartEvent;\n        if (start.toolCallName === GENERATIVE_UI_RENDER_TOOL) {\n          calls.set(start.toolCallId, {\n            stream: new RenderUIArgsStream(),\n            buffered: [],\n            nextIndex: 0,\n            applied: 0,\n            calls: 0,\n            rejected: [],\n            touched: new Set(),\n            finished: false,\n          });\n        }\n      } else if (event.type === EventType.TOOL_CALL_ARGS) {\n        const args = event as ToolCallArgsEvent;\n        const state = calls.get(args.toolCallId);\n        if (!state || state.finished) return;\n        const { target, newOperations } = state.stream.push(args.delta);\n        if (target !== undefined && state.target === undefined) {\n          state.target = target;\n          const buffered = state.buffered;\n          state.buffered = [];\n          for (const op of buffered) processOperation(state, op);\n        }\n        for (const op of newOperations) acceptOperation(state, op);\n      } else if (event.type === EventType.TOOL_CALL_END) {\n        const state = calls.get((event as ToolCallEndEvent).toolCallId);\n        if (state) finishCall(state);\n      }\n    };\n\n    const completeRun = () => {\n      for (const state of calls.values()) finishCall(state);\n      if (heldFinish) {\n        const answered = new Set(\n          heldFinish.messages\n            .filter((m) => m.role === \"tool\")\n            .map((m) => (m as { toolCallId: string }).toolCallId),\n        );\n        for (const [toolCallId, state] of calls) {\n          if (answered.has(toolCallId)) continue;\n          const nothingParsed =\n            state.applied === 0 &&\n            state.calls === 0 &&\n            state.rejected.length === 0;\n          // `applied` counts the surface operations, `calls` the function calls.\n          emit({\n            type: EventType.TOOL_CALL_RESULT,\n            messageId: randomUUID(),\n            toolCallId,\n            content: JSON.stringify(\n              nothingParsed\n                ? {\n                    status: \"error\",\n                    applied: 0,\n                    calls: 0,\n                    rejected: [],\n                    reason:\n                      \"operations must be a JSON array of A2UI operation objects; nothing was rendered\",\n                  }\n                : {\n                    status: \"rendered\",\n                    applied: state.applied,\n                    calls: state.calls,\n                    rejected: state.rejected,\n                  },\n            ),\n          });\n        }\n        flushFinish();\n      }\n      subscriber.complete();\n    };\n\n    let failed = false;\n    const subscription = source.subscribe({\n      next: (item) => {\n        if (failed) return;\n        try {\n          handleEvent(item);\n        } catch (error) {\n          // Anything unexpected must reach the consumer as an error instead of\n          // escaping this callback, where RxJS would rethrow it asynchronously\n          // as an uncaught exception.\n          failed = true;\n          subscriber.error(error);\n        }\n      },\n      error: (error) => {\n        flushFinish();\n        subscriber.error(error);\n      },\n      complete: () => {\n        if (failed) return;\n        try {\n          completeRun();\n        } catch (error) {\n          failed = true;\n          subscriber.error(error);\n        }\n      },\n    });\n    return () => subscription.unsubscribe();\n  });\n}\n"],"mappings":";;;;;;AAwDA,SAAS,SAAS,OAAkD;AAClE,QAAO,OAAO,UAAU,YAAY,UAAU,QAAQ,CAAC,MAAM,QAAQ,MAAM;;;;;;;;;;;;AAa7E,SAAS,aAAa,gBAAyB;CAC7C,MAAM,4DAAoC,eAAe;AACzD,KAAI,CAAC,SAAU,QAAO;CACtB,MAAM,sDAA8B,SAAS;AAE7C,QADkB,SAAS,cAAc,UAAU,SAAS,gBACzC,WAAW;;AAGhC,SAAS,aAAa,OAAsB,UAAgC;CAC1E,MAAM,SAAS,MAAM,SAAS,EAAE,EAAE,QAC/B,MAAM,EAAE,SAASA,6CACnB;AACD,OAAM,KAAK;EACT,MAAMA;EACN,aACE;EACF,YAAY;GACV,MAAM;GACN,YAAY;IACV,QAAQ;KACN,MAAM;KACN,MAAM,SAAS,QAAQ,KAAK,MAAM,EAAE,GAAG;KACvC,aACE;KACH;IACD,YAAY;KACV,MAAM;KACN,aAAa;KACb,OAAO,EAAE,MAAM,UAAU;KAC1B;IACF;GACD,UAAU,CAAC,UAAU,aAAa;GACnC;EACF,CAAC;CACF,MAAM,WAAW,MAAM,WAAW,EAAE,EAAE,QACnC,MAAM,EAAE,gBAAgBC,qDAC1B;AACD,SAAQ,KAAK;EACX,aAAaA;EACb,uDAA+B,SAAS;EACzC,CAAC;AACF,QAAO;EAAE,GAAG;EAAO;EAAO;EAAS;;;;;;;;;;;;AAarC,IAAa,yBAAb,cAA4CC,yBAAW;CACrD,IAAI,OAAsB,MAA4C;EACpE,MAAM,WAAW,aAAa,MAAM,eAAe;AACnD,MAAI,CAAC,SAAU,QAAO,KAAK,QAAQ,OAAO,KAAK;AAC/C,SAAO,cACL,KAAK,iBACH,aAAa,OAAO,SAAS,EAC7B,KACD,EACD,SACD;;;;;;;;;;AAWL,SAAS,mBACP,OACA,WACsB;AACtB,KAAI,MAAM,QAAQ,IAAI,UAAU,CAAE,QAAO,EAAE;AAC3C,OAAM,QAAQ,IAAI,UAAU;AAC5B,QAAO,CACL;EAAE,IAAI;EAAW,MAAM;EAAW,OAAO;EAAa,EACtD;EAAE,IAAI;EAAW,MAAM;EAAa,OAAO,EAAE;EAAE,CAChD;;AAGH,SAAS,cACP,QACA,UACA;AACA,QAAO,IAAIC,iBAAuB,eAAe;EAC/C,MAAM,UAAU,OAAO,QAAQ,SAAS,SAAS;EAGjD,MAAM,WAAW,IAAI,IACnB,QAAQ,QAAQ,GAAG,OAAO,CAAC,EAAE,OAAO,CAAC,KAAK,CAAC,IAAI,OAAO,CAAC,IAAI,EAAE,OAAO,CAAC,CACtE;EAID,MAAM,kBAAkB,IAAI,IAC1B,QAAQ,QAAQ,GAAG,OAAO,EAAE,WAAW,SAAS,CAAC,KAAK,CAAC,QAAQ,GAAG,CACnE;EAED,MAAM,mBAAmB,IAAI,IAC3B,SAAS,QACN,QAAQ,MAAM,EAAE,eAAe,OAAU,CACzC,KAAK,MAAM,CAAC,EAAE,IAAI,IAAI,IAAI,EAAE,WAAW,CAAC,CAAC,CAC7C;EACD,MAAM,kCAAkB,IAAI,KAAa;EACzC,MAAM,wBAAQ,IAAI,KAA8B;EAChD,IAAI,aAAoC;EAExC,MAAM,QAAQ,UAAqB,WAAW,KAAK,MAAM;EACzD,MAAM,SAAS,WAAmB,QAChC,KAAK;GACH,MAAMC,wBAAU;GAChB,gEAAwC,UAAU;GAClD,cAAcC;GACd,OAAO;GACR,CAAC;EAEJ,MAAM,oBACJ,OACA,OACG;GACH,MAAM,QAAQ,MAAM;GACpB,MAAM,UAAU,QAAgB,cAAuB;IACrD,MAAM,QAAQ;KAAE;KAAO;KAAQ;AAC/B,UAAM,SAAS,KAAK,MAAM;AAC1B,QAAI,aAAa,SAAS,IAAI,UAAU,CACtC,OAAM,WAAW,CACf,GAAG,mBAAmB,OAAO,UAAU,EACvC;KAAE,IAAI;KAAO,MAAM;KAAe,OAAO;KAAO,CACjD,CAAC;;AAGN,OAAI,OAAO,KAAM,QAAO,OAAO,8BAA8B;GAC7D,IAAI;AACJ,OAAI;AACF,8DAAkC,GAAG;AACrC,mBAAe,OAAO,cAAc,GAAG,EAAE,WAAW,OAAO;YACpD,OAAO;AAGd,WACE,qCAAqC,iBAAiB,QAAQ,MAAM,UAAU,OAAO,MAAM,IAC3F,UACD;;;;;;;;;EAUL,MAAM,aAAa,SAAS,cAAc,CAAC,SAAS,UAAU;EAC9D,MAAM,iBAAiB,OAAgC;GACrD,MAAM,SAAS,GAAG;AAClB,OAAI,CAAC,SAAS,OAAO,CAAE,QAAO;GAC9B,MAAM,YAAY,WAAW,SAAS,OAAO,UAAoB,GAC7D,OAAO,YACP,SAAS;AACb,UAAO;IAAE,GAAG;IAAI,eAAe;KAAE,GAAG;KAAQ;KAAW;IAAE;;;;;;;EAQ3D,MAAM,wBACJ,OACA,IACA,cACG;GACH,MAAM,kDACa,GAAG,KAAK,kBACrB,MAAM,SACN,SAAS,IAAI,UAAU;GAC7B,MAAM,QAAQ,iBAAiB,IAAI,OAAO;AAC1C,UAAO,SAAS;IAAE;IAAQ;IAAO;;EAGnC,MAAM,kBACJ,OACA,IACA,WACA,WACG;GACH,MAAM,uDAA+B,IAAI;IACvC;IACA,eAAe,IAAI,IAAI,SAAS,MAAM,CAAC;IACvC;IACA,WAAW,SAAS;IACpB,mBAAmB,qBAAqB,OAAO,IAAI,UAAU;IAC9D,CAAC;AAGF,OAAI,WAAW,OAAO,MAAO,QAAO,OAAO,WAAW,QAAQ,UAAU;GAExE,MAAM,WAAW;AACjB,OAAI,0BAA0B,SAC5B,QAAO,qBACL,OACA,SAAS,sBACT,OACD;AAEH,OAAI,mBAAmB,SACrB,QAAO,cAAc,OAAO,WAAW,UAAU,OAAO;AAE1D,SAAM,WAAW,CACf,GAAG,mBAAmB,OAAO,UAAU,EACvC;IAAE,IAAI;IAAO,MAAM;IAAiB,OAAO;IAAU,CACtD,CAAC;AACF,OAAI,mBAAmB,SAAU,UAAS,OAAO,UAAU;AAC3D,SAAM;;EAGR,MAAM,iBACJ,OACA,WACA,IACA,WACG;GACH,MAAM,EAAE,WAAW;AACnB,OAAI,WAAW,OACb,QAAO,OAAO,wCAAwC;AAExD,OAAI,CAAC,SAAS,QAAQ,MAAM,MAAM,EAAE,OAAO,OAAO,CAChD,QAAO,OAAO,WAAW,OAAO,kBAAkB;AAEpD,YAAS,IAAI,WAAW,OAAO;AAC/B,mBAAgB,IAAI,UAAU;GAC9B,MAAM,UAAsC;IAC1C;IACA;IACA,WACG,mBAAmB,MAAM,GAAG,cAAc,aAC3C,SAAS;IACX,YAAY,CAAC,GAAG;IAChB,QAAQ;IACR,UAAU,EAAE;IACb;AACD,QAAK;IACH,MAAMD,wBAAU;IAChB,gEAAwC,UAAU;IAClD,cAAcC;IACd;IACA,SAAS;IACV,CAAC;AACF,SAAM,QAAQ,IAAI,UAAU;AAC5B,SAAM;;;;;;EAOR,MAAM,wBACJ,OACA,SACA,WACG;GACH,MAAM,EAAE,mBAAmB;AAC3B,OAAI,gBAAgB,IAAI,eAAe,CACrC,QAAO,OACL,mBAAmB,eAAe,6CACnC;AAEH,mBAAgB,IAAI,eAAe;AACnC,QAAK;IACH,MAAMD,wBAAU;IAChB,qEAA6C,eAAe;IAC5D,cAAcE;IACd,SAAS;KACP;KACA,cAAc,QAAQ;KACvB;IACD,SAAS;IACV,CAAC;AACF,SAAM;;;EAIR,MAAM,mBACJ,OACA,OACG;AACH,OAAI,MAAM,WAAW,OAAW,OAAM,SAAS,KAAK,GAAG;OAClD,kBAAiB,OAAO,GAAG;;;;;;;;EASlC,MAAM,qBAAqB,UAA2B;GACpD,IAAI;AACJ,OAAI;AACF,WAAO,KAAK,MAAM,MAAM,OAAO,KAAK;WAC9B;AACN;;AAEF,OAAI,CAAC,SAAS,KAAK,CAAE;AACrB,OAAI,MAAM,WAAW,UAAa,OAAO,KAAK,WAAW,SACvD,OAAM,SAAS,KAAK;GAEtB,IAAI,aAAa,KAAK;AACtB,OAAI,OAAO,eAAe,SACxB,KAAI;AACF,iBAAa,KAAK,MAAM,WAAW;WAC7B;AACN;;AAGJ,OAAI,CAAC,MAAM,QAAQ,WAAW,CAAE;AAChC,QAAK,MAAM,MAAM,WACf,iBAAgB,OAAO,SAAS,GAAG,GAAG,KAAK,KAAK;;EAGpD,MAAM,cAAc,UAA2B;AAC7C,OAAI,MAAM,SAAU;AACpB,SAAM,WAAW;AACjB,OAAI,MAAM,cAAc,KAAK,MAAM,SAAS,WAAW,EACrD,mBAAkB,MAAM;AAC1B,OAAI,MAAM,WAAW,QAAW;IAG9B,MAAM,WAAW,MAAM;AACvB,UAAM,WAAW,EAAE;AACnB,SAAK,MAAM,MAAM,UAAU;AACzB,kDAAqB,GAAG,KAAK,wBAAwB;AACnD,uBAAiB,OAAO,GAAG;AAC3B;;AAEF,WAAM,SAAS,KAAK;MAClB,OAAO,MAAM;MACb,QAAQ;MACT,CAAC;;;AAGN,OAAI,MAAM,OAAO,uBAGf,OAAM,SAAS,KAAK;IAClB,OAAO,MAAM;IACb,QAAQ;IACT,CAAC;AAEJ,QAAK,MAAM,MAAM,MAAM,QACrB,OAAM,IAAI,CAAC;IAAE,IAAI;IAAW,MAAM;IAAW,OAAO;IAAS,CAAC,CAAC;;EAInE,MAAM,oBAAoB;AACxB,OAAI,YAAY;AACd,SAAK,WAAW,MAAM;AACtB,iBAAa;;;EAIjB,MAAM,eAAe,EAAE,OAAO,eAA+B;AAC3D,gBAAa;AACb,OAAI,MAAM,SAASF,wBAAU,cAAc;AACzC,iBAAa;KAAE;KAAO;KAAU;AAChC;;AAEF,QAAK,MAAM;AACX,OAAI,MAAM,SAASA,wBAAU,iBAAiB;IAC5C,MAAM,QAAQ;AACd,QAAI,MAAM,iBAAiBJ,6CACzB,OAAM,IAAI,MAAM,YAAY;KAC1B,QAAQ,IAAIO,uCAAoB;KAChC,UAAU,EAAE;KACZ,WAAW;KACX,SAAS;KACT,OAAO;KACP,UAAU,EAAE;KACZ,yBAAS,IAAI,KAAK;KAClB,UAAU;KACX,CAAC;cAEK,MAAM,SAASH,wBAAU,gBAAgB;IAClD,MAAM,OAAO;IACb,MAAM,QAAQ,MAAM,IAAI,KAAK,WAAW;AACxC,QAAI,CAAC,SAAS,MAAM,SAAU;IAC9B,MAAM,EAAE,QAAQ,kBAAkB,MAAM,OAAO,KAAK,KAAK,MAAM;AAC/D,QAAI,WAAW,UAAa,MAAM,WAAW,QAAW;AACtD,WAAM,SAAS;KACf,MAAM,WAAW,MAAM;AACvB,WAAM,WAAW,EAAE;AACnB,UAAK,MAAM,MAAM,SAAU,kBAAiB,OAAO,GAAG;;AAExD,SAAK,MAAM,MAAM,cAAe,iBAAgB,OAAO,GAAG;cACjD,MAAM,SAASA,wBAAU,eAAe;IACjD,MAAM,QAAQ,MAAM,IAAK,MAA2B,WAAW;AAC/D,QAAI,MAAO,YAAW,MAAM;;;EAIhC,MAAM,oBAAoB;AACxB,QAAK,MAAM,SAAS,MAAM,QAAQ,CAAE,YAAW,MAAM;AACrD,OAAI,YAAY;IACd,MAAM,WAAW,IAAI,IACnB,WAAW,SACR,QAAQ,MAAM,EAAE,SAAS,OAAO,CAChC,KAAK,MAAO,EAA6B,WAAW,CACxD;AACD,SAAK,MAAM,CAAC,YAAY,UAAU,OAAO;AACvC,SAAI,SAAS,IAAI,WAAW,CAAE;KAC9B,MAAM,gBACJ,MAAM,YAAY,KAClB,MAAM,UAAU,KAChB,MAAM,SAAS,WAAW;AAE5B,UAAK;MACH,MAAMA,wBAAU;MAChB,+CAAuB;MACvB;MACA,SAAS,KAAK,UACZ,gBACI;OACE,QAAQ;OACR,SAAS;OACT,OAAO;OACP,UAAU,EAAE;OACZ,QACE;OACH,GACD;OACE,QAAQ;OACR,SAAS,MAAM;OACf,OAAO,MAAM;OACb,UAAU,MAAM;OACjB,CACN;MACF,CAAC;;AAEJ,iBAAa;;AAEf,cAAW,UAAU;;EAGvB,IAAI,SAAS;EACb,MAAM,eAAe,OAAO,UAAU;GACpC,OAAO,SAAS;AACd,QAAI,OAAQ;AACZ,QAAI;AACF,iBAAY,KAAK;aACV,OAAO;AAId,cAAS;AACT,gBAAW,MAAM,MAAM;;;GAG3B,QAAQ,UAAU;AAChB,iBAAa;AACb,eAAW,MAAM,MAAM;;GAEzB,gBAAgB;AACd,QAAI,OAAQ;AACZ,QAAI;AACF,kBAAa;aACN,OAAO;AACd,cAAS;AACT,gBAAW,MAAM,MAAM;;;GAG5B,CAAC;AACF,eAAa,aAAa,aAAa;GACvC"}

package/dist/runtime/generative-ui/generative-ui-middleware.d.cts

@@ -0,0 +1,20 @@
+import { AbstractAgent, BaseEvent, Middleware, RunAgentInput } from "@ag-ui/client";
+import { Observable } from "rxjs";
+
+//#region src/runtime/generative-ui/generative-ui-middleware.d.ts
+/**
+ * Experimental. Lets an agent render CopilotKit generative UI surfaces via the
+ * `render_ui` tool. It applies only to runs whose generative UI envelope uses
+ * the "tool" transport. Each complete operation in the streamed arguments is
+ * validated against the catalog in the envelope and committed immediately:
+ * - `ACTIVITY_SNAPSHOT` for a new surface, `ACTIVITY_DELTA` for every further
+ *   operation on it.
+ * - One `ACTIVITY_SNAPSHOT` of type `generative-ui-function-call` for each
+ *   `callRendererFunction`. The client runs the function.
+ */
+declare class GenerativeUIMiddleware extends Middleware {
+  run(input: RunAgentInput, next: AbstractAgent): Observable<BaseEvent>;
+}
+//#endregion
+export { GenerativeUIMiddleware };
+//# sourceMappingURL=generative-ui-middleware.d.cts.map

package/dist/runtime/generative-ui/generative-ui-middleware.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"generative-ui-middleware.d.cts","names":[],"sources":["../../../src/runtime/generative-ui/generative-ui-middleware.ts"],"mappings":";;;;;AA4HA;;;;;;;;;cAAa,sBAAA,SAA+B,UAAA;EAC1C,GAAA,CAAI,KAAA,EAAO,aAAA,EAAe,IAAA,EAAM,aAAA,GAAgB,UAAA,CAAW,SAAA;AAAA"}

package/dist/runtime/generative-ui/generative-ui-middleware.d.mts

@@ -0,0 +1,20 @@
+import { Observable } from "rxjs";
+import { AbstractAgent, BaseEvent, Middleware, RunAgentInput } from "@ag-ui/client";
+
+//#region src/runtime/generative-ui/generative-ui-middleware.d.ts
+/**
+ * Experimental. Lets an agent render CopilotKit generative UI surfaces via the
+ * `render_ui` tool. It applies only to runs whose generative UI envelope uses
+ * the "tool" transport. Each complete operation in the streamed arguments is
+ * validated against the catalog in the envelope and committed immediately:
+ * - `ACTIVITY_SNAPSHOT` for a new surface, `ACTIVITY_DELTA` for every further
+ *   operation on it.
+ * - One `ACTIVITY_SNAPSHOT` of type `generative-ui-function-call` for each
+ *   `callRendererFunction`. The client runs the function.
+ */
+declare class GenerativeUIMiddleware extends Middleware {
+  run(input: RunAgentInput, next: AbstractAgent): Observable<BaseEvent>;
+}
+//#endregion
+export { GenerativeUIMiddleware };
+//# sourceMappingURL=generative-ui-middleware.d.mts.map

package/dist/runtime/generative-ui/generative-ui-middleware.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"generative-ui-middleware.d.mts","names":[],"sources":["../../../src/runtime/generative-ui/generative-ui-middleware.ts"],"mappings":";;;;;AA4HA;;;;;;;;;cAAa,sBAAA,SAA+B,UAAA;EAC1C,GAAA,CAAI,KAAA,EAAO,aAAA,EAAe,IAAA,EAAM,aAAA,GAAgB,UAAA,CAAW,SAAA;AAAA"}