@copilotkit/aimock 1.7.0

package/dist/aimock-cli.js

@@ -0,0 +1,110 @@
+#!/usr/bin/env node
+import { loadConfig, startFromConfig } from "./config-loader.js";
+import { parseArgs } from "node:util";
+import { resolve } from "node:path";
+
+//#region src/aimock-cli.ts
+const HELP = `
+Usage: aimock [options]
+
+Options:
+  -c, --config <path>   Path to aimock config JSON file (required)
+  -p, --port <number>   Port override (default: from config or 0)
+  -h, --host <string>   Host override (default: from config or 127.0.0.1)
+      --help            Show this help message
+`.trim();
+function runAimockCli(deps = {}) {
+	/* v8 ignore next 6 -- defaults used only when called from CLI entry point */
+	const argv = deps.argv ?? process.argv.slice(2);
+	const log = deps.log ?? console.log.bind(console);
+	const logError = deps.logError ?? console.error.bind(console);
+	const exit = deps.exit ?? process.exit.bind(process);
+	const loadConfigFn = deps.loadConfigFn ?? loadConfig;
+	const startFromConfigFn = deps.startFromConfigFn ?? startFromConfig;
+	let values;
+	try {
+		({values} = parseArgs({
+			args: argv,
+			options: {
+				config: {
+					type: "string",
+					short: "c"
+				},
+				port: {
+					type: "string",
+					short: "p"
+				},
+				host: {
+					type: "string",
+					short: "h"
+				},
+				help: {
+					type: "boolean",
+					default: false
+				}
+			},
+			strict: true
+		}));
+	} catch (err) {
+		logError(`Error: ${err instanceof Error ? err.message : String(err)}\n\n${HELP}`);
+		exit(1);
+		return;
+	}
+	if (values.help) {
+		log(HELP);
+		exit(0);
+		return;
+	}
+	if (!values.config) {
+		logError("Error: --config is required.\n\n" + HELP);
+		exit(1);
+		return;
+	}
+	const configPath = resolve(values.config);
+	let config;
+	try {
+		config = loadConfigFn(configPath);
+	} catch (err) {
+		logError(`Failed to load config from ${configPath}: ${err instanceof Error ? err.message : String(err)}`);
+		exit(1);
+		return;
+	}
+	const port = values.port ? Number(values.port) : void 0;
+	if (port !== void 0 && (Number.isNaN(port) || !Number.isInteger(port) || port < 0 || port > 65535)) {
+		logError(`Error: invalid port "${values.port}".\n\n${HELP}`);
+		exit(1);
+		return;
+	}
+	const host = values.host;
+	async function main() {
+		const { llmock, url } = await startFromConfigFn(config, {
+			port,
+			host
+		});
+		log(`aimock server listening on ${url}`);
+		function shutdown() {
+			log("Shutting down...");
+			process.removeListener("SIGINT", shutdown);
+			process.removeListener("SIGTERM", shutdown);
+			llmock.stop().then(() => exit(0), (err) => {
+				logError(`Shutdown error: ${err instanceof Error ? err.message : String(err)}`);
+				exit(1);
+			});
+		}
+		process.on("SIGINT", shutdown);
+		process.on("SIGTERM", shutdown);
+		if (deps.onReady) deps.onReady({ shutdown });
+	}
+	main().catch((err) => {
+		logError(err instanceof Error ? err.message : String(err));
+		exit(1);
+	});
+}
+/* v8 ignore start -- entry-point guard, exercised by integration tests */
+const scriptName = process.argv[1] ?? "";
+if (scriptName.endsWith("aimock-cli.js") || scriptName.endsWith("aimock-cli.ts")) runAimockCli();
+/* v8 ignore stop */
+
+//#endregion
+export { runAimockCli };
+//# sourceMappingURL=aimock-cli.js.map

