@copilotkit/aimock 1.45.1 → 1.46.0

package/dist/ws-realtime.js

@@ -1,10 +1,13 @@
+import { buildOpenAIError } from "./response/errors.js";
 import { DEFAULT_TEST_ID } from "./constants.js";
-import { fixtureToolCallErrorCode, flattenHeaders, generateToolCallId, isCombinedFixtureResponse, isErrorResponse, isFixtureToolCallError, isTextResponse, isToolCallResponse, isTranscriptionResponse, journalFixtureToolCallError, prepareOpenAIChatMisbehavior, rejectResponsesOnlyToolCalls, requireServedFunctionToolCalls, resolveFixtureBlocks, resolveOpenAIChatMisbehaviorUsage, resolveResponse, resolveStrictMode, strictNoMatchLogLine, strictNoMatchMessage, strictOverrideField } from "./helpers.js";
+import { resolveRequestDecisions, strictOverrideField } from "./response/policy.js";
+import { fixtureToolCallErrorCode, flattenHeaders, generateToolCallId, isCombinedFixtureResponse, isErrorResponse, isFixtureToolCallError, isTextResponse, isToolCallResponse, isTranscriptionResponse, journalFixtureToolCallError, prepareOpenAIChatMisbehavior, rejectResponsesOnlyToolCalls, requireServedFunctionToolCalls, resolveFixtureBlocks, resolveOpenAIChatMisbehaviorUsage, resolveResponse, strictNoMatchLogLine, strictNoMatchMessage } from "./helpers.js";
 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 { render } from "./response/usage/codecs/realtime.js";
 import { randomBytes } from "node:crypto";
 
 //#region src/ws-realtime.ts
