@copilotkit/aimock 1.34.0 → 1.35.1

package/dist/recorder.js

@@ -231,6 +231,7 @@ async function proxyAndRecord(req, res, request, providerKey, pathname, fixtures
 	let hardCeilingExceeded = false;
 	let streamedToClient = false;
 	let clientDisconnected = false;
+	let sawDone = false;
 	let frameTimestamps = [];
 	let streamStartTime = 0;
 	const maxProxyBufferBytes = clampMaxBufferBytes(record.maxProxyBufferBytes);
@@ -246,6 +247,7 @@ async function proxyAndRecord(req, res, request, providerKey, pathname, fixtures
 		rawBuffer = result.rawBuffer;
 		streamedToClient = result.streamedToClient;
 		clientDisconnected = result.clientDisconnected;
+		sawDone = result.sawDone;
 		frameTimestamps = result.frameTimestamps;
 		streamStartTime = result.streamStartTime;
 		bufferTruncated = result.bufferTruncated;
@@ -347,15 +349,18 @@ async function proxyAndRecord(req, res, request, providerKey, pathname, fixtures
 					name: tc.name ?? "",
 					arguments: tc.arguments ?? "{}"
 				}));
-				if (collapsed.content) fixtureResponse = {
-					content: collapsed.content,
-					toolCalls: sanitizedToolCalls,
-					...reasoningSpread,
-					...reasoningSignatureSpread,
-					...redactedThinkingSpread,
-					...webSearchesSpread
-				};
-				else fixtureResponse = {
+				if (collapsed.content) {
+					const blocksSpread = collapsed.blocks?.length ? { blocks: collapsed.blocks } : {};
+					fixtureResponse = {
+						content: collapsed.content,
+						toolCalls: sanitizedToolCalls,
+						...blocksSpread,
+						...reasoningSpread,
+						...reasoningSignatureSpread,
+						...redactedThinkingSpread,
+						...webSearchesSpread
+					};
+				} else fixtureResponse = {
 					toolCalls: sanitizedToolCalls,
 					...reasoningSpread,
 					...reasoningSignatureSpread,
@@ -397,8 +402,11 @@ async function proxyAndRecord(req, res, request, providerKey, pathname, fixtures
 		} else fixtureResponse = buildFixtureResponse(parsedResponse, upstreamStatus, defaults.logger);
 	}
 	if (clientDisconnected) {
-		defaults.logger.warn("Client disconnected mid-stream — skipping fixture save to avoid truncated data");
-		return "relayed";
+		if (!sawDone) {
+			defaults.logger.warn("Client disconnected mid-stream — skipping fixture save to avoid truncated data");
+			return "relayed";
+		}
+		defaults.logger.warn("Client closed connection before upstream end — upstream response completed, recording full fixture");
 	}
 	let recordedTimings;
 	if (frameTimestamps.length > 1) {
@@ -545,6 +553,7 @@ function makeUpstreamRequest(target, headers, body, clientRes, method = "POST",
 			let binaryFrameBuffer = Buffer.alloc(0);
 			let streamedToClient = false;
 			let clientDisconnected = false;
+			let sawDone = false;
 			if (isProgressiveStream && clientRes && !clientRes.headersSent) {
 				const relayHeaders = {
 					"Cache-Control": "no-cache, no-transform",
@@ -560,13 +569,15 @@ function makeUpstreamRequest(target, headers, body, clientRes, method = "POST",
 				clientRes.on("close", () => {
 					if (!clientRes.writableFinished) {
 						clientDisconnected = true;
-						req.destroy();
-						res.removeListener("data", onUpstreamData);
-						chunks.length = 0;
-						bufferedBytes = 0;
-						frameTimestamps.length = 0;
-						frameBuffer = "";
-						binaryFrameBuffer = Buffer.alloc(0);
+						if (!sawDone) {
+							req.destroy();
+							res.removeListener("data", onUpstreamData);
+							chunks.length = 0;
+							bufferedBytes = 0;
+							frameTimestamps.length = 0;
+							frameBuffer = "";
+							binaryFrameBuffer = Buffer.alloc(0);
+						}
 					}
 				});
 			}
@@ -618,7 +629,11 @@ function makeUpstreamRequest(target, headers, body, clientRes, method = "POST",
 							tripTruncation("frame");
 							break;
 						}
-						if (parts[fi].trim().length > 0) frameTimestamps.push(Date.now());
+						const frame = parts[fi].trim();
+						if (frame.length > 0) {
+							frameTimestamps.push(Date.now());
+							if (frame === "data: [DONE]") sawDone = true;
+						}
 					}
 					if (!bufferTruncated) frameBuffer = parts[parts.length - 1];
 				}
@@ -670,7 +685,8 @@ function makeUpstreamRequest(target, headers, body, clientRes, method = "POST",
 					bufferTruncated,
 					truncationCap,
 					totalBytes,
-					hardCeilingExceeded
+					hardCeilingExceeded,
+					sawDone
 				});
 			});
 		});