@copilotkit/aimock 1.26.1 → 1.27.1

package/dist/agui-mock.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"agui-mock.cjs","names":["Logger","buildTextResponse","buildToolCallResponse","buildStateUpdate","buildReasoningResponse","readBody","findFixture","writeAGUIEventStream","proxyAndRecordAGUI","http","flattenHeaders"],"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, type LogLevel } 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((options?.logLevel as LogLevel) ?? \"warn\");\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    let body: string;\n    try {\n      body = await readBody(req);\n    } catch (err) {\n      res.writeHead(400, { \"Content-Type\": \"application/json\" });\n      const detail = err instanceof Error ? err.message : \"body read failed\";\n      res.end(JSON.stringify({ error: `Failed to read request body: ${detail}` }));\n      this.journalRequest(req, pathname, 400);\n      return true;\n    }\n\n    let input: AGUIRunAgentInput;\n    try {\n      input = JSON.parse(body) as AGUIRunAgentInput;\n    } catch (err) {\n      res.writeHead(400, { \"Content-Type\": \"application/json\" });\n      const detail = err instanceof Error ? err.message : \"unknown parse error\";\n      res.end(JSON.stringify({ error: `Invalid JSON body: ${detail}` }));\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 result = await proxyAndRecordAGUI(\n        req,\n        res,\n        input,\n        this.fixtures,\n        this.recordConfig,\n        this.logger,\n      );\n      if (result !== false) {\n        this.journalRequest(req, pathname, result);\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,IAAIA,sBAAQ,SAAS,YAAyB,OAAO;;CAKrE,WAAW,SAA4B;AACrC,OAAK,SAAS,KAAK,QAAQ;AAC3B,SAAO;;CAGT,UAAU,SAA0B,MAAc,MAAmC;EACnF,MAAM,SAASC,uCAAkB,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,SAASC,2CAAsB,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,SAASC,sCAAiB,SAAS;AACzC,OAAK,SAAS,KAAK;GACjB,OAAO,EAAE,UAAU,KAAK;GACxB;GACA;GACD,CAAC;AACF,SAAO;;CAGT,YAAY,SAA0B,MAAc,MAAmC;EACrF,MAAM,SAASC,4CAAuB,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,IAAI;AACJ,MAAI;AACF,UAAO,MAAMC,yBAAS,IAAI;WACnB,KAAK;AACZ,OAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;GAC1D,MAAM,SAAS,eAAe,QAAQ,IAAI,UAAU;AACpD,OAAI,IAAI,KAAK,UAAU,EAAE,OAAO,gCAAgC,UAAU,CAAC,CAAC;AAC5E,QAAK,eAAe,KAAK,UAAU,IAAI;AACvC,UAAO;;EAGT,IAAI;AACJ,MAAI;AACF,WAAQ,KAAK,MAAM,KAAK;WACjB,KAAK;AACZ,OAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;GAC1D,MAAM,SAAS,eAAe,QAAQ,IAAI,UAAU;AACpD,OAAI,IAAI,KAAK,UAAU,EAAE,OAAO,sBAAsB,UAAU,CAAC,CAAC;AAClE,QAAK,eAAe,KAAK,UAAU,IAAI;AACvC,UAAO;;EAGT,MAAM,UAAUC,iCAAY,OAAO,KAAK,SAAS;AAEjD,MAAI,SAAS;AACX,SAAMC,0CAAqB,KAAK,QAAQ,QAAQ,EAAE,SAAS,QAAQ,SAAS,CAAC;AAC7E,QAAK,eAAe,KAAK,UAAU,IAAI;AACvC,UAAO;;AAIT,MAAI,KAAK,cAAc;GACrB,MAAM,SAAS,MAAMC,yCACnB,KACA,KACA,OACA,KAAK,UACL,KAAK,cACL,KAAK,OACN;AACD,OAAI,WAAW,OAAO;AACpB,SAAK,eAAe,KAAK,UAAU,OAAO;AAC1C,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,MAAMC,UAAK,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,SAASC,+BAAe,IAAI,QAAQ;GACpC,MAAM;GACN,SAAS;GACT,UAAU;IAAE;IAAQ,SAAS;IAAM;GACpC,CAAC"}
+{"version":3,"file":"agui-mock.cjs","names":["Logger","buildTextResponse","buildToolCallResponse","buildStateUpdate","buildReasoningResponse","readBody","findFixture","writeAGUIEventStream","proxyAndRecordAGUI","http","flattenHeaders"],"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, type LogLevel } 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((options?.logLevel as LogLevel) ?? \"warn\");\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  onToolResult(toolCallId: string, events: AGUIEvent[], delayMs?: number): this {\n    this.fixtures.push({\n      match: { toolCallId },\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    let body: string;\n    try {\n      body = await readBody(req);\n    } catch (err) {\n      res.writeHead(400, { \"Content-Type\": \"application/json\" });\n      const detail = err instanceof Error ? err.message : \"body read failed\";\n      res.end(JSON.stringify({ error: `Failed to read request body: ${detail}` }));\n      this.journalRequest(req, pathname, 400);\n      return true;\n    }\n\n    let input: AGUIRunAgentInput;\n    try {\n      input = JSON.parse(body) as AGUIRunAgentInput;\n    } catch (err) {\n      res.writeHead(400, { \"Content-Type\": \"application/json\" });\n      const detail = err instanceof Error ? err.message : \"unknown parse error\";\n      res.end(JSON.stringify({ error: `Invalid JSON body: ${detail}` }));\n      this.journalRequest(req, pathname, 400);\n      return true;\n    }\n\n    if (input.messages !== undefined && !Array.isArray(input.messages)) {\n      res.writeHead(400, { \"Content-Type\": \"application/json\" });\n      res.end(\n        JSON.stringify({\n          error: \"Invalid input: 'messages' must be an array when provided\",\n        }),\n      );\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 result = await proxyAndRecordAGUI(\n        req,\n        res,\n        input,\n        this.fixtures,\n        this.recordConfig,\n        this.logger,\n      );\n      if (result !== false) {\n        this.journalRequest(req, pathname, result);\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,IAAIA,sBAAQ,SAAS,YAAyB,OAAO;;CAKrE,WAAW,SAA4B;AACrC,OAAK,SAAS,KAAK,QAAQ;AAC3B,SAAO;;CAGT,UAAU,SAA0B,MAAc,MAAmC;EACnF,MAAM,SAASC,uCAAkB,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,SAASC,2CAAsB,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,SAASC,sCAAiB,SAAS;AACzC,OAAK,SAAS,KAAK;GACjB,OAAO,EAAE,UAAU,KAAK;GACxB;GACA;GACD,CAAC;AACF,SAAO;;CAGT,YAAY,SAA0B,MAAc,MAAmC;EACrF,MAAM,SAASC,4CAAuB,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,aAAa,YAAoB,QAAqB,SAAwB;AAC5E,OAAK,SAAS,KAAK;GACjB,OAAO,EAAE,YAAY;GACrB;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,IAAI;AACJ,MAAI;AACF,UAAO,MAAMC,yBAAS,IAAI;WACnB,KAAK;AACZ,OAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;GAC1D,MAAM,SAAS,eAAe,QAAQ,IAAI,UAAU;AACpD,OAAI,IAAI,KAAK,UAAU,EAAE,OAAO,gCAAgC,UAAU,CAAC,CAAC;AAC5E,QAAK,eAAe,KAAK,UAAU,IAAI;AACvC,UAAO;;EAGT,IAAI;AACJ,MAAI;AACF,WAAQ,KAAK,MAAM,KAAK;WACjB,KAAK;AACZ,OAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;GAC1D,MAAM,SAAS,eAAe,QAAQ,IAAI,UAAU;AACpD,OAAI,IAAI,KAAK,UAAU,EAAE,OAAO,sBAAsB,UAAU,CAAC,CAAC;AAClE,QAAK,eAAe,KAAK,UAAU,IAAI;AACvC,UAAO;;AAGT,MAAI,MAAM,aAAa,UAAa,CAAC,MAAM,QAAQ,MAAM,SAAS,EAAE;AAClE,OAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,OAAI,IACF,KAAK,UAAU,EACb,OAAO,4DACR,CAAC,CACH;AACD,QAAK,eAAe,KAAK,UAAU,IAAI;AACvC,UAAO;;EAGT,MAAM,UAAUC,iCAAY,OAAO,KAAK,SAAS;AAEjD,MAAI,SAAS;AACX,SAAMC,0CAAqB,KAAK,QAAQ,QAAQ,EAAE,SAAS,QAAQ,SAAS,CAAC;AAC7E,QAAK,eAAe,KAAK,UAAU,IAAI;AACvC,UAAO;;AAIT,MAAI,KAAK,cAAc;GACrB,MAAM,SAAS,MAAMC,yCACnB,KACA,KACA,OACA,KAAK,UACL,KAAK,cACL,KAAK,OACN;AACD,OAAI,WAAW,OAAO;AACpB,SAAK,eAAe,KAAK,UAAU,OAAO;AAC1C,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,MAAMC,UAAK,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,SAASC,+BAAe,IAAI,QAAQ;GACpC,MAAM;GACN,SAAS;GACT,UAAU;IAAE;IAAQ,SAAS;IAAM;GACpC,CAAC"}

package/dist/agui-mock.d.cts

@@ -29,6 +29,7 @@ declare class AGUIMock implements Mountable {
     delayMs?: number;
   }): this;
   onPredicate(predicate: (input: AGUIRunAgentInput) => boolean, events: AGUIEvent[], delayMs?: number): this;
+  onToolResult(toolCallId: string, events: AGUIEvent[], delayMs?: number): this;
   enableRecording(config: AGUIRecordConfig): this;
   reset(): this;
   handleRequest(req: http$1.IncomingMessage, res: http$1.ServerResponse, pathname: string): Promise<boolean>;

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