@@ -315,11 +318,11 @@ function sendEvent(ws, event, isBeta) {
 function buildErrorRealtimeEvent(ws, message, isBeta, type = "invalid_request_error", code) {
 	sendEvent(ws, {
 		type: "error",
-		error: {
+		error: buildOpenAIError({
 			message,
 			type,
 			code
-		}
+		})
 	}, isBeta);
 }
 /**
@@ -435,11 +438,11 @@ async function processMessage(raw, ws, fixtures, journal, defaults, session, con
 				if (!validTypes.has(s.type)) {
 					sendEvent(ws, {
 						type: "error",
-						error: {
+						error: buildOpenAIError({
 							message: `Invalid session type: ${s.type}`,
 							type: "invalid_request_error",
 							code: "invalid_session_config"
-						}
+						})
 					}, isBeta);
 					return;
 				}
@@ -494,11 +497,11 @@ async function processMessage(raw, ws, fixtures, journal, defaults, session, con
 				Object.assign(session, prevSession);
 				sendEvent(ws, {
 					type: "error",
-					error: {
+					error: buildOpenAIError({
 						message: `Model ${candidateTranscriptionModel} does not support session type transcription`,
 						type: "invalid_request_error",
 						code: "invalid_session_config"
-					}
+					})
 				}, isBeta);
 				return;
 			}
@@ -506,11 +509,11 @@ async function processMessage(raw, ws, fixtures, journal, defaults, session, con
 				Object.assign(session, prevSession);
 				sendEvent(ws, {
 					type: "error",
-					error: {
+					error: buildOpenAIError({
 						message: `Model ${s.model ?? prevSession.model} does not support session type translation`,
 						type: "invalid_request_error",
 						code: "invalid_session_config"
-					}
+					})
 				}, isBeta);
 				return;
 			}
@@ -585,7 +588,7 @@ async function emitLiveTranscriptionEvents(ws, fixtures, journal, defaults, sess
 	const { fixture, skippedBySequenceOrTurn } = matchFixtureDiagnostic(fixtures, request, journal.getFixtureMatchCountsForTest(testId), defaults.requestTransform);
 	const itemId = audioItem.id ?? realtimeId("item");
 	if (!fixture) {
-		if (resolveStrictMode(defaults.strict, defaults.upgradeHeaders)) {
+		if (resolveRequestDecisions(defaults.strict, defaults.upgradeHeaders).fixtureMiss) {
 			const strictMessage = strictNoMatchMessage(skippedBySequenceOrTurn);
 			defaults.logger.error(strictNoMatchLogLine("WS", "/v1/realtime", skippedBySequenceOrTurn));
 			journal.add({
@@ -762,11 +765,7 @@ function checkRealtimeFixtureToolCalls(ws, journalEntry, responseId, isBeta, log
 						code: fixtureToolCallErrorCode(err)
 					}
 				},
-				usage: {
-					total_tokens: 0,
-					input_tokens: 0,
-					output_tokens: 0
-				}
+				usage: render({ kind: "ordinary-zero" })
 			}
 		}, isBeta);
 		return;
@@ -794,7 +793,7 @@ async function handleResponseCreate(ws, fixtures, journal, defaults, session, co
 	const responseId = realtimeId("resp");
 	if (fixture) journal.incrementFixtureMatchCount(fixture, fixtures, testId);
 	if (!fixture) {
-		if (resolveStrictMode(defaults.strict, defaults.upgradeHeaders)) {
+		if (resolveRequestDecisions(defaults.strict, defaults.upgradeHeaders).fixtureMiss) {
 			const strictMessage = strictNoMatchMessage(skippedBySequenceOrTurn);
 			defaults.logger.error(strictNoMatchLogLine("WS", "/v1/realtime", skippedBySequenceOrTurn));
 			journal.add({
@@ -848,11 +847,7 @@ async function handleResponseCreate(ws, fixtures, journal, defaults, session, co
 						code: "no_fixture_match"
 					}
 				},
-				usage: {
-					total_tokens: 0,
-					input_tokens: 0,
-					output_tokens: 0
-				}
+				usage: render({ kind: "ordinary-zero" })
 			}
 		}, isBeta);
 		return;
@@ -916,11 +911,10 @@ async function handleResponseCreate(ws, fixtures, journal, defaults, session, co
 		const usage = resolveOpenAIChatMisbehaviorUsage(misbehavior, completionReq, false);
 		await streamRealtimeBlocks(ws, blocks, responseId, fixture, defaults, latency, chunkSize, isBeta, conversationItems, journalEntry, {
 			stop: misbehavior.stop,
-			usage: {
-				input_tokens: usage.prompt_tokens,
-				output_tokens: usage.completion_tokens,
-				total_tokens: usage.total_tokens
-			}
+			usage: render({
+				kind: "applied-misbehavior",
+				usage
+			})
 		});
 		return;
 	}
@@ -952,11 +946,7 @@ async function handleResponseCreate(ws, fixtures, journal, defaults, session, co
 						code: response.error.code
 					}
 				},
-				usage: {
-					total_tokens: 0,
-					input_tokens: 0,
-					output_tokens: 0
-				}
+				usage: render({ kind: "ordinary-zero" })
 			}
 		}, isBeta);
 		return;
@@ -1234,11 +1224,7 @@ async function handleResponseCreate(ws, fixtures, journal, defaults, session, co
 				object: "realtime.response",
 				status: "completed",
 				output: allOutputItems,
-				usage: {
-					total_tokens: 0,
-					input_tokens: 0,
-					output_tokens: 0
-				}
+				usage: render({ kind: "ordinary-zero" })
 			}
 		}, isBeta);
 		conversationItems.push({
@@ -1394,11 +1380,7 @@ async function handleResponseCreate(ws, fixtures, journal, defaults, session, co
 				object: "realtime.response",
 				status: "completed",
 				output: [outputItem],
-				usage: {
-					total_tokens: 0,
-					input_tokens: 0,
-					output_tokens: 0
-				}
+				usage: render({ kind: "ordinary-zero" })
 			}
 		}, isBeta);
 		conversationItems.push({
@@ -1539,11 +1521,7 @@ async function handleResponseCreate(ws, fixtures, journal, defaults, session, co
 				object: "realtime.response",
 				status: "completed",
 				output: outputItems,
-				usage: {
-					total_tokens: 0,
-					input_tokens: 0,
-					output_tokens: 0
-				}
+				usage: render({ kind: "ordinary-zero" })
 			}
 		}, isBeta);
 		for (const item of outputItems) conversationItems.push(item);
@@ -1808,11 +1786,7 @@ async function streamRealtimeBlocks(ws, blocks, responseId, fixture, defaults, l
 				reason: fault.stop === "length" ? "max_output_tokens" : "content_filter"
 			} } : {},
 			output: allOutputItems,
-			usage: fault?.usage ?? {
-				total_tokens: 0,
-				input_tokens: 0,
-				output_tokens: 0
-			}
+			usage: fault?.usage ?? render({ kind: "ordinary-zero" })
 		}
 	}, isBeta);
 	for (const item of allOutputItems) {