@copilotkit/aimock 1.46.0 → 1.47.0

package/dist/ws-realtime.js

@@ -6,7 +6,7 @@ import { calculateDelay, delay } from "./sse-writer.js";
 import { planMisbehavior, recordMisbehaviorOutcome } from "./misbehavior.js";
 import { matchFixtureDiagnostic } from "./router.js";
 import "./journal.js";
-import { createInterruptionSignal } from "./interruption.js";
+import { markInterrupted, markWsTurnStopped, watchWsClose, wsLifecycle } from "./response/lifecycle.js";
 import { render } from "./response/usage/codecs/realtime.js";
 import { randomBytes } from "node:crypto";
 
@@ -368,6 +368,7 @@ function rejectRetiredBetaShape(ws, logger) {
 function handleWebSocketRealtime(ws, fixtures, journal, defaults) {
 	const { logger } = defaults;
 	const sessionId = realtimeId("sess");
+	watchWsClose(ws);
 	const isBeta = defaults.upgradeHeaders?.["openai-beta"] ? String(defaults.upgradeHeaders["openai-beta"]).includes("realtime=v1") : false;
 	if (isBeta) {
 		rejectRetiredBetaShape(ws, logger);
@@ -676,18 +677,17 @@ async function emitLiveTranscriptionEvents(ws, fixtures, journal, defaults, sess
 	const latency = fixture.latency ?? defaults.latency;
 	const chunkSize = Math.max(1, fixture.chunkSize ?? defaults.chunkSize);
 	const replaySpeed = fixture.replaySpeed ?? defaults.replaySpeed;
-	const interruption = createInterruptionSignal(fixture);
+	const lifecycle = wsLifecycle(ws, fixture);
 	let chunkIndex = 0;
 	for (let index = 0; index < transcript.length; index += chunkSize) {
 		const chunkDelay = calculateDelay(chunkIndex, fixture.streamingProfile, latency, fixture.recordedTimings, replaySpeed);
-		if (chunkDelay > 0) await delay(chunkDelay, interruption?.signal);
-		if (interruption?.signal.aborted || ws.isClosed) {
-			if (interruption?.signal.aborted) {
-				journalEntry.response.interrupted = true;
-				journalEntry.response.interruptReason = interruption.reason();
+		if (chunkDelay > 0) await delay(chunkDelay, lifecycle.signal);
+		if (lifecycle.signal.aborted || ws.isClosed) {
+			if (lifecycle.signal.aborted) {
+				markInterrupted(journalEntry, lifecycle.reason());
 				ws.destroy();
-			}
-			interruption?.cleanup();
+			} else markWsTurnStopped(journalEntry, ws, lifecycle);
+			lifecycle.dispose();
 			return;
 		}
 		sendEvent(ws, {
@@ -696,14 +696,13 @@ async function emitLiveTranscriptionEvents(ws, fixtures, journal, defaults, sess
 			content_index: 0,
 			delta: transcript.slice(index, index + chunkSize)
 		}, isBeta);
-		interruption?.tick();
+		lifecycle.tick();
 		chunkIndex++;
 	}
-	if (interruption?.signal.aborted) {
-		journalEntry.response.interrupted = true;
-		journalEntry.response.interruptReason = interruption.reason();
+	if (lifecycle.signal.aborted || ws.isClosed) {
+		markWsTurnStopped(journalEntry, ws, lifecycle);
 		ws.destroy();
-		interruption.cleanup();
+		lifecycle.dispose();
 		return;
 	}
 	sendEvent(ws, {
@@ -714,7 +713,7 @@ async function emitLiveTranscriptionEvents(ws, fixtures, journal, defaults, sess
 		usage,
 		...response.transcription.languages !== void 0 ? { languages: response.transcription.languages } : {}
 	}, isBeta);
-	interruption?.cleanup();
+	lifecycle.dispose();
 }
 function sendLiveTranscriptionFailure(ws, itemId, error, isBeta) {
 	sendEvent(ws, {
@@ -976,7 +975,7 @@ async function handleResponseCreate(ws, fixtures, journal, defaults, session, co
 				usage: null
 			}
 		}, isBeta);
-		const interruption = createInterruptionSignal(fixture);
+		const lifecycle = wsLifecycle(ws, fixture);
 		let interrupted = false;
 		const allOutputItems = [];
 		const textItemId = realtimeId("item");
@@ -1024,8 +1023,8 @@ async function handleResponseCreate(ws, fixtures, journal, defaults, session, co
 		for (let i = 0; i < content.length; i += chunkSize) {
 			if (ws.isClosed) break;
 			const chunkDelay = calculateDelay(eventIndex, void 0, latency, recordedTimings, replaySpeed);
-			if (chunkDelay > 0) await delay(chunkDelay, interruption?.signal);
-			if (interruption?.signal.aborted) {
+			if (chunkDelay > 0) await delay(chunkDelay, lifecycle.signal);
+			if (lifecycle.signal.aborted) {
 				interrupted = true;
 				break;
 			}
@@ -1045,21 +1044,21 @@ async function handleResponseCreate(ws, fixtures, journal, defaults, session, co
 				break;
 			}
 			eventIndex++;
-			interruption?.tick();
-			if (interruption?.signal.aborted) {
+			lifecycle.tick();
+			if (lifecycle.signal.aborted) {
 				interrupted = true;
 				break;
 			}
 		}
 		if (interrupted) {
 			ws.destroy();
-			journalEntry.response.interrupted = true;
-			journalEntry.response.interruptReason = interruption?.reason();
-			interruption?.cleanup();
+			markInterrupted(journalEntry, lifecycle.reason());
+			lifecycle.dispose();
 			return;
 		}
 		if (ws.isClosed) {
-			interruption?.cleanup();
+			markWsTurnStopped(journalEntry, ws, lifecycle);
+			lifecycle.dispose();
 			return;
 		}
 		sendEvent(ws, {
@@ -1071,7 +1070,8 @@ async function handleResponseCreate(ws, fixtures, journal, defaults, session, co
 			text: content
 		}, isBeta);
 		if (ws.isClosed) {
-			interruption?.cleanup();
+			markWsTurnStopped(journalEntry, ws, lifecycle);
+			lifecycle.dispose();
 			return;
 		}
 		sendEvent(ws, {
@@ -1086,7 +1086,8 @@ async function handleResponseCreate(ws, fixtures, journal, defaults, session, co
 			}
 		}, isBeta);
 		if (ws.isClosed) {
-			interruption?.cleanup();
+			markWsTurnStopped(journalEntry, ws, lifecycle);
+			lifecycle.dispose();
 			return;
 		}
 		sendEvent(ws, {
@@ -1110,7 +1111,8 @@ async function handleResponseCreate(ws, fixtures, journal, defaults, session, co
 			}
 		}, isBeta);
 		if (ws.isClosed) {
-			interruption?.cleanup();
+			markWsTurnStopped(journalEntry, ws, lifecycle);
+			lifecycle.dispose();
 			return;
 		}
 		allOutputItems.push(textOutputItem);
@@ -1146,8 +1148,8 @@ async function handleResponseCreate(ws, fixtures, journal, defaults, session, co
 			for (let i = 0; i < args.length; i += chunkSize) {
 				if (ws.isClosed) break;
 				const chunkDelay = calculateDelay(eventIndex, void 0, latency, recordedTimings, replaySpeed);
-				if (chunkDelay > 0) await delay(chunkDelay, interruption?.signal);
-				if (interruption?.signal.aborted) {
+				if (chunkDelay > 0) await delay(chunkDelay, lifecycle.signal);
+				if (lifecycle.signal.aborted) {
 					interrupted = true;
 					break;
 				}
@@ -1167,8 +1169,8 @@ async function handleResponseCreate(ws, fixtures, journal, defaults, session, co
 					break;
 				}
 				eventIndex++;
-				interruption?.tick();
-				if (interruption?.signal.aborted) {
+				lifecycle.tick();
+				if (lifecycle.signal.aborted) {
 					interrupted = true;
 					break;
 				}
@@ -1210,13 +1212,15 @@ async function handleResponseCreate(ws, fixtures, journal, defaults, session, co
 		}
 		if (interrupted) {
 			ws.destroy();
-			journalEntry.response.interrupted = true;
-			journalEntry.response.interruptReason = interruption?.reason();
-			interruption?.cleanup();
+			markInterrupted(journalEntry, lifecycle.reason());
+			lifecycle.dispose();
+			return;
+		}
+		lifecycle.dispose();
+		if (ws.isClosed) {
+			markWsTurnStopped(journalEntry, ws, lifecycle);
 			return;
 		}
-		interruption?.cleanup();
-		if (ws.isClosed) return;
 		sendEvent(ws, {
 			type: "response.done",
 			response: {
@@ -1292,14 +1296,14 @@ async function handleResponseCreate(ws, fixtures, journal, defaults, session, co
 		const content = response.content ?? "";
 		const replaySpeed = fixture.replaySpeed ?? defaults.replaySpeed;
 		const { recordedTimings } = fixture;
-		const interruption = createInterruptionSignal(fixture);
+		const lifecycle = wsLifecycle(ws, fixture);
 		let interrupted = false;
 		let eventIndex = 0;
 		for (let i = 0; i < content.length; i += chunkSize) {
 			if (ws.isClosed) break;
 			const chunkDelay = calculateDelay(eventIndex, void 0, latency, recordedTimings, replaySpeed);
-			if (chunkDelay > 0) await delay(chunkDelay, interruption?.signal);
-			if (interruption?.signal.aborted) {
+			if (chunkDelay > 0) await delay(chunkDelay, lifecycle.signal);
+			if (lifecycle.signal.aborted) {
 				interrupted = true;
 				break;
 			}
@@ -1319,21 +1323,23 @@ async function handleResponseCreate(ws, fixtures, journal, defaults, session, co
 				break;
 			}
 			eventIndex++;
-			interruption?.tick();
-			if (interruption?.signal.aborted) {
+			lifecycle.tick();
+			if (lifecycle.signal.aborted) {
 				interrupted = true;
 				break;
 			}
 		}
 		if (interrupted) {
 			ws.destroy();
-			journalEntry.response.interrupted = true;
-			journalEntry.response.interruptReason = interruption?.reason();
-			interruption?.cleanup();
+			markInterrupted(journalEntry, lifecycle.reason());
+			lifecycle.dispose();
+			return;
+		}
+		lifecycle.dispose();
+		if (ws.isClosed) {
+			markWsTurnStopped(journalEntry, ws, lifecycle);
 			return;
 		}
-		interruption?.cleanup();
-		if (ws.isClosed) return;
 		sendEvent(ws, {
 			type: "response.output_text.done",
 			response_id: responseId,
@@ -1410,7 +1416,7 @@ async function handleResponseCreate(ws, fixtures, journal, defaults, session, co
 			}
 		}, isBeta);
 		const outputItems = [];
-		const interruption = createInterruptionSignal(fixture);
+		const lifecycle = wsLifecycle(ws, fixture);
 		const replaySpeed = fixture.replaySpeed ?? defaults.replaySpeed;
 		const { recordedTimings } = fixture;
 		let interrupted = false;
@@ -1445,8 +1451,8 @@ async function handleResponseCreate(ws, fixtures, journal, defaults, session, co
 			for (let i = 0; i < args.length; i += chunkSize) {
 				if (ws.isClosed) break;
 				const chunkDelay = calculateDelay(eventIndex, void 0, latency, recordedTimings, replaySpeed);
-				if (chunkDelay > 0) await delay(chunkDelay, interruption?.signal);
-				if (interruption?.signal.aborted) {
+				if (chunkDelay > 0) await delay(chunkDelay, lifecycle.signal);
+				if (lifecycle.signal.aborted) {
 					interrupted = true;
 					break;
 				}
@@ -1466,8 +1472,8 @@ async function handleResponseCreate(ws, fixtures, journal, defaults, session, co
 					break;
 				}
 				eventIndex++;
-				interruption?.tick();
-				if (interruption?.signal.aborted) {
+				lifecycle.tick();
+				if (lifecycle.signal.aborted) {
 					interrupted = true;
 					break;
 				}
@@ -1507,13 +1513,15 @@ async function handleResponseCreate(ws, fixtures, journal, defaults, session, co
 		}
 		if (interrupted) {
 			ws.destroy();
-			journalEntry.response.interrupted = true;
-			journalEntry.response.interruptReason = interruption?.reason();
-			interruption?.cleanup();
+			markInterrupted(journalEntry, lifecycle.reason());
+			lifecycle.dispose();
+			return;
+		}
+		lifecycle.dispose();
+		if (ws.isClosed) {
+			markWsTurnStopped(journalEntry, ws, lifecycle);
 			return;
 		}
-		interruption?.cleanup();
-		if (ws.isClosed) return;
 		sendEvent(ws, {
 			type: "response.done",
 			response: {
@@ -1551,7 +1559,7 @@ async function streamRealtimeBlocks(ws, blocks, responseId, fixture, defaults, l
 			usage: null
 		}
 	}, isBeta);
-	const interruption = createInterruptionSignal(fixture);
+	const lifecycle = wsLifecycle(ws, fixture);
 	let interrupted = false;
 	const allOutputItems = [];
 	const replaySpeed = fixture.replaySpeed ?? defaults.replaySpeed;
@@ -1603,8 +1611,8 @@ async function streamRealtimeBlocks(ws, blocks, responseId, fixture, defaults, l
 			for (let i = 0; i < text.length; i += chunkSize) {
 				if (ws.isClosed) break;
 				const chunkDelay = calculateDelay(eventIndex, void 0, latency, recordedTimings, replaySpeed);
-				if (chunkDelay > 0) await delay(chunkDelay, interruption?.signal);
-				if (interruption?.signal.aborted) {
+				if (chunkDelay > 0) await delay(chunkDelay, lifecycle.signal);
+				if (lifecycle.signal.aborted) {
 					interrupted = true;
 					break;
 				}
@@ -1624,8 +1632,8 @@ async function streamRealtimeBlocks(ws, blocks, responseId, fixture, defaults, l
 					break;
 				}
 				eventIndex++;
-				interruption?.tick();
-				if (interruption?.signal.aborted) {
+				lifecycle.tick();
+				if (lifecycle.signal.aborted) {
 					interrupted = true;
 					break;
 				}
@@ -1704,8 +1712,8 @@ async function streamRealtimeBlocks(ws, blocks, responseId, fixture, defaults, l
 			for (let i = 0; i < args.length; i += chunkSize) {
 				if (ws.isClosed) break;
 				const chunkDelay = calculateDelay(eventIndex, void 0, latency, recordedTimings, replaySpeed);
-				if (chunkDelay > 0) await delay(chunkDelay, interruption?.signal);
-				if (interruption?.signal.aborted) {
+				if (chunkDelay > 0) await delay(chunkDelay, lifecycle.signal);
+				if (lifecycle.signal.aborted) {
 					interrupted = true;
 					break;
 				}
@@ -1725,8 +1733,8 @@ async function streamRealtimeBlocks(ws, blocks, responseId, fixture, defaults, l
 					break;
 				}
 				eventIndex++;
-				interruption?.tick();
-				if (interruption?.signal.aborted) {
+				lifecycle.tick();
+				if (lifecycle.signal.aborted) {
 					interrupted = true;
 					break;
 				}
@@ -1768,13 +1776,15 @@ async function streamRealtimeBlocks(ws, blocks, responseId, fixture, defaults, l
 	}
 	if (interrupted) {
 		ws.destroy();
-		journalEntry.response.interrupted = true;
-		journalEntry.response.interruptReason = interruption?.reason();
-		interruption?.cleanup();
+		markInterrupted(journalEntry, lifecycle.reason());
+		lifecycle.dispose();
+		return;
+	}
+	lifecycle.dispose();
+	if (ws.isClosed) {
+		markWsTurnStopped(journalEntry, ws, lifecycle);
 		return;
 	}
-	interruption?.cleanup();
-	if (ws.isClosed) return;
 	sendEvent(ws, {
 		type: "response.done",
 		response: {