@copilotkit/aimock 1.10.0 → 1.11.0

package/dist/agui-mock.js

@@ -0,0 +1,188 @@
+import { flattenHeaders, readBody } from "./helpers.js";
+import { Logger } from "./logger.js";
+import { buildReasoningResponse, buildStateUpdate, buildTextResponse, buildToolCallResponse, findFixture, writeAGUIEventStream } from "./agui-handler.js";
+import { proxyAndRecordAGUI } from "./agui-recorder.js";
+import * as http from "node:http";
+
+//#region src/agui-mock.ts
+var AGUIMock = class {
+	fixtures = [];
+	server = null;
+	journal = null;
+	registry = null;
+	options;
+	baseUrl = "";
+	recordConfig;
+	logger;
+	constructor(options) {
+		this.options = options ?? {};
+		this.logger = new Logger("silent");
+	}
+	addFixture(fixture) {
+		this.fixtures.push(fixture);
+		return this;
+	}
+	onMessage(pattern, text, opts) {
+		const events = buildTextResponse(text);
+		this.fixtures.push({
+			match: { message: pattern },
+			events,
+			delayMs: opts?.delayMs
+		});
+		return this;
+	}
+	onRun(pattern, events, delayMs) {
+		this.fixtures.push({
+			match: { message: pattern },
+			events,
+			delayMs
+		});
+		return this;
+	}
+	onToolCall(pattern, toolName, args, opts) {
+		const events = buildToolCallResponse(toolName, args, { result: opts?.result });
+		this.fixtures.push({
+			match: { message: pattern },
+			events,
+			delayMs: opts?.delayMs
+		});
+		return this;
+	}
+	onStateKey(key, snapshot, delayMs) {
+		const events = buildStateUpdate(snapshot);
+		this.fixtures.push({
+			match: { stateKey: key },
+			events,
+			delayMs
+		});
+		return this;
+	}
+	onReasoning(pattern, text, opts) {
+		const events = buildReasoningResponse(text);
+		this.fixtures.push({
+			match: { message: pattern },
+			events,
+			delayMs: opts?.delayMs
+		});
+		return this;
+	}
+	onPredicate(predicate, events, delayMs) {
+		this.fixtures.push({
+			match: { predicate },
+			events,
+			delayMs
+		});
+		return this;
+	}
+	enableRecording(config) {
+		this.recordConfig = config;
+		return this;
+	}
+	reset() {
+		this.fixtures = [];
+		this.recordConfig = void 0;
+		return this;
+	}
+	async handleRequest(req, res, pathname) {
+		if (req.method !== "POST" || pathname !== "/" && pathname !== "") return false;
+		if (this.registry) this.registry.incrementCounter("aimock_agui_requests_total", { method: "POST" });
+		const body = await readBody(req);
+		let input;
+		try {
+			input = JSON.parse(body);
+		} catch {
+			res.writeHead(400, { "Content-Type": "application/json" });
+			res.end(JSON.stringify({ error: "Invalid JSON body" }));
+			this.journalRequest(req, pathname, 400);
+			return true;
+		}
+		const fixture = findFixture(input, this.fixtures);
+		if (fixture) {
+			await writeAGUIEventStream(res, fixture.events, { delayMs: fixture.delayMs });
+			this.journalRequest(req, pathname, 200);
+			return true;
+		}
+		if (this.recordConfig) {
+			if (await proxyAndRecordAGUI(req, res, input, this.fixtures, this.recordConfig, this.logger)) {
+				this.journalRequest(req, pathname, 200);
+				return true;
+			}
+		}
+		res.writeHead(404, { "Content-Type": "application/json" });
+		res.end(JSON.stringify({ error: "No matching AG-UI fixture" }));
+		this.journalRequest(req, pathname, 404);
+		return true;
+	}
+	health() {
+		return {
+			status: "ok",
+			fixtures: this.fixtures.length
+		};
+	}
+	setJournal(journal) {
+		this.journal = journal;
+	}
+	setBaseUrl(url) {
+		this.baseUrl = url;
+	}
+	setRegistry(registry) {
+		this.registry = registry;
+	}
+	async start() {
+		if (this.server) throw new Error("AGUIMock server already started");
+		const host = this.options.host ?? "127.0.0.1";
+		const port = this.options.port ?? 0;
+		return new Promise((resolve, reject) => {
+			const srv = http.createServer(async (req, res) => {
+				const url = new URL(req.url ?? "/", `http://${req.headers.host ?? "localhost"}`);
+				if (!await this.handleRequest(req, res, url.pathname).catch((err) => {
+					this.logger.error(`AGUIMock request error: ${err instanceof Error ? err.message : err}`);
+					if (!res.headersSent) {
+						res.writeHead(500);
+						res.end("Internal server error");
+					} else if (!res.writableEnded) res.end();
+					return true;
+				}) && !res.headersSent) {
+					res.writeHead(404, { "Content-Type": "application/json" });
+					res.end(JSON.stringify({ error: "Not found" }));
+				}
+			});
+			srv.on("error", reject);
+			srv.listen(port, host, () => {
+				const addr = srv.address();
+				if (typeof addr === "object" && addr !== null) this.baseUrl = `http://${host}:${addr.port}`;
+				this.server = srv;
+				resolve(this.baseUrl);
+			});
+		});
+	}
+	async stop() {
+		if (!this.server) throw new Error("AGUIMock server not started");
+		const srv = this.server;
+		await new Promise((resolve, reject) => {
+			srv.close((err) => err ? reject(err) : resolve());
+		});
+		this.server = null;
+	}
+	get url() {
+		if (!this.server) throw new Error("AGUIMock server not started");
+		return this.baseUrl;
+	}
+	journalRequest(req, pathname, status) {
+		if (this.journal) this.journal.add({
+			method: req.method ?? "POST",
+			path: req.url ?? pathname,
+			headers: flattenHeaders(req.headers),
+			body: null,
+			service: "agui",
+			response: {
+				status,
+				fixture: null
+			}
+		});
+	}
+};
+
+//#endregion
+export { AGUIMock };
+//# sourceMappingURL=agui-mock.js.map