package/dist/aimock-cli.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"aimock-cli.js","names":[],"sources":["../src/aimock-cli.ts"],"sourcesContent":["#!/usr/bin/env node\nimport { parseArgs } from \"node:util\";\nimport { resolve } from \"node:path\";\nimport { loadConfig, startFromConfig } from \"./config-loader.js\";\n\nconst HELP = `\nUsage: aimock [options]\n\nOptions:\n  -c, --config <path>   Path to aimock config JSON file (required)\n  -p, --port <number>   Port override (default: from config or 0)\n  -h, --host <string>   Host override (default: from config or 127.0.0.1)\n      --help            Show this help message\n`.trim();\n\nexport interface AimockCliDeps {\n  argv?: string[];\n  log?: (msg: string) => void;\n  logError?: (msg: string) => void;\n  exit?: (code: number) => void;\n  loadConfigFn?: typeof loadConfig;\n  startFromConfigFn?: typeof startFromConfig;\n  onReady?: (ctx: { shutdown: () => void }) => void;\n}\n\nexport function runAimockCli(deps: AimockCliDeps = {}): void {\n  /* v8 ignore next 6 -- defaults used only when called from CLI entry point */\n  const argv = deps.argv ?? process.argv.slice(2);\n  const log = deps.log ?? console.log.bind(console);\n  const logError = deps.logError ?? console.error.bind(console);\n  const exit = deps.exit ?? process.exit.bind(process);\n  const loadConfigFn = deps.loadConfigFn ?? loadConfig;\n  const startFromConfigFn = deps.startFromConfigFn ?? startFromConfig;\n\n  let values;\n  try {\n    ({ values } = parseArgs({\n      args: argv,\n      options: {\n        config: { type: \"string\", short: \"c\" },\n        port: { type: \"string\", short: \"p\" },\n        host: { type: \"string\", short: \"h\" },\n        help: { type: \"boolean\", default: false },\n      },\n      strict: true,\n    }));\n  } catch (err) {\n    /* v8 ignore next -- parseArgs always throws Error subclasses */\n    const msg = err instanceof Error ? err.message : String(err);\n    logError(`Error: ${msg}\\n\\n${HELP}`);\n    exit(1);\n    return;\n  }\n\n  if (values.help) {\n    log(HELP);\n    exit(0);\n    return;\n  }\n  if (!values.config) {\n    logError(\"Error: --config is required.\\n\\n\" + HELP);\n    exit(1);\n    return;\n  }\n\n  const configPath = resolve(values.config);\n  let config;\n  try {\n    config = loadConfigFn(configPath);\n  } catch (err) {\n    const msg = err instanceof Error ? err.message : String(err);\n    logError(`Failed to load config from ${configPath}: ${msg}`);\n    exit(1);\n    return;\n  }\n\n  const port = values.port ? Number(values.port) : undefined;\n  if (\n    port !== undefined &&\n    (Number.isNaN(port) || !Number.isInteger(port) || port < 0 || port > 65535)\n  ) {\n    logError(`Error: invalid port \"${values.port}\".\\n\\n${HELP}`);\n    exit(1);\n    return;\n  }\n  const host = values.host;\n\n  async function main() {\n    const { llmock, url } = await startFromConfigFn(config!, { port, host });\n    log(`aimock server listening on ${url}`);\n\n    function shutdown() {\n      log(\"Shutting down...\");\n      process.removeListener(\"SIGINT\", shutdown);\n      process.removeListener(\"SIGTERM\", shutdown);\n      llmock.stop().then(\n        () => exit(0),\n        (err) => {\n          logError(`Shutdown error: ${err instanceof Error ? err.message : String(err)}`);\n          exit(1);\n        },\n      );\n    }\n    process.on(\"SIGINT\", shutdown);\n    process.on(\"SIGTERM\", shutdown);\n\n    if (deps.onReady) {\n      deps.onReady({ shutdown });\n    }\n  }\n\n  main().catch((err) => {\n    logError(err instanceof Error ? err.message : String(err));\n    exit(1);\n  });\n}\n\n// Run when executed as a script (not when imported for testing).\n/* v8 ignore start -- entry-point guard, exercised by integration tests */\nconst scriptName = process.argv[1] ?? \"\";\nif (scriptName.endsWith(\"aimock-cli.js\") || scriptName.endsWith(\"aimock-cli.ts\")) {\n  runAimockCli();\n}\n/* v8 ignore stop */\n"],"mappings":";;;;;;AAKA,MAAM,OAAO;;;;;;;;EAQX,MAAM;AAYR,SAAgB,aAAa,OAAsB,EAAE,EAAQ;;CAE3D,MAAM,OAAO,KAAK,QAAQ,QAAQ,KAAK,MAAM,EAAE;CAC/C,MAAM,MAAM,KAAK,OAAO,QAAQ,IAAI,KAAK,QAAQ;CACjD,MAAM,WAAW,KAAK,YAAY,QAAQ,MAAM,KAAK,QAAQ;CAC7D,MAAM,OAAO,KAAK,QAAQ,QAAQ,KAAK,KAAK,QAAQ;CACpD,MAAM,eAAe,KAAK,gBAAgB;CAC1C,MAAM,oBAAoB,KAAK,qBAAqB;CAEpD,IAAI;AACJ,KAAI;AACF,GAAC,CAAE,UAAW,UAAU;GACtB,MAAM;GACN,SAAS;IACP,QAAQ;KAAE,MAAM;KAAU,OAAO;KAAK;IACtC,MAAM;KAAE,MAAM;KAAU,OAAO;KAAK;IACpC,MAAM;KAAE,MAAM;KAAU,OAAO;KAAK;IACpC,MAAM;KAAE,MAAM;KAAW,SAAS;KAAO;IAC1C;GACD,QAAQ;GACT,CAAC;UACK,KAAK;AAGZ,WAAS,UADG,eAAe,QAAQ,IAAI,UAAU,OAAO,IAAI,CACrC,MAAM,OAAO;AACpC,OAAK,EAAE;AACP;;AAGF,KAAI,OAAO,MAAM;AACf,MAAI,KAAK;AACT,OAAK,EAAE;AACP;;AAEF,KAAI,CAAC,OAAO,QAAQ;AAClB,WAAS,qCAAqC,KAAK;AACnD,OAAK,EAAE;AACP;;CAGF,MAAM,aAAa,QAAQ,OAAO,OAAO;CACzC,IAAI;AACJ,KAAI;AACF,WAAS,aAAa,WAAW;UAC1B,KAAK;AAEZ,WAAS,8BAA8B,WAAW,IADtC,eAAe,QAAQ,IAAI,UAAU,OAAO,IAAI,GACA;AAC5D,OAAK,EAAE;AACP;;CAGF,MAAM,OAAO,OAAO,OAAO,OAAO,OAAO,KAAK,GAAG;AACjD,KACE,SAAS,WACR,OAAO,MAAM,KAAK,IAAI,CAAC,OAAO,UAAU,KAAK,IAAI,OAAO,KAAK,OAAO,QACrE;AACA,WAAS,wBAAwB,OAAO,KAAK,QAAQ,OAAO;AAC5D,OAAK,EAAE;AACP;;CAEF,MAAM,OAAO,OAAO;CAEpB,eAAe,OAAO;EACpB,MAAM,EAAE,QAAQ,QAAQ,MAAM,kBAAkB,QAAS;GAAE;GAAM;GAAM,CAAC;AACxE,MAAI,8BAA8B,MAAM;EAExC,SAAS,WAAW;AAClB,OAAI,mBAAmB;AACvB,WAAQ,eAAe,UAAU,SAAS;AAC1C,WAAQ,eAAe,WAAW,SAAS;AAC3C,UAAO,MAAM,CAAC,WACN,KAAK,EAAE,GACZ,QAAQ;AACP,aAAS,mBAAmB,eAAe,QAAQ,IAAI,UAAU,OAAO,IAAI,GAAG;AAC/E,SAAK,EAAE;KAEV;;AAEH,UAAQ,GAAG,UAAU,SAAS;AAC9B,UAAQ,GAAG,WAAW,SAAS;AAE/B,MAAI,KAAK,QACP,MAAK,QAAQ,EAAE,UAAU,CAAC;;AAI9B,OAAM,CAAC,OAAO,QAAQ;AACpB,WAAS,eAAe,QAAQ,IAAI,UAAU,OAAO,IAAI,CAAC;AAC1D,OAAK,EAAE;GACP;;;AAKJ,MAAM,aAAa,QAAQ,KAAK,MAAM;AACtC,IAAI,WAAW,SAAS,gBAAgB,IAAI,WAAW,SAAS,gBAAgB,CAC9E,eAAc"}

