experimental-a2 0.18.0 → 0.19.0

package/dist/ai-DrUaMyRt.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"ai-DrUaMyRt.d.ts","names":[],"sources":["../src/ai-message-base.ts","../src/ai-control.ts","../src/ai-client-state.ts","../src/ai-projector.ts","../src/ai-progress-batches.ts","../src/ai-message-projection.ts","../src/ai.ts"],"mappings":";;;;KAGK;EAAc;;EAAsB;;EAAmB;;KACvD,aAAa,eAAe;KAC5B;EAAa,OAAO;;EAAiC,QAAQ;;KAEtD;EACV;EACA,SAAS;EACT,QAAQ,eAAe;;;;KCIb,UAAU,UAAU,YAAY;EAC1C;EACA,SAAS;EACT;EACA;;KAGU,eAAe,UAAU,YAAY;EAC3C;EAAgB,SAAS;EAAG;;EAC5B;EAAgB,SAAS;;EACzB;EAAgB;EAAiB;;EACjC;EAAkB;;EAElB;EACA;EACA;EACA;;EAEA;EAAgB;EAAgB;;EAChC;EAAiB;EAAgB,SAAS;EAAG;;EAC7C;EAAiB;;EACjB;;EACA;EAAuB,QAAQ;;EAC/B;EAAoB,UAAU;;EAC9B;EAAiB,UAAU;;EAC3B;EAAyB,SAAS;;EAClC;EAAiB,SAAS;;EAC1B;EAAqB,SAAS;;KAExB;EACV;EACA;EACA;EACA;;KAUU,WAAW,UAAU,YAAY;EAC3C;EACA,OAAO,UAAU;EACjB;;KA+BU;EAAa;EAAc;EAAY;;KACvC;EACV;EACA;EACA;EACA;EACA;EACA;EACA,QAAQ;;KAGE;EACV;EACA;EACA,MAAM;EACN,YAAY;;KAYF;EACV;EACA;EACA;EACA;IAAW;IAAe;IAAqB;;EAC/C;IAAS;IAAc;;;KAGb;EACV,SAAS;EACT;IACE;IACA;IACA,OAAO;IACP;IACA;IACA;IACA,eAAe;IACf;IACA;;;;;KCjIQ,aAAa,UAAU,aAAa,QAC9C,eAAe;EAEb;;KAYQ,oBAAoB,UAAU;EACxC;EACA,SAAS,aAAa;;;;KCfnB;EACH;EACA;EACA;EACA;EACA;EACA;EACA;;KAwJU;EACV,MAAM;EACN,WAAW;EACX,OAAO,eAAe;;;;KChJZ;EAAkB;EAAe,QAAQ;;KAChD,aAAa;EAAoB,MAAM;EAAY,OAAO;;KAEnD;EACV;EACA,MAAM;EACN,QAAQ,MAAM;;;;cCmgBH,mBAAoB;EAC/B;MACE,QAAQ;;;KCtcA;KAGA;KAEA,iBAAiB,YAAY;cAE5B,0BAA0B;KAU3B;EACV;EACA;EACA;EACA;EACA,QAAQ;;;KAIE;EACV;EACA;EACA,MAAM;;;;;;KASI,kBAAkB,UAAU,eAAa,KAAK;EACxD,OAAO,MAAM,qBAAqB;;KAGxB;EAA0B,WAAW;;KACrC;EAAyB;;KACzB,sBAAsB,UAAU;EAC1C,SAAS;;EAET;;KAEU;EAA4B;;KACnC;EACH;EACA;;EAEA;;KAEU,4BAA4B;EAEhC;EAAsB;;EACtB;EAAmB;;KAEf;EACV;IAAY;IAAgB;;EAC5B;EACA,QAAQ;EACR;EACA;;KAEU;EACV;EACA;EACA;;KAEU,wBAAwB;KACxB;EACV;EACA;EACA;EACA;EACA;;EAEA;EACA;;KAEU;;EAEV;EACA,QAAQ;;KAEE;EACV;EACA;EACA;EACA;EACA,eAAe;EACf,QAAQ;EACR;;KAEU;EACV;EACA;EACA;EACA;EACA;;EAEA;EACA;;KAEU;EACV;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;;KAEU;EACV;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;;KAEU;EACV;EACA;EACA;EACA;EACA;EACA;EACA;;KAEU;EACV;EACA;EACA;EACA;EACA;EACA;;KAEU;EACV;EACA;EACA;EACA;EACA;EACA,WAAW;;KAED;EACV;EACA;EACA;EACA;EACA;EACA,OAAO;;KAEG;EACV;EACA;;KAEU;EAAkC;;KAClC;EACV;EACA,QAAQ;;KAEE;EACN;;EACA;EAAoB,QAAQ;;EAC5B;;KAEM;EACV;EACA;EACA;EACA;EACA;EACA;;KAEU,2BAA2B,UAAU;EAC/C;EACA;EACA;EACA,UAAU;EACV;EACA;EACA,QAAQ;;KAGE,YAAY,UAAU,cAAY;EAC5C,wBAAwB,iBAAiB,eAAe;EACxD,wBAAwB,iBAAiB;EACzC,sBAAsB,iBAAiB;EACvC,oBAAoB,iBAAiB;EACrC,+BAA+B,iBAAiB;EAChD,wBAAwB;IACtB;IACA;;EAEF,sBAAsB,iBAAiB;EACvC,qBAAqB,iBAAiB;EACtC,sBAAsB,iBAAiB,sBAAsB;EAC7D,wBAAwB,iBAAiB;EACzC,0BAA0B,iBAAiB;EAC3C,2BAA2B,iBAAiB;EAC5C,sBAAsB,iBAAiB;EACvC,yBAAyB,iBAAiB;EAC1C,0BAA0B,iBAAiB;EAC3C,2BAA2B,iBAAiB;EAC5C,wBAAwB,iBAAiB;EACzC,kBAAkB,iBAAiB;EACnC,kBAAkB,iBAAiB;EACnC,yBAAyB,iBAAiB;EAC1C,yBAAyB,iBAAiB;EAC1C,sBAAsB,iBAAiB;EACvC,sBAAsB,iBAAiB;EACvC,2BAA2B,iBAAiB;EAC5C,2BAA2B,iBAAiB,2BAA2B;EACvE,+BAA+B,iBAAiB;EAChD,8BAA8B,iBAAiB;;KAGrC,mBAAmB;;EAE7B;;KAGU;EACV;EACA;;EAEA;EACA,OAAO;EACP;EACA;IAAW;IAAsB,QAAQ;;EACzC,iBAAiB;EACjB,SAAS;EACT,WAAW;EACX,YAAY;IACN;IAAe;IAAwB,SAAS;;IAChD;IAAe;IAAwB,SAAS;;EAEtD,gBAAgB;IAEV;IACA;IACA,SAAS;;IAGT;IACA;IACA,SAAS;;;KAKL,kBAAkB;KAClB,eAAe;KAEf;EACV;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;EACA;;KAGU;EACV;EACA;EACA;EACA;EACA;EACA,QAAQ;;KAGE;EACV;EACA,OAAO;;KAGG,QAAQ,UAAU,cAAY;EACxC;IAAS;IAAiB,OAAO,UAAU;;EAC3C,uBAAuB,oBAAoB;EAC3C,QAAQ,WAAW;EACnB;EACA,SAAS;EACT,sBAAsB;EACtB;IAAW;IAAkB,WAAW;;EACxC,UAAU,kBAAkB;EAC5B,QAAQ;EACR,kBAAkB;EAClB;EACA,qBAAqB;EACrB;EACA;EACA,kBAAkB;EAClB,gBAAgB,eAEd,KAAK;EAEP,qBAAqB;EACrB,sBAAsB;EAItB,kBAAkB;EAClB,eAAe;EACf,OAAO;EACP,YAAY;EACZ,eAAe,eAAe;EAC9B,OAAO;EACP;;;KAIU,YAAY,KAAK;WAClB,4BAA4B,UAAU;IAE7C,iBAAiB,YAAY,WAAW,4BAChC,UAAU,eAEhB,IACA,cACF;;KAGQ,cACV,UAAU,cAAY,aACtB,UAAU,YAAY,YAAY,IAClC,mBAAmB,wBACjB,SAAS,GAAG,sBAAsB,aAAa,GAAG,GAAG;KAE7C,qBAAqB,YAC/B,KAAK;;KAGK,iBAAiB,UAAU,sBAAsB,eAC3D;;;KAMG,QAAQ,QAAQ,KAAK,cAAc,SAAS;;KAE5C,SAAS,WAAW;;;;;;;KAQb,aACV,UAAU,aACV,UAAU,WACV,2BACS,kBACP,QAAQ,GAAG,QAAQ;WAAiB,aAAa;IACjD,QAAQ,GAAG,QAAQ,KAAK;WAAiB,aAAa;;KAE9C,qBACV,UAAU,aACV,UAAU,WACV;EAEA;EACA,SAAS;;EAET,gBAAgB,iBAAiB;;;;;EAKjC;;IAEE;IACA,cAAc,IAAI,QAAQ,KAAK,SAAS;IACxC,OACE,OAAO,QAAQ,IACf,OAAO,cAAc,YAAY,KAAK,QAAQ,QAC3C,QAAQ;;;iBA6MD,aAAa,UAAU,cAAY,aAAW;EAC5D,gBAAgB,iBAAiB;IAC/B,YAAY;cAsUH,QAAQ,YAAY;;iBA2GjB,cAAc,UAAU,aACtC,OAAO,QAAQ,IACf;EAAS;EAAc;EAAkB;EAAe;IACvD,QAAQ;;iBAmrCK,iBAAiB,UAAU,cAAY,aACrD;EAAoB;EAAc;EAAkB;MACnD,kBAAkB;KAMhB,aAAa;EAAO;EAA8B,SAAS;EAAG;;KAC9D,aACH,UAAU,aACV,UAAU,eAAe,gBACvB,aAAa,QAAQ,eAAe;EAAM,QAAQ;OACnD;EAA+B;;KAEtB;EACV,QAAQ,UAAU,aAChB,SAAS,GACT;IAAY;MACX,aAAa;EAChB,KAAK,UAAU,aAAW,SAAS,IAAI,aAAa;EACpD;IACE,OAAO;MAAW;QAAoB,aAAa;IACnD,QAAQ;MACN;MACA;MACA;QACE,aAAa;IACjB,KAAK,UAAU,aAAW;MACxB,SAAS;QACP,aAAa;IACjB,KAAK;MACH;MACA;QACE,aAAa;;EAEnB,KAAK;IACH;IACA;MACE,aAAa;EACjB,MAAM,UAAU,aAAW;IACzB;IACA,SAAS;IACT;MACE,aAAa;EACjB,MAAM;IACJ;MACE,aAAa;EACjB,UAAU,aAAa;EACvB,SACE,UAAU,2BACT,aAAa;EAChB,MAAM,UAAU,wBAAwB,aAAa;EACrD,aACE,SAAS,wBACR,aAAa;EAChB,MAAM,SAAS,yBAAyB,aAAa;EACrD,UACE,SAAS,4BACR,aAAa;EAChB,WACE,QAAQ,oBACP,aAAa;;;cAsBL,QAAQ;;;;;;;;;iBAgEL,eAAe,UAAU,qBAAqB,aAC5D,WAAW,SAAS,KACnB,iBAAiB;;cAoBP;;;;;;;iBAQG,cACd,UAAU,cAAY,aACtB,UAAU,YAAY,sBACtB,mBAAmB,sBAEnB,SAAS,qBAAqB,GAAG,GAAG,KACnC,cAAc,GAAG,YAAY,KAAK,GAAG"}