package/dist/agui-mock.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agui-mock.js","names":[],"sources":["../src/agui-mock.ts"],"sourcesContent":["import * as http from \"node:http\";\nimport type { Mountable } from \"./types.js\";\nimport type { Journal } from \"./journal.js\";\nimport type { MetricsRegistry } from \"./metrics.js\";\nimport type {\n  AGUIFixture,\n  AGUIMockOptions,\n  AGUIRecordConfig,\n  AGUIEvent,\n  AGUIRunAgentInput,\n} from \"./agui-types.js\";\nimport {\n  findFixture,\n  buildTextResponse,\n  buildToolCallResponse,\n  buildStateUpdate,\n  buildReasoningResponse,\n  writeAGUIEventStream,\n} from \"./agui-handler.js\";\nimport { flattenHeaders, readBody } from \"./helpers.js\";\nimport { proxyAndRecordAGUI } from \"./agui-recorder.js\";\nimport { Logger } from \"./logger.js\";\n\nexport class AGUIMock implements Mountable {\n  private fixtures: AGUIFixture[] = [];\n  private server: http.Server | null = null;\n  private journal: Journal | null = null;\n  private registry: MetricsRegistry | null = null;\n  private options: AGUIMockOptions;\n  private baseUrl = \"\";\n  private recordConfig: AGUIRecordConfig | undefined;\n  private logger: Logger;\n\n  constructor(options?: AGUIMockOptions) {\n    this.options = options ?? {};\n    this.logger = new Logger(\"silent\");\n  }\n\n  // ---- Fluent registration API ----\n\n  addFixture(fixture: AGUIFixture): this {\n    this.fixtures.push(fixture);\n    return this;\n  }\n\n  onMessage(pattern: string | RegExp, text: string, opts?: { delayMs?: number }): this {\n    const events = buildTextResponse(text);\n    this.fixtures.push({\n      match: { message: pattern },\n      events,\n      delayMs: opts?.delayMs,\n    });\n    return this;\n  }\n\n  onRun(pattern: string | RegExp, events: AGUIEvent[], delayMs?: number): this {\n    this.fixtures.push({\n      match: { message: pattern },\n      events,\n      delayMs,\n    });\n    return this;\n  }\n\n  onToolCall(\n    pattern: string | RegExp,\n    toolName: string,\n    args: string,\n    opts?: { result?: string; delayMs?: number },\n  ): this {\n    const events = buildToolCallResponse(toolName, args, {\n      result: opts?.result,\n    });\n    this.fixtures.push({\n      match: { message: pattern },\n      events,\n      delayMs: opts?.delayMs,\n    });\n    return this;\n  }\n\n  onStateKey(key: string, snapshot: Record<string, unknown>, delayMs?: number): this {\n    const events = buildStateUpdate(snapshot);\n    this.fixtures.push({\n      match: { stateKey: key },\n      events,\n      delayMs,\n    });\n    return this;\n  }\n\n  onReasoning(pattern: string | RegExp, text: string, opts?: { delayMs?: number }): this {\n    const events = buildReasoningResponse(text);\n    this.fixtures.push({\n      match: { message: pattern },\n      events,\n      delayMs: opts?.delayMs,\n    });\n    return this;\n  }\n\n  onPredicate(\n    predicate: (input: AGUIRunAgentInput) => boolean,\n    events: AGUIEvent[],\n    delayMs?: number,\n  ): this {\n    this.fixtures.push({\n      match: { predicate },\n      events,\n      delayMs,\n    });\n    return this;\n  }\n\n  enableRecording(config: AGUIRecordConfig): this {\n    this.recordConfig = config;\n    return this;\n  }\n\n  reset(): this {\n    this.fixtures = [];\n    this.recordConfig = undefined;\n    return this;\n  }\n\n  // ---- Mountable interface ----\n\n  async handleRequest(\n    req: http.IncomingMessage,\n    res: http.ServerResponse,\n    pathname: string,\n  ): Promise<boolean> {\n    if (req.method !== \"POST\" || (pathname !== \"/\" && pathname !== \"\")) {\n      return false;\n    }\n\n    if (this.registry) {\n      this.registry.incrementCounter(\"aimock_agui_requests_total\", { method: \"POST\" });\n    }\n\n    const body = await readBody(req);\n\n    let input: AGUIRunAgentInput;\n    try {\n      input = JSON.parse(body) as AGUIRunAgentInput;\n    } catch {\n      res.writeHead(400, { \"Content-Type\": \"application/json\" });\n      res.end(JSON.stringify({ error: \"Invalid JSON body\" }));\n      this.journalRequest(req, pathname, 400);\n      return true;\n    }\n\n    const fixture = findFixture(input, this.fixtures);\n\n    if (fixture) {\n      await writeAGUIEventStream(res, fixture.events, { delayMs: fixture.delayMs });\n      this.journalRequest(req, pathname, 200);\n      return true;\n    }\n\n    // No match — if recording is enabled, proxy to upstream\n    if (this.recordConfig) {\n      const proxied = await proxyAndRecordAGUI(\n        req,\n        res,\n        input,\n        this.fixtures,\n        this.recordConfig,\n        this.logger,\n      );\n      if (proxied) {\n        this.journalRequest(req, pathname, 200);\n        return true;\n      }\n    }\n\n    // No match, no recorder — 404\n    res.writeHead(404, { \"Content-Type\": \"application/json\" });\n    res.end(JSON.stringify({ error: \"No matching AG-UI fixture\" }));\n    this.journalRequest(req, pathname, 404);\n    return true;\n  }\n\n  health(): { status: string; fixtures: number } {\n    return {\n      status: \"ok\",\n      fixtures: this.fixtures.length,\n    };\n  }\n\n  setJournal(journal: Journal): void {\n    this.journal = journal;\n  }\n\n  setBaseUrl(url: string): void {\n    this.baseUrl = url;\n  }\n\n  setRegistry(registry: MetricsRegistry): void {\n    this.registry = registry;\n  }\n\n  // ---- Standalone mode ----\n\n  async start(): Promise<string> {\n    if (this.server) {\n      throw new Error(\"AGUIMock server already started\");\n    }\n\n    const host = this.options.host ?? \"127.0.0.1\";\n    const port = this.options.port ?? 0;\n\n    return new Promise<string>((resolve, reject) => {\n      const srv = http.createServer(async (req, res) => {\n        const url = new URL(req.url ?? \"/\", `http://${req.headers.host ?? \"localhost\"}`);\n        const handled = await this.handleRequest(req, res, url.pathname).catch((err) => {\n          this.logger.error(`AGUIMock request error: ${err instanceof Error ? err.message : err}`);\n          if (!res.headersSent) {\n            res.writeHead(500);\n            res.end(\"Internal server error\");\n          } else if (!res.writableEnded) {\n            res.end();\n          }\n          return true;\n        });\n        if (!handled && !res.headersSent) {\n          res.writeHead(404, { \"Content-Type\": \"application/json\" });\n          res.end(JSON.stringify({ error: \"Not found\" }));\n        }\n      });\n\n      srv.on(\"error\", reject);\n\n      srv.listen(port, host, () => {\n        const addr = srv.address();\n        if (typeof addr === \"object\" && addr !== null) {\n          this.baseUrl = `http://${host}:${addr.port}`;\n        }\n        this.server = srv;\n        resolve(this.baseUrl);\n      });\n    });\n  }\n\n  async stop(): Promise<void> {\n    if (!this.server) {\n      throw new Error(\"AGUIMock server not started\");\n    }\n    const srv = this.server;\n    await new Promise<void>((resolve, reject) => {\n      srv.close((err: Error | undefined) => (err ? reject(err) : resolve()));\n    });\n    this.server = null;\n  }\n\n  get url(): string {\n    if (!this.server) {\n      throw new Error(\"AGUIMock server not started\");\n    }\n    return this.baseUrl;\n  }\n\n  // ---- Private helpers ----\n\n  private journalRequest(req: http.IncomingMessage, pathname: string, status: number): void {\n    if (this.journal) {\n      this.journal.add({\n        method: req.method ?? \"POST\",\n        path: req.url ?? pathname,\n        headers: flattenHeaders(req.headers),\n        body: null,\n        service: \"agui\",\n        response: { status, fixture: null },\n      });\n    }\n  }\n}\n"],"mappings":";;;;;;;AAuBA,IAAa,WAAb,MAA2C;CACzC,AAAQ,WAA0B,EAAE;CACpC,AAAQ,SAA6B;CACrC,AAAQ,UAA0B;CAClC,AAAQ,WAAmC;CAC3C,AAAQ;CACR,AAAQ,UAAU;CAClB,AAAQ;CACR,AAAQ;CAER,YAAY,SAA2B;AACrC,OAAK,UAAU,WAAW,EAAE;AAC5B,OAAK,SAAS,IAAI,OAAO,SAAS;;CAKpC,WAAW,SAA4B;AACrC,OAAK,SAAS,KAAK,QAAQ;AAC3B,SAAO;;CAGT,UAAU,SAA0B,MAAc,MAAmC;EACnF,MAAM,SAAS,kBAAkB,KAAK;AACtC,OAAK,SAAS,KAAK;GACjB,OAAO,EAAE,SAAS,SAAS;GAC3B;GACA,SAAS,MAAM;GAChB,CAAC;AACF,SAAO;;CAGT,MAAM,SAA0B,QAAqB,SAAwB;AAC3E,OAAK,SAAS,KAAK;GACjB,OAAO,EAAE,SAAS,SAAS;GAC3B;GACA;GACD,CAAC;AACF,SAAO;;CAGT,WACE,SACA,UACA,MACA,MACM;EACN,MAAM,SAAS,sBAAsB,UAAU,MAAM,EACnD,QAAQ,MAAM,QACf,CAAC;AACF,OAAK,SAAS,KAAK;GACjB,OAAO,EAAE,SAAS,SAAS;GAC3B;GACA,SAAS,MAAM;GAChB,CAAC;AACF,SAAO;;CAGT,WAAW,KAAa,UAAmC,SAAwB;EACjF,MAAM,SAAS,iBAAiB,SAAS;AACzC,OAAK,SAAS,KAAK;GACjB,OAAO,EAAE,UAAU,KAAK;GACxB;GACA;GACD,CAAC;AACF,SAAO;;CAGT,YAAY,SAA0B,MAAc,MAAmC;EACrF,MAAM,SAAS,uBAAuB,KAAK;AAC3C,OAAK,SAAS,KAAK;GACjB,OAAO,EAAE,SAAS,SAAS;GAC3B;GACA,SAAS,MAAM;GAChB,CAAC;AACF,SAAO;;CAGT,YACE,WACA,QACA,SACM;AACN,OAAK,SAAS,KAAK;GACjB,OAAO,EAAE,WAAW;GACpB;GACA;GACD,CAAC;AACF,SAAO;;CAGT,gBAAgB,QAAgC;AAC9C,OAAK,eAAe;AACpB,SAAO;;CAGT,QAAc;AACZ,OAAK,WAAW,EAAE;AAClB,OAAK,eAAe;AACpB,SAAO;;CAKT,MAAM,cACJ,KACA,KACA,UACkB;AAClB,MAAI,IAAI,WAAW,UAAW,aAAa,OAAO,aAAa,GAC7D,QAAO;AAGT,MAAI,KAAK,SACP,MAAK,SAAS,iBAAiB,8BAA8B,EAAE,QAAQ,QAAQ,CAAC;EAGlF,MAAM,OAAO,MAAM,SAAS,IAAI;EAEhC,IAAI;AACJ,MAAI;AACF,WAAQ,KAAK,MAAM,KAAK;UAClB;AACN,OAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,OAAI,IAAI,KAAK,UAAU,EAAE,OAAO,qBAAqB,CAAC,CAAC;AACvD,QAAK,eAAe,KAAK,UAAU,IAAI;AACvC,UAAO;;EAGT,MAAM,UAAU,YAAY,OAAO,KAAK,SAAS;AAEjD,MAAI,SAAS;AACX,SAAM,qBAAqB,KAAK,QAAQ,QAAQ,EAAE,SAAS,QAAQ,SAAS,CAAC;AAC7E,QAAK,eAAe,KAAK,UAAU,IAAI;AACvC,UAAO;;AAIT,MAAI,KAAK,cASP;OARgB,MAAM,mBACpB,KACA,KACA,OACA,KAAK,UACL,KAAK,cACL,KAAK,OACN,EACY;AACX,SAAK,eAAe,KAAK,UAAU,IAAI;AACvC,WAAO;;;AAKX,MAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,MAAI,IAAI,KAAK,UAAU,EAAE,OAAO,6BAA6B,CAAC,CAAC;AAC/D,OAAK,eAAe,KAAK,UAAU,IAAI;AACvC,SAAO;;CAGT,SAA+C;AAC7C,SAAO;GACL,QAAQ;GACR,UAAU,KAAK,SAAS;GACzB;;CAGH,WAAW,SAAwB;AACjC,OAAK,UAAU;;CAGjB,WAAW,KAAmB;AAC5B,OAAK,UAAU;;CAGjB,YAAY,UAAiC;AAC3C,OAAK,WAAW;;CAKlB,MAAM,QAAyB;AAC7B,MAAI,KAAK,OACP,OAAM,IAAI,MAAM,kCAAkC;EAGpD,MAAM,OAAO,KAAK,QAAQ,QAAQ;EAClC,MAAM,OAAO,KAAK,QAAQ,QAAQ;AAElC,SAAO,IAAI,SAAiB,SAAS,WAAW;GAC9C,MAAM,MAAM,KAAK,aAAa,OAAO,KAAK,QAAQ;IAChD,MAAM,MAAM,IAAI,IAAI,IAAI,OAAO,KAAK,UAAU,IAAI,QAAQ,QAAQ,cAAc;AAWhF,QAAI,CAVY,MAAM,KAAK,cAAc,KAAK,KAAK,IAAI,SAAS,CAAC,OAAO,QAAQ;AAC9E,UAAK,OAAO,MAAM,2BAA2B,eAAe,QAAQ,IAAI,UAAU,MAAM;AACxF,SAAI,CAAC,IAAI,aAAa;AACpB,UAAI,UAAU,IAAI;AAClB,UAAI,IAAI,wBAAwB;gBACvB,CAAC,IAAI,cACd,KAAI,KAAK;AAEX,YAAO;MACP,IACc,CAAC,IAAI,aAAa;AAChC,SAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,SAAI,IAAI,KAAK,UAAU,EAAE,OAAO,aAAa,CAAC,CAAC;;KAEjD;AAEF,OAAI,GAAG,SAAS,OAAO;AAEvB,OAAI,OAAO,MAAM,YAAY;IAC3B,MAAM,OAAO,IAAI,SAAS;AAC1B,QAAI,OAAO,SAAS,YAAY,SAAS,KACvC,MAAK,UAAU,UAAU,KAAK,GAAG,KAAK;AAExC,SAAK,SAAS;AACd,YAAQ,KAAK,QAAQ;KACrB;IACF;;CAGJ,MAAM,OAAsB;AAC1B,MAAI,CAAC,KAAK,OACR,OAAM,IAAI,MAAM,8BAA8B;EAEhD,MAAM,MAAM,KAAK;AACjB,QAAM,IAAI,SAAe,SAAS,WAAW;AAC3C,OAAI,OAAO,QAA4B,MAAM,OAAO,IAAI,GAAG,SAAS,CAAE;IACtE;AACF,OAAK,SAAS;;CAGhB,IAAI,MAAc;AAChB,MAAI,CAAC,KAAK,OACR,OAAM,IAAI,MAAM,8BAA8B;AAEhD,SAAO,KAAK;;CAKd,AAAQ,eAAe,KAA2B,UAAkB,QAAsB;AACxF,MAAI,KAAK,QACP,MAAK,QAAQ,IAAI;GACf,QAAQ,IAAI,UAAU;GACtB,MAAM,IAAI,OAAO;GACjB,SAAS,eAAe,IAAI,QAAQ;GACpC,MAAM;GACN,SAAS;GACT,UAAU;IAAE;IAAQ,SAAS;IAAM;GACpC,CAAC"}

