@copilotkit/aimock 1.46.0 → 1.47.0

package/dist/ws-responses.cjs

@@ -2,11 +2,13 @@ const require_errors = require('./response/errors.cjs');
 const require_constants = require('./constants.cjs');
 const require_policy = require('./response/policy.cjs');
 const require_helpers = require('./helpers.cjs');
-const require_sse_writer = require('./sse-writer.cjs');
+const require_frame_loop = require('./response/frame-loop.cjs');
 const require_misbehavior = require('./misbehavior.cjs');
 const require_router = require('./router.cjs');
 require('./journal.cjs');
-const require_interruption = require('./interruption.cjs');
+const require_lifecycle = require('./response/lifecycle.cjs');
+const require_timing = require('./response/timing.cjs');
+const require_raw = require('./response/raw.cjs');
 const require_responses = require('./responses.cjs');
 
 //#region src/ws-responses.ts
@@ -25,6 +27,7 @@ function buildErrorEvent(message, type = "invalid_request_error", code) {
 }
 function handleWebSocketResponses(ws, fixtures, journal, defaults) {
 	const { logger } = defaults;
+	require_lifecycle.watchWsClose(ws);
 	let pending = Promise.resolve();
 	ws.on("message", (raw) => {
 		pending = pending.then(async () => {
@@ -80,7 +83,7 @@ async function processMessage(raw, ws, fixtures, journal, defaults) {
 	const contextHeader = defaults.upgradeHeaders?.["x-aimock-context"];
 	completionReq._context = typeof contextHeader === "string" ? contextHeader : Array.isArray(contextHeader) && contextHeader.length > 0 ? contextHeader[0] : void 0;
 	const testId = defaults.testId ?? require_constants.DEFAULT_TEST_ID;
-	const { fixture, skippedBySequenceOrTurn } = require_router.matchFixtureDiagnostic(fixtures, completionReq, journal.getFixtureMatchCountsForTest(testId), defaults.requestTransform);
+	const { fixture, skippedBySequenceOrTurn } = require_router.matchFixtureDiagnostic(fixtures, completionReq, journal.getFixtureMatchCountsForTest(testId), defaults.requestTransform, { wire: "openai-responses" });
 	if (fixture) journal.incrementFixtureMatchCount(fixture, fixtures, testId);
 	if (!fixture) {
 		if (require_policy.resolveRequestDecisions(defaults.strict, defaults.upgradeHeaders).fixtureMiss) {
@@ -158,8 +161,6 @@ async function processMessage(raw, ws, fixtures, journal, defaults) {
 	const latency = fixture.latency ?? defaults.latency;
 	const chunkSize = Math.max(1, fixture.chunkSize ?? defaults.chunkSize);
 	if (misbehavior.kind === "applied") {
-		const prepared = require_responses.prepareResponsesMisbehavior(misbehavior);
-		const output = require_responses.buildResponsesMisbehavior(prepared, completionReq, chunkSize, emitEncryptedReasoning);
 		const entry = journal.add({
 			method: "WS",
 			path: "/v1/responses",
@@ -170,28 +171,41 @@ async function processMessage(raw, ws, fixtures, journal, defaults) {
 				fixture
 			}
 		});
+		const prepared = require_responses.prepareResponsesMisbehavior(misbehavior);
+		const output = require_helpers.journalFixtureToolCallError(entry, () => require_responses.buildResponsesMisbehavior(prepared, completionReq, chunkSize, emitEncryptedReasoning));
 		require_misbehavior.recordMisbehaviorOutcome({
 			entry,
 			summary: prepared.summary,
 			defaults: misbehaviorDefaults,
 			testId
 		});
-		const timings = fixture.recordedTimings;
-		const gaps = timings?.interChunkDelaysMs;
-		const appliedTimings = prepared.duplicateId && timings && gaps ? {
-			...timings,
-			interChunkDelaysMs: Array.from({ length: Math.max(gaps.length, output.events.length - 1) }, (_, index) => gaps[index] ?? gaps.at(-1) ?? 0)
-		} : timings;
-		const interruption = require_interruption.createInterruptionSignal(fixture);
-		if (!await sendEvents(ws, output.events, latency, interruption?.signal, interruption?.tick, appliedTimings, fixture.replaySpeed ?? defaults.replaySpeed)) {
-			ws.destroy();
-			entry.response.interrupted = true;
-			entry.response.interruptReason = interruption?.reason();
-		}
-		interruption?.cleanup();
+		const appliedTimings = require_timing.extendRecordedTimings(fixture.recordedTimings, output.events.length, !!prepared.duplicateId);
+		await deliverTurn(ws, entry, fixture, output.events, {
+			latency,
+			recordedTimings: appliedTimings,
+			replaySpeed: fixture.replaySpeed ?? defaults.replaySpeed
+		});
 		return;
 	}
 	const effectiveStrict = require_policy.resolveRequestDecisions(defaults.strict, defaults.upgradeHeaders).reasoningSuppression;
+	if (require_helpers.isRawResponse(response)) {
+		const message = "raw fixtures are not yet supported on the WebSocket transport";
+		const entry = journal.add({
+			method: "WS",
+			path: "/v1/responses",
+			headers: require_helpers.flattenHeaders(defaults.upgradeHeaders ?? {}),
+			body: completionReq,
+			response: {
+				status: 500,
+				fixture
+			}
+		});
+		entry.response.error = message;
+		recordOutcome(entry);
+		defaults.logger.error(`WebSocket responses error: ${require_raw.RAW_MODE_UNAVAILABLE}: ${message}`);
+		ws.send(JSON.stringify(buildErrorEvent(message, "server_error", require_raw.RAW_MODE_UNAVAILABLE)));
+		return;
+	}
 	if (require_helpers.isErrorResponse(response)) {
 		const status = response.status ?? 500;
 		recordOutcome(journal.add({
@@ -229,13 +243,11 @@ async function processMessage(raw, ws, fixtures, journal, defaults) {
 			if (require_helpers.isFixtureToolCallError(error)) journalEntry.response.error = error.message;
 			throw error;
 		}
-		const interruption = require_interruption.createInterruptionSignal(fixture);
-		if (!await sendEvents(ws, events, latency, interruption?.signal, interruption?.tick, fixture.recordedTimings, fixture.replaySpeed ?? defaults.replaySpeed)) {
-			ws.destroy();
-			journalEntry.response.interrupted = true;
-			journalEntry.response.interruptReason = interruption?.reason();
-		}
-		interruption?.cleanup();
+		await deliverTurn(ws, journalEntry, fixture, events, {
+			latency,
+			recordedTimings: fixture.recordedTimings,
+			replaySpeed: fixture.replaySpeed ?? defaults.replaySpeed
+		});
 		return;
 	}
 	if (require_helpers.isTextResponse(response)) {
@@ -250,14 +262,11 @@ async function processMessage(raw, ws, fixtures, journal, defaults) {
 			}
 		});
 		recordOutcome(journalEntry);
-		const events = require_responses.buildTextStreamEvents(response.content, completionReq.model, chunkSize, require_helpers.resolveReasoningForModel(response.reasoning, completionReq.model, effectiveStrict, defaults.logger), response.webSearches, require_helpers.extractOverrides(response), emitEncryptedReasoning, synthesizeSummarylessReasoning);
-		const interruption = require_interruption.createInterruptionSignal(fixture);
-		if (!await sendEvents(ws, events, latency, interruption?.signal, interruption?.tick, fixture.recordedTimings, fixture.replaySpeed ?? defaults.replaySpeed)) {
-			ws.destroy();
-			journalEntry.response.interrupted = true;
-			journalEntry.response.interruptReason = interruption?.reason();
-		}
-		interruption?.cleanup();
+		await deliverTurn(ws, journalEntry, fixture, require_helpers.journalFixtureToolCallError(journalEntry, () => require_responses.buildTextStreamEvents(response.content, completionReq.model, chunkSize, require_helpers.resolveReasoningForModel(response.reasoning, completionReq.model, effectiveStrict, defaults.logger), response.webSearches, require_helpers.extractOverrides(response), emitEncryptedReasoning, synthesizeSummarylessReasoning)), {
+			latency,
+			recordedTimings: fixture.recordedTimings,
+			replaySpeed: fixture.replaySpeed ?? defaults.replaySpeed
+		});
 		return;
 	}
 	if (require_helpers.isToolCallResponse(response)) {
@@ -281,13 +290,11 @@ async function processMessage(raw, ws, fixtures, journal, defaults) {
 			if (require_helpers.isFixtureToolCallError(error)) journalEntry.response.error = error.message;
 			throw error;
 		}
-		const interruption = require_interruption.createInterruptionSignal(fixture);
-		if (!await sendEvents(ws, events, latency, interruption?.signal, interruption?.tick, fixture.recordedTimings, fixture.replaySpeed ?? defaults.replaySpeed)) {
-			ws.destroy();
-			journalEntry.response.interrupted = true;
-			journalEntry.response.interruptReason = interruption?.reason();
-		}
-		interruption?.cleanup();
+		await deliverTurn(ws, journalEntry, fixture, events, {
+			latency,
+			recordedTimings: fixture.recordedTimings,
+			replaySpeed: fixture.replaySpeed ?? defaults.replaySpeed
+		});
 		return;
 	}
 	recordOutcome(journal.add({
@@ -302,20 +309,32 @@ async function processMessage(raw, ws, fixtures, journal, defaults) {
 	}));
 	ws.send(JSON.stringify(buildErrorEvent("Fixture response did not match any known type", "server_error")));
 }
-async function sendEvents(ws, events, latency, signal, onChunkSent, recordedTimings, replaySpeed) {
-	let eventIndex = 0;
-	for (const event of events) {
-		if (ws.isClosed) return false;
-		const chunkDelay = require_sse_writer.calculateDelay(eventIndex, void 0, latency, recordedTimings, replaySpeed);
-		if (chunkDelay > 0) await require_sse_writer.delay(chunkDelay, signal);
-		if (signal?.aborted) return false;
-		if (ws.isClosed) return false;
-		ws.send(JSON.stringify(event));
-		eventIndex++;
-		onChunkSent?.();
-		if (signal?.aborted) return false;
+/** WS Responses frames: one JSON text message per event, no head, no trailer. */
+const wsEventCodec = {
+	headers: {},
+	encode: (event) => JSON.stringify(event),
+	checkGoneBeforeDelay: true
+};
+/**
+* Send one turn's events under a WS lifecycle. The lifecycle owns the fixture
+* interruption and also stops the turn when the client closes, so a pending
+* delay ends at once. A stopped turn marks the entry by the shared WS rule
+* ({@link markWsTurnStopped}) and destroys the socket.
+* `streamingProfile` is not applied on this transport, as before.
+*/
+async function deliverTurn(ws, entry, fixture, events, timing) {
+	const lifecycle = require_lifecycle.wsLifecycle(ws, fixture);
+	try {
+		if (!await require_frame_loop.runFrameLoop(require_frame_loop.wsSink(ws), events, wsEventCodec, timing, {
+			lifecycle,
+			onChunkSent: lifecycle.tick
+		})) {
+			require_lifecycle.markWsTurnStopped(entry, ws, lifecycle);
+			ws.destroy();
+		}
+	} finally {
+		lifecycle.dispose();
 	}
-	return true;
 }
 
 //#endregion

package/dist/ws-responses.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"ws-responses.cjs","names":["buildOpenAIError","isFixtureToolCallError","fixtureToolCallErrorCode","requestWantsEncryptedReasoning","requestIncludesEncryptedReasoning","responsesToCompletionRequest","DEFAULT_TEST_ID","matchFixtureDiagnostic","resolveRequestDecisions","strictNoMatchMessage","strictNoMatchLogLine","flattenHeaders","strictOverrideField","resolveResponse","withoutLegacyNamespaces","planMisbehavior","recordMisbehaviorOutcome","prepareResponsesMisbehavior","buildResponsesMisbehavior","createInterruptionSignal","isErrorResponse","isCombinedFixtureResponse","resolveServedBlockOutcome","contentWithToolCallsStreamEvents","servedToolCalls","resolveReasoningForModel","extractOverrides","isTextResponse","buildTextStreamEvents","isToolCallResponse","toolCallStreamEvents","calculateDelay","delay"],"sources":["../src/ws-responses.ts"],"sourcesContent":["/**\n * WebSocket handler for OpenAI Responses API.\n *\n * Accepts `{ type: \"response.create\", model: \"...\", input: [...] }` messages over\n * WebSocket and sends back the same Responses API SSE events as the HTTP\n * handler, but as individual WebSocket text frames.\n */\n\nimport type { Fixture, HandlerDefaults } from \"./types.js\";\nimport { planMisbehavior, recordMisbehaviorOutcome } from \"./misbehavior.js\";\nimport { matchFixtureDiagnostic } from \"./router.js\";\nimport {\n  responsesToCompletionRequest,\n  requestIncludesEncryptedReasoning,\n  requestWantsEncryptedReasoning,\n  buildTextStreamEvents,\n  toolCallStreamEvents,\n  contentWithToolCallsStreamEvents,\n  prepareResponsesMisbehavior,\n  buildResponsesMisbehavior,\n  type ResponsesSSEEvent,\n  type ResponsesInputItem,\n  type ResponsesToolDef,\n} from \"./responses.js\";\nimport {\n  fixtureToolCallErrorCode,\n  isFixtureToolCallError,\n  isTextResponse,\n  isToolCallResponse,\n  isCombinedFixtureResponse,\n  isErrorResponse,\n  extractOverrides,\n  resolveResponse,\n  resolveReasoningForModel,\n  flattenHeaders,\n  strictNoMatchMessage,\n  strictNoMatchLogLine,\n  assertCustomToolCalls,\n  resolveServedBlockOutcome,\n  servedToolCalls,\n  withoutLegacyNamespaces,\n} from \"./helpers.js\";\nimport { createInterruptionSignal } from \"./interruption.js\";\nimport { delay, calculateDelay } from \"./sse-writer.js\";\nimport { DEFAULT_TEST_ID, type Journal } from \"./journal.js\";\nimport type { WebSocketConnection } from \"./ws-framing.js\";\nimport { resolveRequestDecisions, strictOverrideField } from \"./response/policy.js\";\nimport { buildOpenAIError } from \"./response/errors.js\";\n\ntype ResponsesWebSocketDefaults = Omit<HandlerDefaults, \"replaySpeed\"> & {\n  replaySpeed?: number;\n  model: string;\n  testId?: string;\n  upgradeHeaders?: import(\"node:http\").IncomingHttpHeaders;\n  /** @internal Called before each message is processed (the server assigns fixture positions). */\n  beforeProcessMessage?: () => void;\n};\n\ninterface ResponseCreateMessage {\n  type: \"response.create\";\n  model?: string;\n  input?: unknown[];\n  instructions?: string;\n  tools?: unknown[];\n  tool_choice?: string | object;\n  stream?: boolean;\n  temperature?: number;\n  max_output_tokens?: number;\n  [key: string]: unknown;\n}\n\nfunction isResponseCreateMessage(msg: unknown): msg is ResponseCreateMessage {\n  return (\n    typeof msg === \"object\" &&\n    msg !== null &&\n    (msg as ResponseCreateMessage).type === \"response.create\"\n  );\n}\n\nfunction buildErrorEvent(\n  message: string,\n  type = \"invalid_request_error\",\n  code?: string,\n): ResponsesSSEEvent {\n  return {\n    type: \"error\",\n    error: buildOpenAIError({ message, type, code }),\n  };\n}\n\nexport function handleWebSocketResponses(\n  ws: WebSocketConnection,\n  fixtures: Fixture[],\n  journal: Journal,\n  defaults: ResponsesWebSocketDefaults,\n): void {\n  const { logger } = defaults;\n  // Serialize message processing to prevent event interleaving\n  let pending = Promise.resolve();\n  ws.on(\"message\", (raw: string) => {\n    pending = pending.then(async () => {\n      try {\n        defaults.beforeProcessMessage?.();\n        await processMessage(raw, ws, fixtures, journal, defaults);\n      } catch (err: unknown) {\n        const msg = err instanceof Error ? err.message : \"Internal error\";\n        logger.error(`WebSocket responses error: ${msg}`);\n        try {\n          const code = isFixtureToolCallError(err) ? fixtureToolCallErrorCode(err) : undefined;\n          ws.send(JSON.stringify(buildErrorEvent(msg, \"server_error\", code)));\n        } catch (sendErr) {\n          defaults.logger.debug(\n            `Failed to send error to client: ${sendErr instanceof Error ? sendErr.message : \"unknown\"}`,\n          );\n        }\n      }\n    });\n  });\n}\n\nasync function processMessage(\n  raw: string,\n  ws: WebSocketConnection,\n  fixtures: Fixture[],\n  journal: Journal,\n  defaults: ResponsesWebSocketDefaults,\n): Promise<void> {\n  let parsed: unknown;\n  try {\n    parsed = JSON.parse(raw);\n  } catch (parseErr) {\n    const detail = parseErr instanceof Error ? parseErr.message : \"unknown\";\n    ws.send(\n      JSON.stringify(\n        buildErrorEvent(`Malformed JSON: ${detail}`, \"invalid_request_error\", \"invalid_json\"),\n      ),\n    );\n    return;\n  }\n\n  if (!isResponseCreateMessage(parsed)) {\n    ws.send(\n      JSON.stringify(\n        buildErrorEvent(\n          'Expected message type \"response.create\"',\n          \"invalid_request_error\",\n          \"invalid_message_type\",\n        ),\n      ),\n    );\n    return;\n  }\n\n  const responsesReq = {\n    model: parsed.model ?? defaults.model,\n    // Same shapes as HTTP: namespace / custom tools, additional_tools items\n    // and custom tool call items reach responsesToCompletionRequest unchanged.\n    input: (parsed.input ?? []) as ResponsesInputItem[],\n    instructions: parsed.instructions,\n    tools: parsed.tools as ResponsesToolDef[] | undefined,\n    tool_choice: parsed.tool_choice,\n    stream: parsed.stream,\n    temperature: parsed.temperature,\n    max_output_tokens: parsed.max_output_tokens,\n    include: (parsed as { include?: string[] }).include,\n    store: (parsed as { store?: boolean }).store,\n  };\n\n  // Gate encrypted-reasoning emission identically to the HTTP transport so the\n  // agent-framework#7233 stateless-replay path works over WebSocket too.\n  const emitEncryptedReasoning = requestWantsEncryptedReasoning(responsesReq);\n  // Synthesizing a reasoning item the fixture never declared takes the NARROWER\n  // explicit-`include` gate, transport-identically to HTTP — a `store: false`\n  // request with a summary-less fixture still gets no reasoning item.\n  const synthesizeSummarylessReasoning = requestIncludesEncryptedReasoning(responsesReq);\n\n  const extendedTools = defaults.responsesTools === \"extended\";\n  const completionReq = responsesToCompletionRequest(responsesReq, {\n    extended: extendedTools,\n    logger: defaults.logger,\n  });\n  completionReq._endpointType = \"chat\";\n  const contextHeader = defaults.upgradeHeaders?.[\"x-aimock-context\"];\n  completionReq._context =\n    typeof contextHeader === \"string\"\n      ? contextHeader\n      : Array.isArray(contextHeader) && contextHeader.length > 0\n        ? contextHeader[0]\n        : undefined;\n  const testId = defaults.testId ?? DEFAULT_TEST_ID;\n  const { fixture, skippedBySequenceOrTurn } = matchFixtureDiagnostic(\n    fixtures,\n    completionReq,\n    journal.getFixtureMatchCountsForTest(testId),\n    defaults.requestTransform,\n  );\n\n  if (fixture) {\n    journal.incrementFixtureMatchCount(fixture, fixtures, testId);\n  }\n\n  if (!fixture) {\n    if (resolveRequestDecisions(defaults.strict, defaults.upgradeHeaders).fixtureMiss) {\n      const strictMessage = strictNoMatchMessage(skippedBySequenceOrTurn);\n      defaults.logger.error(strictNoMatchLogLine(\"WS\", \"/v1/responses\", skippedBySequenceOrTurn));\n      journal.add({\n        method: \"WS\",\n        path: \"/v1/responses\",\n        headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n        body: completionReq,\n        response: {\n          status: 503,\n          fixture: null,\n          ...strictOverrideField(defaults.strict, defaults.upgradeHeaders),\n        },\n      });\n      ws.close(1008, strictMessage);\n      return;\n    }\n    journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: {\n        status: 404,\n        fixture: null,\n        ...strictOverrideField(defaults.strict, defaults.upgradeHeaders),\n      },\n    });\n    ws.send(\n      JSON.stringify(\n        buildErrorEvent(\"No fixture matched\", \"invalid_request_error\", \"no_fixture_match\"),\n      ),\n    );\n    return;\n  }\n\n  // Without responsesTools \"extended\", a namespace on a toolCalls entry or a\n  // toolCall block is not emitted, as in earlier releases.\n  const resolvedResponse = await resolveResponse(fixture, completionReq);\n  const response = extendedTools ? resolvedResponse : withoutLegacyNamespaces(resolvedResponse);\n  const misbehaviorDefaults: HandlerDefaults = {\n    ...defaults,\n    replaySpeed: defaults.replaySpeed ?? 1,\n  };\n  const misbehavior = planMisbehavior({\n    wire: \"openai-responses\",\n    fixture,\n    response,\n    request: completionReq,\n    // Responses WebSocket always emits streaming events, even with stream:false.\n    stream: true,\n    emitsToolCallIds: true,\n    defaults: misbehaviorDefaults,\n    // The server has already resolved the upgrade's header/query test ID.\n    rawHeaders: { ...defaults.upgradeHeaders, \"x-test-id\": testId },\n    url: \"/v1/responses\",\n  });\n  const recordOutcome = (entry: ReturnType<Journal[\"add\"]>) =>\n    recordMisbehaviorOutcome({\n      entry,\n      summary: misbehavior.summary,\n      defaults: misbehaviorDefaults,\n      testId,\n    });\n  if (misbehavior.kind === \"error\") {\n    const entry = journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: misbehavior.status, fixture },\n    });\n    recordOutcome(entry);\n    if (!misbehavior.summary?.evaluations.length) {\n      defaults.logger.error(\n        `[misbehavior] ${misbehavior.code}: ${misbehavior.message} (testId=${testId})`,\n      );\n    }\n    ws.send(\n      JSON.stringify(\n        buildErrorEvent(misbehavior.message, \"invalid_request_error\", misbehavior.code),\n      ),\n    );\n    return;\n  }\n  const latency = fixture.latency ?? defaults.latency;\n  const chunkSize = Math.max(1, fixture.chunkSize ?? defaults.chunkSize);\n\n  if (misbehavior.kind === \"applied\") {\n    const prepared = prepareResponsesMisbehavior(misbehavior);\n    const output = buildResponsesMisbehavior(\n      prepared,\n      completionReq,\n      chunkSize,\n      emitEncryptedReasoning,\n    );\n    const entry = journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n    recordMisbehaviorOutcome({\n      entry,\n      summary: prepared.summary,\n      defaults: misbehaviorDefaults,\n      testId,\n    });\n    const timings = fixture.recordedTimings;\n    const gaps = timings?.interChunkDelaysMs;\n    // K4 can add events; extend only its replay using the last recorded gap.\n    const appliedTimings =\n      prepared.duplicateId && timings && gaps\n        ? {\n            ...timings,\n            interChunkDelaysMs: Array.from(\n              { length: Math.max(gaps.length, output.events.length - 1) },\n              (_, index) => gaps[index] ?? gaps.at(-1) ?? 0,\n            ),\n          }\n        : timings;\n    const interruption = createInterruptionSignal(fixture);\n    const completed = await sendEvents(\n      ws,\n      output.events,\n      latency,\n      interruption?.signal,\n      interruption?.tick,\n      appliedTimings,\n      fixture.replaySpeed ?? defaults.replaySpeed,\n    );\n    if (!completed) {\n      ws.destroy();\n      entry.response.interrupted = true;\n      entry.response.interruptReason = interruption?.reason();\n    }\n    interruption?.cleanup();\n    return;\n  }\n\n  // The WS path has no per-request `req.headers`; strict is resolved from the\n  // connection's upgrade headers (see the `!fixture` branch above). Used below\n  // to gate the synthesized reasoning channel on the requested model's capability.\n  const effectiveStrict = resolveRequestDecisions(\n    defaults.strict,\n    defaults.upgradeHeaders,\n  ).reasoningSuppression;\n\n  // Error response\n  if (isErrorResponse(response)) {\n    const status = response.status ?? 500;\n    const entry = journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status, fixture },\n    });\n    recordOutcome(entry);\n    ws.send(\n      JSON.stringify(\n        buildErrorEvent(response.error.message, response.error.type, response.error.code),\n      ),\n    );\n    return;\n  }\n\n  // Content + tool calls response (must be checked before isTextResponse / isToolCallResponse)\n  if (isCombinedFixtureResponse(response)) {\n    const journalEntry = journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n    recordOutcome(journalEntry);\n\n    let events: ResponsesSSEEvent[];\n    try {\n      const ordered = resolveServedBlockOutcome(response)?.ordered;\n      if (!ordered) assertCustomToolCalls(response);\n      events = contentWithToolCallsStreamEvents(\n        response.content ?? \"\",\n        servedToolCalls(response),\n        completionReq.model,\n        chunkSize,\n        resolveReasoningForModel(\n          response.reasoning,\n          completionReq.model,\n          effectiveStrict,\n          defaults.logger,\n        ),\n        response.webSearches,\n        extractOverrides(response),\n        ordered,\n        emitEncryptedReasoning,\n        synthesizeSummarylessReasoning,\n      );\n    } catch (error) {\n      journalEntry.response.status = 500;\n      if (isFixtureToolCallError(error)) journalEntry.response.error = error.message;\n      throw error;\n    }\n\n    const interruption = createInterruptionSignal(fixture);\n    const completed = await sendEvents(\n      ws,\n      events,\n      latency,\n      interruption?.signal,\n      interruption?.tick,\n      fixture.recordedTimings,\n      fixture.replaySpeed ?? defaults.replaySpeed,\n    );\n    if (!completed) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n    }\n    interruption?.cleanup();\n    return;\n  }\n\n  // Text response\n  if (isTextResponse(response)) {\n    const journalEntry = journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n    recordOutcome(journalEntry);\n\n    const events = buildTextStreamEvents(\n      response.content,\n      completionReq.model,\n      chunkSize,\n      resolveReasoningForModel(\n        response.reasoning,\n        completionReq.model,\n        effectiveStrict,\n        defaults.logger,\n      ),\n      response.webSearches,\n      extractOverrides(response),\n      emitEncryptedReasoning,\n      synthesizeSummarylessReasoning,\n    );\n    const interruption = createInterruptionSignal(fixture);\n    const completed = await sendEvents(\n      ws,\n      events,\n      latency,\n      interruption?.signal,\n      interruption?.tick,\n      fixture.recordedTimings,\n      fixture.replaySpeed ?? defaults.replaySpeed,\n    );\n    if (!completed) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n    }\n    interruption?.cleanup();\n    return;\n  }\n\n  // Tool call response\n  if (isToolCallResponse(response)) {\n    const journalEntry = journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n    recordOutcome(journalEntry);\n    let events: ResponsesSSEEvent[];\n    try {\n      assertCustomToolCalls(response);\n      events = toolCallStreamEvents(\n        servedToolCalls(response),\n        completionReq.model,\n        chunkSize,\n        // Gate the synthesized reasoning channel on the requested model's\n        // capability, matching the WS text / content+tool branches and the HTTP\n        // tool-only path so reasoning emission is transport-independent.\n        resolveReasoningForModel(\n          response.reasoning,\n          completionReq.model,\n          effectiveStrict,\n          defaults.logger,\n        ),\n        response.webSearches,\n        extractOverrides(response),\n        emitEncryptedReasoning,\n        synthesizeSummarylessReasoning,\n      );\n    } catch (error) {\n      journalEntry.response.status = 500;\n      if (isFixtureToolCallError(error)) journalEntry.response.error = error.message;\n      throw error;\n    }\n    const interruption = createInterruptionSignal(fixture);\n    const completed = await sendEvents(\n      ws,\n      events,\n      latency,\n      interruption?.signal,\n      interruption?.tick,\n      fixture.recordedTimings,\n      fixture.replaySpeed ?? defaults.replaySpeed,\n    );\n    if (!completed) {\n      ws.destroy();\n      journalEntry.response.interrupted = true;\n      journalEntry.response.interruptReason = interruption?.reason();\n    }\n    interruption?.cleanup();\n    return;\n  }\n\n  // Unknown response type\n  const entry = journal.add({\n    method: \"WS\",\n    path: \"/v1/responses\",\n    headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n    body: completionReq,\n    response: { status: 500, fixture },\n  });\n  recordOutcome(entry);\n  ws.send(\n    JSON.stringify(\n      buildErrorEvent(\"Fixture response did not match any known type\", \"server_error\"),\n    ),\n  );\n}\n\nasync function sendEvents(\n  ws: WebSocketConnection,\n  events: ResponsesSSEEvent[],\n  latency: number,\n  signal?: AbortSignal,\n  onChunkSent?: () => void,\n  recordedTimings?: import(\"./types.js\").RecordedTimings,\n  replaySpeed?: number,\n): Promise<boolean> {\n  let eventIndex = 0;\n  for (const event of events) {\n    if (ws.isClosed) return false;\n    const chunkDelay = calculateDelay(eventIndex, undefined, latency, recordedTimings, replaySpeed);\n    if (chunkDelay > 0) await delay(chunkDelay, signal);\n    if (signal?.aborted) return false;\n    if (ws.isClosed) return false;\n    ws.send(JSON.stringify(event));\n    eventIndex++;\n    onChunkSent?.();\n    if (signal?.aborted) return false;\n  }\n  return true;\n}\n"],"mappings":";;;;;;;;;;;;AAuEA,SAAS,wBAAwB,KAA4C;AAC3E,QACE,OAAO,QAAQ,YACf,QAAQ,QACP,IAA8B,SAAS;;AAI5C,SAAS,gBACP,SACA,OAAO,yBACP,MACmB;AACnB,QAAO;EACL,MAAM;EACN,OAAOA,gCAAiB;GAAE;GAAS;GAAM;GAAM,CAAC;EACjD;;AAGH,SAAgB,yBACd,IACA,UACA,SACA,UACM;CACN,MAAM,EAAE,WAAW;CAEnB,IAAI,UAAU,QAAQ,SAAS;AAC/B,IAAG,GAAG,YAAY,QAAgB;AAChC,YAAU,QAAQ,KAAK,YAAY;AACjC,OAAI;AACF,aAAS,wBAAwB;AACjC,UAAM,eAAe,KAAK,IAAI,UAAU,SAAS,SAAS;YACnD,KAAc;IACrB,MAAM,MAAM,eAAe,QAAQ,IAAI,UAAU;AACjD,WAAO,MAAM,8BAA8B,MAAM;AACjD,QAAI;KACF,MAAM,OAAOC,uCAAuB,IAAI,GAAGC,yCAAyB,IAAI,GAAG;AAC3E,QAAG,KAAK,KAAK,UAAU,gBAAgB,KAAK,gBAAgB,KAAK,CAAC,CAAC;aAC5D,SAAS;AAChB,cAAS,OAAO,MACd,mCAAmC,mBAAmB,QAAQ,QAAQ,UAAU,YACjF;;;IAGL;GACF;;AAGJ,eAAe,eACb,KACA,IACA,UACA,SACA,UACe;CACf,IAAI;AACJ,KAAI;AACF,WAAS,KAAK,MAAM,IAAI;UACjB,UAAU;EACjB,MAAM,SAAS,oBAAoB,QAAQ,SAAS,UAAU;AAC9D,KAAG,KACD,KAAK,UACH,gBAAgB,mBAAmB,UAAU,yBAAyB,eAAe,CACtF,CACF;AACD;;AAGF,KAAI,CAAC,wBAAwB,OAAO,EAAE;AACpC,KAAG,KACD,KAAK,UACH,gBACE,6CACA,yBACA,uBACD,CACF,CACF;AACD;;CAGF,MAAM,eAAe;EACnB,OAAO,OAAO,SAAS,SAAS;EAGhC,OAAQ,OAAO,SAAS,EAAE;EAC1B,cAAc,OAAO;EACrB,OAAO,OAAO;EACd,aAAa,OAAO;EACpB,QAAQ,OAAO;EACf,aAAa,OAAO;EACpB,mBAAmB,OAAO;EAC1B,SAAU,OAAkC;EAC5C,OAAQ,OAA+B;EACxC;CAID,MAAM,yBAAyBC,iDAA+B,aAAa;CAI3E,MAAM,iCAAiCC,oDAAkC,aAAa;CAEtF,MAAM,gBAAgB,SAAS,mBAAmB;CAClD,MAAM,gBAAgBC,+CAA6B,cAAc;EAC/D,UAAU;EACV,QAAQ,SAAS;EAClB,CAAC;AACF,eAAc,gBAAgB;CAC9B,MAAM,gBAAgB,SAAS,iBAAiB;AAChD,eAAc,WACZ,OAAO,kBAAkB,WACrB,gBACA,MAAM,QAAQ,cAAc,IAAI,cAAc,SAAS,IACrD,cAAc,KACd;CACR,MAAM,SAAS,SAAS,UAAUC;CAClC,MAAM,EAAE,SAAS,4BAA4BC,sCAC3C,UACA,eACA,QAAQ,6BAA6B,OAAO,EAC5C,SAAS,iBACV;AAED,KAAI,QACF,SAAQ,2BAA2B,SAAS,UAAU,OAAO;AAG/D,KAAI,CAAC,SAAS;AACZ,MAAIC,uCAAwB,SAAS,QAAQ,SAAS,eAAe,CAAC,aAAa;GACjF,MAAM,gBAAgBC,qCAAqB,wBAAwB;AACnE,YAAS,OAAO,MAAMC,qCAAqB,MAAM,iBAAiB,wBAAwB,CAAC;AAC3F,WAAQ,IAAI;IACV,QAAQ;IACR,MAAM;IACN,SAASC,+BAAe,SAAS,kBAAkB,EAAE,CAAC;IACtD,MAAM;IACN,UAAU;KACR,QAAQ;KACR,SAAS;KACT,GAAGC,mCAAoB,SAAS,QAAQ,SAAS,eAAe;KACjE;IACF,CAAC;AACF,MAAG,MAAM,MAAM,cAAc;AAC7B;;AAEF,UAAQ,IAAI;GACV,QAAQ;GACR,MAAM;GACN,SAASD,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IACR,QAAQ;IACR,SAAS;IACT,GAAGC,mCAAoB,SAAS,QAAQ,SAAS,eAAe;IACjE;GACF,CAAC;AACF,KAAG,KACD,KAAK,UACH,gBAAgB,sBAAsB,yBAAyB,mBAAmB,CACnF,CACF;AACD;;CAKF,MAAM,mBAAmB,MAAMC,gCAAgB,SAAS,cAAc;CACtE,MAAM,WAAW,gBAAgB,mBAAmBC,wCAAwB,iBAAiB;CAC7F,MAAM,sBAAuC;EAC3C,GAAG;EACH,aAAa,SAAS,eAAe;EACtC;CACD,MAAM,cAAcC,oCAAgB;EAClC,MAAM;EACN;EACA;EACA,SAAS;EAET,QAAQ;EACR,kBAAkB;EAClB,UAAU;EAEV,YAAY;GAAE,GAAG,SAAS;GAAgB,aAAa;GAAQ;EAC/D,KAAK;EACN,CAAC;CACF,MAAM,iBAAiB,UACrBC,6CAAyB;EACvB;EACA,SAAS,YAAY;EACrB,UAAU;EACV;EACD,CAAC;AACJ,KAAI,YAAY,SAAS,SAAS;AAQhC,gBAPc,QAAQ,IAAI;GACxB,QAAQ;GACR,MAAM;GACN,SAASL,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ,YAAY;IAAQ;IAAS;GAClD,CAAC,CACkB;AACpB,MAAI,CAAC,YAAY,SAAS,YAAY,OACpC,UAAS,OAAO,MACd,iBAAiB,YAAY,KAAK,IAAI,YAAY,QAAQ,WAAW,OAAO,GAC7E;AAEH,KAAG,KACD,KAAK,UACH,gBAAgB,YAAY,SAAS,yBAAyB,YAAY,KAAK,CAChF,CACF;AACD;;CAEF,MAAM,UAAU,QAAQ,WAAW,SAAS;CAC5C,MAAM,YAAY,KAAK,IAAI,GAAG,QAAQ,aAAa,SAAS,UAAU;AAEtE,KAAI,YAAY,SAAS,WAAW;EAClC,MAAM,WAAWM,8CAA4B,YAAY;EACzD,MAAM,SAASC,4CACb,UACA,eACA,WACA,uBACD;EACD,MAAM,QAAQ,QAAQ,IAAI;GACxB,QAAQ;GACR,MAAM;GACN,SAASP,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;AACF,+CAAyB;GACvB;GACA,SAAS,SAAS;GAClB,UAAU;GACV;GACD,CAAC;EACF,MAAM,UAAU,QAAQ;EACxB,MAAM,OAAO,SAAS;EAEtB,MAAM,iBACJ,SAAS,eAAe,WAAW,OAC/B;GACE,GAAG;GACH,oBAAoB,MAAM,KACxB,EAAE,QAAQ,KAAK,IAAI,KAAK,QAAQ,OAAO,OAAO,SAAS,EAAE,EAAE,GAC1D,GAAG,UAAU,KAAK,UAAU,KAAK,GAAG,GAAG,IAAI,EAC7C;GACF,GACD;EACN,MAAM,eAAeQ,8CAAyB,QAAQ;AAUtD,MAAI,CATc,MAAM,WACtB,IACA,OAAO,QACP,SACA,cAAc,QACd,cAAc,MACd,gBACA,QAAQ,eAAe,SAAS,YACjC,EACe;AACd,MAAG,SAAS;AACZ,SAAM,SAAS,cAAc;AAC7B,SAAM,SAAS,kBAAkB,cAAc,QAAQ;;AAEzD,gBAAc,SAAS;AACvB;;CAMF,MAAM,kBAAkBX,uCACtB,SAAS,QACT,SAAS,eACV,CAAC;AAGF,KAAIY,gCAAgB,SAAS,EAAE;EAC7B,MAAM,SAAS,SAAS,UAAU;AAQlC,gBAPc,QAAQ,IAAI;GACxB,QAAQ;GACR,MAAM;GACN,SAAST,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE;IAAQ;IAAS;GAC9B,CAAC,CACkB;AACpB,KAAG,KACD,KAAK,UACH,gBAAgB,SAAS,MAAM,SAAS,SAAS,MAAM,MAAM,SAAS,MAAM,KAAK,CAClF,CACF;AACD;;AAIF,KAAIU,0CAA0B,SAAS,EAAE;EACvC,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR,MAAM;GACN,SAASV,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;AACF,gBAAc,aAAa;EAE3B,IAAI;AACJ,MAAI;GACF,MAAM,UAAUW,0CAA0B,SAAS,EAAE;AACrD,OAAI,CAAC,QAAS,uCAAsB,SAAS;AAC7C,YAASC,mDACP,SAAS,WAAW,IACpBC,gCAAgB,SAAS,EACzB,cAAc,OACd,WACAC,yCACE,SAAS,WACT,cAAc,OACd,iBACA,SAAS,OACV,EACD,SAAS,aACTC,iCAAiB,SAAS,EAC1B,SACA,wBACA,+BACD;WACM,OAAO;AACd,gBAAa,SAAS,SAAS;AAC/B,OAAIzB,uCAAuB,MAAM,CAAE,cAAa,SAAS,QAAQ,MAAM;AACvE,SAAM;;EAGR,MAAM,eAAekB,8CAAyB,QAAQ;AAUtD,MAAI,CATc,MAAM,WACtB,IACA,QACA,SACA,cAAc,QACd,cAAc,MACd,QAAQ,iBACR,QAAQ,eAAe,SAAS,YACjC,EACe;AACd,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;;AAEhE,gBAAc,SAAS;AACvB;;AAIF,KAAIQ,+BAAe,SAAS,EAAE;EAC5B,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR,MAAM;GACN,SAAShB,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;AACF,gBAAc,aAAa;EAE3B,MAAM,SAASiB,wCACb,SAAS,SACT,cAAc,OACd,WACAH,yCACE,SAAS,WACT,cAAc,OACd,iBACA,SAAS,OACV,EACD,SAAS,aACTC,iCAAiB,SAAS,EAC1B,wBACA,+BACD;EACD,MAAM,eAAeP,8CAAyB,QAAQ;AAUtD,MAAI,CATc,MAAM,WACtB,IACA,QACA,SACA,cAAc,QACd,cAAc,MACd,QAAQ,iBACR,QAAQ,eAAe,SAAS,YACjC,EACe;AACd,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;;AAEhE,gBAAc,SAAS;AACvB;;AAIF,KAAIU,mCAAmB,SAAS,EAAE;EAChC,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR,MAAM;GACN,SAASlB,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;AACF,gBAAc,aAAa;EAC3B,IAAI;AACJ,MAAI;AACF,yCAAsB,SAAS;AAC/B,YAASmB,uCACPN,gCAAgB,SAAS,EACzB,cAAc,OACd,WAIAC,yCACE,SAAS,WACT,cAAc,OACd,iBACA,SAAS,OACV,EACD,SAAS,aACTC,iCAAiB,SAAS,EAC1B,wBACA,+BACD;WACM,OAAO;AACd,gBAAa,SAAS,SAAS;AAC/B,OAAIzB,uCAAuB,MAAM,CAAE,cAAa,SAAS,QAAQ,MAAM;AACvE,SAAM;;EAER,MAAM,eAAekB,8CAAyB,QAAQ;AAUtD,MAAI,CATc,MAAM,WACtB,IACA,QACA,SACA,cAAc,QACd,cAAc,MACd,QAAQ,iBACR,QAAQ,eAAe,SAAS,YACjC,EACe;AACd,MAAG,SAAS;AACZ,gBAAa,SAAS,cAAc;AACpC,gBAAa,SAAS,kBAAkB,cAAc,QAAQ;;AAEhE,gBAAc,SAAS;AACvB;;AAWF,eAPc,QAAQ,IAAI;EACxB,QAAQ;EACR,MAAM;EACN,SAASR,+BAAe,SAAS,kBAAkB,EAAE,CAAC;EACtD,MAAM;EACN,UAAU;GAAE,QAAQ;GAAK;GAAS;EACnC,CAAC,CACkB;AACpB,IAAG,KACD,KAAK,UACH,gBAAgB,iDAAiD,eAAe,CACjF,CACF;;AAGH,eAAe,WACb,IACA,QACA,SACA,QACA,aACA,iBACA,aACkB;CAClB,IAAI,aAAa;AACjB,MAAK,MAAM,SAAS,QAAQ;AAC1B,MAAI,GAAG,SAAU,QAAO;EACxB,MAAM,aAAaoB,kCAAe,YAAY,QAAW,SAAS,iBAAiB,YAAY;AAC/F,MAAI,aAAa,EAAG,OAAMC,yBAAM,YAAY,OAAO;AACnD,MAAI,QAAQ,QAAS,QAAO;AAC5B,MAAI,GAAG,SAAU,QAAO;AACxB,KAAG,KAAK,KAAK,UAAU,MAAM,CAAC;AAC9B;AACA,iBAAe;AACf,MAAI,QAAQ,QAAS,QAAO;;AAE9B,QAAO"}
+{"version":3,"file":"ws-responses.cjs","names":["buildOpenAIError","isFixtureToolCallError","fixtureToolCallErrorCode","requestWantsEncryptedReasoning","requestIncludesEncryptedReasoning","responsesToCompletionRequest","DEFAULT_TEST_ID","matchFixtureDiagnostic","resolveRequestDecisions","strictNoMatchMessage","strictNoMatchLogLine","flattenHeaders","strictOverrideField","resolveResponse","withoutLegacyNamespaces","planMisbehavior","recordMisbehaviorOutcome","prepareResponsesMisbehavior","journalFixtureToolCallError","buildResponsesMisbehavior","extendRecordedTimings","isRawResponse","RAW_MODE_UNAVAILABLE","isErrorResponse","isCombinedFixtureResponse","resolveServedBlockOutcome","contentWithToolCallsStreamEvents","servedToolCalls","resolveReasoningForModel","extractOverrides","isTextResponse","buildTextStreamEvents","isToolCallResponse","toolCallStreamEvents","wsLifecycle","runFrameLoop","wsSink"],"sources":["../src/ws-responses.ts"],"sourcesContent":["/**\n * WebSocket handler for OpenAI Responses API.\n *\n * Accepts `{ type: \"response.create\", model: \"...\", input: [...] }` messages over\n * WebSocket and sends back the same Responses API SSE events as the HTTP\n * handler, but as individual WebSocket text frames.\n */\n\nimport type { Fixture, HandlerDefaults, JournalEntry, RecordedTimings } from \"./types.js\";\nimport { planMisbehavior, recordMisbehaviorOutcome } from \"./misbehavior.js\";\nimport { matchFixtureDiagnostic } from \"./router.js\";\nimport {\n  responsesToCompletionRequest,\n  requestIncludesEncryptedReasoning,\n  requestWantsEncryptedReasoning,\n  buildTextStreamEvents,\n  toolCallStreamEvents,\n  contentWithToolCallsStreamEvents,\n  prepareResponsesMisbehavior,\n  buildResponsesMisbehavior,\n  type ResponsesSSEEvent,\n  type ResponsesInputItem,\n  type ResponsesToolDef,\n} from \"./responses.js\";\nimport {\n  fixtureToolCallErrorCode,\n  isFixtureToolCallError,\n  journalFixtureToolCallError,\n  isTextResponse,\n  isToolCallResponse,\n  isCombinedFixtureResponse,\n  isErrorResponse,\n  isRawResponse,\n  extractOverrides,\n  resolveResponse,\n  resolveReasoningForModel,\n  flattenHeaders,\n  strictNoMatchMessage,\n  strictNoMatchLogLine,\n  assertCustomToolCalls,\n  resolveServedBlockOutcome,\n  servedToolCalls,\n  withoutLegacyNamespaces,\n} from \"./helpers.js\";\nimport { DEFAULT_TEST_ID, type Journal } from \"./journal.js\";\nimport type { WebSocketConnection } from \"./ws-framing.js\";\nimport { resolveRequestDecisions, strictOverrideField } from \"./response/policy.js\";\nimport { buildOpenAIError } from \"./response/errors.js\";\nimport { markWsTurnStopped, watchWsClose, wsLifecycle } from \"./response/lifecycle.js\";\nimport { runFrameLoop, wsSink, type FrameCodec } from \"./response/frame-loop.js\";\nimport { extendRecordedTimings } from \"./response/timing.js\";\nimport { RAW_MODE_UNAVAILABLE } from \"./response/raw.js\";\n\ntype ResponsesWebSocketDefaults = Omit<HandlerDefaults, \"replaySpeed\"> & {\n  replaySpeed?: number;\n  model: string;\n  testId?: string;\n  upgradeHeaders?: import(\"node:http\").IncomingHttpHeaders;\n  /** @internal Called before each message is processed (the server assigns fixture positions). */\n  beforeProcessMessage?: () => void;\n};\n\ninterface ResponseCreateMessage {\n  type: \"response.create\";\n  model?: string;\n  input?: unknown[];\n  instructions?: string;\n  tools?: unknown[];\n  tool_choice?: string | object;\n  stream?: boolean;\n  temperature?: number;\n  max_output_tokens?: number;\n  [key: string]: unknown;\n}\n\nfunction isResponseCreateMessage(msg: unknown): msg is ResponseCreateMessage {\n  return (\n    typeof msg === \"object\" &&\n    msg !== null &&\n    (msg as ResponseCreateMessage).type === \"response.create\"\n  );\n}\n\nfunction buildErrorEvent(\n  message: string,\n  type = \"invalid_request_error\",\n  code?: string,\n): ResponsesSSEEvent {\n  return {\n    type: \"error\",\n    error: buildOpenAIError({ message, type, code }),\n  };\n}\n\nexport function handleWebSocketResponses(\n  ws: WebSocketConnection,\n  fixtures: Fixture[],\n  journal: Journal,\n  defaults: ResponsesWebSocketDefaults,\n): void {\n  const { logger } = defaults;\n  watchWsClose(ws);\n  // Serialize message processing to prevent event interleaving\n  let pending = Promise.resolve();\n  ws.on(\"message\", (raw: string) => {\n    pending = pending.then(async () => {\n      try {\n        defaults.beforeProcessMessage?.();\n        await processMessage(raw, ws, fixtures, journal, defaults);\n      } catch (err: unknown) {\n        const msg = err instanceof Error ? err.message : \"Internal error\";\n        logger.error(`WebSocket responses error: ${msg}`);\n        try {\n          const code = isFixtureToolCallError(err) ? fixtureToolCallErrorCode(err) : undefined;\n          ws.send(JSON.stringify(buildErrorEvent(msg, \"server_error\", code)));\n        } catch (sendErr) {\n          defaults.logger.debug(\n            `Failed to send error to client: ${sendErr instanceof Error ? sendErr.message : \"unknown\"}`,\n          );\n        }\n      }\n    });\n  });\n}\n\nasync function processMessage(\n  raw: string,\n  ws: WebSocketConnection,\n  fixtures: Fixture[],\n  journal: Journal,\n  defaults: ResponsesWebSocketDefaults,\n): Promise<void> {\n  let parsed: unknown;\n  try {\n    parsed = JSON.parse(raw);\n  } catch (parseErr) {\n    const detail = parseErr instanceof Error ? parseErr.message : \"unknown\";\n    ws.send(\n      JSON.stringify(\n        buildErrorEvent(`Malformed JSON: ${detail}`, \"invalid_request_error\", \"invalid_json\"),\n      ),\n    );\n    return;\n  }\n\n  if (!isResponseCreateMessage(parsed)) {\n    ws.send(\n      JSON.stringify(\n        buildErrorEvent(\n          'Expected message type \"response.create\"',\n          \"invalid_request_error\",\n          \"invalid_message_type\",\n        ),\n      ),\n    );\n    return;\n  }\n\n  const responsesReq = {\n    model: parsed.model ?? defaults.model,\n    // Same shapes as HTTP: namespace / custom tools, additional_tools items\n    // and custom tool call items reach responsesToCompletionRequest unchanged.\n    input: (parsed.input ?? []) as ResponsesInputItem[],\n    instructions: parsed.instructions,\n    tools: parsed.tools as ResponsesToolDef[] | undefined,\n    tool_choice: parsed.tool_choice,\n    stream: parsed.stream,\n    temperature: parsed.temperature,\n    max_output_tokens: parsed.max_output_tokens,\n    include: (parsed as { include?: string[] }).include,\n    store: (parsed as { store?: boolean }).store,\n  };\n\n  // Gate encrypted-reasoning emission identically to the HTTP transport so the\n  // agent-framework#7233 stateless-replay path works over WebSocket too.\n  const emitEncryptedReasoning = requestWantsEncryptedReasoning(responsesReq);\n  // Synthesizing a reasoning item the fixture never declared takes the NARROWER\n  // explicit-`include` gate, transport-identically to HTTP — a `store: false`\n  // request with a summary-less fixture still gets no reasoning item.\n  const synthesizeSummarylessReasoning = requestIncludesEncryptedReasoning(responsesReq);\n\n  const extendedTools = defaults.responsesTools === \"extended\";\n  const completionReq = responsesToCompletionRequest(responsesReq, {\n    extended: extendedTools,\n    logger: defaults.logger,\n  });\n  completionReq._endpointType = \"chat\";\n  const contextHeader = defaults.upgradeHeaders?.[\"x-aimock-context\"];\n  completionReq._context =\n    typeof contextHeader === \"string\"\n      ? contextHeader\n      : Array.isArray(contextHeader) && contextHeader.length > 0\n        ? contextHeader[0]\n        : undefined;\n  const testId = defaults.testId ?? DEFAULT_TEST_ID;\n  const { fixture, skippedBySequenceOrTurn } = matchFixtureDiagnostic(\n    fixtures,\n    completionReq,\n    journal.getFixtureMatchCountsForTest(testId),\n    defaults.requestTransform,\n    // Wire only: recordMatchOptions here would turn on strict turnIndex and the\n    // relaxed-turnIndex warning, which this transport has never had.\n    { wire: \"openai-responses\" },\n  );\n\n  if (fixture) {\n    journal.incrementFixtureMatchCount(fixture, fixtures, testId);\n  }\n\n  if (!fixture) {\n    if (resolveRequestDecisions(defaults.strict, defaults.upgradeHeaders).fixtureMiss) {\n      const strictMessage = strictNoMatchMessage(skippedBySequenceOrTurn);\n      defaults.logger.error(strictNoMatchLogLine(\"WS\", \"/v1/responses\", skippedBySequenceOrTurn));\n      journal.add({\n        method: \"WS\",\n        path: \"/v1/responses\",\n        headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n        body: completionReq,\n        response: {\n          status: 503,\n          fixture: null,\n          ...strictOverrideField(defaults.strict, defaults.upgradeHeaders),\n        },\n      });\n      ws.close(1008, strictMessage);\n      return;\n    }\n    journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: {\n        status: 404,\n        fixture: null,\n        ...strictOverrideField(defaults.strict, defaults.upgradeHeaders),\n      },\n    });\n    ws.send(\n      JSON.stringify(\n        buildErrorEvent(\"No fixture matched\", \"invalid_request_error\", \"no_fixture_match\"),\n      ),\n    );\n    return;\n  }\n\n  // Without responsesTools \"extended\", a namespace on a toolCalls entry or a\n  // toolCall block is not emitted, as in earlier releases.\n  const resolvedResponse = await resolveResponse(fixture, completionReq);\n  const response = extendedTools ? resolvedResponse : withoutLegacyNamespaces(resolvedResponse);\n  const misbehaviorDefaults: HandlerDefaults = {\n    ...defaults,\n    replaySpeed: defaults.replaySpeed ?? 1,\n  };\n  const misbehavior = planMisbehavior({\n    wire: \"openai-responses\",\n    fixture,\n    response,\n    request: completionReq,\n    // Responses WebSocket always emits streaming events, even with stream:false.\n    stream: true,\n    emitsToolCallIds: true,\n    defaults: misbehaviorDefaults,\n    // The server has already resolved the upgrade's header/query test ID.\n    rawHeaders: { ...defaults.upgradeHeaders, \"x-test-id\": testId },\n    url: \"/v1/responses\",\n  });\n  const recordOutcome = (entry: ReturnType<Journal[\"add\"]>) =>\n    recordMisbehaviorOutcome({\n      entry,\n      summary: misbehavior.summary,\n      defaults: misbehaviorDefaults,\n      testId,\n    });\n  if (misbehavior.kind === \"error\") {\n    const entry = journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: misbehavior.status, fixture },\n    });\n    recordOutcome(entry);\n    if (!misbehavior.summary?.evaluations.length) {\n      defaults.logger.error(\n        `[misbehavior] ${misbehavior.code}: ${misbehavior.message} (testId=${testId})`,\n      );\n    }\n    ws.send(\n      JSON.stringify(\n        buildErrorEvent(misbehavior.message, \"invalid_request_error\", misbehavior.code),\n      ),\n    );\n    return;\n  }\n  const latency = fixture.latency ?? defaults.latency;\n  const chunkSize = Math.max(1, fixture.chunkSize ?? defaults.chunkSize);\n\n  if (misbehavior.kind === \"applied\") {\n    // Journal first, as HTTP /v1/responses does, so a fixture error thrown\n    // while building leaves a 500 entry that keeps the fixture. Record the\n    // fault only once the reply is built: a thrown error was not served.\n    const entry = journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n    const prepared = prepareResponsesMisbehavior(misbehavior);\n    const output = journalFixtureToolCallError(entry, () =>\n      buildResponsesMisbehavior(prepared, completionReq, chunkSize, emitEncryptedReasoning),\n    );\n    recordMisbehaviorOutcome({\n      entry,\n      summary: prepared.summary,\n      defaults: misbehaviorDefaults,\n      testId,\n    });\n    // K4 can add events; extend only its replay using the last recorded gap.\n    const appliedTimings = extendRecordedTimings(\n      fixture.recordedTimings,\n      output.events.length,\n      !!prepared.duplicateId,\n    );\n    await deliverTurn(ws, entry, fixture, output.events, {\n      latency,\n      recordedTimings: appliedTimings,\n      replaySpeed: fixture.replaySpeed ?? defaults.replaySpeed,\n    });\n    return;\n  }\n\n  // The WS path has no per-request `req.headers`; strict is resolved from the\n  // connection's upgrade headers (see the `!fixture` branch above). Used below\n  // to gate the synthesized reasoning channel on the requested model's capability.\n  const effectiveStrict = resolveRequestDecisions(\n    defaults.strict,\n    defaults.upgradeHeaders,\n  ).reasoningSuppression;\n\n  // Raw passthrough fixtures are served over HTTP only for now. Fail loudly\n  // instead of the generic unknown-type error so the author sees why.\n  if (isRawResponse(response)) {\n    const message = \"raw fixtures are not yet supported on the WebSocket transport\";\n    const entry = journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 500, fixture },\n    });\n    entry.response.error = message;\n    recordOutcome(entry);\n    defaults.logger.error(`WebSocket responses error: ${RAW_MODE_UNAVAILABLE}: ${message}`);\n    ws.send(JSON.stringify(buildErrorEvent(message, \"server_error\", RAW_MODE_UNAVAILABLE)));\n    return;\n  }\n\n  // Error response\n  if (isErrorResponse(response)) {\n    const status = response.status ?? 500;\n    const entry = journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status, fixture },\n    });\n    recordOutcome(entry);\n    ws.send(\n      JSON.stringify(\n        buildErrorEvent(response.error.message, response.error.type, response.error.code),\n      ),\n    );\n    return;\n  }\n\n  // Content + tool calls response (must be checked before isTextResponse / isToolCallResponse)\n  if (isCombinedFixtureResponse(response)) {\n    const journalEntry = journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n    recordOutcome(journalEntry);\n\n    let events: ResponsesSSEEvent[];\n    try {\n      const ordered = resolveServedBlockOutcome(response)?.ordered;\n      if (!ordered) assertCustomToolCalls(response);\n      events = contentWithToolCallsStreamEvents(\n        response.content ?? \"\",\n        servedToolCalls(response),\n        completionReq.model,\n        chunkSize,\n        resolveReasoningForModel(\n          response.reasoning,\n          completionReq.model,\n          effectiveStrict,\n          defaults.logger,\n        ),\n        response.webSearches,\n        extractOverrides(response),\n        ordered,\n        emitEncryptedReasoning,\n        synthesizeSummarylessReasoning,\n      );\n    } catch (error) {\n      journalEntry.response.status = 500;\n      if (isFixtureToolCallError(error)) journalEntry.response.error = error.message;\n      throw error;\n    }\n\n    await deliverTurn(ws, journalEntry, fixture, events, {\n      latency,\n      recordedTimings: fixture.recordedTimings,\n      replaySpeed: fixture.replaySpeed ?? defaults.replaySpeed,\n    });\n    return;\n  }\n\n  // Text response\n  if (isTextResponse(response)) {\n    const journalEntry = journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n    recordOutcome(journalEntry);\n\n    // A fixture error (malformed citations, a non-image file) marks the entry\n    // a 500 carrying the message before the error event goes out.\n    const events = journalFixtureToolCallError(journalEntry, () =>\n      buildTextStreamEvents(\n        response.content,\n        completionReq.model,\n        chunkSize,\n        resolveReasoningForModel(\n          response.reasoning,\n          completionReq.model,\n          effectiveStrict,\n          defaults.logger,\n        ),\n        response.webSearches,\n        extractOverrides(response),\n        emitEncryptedReasoning,\n        synthesizeSummarylessReasoning,\n      ),\n    );\n    await deliverTurn(ws, journalEntry, fixture, events, {\n      latency,\n      recordedTimings: fixture.recordedTimings,\n      replaySpeed: fixture.replaySpeed ?? defaults.replaySpeed,\n    });\n    return;\n  }\n\n  // Tool call response\n  if (isToolCallResponse(response)) {\n    const journalEntry = journal.add({\n      method: \"WS\",\n      path: \"/v1/responses\",\n      headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n      body: completionReq,\n      response: { status: 200, fixture },\n    });\n    recordOutcome(journalEntry);\n    let events: ResponsesSSEEvent[];\n    try {\n      assertCustomToolCalls(response);\n      events = toolCallStreamEvents(\n        servedToolCalls(response),\n        completionReq.model,\n        chunkSize,\n        // Gate the synthesized reasoning channel on the requested model's\n        // capability, matching the WS text / content+tool branches and the HTTP\n        // tool-only path so reasoning emission is transport-independent.\n        resolveReasoningForModel(\n          response.reasoning,\n          completionReq.model,\n          effectiveStrict,\n          defaults.logger,\n        ),\n        response.webSearches,\n        extractOverrides(response),\n        emitEncryptedReasoning,\n        synthesizeSummarylessReasoning,\n      );\n    } catch (error) {\n      journalEntry.response.status = 500;\n      if (isFixtureToolCallError(error)) journalEntry.response.error = error.message;\n      throw error;\n    }\n    await deliverTurn(ws, journalEntry, fixture, events, {\n      latency,\n      recordedTimings: fixture.recordedTimings,\n      replaySpeed: fixture.replaySpeed ?? defaults.replaySpeed,\n    });\n    return;\n  }\n\n  // Unknown response type\n  const entry = journal.add({\n    method: \"WS\",\n    path: \"/v1/responses\",\n    headers: flattenHeaders(defaults.upgradeHeaders ?? {}),\n    body: completionReq,\n    response: { status: 500, fixture },\n  });\n  recordOutcome(entry);\n  ws.send(\n    JSON.stringify(\n      buildErrorEvent(\"Fixture response did not match any known type\", \"server_error\"),\n    ),\n  );\n}\n\n/** WS Responses frames: one JSON text message per event, no head, no trailer. */\nconst wsEventCodec: FrameCodec<ResponsesSSEEvent> = {\n  headers: {},\n  encode: (event) => JSON.stringify(event),\n  checkGoneBeforeDelay: true,\n};\n\n/**\n * Send one turn's events under a WS lifecycle. The lifecycle owns the fixture\n * interruption and also stops the turn when the client closes, so a pending\n * delay ends at once. A stopped turn marks the entry by the shared WS rule\n * ({@link markWsTurnStopped}) and destroys the socket.\n * `streamingProfile` is not applied on this transport, as before.\n */\nasync function deliverTurn(\n  ws: WebSocketConnection,\n  entry: JournalEntry,\n  fixture: Fixture,\n  events: ResponsesSSEEvent[],\n  timing: { latency: number; recordedTimings?: RecordedTimings; replaySpeed?: number },\n): Promise<void> {\n  const lifecycle = wsLifecycle(ws, fixture);\n  try {\n    const completed = await runFrameLoop(wsSink(ws), events, wsEventCodec, timing, {\n      lifecycle,\n      onChunkSent: lifecycle.tick,\n    });\n    if (!completed) {\n      markWsTurnStopped(entry, ws, lifecycle);\n      ws.destroy();\n    }\n  } finally {\n    lifecycle.dispose();\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;AA2EA,SAAS,wBAAwB,KAA4C;AAC3E,QACE,OAAO,QAAQ,YACf,QAAQ,QACP,IAA8B,SAAS;;AAI5C,SAAS,gBACP,SACA,OAAO,yBACP,MACmB;AACnB,QAAO;EACL,MAAM;EACN,OAAOA,gCAAiB;GAAE;GAAS;GAAM;GAAM,CAAC;EACjD;;AAGH,SAAgB,yBACd,IACA,UACA,SACA,UACM;CACN,MAAM,EAAE,WAAW;AACnB,gCAAa,GAAG;CAEhB,IAAI,UAAU,QAAQ,SAAS;AAC/B,IAAG,GAAG,YAAY,QAAgB;AAChC,YAAU,QAAQ,KAAK,YAAY;AACjC,OAAI;AACF,aAAS,wBAAwB;AACjC,UAAM,eAAe,KAAK,IAAI,UAAU,SAAS,SAAS;YACnD,KAAc;IACrB,MAAM,MAAM,eAAe,QAAQ,IAAI,UAAU;AACjD,WAAO,MAAM,8BAA8B,MAAM;AACjD,QAAI;KACF,MAAM,OAAOC,uCAAuB,IAAI,GAAGC,yCAAyB,IAAI,GAAG;AAC3E,QAAG,KAAK,KAAK,UAAU,gBAAgB,KAAK,gBAAgB,KAAK,CAAC,CAAC;aAC5D,SAAS;AAChB,cAAS,OAAO,MACd,mCAAmC,mBAAmB,QAAQ,QAAQ,UAAU,YACjF;;;IAGL;GACF;;AAGJ,eAAe,eACb,KACA,IACA,UACA,SACA,UACe;CACf,IAAI;AACJ,KAAI;AACF,WAAS,KAAK,MAAM,IAAI;UACjB,UAAU;EACjB,MAAM,SAAS,oBAAoB,QAAQ,SAAS,UAAU;AAC9D,KAAG,KACD,KAAK,UACH,gBAAgB,mBAAmB,UAAU,yBAAyB,eAAe,CACtF,CACF;AACD;;AAGF,KAAI,CAAC,wBAAwB,OAAO,EAAE;AACpC,KAAG,KACD,KAAK,UACH,gBACE,6CACA,yBACA,uBACD,CACF,CACF;AACD;;CAGF,MAAM,eAAe;EACnB,OAAO,OAAO,SAAS,SAAS;EAGhC,OAAQ,OAAO,SAAS,EAAE;EAC1B,cAAc,OAAO;EACrB,OAAO,OAAO;EACd,aAAa,OAAO;EACpB,QAAQ,OAAO;EACf,aAAa,OAAO;EACpB,mBAAmB,OAAO;EAC1B,SAAU,OAAkC;EAC5C,OAAQ,OAA+B;EACxC;CAID,MAAM,yBAAyBC,iDAA+B,aAAa;CAI3E,MAAM,iCAAiCC,oDAAkC,aAAa;CAEtF,MAAM,gBAAgB,SAAS,mBAAmB;CAClD,MAAM,gBAAgBC,+CAA6B,cAAc;EAC/D,UAAU;EACV,QAAQ,SAAS;EAClB,CAAC;AACF,eAAc,gBAAgB;CAC9B,MAAM,gBAAgB,SAAS,iBAAiB;AAChD,eAAc,WACZ,OAAO,kBAAkB,WACrB,gBACA,MAAM,QAAQ,cAAc,IAAI,cAAc,SAAS,IACrD,cAAc,KACd;CACR,MAAM,SAAS,SAAS,UAAUC;CAClC,MAAM,EAAE,SAAS,4BAA4BC,sCAC3C,UACA,eACA,QAAQ,6BAA6B,OAAO,EAC5C,SAAS,kBAGT,EAAE,MAAM,oBAAoB,CAC7B;AAED,KAAI,QACF,SAAQ,2BAA2B,SAAS,UAAU,OAAO;AAG/D,KAAI,CAAC,SAAS;AACZ,MAAIC,uCAAwB,SAAS,QAAQ,SAAS,eAAe,CAAC,aAAa;GACjF,MAAM,gBAAgBC,qCAAqB,wBAAwB;AACnE,YAAS,OAAO,MAAMC,qCAAqB,MAAM,iBAAiB,wBAAwB,CAAC;AAC3F,WAAQ,IAAI;IACV,QAAQ;IACR,MAAM;IACN,SAASC,+BAAe,SAAS,kBAAkB,EAAE,CAAC;IACtD,MAAM;IACN,UAAU;KACR,QAAQ;KACR,SAAS;KACT,GAAGC,mCAAoB,SAAS,QAAQ,SAAS,eAAe;KACjE;IACF,CAAC;AACF,MAAG,MAAM,MAAM,cAAc;AAC7B;;AAEF,UAAQ,IAAI;GACV,QAAQ;GACR,MAAM;GACN,SAASD,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IACR,QAAQ;IACR,SAAS;IACT,GAAGC,mCAAoB,SAAS,QAAQ,SAAS,eAAe;IACjE;GACF,CAAC;AACF,KAAG,KACD,KAAK,UACH,gBAAgB,sBAAsB,yBAAyB,mBAAmB,CACnF,CACF;AACD;;CAKF,MAAM,mBAAmB,MAAMC,gCAAgB,SAAS,cAAc;CACtE,MAAM,WAAW,gBAAgB,mBAAmBC,wCAAwB,iBAAiB;CAC7F,MAAM,sBAAuC;EAC3C,GAAG;EACH,aAAa,SAAS,eAAe;EACtC;CACD,MAAM,cAAcC,oCAAgB;EAClC,MAAM;EACN;EACA;EACA,SAAS;EAET,QAAQ;EACR,kBAAkB;EAClB,UAAU;EAEV,YAAY;GAAE,GAAG,SAAS;GAAgB,aAAa;GAAQ;EAC/D,KAAK;EACN,CAAC;CACF,MAAM,iBAAiB,UACrBC,6CAAyB;EACvB;EACA,SAAS,YAAY;EACrB,UAAU;EACV;EACD,CAAC;AACJ,KAAI,YAAY,SAAS,SAAS;AAQhC,gBAPc,QAAQ,IAAI;GACxB,QAAQ;GACR,MAAM;GACN,SAASL,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ,YAAY;IAAQ;IAAS;GAClD,CAAC,CACkB;AACpB,MAAI,CAAC,YAAY,SAAS,YAAY,OACpC,UAAS,OAAO,MACd,iBAAiB,YAAY,KAAK,IAAI,YAAY,QAAQ,WAAW,OAAO,GAC7E;AAEH,KAAG,KACD,KAAK,UACH,gBAAgB,YAAY,SAAS,yBAAyB,YAAY,KAAK,CAChF,CACF;AACD;;CAEF,MAAM,UAAU,QAAQ,WAAW,SAAS;CAC5C,MAAM,YAAY,KAAK,IAAI,GAAG,QAAQ,aAAa,SAAS,UAAU;AAEtE,KAAI,YAAY,SAAS,WAAW;EAIlC,MAAM,QAAQ,QAAQ,IAAI;GACxB,QAAQ;GACR,MAAM;GACN,SAASA,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;EACF,MAAM,WAAWM,8CAA4B,YAAY;EACzD,MAAM,SAASC,4CAA4B,aACzCC,4CAA0B,UAAU,eAAe,WAAW,uBAAuB,CACtF;AACD,+CAAyB;GACvB;GACA,SAAS,SAAS;GAClB,UAAU;GACV;GACD,CAAC;EAEF,MAAM,iBAAiBC,qCACrB,QAAQ,iBACR,OAAO,OAAO,QACd,CAAC,CAAC,SAAS,YACZ;AACD,QAAM,YAAY,IAAI,OAAO,SAAS,OAAO,QAAQ;GACnD;GACA,iBAAiB;GACjB,aAAa,QAAQ,eAAe,SAAS;GAC9C,CAAC;AACF;;CAMF,MAAM,kBAAkBZ,uCACtB,SAAS,QACT,SAAS,eACV,CAAC;AAIF,KAAIa,8BAAc,SAAS,EAAE;EAC3B,MAAM,UAAU;EAChB,MAAM,QAAQ,QAAQ,IAAI;GACxB,QAAQ;GACR,MAAM;GACN,SAASV,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;AACF,QAAM,SAAS,QAAQ;AACvB,gBAAc,MAAM;AACpB,WAAS,OAAO,MAAM,8BAA8BW,iCAAqB,IAAI,UAAU;AACvF,KAAG,KAAK,KAAK,UAAU,gBAAgB,SAAS,gBAAgBA,iCAAqB,CAAC,CAAC;AACvF;;AAIF,KAAIC,gCAAgB,SAAS,EAAE;EAC7B,MAAM,SAAS,SAAS,UAAU;AAQlC,gBAPc,QAAQ,IAAI;GACxB,QAAQ;GACR,MAAM;GACN,SAASZ,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE;IAAQ;IAAS;GAC9B,CAAC,CACkB;AACpB,KAAG,KACD,KAAK,UACH,gBAAgB,SAAS,MAAM,SAAS,SAAS,MAAM,MAAM,SAAS,MAAM,KAAK,CAClF,CACF;AACD;;AAIF,KAAIa,0CAA0B,SAAS,EAAE;EACvC,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR,MAAM;GACN,SAASb,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;AACF,gBAAc,aAAa;EAE3B,IAAI;AACJ,MAAI;GACF,MAAM,UAAUc,0CAA0B,SAAS,EAAE;AACrD,OAAI,CAAC,QAAS,uCAAsB,SAAS;AAC7C,YAASC,mDACP,SAAS,WAAW,IACpBC,gCAAgB,SAAS,EACzB,cAAc,OACd,WACAC,yCACE,SAAS,WACT,cAAc,OACd,iBACA,SAAS,OACV,EACD,SAAS,aACTC,iCAAiB,SAAS,EAC1B,SACA,wBACA,+BACD;WACM,OAAO;AACd,gBAAa,SAAS,SAAS;AAC/B,OAAI5B,uCAAuB,MAAM,CAAE,cAAa,SAAS,QAAQ,MAAM;AACvE,SAAM;;AAGR,QAAM,YAAY,IAAI,cAAc,SAAS,QAAQ;GACnD;GACA,iBAAiB,QAAQ;GACzB,aAAa,QAAQ,eAAe,SAAS;GAC9C,CAAC;AACF;;AAIF,KAAI6B,+BAAe,SAAS,EAAE;EAC5B,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR,MAAM;GACN,SAASnB,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;AACF,gBAAc,aAAa;AAqB3B,QAAM,YAAY,IAAI,cAAc,SAjBrBO,4CAA4B,oBACzCa,wCACE,SAAS,SACT,cAAc,OACd,WACAH,yCACE,SAAS,WACT,cAAc,OACd,iBACA,SAAS,OACV,EACD,SAAS,aACTC,iCAAiB,SAAS,EAC1B,wBACA,+BACD,CACF,EACoD;GACnD;GACA,iBAAiB,QAAQ;GACzB,aAAa,QAAQ,eAAe,SAAS;GAC9C,CAAC;AACF;;AAIF,KAAIG,mCAAmB,SAAS,EAAE;EAChC,MAAM,eAAe,QAAQ,IAAI;GAC/B,QAAQ;GACR,MAAM;GACN,SAASrB,+BAAe,SAAS,kBAAkB,EAAE,CAAC;GACtD,MAAM;GACN,UAAU;IAAE,QAAQ;IAAK;IAAS;GACnC,CAAC;AACF,gBAAc,aAAa;EAC3B,IAAI;AACJ,MAAI;AACF,yCAAsB,SAAS;AAC/B,YAASsB,uCACPN,gCAAgB,SAAS,EACzB,cAAc,OACd,WAIAC,yCACE,SAAS,WACT,cAAc,OACd,iBACA,SAAS,OACV,EACD,SAAS,aACTC,iCAAiB,SAAS,EAC1B,wBACA,+BACD;WACM,OAAO;AACd,gBAAa,SAAS,SAAS;AAC/B,OAAI5B,uCAAuB,MAAM,CAAE,cAAa,SAAS,QAAQ,MAAM;AACvE,SAAM;;AAER,QAAM,YAAY,IAAI,cAAc,SAAS,QAAQ;GACnD;GACA,iBAAiB,QAAQ;GACzB,aAAa,QAAQ,eAAe,SAAS;GAC9C,CAAC;AACF;;AAWF,eAPc,QAAQ,IAAI;EACxB,QAAQ;EACR,MAAM;EACN,SAASU,+BAAe,SAAS,kBAAkB,EAAE,CAAC;EACtD,MAAM;EACN,UAAU;GAAE,QAAQ;GAAK;GAAS;EACnC,CAAC,CACkB;AACpB,IAAG,KACD,KAAK,UACH,gBAAgB,iDAAiD,eAAe,CACjF,CACF;;;AAIH,MAAM,eAA8C;CAClD,SAAS,EAAE;CACX,SAAS,UAAU,KAAK,UAAU,MAAM;CACxC,sBAAsB;CACvB;;;;;;;;AASD,eAAe,YACb,IACA,OACA,SACA,QACA,QACe;CACf,MAAM,YAAYuB,8BAAY,IAAI,QAAQ;AAC1C,KAAI;AAKF,MAAI,CAJc,MAAMC,gCAAaC,0BAAO,GAAG,EAAE,QAAQ,cAAc,QAAQ;GAC7E;GACA,aAAa,UAAU;GACxB,CAAC,EACc;AACd,uCAAkB,OAAO,IAAI,UAAU;AACvC,MAAG,SAAS;;WAEN;AACR,YAAU,SAAS"}

package/dist/ws-responses.d.cts.map

@@ -1 +1 @@
-{"version":3,"file":"ws-responses.d.cts","names":[],"sources":["../src/ws-responses.ts"],"sourcesContent":[],"mappings":";;;;;;;KAiDK,0BAAA,GAA6B,IAIK,CAJA,eAIA,EAAA,aAAA,CAAA,GAAA;EAAmB,WAAA,CAAA,EAAA,MAAA;EAqC1C,KAAA,EAAA,MAAA;EAAwB,MAAA,CAAA,EAAA,MAAA;gBAClC,CAAA,EA1CgC,UAAA,CAIC,mBAsCjC;;sBAEK,CAAA,EAAA,GAAA,GAAA,IAAA;;AAC2B,iBAJtB,wBAAA,CAIsB,EAAA,EAHhC,mBAGgC,EAAA,QAAA,EAF1B,OAE0B,EAAA,EAAA,OAAA,EAD3B,OAC2B,EAAA,QAAA,EAA1B,0BAA0B,CAAA,EAAA,IAAA"}
+{"version":3,"file":"ws-responses.d.cts","names":[],"sources":["../src/ws-responses.ts"],"sourcesContent":[],"mappings":";;;;;;;KAqDK,0BAAA,GAA6B,IAIK,CAJA,eAIA,EAAA,aAAA,CAAA,GAAA;EAAmB,WAAA,CAAA,EAAA,MAAA;EAqC1C,KAAA,EAAA,MAAA;EAAwB,MAAA,CAAA,EAAA,MAAA;gBAClC,CAAA,EA1CgC,UAAA,CAIC,mBAsCjC;;sBAEK,CAAA,EAAA,GAAA,GAAA,IAAA;;AAC2B,iBAJtB,wBAAA,CAIsB,EAAA,EAHhC,mBAGgC,EAAA,QAAA,EAF1B,OAE0B,EAAA,EAAA,OAAA,EAD3B,OAC2B,EAAA,QAAA,EAA1B,0BAA0B,CAAA,EAAA,IAAA"}

package/dist/ws-responses.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ws-responses.d.ts","names":[],"sources":["../src/ws-responses.ts"],"sourcesContent":[],"mappings":";;;;;;;KAiDK,0BAAA,GAA6B,IAIK,CAJA,eAIA,EAAA,aAAA,CAAA,GAAA;EAAmB,WAAA,CAAA,EAAA,MAAA;EAqC1C,KAAA,EAAA,MAAA;EAAwB,MAAA,CAAA,EAAA,MAAA;gBAClC,CAAA,EA1CgC,UAAA,CAIC,mBAsCjC;;sBAEK,CAAA,EAAA,GAAA,GAAA,IAAA;;AAC2B,iBAJtB,wBAAA,CAIsB,EAAA,EAHhC,mBAGgC,EAAA,QAAA,EAF1B,OAE0B,EAAA,EAAA,OAAA,EAD3B,OAC2B,EAAA,QAAA,EAA1B,0BAA0B,CAAA,EAAA,IAAA"}
+{"version":3,"file":"ws-responses.d.ts","names":[],"sources":["../src/ws-responses.ts"],"sourcesContent":[],"mappings":";;;;;;;KAqDK,0BAAA,GAA6B,IAIK,CAJA,eAIA,EAAA,aAAA,CAAA,GAAA;EAAmB,WAAA,CAAA,EAAA,MAAA;EAqC1C,KAAA,EAAA,MAAA;EAAwB,MAAA,CAAA,EAAA,MAAA;gBAClC,CAAA,EA1CgC,UAAA,CAIC,mBAsCjC;;sBAEK,CAAA,EAAA,GAAA,GAAA,IAAA;;AAC2B,iBAJtB,wBAAA,CAIsB,EAAA,EAHhC,mBAGgC,EAAA,QAAA,EAF1B,OAE0B,EAAA,EAAA,OAAA,EAD3B,OAC2B,EAAA,QAAA,EAA1B,0BAA0B,CAAA,EAAA,IAAA"}

package/dist/ws-responses.js

@@ -1,12 +1,14 @@
 import { buildOpenAIError } from "./response/errors.js";
 import { DEFAULT_TEST_ID } from "./constants.js";
 import { resolveRequestDecisions, strictOverrideField } from "./response/policy.js";
-import { assertCustomToolCalls, extractOverrides, fixtureToolCallErrorCode, flattenHeaders, isCombinedFixtureResponse, isErrorResponse, isFixtureToolCallError, isTextResponse, isToolCallResponse, resolveReasoningForModel, resolveResponse, resolveServedBlockOutcome, servedToolCalls, strictNoMatchLogLine, strictNoMatchMessage, withoutLegacyNamespaces } from "./helpers.js";
-import { calculateDelay, delay } from "./sse-writer.js";
+import { assertCustomToolCalls, extractOverrides, fixtureToolCallErrorCode, flattenHeaders, isCombinedFixtureResponse, isErrorResponse, isFixtureToolCallError, isRawResponse, isTextResponse, isToolCallResponse, journalFixtureToolCallError, resolveReasoningForModel, resolveResponse, resolveServedBlockOutcome, servedToolCalls, strictNoMatchLogLine, strictNoMatchMessage, withoutLegacyNamespaces } from "./helpers.js";
+import { runFrameLoop, wsSink } from "./response/frame-loop.js";
 import { planMisbehavior, recordMisbehaviorOutcome } from "./misbehavior.js";
 import { matchFixtureDiagnostic } from "./router.js";
 import "./journal.js";
-import { createInterruptionSignal } from "./interruption.js";
+import { markWsTurnStopped, watchWsClose, wsLifecycle } from "./response/lifecycle.js";
+import { extendRecordedTimings } from "./response/timing.js";
+import { RAW_MODE_UNAVAILABLE } from "./response/raw.js";
 import { buildResponsesMisbehavior, buildTextStreamEvents, contentWithToolCallsStreamEvents, prepareResponsesMisbehavior, requestIncludesEncryptedReasoning, requestWantsEncryptedReasoning, responsesToCompletionRequest, toolCallStreamEvents } from "./responses.js";
 
 //#region src/ws-responses.ts
@@ -25,6 +27,7 @@ function buildErrorEvent(message, type = "invalid_request_error", code) {
 }
 function handleWebSocketResponses(ws, fixtures, journal, defaults) {
 	const { logger } = defaults;
+	watchWsClose(ws);
 	let pending = Promise.resolve();
 	ws.on("message", (raw) => {
 		pending = pending.then(async () => {
@@ -80,7 +83,7 @@ async function processMessage(raw, ws, fixtures, journal, defaults) {
 	const contextHeader = defaults.upgradeHeaders?.["x-aimock-context"];
 	completionReq._context = typeof contextHeader === "string" ? contextHeader : Array.isArray(contextHeader) && contextHeader.length > 0 ? contextHeader[0] : void 0;
 	const testId = defaults.testId ?? DEFAULT_TEST_ID;
-	const { fixture, skippedBySequenceOrTurn } = matchFixtureDiagnostic(fixtures, completionReq, journal.getFixtureMatchCountsForTest(testId), defaults.requestTransform);
+	const { fixture, skippedBySequenceOrTurn } = matchFixtureDiagnostic(fixtures, completionReq, journal.getFixtureMatchCountsForTest(testId), defaults.requestTransform, { wire: "openai-responses" });
 	if (fixture) journal.incrementFixtureMatchCount(fixture, fixtures, testId);
 	if (!fixture) {
 		if (resolveRequestDecisions(defaults.strict, defaults.upgradeHeaders).fixtureMiss) {
@@ -158,8 +161,6 @@ async function processMessage(raw, ws, fixtures, journal, defaults) {
 	const latency = fixture.latency ?? defaults.latency;
 	const chunkSize = Math.max(1, fixture.chunkSize ?? defaults.chunkSize);
 	if (misbehavior.kind === "applied") {
-		const prepared = prepareResponsesMisbehavior(misbehavior);
-		const output = buildResponsesMisbehavior(prepared, completionReq, chunkSize, emitEncryptedReasoning);
 		const entry = journal.add({
 			method: "WS",
 			path: "/v1/responses",
@@ -170,28 +171,41 @@ async function processMessage(raw, ws, fixtures, journal, defaults) {
 				fixture
 			}
 		});
+		const prepared = prepareResponsesMisbehavior(misbehavior);
+		const output = journalFixtureToolCallError(entry, () => buildResponsesMisbehavior(prepared, completionReq, chunkSize, emitEncryptedReasoning));
 		recordMisbehaviorOutcome({
 			entry,
 			summary: prepared.summary,
 			defaults: misbehaviorDefaults,
 			testId
 		});
-		const timings = fixture.recordedTimings;
-		const gaps = timings?.interChunkDelaysMs;
-		const appliedTimings = prepared.duplicateId && timings && gaps ? {
-			...timings,
-			interChunkDelaysMs: Array.from({ length: Math.max(gaps.length, output.events.length - 1) }, (_, index) => gaps[index] ?? gaps.at(-1) ?? 0)
-		} : timings;
-		const interruption = createInterruptionSignal(fixture);
-		if (!await sendEvents(ws, output.events, latency, interruption?.signal, interruption?.tick, appliedTimings, fixture.replaySpeed ?? defaults.replaySpeed)) {
-			ws.destroy();
-			entry.response.interrupted = true;
-			entry.response.interruptReason = interruption?.reason();
-		}
-		interruption?.cleanup();
+		const appliedTimings = extendRecordedTimings(fixture.recordedTimings, output.events.length, !!prepared.duplicateId);
+		await deliverTurn(ws, entry, fixture, output.events, {
+			latency,
+			recordedTimings: appliedTimings,
+			replaySpeed: fixture.replaySpeed ?? defaults.replaySpeed
+		});
 		return;
 	}
 	const effectiveStrict = resolveRequestDecisions(defaults.strict, defaults.upgradeHeaders).reasoningSuppression;
+	if (isRawResponse(response)) {
+		const message = "raw fixtures are not yet supported on the WebSocket transport";
+		const entry = journal.add({
+			method: "WS",
+			path: "/v1/responses",
+			headers: flattenHeaders(defaults.upgradeHeaders ?? {}),
+			body: completionReq,
+			response: {
+				status: 500,
+				fixture
+			}
+		});
+		entry.response.error = message;
+		recordOutcome(entry);
+		defaults.logger.error(`WebSocket responses error: ${RAW_MODE_UNAVAILABLE}: ${message}`);
+		ws.send(JSON.stringify(buildErrorEvent(message, "server_error", RAW_MODE_UNAVAILABLE)));
+		return;
+	}
 	if (isErrorResponse(response)) {
 		const status = response.status ?? 500;
 		recordOutcome(journal.add({
@@ -229,13 +243,11 @@ async function processMessage(raw, ws, fixtures, journal, defaults) {
 			if (isFixtureToolCallError(error)) journalEntry.response.error = error.message;
 			throw error;
 		}
-		const interruption = createInterruptionSignal(fixture);
-		if (!await sendEvents(ws, events, latency, interruption?.signal, interruption?.tick, fixture.recordedTimings, fixture.replaySpeed ?? defaults.replaySpeed)) {
-			ws.destroy();
-			journalEntry.response.interrupted = true;
-			journalEntry.response.interruptReason = interruption?.reason();
-		}
-		interruption?.cleanup();
+		await deliverTurn(ws, journalEntry, fixture, events, {
+			latency,
+			recordedTimings: fixture.recordedTimings,
+			replaySpeed: fixture.replaySpeed ?? defaults.replaySpeed
+		});
 		return;
 	}
 	if (isTextResponse(response)) {
@@ -250,14 +262,11 @@ async function processMessage(raw, ws, fixtures, journal, defaults) {
 			}
 		});
 		recordOutcome(journalEntry);
-		const events = buildTextStreamEvents(response.content, completionReq.model, chunkSize, resolveReasoningForModel(response.reasoning, completionReq.model, effectiveStrict, defaults.logger), response.webSearches, extractOverrides(response), emitEncryptedReasoning, synthesizeSummarylessReasoning);
-		const interruption = createInterruptionSignal(fixture);
-		if (!await sendEvents(ws, events, latency, interruption?.signal, interruption?.tick, fixture.recordedTimings, fixture.replaySpeed ?? defaults.replaySpeed)) {
-			ws.destroy();
-			journalEntry.response.interrupted = true;
-			journalEntry.response.interruptReason = interruption?.reason();
-		}
-		interruption?.cleanup();
+		await deliverTurn(ws, journalEntry, fixture, journalFixtureToolCallError(journalEntry, () => buildTextStreamEvents(response.content, completionReq.model, chunkSize, resolveReasoningForModel(response.reasoning, completionReq.model, effectiveStrict, defaults.logger), response.webSearches, extractOverrides(response), emitEncryptedReasoning, synthesizeSummarylessReasoning)), {
+			latency,
+			recordedTimings: fixture.recordedTimings,
+			replaySpeed: fixture.replaySpeed ?? defaults.replaySpeed
+		});
 		return;
 	}
 	if (isToolCallResponse(response)) {
@@ -281,13 +290,11 @@ async function processMessage(raw, ws, fixtures, journal, defaults) {
 			if (isFixtureToolCallError(error)) journalEntry.response.error = error.message;
 			throw error;
 		}
-		const interruption = createInterruptionSignal(fixture);
-		if (!await sendEvents(ws, events, latency, interruption?.signal, interruption?.tick, fixture.recordedTimings, fixture.replaySpeed ?? defaults.replaySpeed)) {
-			ws.destroy();
-			journalEntry.response.interrupted = true;
-			journalEntry.response.interruptReason = interruption?.reason();
-		}
-		interruption?.cleanup();
+		await deliverTurn(ws, journalEntry, fixture, events, {
+			latency,
+			recordedTimings: fixture.recordedTimings,
+			replaySpeed: fixture.replaySpeed ?? defaults.replaySpeed
+		});
 		return;
 	}
 	recordOutcome(journal.add({
@@ -302,20 +309,32 @@ async function processMessage(raw, ws, fixtures, journal, defaults) {
 	}));
 	ws.send(JSON.stringify(buildErrorEvent("Fixture response did not match any known type", "server_error")));
 }
-async function sendEvents(ws, events, latency, signal, onChunkSent, recordedTimings, replaySpeed) {
-	let eventIndex = 0;
-	for (const event of events) {
-		if (ws.isClosed) return false;
-		const chunkDelay = calculateDelay(eventIndex, void 0, latency, recordedTimings, replaySpeed);
-		if (chunkDelay > 0) await delay(chunkDelay, signal);
-		if (signal?.aborted) return false;
-		if (ws.isClosed) return false;
-		ws.send(JSON.stringify(event));
-		eventIndex++;
-		onChunkSent?.();
-		if (signal?.aborted) return false;
+/** WS Responses frames: one JSON text message per event, no head, no trailer. */
+const wsEventCodec = {
+	headers: {},
+	encode: (event) => JSON.stringify(event),
+	checkGoneBeforeDelay: true
+};
+/**
+* Send one turn's events under a WS lifecycle. The lifecycle owns the fixture
+* interruption and also stops the turn when the client closes, so a pending
+* delay ends at once. A stopped turn marks the entry by the shared WS rule
+* ({@link markWsTurnStopped}) and destroys the socket.
+* `streamingProfile` is not applied on this transport, as before.
+*/
+async function deliverTurn(ws, entry, fixture, events, timing) {
+	const lifecycle = wsLifecycle(ws, fixture);
+	try {
+		if (!await runFrameLoop(wsSink(ws), events, wsEventCodec, timing, {
+			lifecycle,
+			onChunkSent: lifecycle.tick
+		})) {
+			markWsTurnStopped(entry, ws, lifecycle);
+			ws.destroy();
+		}
+	} finally {
+		lifecycle.dispose();
 	}
-	return true;
 }
 
 //#endregion