package/dist/aws-event-stream.cjs

@@ -0,0 +1,117 @@
+const require_runtime = require('./_virtual/_rolldown/runtime.cjs');
+const require_sse_writer = require('./sse-writer.cjs');
+let node_zlib = require("node:zlib");
+
+//#region src/aws-event-stream.ts
+/**
+* AWS Event Stream binary frame encoder.
+*
+* Implements the AWS binary event stream framing protocol used by Bedrock's
+* streaming (invoke-with-response-stream) endpoint. Each frame carries a set of
+* string headers and a raw-bytes payload, wrapped in a prelude with CRC32
+* checksums for integrity.
+*
+* Binary frame layout:
+*   [total_length: 4B uint32-BE]
+*   [headers_length: 4B uint32-BE]
+*   [prelude_crc32: 4B CRC32 of first 8 bytes]
+*   [headers: variable]
+*   [payload: variable, raw JSON bytes]
+*   [message_crc32: 4B CRC32 of entire frame minus last 4 bytes]
+*/
+function encodeHeaders(headers) {
+	const parts = [];
+	for (const [name, value] of Object.entries(headers)) {
+		const nameBytes = Buffer.from(name, "utf8");
+		const valueBytes = Buffer.from(value, "utf8");
+		const header = Buffer.alloc(1 + nameBytes.length + 1 + 2 + valueBytes.length);
+		let offset = 0;
+		header.writeUInt8(nameBytes.length, offset);
+		offset += 1;
+		nameBytes.copy(header, offset);
+		offset += nameBytes.length;
+		header.writeUInt8(7, offset);
+		offset += 1;
+		header.writeUInt16BE(valueBytes.length, offset);
+		offset += 2;
+		valueBytes.copy(header, offset);
+		parts.push(header);
+	}
+	return Buffer.concat(parts);
+}
+/**
+* Encode a single AWS Event Stream binary frame with the given headers and
+* payload buffer.
+*/
+function encodeEventStreamFrame(headers, payload) {
+	const headersBuffer = encodeHeaders(headers);
+	const headersLength = headersBuffer.length;
+	const totalLength = 12 + headersLength + payload.length + 4;
+	const frame = Buffer.alloc(totalLength);
+	let offset = 0;
+	frame.writeUInt32BE(totalLength, offset);
+	offset += 4;
+	frame.writeUInt32BE(headersLength, offset);
+	offset += 4;
+	const preludeCrc = (0, node_zlib.crc32)(frame.subarray(0, 8));
+	frame.writeUInt32BE(preludeCrc >>> 0, offset);
+	offset += 4;
+	headersBuffer.copy(frame, offset);
+	offset += headersLength;
+	payload.copy(frame, offset);
+	offset += payload.length;
+	const messageCrc = (0, node_zlib.crc32)(frame.subarray(0, totalLength - 4));
+	frame.writeUInt32BE(messageCrc >>> 0, offset);
+	return frame;
+}
+/**
+* Encode an event-stream message with standard AWS headers for a JSON event.
+*
+* Sets `:content-type` = `application/json`, `:event-type` = eventType,
+* `:message-type` = `event`.
+*/
+function encodeEventStreamMessage(eventType, jsonPayload) {
+	return encodeEventStreamFrame({
+		":content-type": "application/json",
+		":event-type": eventType,
+		":message-type": "event"
+	}, Buffer.from(JSON.stringify(jsonPayload), "utf8"));
+}
+/**
+* Write a sequence of event-stream frames to an HTTP response with optional
+* timing control. Mirrors the writeSSEStream pattern from sse-writer.ts.
+*
+* Returns `true` when all events are written (including when the response
+* was already ended before writing began), or `false` if interrupted by
+* the provided abort signal.
+*/
+async function writeEventStream(res, events, options) {
+	const opts = options ?? {};
+	const latency = opts.latency ?? 0;
+	const profile = opts.streamingProfile;
+	const signal = opts.signal;
+	const onChunkSent = opts.onChunkSent;
+	if (res.writableEnded) return true;
+	res.setHeader("Content-Type", "application/vnd.amazon.eventstream");
+	res.setHeader("Transfer-Encoding", "chunked");
+	let chunkIndex = 0;
+	for (const event of events) {
+		const chunkDelay = require_sse_writer.calculateDelay(chunkIndex, profile, latency);
+		if (chunkDelay > 0) await require_sse_writer.delay(chunkDelay, signal);
+		if (signal?.aborted) return false;
+		if (res.writableEnded) return true;
+		const frame = encodeEventStreamMessage(event.eventType, event.payload);
+		res.write(frame);
+		onChunkSent?.();
+		if (signal?.aborted) return false;
+		chunkIndex++;
+	}
+	if (!res.writableEnded) res.end();
+	return true;
+}
+
+//#endregion
+exports.encodeEventStreamFrame = encodeEventStreamFrame;
+exports.encodeEventStreamMessage = encodeEventStreamMessage;
+exports.writeEventStream = writeEventStream;
+//# sourceMappingURL=aws-event-stream.cjs.map