package/dist/agui-recorder.cjs

@@ -0,0 +1,153 @@
+const require_runtime = require('./_virtual/_rolldown/runtime.cjs');
+const require_agui_handler = require('./agui-handler.cjs');
+let node_http = require("node:http");
+node_http = require_runtime.__toESM(node_http);
+let node_crypto = require("node:crypto");
+node_crypto = require_runtime.__toESM(node_crypto);
+let node_path = require("node:path");
+node_path = require_runtime.__toESM(node_path);
+let node_fs = require("node:fs");
+node_fs = require_runtime.__toESM(node_fs);
+let node_https = require("node:https");
+node_https = require_runtime.__toESM(node_https);
+
+//#region src/agui-recorder.ts
+/**
+* Proxy an unmatched AG-UI request to a real upstream agent, record the
+* SSE event stream as a fixture on disk and in memory, and relay the
+* response back to the original client in real time.
+*
+* Returns `true` if the request was proxied, `false` if no upstream is configured.
+*/
+async function proxyAndRecordAGUI(req, res, input, fixtures, config, logger) {
+	if (!config.upstream) {
+		logger.warn("No upstream URL configured for AG-UI recording — cannot proxy");
+		return false;
+	}
+	let target;
+	try {
+		target = new URL(config.upstream);
+	} catch {
+		logger.error(`Invalid upstream AG-UI URL: ${config.upstream}`);
+		res.writeHead(502, { "Content-Type": "application/json" });
+		res.end(JSON.stringify({ error: "Invalid upstream AG-UI URL" }));
+		return true;
+	}
+	logger.warn(`NO AG-UI FIXTURE MATCH — proxying to ${config.upstream}`);
+	const forwardHeaders = {
+		"Content-Type": "application/json",
+		Accept: "text/event-stream"
+	};
+	const authorization = req.headers["authorization"];
+	if (authorization) forwardHeaders["Authorization"] = Array.isArray(authorization) ? authorization.join(", ") : authorization;
+	const apiKey = req.headers["x-api-key"];
+	if (apiKey) forwardHeaders["x-api-key"] = Array.isArray(apiKey) ? apiKey.join(", ") : apiKey;
+	const requestBody = JSON.stringify(input);
+	try {
+		await teeUpstreamStream(target, forwardHeaders, requestBody, res, input, fixtures, config, logger);
+	} catch (err) {
+		const msg = err instanceof Error ? err.message : "Unknown proxy error";
+		logger.error(`AG-UI proxy request failed: ${msg}`);
+		if (!res.headersSent) {
+			res.writeHead(502, { "Content-Type": "application/json" });
+			res.end(JSON.stringify({ error: "Upstream AG-UI agent unreachable" }));
+		}
+	}
+	return true;
+}
+function teeUpstreamStream(target, headers, body, clientRes, input, fixtures, config, logger) {
+	return new Promise((resolve, reject) => {
+		const transport = target.protocol === "https:" ? node_https : node_http;
+		const UPSTREAM_TIMEOUT_MS = 3e4;
+		const upstreamReq = transport.request(target, {
+			method: "POST",
+			timeout: UPSTREAM_TIMEOUT_MS,
+			headers: {
+				...headers,
+				"Content-Length": Buffer.byteLength(body).toString()
+			}
+		}, (upstreamRes) => {
+			if (!clientRes.headersSent) clientRes.writeHead(upstreamRes.statusCode ?? 200, {
+				"Content-Type": "text/event-stream",
+				"Cache-Control": "no-cache",
+				Connection: "keep-alive"
+			});
+			const chunks = [];
+			upstreamRes.on("data", (chunk) => {
+				try {
+					clientRes.write(chunk);
+				} catch {}
+				chunks.push(chunk);
+			});
+			upstreamRes.on("error", (err) => {
+				if (!clientRes.headersSent) {
+					clientRes.writeHead(502, { "Content-Type": "application/json" });
+					clientRes.end(JSON.stringify({ error: "Upstream AG-UI agent unreachable" }));
+				} else if (!clientRes.writableEnded) clientRes.end();
+				reject(err);
+			});
+			upstreamRes.on("end", () => {
+				if (!clientRes.writableEnded) clientRes.end();
+				const events = parseSSEEvents(Buffer.concat(chunks).toString(), logger);
+				const message = require_agui_handler.extractLastUserMessage(input);
+				if (!message) logger.warn("Recorded AG-UI fixture has no message match — it will match ALL requests");
+				const fixture = {
+					match: { message: message || void 0 },
+					events
+				};
+				if (!config.proxyOnly) {
+					fixtures.push(fixture);
+					const fixturePath = config.fixturePath ?? "./fixtures/agui-recorded";
+					const filename = `agui-${(/* @__PURE__ */ new Date()).toISOString().replace(/[:.]/g, "-")}-${node_crypto.randomUUID().slice(0, 8)}.json`;
+					const filepath = node_path.join(fixturePath, filename);
+					try {
+						node_fs.mkdirSync(fixturePath, { recursive: true });
+						node_fs.writeFileSync(filepath, JSON.stringify({ fixtures: [{
+							match: fixture.match,
+							events: fixture.events
+						}] }, null, 2), "utf-8");
+						logger.warn(`AG-UI response recorded → ${filepath}`);
+					} catch (err) {
+						const msg = err instanceof Error ? err.message : "Unknown filesystem error";
+						logger.error(`Failed to save AG-UI fixture to disk: ${msg} (fixture retained in memory)`);
+					}
+				} else logger.info("Proxied AG-UI request (proxy-only mode)");
+				resolve();
+			});
+		});
+		upstreamReq.on("timeout", () => {
+			if (!clientRes.writableEnded) clientRes.end();
+			upstreamReq.destroy(/* @__PURE__ */ new Error(`Upstream AG-UI request timed out after ${UPSTREAM_TIMEOUT_MS / 1e3}s`));
+		});
+		upstreamReq.on("error", (err) => {
+			if (!clientRes.headersSent) {
+				clientRes.writeHead(502, { "Content-Type": "application/json" });
+				clientRes.end(JSON.stringify({ error: "Upstream AG-UI agent unreachable" }));
+			} else if (!clientRes.writableEnded) clientRes.end();
+			reject(err);
+		});
+		upstreamReq.write(body);
+		upstreamReq.end();
+	});
+}
+/**
+* Parse SSE data lines from buffered stream text.
+*/
+function parseSSEEvents(text, logger) {
+	const events = [];
+	const blocks = text.split("\n\n");
+	for (const block of blocks) {
+		const lines = block.split("\n");
+		for (const line of lines) if (line.startsWith("data: ")) try {
+			const parsed = JSON.parse(line.slice(6));
+			events.push(parsed);
+		} catch {
+			logger?.warn(`Skipping unparseable SSE data line: ${line.slice(0, 200)}`);
+		}
+	}
+	return events;
+}
+
+//#endregion
+exports.proxyAndRecordAGUI = proxyAndRecordAGUI;
+//# sourceMappingURL=agui-recorder.cjs.map