@@ -1 +1 @@
-{"version":3,"file":"agui-mock.d.cts","names":[],"sources":["../src/agui-mock.ts"],"sourcesContent":[],"mappings":";;;;;;;cAuBa,QAAA,YAAoB;;EAApB,QAAA,MAAS;EAAA,QAAA,OAAA;UAUE,QAAA;UAOF,OAAA;UAKQ,OAAA;UAUJ,YAAA;UAAgB,MAAA;aAUpB,CAAA,OAAA,CAAA,EAhCE,eAgCF;YAgBc,CAAA,OAAA,EAzCd,WAyCc,CAAA,EAAA,IAAA;WAUJ,CAAA,OAAA,EAAA,MAAA,GA9CF,MA8CE,EAAA,IAAA,EAAA,MAAA,EAAA,KAAA,EAAA;IAWT,OAAA,CAAA,EAAA,MAAA;MACX,IAAA;OAWc,CAAA,OAAA,EAAA,MAAA,GA3DA,MA2DA,EAAA,MAAA,EA3DgB,SA2DhB,EAAA,EAAA,OAAA,CAAA,EAAA,MAAA,CAAA,EAAA,IAAA;YAcZ,CAAA,OAAA,EAAA,MAAA,GA/DQ,MA+DR,EAAA,QAAA,EAAA,MAAA,EAAA,IAAA,EAAA,MAAA,EAAA,KAAA,EAAA;IACL,MAAK,CAAA,EAAA,MAAA;IAET,OAAA,CAAA,EAAA,MAAA;MAqEiB,IAAA;YAQE,CAAA,GAAA,EAAA,MAAA,EAAA,QAAA,EA/HY,MA+HZ,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,OAAA,CAAA,EAAA,MAAA,CAAA,EAAA,IAAA;aAMP,CAAA,OAAA,EAAA,MAAA,GA3He,MA2Hf,EAAA,IAAA,EAAA,MAAA,EAAA,KAAA,EAAA;IAwCD,OAAA,CAAA,EAAA,MAAA;MAvOiB,IAAA;EAAS,WAAA,CAAA,SAAA,EAAA,CAAA,KAAA,EA+EnB,iBA/EmB,EAAA,GAAA,OAAA,EAAA,MAAA,EAgF9B,SAhF8B,EAAA,EAAA,OAAA,CAAA,EAAA,MAAA,CAAA,EAAA,IAAA;0BA2FhB;;qBAcjB,MAAA,CAAK,sBACL,MAAA,CAAK,mCAET;;;;;sBAqEiB;;wBAQE;WAMP;UAwCD"}
+{"version":3,"file":"agui-mock.d.cts","names":[],"sources":["../src/agui-mock.ts"],"sourcesContent":[],"mappings":";;;;;;;cAuBa,QAAA,YAAoB;;EAApB,QAAA,MAAS;EAAA,QAAA,OAAA;UAUE,QAAA;UAOF,OAAA;UAKQ,OAAA;UAUJ,YAAA;UAAgB,MAAA;aAUpB,CAAA,OAAA,CAAA,EAhCE,eAgCF;YAgBc,CAAA,OAAA,EAzCd,WAyCc,CAAA,EAAA,IAAA;WAUJ,CAAA,OAAA,EAAA,MAAA,GA9CF,MA8CE,EAAA,IAAA,EAAA,MAAA,EAAA,KAAA,EAAA;IAWT,OAAA,CAAA,EAAA,MAAA;MACX,IAAA;OAW+B,CAAA,OAAA,EAAA,MAAA,GA3DjB,MA2DiB,EAAA,MAAA,EA3DD,SA2DC,EAAA,EAAA,OAAA,CAAA,EAAA,MAAA,CAAA,EAAA,IAAA;YASjB,CAAA,OAAA,EAAA,MAAA,GA1DJ,MA0DI,EAAA,QAAA,EAAA,MAAA,EAAA,IAAA,EAAA,MAAA,EAAA,KAAA,EAAA;IAcjB,MAAK,CAAA,EAAA,MAAA;IACL,OAAK,CAAA,EAAA,MAAA;MAET,IAAA;YAgFiB,CAAA,GAAA,EAAA,MAAA,EAAA,QAAA,EA3Ic,MA2Id,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,OAAA,CAAA,EAAA,MAAA,CAAA,EAAA,IAAA;aAQE,CAAA,OAAA,EAAA,MAAA,GAzIQ,MAyIR,EAAA,IAAA,EAAA,MAAA,EAAA,KAAA,EAAA;IAMP,OAAA,CAAA,EAAA,MAAA;MAwCD,IAAA;aA3PiB,CAAA,SAAA,EAAA,CAAA,KAAA,EA+EV,iBA/EU,EAAA,GAAA,OAAA,EAAA,MAAA,EAgFrB,SAhFqB,EAAA,EAAA,OAAA,CAAA,EAAA,MAAA,CAAA,EAAA,IAAA;EAAS,YAAA,CAAA,UAAA,EAAA,MAAA,EAAA,MAAA,EA2FC,SA3FD,EAAA,EAAA,OAAA,CAAA,EAAA,MAAA,CAAA,EAAA,IAAA;0BAoGhB;;qBAcjB,MAAA,CAAK,sBACL,MAAA,CAAK,mCAET;;;;;sBAgFiB;;wBAQE;WAMP;UAwCD"}

package/dist/agui-mock.d.ts

@@ -29,6 +29,7 @@ declare class AGUIMock implements Mountable {
     delayMs?: number;
   }): this;
   onPredicate(predicate: (input: AGUIRunAgentInput) => boolean, events: AGUIEvent[], delayMs?: number): this;
+  onToolResult(toolCallId: string, events: AGUIEvent[], delayMs?: number): this;
   enableRecording(config: AGUIRecordConfig): this;
   reset(): this;
   handleRequest(req: http$1.IncomingMessage, res: http$1.ServerResponse, pathname: string): Promise<boolean>;

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

@@ -1 +1 @@
-{"version":3,"file":"agui-mock.d.ts","names":[],"sources":["../src/agui-mock.ts"],"sourcesContent":[],"mappings":";;;;;;;cAuBa,QAAA,YAAoB;;EAApB,QAAA,MAAS;EAAA,QAAA,OAAA;UAUE,QAAA;UAOF,OAAA;UAKQ,OAAA;UAUJ,YAAA;UAAgB,MAAA;aAUpB,CAAA,OAAA,CAAA,EAhCE,eAgCF;YAgBc,CAAA,OAAA,EAzCd,WAyCc,CAAA,EAAA,IAAA;WAUJ,CAAA,OAAA,EAAA,MAAA,GA9CF,MA8CE,EAAA,IAAA,EAAA,MAAA,EAAA,KAAA,EAAA;IAWT,OAAA,CAAA,EAAA,MAAA;MACX,IAAA;OAWc,CAAA,OAAA,EAAA,MAAA,GA3DA,MA2DA,EAAA,MAAA,EA3DgB,SA2DhB,EAAA,EAAA,OAAA,CAAA,EAAA,MAAA,CAAA,EAAA,IAAA;YAcZ,CAAA,OAAA,EAAA,MAAA,GA/DQ,MA+DR,EAAA,QAAA,EAAA,MAAA,EAAA,IAAA,EAAA,MAAA,EAAA,KAAA,EAAA;IACL,MAAK,CAAA,EAAA,MAAA;IAET,OAAA,CAAA,EAAA,MAAA;MAqEiB,IAAA;YAQE,CAAA,GAAA,EAAA,MAAA,EAAA,QAAA,EA/HY,MA+HZ,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,OAAA,CAAA,EAAA,MAAA,CAAA,EAAA,IAAA;aAMP,CAAA,OAAA,EAAA,MAAA,GA3He,MA2Hf,EAAA,IAAA,EAAA,MAAA,EAAA,KAAA,EAAA;IAwCD,OAAA,CAAA,EAAA,MAAA;MAvOiB,IAAA;EAAS,WAAA,CAAA,SAAA,EAAA,CAAA,KAAA,EA+EnB,iBA/EmB,EAAA,GAAA,OAAA,EAAA,MAAA,EAgF9B,SAhF8B,EAAA,EAAA,OAAA,CAAA,EAAA,MAAA,CAAA,EAAA,IAAA;0BA2FhB;;qBAcjB,MAAA,CAAK,sBACL,MAAA,CAAK,mCAET;;;;;sBAqEiB;;wBAQE;WAMP;UAwCD"}
+{"version":3,"file":"agui-mock.d.ts","names":[],"sources":["../src/agui-mock.ts"],"sourcesContent":[],"mappings":";;;;;;;cAuBa,QAAA,YAAoB;;EAApB,QAAA,MAAS;EAAA,QAAA,OAAA;UAUE,QAAA;UAOF,OAAA;UAKQ,OAAA;UAUJ,YAAA;UAAgB,MAAA;aAUpB,CAAA,OAAA,CAAA,EAhCE,eAgCF;YAgBc,CAAA,OAAA,EAzCd,WAyCc,CAAA,EAAA,IAAA;WAUJ,CAAA,OAAA,EAAA,MAAA,GA9CF,MA8CE,EAAA,IAAA,EAAA,MAAA,EAAA,KAAA,EAAA;IAWT,OAAA,CAAA,EAAA,MAAA;MACX,IAAA;OAW+B,CAAA,OAAA,EAAA,MAAA,GA3DjB,MA2DiB,EAAA,MAAA,EA3DD,SA2DC,EAAA,EAAA,OAAA,CAAA,EAAA,MAAA,CAAA,EAAA,IAAA;YASjB,CAAA,OAAA,EAAA,MAAA,GA1DJ,MA0DI,EAAA,QAAA,EAAA,MAAA,EAAA,IAAA,EAAA,MAAA,EAAA,KAAA,EAAA;IAcjB,MAAK,CAAA,EAAA,MAAA;IACL,OAAK,CAAA,EAAA,MAAA;MAET,IAAA;YAgFiB,CAAA,GAAA,EAAA,MAAA,EAAA,QAAA,EA3Ic,MA2Id,CAAA,MAAA,EAAA,OAAA,CAAA,EAAA,OAAA,CAAA,EAAA,MAAA,CAAA,EAAA,IAAA;aAQE,CAAA,OAAA,EAAA,MAAA,GAzIQ,MAyIR,EAAA,IAAA,EAAA,MAAA,EAAA,KAAA,EAAA;IAMP,OAAA,CAAA,EAAA,MAAA;MAwCD,IAAA;aA3PiB,CAAA,SAAA,EAAA,CAAA,KAAA,EA+EV,iBA/EU,EAAA,GAAA,OAAA,EAAA,MAAA,EAgFrB,SAhFqB,EAAA,EAAA,OAAA,CAAA,EAAA,MAAA,CAAA,EAAA,IAAA;EAAS,YAAA,CAAA,UAAA,EAAA,MAAA,EAAA,MAAA,EA2FC,SA3FD,EAAA,EAAA,OAAA,CAAA,EAAA,MAAA,CAAA,EAAA,IAAA;0BAoGhB;;qBAcjB,MAAA,CAAK,sBACL,MAAA,CAAK,mCAET;;;;;sBAgFiB;;wBAQE;WAMP;UAwCD"}

package/dist/agui-mock.js

@@ -74,6 +74,14 @@ var AGUIMock = class {
 		});
 		return this;
 	}