package/dist/aws-event-stream.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"aws-event-stream.cjs","names":["calculateDelay","delay"],"sources":["../src/aws-event-stream.ts"],"sourcesContent":["/**\n * AWS Event Stream binary frame encoder.\n *\n * Implements the AWS binary event stream framing protocol used by Bedrock's\n * streaming (invoke-with-response-stream) endpoint. Each frame carries a set of\n * string headers and a raw-bytes payload, wrapped in a prelude with CRC32\n * checksums for integrity.\n *\n * Binary frame layout:\n *   [total_length: 4B uint32-BE]\n *   [headers_length: 4B uint32-BE]\n *   [prelude_crc32: 4B CRC32 of first 8 bytes]\n *   [headers: variable]\n *   [payload: variable, raw JSON bytes]\n *   [message_crc32: 4B CRC32 of entire frame minus last 4 bytes]\n */\n\nimport { crc32 } from \"node:zlib\";\nimport type * as http from \"node:http\";\nimport type { StreamingProfile } from \"./types.js\";\nimport { delay, calculateDelay } from \"./sse-writer.js\";\n\n// ─── Header encoding ────────────────────────────────────────────────────────\n\nfunction encodeHeaders(headers: Record<string, string>): Buffer {\n  const parts: Buffer[] = [];\n  for (const [name, value] of Object.entries(headers)) {\n    const nameBytes = Buffer.from(name, \"utf8\");\n    const valueBytes = Buffer.from(value, \"utf8\");\n\n    // name_length (1 byte) + name + type (1 byte, 7 = STRING) +\n    // value_length (2 bytes BE) + value\n    const header = Buffer.alloc(1 + nameBytes.length + 1 + 2 + valueBytes.length);\n    let offset = 0;\n    header.writeUInt8(nameBytes.length, offset);\n    offset += 1;\n    nameBytes.copy(header, offset);\n    offset += nameBytes.length;\n    header.writeUInt8(7, offset); // STRING type\n    offset += 1;\n    header.writeUInt16BE(valueBytes.length, offset);\n    offset += 2;\n    valueBytes.copy(header, offset);\n\n    parts.push(header);\n  }\n  return Buffer.concat(parts);\n}\n\n// ─── Frame encoding ─────────────────────────────────────────────────────────\n\n/**\n * Encode a single AWS Event Stream binary frame with the given headers and\n * payload buffer.\n */\nexport function encodeEventStreamFrame(headers: Record<string, string>, payload: Buffer): Buffer {\n  const headersBuffer = encodeHeaders(headers);\n  const headersLength = headersBuffer.length;\n\n  // prelude (8) + prelude_crc (4) + headers + payload + message_crc (4)\n  const totalLength = 4 + 4 + 4 + headersLength + payload.length + 4;\n\n  const frame = Buffer.alloc(totalLength);\n  let offset = 0;\n\n  // Prelude\n  frame.writeUInt32BE(totalLength, offset);\n  offset += 4;\n  frame.writeUInt32BE(headersLength, offset);\n  offset += 4;\n\n  // Prelude CRC32 (covers first 8 bytes)\n  const preludeCrc = crc32(frame.subarray(0, 8));\n  frame.writeUInt32BE(preludeCrc >>> 0, offset);\n  offset += 4;\n\n  // Headers\n  headersBuffer.copy(frame, offset);\n  offset += headersLength;\n\n  // Payload\n  payload.copy(frame, offset);\n  offset += payload.length;\n\n  // Message CRC32 (covers entire frame minus last 4 bytes)\n  const messageCrc = crc32(frame.subarray(0, totalLength - 4));\n  frame.writeUInt32BE(messageCrc >>> 0, offset);\n\n  return frame;\n}\n\n// ─── Convenience wrappers ───────────────────────────────────────────────────\n\n/**\n * Encode an event-stream message with standard AWS headers for a JSON event.\n *\n * Sets `:content-type` = `application/json`, `:event-type` = eventType,\n * `:message-type` = `event`.\n */\nexport function encodeEventStreamMessage(eventType: string, jsonPayload: object): Buffer {\n  const headers: Record<string, string> = {\n    \":content-type\": \"application/json\",\n    \":event-type\": eventType,\n    \":message-type\": \"event\",\n  };\n  const payload = Buffer.from(JSON.stringify(jsonPayload), \"utf8\");\n  return encodeEventStreamFrame(headers, payload);\n}\n\n/**\n * Write a sequence of event-stream frames to an HTTP response with optional\n * timing control. Mirrors the writeSSEStream pattern from sse-writer.ts.\n *\n * Returns `true` when all events are written (including when the response\n * was already ended before writing began), or `false` if interrupted by\n * the provided abort signal.\n */\nexport async function writeEventStream(\n  res: http.ServerResponse,\n  events: Array<{ eventType: string; payload: object }>,\n  options?: {\n    latency?: number;\n    streamingProfile?: StreamingProfile;\n    signal?: AbortSignal;\n    onChunkSent?: () => void;\n  },\n): Promise<boolean> {\n  const opts = options ?? {};\n  const latency = opts.latency ?? 0;\n  const profile = opts.streamingProfile;\n  const signal = opts.signal;\n  const onChunkSent = opts.onChunkSent;\n\n  if (res.writableEnded) return true;\n  res.setHeader(\"Content-Type\", \"application/vnd.amazon.eventstream\");\n  res.setHeader(\"Transfer-Encoding\", \"chunked\");\n\n  let chunkIndex = 0;\n  for (const event of events) {\n    const chunkDelay = calculateDelay(chunkIndex, profile, latency);\n    if (chunkDelay > 0) {\n      await delay(chunkDelay, signal);\n    }\n    if (signal?.aborted) return false;\n    if (res.writableEnded) return true;\n\n    const frame = encodeEventStreamMessage(event.eventType, event.payload);\n    res.write(frame);\n    onChunkSent?.();\n    if (signal?.aborted) return false;\n    chunkIndex++;\n  }\n\n  if (!res.writableEnded) {\n    res.end();\n  }\n  return true;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;AAwBA,SAAS,cAAc,SAAyC;CAC9D,MAAM,QAAkB,EAAE;AAC1B,MAAK,MAAM,CAAC,MAAM,UAAU,OAAO,QAAQ,QAAQ,EAAE;EACnD,MAAM,YAAY,OAAO,KAAK,MAAM,OAAO;EAC3C,MAAM,aAAa,OAAO,KAAK,OAAO,OAAO;EAI7C,MAAM,SAAS,OAAO,MAAM,IAAI,UAAU,SAAS,IAAI,IAAI,WAAW,OAAO;EAC7E,IAAI,SAAS;AACb,SAAO,WAAW,UAAU,QAAQ,OAAO;AAC3C,YAAU;AACV,YAAU,KAAK,QAAQ,OAAO;AAC9B,YAAU,UAAU;AACpB,SAAO,WAAW,GAAG,OAAO;AAC5B,YAAU;AACV,SAAO,cAAc,WAAW,QAAQ,OAAO;AAC/C,YAAU;AACV,aAAW,KAAK,QAAQ,OAAO;AAE/B,QAAM,KAAK,OAAO;;AAEpB,QAAO,OAAO,OAAO,MAAM;;;;;;AAS7B,SAAgB,uBAAuB,SAAiC,SAAyB;CAC/F,MAAM,gBAAgB,cAAc,QAAQ;CAC5C,MAAM,gBAAgB,cAAc;CAGpC,MAAM,cAAc,KAAY,gBAAgB,QAAQ,SAAS;CAEjE,MAAM,QAAQ,OAAO,MAAM,YAAY;CACvC,IAAI,SAAS;AAGb,OAAM,cAAc,aAAa,OAAO;AACxC,WAAU;AACV,OAAM,cAAc,eAAe,OAAO;AAC1C,WAAU;CAGV,MAAM,kCAAmB,MAAM,SAAS,GAAG,EAAE,CAAC;AAC9C,OAAM,cAAc,eAAe,GAAG,OAAO;AAC7C,WAAU;AAGV,eAAc,KAAK,OAAO,OAAO;AACjC,WAAU;AAGV,SAAQ,KAAK,OAAO,OAAO;AAC3B,WAAU,QAAQ;CAGlB,MAAM,kCAAmB,MAAM,SAAS,GAAG,cAAc,EAAE,CAAC;AAC5D,OAAM,cAAc,eAAe,GAAG,OAAO;AAE7C,QAAO;;;;;;;;AAWT,SAAgB,yBAAyB,WAAmB,aAA6B;AAOvF,QAAO,uBANiC;EACtC,iBAAiB;EACjB,eAAe;EACf,iBAAiB;EAClB,EACe,OAAO,KAAK,KAAK,UAAU,YAAY,EAAE,OAAO,CACjB;;;;;;;;;;AAWjD,eAAsB,iBACpB,KACA,QACA,SAMkB;CAClB,MAAM,OAAO,WAAW,EAAE;CAC1B,MAAM,UAAU,KAAK,WAAW;CAChC,MAAM,UAAU,KAAK;CACrB,MAAM,SAAS,KAAK;CACpB,MAAM,cAAc,KAAK;AAEzB,KAAI,IAAI,cAAe,QAAO;AAC9B,KAAI,UAAU,gBAAgB,qCAAqC;AACnE,KAAI,UAAU,qBAAqB,UAAU;CAE7C,IAAI,aAAa;AACjB,MAAK,MAAM,SAAS,QAAQ;EAC1B,MAAM,aAAaA,kCAAe,YAAY,SAAS,QAAQ;AAC/D,MAAI,aAAa,EACf,OAAMC,yBAAM,YAAY,OAAO;AAEjC,MAAI,QAAQ,QAAS,QAAO;AAC5B,MAAI,IAAI,cAAe,QAAO;EAE9B,MAAM,QAAQ,yBAAyB,MAAM,WAAW,MAAM,QAAQ;AACtE,MAAI,MAAM,MAAM;AAChB,iBAAe;AACf,MAAI,QAAQ,QAAS,QAAO;AAC5B;;AAGF,KAAI,CAAC,IAAI,cACP,KAAI,KAAK;AAEX,QAAO"}

