@copilotkit/runtime 1.80.0 → 1.81.0

package/dist/v2/runtime/generative-ui/generative-ui-middleware.mjs

@@ -0,0 +1,393 @@
+import "reflect-metadata";
+import { GENERATIVE_UI_ACTIVITY_TYPE, GENERATIVE_UI_CONTEXT_DESCRIPTION, GENERATIVE_UI_FUNCTION_CALL_ACTIVITY_TYPE, GENERATIVE_UI_RENDER_TOOL, RenderUIArgsStream, buildGenerativeUIPrompt, envelopeToManifest, generativeUIFunctionCallMessageId, generativeUISurfaceMessageId, getOperationKind, getOperationSurfaceId, randomUUID, readCopilotKitUIEnvelope, validateOperation } from "@copilotkit/shared";
+import { Observable } from "rxjs";
+import { EventType, Middleware } from "@ag-ui/client";
+
+//#region src/v2/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 = readCopilotKitUIEnvelope(forwardedProps);
+	if (!envelope) return void 0;
+	const manifest = envelopeToManifest(envelope);
+	return manifest.transport === "tool" && manifest.hasComponents ? manifest : void 0;
+}
+function prepareInput(input, manifest) {
+	const tools = (input.tools ?? []).filter((t) => t.name !== GENERATIVE_UI_RENDER_TOOL);
+	tools.push({
+		name: 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 !== GENERATIVE_UI_CONTEXT_DESCRIPTION);
+	context.push({
+		description: GENERATIVE_UI_CONTEXT_DESCRIPTION,
+		value: 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 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 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: EventType.ACTIVITY_DELTA,
+			messageId: generativeUISurfaceMessageId(surfaceId),
+			activityType: 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 = 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 = 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 = 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: EventType.ACTIVITY_SNAPSHOT,
+				messageId: generativeUISurfaceMessageId(surfaceId),
+				activityType: 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: EventType.ACTIVITY_SNAPSHOT,
+				messageId: generativeUIFunctionCallMessageId(functionCallId),
+				activityType: 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 (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 === EventType.RUN_FINISHED) {
+				heldFinish = {
+					event,
+					messages
+				};
+				return;
+			}
+			emit(event);
+			if (event.type === EventType.TOOL_CALL_START) {
+				const start = event;
+				if (start.toolCallName === GENERATIVE_UI_RENDER_TOOL) calls.set(start.toolCallId, {
+					stream: new RenderUIArgsStream(),
+					buffered: [],
+					nextIndex: 0,
+					applied: 0,
+					calls: 0,
+					rejected: [],
+					touched: /* @__PURE__ */ new Set(),
+					finished: false
+				});
+			} else if (event.type === 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 === 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: EventType.TOOL_CALL_RESULT,
+						messageId: 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
+export { GenerativeUIMiddleware };
+//# sourceMappingURL=generative-ui-middleware.mjs.map

package/dist/v2/runtime/generative-ui/generative-ui-middleware.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"generative-ui-middleware.mjs","names":[],"sources":["../../../../src/v2/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,WAAW,yBAAyB,eAAe;AACzD,KAAI,CAAC,SAAU,QAAO;CACtB,MAAM,WAAW,mBAAmB,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,SAAS,0BACnB;AACD,OAAM,KAAK;EACT,MAAM;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,gBAAgB,kCAC1B;AACD,SAAQ,KAAK;EACX,aAAa;EACb,OAAO,wBAAwB,SAAS;EACzC,CAAC;AACF,QAAO;EAAE,GAAG;EAAO;EAAO;EAAS;;;;;;;;;;;;AAarC,IAAa,yBAAb,cAA4C,WAAW;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,IAAI,YAAuB,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,MAAM,UAAU;GAChB,WAAW,6BAA6B,UAAU;GAClD,cAAc;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,gBAAY,sBAAsB,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,SACJ,iBAAiB,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,aAAa,kBAAkB,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,MAAM,UAAU;IAChB,WAAW,6BAA6B,UAAU;IAClD,cAAc;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,MAAM,UAAU;IAChB,WAAW,kCAAkC,eAAe;IAC5D,cAAc;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,SAAI,iBAAiB,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,SAAS,UAAU,cAAc;AACzC,iBAAa;KAAE;KAAO;KAAU;AAChC;;AAEF,QAAK,MAAM;AACX,OAAI,MAAM,SAAS,UAAU,iBAAiB;IAC5C,MAAM,QAAQ;AACd,QAAI,MAAM,iBAAiB,0BACzB,OAAM,IAAI,MAAM,YAAY;KAC1B,QAAQ,IAAI,oBAAoB;KAChC,UAAU,EAAE;KACZ,WAAW;KACX,SAAS;KACT,OAAO;KACP,UAAU,EAAE;KACZ,yBAAS,IAAI,KAAK;KAClB,UAAU;KACX,CAAC;cAEK,MAAM,SAAS,UAAU,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,SAAS,UAAU,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,MAAM,UAAU;MAChB,WAAW,YAAY;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/v2/runtime/generative-ui/index.d.cts

@@ -0,0 +1,3 @@
+
+import { GenerativeUIMiddleware } from "./generative-ui-middleware.cjs";
+import { GenerativeUISurfaceWriter, createSurfaceWriter } from "./surface-writer.cjs";

package/dist/v2/runtime/generative-ui/index.d.mts

@@ -0,0 +1,3 @@
+
+import { GenerativeUIMiddleware } from "./generative-ui-middleware.mjs";
+import { GenerativeUISurfaceWriter, createSurfaceWriter } from "./surface-writer.mjs";

package/dist/v2/runtime/generative-ui/surface-writer.cjs

@@ -0,0 +1,86 @@
+require("reflect-metadata");
+const require_runtime = require('../../../_virtual/_rolldown/runtime.cjs');
+let _copilotkit_shared = require("@copilotkit/shared");
+let _ag_ui_client = require("@ag-ui/client");
+
+//#region src/v2/runtime/generative-ui/surface-writer.ts
+/**
+* Experimental. Programmatic façade for custom agents and workflows: emits the
+* same `generative-ui-surface` activity events as `render_ui`, one per call.
+* Every operation is A2UI v1.0.
+* Does not validate against the browser manifest; the client still validates
+* every operation before applying it.
+*/
+function createSurfaceWriter(emit) {
+	return { openSurface({ target, surfaceId, catalog }) {
+		const messageId = (0, _copilotkit_shared.generativeUISurfaceMessageId)(surfaceId);
+		const patch = (ops) => emit({
+			type: _ag_ui_client.EventType.ACTIVITY_DELTA,
+			messageId,
+			activityType: _copilotkit_shared.GENERATIVE_UI_ACTIVITY_TYPE,
+			patch: ops
+		});
+		const append = (op) => patch([{
+			op: "add",
+			path: "/operations/-",
+			value: op
+		}]);
+		const content = {
+			surfaceId,
+			target,
+			catalogId: catalog,
+			operations: [{
+				version: "v1.0",
+				createSurface: {
+					surfaceId,
+					catalogId: catalog
+				}
+			}],
+			status: "streaming",
+			rejected: []
+		};
+		emit({
+			type: _ag_ui_client.EventType.ACTIVITY_SNAPSHOT,
+			messageId,
+			activityType: _copilotkit_shared.GENERATIVE_UI_ACTIVITY_TYPE,
+			content,
+			replace: true
+		});
+		return {
+			setRoot: (node) => append({
+				version: "v1.0",
+				updateComponents: {
+					surfaceId,
+					components: [{
+						...node,
+						id: _copilotkit_shared.GENERATIVE_UI_ROOT_ID
+					}]
+				}
+			}),
+			upsert: (...nodes) => append({
+				version: "v1.0",
+				updateComponents: {
+					surfaceId,
+					components: nodes
+				}
+			}),
+			updateData: (path, value) => append({
+				version: "v1.0",
+				updateDataModel: {
+					surfaceId,
+					path,
+					value
+				}
+			}),
+			close: () => patch([{
+				op: "replace",
+				path: "/status",
+				value: "ready"
+			}])
+		};
+	} };
+}
+
+//#endregion
+exports.createSurfaceWriter = createSurfaceWriter;
+//# sourceMappingURL=surface-writer.cjs.map

package/dist/v2/runtime/generative-ui/surface-writer.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"surface-writer.cjs","names":["EventType","GENERATIVE_UI_ACTIVITY_TYPE","GENERATIVE_UI_ROOT_ID"],"sources":["../../../../src/v2/runtime/generative-ui/surface-writer.ts"],"sourcesContent":["import { EventType } from \"@ag-ui/client\";\nimport type { BaseEvent } from \"@ag-ui/client\";\nimport {\n  GENERATIVE_UI_ACTIVITY_TYPE,\n  GENERATIVE_UI_ROOT_ID,\n  generativeUISurfaceMessageId,\n} from \"@copilotkit/shared\";\nimport type {\n  A2UIComponentNode,\n  A2UIOperation,\n  GenerativeUISurfaceContent,\n} from \"@copilotkit/shared\";\n\nexport interface GenerativeUISurfaceWriter {\n  setRoot(node: Omit<A2UIComponentNode, \"id\">): void;\n  upsert(...nodes: A2UIComponentNode[]): void;\n  /** Sets `value` at `path`. A `null` value deletes the key at `path`. */\n  updateData(path: string, value: unknown): void;\n  close(): void;\n}\n\n/**\n * Experimental. Programmatic façade for custom agents and workflows: emits the\n * same `generative-ui-surface` activity events as `render_ui`, one per call.\n * Every operation is A2UI v1.0.\n * Does not validate against the browser manifest; the client still validates\n * every operation before applying it.\n */\nexport function createSurfaceWriter(emit: (event: BaseEvent) => void) {\n  return {\n    openSurface({\n      target,\n      surfaceId,\n      catalog,\n    }: {\n      target: string;\n      surfaceId: string;\n      catalog: string;\n    }): GenerativeUISurfaceWriter {\n      const messageId = generativeUISurfaceMessageId(surfaceId);\n      const patch = (ops: unknown[]) =>\n        emit({\n          type: EventType.ACTIVITY_DELTA,\n          messageId,\n          activityType: GENERATIVE_UI_ACTIVITY_TYPE,\n          patch: ops,\n        });\n      const append = (op: A2UIOperation) =>\n        patch([{ op: \"add\", path: \"/operations/-\", value: op }]);\n      const content: GenerativeUISurfaceContent = {\n        surfaceId,\n        target,\n        catalogId: catalog,\n        operations: [\n          { version: \"v1.0\", createSurface: { surfaceId, catalogId: catalog } },\n        ],\n        status: \"streaming\",\n        rejected: [],\n      };\n      emit({\n        type: EventType.ACTIVITY_SNAPSHOT,\n        messageId,\n        activityType: GENERATIVE_UI_ACTIVITY_TYPE,\n        content,\n        replace: true,\n      });\n      return {\n        setRoot: (node) =>\n          append({\n            version: \"v1.0\",\n            updateComponents: {\n              surfaceId,\n              components: [\n                { ...node, id: GENERATIVE_UI_ROOT_ID } as A2UIComponentNode,\n              ],\n            },\n          }),\n        upsert: (...nodes) =>\n          append({\n            version: \"v1.0\",\n            updateComponents: { surfaceId, components: nodes },\n          }),\n        updateData: (path, value) =>\n          append({\n            version: \"v1.0\",\n            updateDataModel: { surfaceId, path, value },\n          }),\n        close: () =>\n          patch([{ op: \"replace\", path: \"/status\", value: \"ready\" }]),\n      };\n    },\n  };\n}\n"],"mappings":";;;;;;;;;;;;;AA4BA,SAAgB,oBAAoB,MAAkC;AACpE,QAAO,EACL,YAAY,EACV,QACA,WACA,WAK4B;EAC5B,MAAM,iEAAyC,UAAU;EACzD,MAAM,SAAS,QACb,KAAK;GACH,MAAMA,wBAAU;GAChB;GACA,cAAcC;GACd,OAAO;GACR,CAAC;EACJ,MAAM,UAAU,OACd,MAAM,CAAC;GAAE,IAAI;GAAO,MAAM;GAAiB,OAAO;GAAI,CAAC,CAAC;EAC1D,MAAM,UAAsC;GAC1C;GACA;GACA,WAAW;GACX,YAAY,CACV;IAAE,SAAS;IAAQ,eAAe;KAAE;KAAW,WAAW;KAAS;IAAE,CACtE;GACD,QAAQ;GACR,UAAU,EAAE;GACb;AACD,OAAK;GACH,MAAMD,wBAAU;GAChB;GACA,cAAcC;GACd;GACA,SAAS;GACV,CAAC;AACF,SAAO;GACL,UAAU,SACR,OAAO;IACL,SAAS;IACT,kBAAkB;KAChB;KACA,YAAY,CACV;MAAE,GAAG;MAAM,IAAIC;MAAuB,CACvC;KACF;IACF,CAAC;GACJ,SAAS,GAAG,UACV,OAAO;IACL,SAAS;IACT,kBAAkB;KAAE;KAAW,YAAY;KAAO;IACnD,CAAC;GACJ,aAAa,MAAM,UACjB,OAAO;IACL,SAAS;IACT,iBAAiB;KAAE;KAAW;KAAM;KAAO;IAC5C,CAAC;GACJ,aACE,MAAM,CAAC;IAAE,IAAI;IAAW,MAAM;IAAW,OAAO;IAAS,CAAC,CAAC;GAC9D;IAEJ"}

package/dist/v2/runtime/generative-ui/surface-writer.d.cts

@@ -0,0 +1,33 @@
+
+import { A2UIComponentNode } from "@copilotkit/shared";
+import { BaseEvent } from "@ag-ui/client";
+
+//#region src/v2/runtime/generative-ui/surface-writer.d.ts
+interface GenerativeUISurfaceWriter {
+  setRoot(node: Omit<A2UIComponentNode, "id">): void;
+  upsert(...nodes: A2UIComponentNode[]): void;
+  /** Sets `value` at `path`. A `null` value deletes the key at `path`. */
+  updateData(path: string, value: unknown): void;
+  close(): void;
+}
+/**
+ * Experimental. Programmatic façade for custom agents and workflows: emits the
+ * same `generative-ui-surface` activity events as `render_ui`, one per call.
+ * Every operation is A2UI v1.0.
+ * Does not validate against the browser manifest; the client still validates
+ * every operation before applying it.
+ */
+declare function createSurfaceWriter(emit: (event: BaseEvent) => void): {
+  openSurface({
+    target,
+    surfaceId,
+    catalog
+  }: {
+    target: string;
+    surfaceId: string;
+    catalog: string;
+  }): GenerativeUISurfaceWriter;
+};
+//#endregion
+export { GenerativeUISurfaceWriter, createSurfaceWriter };
+//# sourceMappingURL=surface-writer.d.cts.map

package/dist/v2/runtime/generative-ui/surface-writer.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"surface-writer.d.cts","names":[],"sources":["../../../../src/v2/runtime/generative-ui/surface-writer.ts"],"mappings":";;;;;UAaiB,yBAAA;EACf,OAAA,CAAQ,IAAA,EAAM,IAAA,CAAK,iBAAA;EACnB,MAAA,IAAU,KAAA,EAAO,iBAAA;EAFuB;EAIxC,UAAA,CAAW,IAAA,UAAc,KAAA;EACzB,KAAA;AAAA;;;;;;;;iBAUc,mBAAA,CAAoB,IAAA,GAAO,KAAA,EAAO,SAAA;;;;;;IAO5C,MAAA;IACA,SAAA;IACA,OAAA;EAAA,IACE,yBAAA;AAAA"}

package/dist/v2/runtime/generative-ui/surface-writer.d.mts

@@ -0,0 +1,33 @@
+
+import { A2UIComponentNode } from "@copilotkit/shared";
+import { BaseEvent } from "@ag-ui/client";
+
+//#region src/v2/runtime/generative-ui/surface-writer.d.ts
+interface GenerativeUISurfaceWriter {
+  setRoot(node: Omit<A2UIComponentNode, "id">): void;
+  upsert(...nodes: A2UIComponentNode[]): void;
+  /** Sets `value` at `path`. A `null` value deletes the key at `path`. */
+  updateData(path: string, value: unknown): void;
+  close(): void;
+}
+/**
+ * Experimental. Programmatic façade for custom agents and workflows: emits the
+ * same `generative-ui-surface` activity events as `render_ui`, one per call.
+ * Every operation is A2UI v1.0.
+ * Does not validate against the browser manifest; the client still validates
+ * every operation before applying it.
+ */
+declare function createSurfaceWriter(emit: (event: BaseEvent) => void): {
+  openSurface({
+    target,
+    surfaceId,
+    catalog
+  }: {
+    target: string;
+    surfaceId: string;
+    catalog: string;
+  }): GenerativeUISurfaceWriter;
+};
+//#endregion
+export { GenerativeUISurfaceWriter, createSurfaceWriter };
+//# sourceMappingURL=surface-writer.d.mts.map

package/dist/v2/runtime/generative-ui/surface-writer.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"surface-writer.d.mts","names":[],"sources":["../../../../src/v2/runtime/generative-ui/surface-writer.ts"],"mappings":";;;;;UAaiB,yBAAA;EACf,OAAA,CAAQ,IAAA,EAAM,IAAA,CAAK,iBAAA;EACnB,MAAA,IAAU,KAAA,EAAO,iBAAA;EAFuB;EAIxC,UAAA,CAAW,IAAA,UAAc,KAAA;EACzB,KAAA;AAAA;;;;;;;;iBAUc,mBAAA,CAAoB,IAAA,GAAO,KAAA,EAAO,SAAA;;;;;;IAO5C,MAAA;IACA,SAAA;IACA,OAAA;EAAA,IACE,yBAAA;AAAA"}