+	onToolResult(toolCallId, events, delayMs) {
+		this.fixtures.push({
+			match: { toolCallId },
+			events,
+			delayMs
+		});
+		return this;
+	}
 	enableRecording(config) {
 		this.recordConfig = config;
 		return this;
@@ -106,6 +114,12 @@ var AGUIMock = class {
 			this.journalRequest(req, pathname, 400);
 			return true;
 		}
+		if (input.messages !== void 0 && !Array.isArray(input.messages)) {
+			res.writeHead(400, { "Content-Type": "application/json" });
+			res.end(JSON.stringify({ error: "Invalid input: 'messages' must be an array when provided" }));
+			this.journalRequest(req, pathname, 400);
+			return true;
+		}
 		const fixture = findFixture(input, this.fixtures);
 		if (fixture) {
 			await writeAGUIEventStream(res, fixture.events, { delayMs: fixture.delayMs });

package/dist/agui-mock.js.map

@@ -1 +1 @@
-{"version":3,"file":"agui-mock.js","names":["http"],"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, type LogLevel } 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((options?.logLevel as LogLevel) ?? \"warn\");\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    let body: string;\n    try {\n      body = await readBody(req);\n    } catch (err) {\n      res.writeHead(400, { \"Content-Type\": \"application/json\" });\n      const detail = err instanceof Error ? err.message : \"body read failed\";\n      res.end(JSON.stringify({ error: `Failed to read request body: ${detail}` }));\n      this.journalRequest(req, pathname, 400);\n      return true;\n    }\n\n    let input: AGUIRunAgentInput;\n    try {\n      input = JSON.parse(body) as AGUIRunAgentInput;\n    } catch (err) {\n      res.writeHead(400, { \"Content-Type\": \"application/json\" });\n      const detail = err instanceof Error ? err.message : \"unknown parse error\";\n      res.end(JSON.stringify({ error: `Invalid JSON body: ${detail}` }));\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 result = await proxyAndRecordAGUI(\n        req,\n        res,\n        input,\n        this.fixtures,\n        this.recordConfig,\n        this.logger,\n      );\n      if (result !== false) {\n        this.journalRequest(req, pathname, result);\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,OAAQ,SAAS,YAAyB,OAAO;;CAKrE,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,IAAI;AACJ,MAAI;AACF,UAAO,MAAM,SAAS,IAAI;WACnB,KAAK;AACZ,OAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;GAC1D,MAAM,SAAS,eAAe,QAAQ,IAAI,UAAU;AACpD,OAAI,IAAI,KAAK,UAAU,EAAE,OAAO,gCAAgC,UAAU,CAAC,CAAC;AAC5E,QAAK,eAAe,KAAK,UAAU,IAAI;AACvC,UAAO;;EAGT,IAAI;AACJ,MAAI;AACF,WAAQ,KAAK,MAAM,KAAK;WACjB,KAAK;AACZ,OAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;GAC1D,MAAM,SAAS,eAAe,QAAQ,IAAI,UAAU;AACpD,OAAI,IAAI,KAAK,UAAU,EAAE,OAAO,sBAAsB,UAAU,CAAC,CAAC;AAClE,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,cAAc;GACrB,MAAM,SAAS,MAAM,mBACnB,KACA,KACA,OACA,KAAK,UACL,KAAK,cACL,KAAK,OACN;AACD,OAAI,WAAW,OAAO;AACpB,SAAK,eAAe,KAAK,UAAU,OAAO;AAC1C,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,MAAMA,OAAK,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"}
+{"version":3,"file":"agui-mock.js","names":["http"],"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, type LogLevel } 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((options?.logLevel as LogLevel) ?? \"warn\");\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  onToolResult(toolCallId: string, events: AGUIEvent[], delayMs?: number): this {\n    this.fixtures.push({\n      match: { toolCallId },\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    let body: string;\n    try {\n      body = await readBody(req);\n    } catch (err) {\n      res.writeHead(400, { \"Content-Type\": \"application/json\" });\n      const detail = err instanceof Error ? err.message : \"body read failed\";\n      res.end(JSON.stringify({ error: `Failed to read request body: ${detail}` }));\n      this.journalRequest(req, pathname, 400);\n      return true;\n    }\n\n    let input: AGUIRunAgentInput;\n    try {\n      input = JSON.parse(body) as AGUIRunAgentInput;\n    } catch (err) {\n      res.writeHead(400, { \"Content-Type\": \"application/json\" });\n      const detail = err instanceof Error ? err.message : \"unknown parse error\";\n      res.end(JSON.stringify({ error: `Invalid JSON body: ${detail}` }));\n      this.journalRequest(req, pathname, 400);\n      return true;\n    }\n\n    if (input.messages !== undefined && !Array.isArray(input.messages)) {\n      res.writeHead(400, { \"Content-Type\": \"application/json\" });\n      res.end(\n        JSON.stringify({\n          error: \"Invalid input: 'messages' must be an array when provided\",\n        }),\n      );\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 result = await proxyAndRecordAGUI(\n        req,\n        res,\n        input,\n        this.fixtures,\n        this.recordConfig,\n        this.logger,\n      );\n      if (result !== false) {\n        this.journalRequest(req, pathname, result);\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,OAAQ,SAAS,YAAyB,OAAO;;CAKrE,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,aAAa,YAAoB,QAAqB,SAAwB;AAC5E,OAAK,SAAS,KAAK;GACjB,OAAO,EAAE,YAAY;GACrB;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,IAAI;AACJ,MAAI;AACF,UAAO,MAAM,SAAS,IAAI;WACnB,KAAK;AACZ,OAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;GAC1D,MAAM,SAAS,eAAe,QAAQ,IAAI,UAAU;AACpD,OAAI,IAAI,KAAK,UAAU,EAAE,OAAO,gCAAgC,UAAU,CAAC,CAAC;AAC5E,QAAK,eAAe,KAAK,UAAU,IAAI;AACvC,UAAO;;EAGT,IAAI;AACJ,MAAI;AACF,WAAQ,KAAK,MAAM,KAAK;WACjB,KAAK;AACZ,OAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;GAC1D,MAAM,SAAS,eAAe,QAAQ,IAAI,UAAU;AACpD,OAAI,IAAI,KAAK,UAAU,EAAE,OAAO,sBAAsB,UAAU,CAAC,CAAC;AAClE,QAAK,eAAe,KAAK,UAAU,IAAI;AACvC,UAAO;;AAGT,MAAI,MAAM,aAAa,UAAa,CAAC,MAAM,QAAQ,MAAM,SAAS,EAAE;AAClE,OAAI,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAC1D,OAAI,IACF,KAAK,UAAU,EACb,OAAO,4DACR,CAAC,CACH;AACD,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,cAAc;GACrB,MAAM,SAAS,MAAM,mBACnB,KACA,KACA,OACA,KAAK,UACL,KAAK,cACL,KAAK,OACN;AACD,OAAI,WAAW,OAAO;AACpB,SAAK,eAAe,KAAK,UAAU,OAAO;AAC1C,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,MAAMA,OAAK,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

@@ -95,7 +95,10 @@ function teeUpstreamStream(target, headers, body, clientRes, input, fixtures, co
 				}
 				chunks.push(chunk);
 			});
+			let settled = false;
 			upstreamRes.on("error", (err) => {
+				if (settled) return;
+				settled = true;
 				try {
 					if (!clientRes.headersSent) {
 						clientRes.writeHead(502, { "Content-Type": "application/json" });
@@ -107,35 +110,60 @@ function teeUpstreamStream(target, headers, body, clientRes, input, fixtures, co
 				reject(err);
 			});
 			upstreamRes.on("end", () => {
+				if (settled) return;
+				settled = true;
+				if (clientStatus !== 200) {
+					try {
+						if (!clientRes.writableEnded) clientRes.end();
+					} catch (writeErr) {
+						logger.warn("Failed to end client response:", writeErr instanceof Error ? writeErr.message : String(writeErr));
+					}
+					resolve(clientStatus);
+					return;
+				}
 				try {
 					if (!clientRes.writableEnded) clientRes.end();
 				} catch (writeErr) {
 					logger.warn("Failed to end client response:", writeErr instanceof Error ? writeErr.message : String(writeErr));
 				}
 				const events = parseSSEEvents(Buffer.concat(chunks).toString(), logger);
-				const message = require_agui_handler.extractLastUserMessage(input);
+				let match;
+				const lastToolResult = require_agui_handler.getLastMessageIfToolResult(input);
+				if (lastToolResult?.toolCallId) {
+					match = { toolCallId: lastToolResult.toolCallId };
+					logger.info(`Recorded AG-UI fixture keyed on toolCallId=${lastToolResult.toolCallId}`);
+				} else {
+					const message = require_agui_handler.extractLastUserMessage(input);
+					if (message) match = { message };
+					else {
+						match = { predicate: (inp) => !inp.messages?.length || !inp.messages.some((m) => m.role === "user") };
+						logger.warn("Recorded AG-UI fixture has no user message — available in-memory only (predicate fixtures cannot be persisted to disk)");
+					}
+				}
 				const fixture = {
-					match: message ? { message } : { predicate: (inp) => !inp.messages?.length || !inp.messages.some((m) => m.role === "user") },
+					match,
 					events
 				};
-				if (!message) logger.warn("Recorded AG-UI fixture has no user message — will use __NO_USER_MESSAGE__ sentinel on disk");
 				if (!config.proxyOnly) {
 					fixtures.push(fixture);
-					const serializableFixture = {
-						match: fixture.match.predicate ? { message: "__NO_USER_MESSAGE__" } : fixture.match,
-						events: fixture.events,
-						...fixture.delayMs !== void 0 ? { delayMs: fixture.delayMs } : {}
-					};
-					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: [serializableFixture] }, 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)`);
+					if (fixture.match.predicate) logger.warn("Skipping disk write for predicate fixture — in-memory only (cannot be persisted)");
+					else {
+						const serializableFixture = {
+							match: fixture.match,
+							events: fixture.events,
+							...fixture.delayMs !== void 0 ? { delayMs: fixture.delayMs } : {}
+						};
+						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: [serializableFixture] }, 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(clientStatus);
@@ -173,9 +201,9 @@ function parseSSEEvents(text, logger) {
 				const parsed = JSON.parse(payload);
 				events.push(parsed);
 			} catch (err) {
-				const msg = err instanceof Error ? err.message : String(err);
-				if (logger) logger.warn(`Skipping unparseable SSE data line: ${payload.slice(0, 200)}`);
-				else console.warn(`Skipping unparseable SSE data line: ${msg}`);
+				const warning = `Skipping unparseable SSE data line (${err instanceof Error ? err.message : String(err)}): ${payload.slice(0, 200)}`;
+				if (logger) logger.warn(warning);
+				else console.warn(warning);
 			}
 		}
 	}

package/dist/agui-recorder.cjs.map

@@ -1 +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 the HTTP status code written to the client if the request was proxied,\n * or `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<number | false> {\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 (err) {\n    const detail = err instanceof Error ? err.message : String(err);\n    logger.error(`Invalid upstream AG-UI URL: ${config.upstream} — ${detail}`);\n    res.writeHead(502, { \"Content-Type\": \"application/json\" });\n    res.end(JSON.stringify({ error: \"Invalid upstream AG-UI URL\" }));\n    return 502;\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  let status: number;\n  try {\n    status = 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    } else if (!res.writableEnded) {\n      res.end();\n    }\n    status = 502;\n  }\n\n  return status;\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<number> {\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        const upstreamStatus = upstreamRes.statusCode ?? 200;\n\n        // Normalize status codes: aimock acts as a gateway, so upstream\n        // provider details (429, 503, etc.) should not leak.\n        // Successes → 200, errors → 502 (Bad Gateway).\n        const clientStatus = upstreamStatus >= 200 && upstreamStatus < 300 ? 200 : 502;\n\n        // Set appropriate headers on the client response.\n        if (!clientRes.headersSent) {\n          if (clientStatus === 200) {\n            clientRes.writeHead(200, {\n              \"Content-Type\": \"text/event-stream\",\n              \"Cache-Control\": \"no-cache\",\n              Connection: \"keep-alive\",\n            });\n          } else {\n            const ct = upstreamRes.headers[\"content-type\"] || \"application/json\";\n            clientRes.writeHead(502, { \"Content-Type\": ct });\n          }\n        }\n\n        const chunks: Buffer[] = [];\n        let clientWriteFailed = false;\n\n        upstreamRes.on(\"data\", (chunk: Buffer) => {\n          // Relay to client in real time\n          try {\n            clientRes.write(chunk);\n          } catch (err) {\n            if (!clientWriteFailed) {\n              clientWriteFailed = true;\n              logger?.warn(\n                \"Client write failed during proxy relay:\",\n                err instanceof Error ? err.message : String(err),\n              );\n            }\n          }\n          // Buffer for fixture construction\n          chunks.push(chunk);\n        });\n\n        upstreamRes.on(\"error\", (err) => {\n          try {\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          } catch (writeErr) {\n            logger.warn(\n              \"Failed to write error response to client:\",\n              writeErr instanceof Error ? writeErr.message : String(writeErr),\n            );\n          }\n          reject(err);\n        });\n\n        upstreamRes.on(\"end\", () => {\n          try {\n            if (!clientRes.writableEnded) clientRes.end();\n          } catch (writeErr) {\n            logger.warn(\n              \"Failed to end client response:\",\n              writeErr instanceof Error ? writeErr.message : String(writeErr),\n            );\n          }\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          const fixture: AGUIFixture = {\n            match: message\n              ? { message }\n              : {\n                  predicate: (inp: AGUIRunAgentInput) =>\n                    !inp.messages?.length || !inp.messages.some((m) => m.role === \"user\"),\n                },\n            events,\n          };\n          if (!message) {\n            logger.warn(\n              \"Recorded AG-UI fixture has no user message — will use __NO_USER_MESSAGE__ sentinel on disk\",\n            );\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 — predicate functions are not serializable,\n            // so replace with a sentinel string that won't match real user messages.\n            const serializableFixture = {\n              match: fixture.match.predicate ? { message: \"__NO_USER_MESSAGE__\" } : fixture.match,\n              events: fixture.events,\n              ...(fixture.delayMs !== undefined ? { delayMs: fixture.delayMs } : {}),\n            };\n\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({ fixtures: [serializableFixture] }, null, 2),\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(clientStatus);\n        });\n      },\n    );\n\n    upstreamReq.on(\"timeout\", () => {\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      try {\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      } catch (writeErr) {\n        logger.warn(\n          \"Failed to write error response to client:\",\n          writeErr instanceof Error ? writeErr.message : String(writeErr),\n        );\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        const payload = line.startsWith(\"data: \") ? line.slice(6) : line.slice(5);\n        try {\n          const parsed = JSON.parse(payload) as AGUIEvent;\n          events.push(parsed);\n        } catch (err) {\n          const msg = err instanceof Error ? err.message : String(err);\n          if (logger) logger.warn(`Skipping unparseable SSE data line: ${payload.slice(0, 200)}`);\n          else console.warn(`Skipping unparseable SSE data line: ${msg}`);\n        }\n      }\n    }\n  }\n  return events;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;AAiBA,eAAsB,mBACpB,KACA,KACA,OACA,UACA,QACA,QACyB;AACzB,KAAI,CAAC,OAAO,UAAU;AACpB,SAAO,KAAK,gEAAgE;AAC5E,SAAO;;CAGT,IAAI;AACJ,KAAI;AACF,WAAS,IAAI,IAAI,OAAO,SAAS;UAC1B,KAAK;EACZ,MAAM,SAAS,eAAe,QAAQ,IAAI,UAAU,OAAO,IAAI;AAC/D,SAAO,MAAM,+BAA+B,OAAO,SAAS,KAAK,SAAS;AAC1E,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;CAEzC,IAAI;AACJ,KAAI;AACF,WAAS,MAAM,kBACb,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;aAC7D,CAAC,IAAI,cACd,KAAI,KAAK;AAEX,WAAS;;AAGX,QAAO;;AAOT,SAAS,kBACP,QACA,SACA,MACA,WACA,OACA,UACA,QACA,QACiB;AACjB,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;GACf,MAAM,iBAAiB,YAAY,cAAc;GAKjD,MAAM,eAAe,kBAAkB,OAAO,iBAAiB,MAAM,MAAM;AAG3E,OAAI,CAAC,UAAU,YACb,KAAI,iBAAiB,IACnB,WAAU,UAAU,KAAK;IACvB,gBAAgB;IAChB,iBAAiB;IACjB,YAAY;IACb,CAAC;QACG;IACL,MAAM,KAAK,YAAY,QAAQ,mBAAmB;AAClD,cAAU,UAAU,KAAK,EAAE,gBAAgB,IAAI,CAAC;;GAIpD,MAAM,SAAmB,EAAE;GAC3B,IAAI,oBAAoB;AAExB,eAAY,GAAG,SAAS,UAAkB;AAExC,QAAI;AACF,eAAU,MAAM,MAAM;aACf,KAAK;AACZ,SAAI,CAAC,mBAAmB;AACtB,0BAAoB;AACpB,cAAQ,KACN,2CACA,eAAe,QAAQ,IAAI,UAAU,OAAO,IAAI,CACjD;;;AAIL,WAAO,KAAK,MAAM;KAClB;AAEF,eAAY,GAAG,UAAU,QAAQ;AAC/B,QAAI;AACF,SAAI,CAAC,UAAU,aAAa;AAC1B,gBAAU,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAChE,gBAAU,IAAI,KAAK,UAAU,EAAE,OAAO,oCAAoC,CAAC,CAAC;gBACnE,CAAC,UAAU,cACpB,WAAU,KAAK;aAEV,UAAU;AACjB,YAAO,KACL,6CACA,oBAAoB,QAAQ,SAAS,UAAU,OAAO,SAAS,CAChE;;AAEH,WAAO,IAAI;KACX;AAEF,eAAY,GAAG,aAAa;AAC1B,QAAI;AACF,SAAI,CAAC,UAAU,cAAe,WAAU,KAAK;aACtC,UAAU;AACjB,YAAO,KACL,kCACA,oBAAoB,QAAQ,SAAS,UAAU,OAAO,SAAS,CAChE;;IAKH,MAAM,SAAS,eADE,OAAO,OAAO,OAAO,CAAC,UAAU,EACT,OAAO;IAG/C,MAAM,UAAUC,4CAAuB,MAAM;IAC7C,MAAM,UAAuB;KAC3B,OAAO,UACH,EAAE,SAAS,GACX,EACE,YAAY,QACV,CAAC,IAAI,UAAU,UAAU,CAAC,IAAI,SAAS,MAAM,MAAM,EAAE,SAAS,OAAO,EACxE;KACL;KACD;AACD,QAAI,CAAC,QACH,QAAO,KACL,6FACD;AAGH,QAAI,CAAC,OAAO,WAAW;AAErB,cAAS,KAAK,QAAQ;KAItB,MAAM,sBAAsB;MAC1B,OAAO,QAAQ,MAAM,YAAY,EAAE,SAAS,uBAAuB,GAAG,QAAQ;MAC9E,QAAQ,QAAQ;MAChB,GAAI,QAAQ,YAAY,SAAY,EAAE,SAAS,QAAQ,SAAS,GAAG,EAAE;MACtE;KAED,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,UAAU,EAAE,UAAU,CAAC,oBAAoB,EAAE,EAAE,MAAM,EAAE,EAC5D,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,YAAQ,aAAa;KACrB;IAEL;AAED,cAAY,GAAG,iBAAiB;AAC9B,eAAY,wBACV,IAAI,MAAM,0CAA0C,sBAAsB,IAAK,GAAG,CACnF;IACD;AAEF,cAAY,GAAG,UAAU,QAAQ;AAC/B,OAAI;AACF,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;YAEV,UAAU;AACjB,WAAO,KACL,6CACA,oBAAoB,QAAQ,SAAS,UAAU,OAAO,SAAS,CAChE;;AAEH,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,QAAQ,EAAE;GAC5B,MAAM,UAAU,KAAK,WAAW,SAAS,GAAG,KAAK,MAAM,EAAE,GAAG,KAAK,MAAM,EAAE;AACzE,OAAI;IACF,MAAM,SAAS,KAAK,MAAM,QAAQ;AAClC,WAAO,KAAK,OAAO;YACZ,KAAK;IACZ,MAAM,MAAM,eAAe,QAAQ,IAAI,UAAU,OAAO,IAAI;AAC5D,QAAI,OAAQ,QAAO,KAAK,uCAAuC,QAAQ,MAAM,GAAG,IAAI,GAAG;QAClF,SAAQ,KAAK,uCAAuC,MAAM;;;;AAKvE,QAAO"}
+{"version":3,"file":"agui-recorder.cjs","names":["https","http","getLastMessageIfToolResult","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 {\n  AGUIFixture,\n  AGUIFixtureMatch,\n  AGUIRecordConfig,\n  AGUIEvent,\n  AGUIRunAgentInput,\n} from \"./agui-types.js\";\nimport { extractLastUserMessage, getLastMessageIfToolResult } from \"./agui-handler.js\";\nimport type { Logger } from \"./logger.js\";\n\n/**\n * Sentinel `match.message` value written to disk when the request had no\n * extractable user text. Keeps the on-disk fixture serializable (predicate\n * matchers aren't) but won't match any real user input on replay.\n */\nexport const NO_USER_MESSAGE_SENTINEL = \"__NO_USER_MESSAGE__\";\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 the HTTP status code written to the client if the request was proxied,\n * or `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<number | false> {\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 (err) {\n    const detail = err instanceof Error ? err.message : String(err);\n    logger.error(`Invalid upstream AG-UI URL: ${config.upstream} — ${detail}`);\n    res.writeHead(502, { \"Content-Type\": \"application/json\" });\n    res.end(JSON.stringify({ error: \"Invalid upstream AG-UI URL\" }));\n    return 502;\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  let status: number;\n  try {\n    status = 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    } else if (!res.writableEnded) {\n      res.end();\n    }\n    status = 502;\n  }\n\n  return status;\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<number> {\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        const upstreamStatus = upstreamRes.statusCode ?? 200;\n\n        // Normalize status codes: aimock acts as a gateway, so upstream\n        // provider details (429, 503, etc.) should not leak.\n        // Successes → 200, errors → 502 (Bad Gateway).\n        const clientStatus = upstreamStatus >= 200 && upstreamStatus < 300 ? 200 : 502;\n\n        // Set appropriate headers on the client response.\n        if (!clientRes.headersSent) {\n          if (clientStatus === 200) {\n            clientRes.writeHead(200, {\n              \"Content-Type\": \"text/event-stream\",\n              \"Cache-Control\": \"no-cache\",\n              Connection: \"keep-alive\",\n            });\n          } else {\n            const ct = upstreamRes.headers[\"content-type\"] || \"application/json\";\n            clientRes.writeHead(502, { \"Content-Type\": ct });\n          }\n        }\n\n        const chunks: Buffer[] = [];\n        let clientWriteFailed = false;\n\n        upstreamRes.on(\"data\", (chunk: Buffer) => {\n          // Relay to client in real time\n          try {\n            clientRes.write(chunk);\n          } catch (err) {\n            if (!clientWriteFailed) {\n              clientWriteFailed = true;\n              logger?.warn(\n                \"Client write failed during proxy relay:\",\n                err instanceof Error ? err.message : String(err),\n              );\n            }\n          }\n          // Buffer for fixture construction\n          chunks.push(chunk);\n        });\n\n        let settled = false;\n\n        upstreamRes.on(\"error\", (err) => {\n          if (settled) return;\n          settled = true;\n          try {\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          } catch (writeErr) {\n            logger.warn(\n              \"Failed to write error response to client:\",\n              writeErr instanceof Error ? writeErr.message : String(writeErr),\n            );\n          }\n          reject(err);\n        });\n\n        upstreamRes.on(\"end\", () => {\n          if (settled) return;\n          settled = true;\n\n          // Don't record fixtures for non-2xx upstream responses\n          if (clientStatus !== 200) {\n            try {\n              if (!clientRes.writableEnded) clientRes.end();\n            } catch (writeErr) {\n              logger.warn(\n                \"Failed to end client response:\",\n                writeErr instanceof Error ? writeErr.message : String(writeErr),\n              );\n            }\n            resolve(clientStatus);\n            return;\n          }\n          try {\n            if (!clientRes.writableEnded) clientRes.end();\n          } catch (writeErr) {\n            logger.warn(\n              \"Failed to end client response:\",\n              writeErr instanceof Error ? writeErr.message : String(writeErr),\n            );\n          }\n\n          // Parse buffered SSE events\n          const buffered = Buffer.concat(chunks).toString();\n          const events = parseSSEEvents(buffered, logger);\n\n          // Build fixture — three-way match priority:\n          // 1. Tool-result continuation (HITL): match by toolCallId\n          // 2. User message: match by last user message content\n          // 3. Fallback predicate: no user message present\n          let match: AGUIFixtureMatch;\n          const lastToolResult = getLastMessageIfToolResult(input);\n          if (lastToolResult?.toolCallId) {\n            match = { toolCallId: lastToolResult.toolCallId };\n            logger.info(`Recorded AG-UI fixture keyed on toolCallId=${lastToolResult.toolCallId}`);\n          } else {\n            const message = extractLastUserMessage(input);\n            if (message) {\n              match = { message };\n            } else {\n              match = {\n                predicate: (inp: AGUIRunAgentInput) =>\n                  !inp.messages?.length || !inp.messages.some((m) => m.role === \"user\"),\n              };\n              logger.warn(\n                \"Recorded AG-UI fixture has no user message — available in-memory only (predicate fixtures cannot be persisted to disk)\",\n              );\n            }\n          }\n          const fixture: AGUIFixture = { match, events };\n\n          if (!config.proxyOnly) {\n            // Register in memory first (always available even if disk write fails)\n            fixtures.push(fixture);\n\n            // Predicate fixtures (no user message, no toolCallId) cannot be\n            // meaningfully serialized — the sentinel becomes a literal string\n            // match that never matches real requests. Keep in-memory only.\n            if (fixture.match.predicate) {\n              logger.warn(\n                \"Skipping disk write for predicate fixture — in-memory only (cannot be persisted)\",\n              );\n            } else {\n              const serializableFixture = {\n                match: fixture.match,\n                events: fixture.events,\n                ...(fixture.delayMs !== undefined ? { delayMs: fixture.delayMs } : {}),\n              };\n\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({ fixtures: [serializableFixture] }, null, 2),\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            }\n          } else {\n            logger.info(\"Proxied AG-UI request (proxy-only mode)\");\n          }\n\n          resolve(clientStatus);\n        });\n      },\n    );\n\n    upstreamReq.on(\"timeout\", () => {\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      try {\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      } catch (writeErr) {\n        logger.warn(\n          \"Failed to write error response to client:\",\n          writeErr instanceof Error ? writeErr.message : String(writeErr),\n        );\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        const payload = line.startsWith(\"data: \") ? line.slice(6) : line.slice(5);\n        try {\n          const parsed = JSON.parse(payload) as AGUIEvent;\n          events.push(parsed);\n        } catch (err) {\n          const msg = err instanceof Error ? err.message : String(err);\n          const warning = `Skipping unparseable SSE data line (${msg}): ${payload.slice(0, 200)}`;\n          if (logger) logger.warn(warning);\n          else console.warn(warning);\n        }\n      }\n    }\n  }\n  return events;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;AA8BA,eAAsB,mBACpB,KACA,KACA,OACA,UACA,QACA,QACyB;AACzB,KAAI,CAAC,OAAO,UAAU;AACpB,SAAO,KAAK,gEAAgE;AAC5E,SAAO;;CAGT,IAAI;AACJ,KAAI;AACF,WAAS,IAAI,IAAI,OAAO,SAAS;UAC1B,KAAK;EACZ,MAAM,SAAS,eAAe,QAAQ,IAAI,UAAU,OAAO,IAAI;AAC/D,SAAO,MAAM,+BAA+B,OAAO,SAAS,KAAK,SAAS;AAC1E,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;CAEzC,IAAI;AACJ,KAAI;AACF,WAAS,MAAM,kBACb,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;aAC7D,CAAC,IAAI,cACd,KAAI,KAAK;AAEX,WAAS;;AAGX,QAAO;;AAOT,SAAS,kBACP,QACA,SACA,MACA,WACA,OACA,UACA,QACA,QACiB;AACjB,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;GACf,MAAM,iBAAiB,YAAY,cAAc;GAKjD,MAAM,eAAe,kBAAkB,OAAO,iBAAiB,MAAM,MAAM;AAG3E,OAAI,CAAC,UAAU,YACb,KAAI,iBAAiB,IACnB,WAAU,UAAU,KAAK;IACvB,gBAAgB;IAChB,iBAAiB;IACjB,YAAY;IACb,CAAC;QACG;IACL,MAAM,KAAK,YAAY,QAAQ,mBAAmB;AAClD,cAAU,UAAU,KAAK,EAAE,gBAAgB,IAAI,CAAC;;GAIpD,MAAM,SAAmB,EAAE;GAC3B,IAAI,oBAAoB;AAExB,eAAY,GAAG,SAAS,UAAkB;AAExC,QAAI;AACF,eAAU,MAAM,MAAM;aACf,KAAK;AACZ,SAAI,CAAC,mBAAmB;AACtB,0BAAoB;AACpB,cAAQ,KACN,2CACA,eAAe,QAAQ,IAAI,UAAU,OAAO,IAAI,CACjD;;;AAIL,WAAO,KAAK,MAAM;KAClB;GAEF,IAAI,UAAU;AAEd,eAAY,GAAG,UAAU,QAAQ;AAC/B,QAAI,QAAS;AACb,cAAU;AACV,QAAI;AACF,SAAI,CAAC,UAAU,aAAa;AAC1B,gBAAU,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAChE,gBAAU,IAAI,KAAK,UAAU,EAAE,OAAO,oCAAoC,CAAC,CAAC;gBACnE,CAAC,UAAU,cACpB,WAAU,KAAK;aAEV,UAAU;AACjB,YAAO,KACL,6CACA,oBAAoB,QAAQ,SAAS,UAAU,OAAO,SAAS,CAChE;;AAEH,WAAO,IAAI;KACX;AAEF,eAAY,GAAG,aAAa;AAC1B,QAAI,QAAS;AACb,cAAU;AAGV,QAAI,iBAAiB,KAAK;AACxB,SAAI;AACF,UAAI,CAAC,UAAU,cAAe,WAAU,KAAK;cACtC,UAAU;AACjB,aAAO,KACL,kCACA,oBAAoB,QAAQ,SAAS,UAAU,OAAO,SAAS,CAChE;;AAEH,aAAQ,aAAa;AACrB;;AAEF,QAAI;AACF,SAAI,CAAC,UAAU,cAAe,WAAU,KAAK;aACtC,UAAU;AACjB,YAAO,KACL,kCACA,oBAAoB,QAAQ,SAAS,UAAU,OAAO,SAAS,CAChE;;IAKH,MAAM,SAAS,eADE,OAAO,OAAO,OAAO,CAAC,UAAU,EACT,OAAO;IAM/C,IAAI;IACJ,MAAM,iBAAiBC,gDAA2B,MAAM;AACxD,QAAI,gBAAgB,YAAY;AAC9B,aAAQ,EAAE,YAAY,eAAe,YAAY;AACjD,YAAO,KAAK,8CAA8C,eAAe,aAAa;WACjF;KACL,MAAM,UAAUC,4CAAuB,MAAM;AAC7C,SAAI,QACF,SAAQ,EAAE,SAAS;UACd;AACL,cAAQ,EACN,YAAY,QACV,CAAC,IAAI,UAAU,UAAU,CAAC,IAAI,SAAS,MAAM,MAAM,EAAE,SAAS,OAAO,EACxE;AACD,aAAO,KACL,yHACD;;;IAGL,MAAM,UAAuB;KAAE;KAAO;KAAQ;AAE9C,QAAI,CAAC,OAAO,WAAW;AAErB,cAAS,KAAK,QAAQ;AAKtB,SAAI,QAAQ,MAAM,UAChB,QAAO,KACL,mFACD;UACI;MACL,MAAM,sBAAsB;OAC1B,OAAO,QAAQ;OACf,QAAQ,QAAQ;OAChB,GAAI,QAAQ,YAAY,SAAY,EAAE,SAAS,QAAQ,SAAS,GAAG,EAAE;OACtE;MAED,MAAM,cAAc,OAAO,eAAe;MAE1C,MAAM,WAAW,yBADC,IAAI,MAAM,EAAC,aAAa,CAAC,QAAQ,SAAS,IAAI,CAC7B,GAAGC,YAAO,YAAY,CAAC,MAAM,GAAG,EAAE,CAAC;MACtE,MAAM,WAAWC,UAAK,KAAK,aAAa,SAAS;AAEjD,UAAI;AACF,eAAG,UAAU,aAAa,EAAE,WAAW,MAAM,CAAC;AAC9C,eAAG,cACD,UACA,KAAK,UAAU,EAAE,UAAU,CAAC,oBAAoB,EAAE,EAAE,MAAM,EAAE,EAC5D,QACD;AACD,cAAO,KAAK,6BAA6B,WAAW;eAC7C,KAAK;OACZ,MAAM,MAAM,eAAe,QAAQ,IAAI,UAAU;AACjD,cAAO,MACL,yCAAyC,IAAI,+BAC9C;;;UAIL,QAAO,KAAK,0CAA0C;AAGxD,YAAQ,aAAa;KACrB;IAEL;AAED,cAAY,GAAG,iBAAiB;AAC9B,eAAY,wBACV,IAAI,MAAM,0CAA0C,sBAAsB,IAAK,GAAG,CACnF;IACD;AAEF,cAAY,GAAG,UAAU,QAAQ;AAC/B,OAAI;AACF,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;YAEV,UAAU;AACjB,WAAO,KACL,6CACA,oBAAoB,QAAQ,SAAS,UAAU,OAAO,SAAS,CAChE;;AAEH,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,QAAQ,EAAE;GAC5B,MAAM,UAAU,KAAK,WAAW,SAAS,GAAG,KAAK,MAAM,EAAE,GAAG,KAAK,MAAM,EAAE;AACzE,OAAI;IACF,MAAM,SAAS,KAAK,MAAM,QAAQ;AAClC,WAAO,KAAK,OAAO;YACZ,KAAK;IAEZ,MAAM,UAAU,uCADJ,eAAe,QAAQ,IAAI,UAAU,OAAO,IAAI,CACD,KAAK,QAAQ,MAAM,GAAG,IAAI;AACrF,QAAI,OAAQ,QAAO,KAAK,QAAQ;QAC3B,SAAQ,KAAK,QAAQ;;;;AAKlC,QAAO"}

package/dist/agui-recorder.d.cts

@@ -14,7 +14,6 @@ import * as http$1 from "node:http";
  */
 declare function proxyAndRecordAGUI(req: http$1.IncomingMessage, res: http$1.ServerResponse, input: AGUIRunAgentInput, fixtures: AGUIFixture[], config: AGUIRecordConfig, logger: Logger): Promise<number | false>;
 //# sourceMappingURL=agui-recorder.d.ts.map
-
 //#endregion
 export { proxyAndRecordAGUI };
 //# sourceMappingURL=agui-recorder.d.cts.map

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

@@ -1 +1 @@
-{"version":3,"file":"agui-recorder.d.cts","names":[],"sources":["../src/agui-recorder.ts"],"sourcesContent":[],"mappings":";;;;;;;;AAiBA;;;;;;AAKU,iBALY,kBAAA,CAKZ,GAAA,EAJH,MAAA,CAAK,eAIF,EAAA,GAAA,EAHH,MAAA,CAAK,cAGF,EAAA,KAAA,EAFD,iBAEC,EAAA,QAAA,EADE,WACF,EAAA,EAAA,MAAA,EAAA,gBAAA,EAAA,MAAA,EACA,MADA,CAAA,EAEP,OAFO,CAAA,MAAA,GAAA,KAAA,CAAA"}
+{"version":3,"file":"agui-recorder.d.cts","names":[],"sources":["../src/agui-recorder.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;;;;;iBA8BsB,kBAAA,MACf,MAAA,CAAK,sBACL,MAAA,CAAK,uBACH,6BACG,uBACF,0BACA,SACP"}

package/dist/agui-recorder.d.ts

@@ -14,7 +14,6 @@ import * as http$1 from "node:http";
  */
 declare function proxyAndRecordAGUI(req: http$1.IncomingMessage, res: http$1.ServerResponse, input: AGUIRunAgentInput, fixtures: AGUIFixture[], config: AGUIRecordConfig, logger: Logger): Promise<number | false>;
 //# sourceMappingURL=agui-recorder.d.ts.map
-
 //#endregion
 export { proxyAndRecordAGUI };
 //# sourceMappingURL=agui-recorder.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"agui-recorder.d.ts","names":[],"sources":["../src/agui-recorder.ts"],"sourcesContent":[],"mappings":";;;;;;;;AAiBA;;;;;;AAKU,iBALY,kBAAA,CAKZ,GAAA,EAJH,MAAA,CAAK,eAIF,EAAA,GAAA,EAHH,MAAA,CAAK,cAGF,EAAA,KAAA,EAFD,iBAEC,EAAA,QAAA,EADE,WACF,EAAA,EAAA,MAAA,EAAA,gBAAA,EAAA,MAAA,EACA,MADA,CAAA,EAEP,OAFO,CAAA,MAAA,GAAA,KAAA,CAAA"}
+{"version":3,"file":"agui-recorder.d.ts","names":[],"sources":["../src/agui-recorder.ts"],"sourcesContent":[],"mappings":";;;;;;;;;;;;;;iBA8BsB,kBAAA,MACf,MAAA,CAAK,sBACL,MAAA,CAAK,uBACH,6BACG,uBACF,0BACA,SACP"}

package/dist/agui-recorder.js

@@ -1,4 +1,4 @@
-import { extractLastUserMessage } from "./agui-handler.js";
+import { extractLastUserMessage, getLastMessageIfToolResult } from "./agui-handler.js";
 import * as http$1 from "node:http";
 import * as crypto$1 from "node:crypto";
 import * as https from "node:https";
@@ -89,7 +89,10 @@ function teeUpstreamStream(target, headers, body, clientRes, input, fixtures, co
 				}
 				chunks.push(chunk);
 			});
+			let settled = false;
 			upstreamRes.on("error", (err) => {
+				if (settled) return;
+				settled = true;
 				try {
 					if (!clientRes.headersSent) {
 						clientRes.writeHead(502, { "Content-Type": "application/json" });
@@ -101,35 +104,60 @@ function teeUpstreamStream(target, headers, body, clientRes, input, fixtures, co
 				reject(err);
 			});
 			upstreamRes.on("end", () => {
+				if (settled) return;
+				settled = true;
+				if (clientStatus !== 200) {
+					try {
+						if (!clientRes.writableEnded) clientRes.end();
+					} catch (writeErr) {
+						logger.warn("Failed to end client response:", writeErr instanceof Error ? writeErr.message : String(writeErr));
+					}
+					resolve(clientStatus);
+					return;
+				}
 				try {
 					if (!clientRes.writableEnded) clientRes.end();
 				} catch (writeErr) {
 					logger.warn("Failed to end client response:", writeErr instanceof Error ? writeErr.message : String(writeErr));
 				}
 				const events = parseSSEEvents(Buffer.concat(chunks).toString(), logger);
-				const message = extractLastUserMessage(input);
+				let match;
+				const lastToolResult = getLastMessageIfToolResult(input);
+				if (lastToolResult?.toolCallId) {
+					match = { toolCallId: lastToolResult.toolCallId };
+					logger.info(`Recorded AG-UI fixture keyed on toolCallId=${lastToolResult.toolCallId}`);
+				} else {
+					const message = extractLastUserMessage(input);
+					if (message) match = { message };
+					else {
+						match = { predicate: (inp) => !inp.messages?.length || !inp.messages.some((m) => m.role === "user") };
+						logger.warn("Recorded AG-UI fixture has no user message — available in-memory only (predicate fixtures cannot be persisted to disk)");
+					}
+				}
 				const fixture = {
-					match: message ? { message } : { predicate: (inp) => !inp.messages?.length || !inp.messages.some((m) => m.role === "user") },
+					match,
 					events
 				};
-				if (!message) logger.warn("Recorded AG-UI fixture has no user message — will use __NO_USER_MESSAGE__ sentinel on disk");
 				if (!config.proxyOnly) {
 					fixtures.push(fixture);
-					const serializableFixture = {
-						match: fixture.match.predicate ? { message: "__NO_USER_MESSAGE__" } : fixture.match,
-						events: fixture.events,
-						...fixture.delayMs !== void 0 ? { delayMs: fixture.delayMs } : {}
-					};
-					const fixturePath = config.fixturePath ?? "./fixtures/agui-recorded";
-					const filename = `agui-${(/* @__PURE__ */ new Date()).toISOString().replace(/[:.]/g, "-")}-${crypto$1.randomUUID().slice(0, 8)}.json`;
-					const filepath = path.join(fixturePath, filename);
-					try {
-						fs.mkdirSync(fixturePath, { recursive: true });
-						fs.writeFileSync(filepath, JSON.stringify({ fixtures: [serializableFixture] }, 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)`);
+					if (fixture.match.predicate) logger.warn("Skipping disk write for predicate fixture — in-memory only (cannot be persisted)");
+					else {
+						const serializableFixture = {
+							match: fixture.match,
+							events: fixture.events,
+							...fixture.delayMs !== void 0 ? { delayMs: fixture.delayMs } : {}
+						};
+						const fixturePath = config.fixturePath ?? "./fixtures/agui-recorded";
+						const filename = `agui-${(/* @__PURE__ */ new Date()).toISOString().replace(/[:.]/g, "-")}-${crypto$1.randomUUID().slice(0, 8)}.json`;
+						const filepath = path.join(fixturePath, filename);
+						try {
+							fs.mkdirSync(fixturePath, { recursive: true });
+							fs.writeFileSync(filepath, JSON.stringify({ fixtures: [serializableFixture] }, 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(clientStatus);
@@ -167,9 +195,9 @@ function parseSSEEvents(text, logger) {
 				const parsed = JSON.parse(payload);
 				events.push(parsed);
 			} catch (err) {
-				const msg = err instanceof Error ? err.message : String(err);
-				if (logger) logger.warn(`Skipping unparseable SSE data line: ${payload.slice(0, 200)}`);
-				else console.warn(`Skipping unparseable SSE data line: ${msg}`);
+				const warning = `Skipping unparseable SSE data line (${err instanceof Error ? err.message : String(err)}): ${payload.slice(0, 200)}`;
+				if (logger) logger.warn(warning);
+				else console.warn(warning);
 			}
 		}
 	}

package/dist/agui-recorder.js.map

@@ -1 +1 @@
-{"version":3,"file":"agui-recorder.js","names":["http","crypto"],"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 the HTTP status code written to the client if the request was proxied,\n * or `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<number | false> {\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 (err) {\n    const detail = err instanceof Error ? err.message : String(err);\n    logger.error(`Invalid upstream AG-UI URL: ${config.upstream} — ${detail}`);\n    res.writeHead(502, { \"Content-Type\": \"application/json\" });\n    res.end(JSON.stringify({ error: \"Invalid upstream AG-UI URL\" }));\n    return 502;\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  let status: number;\n  try {\n    status = 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    } else if (!res.writableEnded) {\n      res.end();\n    }\n    status = 502;\n  }\n\n  return status;\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<number> {\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        const upstreamStatus = upstreamRes.statusCode ?? 200;\n\n        // Normalize status codes: aimock acts as a gateway, so upstream\n        // provider details (429, 503, etc.) should not leak.\n        // Successes → 200, errors → 502 (Bad Gateway).\n        const clientStatus = upstreamStatus >= 200 && upstreamStatus < 300 ? 200 : 502;\n\n        // Set appropriate headers on the client response.\n        if (!clientRes.headersSent) {\n          if (clientStatus === 200) {\n            clientRes.writeHead(200, {\n              \"Content-Type\": \"text/event-stream\",\n              \"Cache-Control\": \"no-cache\",\n              Connection: \"keep-alive\",\n            });\n          } else {\n            const ct = upstreamRes.headers[\"content-type\"] || \"application/json\";\n            clientRes.writeHead(502, { \"Content-Type\": ct });\n          }\n        }\n\n        const chunks: Buffer[] = [];\n        let clientWriteFailed = false;\n\n        upstreamRes.on(\"data\", (chunk: Buffer) => {\n          // Relay to client in real time\n          try {\n            clientRes.write(chunk);\n          } catch (err) {\n            if (!clientWriteFailed) {\n              clientWriteFailed = true;\n              logger?.warn(\n                \"Client write failed during proxy relay:\",\n                err instanceof Error ? err.message : String(err),\n              );\n            }\n          }\n          // Buffer for fixture construction\n          chunks.push(chunk);\n        });\n\n        upstreamRes.on(\"error\", (err) => {\n          try {\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          } catch (writeErr) {\n            logger.warn(\n              \"Failed to write error response to client:\",\n              writeErr instanceof Error ? writeErr.message : String(writeErr),\n            );\n          }\n          reject(err);\n        });\n\n        upstreamRes.on(\"end\", () => {\n          try {\n            if (!clientRes.writableEnded) clientRes.end();\n          } catch (writeErr) {\n            logger.warn(\n              \"Failed to end client response:\",\n              writeErr instanceof Error ? writeErr.message : String(writeErr),\n            );\n          }\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          const fixture: AGUIFixture = {\n            match: message\n              ? { message }\n              : {\n                  predicate: (inp: AGUIRunAgentInput) =>\n                    !inp.messages?.length || !inp.messages.some((m) => m.role === \"user\"),\n                },\n            events,\n          };\n          if (!message) {\n            logger.warn(\n              \"Recorded AG-UI fixture has no user message — will use __NO_USER_MESSAGE__ sentinel on disk\",\n            );\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 — predicate functions are not serializable,\n            // so replace with a sentinel string that won't match real user messages.\n            const serializableFixture = {\n              match: fixture.match.predicate ? { message: \"__NO_USER_MESSAGE__\" } : fixture.match,\n              events: fixture.events,\n              ...(fixture.delayMs !== undefined ? { delayMs: fixture.delayMs } : {}),\n            };\n\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({ fixtures: [serializableFixture] }, null, 2),\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(clientStatus);\n        });\n      },\n    );\n\n    upstreamReq.on(\"timeout\", () => {\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      try {\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      } catch (writeErr) {\n        logger.warn(\n          \"Failed to write error response to client:\",\n          writeErr instanceof Error ? writeErr.message : String(writeErr),\n        );\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        const payload = line.startsWith(\"data: \") ? line.slice(6) : line.slice(5);\n        try {\n          const parsed = JSON.parse(payload) as AGUIEvent;\n          events.push(parsed);\n        } catch (err) {\n          const msg = err instanceof Error ? err.message : String(err);\n          if (logger) logger.warn(`Skipping unparseable SSE data line: ${payload.slice(0, 200)}`);\n          else console.warn(`Skipping unparseable SSE data line: ${msg}`);\n        }\n      }\n    }\n  }\n  return events;\n}\n"],"mappings":";;;;;;;;;;;;;;;;AAiBA,eAAsB,mBACpB,KACA,KACA,OACA,UACA,QACA,QACyB;AACzB,KAAI,CAAC,OAAO,UAAU;AACpB,SAAO,KAAK,gEAAgE;AAC5E,SAAO;;CAGT,IAAI;AACJ,KAAI;AACF,WAAS,IAAI,IAAI,OAAO,SAAS;UAC1B,KAAK;EACZ,MAAM,SAAS,eAAe,QAAQ,IAAI,UAAU,OAAO,IAAI;AAC/D,SAAO,MAAM,+BAA+B,OAAO,SAAS,KAAK,SAAS;AAC1E,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;CAEzC,IAAI;AACJ,KAAI;AACF,WAAS,MAAM,kBACb,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;aAC7D,CAAC,IAAI,cACd,KAAI,KAAK;AAEX,WAAS;;AAGX,QAAO;;AAOT,SAAS,kBACP,QACA,SACA,MACA,WACA,OACA,UACA,QACA,QACiB;AACjB,QAAO,IAAI,SAAS,SAAS,WAAW;EACtC,MAAM,YAAY,OAAO,aAAa,WAAW,QAAQA;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;GACf,MAAM,iBAAiB,YAAY,cAAc;GAKjD,MAAM,eAAe,kBAAkB,OAAO,iBAAiB,MAAM,MAAM;AAG3E,OAAI,CAAC,UAAU,YACb,KAAI,iBAAiB,IACnB,WAAU,UAAU,KAAK;IACvB,gBAAgB;IAChB,iBAAiB;IACjB,YAAY;IACb,CAAC;QACG;IACL,MAAM,KAAK,YAAY,QAAQ,mBAAmB;AAClD,cAAU,UAAU,KAAK,EAAE,gBAAgB,IAAI,CAAC;;GAIpD,MAAM,SAAmB,EAAE;GAC3B,IAAI,oBAAoB;AAExB,eAAY,GAAG,SAAS,UAAkB;AAExC,QAAI;AACF,eAAU,MAAM,MAAM;aACf,KAAK;AACZ,SAAI,CAAC,mBAAmB;AACtB,0BAAoB;AACpB,cAAQ,KACN,2CACA,eAAe,QAAQ,IAAI,UAAU,OAAO,IAAI,CACjD;;;AAIL,WAAO,KAAK,MAAM;KAClB;AAEF,eAAY,GAAG,UAAU,QAAQ;AAC/B,QAAI;AACF,SAAI,CAAC,UAAU,aAAa;AAC1B,gBAAU,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAChE,gBAAU,IAAI,KAAK,UAAU,EAAE,OAAO,oCAAoC,CAAC,CAAC;gBACnE,CAAC,UAAU,cACpB,WAAU,KAAK;aAEV,UAAU;AACjB,YAAO,KACL,6CACA,oBAAoB,QAAQ,SAAS,UAAU,OAAO,SAAS,CAChE;;AAEH,WAAO,IAAI;KACX;AAEF,eAAY,GAAG,aAAa;AAC1B,QAAI;AACF,SAAI,CAAC,UAAU,cAAe,WAAU,KAAK;aACtC,UAAU;AACjB,YAAO,KACL,kCACA,oBAAoB,QAAQ,SAAS,UAAU,OAAO,SAAS,CAChE;;IAKH,MAAM,SAAS,eADE,OAAO,OAAO,OAAO,CAAC,UAAU,EACT,OAAO;IAG/C,MAAM,UAAU,uBAAuB,MAAM;IAC7C,MAAM,UAAuB;KAC3B,OAAO,UACH,EAAE,SAAS,GACX,EACE,YAAY,QACV,CAAC,IAAI,UAAU,UAAU,CAAC,IAAI,SAAS,MAAM,MAAM,EAAE,SAAS,OAAO,EACxE;KACL;KACD;AACD,QAAI,CAAC,QACH,QAAO,KACL,6FACD;AAGH,QAAI,CAAC,OAAO,WAAW;AAErB,cAAS,KAAK,QAAQ;KAItB,MAAM,sBAAsB;MAC1B,OAAO,QAAQ,MAAM,YAAY,EAAE,SAAS,uBAAuB,GAAG,QAAQ;MAC9E,QAAQ,QAAQ;MAChB,GAAI,QAAQ,YAAY,SAAY,EAAE,SAAS,QAAQ,SAAS,GAAG,EAAE;MACtE;KAED,MAAM,cAAc,OAAO,eAAe;KAE1C,MAAM,WAAW,yBADC,IAAI,MAAM,EAAC,aAAa,CAAC,QAAQ,SAAS,IAAI,CAC7B,GAAGC,SAAO,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,UAAU,EAAE,UAAU,CAAC,oBAAoB,EAAE,EAAE,MAAM,EAAE,EAC5D,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,YAAQ,aAAa;KACrB;IAEL;AAED,cAAY,GAAG,iBAAiB;AAC9B,eAAY,wBACV,IAAI,MAAM,0CAA0C,sBAAsB,IAAK,GAAG,CACnF;IACD;AAEF,cAAY,GAAG,UAAU,QAAQ;AAC/B,OAAI;AACF,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;YAEV,UAAU;AACjB,WAAO,KACL,6CACA,oBAAoB,QAAQ,SAAS,UAAU,OAAO,SAAS,CAChE;;AAEH,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,QAAQ,EAAE;GAC5B,MAAM,UAAU,KAAK,WAAW,SAAS,GAAG,KAAK,MAAM,EAAE,GAAG,KAAK,MAAM,EAAE;AACzE,OAAI;IACF,MAAM,SAAS,KAAK,MAAM,QAAQ;AAClC,WAAO,KAAK,OAAO;YACZ,KAAK;IACZ,MAAM,MAAM,eAAe,QAAQ,IAAI,UAAU,OAAO,IAAI;AAC5D,QAAI,OAAQ,QAAO,KAAK,uCAAuC,QAAQ,MAAM,GAAG,IAAI,GAAG;QAClF,SAAQ,KAAK,uCAAuC,MAAM;;;;AAKvE,QAAO"}
+{"version":3,"file":"agui-recorder.js","names":["http","crypto"],"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 {\n  AGUIFixture,\n  AGUIFixtureMatch,\n  AGUIRecordConfig,\n  AGUIEvent,\n  AGUIRunAgentInput,\n} from \"./agui-types.js\";\nimport { extractLastUserMessage, getLastMessageIfToolResult } from \"./agui-handler.js\";\nimport type { Logger } from \"./logger.js\";\n\n/**\n * Sentinel `match.message` value written to disk when the request had no\n * extractable user text. Keeps the on-disk fixture serializable (predicate\n * matchers aren't) but won't match any real user input on replay.\n */\nexport const NO_USER_MESSAGE_SENTINEL = \"__NO_USER_MESSAGE__\";\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 the HTTP status code written to the client if the request was proxied,\n * or `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<number | false> {\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 (err) {\n    const detail = err instanceof Error ? err.message : String(err);\n    logger.error(`Invalid upstream AG-UI URL: ${config.upstream} — ${detail}`);\n    res.writeHead(502, { \"Content-Type\": \"application/json\" });\n    res.end(JSON.stringify({ error: \"Invalid upstream AG-UI URL\" }));\n    return 502;\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  let status: number;\n  try {\n    status = 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    } else if (!res.writableEnded) {\n      res.end();\n    }\n    status = 502;\n  }\n\n  return status;\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<number> {\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        const upstreamStatus = upstreamRes.statusCode ?? 200;\n\n        // Normalize status codes: aimock acts as a gateway, so upstream\n        // provider details (429, 503, etc.) should not leak.\n        // Successes → 200, errors → 502 (Bad Gateway).\n        const clientStatus = upstreamStatus >= 200 && upstreamStatus < 300 ? 200 : 502;\n\n        // Set appropriate headers on the client response.\n        if (!clientRes.headersSent) {\n          if (clientStatus === 200) {\n            clientRes.writeHead(200, {\n              \"Content-Type\": \"text/event-stream\",\n              \"Cache-Control\": \"no-cache\",\n              Connection: \"keep-alive\",\n            });\n          } else {\n            const ct = upstreamRes.headers[\"content-type\"] || \"application/json\";\n            clientRes.writeHead(502, { \"Content-Type\": ct });\n          }\n        }\n\n        const chunks: Buffer[] = [];\n        let clientWriteFailed = false;\n\n        upstreamRes.on(\"data\", (chunk: Buffer) => {\n          // Relay to client in real time\n          try {\n            clientRes.write(chunk);\n          } catch (err) {\n            if (!clientWriteFailed) {\n              clientWriteFailed = true;\n              logger?.warn(\n                \"Client write failed during proxy relay:\",\n                err instanceof Error ? err.message : String(err),\n              );\n            }\n          }\n          // Buffer for fixture construction\n          chunks.push(chunk);\n        });\n\n        let settled = false;\n\n        upstreamRes.on(\"error\", (err) => {\n          if (settled) return;\n          settled = true;\n          try {\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          } catch (writeErr) {\n            logger.warn(\n              \"Failed to write error response to client:\",\n              writeErr instanceof Error ? writeErr.message : String(writeErr),\n            );\n          }\n          reject(err);\n        });\n\n        upstreamRes.on(\"end\", () => {\n          if (settled) return;\n          settled = true;\n\n          // Don't record fixtures for non-2xx upstream responses\n          if (clientStatus !== 200) {\n            try {\n              if (!clientRes.writableEnded) clientRes.end();\n            } catch (writeErr) {\n              logger.warn(\n                \"Failed to end client response:\",\n                writeErr instanceof Error ? writeErr.message : String(writeErr),\n              );\n            }\n            resolve(clientStatus);\n            return;\n          }\n          try {\n            if (!clientRes.writableEnded) clientRes.end();\n          } catch (writeErr) {\n            logger.warn(\n              \"Failed to end client response:\",\n              writeErr instanceof Error ? writeErr.message : String(writeErr),\n            );\n          }\n\n          // Parse buffered SSE events\n          const buffered = Buffer.concat(chunks).toString();\n          const events = parseSSEEvents(buffered, logger);\n\n          // Build fixture — three-way match priority:\n          // 1. Tool-result continuation (HITL): match by toolCallId\n          // 2. User message: match by last user message content\n          // 3. Fallback predicate: no user message present\n          let match: AGUIFixtureMatch;\n          const lastToolResult = getLastMessageIfToolResult(input);\n          if (lastToolResult?.toolCallId) {\n            match = { toolCallId: lastToolResult.toolCallId };\n            logger.info(`Recorded AG-UI fixture keyed on toolCallId=${lastToolResult.toolCallId}`);\n          } else {\n            const message = extractLastUserMessage(input);\n            if (message) {\n              match = { message };\n            } else {\n              match = {\n                predicate: (inp: AGUIRunAgentInput) =>\n                  !inp.messages?.length || !inp.messages.some((m) => m.role === \"user\"),\n              };\n              logger.warn(\n                \"Recorded AG-UI fixture has no user message — available in-memory only (predicate fixtures cannot be persisted to disk)\",\n              );\n            }\n          }\n          const fixture: AGUIFixture = { match, events };\n\n          if (!config.proxyOnly) {\n            // Register in memory first (always available even if disk write fails)\n            fixtures.push(fixture);\n\n            // Predicate fixtures (no user message, no toolCallId) cannot be\n            // meaningfully serialized — the sentinel becomes a literal string\n            // match that never matches real requests. Keep in-memory only.\n            if (fixture.match.predicate) {\n              logger.warn(\n                \"Skipping disk write for predicate fixture — in-memory only (cannot be persisted)\",\n              );\n            } else {\n              const serializableFixture = {\n                match: fixture.match,\n                events: fixture.events,\n                ...(fixture.delayMs !== undefined ? { delayMs: fixture.delayMs } : {}),\n              };\n\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({ fixtures: [serializableFixture] }, null, 2),\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            }\n          } else {\n            logger.info(\"Proxied AG-UI request (proxy-only mode)\");\n          }\n\n          resolve(clientStatus);\n        });\n      },\n    );\n\n    upstreamReq.on(\"timeout\", () => {\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      try {\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      } catch (writeErr) {\n        logger.warn(\n          \"Failed to write error response to client:\",\n          writeErr instanceof Error ? writeErr.message : String(writeErr),\n        );\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        const payload = line.startsWith(\"data: \") ? line.slice(6) : line.slice(5);\n        try {\n          const parsed = JSON.parse(payload) as AGUIEvent;\n          events.push(parsed);\n        } catch (err) {\n          const msg = err instanceof Error ? err.message : String(err);\n          const warning = `Skipping unparseable SSE data line (${msg}): ${payload.slice(0, 200)}`;\n          if (logger) logger.warn(warning);\n          else console.warn(warning);\n        }\n      }\n    }\n  }\n  return events;\n}\n"],"mappings":";;;;;;;;;;;;;;;;AA8BA,eAAsB,mBACpB,KACA,KACA,OACA,UACA,QACA,QACyB;AACzB,KAAI,CAAC,OAAO,UAAU;AACpB,SAAO,KAAK,gEAAgE;AAC5E,SAAO;;CAGT,IAAI;AACJ,KAAI;AACF,WAAS,IAAI,IAAI,OAAO,SAAS;UAC1B,KAAK;EACZ,MAAM,SAAS,eAAe,QAAQ,IAAI,UAAU,OAAO,IAAI;AAC/D,SAAO,MAAM,+BAA+B,OAAO,SAAS,KAAK,SAAS;AAC1E,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;CAEzC,IAAI;AACJ,KAAI;AACF,WAAS,MAAM,kBACb,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;aAC7D,CAAC,IAAI,cACd,KAAI,KAAK;AAEX,WAAS;;AAGX,QAAO;;AAOT,SAAS,kBACP,QACA,SACA,MACA,WACA,OACA,UACA,QACA,QACiB;AACjB,QAAO,IAAI,SAAS,SAAS,WAAW;EACtC,MAAM,YAAY,OAAO,aAAa,WAAW,QAAQA;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;GACf,MAAM,iBAAiB,YAAY,cAAc;GAKjD,MAAM,eAAe,kBAAkB,OAAO,iBAAiB,MAAM,MAAM;AAG3E,OAAI,CAAC,UAAU,YACb,KAAI,iBAAiB,IACnB,WAAU,UAAU,KAAK;IACvB,gBAAgB;IAChB,iBAAiB;IACjB,YAAY;IACb,CAAC;QACG;IACL,MAAM,KAAK,YAAY,QAAQ,mBAAmB;AAClD,cAAU,UAAU,KAAK,EAAE,gBAAgB,IAAI,CAAC;;GAIpD,MAAM,SAAmB,EAAE;GAC3B,IAAI,oBAAoB;AAExB,eAAY,GAAG,SAAS,UAAkB;AAExC,QAAI;AACF,eAAU,MAAM,MAAM;aACf,KAAK;AACZ,SAAI,CAAC,mBAAmB;AACtB,0BAAoB;AACpB,cAAQ,KACN,2CACA,eAAe,QAAQ,IAAI,UAAU,OAAO,IAAI,CACjD;;;AAIL,WAAO,KAAK,MAAM;KAClB;GAEF,IAAI,UAAU;AAEd,eAAY,GAAG,UAAU,QAAQ;AAC/B,QAAI,QAAS;AACb,cAAU;AACV,QAAI;AACF,SAAI,CAAC,UAAU,aAAa;AAC1B,gBAAU,UAAU,KAAK,EAAE,gBAAgB,oBAAoB,CAAC;AAChE,gBAAU,IAAI,KAAK,UAAU,EAAE,OAAO,oCAAoC,CAAC,CAAC;gBACnE,CAAC,UAAU,cACpB,WAAU,KAAK;aAEV,UAAU;AACjB,YAAO,KACL,6CACA,oBAAoB,QAAQ,SAAS,UAAU,OAAO,SAAS,CAChE;;AAEH,WAAO,IAAI;KACX;AAEF,eAAY,GAAG,aAAa;AAC1B,QAAI,QAAS;AACb,cAAU;AAGV,QAAI,iBAAiB,KAAK;AACxB,SAAI;AACF,UAAI,CAAC,UAAU,cAAe,WAAU,KAAK;cACtC,UAAU;AACjB,aAAO,KACL,kCACA,oBAAoB,QAAQ,SAAS,UAAU,OAAO,SAAS,CAChE;;AAEH,aAAQ,aAAa;AACrB;;AAEF,QAAI;AACF,SAAI,CAAC,UAAU,cAAe,WAAU,KAAK;aACtC,UAAU;AACjB,YAAO,KACL,kCACA,oBAAoB,QAAQ,SAAS,UAAU,OAAO,SAAS,CAChE;;IAKH,MAAM,SAAS,eADE,OAAO,OAAO,OAAO,CAAC,UAAU,EACT,OAAO;IAM/C,IAAI;IACJ,MAAM,iBAAiB,2BAA2B,MAAM;AACxD,QAAI,gBAAgB,YAAY;AAC9B,aAAQ,EAAE,YAAY,eAAe,YAAY;AACjD,YAAO,KAAK,8CAA8C,eAAe,aAAa;WACjF;KACL,MAAM,UAAU,uBAAuB,MAAM;AAC7C,SAAI,QACF,SAAQ,EAAE,SAAS;UACd;AACL,cAAQ,EACN,YAAY,QACV,CAAC,IAAI,UAAU,UAAU,CAAC,IAAI,SAAS,MAAM,MAAM,EAAE,SAAS,OAAO,EACxE;AACD,aAAO,KACL,yHACD;;;IAGL,MAAM,UAAuB;KAAE;KAAO;KAAQ;AAE9C,QAAI,CAAC,OAAO,WAAW;AAErB,cAAS,KAAK,QAAQ;AAKtB,SAAI,QAAQ,MAAM,UAChB,QAAO,KACL,mFACD;UACI;MACL,MAAM,sBAAsB;OAC1B,OAAO,QAAQ;OACf,QAAQ,QAAQ;OAChB,GAAI,QAAQ,YAAY,SAAY,EAAE,SAAS,QAAQ,SAAS,GAAG,EAAE;OACtE;MAED,MAAM,cAAc,OAAO,eAAe;MAE1C,MAAM,WAAW,yBADC,IAAI,MAAM,EAAC,aAAa,CAAC,QAAQ,SAAS,IAAI,CAC7B,GAAGC,SAAO,YAAY,CAAC,MAAM,GAAG,EAAE,CAAC;MACtE,MAAM,WAAW,KAAK,KAAK,aAAa,SAAS;AAEjD,UAAI;AACF,UAAG,UAAU,aAAa,EAAE,WAAW,MAAM,CAAC;AAC9C,UAAG,cACD,UACA,KAAK,UAAU,EAAE,UAAU,CAAC,oBAAoB,EAAE,EAAE,MAAM,EAAE,EAC5D,QACD;AACD,cAAO,KAAK,6BAA6B,WAAW;eAC7C,KAAK;OACZ,MAAM,MAAM,eAAe,QAAQ,IAAI,UAAU;AACjD,cAAO,MACL,yCAAyC,IAAI,+BAC9C;;;UAIL,QAAO,KAAK,0CAA0C;AAGxD,YAAQ,aAAa;KACrB;IAEL;AAED,cAAY,GAAG,iBAAiB;AAC9B,eAAY,wBACV,IAAI,MAAM,0CAA0C,sBAAsB,IAAK,GAAG,CACnF;IACD;AAEF,cAAY,GAAG,UAAU,QAAQ;AAC/B,OAAI;AACF,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;YAEV,UAAU;AACjB,WAAO,KACL,6CACA,oBAAoB,QAAQ,SAAS,UAAU,OAAO,SAAS,CAChE;;AAEH,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,QAAQ,EAAE;GAC5B,MAAM,UAAU,KAAK,WAAW,SAAS,GAAG,KAAK,MAAM,EAAE,GAAG,KAAK,MAAM,EAAE;AACzE,OAAI;IACF,MAAM,SAAS,KAAK,MAAM,QAAQ;AAClC,WAAO,KAAK,OAAO;YACZ,KAAK;IAEZ,MAAM,UAAU,uCADJ,eAAe,QAAQ,IAAI,UAAU,OAAO,IAAI,CACD,KAAK,QAAQ,MAAM,GAAG,IAAI;AACrF,QAAI,OAAQ,QAAO,KAAK,QAAQ;QAC3B,SAAQ,KAAK,QAAQ;;;;AAKlC,QAAO"}

package/dist/agui-stub.cjs

@@ -23,6 +23,7 @@ exports.buildToolCallChunk = require_agui_handler.buildToolCallChunk;
 exports.buildToolCallResponse = require_agui_handler.buildToolCallResponse;
 exports.extractLastUserMessage = require_agui_handler.extractLastUserMessage;
 exports.findFixture = require_agui_handler.findFixture;
+exports.getLastMessageIfToolResult = require_agui_handler.getLastMessageIfToolResult;
 exports.matchesFixture = require_agui_handler.matchesFixture;
 exports.proxyAndRecordAGUI = require_agui_recorder.proxyAndRecordAGUI;
 exports.writeAGUIEventStream = require_agui_handler.writeAGUIEventStream;