package/dist/agui-recorder.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"agui-recorder.cjs","names":["https","http","extractLastUserMessage","crypto","path"],"sources":["../src/agui-recorder.ts"],"sourcesContent":["import * as http from \"node:http\";\nimport * as https from \"node:https\";\nimport * as fs from \"node:fs\";\nimport * as path from \"node:path\";\nimport * as crypto from \"node:crypto\";\nimport type { AGUIFixture, AGUIRecordConfig, AGUIEvent, AGUIRunAgentInput } from \"./agui-types.js\";\nimport { extractLastUserMessage } from \"./agui-handler.js\";\nimport type { Logger } from \"./logger.js\";\n\n/**\n * Proxy an unmatched AG-UI request to a real upstream agent, record the\n * SSE event stream as a fixture on disk and in memory, and relay the\n * response back to the original client in real time.\n *\n * Returns `true` if the request was proxied, `false` if no upstream is configured.\n */\nexport async function proxyAndRecordAGUI(\n  req: http.IncomingMessage,\n  res: http.ServerResponse,\n  input: AGUIRunAgentInput,\n  fixtures: AGUIFixture[],\n  config: AGUIRecordConfig,\n  logger: Logger,\n): Promise<boolean> {\n  if (!config.upstream) {\n    logger.warn(\"No upstream URL configured for AG-UI recording — cannot proxy\");\n    return false;\n  }\n\n  let target: URL;\n  try {\n    target = new URL(config.upstream);\n  } catch {\n    logger.error(`Invalid upstream AG-UI URL: ${config.upstream}`);\n    res.writeHead(502, { \"Content-Type\": \"application/json\" });\n    res.end(JSON.stringify({ error: \"Invalid upstream AG-UI URL\" }));\n    return true;\n  }\n\n  logger.warn(`NO AG-UI FIXTURE MATCH — proxying to ${config.upstream}`);\n\n  // Build upstream request headers\n  const forwardHeaders: Record<string, string> = {\n    \"Content-Type\": \"application/json\",\n    Accept: \"text/event-stream\",\n  };\n  // Forward auth headers if present\n  const authorization = req.headers[\"authorization\"];\n  if (authorization) {\n    forwardHeaders[\"Authorization\"] = Array.isArray(authorization)\n      ? authorization.join(\", \")\n      : authorization;\n  }\n  const apiKey = req.headers[\"x-api-key\"];\n  if (apiKey) {\n    forwardHeaders[\"x-api-key\"] = Array.isArray(apiKey) ? apiKey.join(\", \") : apiKey;\n  }\n\n  const requestBody = JSON.stringify(input);\n\n  try {\n    await teeUpstreamStream(\n      target,\n      forwardHeaders,\n      requestBody,\n      res,\n      input,\n      fixtures,\n      config,\n      logger,\n    );\n  } catch (err) {\n    const msg = err instanceof Error ? err.message : \"Unknown proxy error\";\n    logger.error(`AG-UI proxy request failed: ${msg}`);\n    if (!res.headersSent) {\n      res.writeHead(502, { \"Content-Type\": \"application/json\" });\n      res.end(JSON.stringify({ error: \"Upstream AG-UI agent unreachable\" }));\n    }\n  }\n\n  return true;\n}\n\n// ---------------------------------------------------------------------------\n// Internal: tee the upstream SSE stream to the client and buffer for recording\n// ---------------------------------------------------------------------------\n\nfunction teeUpstreamStream(\n  target: URL,\n  headers: Record<string, string>,\n  body: string,\n  clientRes: http.ServerResponse,\n  input: AGUIRunAgentInput,\n  fixtures: AGUIFixture[],\n  config: AGUIRecordConfig,\n  logger: Logger,\n): Promise<void> {\n  return new Promise((resolve, reject) => {\n    const transport = target.protocol === \"https:\" ? https : http;\n    const UPSTREAM_TIMEOUT_MS = 30_000;\n\n    const upstreamReq = transport.request(\n      target,\n      {\n        method: \"POST\",\n        timeout: UPSTREAM_TIMEOUT_MS,\n        headers: {\n          ...headers,\n          \"Content-Length\": Buffer.byteLength(body).toString(),\n        },\n      },\n      (upstreamRes) => {\n        // Set SSE headers on the client response\n        if (!clientRes.headersSent) {\n          clientRes.writeHead(upstreamRes.statusCode ?? 200, {\n            \"Content-Type\": \"text/event-stream\",\n            \"Cache-Control\": \"no-cache\",\n            Connection: \"keep-alive\",\n          });\n        }\n\n        const chunks: Buffer[] = [];\n\n        upstreamRes.on(\"data\", (chunk: Buffer) => {\n          // Relay to client in real time\n          try {\n            clientRes.write(chunk);\n          } catch {\n            // Client connection may have closed — continue buffering for recording\n          }\n          // Buffer for fixture construction\n          chunks.push(chunk);\n        });\n\n        upstreamRes.on(\"error\", (err) => {\n          if (!clientRes.headersSent) {\n            clientRes.writeHead(502, { \"Content-Type\": \"application/json\" });\n            clientRes.end(JSON.stringify({ error: \"Upstream AG-UI agent unreachable\" }));\n          } else if (!clientRes.writableEnded) {\n            clientRes.end();\n          }\n          reject(err);\n        });\n\n        upstreamRes.on(\"end\", () => {\n          if (!clientRes.writableEnded) clientRes.end();\n\n          // Parse buffered SSE events\n          const buffered = Buffer.concat(chunks).toString();\n          const events = parseSSEEvents(buffered, logger);\n\n          // Build fixture\n          const message = extractLastUserMessage(input);\n          if (!message) {\n            logger.warn(\"Recorded AG-UI fixture has no message match — it will match ALL requests\");\n          }\n          const fixture: AGUIFixture = {\n            match: { message: message || undefined },\n            events,\n          };\n\n          if (!config.proxyOnly) {\n            // Register in memory first (always available even if disk write fails)\n            fixtures.push(fixture);\n\n            // Write to disk\n            const fixturePath = config.fixturePath ?? \"./fixtures/agui-recorded\";\n            const timestamp = new Date().toISOString().replace(/[:.]/g, \"-\");\n            const filename = `agui-${timestamp}-${crypto.randomUUID().slice(0, 8)}.json`;\n            const filepath = path.join(fixturePath, filename);\n\n            try {\n              fs.mkdirSync(fixturePath, { recursive: true });\n              fs.writeFileSync(\n                filepath,\n                JSON.stringify(\n                  { fixtures: [{ match: fixture.match, events: fixture.events }] },\n                  null,\n                  2,\n                ),\n                \"utf-8\",\n              );\n              logger.warn(`AG-UI response recorded → ${filepath}`);\n            } catch (err) {\n              const msg = err instanceof Error ? err.message : \"Unknown filesystem error\";\n              logger.error(\n                `Failed to save AG-UI fixture to disk: ${msg} (fixture retained in memory)`,\n              );\n            }\n          } else {\n            logger.info(\"Proxied AG-UI request (proxy-only mode)\");\n          }\n\n          resolve();\n        });\n      },\n    );\n\n    upstreamReq.on(\"timeout\", () => {\n      if (!clientRes.writableEnded) clientRes.end();\n      upstreamReq.destroy(\n        new Error(`Upstream AG-UI request timed out after ${UPSTREAM_TIMEOUT_MS / 1000}s`),\n      );\n    });\n\n    upstreamReq.on(\"error\", (err) => {\n      if (!clientRes.headersSent) {\n        clientRes.writeHead(502, { \"Content-Type\": \"application/json\" });\n        clientRes.end(JSON.stringify({ error: \"Upstream AG-UI agent unreachable\" }));\n      } else if (!clientRes.writableEnded) {\n        clientRes.end();\n      }\n      reject(err);\n    });\n\n    upstreamReq.write(body);\n    upstreamReq.end();\n  });\n}\n\n/**\n * Parse SSE data lines from buffered stream text.\n */\nfunction parseSSEEvents(text: string, logger?: Logger): AGUIEvent[] {\n  const events: AGUIEvent[] = [];\n  const blocks = text.split(\"\\n\\n\");\n  for (const block of blocks) {\n    const lines = block.split(\"\\n\");\n    for (const line of lines) {\n      if (line.startsWith(\"data: \")) {\n        try {\n          const parsed = JSON.parse(line.slice(6)) as AGUIEvent;\n          events.push(parsed);\n        } catch {\n          logger?.warn(`Skipping unparseable SSE data line: ${line.slice(0, 200)}`);\n        }\n      }\n    }\n  }\n  return events;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;AAgBA,eAAsB,mBACpB,KACA,KACA,OACA,UACA,QACA,QACkB;AAClB,KAAI,CAAC,OAAO,UAAU;AACpB,SAAO,KAAK,gEAAgE;AAC5E,SAAO;;CAGT,IAAI;AACJ,KAAI;AACF,WAAS,IAAI,IAAI,OAAO,SAAS;SAC3B;AACN,SAAO,MAAM,+BAA+B,OAAO,WAAW;AAC9D,MAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,MAAI,IAAI,KAAK,UAAU,EAAE,OAAO,8BAA8B,CAAC,CAAC;AAChE,SAAO;;AAGT,QAAO,KAAK,wCAAwC,OAAO,WAAW;CAGtE,MAAM,iBAAyC;EAC7C,gBAAgB;EAChB,QAAQ;EACT;CAED,MAAM,gBAAgB,IAAI,QAAQ;AAClC,KAAI,cACF,gBAAe,mBAAmB,MAAM,QAAQ,cAAc,GAC1D,cAAc,KAAK,KAAK,GACxB;CAEN,MAAM,SAAS,IAAI,QAAQ;AAC3B,KAAI,OACF,gBAAe,eAAe,MAAM,QAAQ,OAAO,GAAG,OAAO,KAAK,KAAK,GAAG;CAG5E,MAAM,cAAc,KAAK,UAAU,MAAM;AAEzC,KAAI;AACF,QAAM,kBACJ,QACA,gBACA,aACA,KACA,OACA,UACA,QACA,OACD;UACM,KAAK;EACZ,MAAM,MAAM,eAAe,QAAQ,IAAI,UAAU;AACjD,SAAO,MAAM,+BAA+B,MAAM;AAClD,MAAI,CAAC,IAAI,aAAa;AACpB,OAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,OAAI,IAAI,KAAK,UAAU,EAAE,OAAO,oCAAoC,CAAC,CAAC;;;AAI1E,QAAO;;AAOT,SAAS,kBACP,QACA,SACA,MACA,WACA,OACA,UACA,QACA,QACe;AACf,QAAO,IAAI,SAAS,SAAS,WAAW;EACtC,MAAM,YAAY,OAAO,aAAa,WAAWA,aAAQC;EACzD,MAAM,sBAAsB;EAE5B,MAAM,cAAc,UAAU,QAC5B,QACA;GACE,QAAQ;GACR,SAAS;GACT,SAAS;IACP,GAAG;IACH,kBAAkB,OAAO,WAAW,KAAK,CAAC,UAAU;IACrD;GACF,GACA,gBAAgB;AAEf,OAAI,CAAC,UAAU,YACb,WAAU,UAAU,YAAY,cAAc,KAAK;IACjD,gBAAgB;IAChB,iBAAiB;IACjB,YAAY;IACb,CAAC;GAGJ,MAAM,SAAmB,EAAE;AAE3B,eAAY,GAAG,SAAS,UAAkB;AAExC,QAAI;AACF,eAAU,MAAM,MAAM;YAChB;AAIR,WAAO,KAAK,MAAM;KAClB;AAEF,eAAY,GAAG,UAAU,QAAQ;AAC/B,QAAI,CAAC,UAAU,aAAa;AAC1B,eAAU,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAChE,eAAU,IAAI,KAAK,UAAU,EAAE,OAAO,oCAAoC,CAAC,CAAC;eACnE,CAAC,UAAU,cACpB,WAAU,KAAK;AAEjB,WAAO,IAAI;KACX;AAEF,eAAY,GAAG,aAAa;AAC1B,QAAI,CAAC,UAAU,cAAe,WAAU,KAAK;IAI7C,MAAM,SAAS,eADE,OAAO,OAAO,OAAO,CAAC,UAAU,EACT,OAAO;IAG/C,MAAM,UAAUC,4CAAuB,MAAM;AAC7C,QAAI,CAAC,QACH,QAAO,KAAK,2EAA2E;IAEzF,MAAM,UAAuB;KAC3B,OAAO,EAAE,SAAS,WAAW,QAAW;KACxC;KACD;AAED,QAAI,CAAC,OAAO,WAAW;AAErB,cAAS,KAAK,QAAQ;KAGtB,MAAM,cAAc,OAAO,eAAe;KAE1C,MAAM,WAAW,yBADC,IAAI,MAAM,EAAC,aAAa,CAAC,QAAQ,SAAS,IAAI,CAC7B,GAAGC,YAAO,YAAY,CAAC,MAAM,GAAG,EAAE,CAAC;KACtE,MAAM,WAAWC,UAAK,KAAK,aAAa,SAAS;AAEjD,SAAI;AACF,cAAG,UAAU,aAAa,EAAE,WAAW,MAAM,CAAC;AAC9C,cAAG,cACD,UACA,KAAK,UACH,EAAE,UAAU,CAAC;OAAE,OAAO,QAAQ;OAAO,QAAQ,QAAQ;OAAQ,CAAC,EAAE,EAChE,MACA,EACD,EACD,QACD;AACD,aAAO,KAAK,6BAA6B,WAAW;cAC7C,KAAK;MACZ,MAAM,MAAM,eAAe,QAAQ,IAAI,UAAU;AACjD,aAAO,MACL,yCAAyC,IAAI,+BAC9C;;UAGH,QAAO,KAAK,0CAA0C;AAGxD,aAAS;KACT;IAEL;AAED,cAAY,GAAG,iBAAiB;AAC9B,OAAI,CAAC,UAAU,cAAe,WAAU,KAAK;AAC7C,eAAY,wBACV,IAAI,MAAM,0CAA0C,sBAAsB,IAAK,GAAG,CACnF;IACD;AAEF,cAAY,GAAG,UAAU,QAAQ;AAC/B,OAAI,CAAC,UAAU,aAAa;AAC1B,cAAU,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAChE,cAAU,IAAI,KAAK,UAAU,EAAE,OAAO,oCAAoC,CAAC,CAAC;cACnE,CAAC,UAAU,cACpB,WAAU,KAAK;AAEjB,UAAO,IAAI;IACX;AAEF,cAAY,MAAM,KAAK;AACvB,cAAY,KAAK;GACjB;;;;;AAMJ,SAAS,eAAe,MAAc,QAA8B;CAClE,MAAM,SAAsB,EAAE;CAC9B,MAAM,SAAS,KAAK,MAAM,OAAO;AACjC,MAAK,MAAM,SAAS,QAAQ;EAC1B,MAAM,QAAQ,MAAM,MAAM,KAAK;AAC/B,OAAK,MAAM,QAAQ,MACjB,KAAI,KAAK,WAAW,SAAS,CAC3B,KAAI;GACF,MAAM,SAAS,KAAK,MAAM,KAAK,MAAM,EAAE,CAAC;AACxC,UAAO,KAAK,OAAO;UACb;AACN,WAAQ,KAAK,uCAAuC,KAAK,MAAM,GAAG,IAAI,GAAG;;;AAKjF,QAAO"}