package/dist/aws-event-stream.d.cts

@@ -0,0 +1,38 @@
+import { StreamingProfile } from "./types.cjs";
+import * as http from "node:http";
+
+//#region src/aws-event-stream.d.ts
+
+/**
+ * Encode a single AWS Event Stream binary frame with the given headers and
+ * payload buffer.
+ */
+declare function encodeEventStreamFrame(headers: Record<string, string>, payload: Buffer): Buffer;
+/**
+ * Encode an event-stream message with standard AWS headers for a JSON event.
+ *
+ * Sets `:content-type` = `application/json`, `:event-type` = eventType,
+ * `:message-type` = `event`.
+ */
+declare function encodeEventStreamMessage(eventType: string, jsonPayload: object): Buffer;
+/**
+ * Write a sequence of event-stream frames to an HTTP response with optional
+ * timing control. Mirrors the writeSSEStream pattern from sse-writer.ts.
+ *
+ * Returns `true` when all events are written (including when the response
+ * was already ended before writing began), or `false` if interrupted by
+ * the provided abort signal.
+ */
+declare function writeEventStream(res: http.ServerResponse, events: Array<{
+  eventType: string;
+  payload: object;
+}>, options?: {
+  latency?: number;
+  streamingProfile?: StreamingProfile;
+  signal?: AbortSignal;
+  onChunkSent?: () => void;
+}): Promise<boolean>;
+//# sourceMappingURL=aws-event-stream.d.ts.map
+//#endregion
+export { encodeEventStreamFrame, encodeEventStreamMessage, writeEventStream };
+//# sourceMappingURL=aws-event-stream.d.cts.map

