ai 7.0.136 → 7.0.138

package/dist/index.js

@@ -1174,7 +1174,7 @@ function cloneValue(value) {
 }
 //#endregion
 //#region src/version.ts
-const VERSION = "7.0.136";
+const VERSION = "7.0.138";
 //#endregion
 //#region src/util/download/download.ts
 /**
@@ -1396,17 +1396,6 @@ function convertToLanguageModelV4FilePart(content) {
 //#region src/prompt/convert-to-language-model-prompt.ts
 async function convertToLanguageModelPrompt({ prompt, supportedUrls, download, abortSignal, provider }) {
 	const downloadedAssets = await downloadAssets(prompt.messages, download ?? createDefaultDownloadFunction(void 0, abortSignal), supportedUrls);
-	const approvalIdToToolCallId = /* @__PURE__ */ new Map();
-	for (const message of prompt.messages) if (message.role === "assistant" && Array.isArray(message.content)) {
-		for (const part of message.content) if (part.type === "tool-approval-request" && "approvalId" in part && "toolCallId" in part) approvalIdToToolCallId.set(part.approvalId, part.toolCallId);
-	}
-	const approvedToolCallIds = /* @__PURE__ */ new Set();
-	for (const message of prompt.messages) if (message.role === "tool") {
-		for (const part of message.content) if (part.type === "tool-approval-response") {
-			const toolCallId = approvalIdToToolCallId.get(part.approvalId);
-			if (toolCallId) approvedToolCallIds.add(toolCallId);
-		}
-	}
 	const messages = [...prompt.instructions != null ? typeof prompt.instructions === "string" ? [{
 		role: "system",
 		content: prompt.instructions
@@ -1442,11 +1431,8 @@ async function convertToLanguageModelPrompt({ prompt, supportedUrls, download, a
 			for (const content of message.content) if (content.type === "tool-result") toolCallIds.delete(content.toolCallId);
 			break;
 		case "user":
-		case "system":
-			for (const id of approvedToolCallIds) toolCallIds.delete(id);
-			if (toolCallIds.size > 0) throw new MissingToolResultsError({ toolCallIds: Array.from(toolCallIds) });
+		case "system": if (toolCallIds.size > 0) throw new MissingToolResultsError({ toolCallIds: Array.from(toolCallIds) });
 	}
-	for (const id of approvedToolCallIds) toolCallIds.delete(id);
 	if (toolCallIds.size > 0) throw new MissingToolResultsError({ toolCallIds: Array.from(toolCallIds) });
 	return combinedMessages.filter((message) => message.role !== "tool" || message.content.length > 0);
 }
@@ -6276,6 +6262,29 @@ function getResponseUIMessageId({ originalMessages, responseMessageId }) {
 	return lastMessage?.role === "assistant" ? lastMessage.id : typeof responseMessageId === "function" ? responseMessageId() : responseMessageId;
 }
 //#endregion
+//#region src/types/citation.ts
+const citationSchema = z.object({
+	source: z.discriminatedUnion("sourceType", [z.object({
+		type: z.literal("source"),
+		sourceType: z.literal("url"),
+		id: z.string(),
+		url: z.string(),
+		title: z.string().optional(),
+		providerMetadata: providerMetadataSchema.optional()
+	}), z.object({
+		type: z.literal("source"),
+		sourceType: z.literal("document"),
+		id: z.string(),
+		mediaType: z.string(),
+		title: z.string(),
+		filename: z.string().optional(),
+		providerMetadata: providerMetadataSchema.optional()
+	})]),
+	startIndex: z.number().int().nonnegative().optional(),
+	endIndex: z.number().int().nonnegative().optional(),
+	citedText: z.string().optional()
+});
+//#endregion
 //#region src/ui-message-stream/ui-message-chunks.ts
 const toolMetadataSchema$1 = z.record(z.string(), jsonValueSchema.optional());
 const uiMessageChunkSchema = lazySchema(() => zodSchema$1(z.union([
@@ -6293,6 +6302,7 @@ const uiMessageChunkSchema = lazySchema(() => zodSchema$1(z.union([
 	z.looseObject({
 		type: z.literal("text-end"),
 		id: z.string(),
+		citations: z.array(citationSchema).optional(),
 		providerMetadata: providerMetadataSchema.optional()
 	}),
 	z.looseObject({
@@ -6604,6 +6614,7 @@ function createStreamingUIMessageState({ lastMessage, messageId }) {
 	};
 }
 function processUIMessageStream({ stream, messageMetadataSchema, dataPartSchemas, runUpdateMessageJob, onError, onToolCall, onData, resetStateOnMessageIdChange = false, resetStateOnFirstMessageStart = false }) {
+	let isFirstMessageStart = true;
 	return stream.pipeThrough(new TransformStream({ async transform(chunk, controller) {
 		await runUpdateMessageJob(async ({ state, write }) => {
 			function getCurrentStepParts() {
@@ -6765,6 +6776,7 @@ function processUIMessageStream({ stream, messageMetadataSchema, dataPartSchemas
 						message: `Received text-end for missing text part with ID "${chunk.id}". Ensure a "text-start" chunk is sent before any "text-end" chunks.`
 					});
 					textPart.state = "done";
+					if (chunk.citations != null) textPart.citations = chunk.citations;
 					textPart.providerMetadata = chunk.providerMetadata ?? textPart.providerMetadata;
 					delete state.activeTextParts[chunk.id];
 					write();
@@ -7068,15 +7080,18 @@ function processUIMessageStream({ stream, messageMetadataSchema, dataPartSchemas
 					}
 					break;
 				}
-				case "start":
-					if (chunk.messageId != null && (resetStateOnFirstMessageStart || resetStateOnMessageIdChange && chunk.messageId !== state.message.id)) Object.assign(state, createStreamingUIMessageState({
+				case "start": {
+					const shouldReset = resetStateOnFirstMessageStart && isFirstMessageStart || resetStateOnMessageIdChange && chunk.messageId != null && chunk.messageId !== state.message.id;
+					isFirstMessageStart = false;
+					if (shouldReset) Object.assign(state, createStreamingUIMessageState({
 						lastMessage: void 0,
-						messageId: chunk.messageId
+						messageId: chunk.messageId ?? state.message.id
 					}), { finishReason: void 0 });
 					if (chunk.messageId != null) state.message.id = chunk.messageId;
 					await updateMessageMetadata(chunk.messageMetadata);
-					if (chunk.messageId != null || chunk.messageMetadata != null) write({ updateStatus: false });
+					if (shouldReset || chunk.messageId != null || chunk.messageMetadata != null) write({ updateStatus: false });
 					break;
+				}
 				case "finish":
 					if (chunk.finishReason != null) state.finishReason = chunk.finishReason;
 					await updateMessageMetadata(chunk.messageMetadata);
@@ -7248,6 +7263,7 @@ function toUIMessageChunk(part, { tools, sendReasoning = true, sendSources = fal
 		case "text-end": return {
 			type: "text-end",
 			id: part.id,
+			...part.citations != null ? { citations: part.citations } : {},
 			...part.providerMetadata != null ? { providerMetadata: part.providerMetadata } : {}
 		};
 		case "reasoning-start":
@@ -7917,6 +7933,47 @@ function executeToolsFromStream({ stream, tools, callId, messages, abortSignal,
 	} }));
 }
 //#endregion
+//#region src/generate-text/is-output-chunk.ts
+const isOutputChunkType = {
+	file: true,
+	custom: false,
+	source: false,
+	"text-start": false,
+	"text-end": false,
+	"text-delta": true,
+	"reasoning-start": false,
+	"reasoning-end": false,
+	"reasoning-delta": true,
+	"reasoning-file": true,
+	"tool-input-start": false,
+	"tool-input-end": false,
+	"tool-input-delta": true,
+	"tool-approval-request": false,
+	"tool-approval-response": false,
+	"tool-call": true,
+	"tool-result": false,
+	"tool-error": false,
+	"tool-output-denied": false,
+	"tool-execution-end": false,
+	"model-call-start": false,
+	"model-call-response-metadata": false,
+	"model-call-end": false,
+	error: false,
+	raw: false
+};
+function isOutputChunk$1(chunk) {
+	if (!isOutputChunkType[chunk.type]) return false;
+	switch (chunk.type) {
+		case "text-delta":
+		case "reasoning-delta": return chunk.text.length > 0;
+		case "tool-input-delta": return chunk.delta.length > 0;
+		case "file":
+		case "reasoning-file":
+		case "tool-call": return true;
+		default: return false;
+	}
+}
+//#endregion
 //#region src/generate-text/invoke-tool-callbacks-from-stream.ts
 function invokeToolCallbacksFromStream({ stream, tools, stepInputMessages, abortSignal, toolsContext }) {
 	if (tools == null) return stream;
@@ -8216,7 +8273,7 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform({ too
 	let previousOutputChunkTimestampMs;
 	const timeBetweenOutputChunksMs = [];
 	return new TransformStream({ async transform(chunk, controller) {
-		if (isOutputChunk$1(chunk)) {
+		if (isOutputChunk(chunk)) {
 			const outputChunkTimestampMs = now();
 			if (timeToFirstOutputMs == null) timeToFirstOutputMs = outputChunkTimestampMs - callStartTimestampMs;
 			else if (previousOutputChunkTimestampMs != null) timeBetweenOutputChunksMs.push(outputChunkTimestampMs - previousOutputChunkTimestampMs);
@@ -8528,7 +8585,7 @@ function createLanguageModelV4StreamPartToLanguageModelStreamPartTransform({ too
 * Used to measure time-to-first-output for text, reasoning, generated files,
 * and tool calls.
 */
-function isOutputChunk$1(chunk) {
+function isOutputChunk(chunk) {
 	return chunk.type === "text-delta" && chunk.delta.length > 0 || chunk.type === "reasoning-delta" && chunk.delta.length > 0 || chunk.type === "tool-input-delta" && chunk.delta.length > 0 || chunk.type === "file" || chunk.type === "reasoning-file" || chunk.type === "tool-call";
 }
 function calculateOutputChunkTimingStats(timingsMs) {
@@ -8589,45 +8646,6 @@ const originalGenerateCallId$7 = createIdGenerator$1({
 	prefix: "call",
 	size: 24
 });
-const isOutputChunkType = {
-	file: true,
-	custom: false,
-	source: false,
-	"text-start": false,
-	"text-end": false,
-	"text-delta": true,
-	"reasoning-start": false,
-	"reasoning-end": false,
-	"reasoning-delta": true,
-	"reasoning-file": true,
-	"tool-input-start": false,
-	"tool-input-end": false,
-	"tool-input-delta": true,
-	"tool-approval-request": false,
-	"tool-approval-response": false,
-	"tool-call": true,
-	"tool-result": false,
-	"tool-error": false,
-	"tool-output-denied": false,
-	"tool-execution-end": false,
-	"model-call-start": false,
-	"model-call-response-metadata": false,
-	"model-call-end": false,
-	error: false,
-	raw: false
-};
-function isOutputChunk(chunk) {
-	if (!isOutputChunkType[chunk.type]) return false;
-	switch (chunk.type) {
-		case "text-delta":
-		case "reasoning-delta": return chunk.text.length > 0;
-		case "tool-input-delta": return chunk.delta.length > 0;
-		case "file":
-		case "reasoning-file":
-		case "tool-call": return true;
-		default: return false;
-	}
-}
 /**
 * Generate a text and call tools for a given prompt using a language model.
 *
@@ -9070,6 +9088,7 @@ var DefaultStreamTextResult = class {
 						return;
 					}
 					activeText.providerMetadata = part.providerMetadata ?? activeText.providerMetadata;
+					if (part.citations != null) activeText.citations = part.citations;
 					delete activeTextContent[part.id];
 				}
 				if (part.type === "reasoning-start") {
@@ -9731,7 +9750,7 @@ var DefaultStreamTextResult = class {
 									enqueueAttemptPart(value);
 									return;
 								}
-								if (isOutputChunk(value)) {
+								if (isOutputChunk$1(value)) {
 									clearFirstChunkTimeout();
 									resetChunkTimeout();
 								}
@@ -9889,7 +9908,7 @@ var DefaultStreamTextResult = class {
 								});
 							}
 							const chunkType = chunk.type;
-							if (isOutputChunk(chunk)) hasReceivedOutputChunk = true;
+							if (isOutputChunk$1(chunk)) hasReceivedOutputChunk = true;
 							switch (chunkType) {
 								case "file":
 								case "custom":
@@ -10654,6 +10673,7 @@ function readUIMessageStream({ message, stream, onError, terminateOnError = fals
 	consumeStream({
 		stream: processUIMessageStream({
 			stream,
+			resetStateOnMessageIdChange: true,
 			runUpdateMessageJob(job) {
 				return job({
 					state,
@@ -10980,6 +11000,7 @@ const uiMessagesSchema = lazySchema(() => {
 			z.object({
 				type: z.literal("text"),
 				text: z.string(),
+				citations: z.array(citationSchema).optional(),
 				state: z.enum(["streaming", "done"]).optional(),
 				providerMetadata: providerMetadataSchema.optional()
 			}),
@@ -12878,7 +12899,7 @@ function validateDecisionAnswers({ questions, answers, rounding }) {
 				if (answer.probabilities !== void 0) {
 					validateDistribution(answer.probabilities, Object.keys(question.criteria), answers, id, probabilityError);
 					const selected = answer.probabilities[answer.choice];
-					if (Object.values(answer.probabilities).some((probability) => probability > selected + tolerance)) invalidAnswer(answers, `Question "${id}" did not select a highest-probability option.`);
+					if (Object.values(answer.probabilities).some((probability) => probability > selected + tolerance + 2 * probabilityError)) invalidAnswer(answers, `Question "${id}" did not select a highest-probability option.`);
 				}
 				break;
 			case "score":
@@ -15677,7 +15698,7 @@ function simulateStreamingMiddleware() {
 					});
 					for (const part of result.content) switch (part.type) {
 						case "text":
-							if (part.text.length > 0) {
+							if (part.text.length > 0 || (part.citations?.length ?? 0) > 0) {
 								controller.enqueue({
 									type: "text-start",
 									id: String(id),
@@ -15690,7 +15711,8 @@ function simulateStreamingMiddleware() {
 								});
 								controller.enqueue({
 									type: "text-end",
-									id: String(id)
+									id: String(id),
+									...part.citations != null ? { citations: part.citations } : {}
 								});
 								id++;
 							}
@@ -19719,8 +19741,9 @@ var HttpChatTransport = class {
 //#endregion
 //#region src/ui/default-chat-transport.ts
 var DefaultChatTransport = class extends HttpChatTransport {
-	constructor(options = {}) {
+	constructor({ resumeStreamIsReplay = true, ...options } = {}) {
 		super(options);
+		this.resumeStreamIsReplay = resumeStreamIsReplay;
 	}
 	processResponseStream(stream) {
 		return parseJsonEventStream$1({
@@ -19831,6 +19854,14 @@ var AbstractChat = class {
 				...options
 			});
 		};
+		this["~resumeStreamIfDisconnected"] = async ({ shouldResume = () => true } = {}) => {
+			const activeResponse = this.activeResponse;
+			if (this.resumableStreamState == null && activeResponse != null) {
+				await activeResponse.completionPromise;
+				if (this.resumableStreamState !== activeResponse.state) return;
+			}
+			if (this.resumableStreamState != null && shouldResume()) await this.resumeStream();
+		};
 		this.clearError = () => {
 			if (this.status === "error") {
 				this.state.error = void 0;
@@ -19873,7 +19904,6 @@ var AbstractChat = class {
 		});
 		this.addToolOutput = async ({ state, toolCallId, output, errorText, options }) => this.jobExecutor.run(async () => {
 			const messages = this.state.messages;
-			const lastMessage = messages[messages.length - 1];
 			const updatePart = (part) => {
 				if (!isToolUIPart(part) || part.toolCallId !== toolCallId) return part;
 				const { approval: existingApproval, ...toolPart } = part;
@@ -19897,16 +19927,20 @@ var AbstractChat = class {
 					errorText: void 0
 				};
 			};
-			this.state.replaceMessage(messages.length - 1, {
-				...lastMessage,
-				parts: lastMessage.parts.map(updatePart)
-			});
+			const messageIndex = messages.findIndex((message) => message.parts.some((part) => isToolUIPart(part) && part.toolCallId === toolCallId));
+			if (messageIndex !== -1) {
+				const message = messages[messageIndex];
+				this.state.replaceMessage(messageIndex, {
+					...message,
+					parts: message.parts.map(updatePart)
+				});
+			}
 			if (this.activeResponse) this.activeResponse.state.message.parts = this.activeResponse.state.message.parts.map(updatePart);
-			if (this.status !== "streaming" && this.status !== "submitted" && this.sendAutomaticallyWhen) {
+			if (messageIndex !== -1 && this.status !== "streaming" && this.status !== "submitted" && this.sendAutomaticallyWhen) {
 				const shouldSend = await this.shouldSendAutomatically();
 				if (shouldSend) this.runAutomaticRequest(() => this.makeRequest({
 					trigger: "submit-message",
-					messageId: this.lastMessage?.id,
+					messageId: messages[messageIndex].id,
 					...options
 				}), shouldSend);
 			}
@@ -20073,6 +20107,7 @@ var AbstractChat = class {
 		const lastMessage = this.lastMessage;
 		const responseMessageIndex = trigger === "submit-message" && messageId != null ? this.state.messages.findIndex((message) => message.id === messageId) : this.state.messages.length - 1;
 		const responseMessage = responseMessageIndex === -1 ? lastMessage : this.state.messages[responseMessageIndex];
+		const originalResponseMessageId = responseMessage?.id;
 		const resumableResponseMessage = trigger === "resume-stream" && responseMessage?.role === "assistant" && responseMessage.parts.some((part) => isToolUIPart(part) && (part.state === "input-streaming" || part.state === "input-available" && part.providerExecuted || part.state === "approval-responded")) ? this.state.snapshot(responseMessage) : void 0;
 		const usesEarlierAssistantMessage = responseMessageIndex !== -1 && responseMessageIndex < this.state.messages.length - 1 && responseMessage?.role === "assistant";
 		let isAbort = false;
@@ -20112,9 +20147,13 @@ var AbstractChat = class {
 					write: ({ updateStatus = true } = {}) => {
 						if (response.abortController.signal.aborted) return;
 						if (updateStatus) this.setStatus({ status: "streaming" });
-						if (usesEarlierAssistantMessage) this.state.replaceMessage(responseMessageIndex, response.state.message);
-						else if (response.state.message.id === this.lastMessage?.id) this.state.replaceMessage(this.state.messages.length - 1, response.state.message);
-						else this.state.pushMessage(response.state.message);
+						const existingMessageIndex = this.state.messages.findLastIndex((message) => message.id === response.state.message.id);
+						if (existingMessageIndex !== -1) this.state.replaceMessage(existingMessageIndex, response.state.message);
+						else if (usesEarlierAssistantMessage) {
+							const originalMessageIndex = this.state.messages.findLastIndex((message) => message.id === originalResponseMessageId);
+							if (originalMessageIndex !== -1) this.state.replaceMessage(originalMessageIndex, response.state.message);
+							else this.state.pushMessage(response.state.message);
+						} else this.state.pushMessage(response.state.message);
 					}
 				});
 			});