package/dist/agui-recorder.d.cts

@@ -0,0 +1,19 @@
+import { Logger } from "./logger.cjs";
+import { AGUIFixture, AGUIRecordConfig, AGUIRunAgentInput } from "./agui-types.cjs";
+import * as http from "node:http";
+
+//#region src/agui-recorder.d.ts
+
+/**
+ * Proxy an unmatched AG-UI request to a real upstream agent, record the
+ * SSE event stream as a fixture on disk and in memory, and relay the
+ * response back to the original client in real time.
+ *
+ * Returns `true` if the request was proxied, `false` if no upstream is configured.
+ */
+declare function proxyAndRecordAGUI(req: http.IncomingMessage, res: http.ServerResponse, input: AGUIRunAgentInput, fixtures: AGUIFixture[], config: AGUIRecordConfig, logger: Logger): Promise<boolean>;
+//# sourceMappingURL=agui-recorder.d.ts.map
+
+//#endregion
+export { proxyAndRecordAGUI };
+//# sourceMappingURL=agui-recorder.d.cts.map

package/dist/agui-recorder.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agui-recorder.d.cts","names":[],"sources":["../src/agui-recorder.ts"],"sourcesContent":[],"mappings":";;;;;;;;AAgBA;;;;;AAIY,iBAJU,kBAAA,CAIV,GAAA,EAHL,IAAA,CAAK,eAGA,EAAA,GAAA,EAFL,IAAA,CAAK,cAEA,EAAA,KAAA,EADH,iBACG,EAAA,QAAA,EAAA,WAAA,EAAA,EAAA,MAAA,EACF,gBADE,EAAA,MAAA,EAEF,MAFE,CAAA,EAGT,OAHS,CAAA,OAAA,CAAA"}