package/dist/aws-event-stream.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"aws-event-stream.d.cts","names":[],"sources":["../src/aws-event-stream.ts"],"sourcesContent":[],"mappings":";;;;;;;;;iBAuDgB,sBAAA,UAAgC,iCAAiC,SAAS;;;;;;;iBA4C1E,wBAAA,0CAAkE;;;;;;;;;iBAkB5D,gBAAA,MACf,IAAA,CAAK,wBACF;;;;;qBAGa;WACV;;IAGV"}

package/dist/aws-event-stream.d.ts

@@ -0,0 +1,38 @@
+import { StreamingProfile } from "./types.js";
+import * as http from "node:http";
+
+//#region src/aws-event-stream.d.ts
+
+/**
+ * Encode a single AWS Event Stream binary frame with the given headers and
+ * payload buffer.
+ */
+declare function encodeEventStreamFrame(headers: Record<string, string>, payload: Buffer): Buffer;
+/**
+ * Encode an event-stream message with standard AWS headers for a JSON event.
+ *
+ * Sets `:content-type` = `application/json`, `:event-type` = eventType,
+ * `:message-type` = `event`.
+ */
+declare function encodeEventStreamMessage(eventType: string, jsonPayload: object): Buffer;
+/**
+ * Write a sequence of event-stream frames to an HTTP response with optional
+ * timing control. Mirrors the writeSSEStream pattern from sse-writer.ts.
+ *
+ * Returns `true` when all events are written (including when the response
+ * was already ended before writing began), or `false` if interrupted by
+ * the provided abort signal.
+ */
+declare function writeEventStream(res: http.ServerResponse, events: Array<{
+  eventType: string;
+  payload: object;
+}>, options?: {
+  latency?: number;
+  streamingProfile?: StreamingProfile;
+  signal?: AbortSignal;
+  onChunkSent?: () => void;
+}): Promise<boolean>;
+//# sourceMappingURL=aws-event-stream.d.ts.map
+//#endregion
+export { encodeEventStreamFrame, encodeEventStreamMessage, writeEventStream };
+//# sourceMappingURL=aws-event-stream.d.ts.map

package/dist/aws-event-stream.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"aws-event-stream.d.ts","names":[],"sources":["../src/aws-event-stream.ts"],"sourcesContent":[],"mappings":";;;;;;;;;iBAuDgB,sBAAA,UAAgC,iCAAiC,SAAS;;;;;;;iBA4C1E,wBAAA,0CAAkE;;;;;;;;;iBAkB5D,gBAAA,MACf,IAAA,CAAK,wBACF;;;;;qBAGa;WACV;;IAGV"}

package/dist/aws-event-stream.js

@@ -0,0 +1,114 @@
+import { calculateDelay, delay } from "./sse-writer.js";
+import { crc32 } from "node:zlib";
+
+//#region src/aws-event-stream.ts
+/**
+* AWS Event Stream binary frame encoder.
+*
+* Implements the AWS binary event stream framing protocol used by Bedrock's
+* streaming (invoke-with-response-stream) endpoint. Each frame carries a set of
+* string headers and a raw-bytes payload, wrapped in a prelude with CRC32
+* checksums for integrity.
+*
+* Binary frame layout:
+*   [total_length: 4B uint32-BE]
+*   [headers_length: 4B uint32-BE]
+*   [prelude_crc32: 4B CRC32 of first 8 bytes]
+*   [headers: variable]
+*   [payload: variable, raw JSON bytes]
+*   [message_crc32: 4B CRC32 of entire frame minus last 4 bytes]
+*/
+function encodeHeaders(headers) {
+	const parts = [];
+	for (const [name, value] of Object.entries(headers)) {
+		const nameBytes = Buffer.from(name, "utf8");
+		const valueBytes = Buffer.from(value, "utf8");
+		const header = Buffer.alloc(1 + nameBytes.length + 1 + 2 + valueBytes.length);
+		let offset = 0;
+		header.writeUInt8(nameBytes.length, offset);
+		offset += 1;
+		nameBytes.copy(header, offset);
+		offset += nameBytes.length;
+		header.writeUInt8(7, offset);
+		offset += 1;
+		header.writeUInt16BE(valueBytes.length, offset);
+		offset += 2;
+		valueBytes.copy(header, offset);
+		parts.push(header);
+	}
+	return Buffer.concat(parts);
+}
+/**
+* Encode a single AWS Event Stream binary frame with the given headers and
+* payload buffer.
+*/
+function encodeEventStreamFrame(headers, payload) {
+	const headersBuffer = encodeHeaders(headers);
+	const headersLength = headersBuffer.length;
+	const totalLength = 12 + headersLength + payload.length + 4;
+	const frame = Buffer.alloc(totalLength);
+	let offset = 0;
+	frame.writeUInt32BE(totalLength, offset);
+	offset += 4;
+	frame.writeUInt32BE(headersLength, offset);
+	offset += 4;
+	const preludeCrc = crc32(frame.subarray(0, 8));
+	frame.writeUInt32BE(preludeCrc >>> 0, offset);
+	offset += 4;
+	headersBuffer.copy(frame, offset);
+	offset += headersLength;
+	payload.copy(frame, offset);
+	offset += payload.length;
+	const messageCrc = crc32(frame.subarray(0, totalLength - 4));
+	frame.writeUInt32BE(messageCrc >>> 0, offset);
+	return frame;
+}
+/**
+* Encode an event-stream message with standard AWS headers for a JSON event.
+*
+* Sets `:content-type` = `application/json`, `:event-type` = eventType,
+* `:message-type` = `event`.
+*/
+function encodeEventStreamMessage(eventType, jsonPayload) {
+	return encodeEventStreamFrame({
+		":content-type": "application/json",
+		":event-type": eventType,
+		":message-type": "event"
+	}, Buffer.from(JSON.stringify(jsonPayload), "utf8"));
+}
+/**
+* Write a sequence of event-stream frames to an HTTP response with optional
+* timing control. Mirrors the writeSSEStream pattern from sse-writer.ts.
+*
+* Returns `true` when all events are written (including when the response
+* was already ended before writing began), or `false` if interrupted by
+* the provided abort signal.
+*/
+async function writeEventStream(res, events, options) {
+	const opts = options ?? {};
+	const latency = opts.latency ?? 0;
+	const profile = opts.streamingProfile;
+	const signal = opts.signal;
+	const onChunkSent = opts.onChunkSent;
+	if (res.writableEnded) return true;
+	res.setHeader("Content-Type", "application/vnd.amazon.eventstream");
+	res.setHeader("Transfer-Encoding", "chunked");
+	let chunkIndex = 0;
+	for (const event of events) {
+		const chunkDelay = calculateDelay(chunkIndex, profile, latency);
+		if (chunkDelay > 0) await delay(chunkDelay, signal);
+		if (signal?.aborted) return false;
+		if (res.writableEnded) return true;
+		const frame = encodeEventStreamMessage(event.eventType, event.payload);
+		res.write(frame);
+		onChunkSent?.();
+		if (signal?.aborted) return false;
+		chunkIndex++;
+	}
+	if (!res.writableEnded) res.end();
+	return true;
+}
+
+//#endregion
+export { encodeEventStreamFrame, encodeEventStreamMessage, writeEventStream };
+//# sourceMappingURL=aws-event-stream.js.map

