@copilotkit/aimock 1.35.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;
@@ -400,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) {
@@ -548,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",
@@ -563,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);
+						}
 					}
 				});
 			}
@@ -621,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];
 				}
@@ -673,7 +685,8 @@ function makeUpstreamRequest(target, headers, body, clientRes, method = "POST",
 					bufferTruncated,
 					truncationCap,
 					totalBytes,
-					hardCeilingExceeded
+					hardCeilingExceeded,
+					sawDone
 				});
 			});
 		});