package/dist/agui-recorder.d.ts

@@ -0,0 +1,19 @@
+import { Logger } from "./logger.js";
+import { AGUIFixture, AGUIRecordConfig, AGUIRunAgentInput } from "./agui-types.js";
+import * as http from "node:http";
+
+//#region src/agui-recorder.d.ts
+
+/**
+ * Proxy an unmatched AG-UI request to a real upstream agent, record the
+ * SSE event stream as a fixture on disk and in memory, and relay the
+ * response back to the original client in real time.
+ *
+ * Returns `true` if the request was proxied, `false` if no upstream is configured.
+ */
+declare function proxyAndRecordAGUI(req: http.IncomingMessage, res: http.ServerResponse, input: AGUIRunAgentInput, fixtures: AGUIFixture[], config: AGUIRecordConfig, logger: Logger): Promise<boolean>;
+//# sourceMappingURL=agui-recorder.d.ts.map
+
+//#endregion
+export { proxyAndRecordAGUI };
+//# sourceMappingURL=agui-recorder.d.ts.map

package/dist/agui-recorder.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agui-recorder.d.ts","names":[],"sources":["../src/agui-recorder.ts"],"sourcesContent":[],"mappings":";;;;;;;;AAgBA;;;;;AAIY,iBAJU,kBAAA,CAIV,GAAA,EAHL,IAAA,CAAK,eAGA,EAAA,GAAA,EAFL,IAAA,CAAK,cAEA,EAAA,KAAA,EADH,iBACG,EAAA,QAAA,EAAA,WAAA,EAAA,EAAA,MAAA,EACF,gBADE,EAAA,MAAA,EAEF,MAFE,CAAA,EAGT,OAHS,CAAA,OAAA,CAAA"}

package/dist/agui-recorder.js

@@ -0,0 +1,147 @@
+import { extractLastUserMessage } from "./agui-handler.js";
+import * as http from "node:http";
+import * as crypto from "node:crypto";
+import * as path from "node:path";
+import * as fs from "node:fs";
+import * as https from "node:https";
+
+//#region src/agui-recorder.ts
+/**
+* Proxy an unmatched AG-UI request to a real upstream agent, record the
+* SSE event stream as a fixture on disk and in memory, and relay the
+* response back to the original client in real time.
+*
+* Returns `true` if the request was proxied, `false` if no upstream is configured.
+*/
+async function proxyAndRecordAGUI(req, res, input, fixtures, config, logger) {
+	if (!config.upstream) {
+		logger.warn("No upstream URL configured for AG-UI recording — cannot proxy");
+		return false;
+	}
+	let target;
+	try {
+		target = new URL(config.upstream);
+	} catch {
+		logger.error(`Invalid upstream AG-UI URL: ${config.upstream}`);
+		res.writeHead(502, { "Content-Type": "application/json" });
+		res.end(JSON.stringify({ error: "Invalid upstream AG-UI URL" }));
+		return true;
+	}
+	logger.warn(`NO AG-UI FIXTURE MATCH — proxying to ${config.upstream}`);
+	const forwardHeaders = {
+		"Content-Type": "application/json",
+		Accept: "text/event-stream"
+	};
+	const authorization = req.headers["authorization"];
+	if (authorization) forwardHeaders["Authorization"] = Array.isArray(authorization) ? authorization.join(", ") : authorization;
+	const apiKey = req.headers["x-api-key"];
+	if (apiKey) forwardHeaders["x-api-key"] = Array.isArray(apiKey) ? apiKey.join(", ") : apiKey;
+	const requestBody = JSON.stringify(input);
+	try {
+		await teeUpstreamStream(target, forwardHeaders, requestBody, res, input, fixtures, config, logger);
+	} catch (err) {
+		const msg = err instanceof Error ? err.message : "Unknown proxy error";
+		logger.error(`AG-UI proxy request failed: ${msg}`);
+		if (!res.headersSent) {
+			res.writeHead(502, { "Content-Type": "application/json" });
+			res.end(JSON.stringify({ error: "Upstream AG-UI agent unreachable" }));
+		}
+	}
+	return true;
+}
+function teeUpstreamStream(target, headers, body, clientRes, input, fixtures, config, logger) {
+	return new Promise((resolve, reject) => {
+		const transport = target.protocol === "https:" ? https : http;
+		const UPSTREAM_TIMEOUT_MS = 3e4;
+		const upstreamReq = transport.request(target, {
+			method: "POST",
+			timeout: UPSTREAM_TIMEOUT_MS,
+			headers: {
+				...headers,
+				"Content-Length": Buffer.byteLength(body).toString()
+			}
+		}, (upstreamRes) => {
+			if (!clientRes.headersSent) clientRes.writeHead(upstreamRes.statusCode ?? 200, {
+				"Content-Type": "text/event-stream",
+				"Cache-Control": "no-cache",
+				Connection: "keep-alive"
+			});
+			const chunks = [];
+			upstreamRes.on("data", (chunk) => {
+				try {
+					clientRes.write(chunk);
+				} catch {}
+				chunks.push(chunk);
+			});
+			upstreamRes.on("error", (err) => {
+				if (!clientRes.headersSent) {
+					clientRes.writeHead(502, { "Content-Type": "application/json" });
+					clientRes.end(JSON.stringify({ error: "Upstream AG-UI agent unreachable" }));
+				} else if (!clientRes.writableEnded) clientRes.end();
+				reject(err);
+			});
+			upstreamRes.on("end", () => {
+				if (!clientRes.writableEnded) clientRes.end();
+				const events = parseSSEEvents(Buffer.concat(chunks).toString(), logger);
+				const message = extractLastUserMessage(input);
+				if (!message) logger.warn("Recorded AG-UI fixture has no message match — it will match ALL requests");
+				const fixture = {
+					match: { message: message || void 0 },
+					events
+				};
+				if (!config.proxyOnly) {
+					fixtures.push(fixture);
+					const fixturePath = config.fixturePath ?? "./fixtures/agui-recorded";
+					const filename = `agui-${(/* @__PURE__ */ new Date()).toISOString().replace(/[:.]/g, "-")}-${crypto.randomUUID().slice(0, 8)}.json`;
+					const filepath = path.join(fixturePath, filename);
+					try {
+						fs.mkdirSync(fixturePath, { recursive: true });
+						fs.writeFileSync(filepath, JSON.stringify({ fixtures: [{
+							match: fixture.match,
+							events: fixture.events
+						}] }, null, 2), "utf-8");
+						logger.warn(`AG-UI response recorded → ${filepath}`);
+					} catch (err) {
+						const msg = err instanceof Error ? err.message : "Unknown filesystem error";
+						logger.error(`Failed to save AG-UI fixture to disk: ${msg} (fixture retained in memory)`);
+					}
+				} else logger.info("Proxied AG-UI request (proxy-only mode)");
+				resolve();
+			});
+		});
+		upstreamReq.on("timeout", () => {
+			if (!clientRes.writableEnded) clientRes.end();
+			upstreamReq.destroy(/* @__PURE__ */ new Error(`Upstream AG-UI request timed out after ${UPSTREAM_TIMEOUT_MS / 1e3}s`));
+		});
+		upstreamReq.on("error", (err) => {
+			if (!clientRes.headersSent) {
+				clientRes.writeHead(502, { "Content-Type": "application/json" });
+				clientRes.end(JSON.stringify({ error: "Upstream AG-UI agent unreachable" }));
+			} else if (!clientRes.writableEnded) clientRes.end();
+			reject(err);
+		});
+		upstreamReq.write(body);
+		upstreamReq.end();
+	});
+}
+/**
+* Parse SSE data lines from buffered stream text.
+*/
+function parseSSEEvents(text, logger) {
+	const events = [];
+	const blocks = text.split("\n\n");
+	for (const block of blocks) {
+		const lines = block.split("\n");
+		for (const line of lines) if (line.startsWith("data: ")) try {
+			const parsed = JSON.parse(line.slice(6));
+			events.push(parsed);
+		} catch {
+			logger?.warn(`Skipping unparseable SSE data line: ${line.slice(0, 200)}`);
+		}
+	}
+	return events;
+}
+
+//#endregion
+export { proxyAndRecordAGUI };
+//# sourceMappingURL=agui-recorder.js.map

