@copilotkit/aimock 1.35.0 → 1.36.0

package/dist/recorder.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"recorder.d.cts","names":[],"sources":["../src/recorder.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;UAiIiB,qBAAA;;;QAGT;;UAGS,YAAA;;;;;;;;;;;;;;;;;;mCAkBkB,oCAAoC;;;;;;;;;;;;;;;;;;;;;;KAuB3D,YAAA;;;;;;;;;;;;;;;iBA6MU,cAAA,MACf,MAAA,CAAK,sBACL,MAAA,CAAK,yBACD,oCACI,+CAEH;WAEC;UACD;2BACiB,0BAA0B;+BAG3C,eACT,QAAQ"}
+{"version":3,"file":"recorder.d.cts","names":[],"sources":["../src/recorder.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;UAkIiB,qBAAA;;;QAGT;;UAGS,YAAA;;;;;;;;;;;;;;;;;;mCAkBkB,oCAAoC;;;;;;;;;;;;;;;;;;;;;;KAuB3D,YAAA;;;;;;;;;;;;;;;iBA6MU,cAAA,MACf,MAAA,CAAK,sBACL,MAAA,CAAK,yBACD,oCACI,+CAEH;WAEC;UACD;2BACiB,0BAA0B;+BAG3C,eACT,QAAQ"}

package/dist/recorder.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"recorder.d.ts","names":[],"sources":["../src/recorder.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;UAiIiB,qBAAA;;;QAGT;;UAGS,YAAA;;;;;;;;;;;;;;;;;;mCAkBkB,oCAAoC;;;;;;;;;;;;;;;;;;;;;;KAuB3D,YAAA;;;;;;;;;;;;;;;iBA6MU,cAAA,MACf,MAAA,CAAK,sBACL,MAAA,CAAK,yBACD,oCACI,+CAEH;WAEC;UACD;2BACiB,0BAA0B;+BAG3C,eACT,QAAQ"}
+{"version":3,"file":"recorder.d.ts","names":[],"sources":["../src/recorder.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;UAkIiB,qBAAA;;;QAGT;;UAGS,YAAA;;;;;;;;;;;;;;;;;;mCAkBkB,oCAAoC;;;;;;;;;;;;;;;;;;;;;;KAuB3D,YAAA;;;;;;;;;;;;;;;iBA6MU,cAAA,MACf,MAAA,CAAK,sBACL,MAAA,CAAK,yBACD,oCACI,+CAEH;WAEC;UACD;2BACiB,0BAA0B;+BAG3C,eACT,QAAQ"}

package/dist/recorder.js

@@ -5,6 +5,7 @@ import { getLastMessageByRole, getTextContent } from "./router.js";
 import { writeErrorResponse } from "./sse-writer.js";
 import { capturedRedactedData, collapseStreamingResponse } from "./stream-collapse.js";
 import { resolveUpstreamUrl } from "./url.js";
+import { applyProviderAuth } from "./provider-auth.js";
 import * as http$1 from "node:http";
 import * as crypto$1 from "node:crypto";
 import * as https from "node:https";
@@ -201,7 +202,9 @@ function persistFixture(opts) {
 async function proxyAndRecord(req, res, request, providerKey, pathname, fixtures, defaults, rawBody, options) {
 	const record = defaults.record;
 	if (!record) return "not_configured";
-	const upstreamUrl = record.providers[providerKey === "gemini-interactions" ? "gemini" : providerKey];
+	const providers = record.providers;
+	const lookupKey = providerKey === "gemini-interactions" ? "gemini" : providerKey;
+	const upstreamUrl = providers[lookupKey];
 	if (!upstreamUrl) {
 		defaults.logger.warn(`No upstream URL configured for provider "${providerKey}" — cannot proxy`);
 		return "not_configured";
@@ -220,6 +223,7 @@ async function proxyAndRecord(req, res, request, providerKey, pathname, fixtures
 	}
 	defaults.logger.warn(`NO FIXTURE MATCH — proxying to ${upstreamUrl}${pathname}`);
 	const forwardHeaders = buildForwardHeaders(req);
+	applyProviderAuth(forwardHeaders, target, providerKey, record.providerKeys?.[lookupKey]);
 	const requestBody = rawBody ?? JSON.stringify(request);
 	let upstreamStatus;
 	let upstreamHeaders;
@@ -231,6 +235,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 +251,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 +406,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 +557,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 +573,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 +633,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 +689,8 @@ function makeUpstreamRequest(target, headers, body, clientRes, method = "POST",
 					bufferTruncated,
 					truncationCap,
 					totalBytes,
-					hardCeilingExceeded
+					hardCeilingExceeded,
+					sawDone
 				});
 			});
 		});