@copilotkit/aimock 1.38.0 → 1.40.0

package/dist/stream-collapse.cjs

@@ -245,6 +245,7 @@ function collapseOpenAISSE(rawBody) {
 	let transcriptUsage;
 	let reasoning = "";
 	const webSearchQueries = [];
+	let usage;
 	let droppedChunks = 0;
 	let firstDroppedSample;
 	let harmonyUnparsed = false;
@@ -252,6 +253,7 @@ function collapseOpenAISSE(rawBody) {
 	const toolCallMap = /* @__PURE__ */ new Map();
 	let nextSyntheticIndex = 1e6;
 	const idKeyMap = /* @__PURE__ */ new Map();
+	let lastToolCallKey;
 	const orderAtoms = [];
 	for (const line of lines) {
 		const data = extractSSEData(splitSSELines(line));
@@ -275,7 +277,7 @@ function collapseOpenAISSE(rawBody) {
 			transcript = parsed.text;
 			transcriptSeen = true;
 			if (Array.isArray(parsed.languages)) transcriptLanguages = parsed.languages.filter((language) => typeof language === "object" && language !== null && typeof language.code === "string").map((language) => ({ code: language.code }));
-			if (parsed.usage && typeof parsed.usage === "object") transcriptUsage = parsed.usage;
+			if (parsed.usage && typeof parsed.usage === "object" && !Array.isArray(parsed.usage) && Object.keys(parsed.usage).length > 0) transcriptUsage = parsed.usage;
 			continue;
 		}
 		if (parsed.type === "response.reasoning_summary_text.delta" && typeof parsed.delta === "string") {
@@ -291,12 +293,90 @@ function collapseOpenAISSE(rawBody) {
 					continue;
 				}
 			}
+			if (item?.type === "function_call" && typeof parsed.output_index === "number") {
+				const entry = toolCallMap.get(parsed.output_index);
+				if (entry) {
+					if (!entry.id && typeof item.call_id === "string") entry.id = item.call_id;
+					if (!entry.name && typeof item.name === "string") entry.name = item.name;
+					if (entry.arguments === "" && typeof item.arguments === "string") entry.arguments = item.arguments;
+				} else {
+					const created = {
+						id: typeof item.call_id === "string" ? item.call_id : "",
+						name: typeof item.name === "string" ? item.name : "",
+						arguments: typeof item.arguments === "string" ? item.arguments : ""
+					};
+					toolCallMap.set(parsed.output_index, created);
+					orderAtoms.push({
+						kind: "toolCall",
+						ref: created
+					});
+				}
+				continue;
+			}
+		}
+		if (parsed.type === "response.output_item.added") {
+			const item = parsed.item;
+			if (item?.type === "function_call" && typeof parsed.output_index === "number") {
+				if (!toolCallMap.has(parsed.output_index)) {
+					const created = {
+						id: typeof item.call_id === "string" ? item.call_id : "",
+						name: typeof item.name === "string" ? item.name : "",
+						arguments: ""
+					};
+					toolCallMap.set(parsed.output_index, created);
+					orderAtoms.push({
+						kind: "toolCall",
+						ref: created
+					});
+				}
+				continue;
+			}
+		}
+		if (parsed.type === "response.function_call_arguments.delta" && typeof parsed.delta === "string" && typeof parsed.output_index === "number") {
+			let entry = toolCallMap.get(parsed.output_index);
+			if (!entry) {
+				entry = {
+					id: "",
+					name: "",
+					arguments: ""
+				};
+				toolCallMap.set(parsed.output_index, entry);
+				orderAtoms.push({
+					kind: "toolCall",
+					ref: entry
+				});
+			}
+			entry.arguments += parsed.delta;
+			continue;
+		}
+		if (parsed.type === "response.function_call_arguments.done" && typeof parsed.output_index === "number") {
+			const entry = toolCallMap.get(parsed.output_index);
+			if (entry) {
+				if (entry.arguments === "" && typeof parsed.arguments === "string") entry.arguments = parsed.arguments;
+			} else if (typeof parsed.arguments === "string") {
+				const created = {
+					id: "",
+					name: "",
+					arguments: parsed.arguments
+				};
+				toolCallMap.set(parsed.output_index, created);
+				orderAtoms.push({
+					kind: "toolCall",
+					ref: created
+				});
+			}
+			continue;
 		}
 		if (parsed.type === "response.output_text.delta" && typeof parsed.delta === "string") {
 			content += parsed.delta;
+			if (parsed.delta.length > 0) orderAtoms.push({
+				kind: "text",
+				text: parsed.delta
+			});
 			continue;
 		}
 		if (typeof parsed.type === "string" && parsed.type.startsWith("response.")) continue;
+		if (parsed.usage && typeof parsed.usage === "object" && !Array.isArray(parsed.usage) && Object.keys(parsed.usage).length > 0) usage = parsed.usage;
 		const choices = parsed.choices;
 		if (!choices || choices.length === 0) continue;
 		const delta = choices[0].delta;
@@ -322,7 +402,9 @@ function collapseOpenAISSE(rawBody) {
 					index = nextSyntheticIndex++;
 					idKeyMap.set(rawId, index);
 				}
-			} else index = nextSyntheticIndex++;
+			} else if (lastToolCallKey !== void 0) index = lastToolCallKey;
+			else index = nextSyntheticIndex++;
+			lastToolCallKey = index;
 			if (!toolCallMap.has(index)) {
 				const created = {
 					id: rawId ?? "",
@@ -376,6 +458,7 @@ function collapseOpenAISSE(rawBody) {
 			toolCalls: [...blocks ? orderedToolCalls : indexSortedToolCalls, ...harmonyToolCalls],
 			...reasoning ? { reasoning } : {},
 			...webSearchQueries.length > 0 ? { webSearches: webSearchQueries } : {},
+			...usage ? { usage } : {},
 			...droppedChunks > 0 ? { droppedChunks } : {},
 			...firstDroppedSample ? { firstDroppedSample } : {},
 			...harmonyUnparsed ? { harmonyUnparsed: true } : {},
@@ -392,6 +475,7 @@ function collapseOpenAISSE(rawBody) {
 		content,
 		...reasoning ? { reasoning } : {},
 		...webSearchQueries.length > 0 ? { webSearches: webSearchQueries } : {},
+		...usage ? { usage } : {},
 		...droppedChunks > 0 ? { droppedChunks } : {},
 		...firstDroppedSample ? { firstDroppedSample } : {},
 		...harmonyUnparsed ? { harmonyUnparsed: true } : {},
@@ -756,7 +840,6 @@ function collapseCohereSSE(rawBody) {
 			let index;
 			if (typeof parsed.index === "number") index = parsed.index;
 			else index = nextSyntheticIndex++;
-			lastStartKey = index;
 			const toolCalls = (parsed.delta?.message)?.tool_calls;
 			if (toolCalls) {
 				const fn = toolCalls.function;
@@ -766,6 +849,7 @@ function collapseCohereSSE(rawBody) {
 					arguments: ""
 				};
 				toolCallMap.set(index, created);
+				lastStartKey = index;
 				orderAtoms.push({
 					kind: "toolCall",
 					ref: created
@@ -959,7 +1043,7 @@ function collapseBedrockEventStream(rawBody) {
 			const index = parsed.index;
 			const redactedData = capturedRedactedData(block);
 			if (redactedData !== void 0) redactedThinking.push(redactedData);
-			if (block?.type === "tool_use" && index !== void 0) {
+			if (block?.type === "tool_use") if (index !== void 0) {
 				const created = {
 					id: block.id ?? "",
 					name: block.name ?? "",
@@ -970,6 +1054,9 @@ function collapseBedrockEventStream(rawBody) {
 					kind: "toolCall",
 					ref: created
 				});
+			} else {
+				droppedChunks++;
+				if (droppedChunks === 1) firstDroppedSample = `tool_use content_block_start with no index — tool call identity lost: ${surrogateSafeSlice(frameStr, 200)}`;
 			}
 			continue;
 		}
@@ -977,7 +1064,7 @@ function collapseBedrockEventStream(rawBody) {
 			const blockStart = parsed.contentBlockStart;
 			const index = parsed.contentBlockIndex ?? blockStart.contentBlockIndex;
 			const start = blockStart.start;
-			if (start?.toolUse && index !== void 0) {
+			if (start?.toolUse) if (index !== void 0) {
 				const toolUse = start.toolUse;
 				const created = {
 					id: toolUse.toolUseId ?? "",
@@ -989,6 +1076,9 @@ function collapseBedrockEventStream(rawBody) {
 					kind: "toolCall",
 					ref: created
 				});
+			} else {
+				droppedChunks++;
+				if (droppedChunks === 1) firstDroppedSample = `contentBlockStart toolUse with no index — tool call identity lost: ${surrogateSafeSlice(frameStr, 200)}`;
 			}
 		}
 		if (parsed.contentBlockDelta) {
@@ -1175,7 +1265,8 @@ function collapseStreamingResponse(contentType, providerKey, body, logger) {
 		const str = typeof body === "string" ? body : body.toString("utf8");
 		switch (providerKey) {
 			case "openai":
-			case "azure": return collapseOpenAISSE(str);
+			case "azure":
+			case "openrouter": return collapseOpenAISSE(str);
 			case "anthropic": return collapseAnthropicSSE(str);
 			case "gemini":
 			case "vertexai": return collapseGeminiSSE(str);