package/dist/aws-event-stream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"aws-event-stream.js","names":[],"sources":["../src/aws-event-stream.ts"],"sourcesContent":["/**\n * AWS Event Stream binary frame encoder.\n *\n * Implements the AWS binary event stream framing protocol used by Bedrock's\n * streaming (invoke-with-response-stream) endpoint. Each frame carries a set of\n * string headers and a raw-bytes payload, wrapped in a prelude with CRC32\n * checksums for integrity.\n *\n * Binary frame layout:\n *   [total_length: 4B uint32-BE]\n *   [headers_length: 4B uint32-BE]\n *   [prelude_crc32: 4B CRC32 of first 8 bytes]\n *   [headers: variable]\n *   [payload: variable, raw JSON bytes]\n *   [message_crc32: 4B CRC32 of entire frame minus last 4 bytes]\n */\n\nimport { crc32 } from \"node:zlib\";\nimport type * as http from \"node:http\";\nimport type { StreamingProfile } from \"./types.js\";\nimport { delay, calculateDelay } from \"./sse-writer.js\";\n\n// ─── Header encoding ────────────────────────────────────────────────────────\n\nfunction encodeHeaders(headers: Record<string, string>): Buffer {\n  const parts: Buffer[] = [];\n  for (const [name, value] of Object.entries(headers)) {\n    const nameBytes = Buffer.from(name, \"utf8\");\n    const valueBytes = Buffer.from(value, \"utf8\");\n\n    // name_length (1 byte) + name + type (1 byte, 7 = STRING) +\n    // value_length (2 bytes BE) + value\n    const header = Buffer.alloc(1 + nameBytes.length + 1 + 2 + valueBytes.length);\n    let offset = 0;\n    header.writeUInt8(nameBytes.length, offset);\n    offset += 1;\n    nameBytes.copy(header, offset);\n    offset += nameBytes.length;\n    header.writeUInt8(7, offset); // STRING type\n    offset += 1;\n    header.writeUInt16BE(valueBytes.length, offset);\n    offset += 2;\n    valueBytes.copy(header, offset);\n\n    parts.push(header);\n  }\n  return Buffer.concat(parts);\n}\n\n// ─── Frame encoding ─────────────────────────────────────────────────────────\n\n/**\n * Encode a single AWS Event Stream binary frame with the given headers and\n * payload buffer.\n */\nexport function encodeEventStreamFrame(headers: Record<string, string>, payload: Buffer): Buffer {\n  const headersBuffer = encodeHeaders(headers);\n  const headersLength = headersBuffer.length;\n\n  // prelude (8) + prelude_crc (4) + headers + payload + message_crc (4)\n  const totalLength = 4 + 4 + 4 + headersLength + payload.length + 4;\n\n  const frame = Buffer.alloc(totalLength);\n  let offset = 0;\n\n  // Prelude\n  frame.writeUInt32BE(totalLength, offset);\n  offset += 4;\n  frame.writeUInt32BE(headersLength, offset);\n  offset += 4;\n\n  // Prelude CRC32 (covers first 8 bytes)\n  const preludeCrc = crc32(frame.subarray(0, 8));\n  frame.writeUInt32BE(preludeCrc >>> 0, offset);\n  offset += 4;\n\n  // Headers\n  headersBuffer.copy(frame, offset);\n  offset += headersLength;\n\n  // Payload\n  payload.copy(frame, offset);\n  offset += payload.length;\n\n  // Message CRC32 (covers entire frame minus last 4 bytes)\n  const messageCrc = crc32(frame.subarray(0, totalLength - 4));\n  frame.writeUInt32BE(messageCrc >>> 0, offset);\n\n  return frame;\n}\n\n// ─── Convenience wrappers ───────────────────────────────────────────────────\n\n/**\n * Encode an event-stream message with standard AWS headers for a JSON event.\n *\n * Sets `:content-type` = `application/json`, `:event-type` = eventType,\n * `:message-type` = `event`.\n */\nexport function encodeEventStreamMessage(eventType: string, jsonPayload: object): Buffer {\n  const headers: Record<string, string> = {\n    \":content-type\": \"application/json\",\n    \":event-type\": eventType,\n    \":message-type\": \"event\",\n  };\n  const payload = Buffer.from(JSON.stringify(jsonPayload), \"utf8\");\n  return encodeEventStreamFrame(headers, payload);\n}\n\n/**\n * Write a sequence of event-stream frames to an HTTP response with optional\n * timing control. Mirrors the writeSSEStream pattern from sse-writer.ts.\n *\n * Returns `true` when all events are written (including when the response\n * was already ended before writing began), or `false` if interrupted by\n * the provided abort signal.\n */\nexport async function writeEventStream(\n  res: http.ServerResponse,\n  events: Array<{ eventType: string; payload: object }>,\n  options?: {\n    latency?: number;\n    streamingProfile?: StreamingProfile;\n    signal?: AbortSignal;\n    onChunkSent?: () => void;\n  },\n): Promise<boolean> {\n  const opts = options ?? {};\n  const latency = opts.latency ?? 0;\n  const profile = opts.streamingProfile;\n  const signal = opts.signal;\n  const onChunkSent = opts.onChunkSent;\n\n  if (res.writableEnded) return true;\n  res.setHeader(\"Content-Type\", \"application/vnd.amazon.eventstream\");\n  res.setHeader(\"Transfer-Encoding\", \"chunked\");\n\n  let chunkIndex = 0;\n  for (const event of events) {\n    const chunkDelay = calculateDelay(chunkIndex, profile, latency);\n    if (chunkDelay > 0) {\n      await delay(chunkDelay, signal);\n    }\n    if (signal?.aborted) return false;\n    if (res.writableEnded) return true;\n\n    const frame = encodeEventStreamMessage(event.eventType, event.payload);\n    res.write(frame);\n    onChunkSent?.();\n    if (signal?.aborted) return false;\n    chunkIndex++;\n  }\n\n  if (!res.writableEnded) {\n    res.end();\n  }\n  return true;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAwBA,SAAS,cAAc,SAAyC;CAC9D,MAAM,QAAkB,EAAE;AAC1B,MAAK,MAAM,CAAC,MAAM,UAAU,OAAO,QAAQ,QAAQ,EAAE;EACnD,MAAM,YAAY,OAAO,KAAK,MAAM,OAAO;EAC3C,MAAM,aAAa,OAAO,KAAK,OAAO,OAAO;EAI7C,MAAM,SAAS,OAAO,MAAM,IAAI,UAAU,SAAS,IAAI,IAAI,WAAW,OAAO;EAC7E,IAAI,SAAS;AACb,SAAO,WAAW,UAAU,QAAQ,OAAO;AAC3C,YAAU;AACV,YAAU,KAAK,QAAQ,OAAO;AAC9B,YAAU,UAAU;AACpB,SAAO,WAAW,GAAG,OAAO;AAC5B,YAAU;AACV,SAAO,cAAc,WAAW,QAAQ,OAAO;AAC/C,YAAU;AACV,aAAW,KAAK,QAAQ,OAAO;AAE/B,QAAM,KAAK,OAAO;;AAEpB,QAAO,OAAO,OAAO,MAAM;;;;;;AAS7B,SAAgB,uBAAuB,SAAiC,SAAyB;CAC/F,MAAM,gBAAgB,cAAc,QAAQ;CAC5C,MAAM,gBAAgB,cAAc;CAGpC,MAAM,cAAc,KAAY,gBAAgB,QAAQ,SAAS;CAEjE,MAAM,QAAQ,OAAO,MAAM,YAAY;CACvC,IAAI,SAAS;AAGb,OAAM,cAAc,aAAa,OAAO;AACxC,WAAU;AACV,OAAM,cAAc,eAAe,OAAO;AAC1C,WAAU;CAGV,MAAM,aAAa,MAAM,MAAM,SAAS,GAAG,EAAE,CAAC;AAC9C,OAAM,cAAc,eAAe,GAAG,OAAO;AAC7C,WAAU;AAGV,eAAc,KAAK,OAAO,OAAO;AACjC,WAAU;AAGV,SAAQ,KAAK,OAAO,OAAO;AAC3B,WAAU,QAAQ;CAGlB,MAAM,aAAa,MAAM,MAAM,SAAS,GAAG,cAAc,EAAE,CAAC;AAC5D,OAAM,cAAc,eAAe,GAAG,OAAO;AAE7C,QAAO;;;;;;;;AAWT,SAAgB,yBAAyB,WAAmB,aAA6B;AAOvF,QAAO,uBANiC;EACtC,iBAAiB;EACjB,eAAe;EACf,iBAAiB;EAClB,EACe,OAAO,KAAK,KAAK,UAAU,YAAY,EAAE,OAAO,CACjB;;;;;;;;;;AAWjD,eAAsB,iBACpB,KACA,QACA,SAMkB;CAClB,MAAM,OAAO,WAAW,EAAE;CAC1B,MAAM,UAAU,KAAK,WAAW;CAChC,MAAM,UAAU,KAAK;CACrB,MAAM,SAAS,KAAK;CACpB,MAAM,cAAc,KAAK;AAEzB,KAAI,IAAI,cAAe,QAAO;AAC9B,KAAI,UAAU,gBAAgB,qCAAqC;AACnE,KAAI,UAAU,qBAAqB,UAAU;CAE7C,IAAI,aAAa;AACjB,MAAK,MAAM,SAAS,QAAQ;EAC1B,MAAM,aAAa,eAAe,YAAY,SAAS,QAAQ;AAC/D,MAAI,aAAa,EACf,OAAM,MAAM,YAAY,OAAO;AAEjC,MAAI,QAAQ,QAAS,QAAO;AAC5B,MAAI,IAAI,cAAe,QAAO;EAE9B,MAAM,QAAQ,yBAAyB,MAAM,WAAW,MAAM,QAAQ;AACtE,MAAI,MAAM,MAAM;AAChB,iBAAe;AACf,MAAI,QAAQ,QAAS,QAAO;AAC5B;;AAGF,KAAI,CAAC,IAAI,cACP,KAAI,KAAK;AAEX,QAAO"}