package/dist/agui-recorder.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agui-recorder.js","names":[],"sources":["../src/agui-recorder.ts"],"sourcesContent":["import * as http from \"node:http\";\nimport * as https from \"node:https\";\nimport * as fs from \"node:fs\";\nimport * as path from \"node:path\";\nimport * as crypto from \"node:crypto\";\nimport type { AGUIFixture, AGUIRecordConfig, AGUIEvent, AGUIRunAgentInput } from \"./agui-types.js\";\nimport { extractLastUserMessage } from \"./agui-handler.js\";\nimport type { Logger } from \"./logger.js\";\n\n/**\n * Proxy an unmatched AG-UI request to a real upstream agent, record the\n * SSE event stream as a fixture on disk and in memory, and relay the\n * response back to the original client in real time.\n *\n * Returns `true` if the request was proxied, `false` if no upstream is configured.\n */\nexport async function proxyAndRecordAGUI(\n  req: http.IncomingMessage,\n  res: http.ServerResponse,\n  input: AGUIRunAgentInput,\n  fixtures: AGUIFixture[],\n  config: AGUIRecordConfig,\n  logger: Logger,\n): Promise<boolean> {\n  if (!config.upstream) {\n    logger.warn(\"No upstream URL configured for AG-UI recording — cannot proxy\");\n    return false;\n  }\n\n  let target: URL;\n  try {\n    target = new URL(config.upstream);\n  } catch {\n    logger.error(`Invalid upstream AG-UI URL: ${config.upstream}`);\n    res.writeHead(502, { \"Content-Type\": \"application/json\" });\n    res.end(JSON.stringify({ error: \"Invalid upstream AG-UI URL\" }));\n    return true;\n  }\n\n  logger.warn(`NO AG-UI FIXTURE MATCH — proxying to ${config.upstream}`);\n\n  // Build upstream request headers\n  const forwardHeaders: Record<string, string> = {\n    \"Content-Type\": \"application/json\",\n    Accept: \"text/event-stream\",\n  };\n  // Forward auth headers if present\n  const authorization = req.headers[\"authorization\"];\n  if (authorization) {\n    forwardHeaders[\"Authorization\"] = Array.isArray(authorization)\n      ? authorization.join(\", \")\n      : authorization;\n  }\n  const apiKey = req.headers[\"x-api-key\"];\n  if (apiKey) {\n    forwardHeaders[\"x-api-key\"] = Array.isArray(apiKey) ? apiKey.join(\", \") : apiKey;\n  }\n\n  const requestBody = JSON.stringify(input);\n\n  try {\n    await teeUpstreamStream(\n      target,\n      forwardHeaders,\n      requestBody,\n      res,\n      input,\n      fixtures,\n      config,\n      logger,\n    );\n  } catch (err) {\n    const msg = err instanceof Error ? err.message : \"Unknown proxy error\";\n    logger.error(`AG-UI proxy request failed: ${msg}`);\n    if (!res.headersSent) {\n      res.writeHead(502, { \"Content-Type\": \"application/json\" });\n      res.end(JSON.stringify({ error: \"Upstream AG-UI agent unreachable\" }));\n    }\n  }\n\n  return true;\n}\n\n// ---------------------------------------------------------------------------\n// Internal: tee the upstream SSE stream to the client and buffer for recording\n// ---------------------------------------------------------------------------\n\nfunction teeUpstreamStream(\n  target: URL,\n  headers: Record<string, string>,\n  body: string,\n  clientRes: http.ServerResponse,\n  input: AGUIRunAgentInput,\n  fixtures: AGUIFixture[],\n  config: AGUIRecordConfig,\n  logger: Logger,\n): Promise<void> {\n  return new Promise((resolve, reject) => {\n    const transport = target.protocol === \"https:\" ? https : http;\n    const UPSTREAM_TIMEOUT_MS = 30_000;\n\n    const upstreamReq = transport.request(\n      target,\n      {\n        method: \"POST\",\n        timeout: UPSTREAM_TIMEOUT_MS,\n        headers: {\n          ...headers,\n          \"Content-Length\": Buffer.byteLength(body).toString(),\n        },\n      },\n      (upstreamRes) => {\n        // Set SSE headers on the client response\n        if (!clientRes.headersSent) {\n          clientRes.writeHead(upstreamRes.statusCode ?? 200, {\n            \"Content-Type\": \"text/event-stream\",\n            \"Cache-Control\": \"no-cache\",\n            Connection: \"keep-alive\",\n          });\n        }\n\n        const chunks: Buffer[] = [];\n\n        upstreamRes.on(\"data\", (chunk: Buffer) => {\n          // Relay to client in real time\n          try {\n            clientRes.write(chunk);\n          } catch {\n            // Client connection may have closed — continue buffering for recording\n          }\n          // Buffer for fixture construction\n          chunks.push(chunk);\n        });\n\n        upstreamRes.on(\"error\", (err) => {\n          if (!clientRes.headersSent) {\n            clientRes.writeHead(502, { \"Content-Type\": \"application/json\" });\n            clientRes.end(JSON.stringify({ error: \"Upstream AG-UI agent unreachable\" }));\n          } else if (!clientRes.writableEnded) {\n            clientRes.end();\n          }\n          reject(err);\n        });\n\n        upstreamRes.on(\"end\", () => {\n          if (!clientRes.writableEnded) clientRes.end();\n\n          // Parse buffered SSE events\n          const buffered = Buffer.concat(chunks).toString();\n          const events = parseSSEEvents(buffered, logger);\n\n          // Build fixture\n          const message = extractLastUserMessage(input);\n          if (!message) {\n            logger.warn(\"Recorded AG-UI fixture has no message match — it will match ALL requests\");\n          }\n          const fixture: AGUIFixture = {\n            match: { message: message || undefined },\n            events,\n          };\n\n          if (!config.proxyOnly) {\n            // Register in memory first (always available even if disk write fails)\n            fixtures.push(fixture);\n\n            // Write to disk\n            const fixturePath = config.fixturePath ?? \"./fixtures/agui-recorded\";\n            const timestamp = new Date().toISOString().replace(/[:.]/g, \"-\");\n            const filename = `agui-${timestamp}-${crypto.randomUUID().slice(0, 8)}.json`;\n            const filepath = path.join(fixturePath, filename);\n\n            try {\n              fs.mkdirSync(fixturePath, { recursive: true });\n              fs.writeFileSync(\n                filepath,\n                JSON.stringify(\n                  { fixtures: [{ match: fixture.match, events: fixture.events }] },\n                  null,\n                  2,\n                ),\n                \"utf-8\",\n              );\n              logger.warn(`AG-UI response recorded → ${filepath}`);\n            } catch (err) {\n              const msg = err instanceof Error ? err.message : \"Unknown filesystem error\";\n              logger.error(\n                `Failed to save AG-UI fixture to disk: ${msg} (fixture retained in memory)`,\n              );\n            }\n          } else {\n            logger.info(\"Proxied AG-UI request (proxy-only mode)\");\n          }\n\n          resolve();\n        });\n      },\n    );\n\n    upstreamReq.on(\"timeout\", () => {\n      if (!clientRes.writableEnded) clientRes.end();\n      upstreamReq.destroy(\n        new Error(`Upstream AG-UI request timed out after ${UPSTREAM_TIMEOUT_MS / 1000}s`),\n      );\n    });\n\n    upstreamReq.on(\"error\", (err) => {\n      if (!clientRes.headersSent) {\n        clientRes.writeHead(502, { \"Content-Type\": \"application/json\" });\n        clientRes.end(JSON.stringify({ error: \"Upstream AG-UI agent unreachable\" }));\n      } else if (!clientRes.writableEnded) {\n        clientRes.end();\n      }\n      reject(err);\n    });\n\n    upstreamReq.write(body);\n    upstreamReq.end();\n  });\n}\n\n/**\n * Parse SSE data lines from buffered stream text.\n */\nfunction parseSSEEvents(text: string, logger?: Logger): AGUIEvent[] {\n  const events: AGUIEvent[] = [];\n  const blocks = text.split(\"\\n\\n\");\n  for (const block of blocks) {\n    const lines = block.split(\"\\n\");\n    for (const line of lines) {\n      if (line.startsWith(\"data: \")) {\n        try {\n          const parsed = JSON.parse(line.slice(6)) as AGUIEvent;\n          events.push(parsed);\n        } catch {\n          logger?.warn(`Skipping unparseable SSE data line: ${line.slice(0, 200)}`);\n        }\n      }\n    }\n  }\n  return events;\n}\n"],"mappings":";;;;;;;;;;;;;;;AAgBA,eAAsB,mBACpB,KACA,KACA,OACA,UACA,QACA,QACkB;AAClB,KAAI,CAAC,OAAO,UAAU;AACpB,SAAO,KAAK,gEAAgE;AAC5E,SAAO;;CAGT,IAAI;AACJ,KAAI;AACF,WAAS,IAAI,IAAI,OAAO,SAAS;SAC3B;AACN,SAAO,MAAM,+BAA+B,OAAO,WAAW;AAC9D,MAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,MAAI,IAAI,KAAK,UAAU,EAAE,OAAO,8BAA8B,CAAC,CAAC;AAChE,SAAO;;AAGT,QAAO,KAAK,wCAAwC,OAAO,WAAW;CAGtE,MAAM,iBAAyC;EAC7C,gBAAgB;EAChB,QAAQ;EACT;CAED,MAAM,gBAAgB,IAAI,QAAQ;AAClC,KAAI,cACF,gBAAe,mBAAmB,MAAM,QAAQ,cAAc,GAC1D,cAAc,KAAK,KAAK,GACxB;CAEN,MAAM,SAAS,IAAI,QAAQ;AAC3B,KAAI,OACF,gBAAe,eAAe,MAAM,QAAQ,OAAO,GAAG,OAAO,KAAK,KAAK,GAAG;CAG5E,MAAM,cAAc,KAAK,UAAU,MAAM;AAEzC,KAAI;AACF,QAAM,kBACJ,QACA,gBACA,aACA,KACA,OACA,UACA,QACA,OACD;UACM,KAAK;EACZ,MAAM,MAAM,eAAe,QAAQ,IAAI,UAAU;AACjD,SAAO,MAAM,+BAA+B,MAAM;AAClD,MAAI,CAAC,IAAI,aAAa;AACpB,OAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,OAAI,IAAI,KAAK,UAAU,EAAE,OAAO,oCAAoC,CAAC,CAAC;;;AAI1E,QAAO;;AAOT,SAAS,kBACP,QACA,SACA,MACA,WACA,OACA,UACA,QACA,QACe;AACf,QAAO,IAAI,SAAS,SAAS,WAAW;EACtC,MAAM,YAAY,OAAO,aAAa,WAAW,QAAQ;EACzD,MAAM,sBAAsB;EAE5B,MAAM,cAAc,UAAU,QAC5B,QACA;GACE,QAAQ;GACR,SAAS;GACT,SAAS;IACP,GAAG;IACH,kBAAkB,OAAO,WAAW,KAAK,CAAC,UAAU;IACrD;GACF,GACA,gBAAgB;AAEf,OAAI,CAAC,UAAU,YACb,WAAU,UAAU,YAAY,cAAc,KAAK;IACjD,gBAAgB;IAChB,iBAAiB;IACjB,YAAY;IACb,CAAC;GAGJ,MAAM,SAAmB,EAAE;AAE3B,eAAY,GAAG,SAAS,UAAkB;AAExC,QAAI;AACF,eAAU,MAAM,MAAM;YAChB;AAIR,WAAO,KAAK,MAAM;KAClB;AAEF,eAAY,GAAG,UAAU,QAAQ;AAC/B,QAAI,CAAC,UAAU,aAAa;AAC1B,eAAU,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAChE,eAAU,IAAI,KAAK,UAAU,EAAE,OAAO,oCAAoC,CAAC,CAAC;eACnE,CAAC,UAAU,cACpB,WAAU,KAAK;AAEjB,WAAO,IAAI;KACX;AAEF,eAAY,GAAG,aAAa;AAC1B,QAAI,CAAC,UAAU,cAAe,WAAU,KAAK;IAI7C,MAAM,SAAS,eADE,OAAO,OAAO,OAAO,CAAC,UAAU,EACT,OAAO;IAG/C,MAAM,UAAU,uBAAuB,MAAM;AAC7C,QAAI,CAAC,QACH,QAAO,KAAK,2EAA2E;IAEzF,MAAM,UAAuB;KAC3B,OAAO,EAAE,SAAS,WAAW,QAAW;KACxC;KACD;AAED,QAAI,CAAC,OAAO,WAAW;AAErB,cAAS,KAAK,QAAQ;KAGtB,MAAM,cAAc,OAAO,eAAe;KAE1C,MAAM,WAAW,yBADC,IAAI,MAAM,EAAC,aAAa,CAAC,QAAQ,SAAS,IAAI,CAC7B,GAAG,OAAO,YAAY,CAAC,MAAM,GAAG,EAAE,CAAC;KACtE,MAAM,WAAW,KAAK,KAAK,aAAa,SAAS;AAEjD,SAAI;AACF,SAAG,UAAU,aAAa,EAAE,WAAW,MAAM,CAAC;AAC9C,SAAG,cACD,UACA,KAAK,UACH,EAAE,UAAU,CAAC;OAAE,OAAO,QAAQ;OAAO,QAAQ,QAAQ;OAAQ,CAAC,EAAE,EAChE,MACA,EACD,EACD,QACD;AACD,aAAO,KAAK,6BAA6B,WAAW;cAC7C,KAAK;MACZ,MAAM,MAAM,eAAe,QAAQ,IAAI,UAAU;AACjD,aAAO,MACL,yCAAyC,IAAI,+BAC9C;;UAGH,QAAO,KAAK,0CAA0C;AAGxD,aAAS;KACT;IAEL;AAED,cAAY,GAAG,iBAAiB;AAC9B,OAAI,CAAC,UAAU,cAAe,WAAU,KAAK;AAC7C,eAAY,wBACV,IAAI,MAAM,0CAA0C,sBAAsB,IAAK,GAAG,CACnF;IACD;AAEF,cAAY,GAAG,UAAU,QAAQ;AAC/B,OAAI,CAAC,UAAU,aAAa;AAC1B,cAAU,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAChE,cAAU,IAAI,KAAK,UAAU,EAAE,OAAO,oCAAoC,CAAC,CAAC;cACnE,CAAC,UAAU,cACpB,WAAU,KAAK;AAEjB,UAAO,IAAI;IACX;AAEF,cAAY,MAAM,KAAK;AACvB,cAAY,KAAK;GACjB;;;;;AAMJ,SAAS,eAAe,MAAc,QAA8B;CAClE,MAAM,SAAsB,EAAE;CAC9B,MAAM,SAAS,KAAK,MAAM,OAAO;AACjC,MAAK,MAAM,SAAS,QAAQ;EAC1B,MAAM,QAAQ,MAAM,MAAM,KAAK;AAC/B,OAAK,MAAM,QAAQ,MACjB,KAAI,KAAK,WAAW,SAAS,CAC3B,KAAI;GACF,MAAM,SAAS,KAAK,MAAM,KAAK,MAAM,EAAE,CAAC;AACxC,UAAO,KAAK,OAAO;UACb;AACN,WAAQ,KAAK,uCAAuC,KAAK,MAAM,GAAG,IAAI,GAAG;;;AAKjF,QAAO"}