package/dist/ai-harness-BuLh_EYi.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"ai-harness-BuLh_EYi.js","names":[],"sources":["../src/ai-harness.ts"],"sourcesContent":["import type {\n  Experimental_SandboxSession,\n  FinishReason,\n  LanguageModelUsage,\n  ToolSet,\n  UIMessage,\n  UIMessageChunk,\n} from 'ai'\nimport type {\n  GenerationCompletedPayload,\n  ReasoningLevel,\n  ToolResultPayload,\n} from './ai.ts'\nimport { boundedCache, type BoundedCache } from './bounded-cache.ts'\nimport type { ContractEvent, EventDefs } from './contract.ts'\n\ndeclare const harnessBrand: unique symbol\n\n/**\n * A coding agent's own loop for `agent({ harness })`, from `fx()` or `pi()`.\n * `Tools` names the tools the harness brings, which `toolRecovery` may name.\n */\nexport type Harness<\n  Type extends 'fx' | 'pi' = 'fx' | 'pi',\n  Tools extends string = string,\n> = {\n  readonly type: Type\n  readonly [harnessBrand]: { readonly tools: Tools }\n}\n\n/** What a harness's sandbox resolver receives for one step or tool execution. */\nexport type HarnessSandboxContext = {\n  session: { readonly id: string }\n  signal: AbortSignal\n}\n\n/** Resolves the AI SDK sandbox session for an agent session, the same one every time. */\nexport type HarnessSandbox = (\n  context: HarnessSandboxContext,\n) => Experimental_SandboxSession | Promise<Experimental_SandboxSession>\n\n/** What a harness step streams, as the default generator does. */\nexport type HarnessGeneration = {\n  stream: ReadableStream<UIMessageChunk>\n  completion: () => { finishReason?: FinishReason; usage?: LanguageModelUsage }\n}\n\n/** One stored snapshot and the log after it, read in one store operation. */\nexport type HarnessLogRead = {\n  snapshot: { index: number; state: unknown } | null\n  events: ContractEvent<EventDefs>[]\n}\n\n/** The session log as a harness step reads it, plus the state saved in it. */\nexport type HarnessLog = {\n  /** The latest saved state under `name` and the log after it; `fresh` reads it all. */\n  read(options: {\n    name: string\n    throughIndex: number\n    fresh?: boolean\n  }): Promise<HarnessLogRead>\n  write(options: { name: string; index: number; state: unknown }): Promise<void>\n}\n\n/** Everything one durable generation hands its harness. */\nexport type HarnessStep = {\n  agentName: string\n  sessionId: string\n  requestIndex: number\n  /** This attempt at the generation. */\n  generationId: string\n  responseMessageId: string\n  /** The log index this step's context was read through. */\n  throughIndex: number\n  /** The transcript through that index. */\n  messages: readonly UIMessage[]\n  model: string\n  reasoning?: ReasoningLevel\n  speed?: 'fast' | 'standard'\n  instructions: string[]\n  tools: ToolSet\n  log: HarnessLog\n  signal: AbortSignal\n}\n\n/** The internal side of a harness value. */\nexport type HarnessDriver = {\n  /** The session's sandbox; every local tool execution receives it. */\n  readonly sandbox?: HarnessSandbox\n  /** Tools the harness brings; A2 executes them like the agent's own. */\n  readonly tools: ToolSet\n  step(step: HarnessStep): Promise<HarnessGeneration>\n}\n\n// A global symbol, so harness values work across separately bundled copies.\nconst DRIVER = Symbol.for('experimental-a2.ai.harness')\n\nexport function defineHarness<\n  Type extends 'fx' | 'pi',\n  Tools extends string = never,\n>(type: Type, driver: HarnessDriver): Harness<Type, Tools> {\n  const harness = { type }\n  Object.defineProperty(harness, DRIVER, { value: driver })\n  return Object.freeze(harness) as unknown as Harness<Type, Tools>\n}\n\nexport function harnessDriver(harness: unknown): HarnessDriver {\n  const driver =\n    typeof harness === 'object' && harness !== null\n      ? (harness as { [DRIVER]?: HarnessDriver })[DRIVER]\n      : undefined\n  if (driver === undefined)\n    throw new TypeError(\n      \"harness must be { type: 'ai-sdk' }, or come from fx() or pi()\",\n    )\n  return driver\n}\n\n// ------------------------------------------------------- saved state\n\n/**\n * What an agent's steps saved in this process at the end of a response, per\n * session and harness. A cache: a cold process rebuilds from the whole log.\n */\nexport type HarnessStates = BoundedCache<\n  string,\n  { index: number; state: unknown }\n>\n\nexport const harnessStates = (): HarnessStates =>\n  boundedCache({ maxEntries: 1_000, maxBytes: 64 * 1024 * 1024 })\n\nconst savedKey = (sessionId: string, name: string): string =>\n  `${sessionId}\\u0000${name}`\n\n// Binary state counts its bytes; everything else its JSON text.\nconst sizeOf = (value: unknown): number => {\n  let binary = 0\n  const text = JSON.stringify(value, (_key, item: unknown) => {\n    if (!(item instanceof Uint8Array)) return item\n    binary += item.byteLength\n    return null\n  })\n  return binary + (text?.length ?? 0)\n}\n\n/**\n * The session log as a harness step reads it: the state this process saved\n * at the end of the last response, when it has one, and the log after it.\n * Saved states are copied in and out, so a step cannot change one in place.\n */\nexport function harnessLog(options: {\n  states: HarnessStates\n  session: {\n    readonly id: string\n    history(bounds: {\n      gte?: number\n      lte?: number\n    }): Promise<ContractEvent<EventDefs>[]>\n  }\n}): HarnessLog {\n  const { states, session } = options\n  return {\n    read: async ({ name, throughIndex, fresh }) => {\n      const entry = fresh ? undefined : states.get(savedKey(session.id, name))\n      const snapshot =\n        entry === undefined || entry.index > throughIndex\n          ? null\n          : { index: entry.index, state: structuredClone(entry.state) }\n      const events = await session.history(\n        snapshot === null\n          ? { lte: throughIndex }\n          : { gte: snapshot.index + 1, lte: throughIndex },\n      )\n      return { snapshot, events }\n    },\n    write: async ({ name, index, state }) => {\n      const copy = structuredClone(state)\n      states.set(\n        savedKey(session.id, name),\n        { index, state: copy },\n        sizeOf(copy),\n      )\n    },\n  }\n}\n\nconst agentStates = new WeakMap<object, HarnessStates>()\n\n/** Registers an agent server's saved harness states, for `savedHarnessState`. */\nexport function registerHarnessStates(\n  server: object,\n  states: HarnessStates,\n): void {\n  agentStates.set(server, states)\n}\n\n/** What an agent's process saved for one session's harness; tests read it. */\nexport function savedHarnessState(\n  server: object,\n  options: { sessionId: string; name: string },\n): { index: number; state: unknown } | undefined {\n  return agentStates.get(server)?.get(savedKey(options.sessionId, options.name))\n}\n\n/** A harness's state at the end of a response, from the attempt that ran it. */\ntype SavedEnvelope = {\n  generationId: string\n  responseMessageId: string\n  state: unknown\n}\n\nexport type HarnessRead<S> = {\n  /** The state saved at the end of the last response, when it is usable. */\n  saved: S | undefined\n  turns: HarnessTurn[]\n  results: Map<string, ToolResultPayload>\n}\n\n/**\n * Reads a harness's saved state and the log since, as turns to replay. A\n * snapshot counts only when the attempt that wrote it committed: a stopped,\n * superseded, or still-running attempt can write one the log never kept.\n * `replayAll` reads the whole log instead, for a state that fails to restore.\n */\nexport async function readHarness<S>({\n  step,\n  name,\n  parse,\n}: {\n  step: HarnessStep\n  name: string\n  parse: (state: unknown) => S | undefined\n}): Promise<HarnessRead<S> & { replayAll: () => Promise<HarnessRead<S>> }> {\n  const replayAll = async (): Promise<HarnessRead<S>> => {\n    const read = await step.log.read({\n      name,\n      throughIndex: step.throughIndex,\n      fresh: true,\n    })\n    return { saved: undefined, ...harnessTurns(step, read.events, undefined) }\n  }\n  const read = await step.log.read({ name, throughIndex: step.throughIndex })\n  if (read.snapshot === null)\n    return {\n      saved: undefined,\n      ...harnessTurns(step, read.events, undefined),\n      replayAll,\n    }\n  const envelope = savedEnvelope(read.snapshot.state)\n  const saved = envelope === undefined ? undefined : parse(envelope.state)\n  const committed =\n    envelope !== undefined &&\n    read.events.some(\n      (event) =>\n        event.type === 'ai.generation.completed' &&\n        (event.payload as { generationId?: unknown }).generationId ===\n          envelope.generationId,\n    )\n  // A snapshot of this very response belongs to an earlier attempt at it.\n  if (\n    saved === undefined ||\n    !committed ||\n    envelope.responseMessageId === step.responseMessageId\n  )\n    return { ...(await replayAll()), replayAll }\n  return {\n    saved,\n    ...harnessTurns(step, read.events, envelope.responseMessageId),\n    replayAll,\n  }\n}\n\n/** Saves a harness's state at the end of a response; losing it costs a longer replay. */\nexport async function saveHarness({\n  step,\n  name,\n  state,\n}: {\n  step: HarnessStep\n  name: string\n  state: unknown\n}): Promise<void> {\n  const envelope: SavedEnvelope = {\n    generationId: step.generationId,\n    responseMessageId: step.responseMessageId,\n    state,\n  }\n  try {\n    await step.log.write({ name, index: step.requestIndex, state: envelope })\n  } catch {\n    // The snapshot is a cache.\n  }\n}\n\nfunction savedEnvelope(value: unknown): SavedEnvelope | undefined {\n  if (typeof value !== 'object' || value === null) return undefined\n  const { generationId, responseMessageId, state } = value as Record<\n    string,\n    unknown\n  >\n  return typeof generationId === 'string' &&\n    typeof responseMessageId === 'string'\n    ? { generationId, responseMessageId, state }\n    : undefined\n}\n\n// ------------------------------------------------- the log as harness turns\n\n// A2's reason on `ai.message.interrupted` when a steer ends a response.\nconst STEERED = 'steered after step'\n\n/** A completed generation as the log recorded it. */\nexport type RecordedGeneration = {\n  chunks: UIMessageChunk[]\n  completed: GenerationCompletedPayload\n}\n\n/** One harness turn to rebuild: its prompt, completed generations, and end. */\nexport type HarnessTurn = {\n  prompt: string\n  generations: RecordedGeneration[]\n  /** Steering that entered after the generation at that index. */\n  steers: Map<number, string[]>\n  end: 'stop' | 'interrupted' | 'live'\n}\n\n/**\n * Groups the log since a harness snapshot into harness turns. A response\n * steered after a step continues its turn, with the next response's prompt as\n * steering. `covered` is the response the snapshot already holds.\n */\nexport function harnessTurns(\n  step: HarnessStep,\n  events: readonly ContractEvent<EventDefs>[],\n  covered: string | undefined,\n): { turns: HarnessTurn[]; results: Map<string, ToolResultPayload> } {\n  type Generation = {\n    id: string\n    chunks: UIMessageChunk[]\n    completed?: GenerationCompletedPayload\n  }\n  type Response = {\n    id: string\n    prompt: string[]\n    generations: Generation[]\n    end?: 'stop' | 'steered' | 'interrupted'\n  }\n  // Queued messages are admitted together, possibly before the snapshot.\n  const userText = new Map<string, string>(\n    step.messages\n      .filter((message) => message.role === 'user')\n      .map((message) => [message.id, textOf(message)]),\n  )\n  let passive: string[] = []\n  const responses = new Map<string, Response>()\n  const byStart = new Map<number, Generation>()\n  const byId = new Map<string, Generation>()\n  const results = new Map<string, ToolResultPayload>()\n\n  for (const event of events) {\n    const payload = event.payload as Record<string, unknown>\n    switch (event.type) {\n      case 'ai.message.created': {\n        const message = payload['message'] as UIMessage\n        if (message.role !== 'user') break\n        userText.set(message.id, textOf(message))\n        if (payload['generate'] === false) passive.push(message.id)\n        break\n      }\n      case 'ai.generation.requested': {\n        const id = payload['responseMessageId'] as string\n        if (id === covered || responses.has(id)) break\n        const messageId = payload['messageId'] as string\n        const ids = [...passive.filter((p) => p !== messageId), messageId]\n        passive = []\n        responses.set(id, {\n          id,\n          prompt: ids.flatMap((p) => userText.get(p) ?? []),\n          generations: [],\n        })\n        break\n      }\n      case 'ai.generation.started': {\n        const response = responses.get(payload['responseMessageId'] as string)\n        if (response === undefined) break\n        const generation: Generation = {\n          id: payload['generationId'] as string,\n          chunks: [],\n        }\n        response.generations.push(generation)\n        byStart.set(event.index, generation)\n        byId.set(generation.id, generation)\n        break\n      }\n      case 'ai.generation.progress':\n        byStart\n          .get(payload['startIndex'] as number)\n          ?.chunks.push(...(payload['chunks'] as UIMessageChunk[]))\n        break\n      case 'ai.generation.completed': {\n        const generation = byId.get(payload['generationId'] as string)\n        if (generation)\n          generation.completed = event.payload as GenerationCompletedPayload\n        break\n      }\n      case 'ai.tool.result':\n        if (payload['preliminary'] !== true)\n          results.set(\n            payload['toolCallId'] as string,\n            event.payload as ToolResultPayload,\n          )\n        break\n      case 'ai.message.completed': {\n        const response = responses.get(payload['messageId'] as string)\n        if (response) response.end = 'stop'\n        break\n      }\n      case 'ai.message.interrupted': {\n        const response = responses.get(payload['messageId'] as string)\n        if (response)\n          response.end =\n            payload['reason'] === STEERED ? 'steered' : 'interrupted'\n        break\n      }\n    }\n  }\n\n  const turns: HarnessTurn[] = []\n  let steered = false\n  for (const response of responses.values()) {\n    const text = response.prompt.join('\\n\\n')\n    let turn = turns.at(-1)\n    if (steered && turn !== undefined) {\n      const at = Math.max(0, turn.generations.length - 1)\n      turn.steers.set(at, [...(turn.steers.get(at) ?? []), text])\n    } else {\n      turn = { prompt: text, generations: [], steers: new Map(), end: 'live' }\n      turns.push(turn)\n    }\n    for (const generation of response.generations)\n      if (generation.completed !== undefined)\n        turn.generations.push({\n          chunks: generation.chunks,\n          completed: generation.completed,\n        })\n    turn.end =\n      response.id === step.responseMessageId\n        ? 'live'\n        : response.end === 'stop'\n          ? 'stop'\n          : 'interrupted'\n    steered = response.end === 'steered'\n  }\n  return { turns, results }\n}\n\nfunction textOf(message: UIMessage): string {\n  return message.parts\n    .flatMap((part) => (part.type === 'text' ? [part.text] : []))\n    .join('\\n')\n}\n\n// ------------------------------------------------------------- Gateway\n\nexport type GatewayCredential = { token: string; method: 'api-key' | 'oidc' }\n\n/** Read per step: on Vercel the OIDC token rides on each function request. */\nexport function gatewayCredential(harness: string): GatewayCredential {\n  const key = process.env['AI_GATEWAY_API_KEY']\n  if (key) return { token: key, method: 'api-key' }\n  const oidc = requestOidcToken() ?? process.env['VERCEL_OIDC_TOKEN']\n  if (oidc) return { token: oidc, method: 'oidc' }\n  throw new Error(`${harness}() needs AI_GATEWAY_API_KEY or VERCEL_OIDC_TOKEN`)\n}\n\n// Where Vercel Functions keep the current request, as @vercel/oidc reads it.\nconst REQUEST_CONTEXT = Symbol.for('@vercel/request-context')\n\nfunction requestOidcToken(): string | undefined {\n  const context = (\n    globalThis as {\n      [REQUEST_CONTEXT]?: { get?: () => { headers?: Record<string, unknown> } }\n    }\n  )[REQUEST_CONTEXT]\n  const token = context?.get?.()?.headers?.['x-vercel-oidc-token']\n  return typeof token === 'string' && token !== '' ? token : undefined\n}\n\n// ------------------------------------------------------------- helpers\n\nexport class Divergence extends Error {\n  constructor(harness: string, message: string) {\n    super(`${harness} replay diverged from the log: ${message}`)\n  }\n}\n\nexport type Deferred = { promise: Promise<void>; resolve: () => void }\n\nexport function deferred(): Deferred {\n  let resolve!: () => void\n  const promise = new Promise<void>((done) => {\n    resolve = done\n  })\n  return { promise, resolve }\n}\n\n/** A result that never settles; the harness is stopped instead. */\nexport function pending(\n  signal: AbortSignal | null | undefined,\n): Promise<never> {\n  return new Promise((_, reject) => {\n    if (signal?.aborted) reject(signal.reason)\n    signal?.addEventListener('abort', () => reject(signal.reason), {\n      once: true,\n    })\n  })\n}\n"],"mappings":";;AA+FA,MAAM,SAAS,OAAO,IAAI,4BAA4B;AAEtD,SAAgB,cAGd,MAAY,QAA6C;CACzD,MAAM,UAAU,EAAE,KAAK;CACvB,OAAO,eAAe,SAAS,QAAQ,EAAE,OAAO,OAAO,CAAC;CACxD,OAAO,OAAO,OAAO,OAAO;AAC9B;AAEA,SAAgB,cAAc,SAAiC;CAC7D,MAAM,SACJ,OAAO,YAAY,YAAY,YAAY,OACtC,QAAyC,UAC1C,KAAA;CACN,IAAI,WAAW,KAAA,GACb,MAAM,IAAI,UACR,+DACF;CACF,OAAO;AACT;AAaA,MAAa,sBACX,aAAa;CAAE,YAAY;CAAO,UAAU;AAAiB,CAAC;AAEhE,MAAM,YAAY,WAAmB,SACnC,GAAG,UAAU,QAAQ;AAGvB,MAAM,UAAU,UAA2B;CACzC,IAAI,SAAS;CACb,MAAM,OAAO,KAAK,UAAU,QAAQ,MAAM,SAAkB;EAC1D,IAAI,EAAE,gBAAgB,aAAa,OAAO;EAC1C,UAAU,KAAK;EACf,OAAO;CACT,CAAC;CACD,OAAO,UAAU,MAAM,UAAU;AACnC;;;;;;AAOA,SAAgB,WAAW,SASZ;CACb,MAAM,EAAE,QAAQ,YAAY;CAC5B,OAAO;EACL,MAAM,OAAO,EAAE,MAAM,cAAc,YAAY;GAC7C,MAAM,QAAQ,QAAQ,KAAA,IAAY,OAAO,IAAI,SAAS,QAAQ,IAAI,IAAI,CAAC;GACvE,MAAM,WACJ,UAAU,KAAA,KAAa,MAAM,QAAQ,eACjC,OACA;IAAE,OAAO,MAAM;IAAO,OAAO,gBAAgB,MAAM,KAAK;GAAE;GAMhE,OAAO;IAAE;IAAU,QAAA,MALE,QAAQ,QAC3B,aAAa,OACT,EAAE,KAAK,aAAa,IACpB;KAAE,KAAK,SAAS,QAAQ;KAAG,KAAK;IAAa,CACnD;GAC0B;EAC5B;EACA,OAAO,OAAO,EAAE,MAAM,OAAO,YAAY;GACvC,MAAM,OAAO,gBAAgB,KAAK;GAClC,OAAO,IACL,SAAS,QAAQ,IAAI,IAAI,GACzB;IAAE;IAAO,OAAO;GAAK,GACrB,OAAO,IAAI,CACb;EACF;CACF;AACF;AAEA,MAAM,8BAAc,IAAI,QAA+B;;AAGvD,SAAgB,sBACd,QACA,QACM;CACN,YAAY,IAAI,QAAQ,MAAM;AAChC;;;;;;;AA8BA,eAAsB,YAAe,EACnC,MACA,MACA,SAKyE;CACzE,MAAM,YAAY,YAAqC;EAMrD,OAAO;GAAE,OAAO,KAAA;GAAW,GAAG,aAAa,OAAM,MAL9B,KAAK,IAAI,KAAK;IAC/B;IACA,cAAc,KAAK;IACnB,OAAO;GACT,CAAC,EAAA,CACqD,QAAQ,KAAA,CAAS;EAAE;CAC3E;CACA,MAAM,OAAO,MAAM,KAAK,IAAI,KAAK;EAAE;EAAM,cAAc,KAAK;CAAa,CAAC;CAC1E,IAAI,KAAK,aAAa,MACpB,OAAO;EACL,OAAO,KAAA;EACP,GAAG,aAAa,MAAM,KAAK,QAAQ,KAAA,CAAS;EAC5C;CACF;CACF,MAAM,WAAW,cAAc,KAAK,SAAS,KAAK;CAClD,MAAM,QAAQ,aAAa,KAAA,IAAY,KAAA,IAAY,MAAM,SAAS,KAAK;CACvE,MAAM,YACJ,aAAa,KAAA,KACb,KAAK,OAAO,MACT,UACC,MAAM,SAAS,6BACd,MAAM,QAAuC,iBAC5C,SAAS,YACf;CAEF,IACE,UAAU,KAAA,KACV,CAAC,aACD,SAAS,sBAAsB,KAAK,mBAEpC,OAAO;EAAE,GAAI,MAAM,UAAU;EAAI;CAAU;CAC7C,OAAO;EACL;EACA,GAAG,aAAa,MAAM,KAAK,QAAQ,SAAS,iBAAiB;EAC7D;CACF;AACF;;AAGA,eAAsB,YAAY,EAChC,MACA,MACA,SAKgB;CAChB,MAAM,WAA0B;EAC9B,cAAc,KAAK;EACnB,mBAAmB,KAAK;EACxB;CACF;CACA,IAAI;EACF,MAAM,KAAK,IAAI,MAAM;GAAE;GAAM,OAAO,KAAK;GAAc,OAAO;EAAS,CAAC;CAC1E,QAAQ,CAER;AACF;AAEA,SAAS,cAAc,OAA2C;CAChE,IAAI,OAAO,UAAU,YAAY,UAAU,MAAM,OAAO,KAAA;CACxD,MAAM,EAAE,cAAc,mBAAmB,UAAU;CAInD,OAAO,OAAO,iBAAiB,YAC7B,OAAO,sBAAsB,WAC3B;EAAE;EAAc;EAAmB;CAAM,IACzC,KAAA;AACN;AAKA,MAAM,UAAU;;;;;;AAsBhB,SAAgB,aACd,MACA,QACA,SACmE;CAanE,MAAM,WAAW,IAAI,IACnB,KAAK,SACF,QAAQ,YAAY,QAAQ,SAAS,MAAM,CAAC,CAC5C,KAAK,YAAY,CAAC,QAAQ,IAAI,OAAO,OAAO,CAAC,CAAC,CACnD;CACA,IAAI,UAAoB,CAAC;CACzB,MAAM,4BAAY,IAAI,IAAsB;CAC5C,MAAM,0BAAU,IAAI,IAAwB;CAC5C,MAAM,uBAAO,IAAI,IAAwB;CACzC,MAAM,0BAAU,IAAI,IAA+B;CAEnD,KAAK,MAAM,SAAS,QAAQ;EAC1B,MAAM,UAAU,MAAM;EACtB,QAAQ,MAAM,MAAd;GACE,KAAK,sBAAsB;IACzB,MAAM,UAAU,QAAQ;IACxB,IAAI,QAAQ,SAAS,QAAQ;IAC7B,SAAS,IAAI,QAAQ,IAAI,OAAO,OAAO,CAAC;IACxC,IAAI,QAAQ,gBAAgB,OAAO,QAAQ,KAAK,QAAQ,EAAE;IAC1D;GACF;GACA,KAAK,2BAA2B;IAC9B,MAAM,KAAK,QAAQ;IACnB,IAAI,OAAO,WAAW,UAAU,IAAI,EAAE,GAAG;IACzC,MAAM,YAAY,QAAQ;IAC1B,MAAM,MAAM,CAAC,GAAG,QAAQ,QAAQ,MAAM,MAAM,SAAS,GAAG,SAAS;IACjE,UAAU,CAAC;IACX,UAAU,IAAI,IAAI;KAChB;KACA,QAAQ,IAAI,SAAS,MAAM,SAAS,IAAI,CAAC,KAAK,CAAC,CAAC;KAChD,aAAa,CAAC;IAChB,CAAC;IACD;GACF;GACA,KAAK,yBAAyB;IAC5B,MAAM,WAAW,UAAU,IAAI,QAAQ,oBAA8B;IACrE,IAAI,aAAa,KAAA,GAAW;IAC5B,MAAM,aAAyB;KAC7B,IAAI,QAAQ;KACZ,QAAQ,CAAC;IACX;IACA,SAAS,YAAY,KAAK,UAAU;IACpC,QAAQ,IAAI,MAAM,OAAO,UAAU;IACnC,KAAK,IAAI,WAAW,IAAI,UAAU;IAClC;GACF;GACA,KAAK;IACH,QACG,IAAI,QAAQ,aAAuB,CAAC,EACnC,OAAO,KAAK,GAAI,QAAQ,SAA8B;IAC1D;GACF,KAAK,2BAA2B;IAC9B,MAAM,aAAa,KAAK,IAAI,QAAQ,eAAyB;IAC7D,IAAI,YACF,WAAW,YAAY,MAAM;IAC/B;GACF;GACA,KAAK;IACH,IAAI,QAAQ,mBAAmB,MAC7B,QAAQ,IACN,QAAQ,eACR,MAAM,OACR;IACF;GACF,KAAK,wBAAwB;IAC3B,MAAM,WAAW,UAAU,IAAI,QAAQ,YAAsB;IAC7D,IAAI,UAAU,SAAS,MAAM;IAC7B;GACF;GACA,KAAK,0BAA0B;IAC7B,MAAM,WAAW,UAAU,IAAI,QAAQ,YAAsB;IAC7D,IAAI,UACF,SAAS,MACP,QAAQ,cAAc,UAAU,YAAY;IAChD;GACF;EACF;CACF;CAEA,MAAM,QAAuB,CAAC;CAC9B,IAAI,UAAU;CACd,KAAK,MAAM,YAAY,UAAU,OAAO,GAAG;EACzC,MAAM,OAAO,SAAS,OAAO,KAAK,MAAM;EACxC,IAAI,OAAO,MAAM,GAAG,EAAE;EACtB,IAAI,WAAW,SAAS,KAAA,GAAW;GACjC,MAAM,KAAK,KAAK,IAAI,GAAG,KAAK,YAAY,SAAS,CAAC;GAClD,KAAK,OAAO,IAAI,IAAI,CAAC,GAAI,KAAK,OAAO,IAAI,EAAE,KAAK,CAAC,GAAI,IAAI,CAAC;EAC5D,OAAO;GACL,OAAO;IAAE,QAAQ;IAAM,aAAa,CAAC;IAAG,wBAAQ,IAAI,IAAI;IAAG,KAAK;GAAO;GACvE,MAAM,KAAK,IAAI;EACjB;EACA,KAAK,MAAM,cAAc,SAAS,aAChC,IAAI,WAAW,cAAc,KAAA,GAC3B,KAAK,YAAY,KAAK;GACpB,QAAQ,WAAW;GACnB,WAAW,WAAW;EACxB,CAAC;EACL,KAAK,MACH,SAAS,OAAO,KAAK,oBACjB,SACA,SAAS,QAAQ,SACf,SACA;EACR,UAAU,SAAS,QAAQ;CAC7B;CACA,OAAO;EAAE;EAAO;CAAQ;AAC1B;AAEA,SAAS,OAAO,SAA4B;CAC1C,OAAO,QAAQ,MACZ,SAAS,SAAU,KAAK,SAAS,SAAS,CAAC,KAAK,IAAI,IAAI,CAAC,CAAE,CAAC,CAC5D,KAAK,IAAI;AACd;;AAOA,SAAgB,kBAAkB,SAAoC;CACpE,MAAM,MAAM,QAAQ,IAAI;CACxB,IAAI,KAAK,OAAO;EAAE,OAAO;EAAK,QAAQ;CAAU;CAChD,MAAM,OAAO,iBAAiB,KAAK,QAAQ,IAAI;CAC/C,IAAI,MAAM,OAAO;EAAE,OAAO;EAAM,QAAQ;CAAO;CAC/C,MAAM,IAAI,MAAM,GAAG,QAAQ,iDAAiD;AAC9E;AAGA,MAAM,kBAAkB,OAAO,IAAI,yBAAyB;AAE5D,SAAS,mBAAuC;CAM9C,MAAM,QAJJ,WAGA,gBACmB,EAAE,MAAM,CAAC,EAAE,UAAU;CAC1C,OAAO,OAAO,UAAU,YAAY,UAAU,KAAK,QAAQ,KAAA;AAC7D;AAIA,IAAa,aAAb,cAAgC,MAAM;CACpC,YAAY,SAAiB,SAAiB;EAC5C,MAAM,GAAG,QAAQ,iCAAiC,SAAS;CAC7D;AACF;AAIA,SAAgB,WAAqB;CACnC,IAAI;CAIJ,OAAO;EAAE,SAAA,IAHW,SAAe,SAAS;GAC1C,UAAU;EACZ,CACe;EAAG;CAAQ;AAC5B;;AAGA,SAAgB,QACd,QACgB;CAChB,OAAO,IAAI,SAAS,GAAG,WAAW;EAChC,IAAI,QAAQ,SAAS,OAAO,OAAO,MAAM;EACzC,QAAQ,iBAAiB,eAAe,OAAO,OAAO,MAAM,GAAG,EAC7D,MAAM,KACR,CAAC;CACH,CAAC;AACH"}

package/dist/ai-harness-DvsPeJeB.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"ai-harness-DvsPeJeB.d.ts","names":[],"sources":["../src/ai-harness.ts"],"mappings":";;;cAgBc;;;;;KAMF,QACV,wCACA;WAES,MAAM;YACL;aAA0B,OAAO;;;;KAIjC;EACV;aAAoB;;EACpB,QAAQ;;;KAIE,kBACV,SAAS,0BACN,8BAA8B,QAAQ"}