@copilotkit/aimock 1.39.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") {
@@ -374,6 +376,7 @@ function collapseOpenAISSE(rawBody) {
 			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;
@@ -399,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 ?? "",
@@ -453,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 } : {},
@@ -469,6 +475,7 @@ function collapseOpenAISSE(rawBody) {
 		content,
 		...reasoning ? { reasoning } : {},
 		...webSearchQueries.length > 0 ? { webSearches: webSearchQueries } : {},
+		...usage ? { usage } : {},
 		...droppedChunks > 0 ? { droppedChunks } : {},
 		...firstDroppedSample ? { firstDroppedSample } : {},
 		...harmonyUnparsed ? { harmonyUnparsed: true } : {},
@@ -833,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;
@@ -843,6 +849,7 @@ function collapseCohereSSE(rawBody) {
 					arguments: ""
 				};
 				toolCallMap.set(index, created);
+				lastStartKey = index;
 				orderAtoms.push({
 					kind: "toolCall",
 					ref: created
@@ -1036,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 ?? "",
@@ -1047,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;
 		}
@@ -1054,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 ?? "",
@@ -1066,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) {
@@ -1252,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);