@athenaintel/react 0.41.3 → 0.42.1

package/dist/agent-runtime/src/message-window.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"message-window.cjs","sources":["../../../../agent-runtime/src/message-window.ts"],"sourcesContent":["import type { LangChainMessage } from '@assistant-ui/react-langgraph';\nimport type { StatewireDocuments } from 'statewire';\nimport {\n  DEEP_AGENT_DEFAULT_MESSAGE_WINDOW,\n  DEEP_AGENT_HISTORY_MESSAGE_COMMAND,\n  DEEP_AGENT_HISTORY_TOOL_OUTPUT_COMMAND,\n  DEEP_AGENT_INTEREST_PROTOCOL,\n  DEEP_AGENT_MAIN_NS,\n  DEEP_AGENT_MAX_INTERESTS,\n  DEEP_AGENT_MAX_MESSAGE_WINDOW,\n  DEEP_AGENT_MESSAGE_WINDOW_HEADER,\n  DEEP_AGENT_MIN_MESSAGE_WINDOW,\n  DEEP_AGENT_TRUNCATED_CONTENT_KEY,\n} from './generated/runconfig-contract';\nimport { DEEP_AGENT_STATEWIRE_PROTOCOL } from './statewire-transport';\n\n/**\n * The interest window — the client half of agora's\n * `agora/graphs/athena_deep_agent/message_window.py`.\n *\n * The wire main document carries no `messages`. A client states its interests\n * on the `athena/interest` lane as one array (`set(requests)`): the first is\n * the head (the newest `before` messages of the `main` namespace), the rest\n * are pages anchored `at` the first message of the window they extend. The\n * host holds one ephemeral main document per attach, `{ windows }`, one\n * resolution per request in order (the request echoed plus the covered\n * `chain` and `more` flags; an anchor that vanished leaves an empty window in\n * place), and one shared `athena` document per covered message, keyed\n * `[threadId, \"main\", messageId]`. {@link readDeepAgentMessageWindow} joins\n * them into the transcript the converter renders.\n *\n * Two host rules shape the client half. The head window never drops below\n * the oldest message it has carried on an attach (a run's appends grow the\n * view, nothing loaded slides out), so a page stays anchored at the head's\n * first message while the transcript grows; and the host remembers a client\n * id's interests across its attaches, so a reconnect's hello carries the\n * view the client held. A fresh client's hello carries the head it declared\n * on the attach headers ({@link withDeclaredMessageWindow}) — a statement\n * cannot ride the first attach, the hello negotiates the lanes.\n *\n * Pages are appended one per load until the host's interest cap, then the\n * loaded range is restated in as few maximal requests as it takes\n * ({@link pageInterestRequests}); the whole transcript stays reachable up to\n * `MAX_INTERESTS × MAX_MESSAGE_WINDOW` messages. A truncated tool output\n * comes back whole through `history/tool_output` ({@link buildToolOutputCommand});\n * a message's stored form — provenance blobs and all — through\n * `history/message` ({@link buildHistoryMessageCommand}).\n */\n\nconst isIndex = (value: unknown): value is number =>\n  typeof value === 'number' && Number.isInteger(value) && value >= 0;\n\n/** What a surface declares for its window: the interest size (the host clamps to its bounds). */\nexport interface DeepAgentMessageWindowDeclaration {\n  size?: number;\n}\n\nexport function clampMessageWindowSize(size?: number): number {\n  if (size === undefined || !Number.isFinite(size)) return DEEP_AGENT_DEFAULT_MESSAGE_WINDOW;\n  return Math.max(\n    DEEP_AGENT_MIN_MESSAGE_WINDOW,\n    Math.min(DEEP_AGENT_MAX_MESSAGE_WINDOW, Math.trunc(size))\n  );\n}\n\n/** An `athena/interest` request over the linear transcript. */\nexport interface DeepAgentInterestRequest {\n  readonly ns: typeof DEEP_AGENT_MAIN_NS;\n  /** Messages ending at the anchor (the tail without `at`). */\n  readonly before: number;\n  /** Anchor message id; a page's is the first message of the window it extends. */\n  readonly at?: string;\n  /** Messages past the anchor (requires `at`). */\n  readonly after?: number;\n}\n\nexport function headInterestRequest(size?: number): DeepAgentInterestRequest {\n  return { ns: DEEP_AGENT_MAIN_NS, before: clampMessageWindowSize(size) };\n}\n\nexport function pageInterestRequest(at: string, size?: number): DeepAgentInterestRequest {\n  return { ns: DEEP_AGENT_MAIN_NS, before: clampMessageWindowSize(size), at };\n}\n\n/**\n * Attach headers with the head window declared (`x-athena-deep-agent-window`):\n * the hello of a client the host does not remember carries the newest `size`\n * messages. Part of the one attach recipe, so a pooled dial asks for the shape\n * the thread's own attach would. An explicit entry (tests, overrides) wins.\n */\nexport function withDeclaredMessageWindow(\n  headers: Record<string, string>,\n  declaration: DeepAgentMessageWindowDeclaration = {}\n): Record<string, string> {\n  if (DEEP_AGENT_MESSAGE_WINDOW_HEADER in headers) return headers;\n  return {\n    ...headers,\n    [DEEP_AGENT_MESSAGE_WINDOW_HEADER]: String(clampMessageWindowSize(declaration.size)),\n  };\n}\n\nexport interface DeepAgentInterestChainEntry {\n  id: string;\n  siblings: string[];\n}\n\n/** The host's resolution document of one interest: the request echoed plus its coverage. */\nexport interface DeepAgentInterestResolution extends DeepAgentInterestRequest {\n  /** Covered message ids, oldest first (a page ends at its anchor). */\n  chain: DeepAgentInterestChainEntry[];\n  more: { before: boolean; after: boolean };\n}\n\nexport function readDeepAgentInterestResolution(value: unknown): DeepAgentInterestResolution | null {\n  if (typeof value !== 'object' || value === null) return null;\n  const { ns, before, at, after, chain, more } = value as Record<string, unknown>;\n  if (ns !== DEEP_AGENT_MAIN_NS || !isIndex(before) || !Array.isArray(chain)) return null;\n  if (at !== undefined && typeof at !== 'string') return null;\n  if (after !== undefined && !isIndex(after)) return null;\n  if (typeof more !== 'object' || more === null) return null;\n  const { before: moreBefore, after: moreAfter } = more as Record<string, unknown>;\n  if (typeof moreBefore !== 'boolean' || typeof moreAfter !== 'boolean') return null;\n  const entries: DeepAgentInterestChainEntry[] = [];\n  for (const entry of chain) {\n    const { id, siblings } = (entry ?? {}) as Record<string, unknown>;\n    if (typeof id !== 'string') return null;\n    entries.push({\n      id,\n      siblings: Array.isArray(siblings) ? siblings.filter((s) => typeof s === 'string') : [id],\n    });\n  }\n  return {\n    ns: DEEP_AGENT_MAIN_NS,\n    before,\n    ...(at !== undefined && { at }),\n    ...(after !== undefined && { after }),\n    chain: entries,\n    more: { before: moreBefore, after: moreAfter },\n  };\n}\n\n/** The attach's interest document: one resolution per stated request, in order. */\nexport interface DeepAgentInterestDocument {\n  windows: DeepAgentInterestResolution[];\n}\n\nexport function readDeepAgentInterestDocument(value: unknown): DeepAgentInterestDocument | null {\n  if (typeof value !== 'object' || value === null) return null;\n  const { windows } = value as Record<string, unknown>;\n  if (!Array.isArray(windows)) return null;\n  const resolutions: DeepAgentInterestResolution[] = [];\n  for (const window of windows) {\n    const resolution = readDeepAgentInterestResolution(window);\n    if (resolution === null) return null;\n    resolutions.push(resolution);\n  }\n  return { windows: resolutions };\n}\n\nconst interestRequest = ({ ns, before, at, after }: DeepAgentInterestRequest): DeepAgentInterestRequest => ({\n  ns,\n  before,\n  ...(at !== undefined && { at }),\n  ...(after !== undefined && { after }),\n});\n\n/** The loaded transcript window, joined from the head interest and its anchored pages. */\nexport interface DeepAgentMessageWindow {\n  /** The requests the joined view stands on, in order (the head first): what a `set` restates. A stale page (its anchor gone) or one the head already covers is left out. */\n  requests: DeepAgentInterestRequest[];\n  /** Covered chain, oldest first. */\n  chain: DeepAgentInterestChainEntry[];\n  /** The chain's message documents, in chain order (an unmounted one is skipped). */\n  messages: LangChainMessage[];\n  /** Whether the host holds messages older than the window. */\n  hasOlder: boolean;\n  /** Pages joined onto the head. */\n  pages: number;\n}\n\n/**\n * The loaded chain plus `grow` older messages, restated in as few requests as\n * it takes: a maximal head, then maximal pages each anchored at the first\n * message of the window above it. `null` once even that exceeds the host's\n * interest cap — the reachable depth is `MAX_INTERESTS × MAX_MESSAGE_WINDOW`.\n */\nfunction coalescedInterestRequests(\n  chain: readonly DeepAgentInterestChainEntry[],\n  grow: number\n): DeepAgentInterestRequest[] | null {\n  const loaded = chain.length;\n  const requests: DeepAgentInterestRequest[] = [\n    { ns: DEEP_AGENT_MAIN_NS, before: Math.min(DEEP_AGENT_MAX_MESSAGE_WINDOW, loaded) },\n  ];\n  // `low` is the chain index of the lowest message the plan covers so far.\n  let low = loaded - (requests[0] as DeepAgentInterestRequest).before;\n  while (low > 0) {\n    const before = Math.min(DEEP_AGENT_MAX_MESSAGE_WINDOW, low + 1);\n    requests.push({\n      ns: DEEP_AGENT_MAIN_NS,\n      before,\n      at: (chain[low] as DeepAgentInterestChainEntry).id,\n    });\n    low = low + 1 - before;\n  }\n  // The plan covers the chain; the new messages extend its lowest request\n  // when that stays within one window, else they are a page of their own.\n  const last = requests[requests.length - 1] as DeepAgentInterestRequest;\n  if (last.before + grow <= DEEP_AGENT_MAX_MESSAGE_WINDOW) {\n    requests[requests.length - 1] = { ...last, before: last.before + grow };\n  } else {\n    requests.push({\n      ns: DEEP_AGENT_MAIN_NS,\n      before: grow + 1,\n      at: (chain[0] as DeepAgentInterestChainEntry).id,\n    });\n  }\n  return requests.length > DEEP_AGENT_MAX_INTERESTS ? null : requests;\n}\n\n/**\n * The requests to state for one more page of older history, or `null` when\n * there is nothing to load (no anchor, the beginning reached) or the host's\n * interest cap is spent. One page is appended, anchored at the window's first\n * message, while the cap allows; at the cap the loaded range is coalesced\n * into maximal requests first (see {@link coalescedInterestRequests}).\n */\nexport function pageInterestRequests(\n  window: DeepAgentMessageWindow,\n  size?: number\n): DeepAgentInterestRequest[] | null {\n  const anchor = window.chain[0]?.id;\n  if (anchor === undefined || !window.hasOlder) return null;\n  const page = pageInterestRequest(anchor, size);\n  const appended = [...window.requests, page];\n  if (appended.length <= DEEP_AGENT_MAX_INTERESTS) return appended;\n  // A page's `before` counts its anchor: it adds `before - 1` older messages.\n  return coalescedInterestRequests(window.chain, page.before - 1);\n}\n\n/**\n * Whether every `stated` request is carried by `carried` (the same anchor, at\n * least as wide): a reconnect's hello that carries what the client last\n * stated — the host remembers a client id's interests — needs no restatement.\n */\nexport function interestsCovered(\n  carried: readonly DeepAgentInterestRequest[],\n  stated: readonly DeepAgentInterestRequest[]\n): boolean {\n  return stated.every((request) =>\n    carried.some((have) => have.at === request.at && have.before >= request.before)\n  );\n}\n\n/**\n * The window off the mounted documents, or `null` while the head interest has\n * not resolved (pre-attach, before the first `set` answer, a wire without\n * documents). A page joins wherever its anchor sits in the chain so far — at\n * the first message it extends the view, inside the head (the head widened\n * past it) it adds nothing and is left out of `requests`; a page whose anchor\n * left the transcript (a rewind emptied it in place) is left out too. A\n * malformed document reads as absent.\n */\nexport function readDeepAgentMessageWindow(\n  mounted: ReadonlyMap<number, StatewireDocuments.Document> | undefined\n): DeepAgentMessageWindow | null {\n  if (mounted === undefined) return null;\n  let windows: DeepAgentInterestResolution[] | undefined;\n  const documents = new Map<string, LangChainMessage>();\n  for (const document of mounted.values()) {\n    const key = document.key;\n    if (document.protocol === DEEP_AGENT_INTEREST_PROTOCOL && document.main === true) {\n      windows = readDeepAgentInterestDocument(document.value)?.windows;\n    } else if (\n      document.protocol === DEEP_AGENT_STATEWIRE_PROTOCOL &&\n      key !== null &&\n      key.length === 3 &&\n      key[1] === DEEP_AGENT_MAIN_NS\n    ) {\n      documents.set(key[2] as string, document.value as LangChainMessage);\n    }\n  }\n  const head = windows?.[0];\n  if (windows === undefined || head === undefined) return null;\n  let chain = head.chain;\n  let hasOlder = head.more.before;\n  const requests = [interestRequest(head)];\n  const loaded = new Set(chain.map((entry) => entry.id));\n  let pages = 0;\n  for (const page of windows.slice(1)) {\n    if (page.at === undefined || !loaded.has(page.at)) continue;\n    const anchorAt = page.chain.findIndex((entry) => entry.id === page.at);\n    if (anchorAt < 0) continue;\n    // Both windows are ranges of one linear transcript sharing the anchor:\n    // what the page holds below it and the chain does not is a prefix.\n    const fresh = page.chain.slice(0, anchorAt).filter((entry) => !loaded.has(entry.id));\n    if (fresh.length === 0) {\n      // Nothing new, but a page that starts where the loaded chain starts\n      // has looked below it: its `more.before` is the answer for the edge.\n      if (page.chain[0]?.id === chain[0]?.id) hasOlder = page.more.before;\n      continue;\n    }\n    chain = [...fresh, ...chain];\n    for (const entry of fresh) loaded.add(entry.id);\n    hasOlder = page.more.before;\n    requests.push(interestRequest(page));\n    pages += 1;\n  }\n  const messages: LangChainMessage[] = [];\n  for (const entry of chain) {\n    const message = documents.get(entry.id);\n    if (message !== undefined) messages.push(message);\n  }\n  return { requests, chain, messages, hasOlder, pages };\n}\n\n/**\n * The windowed wire's read command on the statewire command union, so\n * `sendCommand({ type: 'history/tool_output', … })` typechecks wherever the\n * deep-agent host is driven (`DeepAgentThread.history_tool_output`, an\n * attach-ACL read).\n */\ndeclare module '@assistant-ui/core' {\n  namespace Assistant {\n    interface Commands {\n      /** A tool output the windowed wire carried cut, whole. */\n      historyToolOutput: ToolOutputCommand;\n      /** One stored message whole — the per-client shape, by command. */\n      historyMessage: HistoryMessageCommand;\n    }\n  }\n}\n\n/** The `history/tool_output` command: a truncated tool output, whole. */\nexport interface ToolOutputCommand {\n  readonly type: typeof DEEP_AGENT_HISTORY_TOOL_OUTPUT_COMMAND;\n  readonly messageId: string;\n}\n\nexport function buildToolOutputCommand({ messageId }: { messageId: string }): ToolOutputCommand {\n  return { type: DEEP_AGENT_HISTORY_TOOL_OUTPUT_COMMAND, messageId };\n}\n\nexport interface ToolOutputResult {\n  messageId: string;\n  content: unknown;\n}\n\nexport function readToolOutputResult(payload: unknown): ToolOutputResult | null {\n  if (typeof payload !== 'object' || payload === null) return null;\n  const { messageId, content } = payload as Record<string, unknown>;\n  if (typeof messageId !== 'string' || !messageId) return null;\n  return { messageId, content };\n}\n\n/**\n * The `history/message` command: a message as the host stores it — the\n * model-provenance blobs the wire strips and an untruncated tool output\n * included. Every client receives one projected wire form; this is the one\n * per-client shape, fetched on demand like `history/tool_output`.\n */\nexport interface HistoryMessageCommand {\n  readonly type: typeof DEEP_AGENT_HISTORY_MESSAGE_COMMAND;\n  readonly messageId: string;\n}\n\nexport function buildHistoryMessageCommand({\n  messageId,\n}: {\n  messageId: string;\n}): HistoryMessageCommand {\n  return { type: DEEP_AGENT_HISTORY_MESSAGE_COMMAND, messageId };\n}\n\nexport interface HistoryMessageResult {\n  messageId: string;\n  message: LangChainMessage;\n}\n\nexport function readHistoryMessageResult(payload: unknown): HistoryMessageResult | null {\n  if (typeof payload !== 'object' || payload === null) return null;\n  const { messageId, message } = payload as Record<string, unknown>;\n  if (typeof messageId !== 'string' || !messageId) return null;\n  if (typeof message !== 'object' || message === null) return null;\n  return { messageId, message: message as LangChainMessage };\n}\n\n/** The wire marker on a tool message whose content the host cut. */\nexport interface TruncatedToolContent {\n  /** Length of the full output, in characters. */\n  chars: number;\n}\n\n/**\n * The truncation marker off a raw wire message, or `null` when the message's\n * content is whole. Pure and selector-safe on the marker object.\n */\nexport function readTruncatedToolContent(message: unknown): TruncatedToolContent | null {\n  if (typeof message !== 'object' || message === null) return null;\n  const marker = (message as Record<string, unknown>)[DEEP_AGENT_TRUNCATED_CONTENT_KEY];\n  if (typeof marker !== 'object' || marker === null) return null;\n  const chars = (marker as Record<string, unknown>).chars;\n  return isIndex(chars) ? { chars } : null;\n}\n\n/**\n * `artifact` key the shared converter lifts a truncated tool output's marker\n * into (with the tool message id), so a tool card can ask for the full output\n * — the converter joins the ToolMessage into its tool-call part and would\n * otherwise drop the wire marker.\n */\nexport const ATHENA_TRUNCATED_TOOL_OUTPUT_ARTIFACT_KEY = '_athena_content_truncated';\n\nexport interface TruncatedToolOutputArtifact extends TruncatedToolContent {\n  /** The tool message to name in `history/tool_output`. */\n  messageId: string;\n}\n\n/**\n * The lifted truncation marker off a tool-call part's `artifact`, or `null`.\n */\nexport function readTruncatedToolOutputArtifact(\n  artifact: unknown\n): TruncatedToolOutputArtifact | null {\n  if (typeof artifact !== 'object' || artifact === null) return null;\n  const lifted = (artifact as Record<string, unknown>)[ATHENA_TRUNCATED_TOOL_OUTPUT_ARTIFACT_KEY];\n  if (typeof lifted !== 'object' || lifted === null) return null;\n  const { chars, messageId } = lifted as Record<string, unknown>;\n  if (!isIndex(chars) || typeof messageId !== 'string' || !messageId) return null;\n  return { chars, messageId };\n}\n"],"names":["DEEP_AGENT_DEFAULT_MESSAGE_WINDOW","DEEP_AGENT_MIN_MESSAGE_WINDOW","DEEP_AGENT_MAX_MESSAGE_WINDOW","DEEP_AGENT_MAIN_NS","DEEP_AGENT_MESSAGE_WINDOW_HEADER","DEEP_AGENT_MAX_INTERESTS","DEEP_AGENT_INTEREST_PROTOCOL","DEEP_AGENT_STATEWIRE_PROTOCOL","DEEP_AGENT_HISTORY_TOOL_OUTPUT_COMMAND","DEEP_AGENT_TRUNCATED_CONTENT_KEY"],"mappings":";;;;AAiDA,MAAM,UAAU,CAAC,UACf,OAAO,UAAU,YAAY,OAAO,UAAU,KAAK,KAAK,SAAS;AAO5D,SAAS,uBAAuB,MAAuB;AAC5D,MAAI,SAAS,UAAa,CAAC,OAAO,SAAS,IAAI,EAAG,QAAOA,kBAAAA;AACzD,SAAO,KAAK;AAAA,IACVC,kBAAAA;AAAAA,IACA,KAAK,IAAIC,kBAAAA,+BAA+B,KAAK,MAAM,IAAI,CAAC;AAAA,EAAA;AAE5D;AAaO,SAAS,oBAAoB,MAAyC;AAC3E,SAAO,EAAE,IAAIC,kBAAAA,oBAAoB,QAAQ,uBAAuB,IAAI,EAAA;AACtE;AAEO,SAAS,oBAAoB,IAAY,MAAyC;AACvF,SAAO,EAAE,IAAIA,sCAAoB,QAAQ,uBAAuB,IAAI,GAAG,GAAA;AACzE;AAQO,SAAS,0BACd,SACA,cAAiD,IACzB;AACxB,MAAIC,kBAAAA,oCAAoC,QAAS,QAAO;AACxD,SAAO;AAAA,IACL,GAAG;AAAA,IACH,CAACA,kBAAAA,gCAAgC,GAAG,OAAO,uBAAuB,YAAY,IAAI,CAAC;AAAA,EAAA;AAEvF;AAcO,SAAS,gCAAgC,OAAoD;AAClG,MAAI,OAAO,UAAU,YAAY,UAAU,KAAM,QAAO;AACxD,QAAM,EAAE,IAAI,QAAQ,IAAI,OAAO,OAAO,SAAS;AAC/C,MAAI,OAAOD,kBAAAA,sBAAsB,CAAC,QAAQ,MAAM,KAAK,CAAC,MAAM,QAAQ,KAAK,EAAG,QAAO;AACnF,MAAI,OAAO,UAAa,OAAO,OAAO,SAAU,QAAO;AACvD,MAAI,UAAU,UAAa,CAAC,QAAQ,KAAK,EAAG,QAAO;AACnD,MAAI,OAAO,SAAS,YAAY,SAAS,KAAM,QAAO;AACtD,QAAM,EAAE,QAAQ,YAAY,OAAO,cAAc;AACjD,MAAI,OAAO,eAAe,aAAa,OAAO,cAAc,UAAW,QAAO;AAC9E,QAAM,UAAyC,CAAA;AAC/C,aAAW,SAAS,OAAO;AACzB,UAAM,EAAE,IAAI,SAAA,IAAc,SAAS,CAAA;AACnC,QAAI,OAAO,OAAO,SAAU,QAAO;AACnC,YAAQ,KAAK;AAAA,MACX;AAAA,MACA,UAAU,MAAM,QAAQ,QAAQ,IAAI,SAAS,OAAO,CAAC,MAAM,OAAO,MAAM,QAAQ,IAAI,CAAC,EAAE;AAAA,IAAA,CACxF;AAAA,EACH;AACA,SAAO;AAAA,IACL,IAAIA,kBAAAA;AAAAA,IACJ;AAAA,IACA,GAAI,OAAO,UAAa,EAAE,GAAA;AAAA,IAC1B,GAAI,UAAU,UAAa,EAAE,MAAA;AAAA,IAC7B,OAAO;AAAA,IACP,MAAM,EAAE,QAAQ,YAAY,OAAO,UAAA;AAAA,EAAU;AAEjD;AAOO,SAAS,8BAA8B,OAAkD;AAC9F,MAAI,OAAO,UAAU,YAAY,UAAU,KAAM,QAAO;AACxD,QAAM,EAAE,YAAY;AACpB,MAAI,CAAC,MAAM,QAAQ,OAAO,EAAG,QAAO;AACpC,QAAM,cAA6C,CAAA;AACnD,aAAW,UAAU,SAAS;AAC5B,UAAM,aAAa,gCAAgC,MAAM;AACzD,QAAI,eAAe,KAAM,QAAO;AAChC,gBAAY,KAAK,UAAU;AAAA,EAC7B;AACA,SAAO,EAAE,SAAS,YAAA;AACpB;AAEA,MAAM,kBAAkB,CAAC,EAAE,IAAI,QAAQ,IAAI,aAAiE;AAAA,EAC1G;AAAA,EACA;AAAA,EACA,GAAI,OAAO,UAAa,EAAE,GAAA;AAAA,EAC1B,GAAI,UAAU,UAAa,EAAE,MAAA;AAC/B;AAsBA,SAAS,0BACP,OACA,MACmC;AACnC,QAAM,SAAS,MAAM;AACrB,QAAM,WAAuC;AAAA,IAC3C,EAAE,IAAIA,kBAAAA,oBAAoB,QAAQ,KAAK,IAAID,kBAAAA,+BAA+B,MAAM,EAAA;AAAA,EAAE;AAGpF,MAAI,MAAM,SAAU,SAAS,CAAC,EAA+B;AAC7D,SAAO,MAAM,GAAG;AACd,UAAM,SAAS,KAAK,IAAIA,kBAAAA,+BAA+B,MAAM,CAAC;AAC9D,aAAS,KAAK;AAAA,MACZ,IAAIC,kBAAAA;AAAAA,MACJ;AAAA,MACA,IAAK,MAAM,GAAG,EAAkC;AAAA,IAAA,CACjD;AACD,UAAM,MAAM,IAAI;AAAA,EAClB;AAGA,QAAM,OAAO,SAAS,SAAS,SAAS,CAAC;AACzC,MAAI,KAAK,SAAS,QAAQD,iDAA+B;AACvD,aAAS,SAAS,SAAS,CAAC,IAAI,EAAE,GAAG,MAAM,QAAQ,KAAK,SAAS,KAAA;AAAA,EACnE,OAAO;AACL,aAAS,KAAK;AAAA,MACZ,IAAIC,kBAAAA;AAAAA,MACJ,QAAQ,OAAO;AAAA,MACf,IAAK,MAAM,CAAC,EAAkC;AAAA,IAAA,CAC/C;AAAA,EACH;AACA,SAAO,SAAS,SAASE,kBAAAA,2BAA2B,OAAO;AAC7D;AASO,SAAS,qBACd,QACA,MACmC;;AACnC,QAAM,UAAS,YAAO,MAAM,CAAC,MAAd,mBAAiB;AAChC,MAAI,WAAW,UAAa,CAAC,OAAO,SAAU,QAAO;AACrD,QAAM,OAAO,oBAAoB,QAAQ,IAAI;AAC7C,QAAM,WAAW,CAAC,GAAG,OAAO,UAAU,IAAI;AAC1C,MAAI,SAAS,UAAUA,kBAAAA,yBAA0B,QAAO;AAExD,SAAO,0BAA0B,OAAO,OAAO,KAAK,SAAS,CAAC;AAChE;AAOO,SAAS,iBACd,SACA,QACS;AACT,SAAO,OAAO;AAAA,IAAM,CAAC,YACnB,QAAQ,KAAK,CAAC,SAAS,KAAK,OAAO,QAAQ,MAAM,KAAK,UAAU,QAAQ,MAAM;AAAA,EAAA;AAElF;AAWO,SAAS,2BACd,SAC+B;;AAC/B,MAAI,YAAY,OAAW,QAAO;AAClC,MAAI;AACJ,QAAM,gCAAgB,IAAA;AACtB,aAAW,YAAY,QAAQ,UAAU;AACvC,UAAM,MAAM,SAAS;AACrB,QAAI,SAAS,aAAaC,kBAAAA,gCAAgC,SAAS,SAAS,MAAM;AAChF,iBAAU,mCAA8B,SAAS,KAAK,MAA5C,mBAA+C;AAAA,IAC3D,WACE,SAAS,aAAaC,mBAAAA,iCACtB,QAAQ,QACR,IAAI,WAAW,KACf,IAAI,CAAC,MAAMJ,kBAAAA,oBACX;AACA,gBAAU,IAAI,IAAI,CAAC,GAAa,SAAS,KAAyB;AAAA,IACpE;AAAA,EACF;AACA,QAAM,OAAO,mCAAU;AACvB,MAAI,YAAY,UAAa,SAAS,OAAW,QAAO;AACxD,MAAI,QAAQ,KAAK;AACjB,MAAI,WAAW,KAAK,KAAK;AACzB,QAAM,WAAW,CAAC,gBAAgB,IAAI,CAAC;AACvC,QAAM,SAAS,IAAI,IAAI,MAAM,IAAI,CAAC,UAAU,MAAM,EAAE,CAAC;AACrD,MAAI,QAAQ;AACZ,aAAW,QAAQ,QAAQ,MAAM,CAAC,GAAG;AACnC,QAAI,KAAK,OAAO,UAAa,CAAC,OAAO,IAAI,KAAK,EAAE,EAAG;AACnD,UAAM,WAAW,KAAK,MAAM,UAAU,CAAC,UAAU,MAAM,OAAO,KAAK,EAAE;AACrE,QAAI,WAAW,EAAG;AAGlB,UAAM,QAAQ,KAAK,MAAM,MAAM,GAAG,QAAQ,EAAE,OAAO,CAAC,UAAU,CAAC,OAAO,IAAI,MAAM,EAAE,CAAC;AACnF,QAAI,MAAM,WAAW,GAAG;AAGtB,YAAI,UAAK,MAAM,CAAC,MAAZ,mBAAe,UAAO,WAAM,CAAC,MAAP,mBAAU,IAAI,YAAW,KAAK,KAAK;AAC7D;AAAA,IACF;AACA,YAAQ,CAAC,GAAG,OAAO,GAAG,KAAK;AAC3B,eAAW,SAAS,MAAO,QAAO,IAAI,MAAM,EAAE;AAC9C,eAAW,KAAK,KAAK;AACrB,aAAS,KAAK,gBAAgB,IAAI,CAAC;AACnC,aAAS;AAAA,EACX;AACA,QAAM,WAA+B,CAAA;AACrC,aAAW,SAAS,OAAO;AACzB,UAAM,UAAU,UAAU,IAAI,MAAM,EAAE;AACtC,QAAI,YAAY,OAAW,UAAS,KAAK,OAAO;AAAA,EAClD;AACA,SAAO,EAAE,UAAU,OAAO,UAAU,UAAU,MAAA;AAChD;AAyBO,SAAS,uBAAuB,EAAE,aAAuD;AAC9F,SAAO,EAAE,MAAMK,kBAAAA,wCAAwC,UAAA;AACzD;AAOO,SAAS,qBAAqB,SAA2C;AAC9E,MAAI,OAAO,YAAY,YAAY,YAAY,KAAM,QAAO;AAC5D,QAAM,EAAE,WAAW,QAAA,IAAY;AAC/B,MAAI,OAAO,cAAc,YAAY,CAAC,UAAW,QAAO;AACxD,SAAO,EAAE,WAAW,QAAA;AACtB;AA4CO,SAAS,yBAAyB,SAA+C;AACtF,MAAI,OAAO,YAAY,YAAY,YAAY,KAAM,QAAO;AAC5D,QAAM,SAAU,QAAoCC,kDAAgC;AACpF,MAAI,OAAO,WAAW,YAAY,WAAW,KAAM,QAAO;AAC1D,QAAM,QAAS,OAAmC;AAClD,SAAO,QAAQ,KAAK,IAAI,EAAE,UAAU;AACtC;AAQO,MAAM,4CAA4C;AAUlD,SAAS,gCACd,UACoC;AACpC,MAAI,OAAO,aAAa,YAAY,aAAa,KAAM,QAAO;AAC9D,QAAM,SAAU,SAAqC,yCAAyC;AAC9F,MAAI,OAAO,WAAW,YAAY,WAAW,KAAM,QAAO;AAC1D,QAAM,EAAE,OAAO,UAAA,IAAc;AAC7B,MAAI,CAAC,QAAQ,KAAK,KAAK,OAAO,cAAc,YAAY,CAAC,UAAW,QAAO;AAC3E,SAAO,EAAE,OAAO,UAAA;AAClB;;;;;;;;;;;;;;;"}
+{"version":3,"file":"message-window.cjs","sources":["../../../../agent-runtime/src/message-window.ts"],"sourcesContent":["import type { LangChainMessage } from '@assistant-ui/react-langgraph';\nimport type { StatewireDocuments } from 'statewire';\nimport {\n  DEEP_AGENT_CUT_LANE_DOCUMENT,\n  DEEP_AGENT_DEFAULT_MESSAGE_WINDOW,\n  DEEP_AGENT_HISTORY_MESSAGE_COMMAND,\n  DEEP_AGENT_HISTORY_TOOL_OUTPUT_COMMAND,\n  DEEP_AGENT_INTEREST_PROTOCOL,\n  DEEP_AGENT_MAIN_NS,\n  DEEP_AGENT_MAX_INTERESTS,\n  DEEP_AGENT_MAX_MESSAGE_WINDOW,\n  DEEP_AGENT_MESSAGE_WINDOW_HEADER,\n  DEEP_AGENT_MIN_MESSAGE_WINDOW,\n  DEEP_AGENT_TRUNCATED_CONTENT_KEY,\n} from './generated/runconfig-contract';\nimport { DEEP_AGENT_STATEWIRE_PROTOCOL } from './statewire-transport';\n\n/**\n * The interest window — the client half of agora's\n * `agora/graphs/athena_deep_agent/message_window.py`.\n *\n * The wire main document carries no `messages`. A client states its interests\n * on the `athena/interest` lane as one array (`set(requests)`): the first is\n * the head (the newest `before` messages of the `main` namespace), the rest\n * are pages anchored `at` the first message of the window they extend. The\n * host holds one ephemeral main document per attach, `{ windows }`, one\n * resolution per request in order (the request echoed plus the covered\n * `chain` and `more` flags; an anchor that vanished leaves an empty window in\n * place), and one shared `athena` document per covered message, keyed\n * `[threadId, \"main\", messageId]` — or `[threadId, \"main\", messageId, \"cut\"]`\n * for a finished card a version-2 view carries cut to its newest steps (its\n * whole lane pages in through `history/lane`, see `lane-history.ts`).\n * {@link readDeepAgentMessageWindow} joins them into the transcript the\n * converter renders.\n *\n * Two host rules shape the client half. The head window never drops below\n * the oldest message it has carried on an attach (a run's appends grow the\n * view, nothing loaded slides out), so a page stays anchored at the head's\n * first message while the transcript grows; and the host remembers a client\n * id's interests across its attaches, so a reconnect's hello carries the\n * view the client held. A fresh client's hello carries the head it declared\n * on the attach headers ({@link withDeclaredMessageWindow}) — a statement\n * cannot ride the first attach, the hello negotiates the lanes.\n *\n * Pages are appended one per load until the host's interest cap, then the\n * loaded range is restated in as few maximal requests as it takes\n * ({@link pageInterestRequests}); the whole transcript stays reachable up to\n * `MAX_INTERESTS × MAX_MESSAGE_WINDOW` messages. A truncated tool output\n * comes back whole through `history/tool_output` ({@link buildToolOutputCommand});\n * a message's stored form — provenance blobs and all — through\n * `history/message` ({@link buildHistoryMessageCommand}).\n */\n\nconst isIndex = (value: unknown): value is number =>\n  typeof value === 'number' && Number.isInteger(value) && value >= 0;\n\n/** What a surface declares for its window: the interest size (the host clamps to its bounds). */\nexport interface DeepAgentMessageWindowDeclaration {\n  size?: number;\n}\n\nexport function clampMessageWindowSize(size?: number): number {\n  if (size === undefined || !Number.isFinite(size)) return DEEP_AGENT_DEFAULT_MESSAGE_WINDOW;\n  return Math.max(\n    DEEP_AGENT_MIN_MESSAGE_WINDOW,\n    Math.min(DEEP_AGENT_MAX_MESSAGE_WINDOW, Math.trunc(size))\n  );\n}\n\n/** An `athena/interest` request over the linear transcript. */\nexport interface DeepAgentInterestRequest {\n  readonly ns: typeof DEEP_AGENT_MAIN_NS;\n  /** Messages ending at the anchor (the tail without `at`). */\n  readonly before: number;\n  /** Anchor message id; a page's is the first message of the window it extends. */\n  readonly at?: string;\n  /** Messages past the anchor (requires `at`). */\n  readonly after?: number;\n  /** The newest steps of every subagent lane a covered card names (see `lane-follow.ts`). */\n  readonly follow?: { readonly before: number };\n}\n\nexport function headInterestRequest(size?: number): DeepAgentInterestRequest {\n  return { ns: DEEP_AGENT_MAIN_NS, before: clampMessageWindowSize(size) };\n}\n\nexport function pageInterestRequest(at: string, size?: number): DeepAgentInterestRequest {\n  return { ns: DEEP_AGENT_MAIN_NS, before: clampMessageWindowSize(size), at };\n}\n\n/**\n * Attach headers with the head window declared (`x-athena-deep-agent-window`):\n * the hello of a client the host does not remember carries the newest `size`\n * messages. Part of the one attach recipe, so a pooled dial asks for the shape\n * the thread's own attach would. An explicit entry (tests, overrides) wins.\n */\nexport function withDeclaredMessageWindow(\n  headers: Record<string, string>,\n  declaration: DeepAgentMessageWindowDeclaration = {}\n): Record<string, string> {\n  if (DEEP_AGENT_MESSAGE_WINDOW_HEADER in headers) return headers;\n  return {\n    ...headers,\n    [DEEP_AGENT_MESSAGE_WINDOW_HEADER]: String(clampMessageWindowSize(declaration.size)),\n  };\n}\n\nexport interface DeepAgentInterestChainEntry {\n  id: string;\n  siblings: string[];\n}\n\n/** The host's resolution document of one interest: the request echoed plus its coverage. */\nexport interface DeepAgentInterestResolution extends DeepAgentInterestRequest {\n  /** Covered message ids, oldest first (a page ends at its anchor). */\n  chain: DeepAgentInterestChainEntry[];\n  more: { before: boolean; after: boolean };\n}\n\nexport function readDeepAgentInterestResolution(value: unknown): DeepAgentInterestResolution | null {\n  if (typeof value !== 'object' || value === null) return null;\n  const { ns, before, at, after, chain, more } = value as Record<string, unknown>;\n  if (ns !== DEEP_AGENT_MAIN_NS || !isIndex(before) || !Array.isArray(chain)) return null;\n  if (at !== undefined && typeof at !== 'string') return null;\n  if (after !== undefined && !isIndex(after)) return null;\n  if (typeof more !== 'object' || more === null) return null;\n  const { before: moreBefore, after: moreAfter } = more as Record<string, unknown>;\n  if (typeof moreBefore !== 'boolean' || typeof moreAfter !== 'boolean') return null;\n  const entries: DeepAgentInterestChainEntry[] = [];\n  for (const entry of chain) {\n    const { id, siblings } = (entry ?? {}) as Record<string, unknown>;\n    if (typeof id !== 'string') return null;\n    entries.push({\n      id,\n      siblings: Array.isArray(siblings) ? siblings.filter((s) => typeof s === 'string') : [id],\n    });\n  }\n  return {\n    ns: DEEP_AGENT_MAIN_NS,\n    before,\n    ...(at !== undefined && { at }),\n    ...(after !== undefined && { after }),\n    chain: entries,\n    more: { before: moreBefore, after: moreAfter },\n  };\n}\n\n/** The attach's interest document: one resolution per stated request, in order. */\nexport interface DeepAgentInterestDocument {\n  windows: DeepAgentInterestResolution[];\n}\n\nexport function readDeepAgentInterestDocument(value: unknown): DeepAgentInterestDocument | null {\n  if (typeof value !== 'object' || value === null) return null;\n  const { windows } = value as Record<string, unknown>;\n  if (!Array.isArray(windows)) return null;\n  const resolutions: DeepAgentInterestResolution[] = [];\n  for (const window of windows) {\n    const resolution = readDeepAgentInterestResolution(window);\n    if (resolution === null) return null;\n    resolutions.push(resolution);\n  }\n  return { windows: resolutions };\n}\n\nconst interestRequest = ({ ns, before, at, after }: DeepAgentInterestRequest): DeepAgentInterestRequest => ({\n  ns,\n  before,\n  ...(at !== undefined && { at }),\n  ...(after !== undefined && { after }),\n});\n\n/** The loaded transcript window, joined from the head interest and its anchored pages. */\nexport interface DeepAgentMessageWindow {\n  /** The requests the joined view stands on, in order (the head first): what a `set` restates. A stale page (its anchor gone) or one the head already covers is left out. */\n  requests: DeepAgentInterestRequest[];\n  /** Covered chain, oldest first. */\n  chain: DeepAgentInterestChainEntry[];\n  /** The chain's message documents, in chain order (an unmounted one is skipped). */\n  messages: LangChainMessage[];\n  /** Whether the host holds messages older than the window. */\n  hasOlder: boolean;\n  /** Pages joined onto the head. */\n  pages: number;\n}\n\n/**\n * The loaded chain plus `grow` older messages, restated in as few requests as\n * it takes: a maximal head, then maximal pages each anchored at the first\n * message of the window above it. `null` once even that exceeds the host's\n * interest cap — the reachable depth is `MAX_INTERESTS × MAX_MESSAGE_WINDOW`.\n */\nfunction coalescedInterestRequests(\n  chain: readonly DeepAgentInterestChainEntry[],\n  grow: number\n): DeepAgentInterestRequest[] | null {\n  const loaded = chain.length;\n  const requests: DeepAgentInterestRequest[] = [\n    { ns: DEEP_AGENT_MAIN_NS, before: Math.min(DEEP_AGENT_MAX_MESSAGE_WINDOW, loaded) },\n  ];\n  // `low` is the chain index of the lowest message the plan covers so far.\n  let low = loaded - (requests[0] as DeepAgentInterestRequest).before;\n  while (low > 0) {\n    const before = Math.min(DEEP_AGENT_MAX_MESSAGE_WINDOW, low + 1);\n    requests.push({\n      ns: DEEP_AGENT_MAIN_NS,\n      before,\n      at: (chain[low] as DeepAgentInterestChainEntry).id,\n    });\n    low = low + 1 - before;\n  }\n  // The plan covers the chain; the new messages extend its lowest request\n  // when that stays within one window, else they are a page of their own.\n  const last = requests[requests.length - 1] as DeepAgentInterestRequest;\n  if (last.before + grow <= DEEP_AGENT_MAX_MESSAGE_WINDOW) {\n    requests[requests.length - 1] = { ...last, before: last.before + grow };\n  } else {\n    requests.push({\n      ns: DEEP_AGENT_MAIN_NS,\n      before: grow + 1,\n      at: (chain[0] as DeepAgentInterestChainEntry).id,\n    });\n  }\n  return requests.length > DEEP_AGENT_MAX_INTERESTS ? null : requests;\n}\n\n/**\n * The requests to state for one more page of older history, or `null` when\n * there is nothing to load (no anchor, the beginning reached) or the host's\n * interest cap is spent. One page is appended, anchored at the window's first\n * message, while the cap allows; at the cap the loaded range is coalesced\n * into maximal requests first (see {@link coalescedInterestRequests}).\n */\nexport function pageInterestRequests(\n  window: DeepAgentMessageWindow,\n  size?: number\n): DeepAgentInterestRequest[] | null {\n  const anchor = window.chain[0]?.id;\n  if (anchor === undefined || !window.hasOlder) return null;\n  const page = pageInterestRequest(anchor, size);\n  const appended = [...window.requests, page];\n  if (appended.length <= DEEP_AGENT_MAX_INTERESTS) return appended;\n  // A page's `before` counts its anchor: it adds `before - 1` older messages.\n  return coalescedInterestRequests(window.chain, page.before - 1);\n}\n\n/**\n * Whether every `stated` request is carried by `carried` (the same anchor, at\n * least as wide): a reconnect's hello that carries what the client last\n * stated — the host remembers a client id's interests — needs no restatement.\n */\nexport function interestsCovered(\n  carried: readonly DeepAgentInterestRequest[],\n  stated: readonly DeepAgentInterestRequest[]\n): boolean {\n  return stated.every((request) =>\n    carried.some((have) => have.at === request.at && have.before >= request.before)\n  );\n}\n\n/**\n * The window off the mounted documents, or `null` while the head interest has\n * not resolved (pre-attach, before the first `set` answer, a wire without\n * documents). A page joins wherever its anchor sits in the chain so far — at\n * the first message it extends the view, inside the head (the head widened\n * past it) it adds nothing and is left out of `requests`; a page whose anchor\n * left the transcript (a rewind emptied it in place) is left out too. A\n * malformed document reads as absent.\n */\nexport function readDeepAgentMessageWindow(\n  mounted: ReadonlyMap<number, StatewireDocuments.Document> | undefined\n): DeepAgentMessageWindow | null {\n  if (mounted === undefined) return null;\n  let windows: DeepAgentInterestResolution[] | undefined;\n  const documents = new Map<string, LangChainMessage>();\n  for (const document of mounted.values()) {\n    const key = document.key;\n    if (document.protocol === DEEP_AGENT_INTEREST_PROTOCOL && document.main === true) {\n      windows = readDeepAgentInterestDocument(document.value)?.windows;\n    } else if (\n      document.protocol === DEEP_AGENT_STATEWIRE_PROTOCOL &&\n      key !== null &&\n      key[1] === DEEP_AGENT_MAIN_NS &&\n      (key.length === 3 || (key.length === 4 && key[3] === DEEP_AGENT_CUT_LANE_DOCUMENT))\n    ) {\n      documents.set(key[2] as string, document.value as LangChainMessage);\n    }\n  }\n  const head = windows?.[0];\n  if (windows === undefined || head === undefined) return null;\n  let chain = head.chain;\n  let hasOlder = head.more.before;\n  const requests = [interestRequest(head)];\n  const loaded = new Set(chain.map((entry) => entry.id));\n  let pages = 0;\n  for (const page of windows.slice(1)) {\n    if (page.at === undefined || !loaded.has(page.at)) continue;\n    const anchorAt = page.chain.findIndex((entry) => entry.id === page.at);\n    if (anchorAt < 0) continue;\n    // Both windows are ranges of one linear transcript sharing the anchor:\n    // what the page holds below it and the chain does not is a prefix.\n    const fresh = page.chain.slice(0, anchorAt).filter((entry) => !loaded.has(entry.id));\n    if (fresh.length === 0) {\n      // Nothing new, but a page that starts where the loaded chain starts\n      // has looked below it: its `more.before` is the answer for the edge.\n      if (page.chain[0]?.id === chain[0]?.id) hasOlder = page.more.before;\n      continue;\n    }\n    chain = [...fresh, ...chain];\n    for (const entry of fresh) loaded.add(entry.id);\n    hasOlder = page.more.before;\n    requests.push(interestRequest(page));\n    pages += 1;\n  }\n  const messages: LangChainMessage[] = [];\n  for (const entry of chain) {\n    const message = documents.get(entry.id);\n    if (message !== undefined) messages.push(message);\n  }\n  return { requests, chain, messages, hasOlder, pages };\n}\n\n/**\n * The windowed wire's read command on the statewire command union, so\n * `sendCommand({ type: 'history/tool_output', … })` typechecks wherever the\n * deep-agent host is driven (`DeepAgentThread.history_tool_output`, an\n * attach-ACL read).\n */\ndeclare module '@assistant-ui/core' {\n  namespace Assistant {\n    interface Commands {\n      /** A tool output the windowed wire carried cut, whole. */\n      historyToolOutput: ToolOutputCommand;\n      /** One stored message whole — the per-client shape, by command. */\n      historyMessage: HistoryMessageCommand;\n    }\n  }\n}\n\n/** The `history/tool_output` command: a truncated tool output, whole. */\nexport interface ToolOutputCommand {\n  readonly type: typeof DEEP_AGENT_HISTORY_TOOL_OUTPUT_COMMAND;\n  readonly messageId: string;\n}\n\nexport function buildToolOutputCommand({ messageId }: { messageId: string }): ToolOutputCommand {\n  return { type: DEEP_AGENT_HISTORY_TOOL_OUTPUT_COMMAND, messageId };\n}\n\nexport interface ToolOutputResult {\n  messageId: string;\n  content: unknown;\n}\n\nexport function readToolOutputResult(payload: unknown): ToolOutputResult | null {\n  if (typeof payload !== 'object' || payload === null) return null;\n  const { messageId, content } = payload as Record<string, unknown>;\n  if (typeof messageId !== 'string' || !messageId) return null;\n  return { messageId, content };\n}\n\n/**\n * The `history/message` command: a message as the host stores it — the\n * model-provenance blobs the wire strips and an untruncated tool output\n * included. Every client receives one projected wire form; this is the one\n * per-client shape, fetched on demand like `history/tool_output`.\n */\nexport interface HistoryMessageCommand {\n  readonly type: typeof DEEP_AGENT_HISTORY_MESSAGE_COMMAND;\n  readonly messageId: string;\n}\n\nexport function buildHistoryMessageCommand({\n  messageId,\n}: {\n  messageId: string;\n}): HistoryMessageCommand {\n  return { type: DEEP_AGENT_HISTORY_MESSAGE_COMMAND, messageId };\n}\n\nexport interface HistoryMessageResult {\n  messageId: string;\n  message: LangChainMessage;\n}\n\nexport function readHistoryMessageResult(payload: unknown): HistoryMessageResult | null {\n  if (typeof payload !== 'object' || payload === null) return null;\n  const { messageId, message } = payload as Record<string, unknown>;\n  if (typeof messageId !== 'string' || !messageId) return null;\n  if (typeof message !== 'object' || message === null) return null;\n  return { messageId, message: message as LangChainMessage };\n}\n\n/** The wire marker on a tool message whose content the host cut. */\nexport interface TruncatedToolContent {\n  /** Length of the full output, in characters. */\n  chars: number;\n}\n\n/**\n * The truncation marker off a raw wire message, or `null` when the message's\n * content is whole. Pure and selector-safe on the marker object.\n */\nexport function readTruncatedToolContent(message: unknown): TruncatedToolContent | null {\n  if (typeof message !== 'object' || message === null) return null;\n  const marker = (message as Record<string, unknown>)[DEEP_AGENT_TRUNCATED_CONTENT_KEY];\n  if (typeof marker !== 'object' || marker === null) return null;\n  const chars = (marker as Record<string, unknown>).chars;\n  return isIndex(chars) ? { chars } : null;\n}\n\n/**\n * `artifact` key the shared converter lifts a truncated tool output's marker\n * into (with the tool message id), so a tool card can ask for the full output\n * — the converter joins the ToolMessage into its tool-call part and would\n * otherwise drop the wire marker.\n */\nexport const ATHENA_TRUNCATED_TOOL_OUTPUT_ARTIFACT_KEY = '_athena_content_truncated';\n\nexport interface TruncatedToolOutputArtifact extends TruncatedToolContent {\n  /** The tool message to name in `history/tool_output`. */\n  messageId: string;\n}\n\n/**\n * The lifted truncation marker off a tool-call part's `artifact`, or `null`.\n */\nexport function readTruncatedToolOutputArtifact(\n  artifact: unknown\n): TruncatedToolOutputArtifact | null {\n  if (typeof artifact !== 'object' || artifact === null) return null;\n  const lifted = (artifact as Record<string, unknown>)[ATHENA_TRUNCATED_TOOL_OUTPUT_ARTIFACT_KEY];\n  if (typeof lifted !== 'object' || lifted === null) return null;\n  const { chars, messageId } = lifted as Record<string, unknown>;\n  if (!isIndex(chars) || typeof messageId !== 'string' || !messageId) return null;\n  return { chars, messageId };\n}\n"],"names":["DEEP_AGENT_DEFAULT_MESSAGE_WINDOW","DEEP_AGENT_MIN_MESSAGE_WINDOW","DEEP_AGENT_MAX_MESSAGE_WINDOW","DEEP_AGENT_MAIN_NS","DEEP_AGENT_MESSAGE_WINDOW_HEADER","DEEP_AGENT_MAX_INTERESTS","DEEP_AGENT_INTEREST_PROTOCOL","DEEP_AGENT_STATEWIRE_PROTOCOL","DEEP_AGENT_CUT_LANE_DOCUMENT","DEEP_AGENT_HISTORY_TOOL_OUTPUT_COMMAND","DEEP_AGENT_HISTORY_MESSAGE_COMMAND","DEEP_AGENT_TRUNCATED_CONTENT_KEY"],"mappings":";;;;AAqDA,MAAM,UAAU,CAAC,UACf,OAAO,UAAU,YAAY,OAAO,UAAU,KAAK,KAAK,SAAS;AAO5D,SAAS,uBAAuB,MAAuB;AAC5D,MAAI,SAAS,UAAa,CAAC,OAAO,SAAS,IAAI,EAAG,QAAOA,kBAAAA;AACzD,SAAO,KAAK;AAAA,IACVC,kBAAAA;AAAAA,IACA,KAAK,IAAIC,kBAAAA,+BAA+B,KAAK,MAAM,IAAI,CAAC;AAAA,EAAA;AAE5D;AAeO,SAAS,oBAAoB,MAAyC;AAC3E,SAAO,EAAE,IAAIC,kBAAAA,oBAAoB,QAAQ,uBAAuB,IAAI,EAAA;AACtE;AAEO,SAAS,oBAAoB,IAAY,MAAyC;AACvF,SAAO,EAAE,IAAIA,sCAAoB,QAAQ,uBAAuB,IAAI,GAAG,GAAA;AACzE;AAQO,SAAS,0BACd,SACA,cAAiD,IACzB;AACxB,MAAIC,kBAAAA,oCAAoC,QAAS,QAAO;AACxD,SAAO;AAAA,IACL,GAAG;AAAA,IACH,CAACA,kBAAAA,gCAAgC,GAAG,OAAO,uBAAuB,YAAY,IAAI,CAAC;AAAA,EAAA;AAEvF;AAcO,SAAS,gCAAgC,OAAoD;AAClG,MAAI,OAAO,UAAU,YAAY,UAAU,KAAM,QAAO;AACxD,QAAM,EAAE,IAAI,QAAQ,IAAI,OAAO,OAAO,SAAS;AAC/C,MAAI,OAAOD,kBAAAA,sBAAsB,CAAC,QAAQ,MAAM,KAAK,CAAC,MAAM,QAAQ,KAAK,EAAG,QAAO;AACnF,MAAI,OAAO,UAAa,OAAO,OAAO,SAAU,QAAO;AACvD,MAAI,UAAU,UAAa,CAAC,QAAQ,KAAK,EAAG,QAAO;AACnD,MAAI,OAAO,SAAS,YAAY,SAAS,KAAM,QAAO;AACtD,QAAM,EAAE,QAAQ,YAAY,OAAO,cAAc;AACjD,MAAI,OAAO,eAAe,aAAa,OAAO,cAAc,UAAW,QAAO;AAC9E,QAAM,UAAyC,CAAA;AAC/C,aAAW,SAAS,OAAO;AACzB,UAAM,EAAE,IAAI,SAAA,IAAc,SAAS,CAAA;AACnC,QAAI,OAAO,OAAO,SAAU,QAAO;AACnC,YAAQ,KAAK;AAAA,MACX;AAAA,MACA,UAAU,MAAM,QAAQ,QAAQ,IAAI,SAAS,OAAO,CAAC,MAAM,OAAO,MAAM,QAAQ,IAAI,CAAC,EAAE;AAAA,IAAA,CACxF;AAAA,EACH;AACA,SAAO;AAAA,IACL,IAAIA,kBAAAA;AAAAA,IACJ;AAAA,IACA,GAAI,OAAO,UAAa,EAAE,GAAA;AAAA,IAC1B,GAAI,UAAU,UAAa,EAAE,MAAA;AAAA,IAC7B,OAAO;AAAA,IACP,MAAM,EAAE,QAAQ,YAAY,OAAO,UAAA;AAAA,EAAU;AAEjD;AAOO,SAAS,8BAA8B,OAAkD;AAC9F,MAAI,OAAO,UAAU,YAAY,UAAU,KAAM,QAAO;AACxD,QAAM,EAAE,YAAY;AACpB,MAAI,CAAC,MAAM,QAAQ,OAAO,EAAG,QAAO;AACpC,QAAM,cAA6C,CAAA;AACnD,aAAW,UAAU,SAAS;AAC5B,UAAM,aAAa,gCAAgC,MAAM;AACzD,QAAI,eAAe,KAAM,QAAO;AAChC,gBAAY,KAAK,UAAU;AAAA,EAC7B;AACA,SAAO,EAAE,SAAS,YAAA;AACpB;AAEA,MAAM,kBAAkB,CAAC,EAAE,IAAI,QAAQ,IAAI,aAAiE;AAAA,EAC1G;AAAA,EACA;AAAA,EACA,GAAI,OAAO,UAAa,EAAE,GAAA;AAAA,EAC1B,GAAI,UAAU,UAAa,EAAE,MAAA;AAC/B;AAsBA,SAAS,0BACP,OACA,MACmC;AACnC,QAAM,SAAS,MAAM;AACrB,QAAM,WAAuC;AAAA,IAC3C,EAAE,IAAIA,kBAAAA,oBAAoB,QAAQ,KAAK,IAAID,kBAAAA,+BAA+B,MAAM,EAAA;AAAA,EAAE;AAGpF,MAAI,MAAM,SAAU,SAAS,CAAC,EAA+B;AAC7D,SAAO,MAAM,GAAG;AACd,UAAM,SAAS,KAAK,IAAIA,kBAAAA,+BAA+B,MAAM,CAAC;AAC9D,aAAS,KAAK;AAAA,MACZ,IAAIC,kBAAAA;AAAAA,MACJ;AAAA,MACA,IAAK,MAAM,GAAG,EAAkC;AAAA,IAAA,CACjD;AACD,UAAM,MAAM,IAAI;AAAA,EAClB;AAGA,QAAM,OAAO,SAAS,SAAS,SAAS,CAAC;AACzC,MAAI,KAAK,SAAS,QAAQD,iDAA+B;AACvD,aAAS,SAAS,SAAS,CAAC,IAAI,EAAE,GAAG,MAAM,QAAQ,KAAK,SAAS,KAAA;AAAA,EACnE,OAAO;AACL,aAAS,KAAK;AAAA,MACZ,IAAIC,kBAAAA;AAAAA,MACJ,QAAQ,OAAO;AAAA,MACf,IAAK,MAAM,CAAC,EAAkC;AAAA,IAAA,CAC/C;AAAA,EACH;AACA,SAAO,SAAS,SAASE,kBAAAA,2BAA2B,OAAO;AAC7D;AASO,SAAS,qBACd,QACA,MACmC;;AACnC,QAAM,UAAS,YAAO,MAAM,CAAC,MAAd,mBAAiB;AAChC,MAAI,WAAW,UAAa,CAAC,OAAO,SAAU,QAAO;AACrD,QAAM,OAAO,oBAAoB,QAAQ,IAAI;AAC7C,QAAM,WAAW,CAAC,GAAG,OAAO,UAAU,IAAI;AAC1C,MAAI,SAAS,UAAUA,kBAAAA,yBAA0B,QAAO;AAExD,SAAO,0BAA0B,OAAO,OAAO,KAAK,SAAS,CAAC;AAChE;AAOO,SAAS,iBACd,SACA,QACS;AACT,SAAO,OAAO;AAAA,IAAM,CAAC,YACnB,QAAQ,KAAK,CAAC,SAAS,KAAK,OAAO,QAAQ,MAAM,KAAK,UAAU,QAAQ,MAAM;AAAA,EAAA;AAElF;AAWO,SAAS,2BACd,SAC+B;;AAC/B,MAAI,YAAY,OAAW,QAAO;AAClC,MAAI;AACJ,QAAM,gCAAgB,IAAA;AACtB,aAAW,YAAY,QAAQ,UAAU;AACvC,UAAM,MAAM,SAAS;AACrB,QAAI,SAAS,aAAaC,kBAAAA,gCAAgC,SAAS,SAAS,MAAM;AAChF,iBAAU,mCAA8B,SAAS,KAAK,MAA5C,mBAA+C;AAAA,IAC3D,WACE,SAAS,aAAaC,mBAAAA,iCACtB,QAAQ,QACR,IAAI,CAAC,MAAMJ,kBAAAA,uBACV,IAAI,WAAW,KAAM,IAAI,WAAW,KAAK,IAAI,CAAC,MAAMK,kBAAAA,+BACrD;AACA,gBAAU,IAAI,IAAI,CAAC,GAAa,SAAS,KAAyB;AAAA,IACpE;AAAA,EACF;AACA,QAAM,OAAO,mCAAU;AACvB,MAAI,YAAY,UAAa,SAAS,OAAW,QAAO;AACxD,MAAI,QAAQ,KAAK;AACjB,MAAI,WAAW,KAAK,KAAK;AACzB,QAAM,WAAW,CAAC,gBAAgB,IAAI,CAAC;AACvC,QAAM,SAAS,IAAI,IAAI,MAAM,IAAI,CAAC,UAAU,MAAM,EAAE,CAAC;AACrD,MAAI,QAAQ;AACZ,aAAW,QAAQ,QAAQ,MAAM,CAAC,GAAG;AACnC,QAAI,KAAK,OAAO,UAAa,CAAC,OAAO,IAAI,KAAK,EAAE,EAAG;AACnD,UAAM,WAAW,KAAK,MAAM,UAAU,CAAC,UAAU,MAAM,OAAO,KAAK,EAAE;AACrE,QAAI,WAAW,EAAG;AAGlB,UAAM,QAAQ,KAAK,MAAM,MAAM,GAAG,QAAQ,EAAE,OAAO,CAAC,UAAU,CAAC,OAAO,IAAI,MAAM,EAAE,CAAC;AACnF,QAAI,MAAM,WAAW,GAAG;AAGtB,YAAI,UAAK,MAAM,CAAC,MAAZ,mBAAe,UAAO,WAAM,CAAC,MAAP,mBAAU,IAAI,YAAW,KAAK,KAAK;AAC7D;AAAA,IACF;AACA,YAAQ,CAAC,GAAG,OAAO,GAAG,KAAK;AAC3B,eAAW,SAAS,MAAO,QAAO,IAAI,MAAM,EAAE;AAC9C,eAAW,KAAK,KAAK;AACrB,aAAS,KAAK,gBAAgB,IAAI,CAAC;AACnC,aAAS;AAAA,EACX;AACA,QAAM,WAA+B,CAAA;AACrC,aAAW,SAAS,OAAO;AACzB,UAAM,UAAU,UAAU,IAAI,MAAM,EAAE;AACtC,QAAI,YAAY,OAAW,UAAS,KAAK,OAAO;AAAA,EAClD;AACA,SAAO,EAAE,UAAU,OAAO,UAAU,UAAU,MAAA;AAChD;AAyBO,SAAS,uBAAuB,EAAE,aAAuD;AAC9F,SAAO,EAAE,MAAMC,kBAAAA,wCAAwC,UAAA;AACzD;AAOO,SAAS,qBAAqB,SAA2C;AAC9E,MAAI,OAAO,YAAY,YAAY,YAAY,KAAM,QAAO;AAC5D,QAAM,EAAE,WAAW,QAAA,IAAY;AAC/B,MAAI,OAAO,cAAc,YAAY,CAAC,UAAW,QAAO;AACxD,SAAO,EAAE,WAAW,QAAA;AACtB;AAaO,SAAS,2BAA2B;AAAA,EACzC;AACF,GAE0B;AACxB,SAAO,EAAE,MAAMC,kBAAAA,oCAAoC,UAAA;AACrD;AAOO,SAAS,yBAAyB,SAA+C;AACtF,MAAI,OAAO,YAAY,YAAY,YAAY,KAAM,QAAO;AAC5D,QAAM,EAAE,WAAW,QAAA,IAAY;AAC/B,MAAI,OAAO,cAAc,YAAY,CAAC,UAAW,QAAO;AACxD,MAAI,OAAO,YAAY,YAAY,YAAY,KAAM,QAAO;AAC5D,SAAO,EAAE,WAAW,QAAA;AACtB;AAYO,SAAS,yBAAyB,SAA+C;AACtF,MAAI,OAAO,YAAY,YAAY,YAAY,KAAM,QAAO;AAC5D,QAAM,SAAU,QAAoCC,kDAAgC;AACpF,MAAI,OAAO,WAAW,YAAY,WAAW,KAAM,QAAO;AAC1D,QAAM,QAAS,OAAmC;AAClD,SAAO,QAAQ,KAAK,IAAI,EAAE,UAAU;AACtC;AAQO,MAAM,4CAA4C;AAUlD,SAAS,gCACd,UACoC;AACpC,MAAI,OAAO,aAAa,YAAY,aAAa,KAAM,QAAO;AAC9D,QAAM,SAAU,SAAqC,yCAAyC;AAC9F,MAAI,OAAO,WAAW,YAAY,WAAW,KAAM,QAAO;AAC1D,QAAM,EAAE,OAAO,UAAA,IAAc;AAC7B,MAAI,CAAC,QAAQ,KAAK,KAAK,OAAO,cAAc,YAAY,CAAC,UAAW,QAAO;AAC3E,SAAO,EAAE,OAAO,UAAA;AAClB;;;;;;;;;;;;;;;;;"}

package/dist/agent-runtime/src/message-window.d.ts

@@ -0,0 +1,159 @@
+import type { LangChainMessage } from '@assistant-ui/react-langgraph';
+import type { StatewireDocuments } from 'statewire';
+import { DEEP_AGENT_HISTORY_MESSAGE_COMMAND, DEEP_AGENT_HISTORY_TOOL_OUTPUT_COMMAND, DEEP_AGENT_MAIN_NS } from './generated/runconfig-contract';
+/** What a surface declares for its window: the interest size (the host clamps to its bounds). */
+export interface DeepAgentMessageWindowDeclaration {
+    size?: number;
+}
+export declare function clampMessageWindowSize(size?: number): number;
+/** An `athena/interest` request over the linear transcript. */
+export interface DeepAgentInterestRequest {
+    readonly ns: typeof DEEP_AGENT_MAIN_NS;
+    /** Messages ending at the anchor (the tail without `at`). */
+    readonly before: number;
+    /** Anchor message id; a page's is the first message of the window it extends. */
+    readonly at?: string;
+    /** Messages past the anchor (requires `at`). */
+    readonly after?: number;
+    /** The newest steps of every subagent lane a covered card names (see `lane-follow.ts`). */
+    readonly follow?: {
+        readonly before: number;
+    };
+}
+export declare function headInterestRequest(size?: number): DeepAgentInterestRequest;
+export declare function pageInterestRequest(at: string, size?: number): DeepAgentInterestRequest;
+/**
+ * Attach headers with the head window declared (`x-athena-deep-agent-window`):
+ * the hello of a client the host does not remember carries the newest `size`
+ * messages. Part of the one attach recipe, so a pooled dial asks for the shape
+ * the thread's own attach would. An explicit entry (tests, overrides) wins.
+ */
+export declare function withDeclaredMessageWindow(headers: Record<string, string>, declaration?: DeepAgentMessageWindowDeclaration): Record<string, string>;
+export interface DeepAgentInterestChainEntry {
+    id: string;
+    siblings: string[];
+}
+/** The host's resolution document of one interest: the request echoed plus its coverage. */
+export interface DeepAgentInterestResolution extends DeepAgentInterestRequest {
+    /** Covered message ids, oldest first (a page ends at its anchor). */
+    chain: DeepAgentInterestChainEntry[];
+    more: {
+        before: boolean;
+        after: boolean;
+    };
+}
+export declare function readDeepAgentInterestResolution(value: unknown): DeepAgentInterestResolution | null;
+/** The attach's interest document: one resolution per stated request, in order. */
+export interface DeepAgentInterestDocument {
+    windows: DeepAgentInterestResolution[];
+}
+export declare function readDeepAgentInterestDocument(value: unknown): DeepAgentInterestDocument | null;
+/** The loaded transcript window, joined from the head interest and its anchored pages. */
+export interface DeepAgentMessageWindow {
+    /** The requests the joined view stands on, in order (the head first): what a `set` restates. A stale page (its anchor gone) or one the head already covers is left out. */
+    requests: DeepAgentInterestRequest[];
+    /** Covered chain, oldest first. */
+    chain: DeepAgentInterestChainEntry[];
+    /** The chain's message documents, in chain order (an unmounted one is skipped). */
+    messages: LangChainMessage[];
+    /** Whether the host holds messages older than the window. */
+    hasOlder: boolean;
+    /** Pages joined onto the head. */
+    pages: number;
+}
+/**
+ * The requests to state for one more page of older history, or `null` when
+ * there is nothing to load (no anchor, the beginning reached) or the host's
+ * interest cap is spent. One page is appended, anchored at the window's first
+ * message, while the cap allows; at the cap the loaded range is coalesced
+ * into maximal requests first (see {@link coalescedInterestRequests}).
+ */
+export declare function pageInterestRequests(window: DeepAgentMessageWindow, size?: number): DeepAgentInterestRequest[] | null;
+/**
+ * Whether every `stated` request is carried by `carried` (the same anchor, at
+ * least as wide): a reconnect's hello that carries what the client last
+ * stated — the host remembers a client id's interests — needs no restatement.
+ */
+export declare function interestsCovered(carried: readonly DeepAgentInterestRequest[], stated: readonly DeepAgentInterestRequest[]): boolean;
+/**
+ * The window off the mounted documents, or `null` while the head interest has
+ * not resolved (pre-attach, before the first `set` answer, a wire without
+ * documents). A page joins wherever its anchor sits in the chain so far — at
+ * the first message it extends the view, inside the head (the head widened
+ * past it) it adds nothing and is left out of `requests`; a page whose anchor
+ * left the transcript (a rewind emptied it in place) is left out too. A
+ * malformed document reads as absent.
+ */
+export declare function readDeepAgentMessageWindow(mounted: ReadonlyMap<number, StatewireDocuments.Document> | undefined): DeepAgentMessageWindow | null;
+/**
+ * The windowed wire's read command on the statewire command union, so
+ * `sendCommand({ type: 'history/tool_output', … })` typechecks wherever the
+ * deep-agent host is driven (`DeepAgentThread.history_tool_output`, an
+ * attach-ACL read).
+ */
+declare module '@assistant-ui/core' {
+    namespace Assistant {
+        interface Commands {
+            /** A tool output the windowed wire carried cut, whole. */
+            historyToolOutput: ToolOutputCommand;
+            /** One stored message whole — the per-client shape, by command. */
+            historyMessage: HistoryMessageCommand;
+        }
+    }
+}
+/** The `history/tool_output` command: a truncated tool output, whole. */
+export interface ToolOutputCommand {
+    readonly type: typeof DEEP_AGENT_HISTORY_TOOL_OUTPUT_COMMAND;
+    readonly messageId: string;
+}
+export declare function buildToolOutputCommand({ messageId }: {
+    messageId: string;
+}): ToolOutputCommand;
+export interface ToolOutputResult {
+    messageId: string;
+    content: unknown;
+}
+export declare function readToolOutputResult(payload: unknown): ToolOutputResult | null;
+/**
+ * The `history/message` command: a message as the host stores it — the
+ * model-provenance blobs the wire strips and an untruncated tool output
+ * included. Every client receives one projected wire form; this is the one
+ * per-client shape, fetched on demand like `history/tool_output`.
+ */
+export interface HistoryMessageCommand {
+    readonly type: typeof DEEP_AGENT_HISTORY_MESSAGE_COMMAND;
+    readonly messageId: string;
+}
+export declare function buildHistoryMessageCommand({ messageId, }: {
+    messageId: string;
+}): HistoryMessageCommand;
+export interface HistoryMessageResult {
+    messageId: string;
+    message: LangChainMessage;
+}
+export declare function readHistoryMessageResult(payload: unknown): HistoryMessageResult | null;
+/** The wire marker on a tool message whose content the host cut. */
+export interface TruncatedToolContent {
+    /** Length of the full output, in characters. */
+    chars: number;
+}
+/**
+ * The truncation marker off a raw wire message, or `null` when the message's
+ * content is whole. Pure and selector-safe on the marker object.
+ */
+export declare function readTruncatedToolContent(message: unknown): TruncatedToolContent | null;
+/**
+ * `artifact` key the shared converter lifts a truncated tool output's marker
+ * into (with the tool message id), so a tool card can ask for the full output
+ * — the converter joins the ToolMessage into its tool-call part and would
+ * otherwise drop the wire marker.
+ */
+export declare const ATHENA_TRUNCATED_TOOL_OUTPUT_ARTIFACT_KEY = "_athena_content_truncated";
+export interface TruncatedToolOutputArtifact extends TruncatedToolContent {
+    /** The tool message to name in `history/tool_output`. */
+    messageId: string;
+}
+/**
+ * The lifted truncation marker off a tool-call part's `artifact`, or `null`.
+ */
+export declare function readTruncatedToolOutputArtifact(artifact: unknown): TruncatedToolOutputArtifact | null;

package/dist/agent-runtime/src/message-window.js

@@ -1,4 +1,4 @@
-import { DEEP_AGENT_MAX_INTERESTS, DEEP_AGENT_HISTORY_TOOL_OUTPUT_COMMAND, DEEP_AGENT_INTEREST_PROTOCOL, DEEP_AGENT_MAIN_NS, DEEP_AGENT_MAX_MESSAGE_WINDOW, DEEP_AGENT_DEFAULT_MESSAGE_WINDOW, DEEP_AGENT_MIN_MESSAGE_WINDOW, DEEP_AGENT_MESSAGE_WINDOW_HEADER, DEEP_AGENT_TRUNCATED_CONTENT_KEY } from "./generated/runconfig-contract.js";
+import { DEEP_AGENT_MAX_INTERESTS, DEEP_AGENT_HISTORY_TOOL_OUTPUT_COMMAND, DEEP_AGENT_INTEREST_PROTOCOL, DEEP_AGENT_MAIN_NS, DEEP_AGENT_CUT_LANE_DOCUMENT, DEEP_AGENT_MAX_MESSAGE_WINDOW, DEEP_AGENT_DEFAULT_MESSAGE_WINDOW, DEEP_AGENT_MIN_MESSAGE_WINDOW, DEEP_AGENT_MESSAGE_WINDOW_HEADER, DEEP_AGENT_HISTORY_MESSAGE_COMMAND, DEEP_AGENT_TRUNCATED_CONTENT_KEY } from "./generated/runconfig-contract.js";
 import { DEEP_AGENT_STATEWIRE_PROTOCOL } from "./statewire-transport.js";
 const isIndex = (value) => typeof value === "number" && Number.isInteger(value) && value >= 0;
 function clampMessageWindowSize(size) {
@@ -116,7 +116,7 @@ function readDeepAgentMessageWindow(mounted) {
     const key = document.key;
     if (document.protocol === DEEP_AGENT_INTEREST_PROTOCOL && document.main === true) {
       windows = (_a = readDeepAgentInterestDocument(document.value)) == null ? void 0 : _a.windows;
-    } else if (document.protocol === DEEP_AGENT_STATEWIRE_PROTOCOL && key !== null && key.length === 3 && key[1] === DEEP_AGENT_MAIN_NS) {
+    } else if (document.protocol === DEEP_AGENT_STATEWIRE_PROTOCOL && key !== null && key[1] === DEEP_AGENT_MAIN_NS && (key.length === 3 || key.length === 4 && key[3] === DEEP_AGENT_CUT_LANE_DOCUMENT)) {
       documents.set(key[2], document.value);
     }
   }
@@ -158,6 +158,18 @@ function readToolOutputResult(payload) {
   if (typeof messageId !== "string" || !messageId) return null;
   return { messageId, content };
 }
+function buildHistoryMessageCommand({
+  messageId
+}) {
+  return { type: DEEP_AGENT_HISTORY_MESSAGE_COMMAND, messageId };
+}
+function readHistoryMessageResult(payload) {
+  if (typeof payload !== "object" || payload === null) return null;
+  const { messageId, message } = payload;
+  if (typeof messageId !== "string" || !messageId) return null;
+  if (typeof message !== "object" || message === null) return null;
+  return { messageId, message };
+}
 function readTruncatedToolContent(message) {
   if (typeof message !== "object" || message === null) return null;
   const marker = message[DEEP_AGENT_TRUNCATED_CONTENT_KEY];
@@ -176,6 +188,7 @@ function readTruncatedToolOutputArtifact(artifact) {
 }
 export {
   ATHENA_TRUNCATED_TOOL_OUTPUT_ARTIFACT_KEY,
+  buildHistoryMessageCommand,
   buildToolOutputCommand,
   clampMessageWindowSize,
   headInterestRequest,
@@ -185,6 +198,7 @@ export {
   readDeepAgentInterestDocument,
   readDeepAgentInterestResolution,
   readDeepAgentMessageWindow,
+  readHistoryMessageResult,
   readToolOutputResult,
   readTruncatedToolContent,
   readTruncatedToolOutputArtifact,

package/dist/agent-runtime/src/message-window.js.map

@@ -1 +1 @@
-{"version":3,"file":"message-window.js","sources":["../../../../agent-runtime/src/message-window.ts"],"sourcesContent":["import type { LangChainMessage } from '@assistant-ui/react-langgraph';\nimport type { StatewireDocuments } from 'statewire';\nimport {\n  DEEP_AGENT_DEFAULT_MESSAGE_WINDOW,\n  DEEP_AGENT_HISTORY_MESSAGE_COMMAND,\n  DEEP_AGENT_HISTORY_TOOL_OUTPUT_COMMAND,\n  DEEP_AGENT_INTEREST_PROTOCOL,\n  DEEP_AGENT_MAIN_NS,\n  DEEP_AGENT_MAX_INTERESTS,\n  DEEP_AGENT_MAX_MESSAGE_WINDOW,\n  DEEP_AGENT_MESSAGE_WINDOW_HEADER,\n  DEEP_AGENT_MIN_MESSAGE_WINDOW,\n  DEEP_AGENT_TRUNCATED_CONTENT_KEY,\n} from './generated/runconfig-contract';\nimport { DEEP_AGENT_STATEWIRE_PROTOCOL } from './statewire-transport';\n\n/**\n * The interest window — the client half of agora's\n * `agora/graphs/athena_deep_agent/message_window.py`.\n *\n * The wire main document carries no `messages`. A client states its interests\n * on the `athena/interest` lane as one array (`set(requests)`): the first is\n * the head (the newest `before` messages of the `main` namespace), the rest\n * are pages anchored `at` the first message of the window they extend. The\n * host holds one ephemeral main document per attach, `{ windows }`, one\n * resolution per request in order (the request echoed plus the covered\n * `chain` and `more` flags; an anchor that vanished leaves an empty window in\n * place), and one shared `athena` document per covered message, keyed\n * `[threadId, \"main\", messageId]`. {@link readDeepAgentMessageWindow} joins\n * them into the transcript the converter renders.\n *\n * Two host rules shape the client half. The head window never drops below\n * the oldest message it has carried on an attach (a run's appends grow the\n * view, nothing loaded slides out), so a page stays anchored at the head's\n * first message while the transcript grows; and the host remembers a client\n * id's interests across its attaches, so a reconnect's hello carries the\n * view the client held. A fresh client's hello carries the head it declared\n * on the attach headers ({@link withDeclaredMessageWindow}) — a statement\n * cannot ride the first attach, the hello negotiates the lanes.\n *\n * Pages are appended one per load until the host's interest cap, then the\n * loaded range is restated in as few maximal requests as it takes\n * ({@link pageInterestRequests}); the whole transcript stays reachable up to\n * `MAX_INTERESTS × MAX_MESSAGE_WINDOW` messages. A truncated tool output\n * comes back whole through `history/tool_output` ({@link buildToolOutputCommand});\n * a message's stored form — provenance blobs and all — through\n * `history/message` ({@link buildHistoryMessageCommand}).\n */\n\nconst isIndex = (value: unknown): value is number =>\n  typeof value === 'number' && Number.isInteger(value) && value >= 0;\n\n/** What a surface declares for its window: the interest size (the host clamps to its bounds). */\nexport interface DeepAgentMessageWindowDeclaration {\n  size?: number;\n}\n\nexport function clampMessageWindowSize(size?: number): number {\n  if (size === undefined || !Number.isFinite(size)) return DEEP_AGENT_DEFAULT_MESSAGE_WINDOW;\n  return Math.max(\n    DEEP_AGENT_MIN_MESSAGE_WINDOW,\n    Math.min(DEEP_AGENT_MAX_MESSAGE_WINDOW, Math.trunc(size))\n  );\n}\n\n/** An `athena/interest` request over the linear transcript. */\nexport interface DeepAgentInterestRequest {\n  readonly ns: typeof DEEP_AGENT_MAIN_NS;\n  /** Messages ending at the anchor (the tail without `at`). */\n  readonly before: number;\n  /** Anchor message id; a page's is the first message of the window it extends. */\n  readonly at?: string;\n  /** Messages past the anchor (requires `at`). */\n  readonly after?: number;\n}\n\nexport function headInterestRequest(size?: number): DeepAgentInterestRequest {\n  return { ns: DEEP_AGENT_MAIN_NS, before: clampMessageWindowSize(size) };\n}\n\nexport function pageInterestRequest(at: string, size?: number): DeepAgentInterestRequest {\n  return { ns: DEEP_AGENT_MAIN_NS, before: clampMessageWindowSize(size), at };\n}\n\n/**\n * Attach headers with the head window declared (`x-athena-deep-agent-window`):\n * the hello of a client the host does not remember carries the newest `size`\n * messages. Part of the one attach recipe, so a pooled dial asks for the shape\n * the thread's own attach would. An explicit entry (tests, overrides) wins.\n */\nexport function withDeclaredMessageWindow(\n  headers: Record<string, string>,\n  declaration: DeepAgentMessageWindowDeclaration = {}\n): Record<string, string> {\n  if (DEEP_AGENT_MESSAGE_WINDOW_HEADER in headers) return headers;\n  return {\n    ...headers,\n    [DEEP_AGENT_MESSAGE_WINDOW_HEADER]: String(clampMessageWindowSize(declaration.size)),\n  };\n}\n\nexport interface DeepAgentInterestChainEntry {\n  id: string;\n  siblings: string[];\n}\n\n/** The host's resolution document of one interest: the request echoed plus its coverage. */\nexport interface DeepAgentInterestResolution extends DeepAgentInterestRequest {\n  /** Covered message ids, oldest first (a page ends at its anchor). */\n  chain: DeepAgentInterestChainEntry[];\n  more: { before: boolean; after: boolean };\n}\n\nexport function readDeepAgentInterestResolution(value: unknown): DeepAgentInterestResolution | null {\n  if (typeof value !== 'object' || value === null) return null;\n  const { ns, before, at, after, chain, more } = value as Record<string, unknown>;\n  if (ns !== DEEP_AGENT_MAIN_NS || !isIndex(before) || !Array.isArray(chain)) return null;\n  if (at !== undefined && typeof at !== 'string') return null;\n  if (after !== undefined && !isIndex(after)) return null;\n  if (typeof more !== 'object' || more === null) return null;\n  const { before: moreBefore, after: moreAfter } = more as Record<string, unknown>;\n  if (typeof moreBefore !== 'boolean' || typeof moreAfter !== 'boolean') return null;\n  const entries: DeepAgentInterestChainEntry[] = [];\n  for (const entry of chain) {\n    const { id, siblings } = (entry ?? {}) as Record<string, unknown>;\n    if (typeof id !== 'string') return null;\n    entries.push({\n      id,\n      siblings: Array.isArray(siblings) ? siblings.filter((s) => typeof s === 'string') : [id],\n    });\n  }\n  return {\n    ns: DEEP_AGENT_MAIN_NS,\n    before,\n    ...(at !== undefined && { at }),\n    ...(after !== undefined && { after }),\n    chain: entries,\n    more: { before: moreBefore, after: moreAfter },\n  };\n}\n\n/** The attach's interest document: one resolution per stated request, in order. */\nexport interface DeepAgentInterestDocument {\n  windows: DeepAgentInterestResolution[];\n}\n\nexport function readDeepAgentInterestDocument(value: unknown): DeepAgentInterestDocument | null {\n  if (typeof value !== 'object' || value === null) return null;\n  const { windows } = value as Record<string, unknown>;\n  if (!Array.isArray(windows)) return null;\n  const resolutions: DeepAgentInterestResolution[] = [];\n  for (const window of windows) {\n    const resolution = readDeepAgentInterestResolution(window);\n    if (resolution === null) return null;\n    resolutions.push(resolution);\n  }\n  return { windows: resolutions };\n}\n\nconst interestRequest = ({ ns, before, at, after }: DeepAgentInterestRequest): DeepAgentInterestRequest => ({\n  ns,\n  before,\n  ...(at !== undefined && { at }),\n  ...(after !== undefined && { after }),\n});\n\n/** The loaded transcript window, joined from the head interest and its anchored pages. */\nexport interface DeepAgentMessageWindow {\n  /** The requests the joined view stands on, in order (the head first): what a `set` restates. A stale page (its anchor gone) or one the head already covers is left out. */\n  requests: DeepAgentInterestRequest[];\n  /** Covered chain, oldest first. */\n  chain: DeepAgentInterestChainEntry[];\n  /** The chain's message documents, in chain order (an unmounted one is skipped). */\n  messages: LangChainMessage[];\n  /** Whether the host holds messages older than the window. */\n  hasOlder: boolean;\n  /** Pages joined onto the head. */\n  pages: number;\n}\n\n/**\n * The loaded chain plus `grow` older messages, restated in as few requests as\n * it takes: a maximal head, then maximal pages each anchored at the first\n * message of the window above it. `null` once even that exceeds the host's\n * interest cap — the reachable depth is `MAX_INTERESTS × MAX_MESSAGE_WINDOW`.\n */\nfunction coalescedInterestRequests(\n  chain: readonly DeepAgentInterestChainEntry[],\n  grow: number\n): DeepAgentInterestRequest[] | null {\n  const loaded = chain.length;\n  const requests: DeepAgentInterestRequest[] = [\n    { ns: DEEP_AGENT_MAIN_NS, before: Math.min(DEEP_AGENT_MAX_MESSAGE_WINDOW, loaded) },\n  ];\n  // `low` is the chain index of the lowest message the plan covers so far.\n  let low = loaded - (requests[0] as DeepAgentInterestRequest).before;\n  while (low > 0) {\n    const before = Math.min(DEEP_AGENT_MAX_MESSAGE_WINDOW, low + 1);\n    requests.push({\n      ns: DEEP_AGENT_MAIN_NS,\n      before,\n      at: (chain[low] as DeepAgentInterestChainEntry).id,\n    });\n    low = low + 1 - before;\n  }\n  // The plan covers the chain; the new messages extend its lowest request\n  // when that stays within one window, else they are a page of their own.\n  const last = requests[requests.length - 1] as DeepAgentInterestRequest;\n  if (last.before + grow <= DEEP_AGENT_MAX_MESSAGE_WINDOW) {\n    requests[requests.length - 1] = { ...last, before: last.before + grow };\n  } else {\n    requests.push({\n      ns: DEEP_AGENT_MAIN_NS,\n      before: grow + 1,\n      at: (chain[0] as DeepAgentInterestChainEntry).id,\n    });\n  }\n  return requests.length > DEEP_AGENT_MAX_INTERESTS ? null : requests;\n}\n\n/**\n * The requests to state for one more page of older history, or `null` when\n * there is nothing to load (no anchor, the beginning reached) or the host's\n * interest cap is spent. One page is appended, anchored at the window's first\n * message, while the cap allows; at the cap the loaded range is coalesced\n * into maximal requests first (see {@link coalescedInterestRequests}).\n */\nexport function pageInterestRequests(\n  window: DeepAgentMessageWindow,\n  size?: number\n): DeepAgentInterestRequest[] | null {\n  const anchor = window.chain[0]?.id;\n  if (anchor === undefined || !window.hasOlder) return null;\n  const page = pageInterestRequest(anchor, size);\n  const appended = [...window.requests, page];\n  if (appended.length <= DEEP_AGENT_MAX_INTERESTS) return appended;\n  // A page's `before` counts its anchor: it adds `before - 1` older messages.\n  return coalescedInterestRequests(window.chain, page.before - 1);\n}\n\n/**\n * Whether every `stated` request is carried by `carried` (the same anchor, at\n * least as wide): a reconnect's hello that carries what the client last\n * stated — the host remembers a client id's interests — needs no restatement.\n */\nexport function interestsCovered(\n  carried: readonly DeepAgentInterestRequest[],\n  stated: readonly DeepAgentInterestRequest[]\n): boolean {\n  return stated.every((request) =>\n    carried.some((have) => have.at === request.at && have.before >= request.before)\n  );\n}\n\n/**\n * The window off the mounted documents, or `null` while the head interest has\n * not resolved (pre-attach, before the first `set` answer, a wire without\n * documents). A page joins wherever its anchor sits in the chain so far — at\n * the first message it extends the view, inside the head (the head widened\n * past it) it adds nothing and is left out of `requests`; a page whose anchor\n * left the transcript (a rewind emptied it in place) is left out too. A\n * malformed document reads as absent.\n */\nexport function readDeepAgentMessageWindow(\n  mounted: ReadonlyMap<number, StatewireDocuments.Document> | undefined\n): DeepAgentMessageWindow | null {\n  if (mounted === undefined) return null;\n  let windows: DeepAgentInterestResolution[] | undefined;\n  const documents = new Map<string, LangChainMessage>();\n  for (const document of mounted.values()) {\n    const key = document.key;\n    if (document.protocol === DEEP_AGENT_INTEREST_PROTOCOL && document.main === true) {\n      windows = readDeepAgentInterestDocument(document.value)?.windows;\n    } else if (\n      document.protocol === DEEP_AGENT_STATEWIRE_PROTOCOL &&\n      key !== null &&\n      key.length === 3 &&\n      key[1] === DEEP_AGENT_MAIN_NS\n    ) {\n      documents.set(key[2] as string, document.value as LangChainMessage);\n    }\n  }\n  const head = windows?.[0];\n  if (windows === undefined || head === undefined) return null;\n  let chain = head.chain;\n  let hasOlder = head.more.before;\n  const requests = [interestRequest(head)];\n  const loaded = new Set(chain.map((entry) => entry.id));\n  let pages = 0;\n  for (const page of windows.slice(1)) {\n    if (page.at === undefined || !loaded.has(page.at)) continue;\n    const anchorAt = page.chain.findIndex((entry) => entry.id === page.at);\n    if (anchorAt < 0) continue;\n    // Both windows are ranges of one linear transcript sharing the anchor:\n    // what the page holds below it and the chain does not is a prefix.\n    const fresh = page.chain.slice(0, anchorAt).filter((entry) => !loaded.has(entry.id));\n    if (fresh.length === 0) {\n      // Nothing new, but a page that starts where the loaded chain starts\n      // has looked below it: its `more.before` is the answer for the edge.\n      if (page.chain[0]?.id === chain[0]?.id) hasOlder = page.more.before;\n      continue;\n    }\n    chain = [...fresh, ...chain];\n    for (const entry of fresh) loaded.add(entry.id);\n    hasOlder = page.more.before;\n    requests.push(interestRequest(page));\n    pages += 1;\n  }\n  const messages: LangChainMessage[] = [];\n  for (const entry of chain) {\n    const message = documents.get(entry.id);\n    if (message !== undefined) messages.push(message);\n  }\n  return { requests, chain, messages, hasOlder, pages };\n}\n\n/**\n * The windowed wire's read command on the statewire command union, so\n * `sendCommand({ type: 'history/tool_output', … })` typechecks wherever the\n * deep-agent host is driven (`DeepAgentThread.history_tool_output`, an\n * attach-ACL read).\n */\ndeclare module '@assistant-ui/core' {\n  namespace Assistant {\n    interface Commands {\n      /** A tool output the windowed wire carried cut, whole. */\n      historyToolOutput: ToolOutputCommand;\n      /** One stored message whole — the per-client shape, by command. */\n      historyMessage: HistoryMessageCommand;\n    }\n  }\n}\n\n/** The `history/tool_output` command: a truncated tool output, whole. */\nexport interface ToolOutputCommand {\n  readonly type: typeof DEEP_AGENT_HISTORY_TOOL_OUTPUT_COMMAND;\n  readonly messageId: string;\n}\n\nexport function buildToolOutputCommand({ messageId }: { messageId: string }): ToolOutputCommand {\n  return { type: DEEP_AGENT_HISTORY_TOOL_OUTPUT_COMMAND, messageId };\n}\n\nexport interface ToolOutputResult {\n  messageId: string;\n  content: unknown;\n}\n\nexport function readToolOutputResult(payload: unknown): ToolOutputResult | null {\n  if (typeof payload !== 'object' || payload === null) return null;\n  const { messageId, content } = payload as Record<string, unknown>;\n  if (typeof messageId !== 'string' || !messageId) return null;\n  return { messageId, content };\n}\n\n/**\n * The `history/message` command: a message as the host stores it — the\n * model-provenance blobs the wire strips and an untruncated tool output\n * included. Every client receives one projected wire form; this is the one\n * per-client shape, fetched on demand like `history/tool_output`.\n */\nexport interface HistoryMessageCommand {\n  readonly type: typeof DEEP_AGENT_HISTORY_MESSAGE_COMMAND;\n  readonly messageId: string;\n}\n\nexport function buildHistoryMessageCommand({\n  messageId,\n}: {\n  messageId: string;\n}): HistoryMessageCommand {\n  return { type: DEEP_AGENT_HISTORY_MESSAGE_COMMAND, messageId };\n}\n\nexport interface HistoryMessageResult {\n  messageId: string;\n  message: LangChainMessage;\n}\n\nexport function readHistoryMessageResult(payload: unknown): HistoryMessageResult | null {\n  if (typeof payload !== 'object' || payload === null) return null;\n  const { messageId, message } = payload as Record<string, unknown>;\n  if (typeof messageId !== 'string' || !messageId) return null;\n  if (typeof message !== 'object' || message === null) return null;\n  return { messageId, message: message as LangChainMessage };\n}\n\n/** The wire marker on a tool message whose content the host cut. */\nexport interface TruncatedToolContent {\n  /** Length of the full output, in characters. */\n  chars: number;\n}\n\n/**\n * The truncation marker off a raw wire message, or `null` when the message's\n * content is whole. Pure and selector-safe on the marker object.\n */\nexport function readTruncatedToolContent(message: unknown): TruncatedToolContent | null {\n  if (typeof message !== 'object' || message === null) return null;\n  const marker = (message as Record<string, unknown>)[DEEP_AGENT_TRUNCATED_CONTENT_KEY];\n  if (typeof marker !== 'object' || marker === null) return null;\n  const chars = (marker as Record<string, unknown>).chars;\n  return isIndex(chars) ? { chars } : null;\n}\n\n/**\n * `artifact` key the shared converter lifts a truncated tool output's marker\n * into (with the tool message id), so a tool card can ask for the full output\n * — the converter joins the ToolMessage into its tool-call part and would\n * otherwise drop the wire marker.\n */\nexport const ATHENA_TRUNCATED_TOOL_OUTPUT_ARTIFACT_KEY = '_athena_content_truncated';\n\nexport interface TruncatedToolOutputArtifact extends TruncatedToolContent {\n  /** The tool message to name in `history/tool_output`. */\n  messageId: string;\n}\n\n/**\n * The lifted truncation marker off a tool-call part's `artifact`, or `null`.\n */\nexport function readTruncatedToolOutputArtifact(\n  artifact: unknown\n): TruncatedToolOutputArtifact | null {\n  if (typeof artifact !== 'object' || artifact === null) return null;\n  const lifted = (artifact as Record<string, unknown>)[ATHENA_TRUNCATED_TOOL_OUTPUT_ARTIFACT_KEY];\n  if (typeof lifted !== 'object' || lifted === null) return null;\n  const { chars, messageId } = lifted as Record<string, unknown>;\n  if (!isIndex(chars) || typeof messageId !== 'string' || !messageId) return null;\n  return { chars, messageId };\n}\n"],"names":[],"mappings":";;AAiDA,MAAM,UAAU,CAAC,UACf,OAAO,UAAU,YAAY,OAAO,UAAU,KAAK,KAAK,SAAS;AAO5D,SAAS,uBAAuB,MAAuB;AAC5D,MAAI,SAAS,UAAa,CAAC,OAAO,SAAS,IAAI,EAAG,QAAO;AACzD,SAAO,KAAK;AAAA,IACV;AAAA,IACA,KAAK,IAAI,+BAA+B,KAAK,MAAM,IAAI,CAAC;AAAA,EAAA;AAE5D;AAaO,SAAS,oBAAoB,MAAyC;AAC3E,SAAO,EAAE,IAAI,oBAAoB,QAAQ,uBAAuB,IAAI,EAAA;AACtE;AAEO,SAAS,oBAAoB,IAAY,MAAyC;AACvF,SAAO,EAAE,IAAI,oBAAoB,QAAQ,uBAAuB,IAAI,GAAG,GAAA;AACzE;AAQO,SAAS,0BACd,SACA,cAAiD,IACzB;AACxB,MAAI,oCAAoC,QAAS,QAAO;AACxD,SAAO;AAAA,IACL,GAAG;AAAA,IACH,CAAC,gCAAgC,GAAG,OAAO,uBAAuB,YAAY,IAAI,CAAC;AAAA,EAAA;AAEvF;AAcO,SAAS,gCAAgC,OAAoD;AAClG,MAAI,OAAO,UAAU,YAAY,UAAU,KAAM,QAAO;AACxD,QAAM,EAAE,IAAI,QAAQ,IAAI,OAAO,OAAO,SAAS;AAC/C,MAAI,OAAO,sBAAsB,CAAC,QAAQ,MAAM,KAAK,CAAC,MAAM,QAAQ,KAAK,EAAG,QAAO;AACnF,MAAI,OAAO,UAAa,OAAO,OAAO,SAAU,QAAO;AACvD,MAAI,UAAU,UAAa,CAAC,QAAQ,KAAK,EAAG,QAAO;AACnD,MAAI,OAAO,SAAS,YAAY,SAAS,KAAM,QAAO;AACtD,QAAM,EAAE,QAAQ,YAAY,OAAO,cAAc;AACjD,MAAI,OAAO,eAAe,aAAa,OAAO,cAAc,UAAW,QAAO;AAC9E,QAAM,UAAyC,CAAA;AAC/C,aAAW,SAAS,OAAO;AACzB,UAAM,EAAE,IAAI,SAAA,IAAc,SAAS,CAAA;AACnC,QAAI,OAAO,OAAO,SAAU,QAAO;AACnC,YAAQ,KAAK;AAAA,MACX;AAAA,MACA,UAAU,MAAM,QAAQ,QAAQ,IAAI,SAAS,OAAO,CAAC,MAAM,OAAO,MAAM,QAAQ,IAAI,CAAC,EAAE;AAAA,IAAA,CACxF;AAAA,EACH;AACA,SAAO;AAAA,IACL,IAAI;AAAA,IACJ;AAAA,IACA,GAAI,OAAO,UAAa,EAAE,GAAA;AAAA,IAC1B,GAAI,UAAU,UAAa,EAAE,MAAA;AAAA,IAC7B,OAAO;AAAA,IACP,MAAM,EAAE,QAAQ,YAAY,OAAO,UAAA;AAAA,EAAU;AAEjD;AAOO,SAAS,8BAA8B,OAAkD;AAC9F,MAAI,OAAO,UAAU,YAAY,UAAU,KAAM,QAAO;AACxD,QAAM,EAAE,YAAY;AACpB,MAAI,CAAC,MAAM,QAAQ,OAAO,EAAG,QAAO;AACpC,QAAM,cAA6C,CAAA;AACnD,aAAW,UAAU,SAAS;AAC5B,UAAM,aAAa,gCAAgC,MAAM;AACzD,QAAI,eAAe,KAAM,QAAO;AAChC,gBAAY,KAAK,UAAU;AAAA,EAC7B;AACA,SAAO,EAAE,SAAS,YAAA;AACpB;AAEA,MAAM,kBAAkB,CAAC,EAAE,IAAI,QAAQ,IAAI,aAAiE;AAAA,EAC1G;AAAA,EACA;AAAA,EACA,GAAI,OAAO,UAAa,EAAE,GAAA;AAAA,EAC1B,GAAI,UAAU,UAAa,EAAE,MAAA;AAC/B;AAsBA,SAAS,0BACP,OACA,MACmC;AACnC,QAAM,SAAS,MAAM;AACrB,QAAM,WAAuC;AAAA,IAC3C,EAAE,IAAI,oBAAoB,QAAQ,KAAK,IAAI,+BAA+B,MAAM,EAAA;AAAA,EAAE;AAGpF,MAAI,MAAM,SAAU,SAAS,CAAC,EAA+B;AAC7D,SAAO,MAAM,GAAG;AACd,UAAM,SAAS,KAAK,IAAI,+BAA+B,MAAM,CAAC;AAC9D,aAAS,KAAK;AAAA,MACZ,IAAI;AAAA,MACJ;AAAA,MACA,IAAK,MAAM,GAAG,EAAkC;AAAA,IAAA,CACjD;AACD,UAAM,MAAM,IAAI;AAAA,EAClB;AAGA,QAAM,OAAO,SAAS,SAAS,SAAS,CAAC;AACzC,MAAI,KAAK,SAAS,QAAQ,+BAA+B;AACvD,aAAS,SAAS,SAAS,CAAC,IAAI,EAAE,GAAG,MAAM,QAAQ,KAAK,SAAS,KAAA;AAAA,EACnE,OAAO;AACL,aAAS,KAAK;AAAA,MACZ,IAAI;AAAA,MACJ,QAAQ,OAAO;AAAA,MACf,IAAK,MAAM,CAAC,EAAkC;AAAA,IAAA,CAC/C;AAAA,EACH;AACA,SAAO,SAAS,SAAS,2BAA2B,OAAO;AAC7D;AASO,SAAS,qBACd,QACA,MACmC;;AACnC,QAAM,UAAS,YAAO,MAAM,CAAC,MAAd,mBAAiB;AAChC,MAAI,WAAW,UAAa,CAAC,OAAO,SAAU,QAAO;AACrD,QAAM,OAAO,oBAAoB,QAAQ,IAAI;AAC7C,QAAM,WAAW,CAAC,GAAG,OAAO,UAAU,IAAI;AAC1C,MAAI,SAAS,UAAU,yBAA0B,QAAO;AAExD,SAAO,0BAA0B,OAAO,OAAO,KAAK,SAAS,CAAC;AAChE;AAOO,SAAS,iBACd,SACA,QACS;AACT,SAAO,OAAO;AAAA,IAAM,CAAC,YACnB,QAAQ,KAAK,CAAC,SAAS,KAAK,OAAO,QAAQ,MAAM,KAAK,UAAU,QAAQ,MAAM;AAAA,EAAA;AAElF;AAWO,SAAS,2BACd,SAC+B;;AAC/B,MAAI,YAAY,OAAW,QAAO;AAClC,MAAI;AACJ,QAAM,gCAAgB,IAAA;AACtB,aAAW,YAAY,QAAQ,UAAU;AACvC,UAAM,MAAM,SAAS;AACrB,QAAI,SAAS,aAAa,gCAAgC,SAAS,SAAS,MAAM;AAChF,iBAAU,mCAA8B,SAAS,KAAK,MAA5C,mBAA+C;AAAA,IAC3D,WACE,SAAS,aAAa,iCACtB,QAAQ,QACR,IAAI,WAAW,KACf,IAAI,CAAC,MAAM,oBACX;AACA,gBAAU,IAAI,IAAI,CAAC,GAAa,SAAS,KAAyB;AAAA,IACpE;AAAA,EACF;AACA,QAAM,OAAO,mCAAU;AACvB,MAAI,YAAY,UAAa,SAAS,OAAW,QAAO;AACxD,MAAI,QAAQ,KAAK;AACjB,MAAI,WAAW,KAAK,KAAK;AACzB,QAAM,WAAW,CAAC,gBAAgB,IAAI,CAAC;AACvC,QAAM,SAAS,IAAI,IAAI,MAAM,IAAI,CAAC,UAAU,MAAM,EAAE,CAAC;AACrD,MAAI,QAAQ;AACZ,aAAW,QAAQ,QAAQ,MAAM,CAAC,GAAG;AACnC,QAAI,KAAK,OAAO,UAAa,CAAC,OAAO,IAAI,KAAK,EAAE,EAAG;AACnD,UAAM,WAAW,KAAK,MAAM,UAAU,CAAC,UAAU,MAAM,OAAO,KAAK,EAAE;AACrE,QAAI,WAAW,EAAG;AAGlB,UAAM,QAAQ,KAAK,MAAM,MAAM,GAAG,QAAQ,EAAE,OAAO,CAAC,UAAU,CAAC,OAAO,IAAI,MAAM,EAAE,CAAC;AACnF,QAAI,MAAM,WAAW,GAAG;AAGtB,YAAI,UAAK,MAAM,CAAC,MAAZ,mBAAe,UAAO,WAAM,CAAC,MAAP,mBAAU,IAAI,YAAW,KAAK,KAAK;AAC7D;AAAA,IACF;AACA,YAAQ,CAAC,GAAG,OAAO,GAAG,KAAK;AAC3B,eAAW,SAAS,MAAO,QAAO,IAAI,MAAM,EAAE;AAC9C,eAAW,KAAK,KAAK;AACrB,aAAS,KAAK,gBAAgB,IAAI,CAAC;AACnC,aAAS;AAAA,EACX;AACA,QAAM,WAA+B,CAAA;AACrC,aAAW,SAAS,OAAO;AACzB,UAAM,UAAU,UAAU,IAAI,MAAM,EAAE;AACtC,QAAI,YAAY,OAAW,UAAS,KAAK,OAAO;AAAA,EAClD;AACA,SAAO,EAAE,UAAU,OAAO,UAAU,UAAU,MAAA;AAChD;AAyBO,SAAS,uBAAuB,EAAE,aAAuD;AAC9F,SAAO,EAAE,MAAM,wCAAwC,UAAA;AACzD;AAOO,SAAS,qBAAqB,SAA2C;AAC9E,MAAI,OAAO,YAAY,YAAY,YAAY,KAAM,QAAO;AAC5D,QAAM,EAAE,WAAW,QAAA,IAAY;AAC/B,MAAI,OAAO,cAAc,YAAY,CAAC,UAAW,QAAO;AACxD,SAAO,EAAE,WAAW,QAAA;AACtB;AA4CO,SAAS,yBAAyB,SAA+C;AACtF,MAAI,OAAO,YAAY,YAAY,YAAY,KAAM,QAAO;AAC5D,QAAM,SAAU,QAAoC,gCAAgC;AACpF,MAAI,OAAO,WAAW,YAAY,WAAW,KAAM,QAAO;AAC1D,QAAM,QAAS,OAAmC;AAClD,SAAO,QAAQ,KAAK,IAAI,EAAE,UAAU;AACtC;AAQO,MAAM,4CAA4C;AAUlD,SAAS,gCACd,UACoC;AACpC,MAAI,OAAO,aAAa,YAAY,aAAa,KAAM,QAAO;AAC9D,QAAM,SAAU,SAAqC,yCAAyC;AAC9F,MAAI,OAAO,WAAW,YAAY,WAAW,KAAM,QAAO;AAC1D,QAAM,EAAE,OAAO,UAAA,IAAc;AAC7B,MAAI,CAAC,QAAQ,KAAK,KAAK,OAAO,cAAc,YAAY,CAAC,UAAW,QAAO;AAC3E,SAAO,EAAE,OAAO,UAAA;AAClB;"}
+{"version":3,"file":"message-window.js","sources":["../../../../agent-runtime/src/message-window.ts"],"sourcesContent":["import type { LangChainMessage } from '@assistant-ui/react-langgraph';\nimport type { StatewireDocuments } from 'statewire';\nimport {\n  DEEP_AGENT_CUT_LANE_DOCUMENT,\n  DEEP_AGENT_DEFAULT_MESSAGE_WINDOW,\n  DEEP_AGENT_HISTORY_MESSAGE_COMMAND,\n  DEEP_AGENT_HISTORY_TOOL_OUTPUT_COMMAND,\n  DEEP_AGENT_INTEREST_PROTOCOL,\n  DEEP_AGENT_MAIN_NS,\n  DEEP_AGENT_MAX_INTERESTS,\n  DEEP_AGENT_MAX_MESSAGE_WINDOW,\n  DEEP_AGENT_MESSAGE_WINDOW_HEADER,\n  DEEP_AGENT_MIN_MESSAGE_WINDOW,\n  DEEP_AGENT_TRUNCATED_CONTENT_KEY,\n} from './generated/runconfig-contract';\nimport { DEEP_AGENT_STATEWIRE_PROTOCOL } from './statewire-transport';\n\n/**\n * The interest window — the client half of agora's\n * `agora/graphs/athena_deep_agent/message_window.py`.\n *\n * The wire main document carries no `messages`. A client states its interests\n * on the `athena/interest` lane as one array (`set(requests)`): the first is\n * the head (the newest `before` messages of the `main` namespace), the rest\n * are pages anchored `at` the first message of the window they extend. The\n * host holds one ephemeral main document per attach, `{ windows }`, one\n * resolution per request in order (the request echoed plus the covered\n * `chain` and `more` flags; an anchor that vanished leaves an empty window in\n * place), and one shared `athena` document per covered message, keyed\n * `[threadId, \"main\", messageId]` — or `[threadId, \"main\", messageId, \"cut\"]`\n * for a finished card a version-2 view carries cut to its newest steps (its\n * whole lane pages in through `history/lane`, see `lane-history.ts`).\n * {@link readDeepAgentMessageWindow} joins them into the transcript the\n * converter renders.\n *\n * Two host rules shape the client half. The head window never drops below\n * the oldest message it has carried on an attach (a run's appends grow the\n * view, nothing loaded slides out), so a page stays anchored at the head's\n * first message while the transcript grows; and the host remembers a client\n * id's interests across its attaches, so a reconnect's hello carries the\n * view the client held. A fresh client's hello carries the head it declared\n * on the attach headers ({@link withDeclaredMessageWindow}) — a statement\n * cannot ride the first attach, the hello negotiates the lanes.\n *\n * Pages are appended one per load until the host's interest cap, then the\n * loaded range is restated in as few maximal requests as it takes\n * ({@link pageInterestRequests}); the whole transcript stays reachable up to\n * `MAX_INTERESTS × MAX_MESSAGE_WINDOW` messages. A truncated tool output\n * comes back whole through `history/tool_output` ({@link buildToolOutputCommand});\n * a message's stored form — provenance blobs and all — through\n * `history/message` ({@link buildHistoryMessageCommand}).\n */\n\nconst isIndex = (value: unknown): value is number =>\n  typeof value === 'number' && Number.isInteger(value) && value >= 0;\n\n/** What a surface declares for its window: the interest size (the host clamps to its bounds). */\nexport interface DeepAgentMessageWindowDeclaration {\n  size?: number;\n}\n\nexport function clampMessageWindowSize(size?: number): number {\n  if (size === undefined || !Number.isFinite(size)) return DEEP_AGENT_DEFAULT_MESSAGE_WINDOW;\n  return Math.max(\n    DEEP_AGENT_MIN_MESSAGE_WINDOW,\n    Math.min(DEEP_AGENT_MAX_MESSAGE_WINDOW, Math.trunc(size))\n  );\n}\n\n/** An `athena/interest` request over the linear transcript. */\nexport interface DeepAgentInterestRequest {\n  readonly ns: typeof DEEP_AGENT_MAIN_NS;\n  /** Messages ending at the anchor (the tail without `at`). */\n  readonly before: number;\n  /** Anchor message id; a page's is the first message of the window it extends. */\n  readonly at?: string;\n  /** Messages past the anchor (requires `at`). */\n  readonly after?: number;\n  /** The newest steps of every subagent lane a covered card names (see `lane-follow.ts`). */\n  readonly follow?: { readonly before: number };\n}\n\nexport function headInterestRequest(size?: number): DeepAgentInterestRequest {\n  return { ns: DEEP_AGENT_MAIN_NS, before: clampMessageWindowSize(size) };\n}\n\nexport function pageInterestRequest(at: string, size?: number): DeepAgentInterestRequest {\n  return { ns: DEEP_AGENT_MAIN_NS, before: clampMessageWindowSize(size), at };\n}\n\n/**\n * Attach headers with the head window declared (`x-athena-deep-agent-window`):\n * the hello of a client the host does not remember carries the newest `size`\n * messages. Part of the one attach recipe, so a pooled dial asks for the shape\n * the thread's own attach would. An explicit entry (tests, overrides) wins.\n */\nexport function withDeclaredMessageWindow(\n  headers: Record<string, string>,\n  declaration: DeepAgentMessageWindowDeclaration = {}\n): Record<string, string> {\n  if (DEEP_AGENT_MESSAGE_WINDOW_HEADER in headers) return headers;\n  return {\n    ...headers,\n    [DEEP_AGENT_MESSAGE_WINDOW_HEADER]: String(clampMessageWindowSize(declaration.size)),\n  };\n}\n\nexport interface DeepAgentInterestChainEntry {\n  id: string;\n  siblings: string[];\n}\n\n/** The host's resolution document of one interest: the request echoed plus its coverage. */\nexport interface DeepAgentInterestResolution extends DeepAgentInterestRequest {\n  /** Covered message ids, oldest first (a page ends at its anchor). */\n  chain: DeepAgentInterestChainEntry[];\n  more: { before: boolean; after: boolean };\n}\n\nexport function readDeepAgentInterestResolution(value: unknown): DeepAgentInterestResolution | null {\n  if (typeof value !== 'object' || value === null) return null;\n  const { ns, before, at, after, chain, more } = value as Record<string, unknown>;\n  if (ns !== DEEP_AGENT_MAIN_NS || !isIndex(before) || !Array.isArray(chain)) return null;\n  if (at !== undefined && typeof at !== 'string') return null;\n  if (after !== undefined && !isIndex(after)) return null;\n  if (typeof more !== 'object' || more === null) return null;\n  const { before: moreBefore, after: moreAfter } = more as Record<string, unknown>;\n  if (typeof moreBefore !== 'boolean' || typeof moreAfter !== 'boolean') return null;\n  const entries: DeepAgentInterestChainEntry[] = [];\n  for (const entry of chain) {\n    const { id, siblings } = (entry ?? {}) as Record<string, unknown>;\n    if (typeof id !== 'string') return null;\n    entries.push({\n      id,\n      siblings: Array.isArray(siblings) ? siblings.filter((s) => typeof s === 'string') : [id],\n    });\n  }\n  return {\n    ns: DEEP_AGENT_MAIN_NS,\n    before,\n    ...(at !== undefined && { at }),\n    ...(after !== undefined && { after }),\n    chain: entries,\n    more: { before: moreBefore, after: moreAfter },\n  };\n}\n\n/** The attach's interest document: one resolution per stated request, in order. */\nexport interface DeepAgentInterestDocument {\n  windows: DeepAgentInterestResolution[];\n}\n\nexport function readDeepAgentInterestDocument(value: unknown): DeepAgentInterestDocument | null {\n  if (typeof value !== 'object' || value === null) return null;\n  const { windows } = value as Record<string, unknown>;\n  if (!Array.isArray(windows)) return null;\n  const resolutions: DeepAgentInterestResolution[] = [];\n  for (const window of windows) {\n    const resolution = readDeepAgentInterestResolution(window);\n    if (resolution === null) return null;\n    resolutions.push(resolution);\n  }\n  return { windows: resolutions };\n}\n\nconst interestRequest = ({ ns, before, at, after }: DeepAgentInterestRequest): DeepAgentInterestRequest => ({\n  ns,\n  before,\n  ...(at !== undefined && { at }),\n  ...(after !== undefined && { after }),\n});\n\n/** The loaded transcript window, joined from the head interest and its anchored pages. */\nexport interface DeepAgentMessageWindow {\n  /** The requests the joined view stands on, in order (the head first): what a `set` restates. A stale page (its anchor gone) or one the head already covers is left out. */\n  requests: DeepAgentInterestRequest[];\n  /** Covered chain, oldest first. */\n  chain: DeepAgentInterestChainEntry[];\n  /** The chain's message documents, in chain order (an unmounted one is skipped). */\n  messages: LangChainMessage[];\n  /** Whether the host holds messages older than the window. */\n  hasOlder: boolean;\n  /** Pages joined onto the head. */\n  pages: number;\n}\n\n/**\n * The loaded chain plus `grow` older messages, restated in as few requests as\n * it takes: a maximal head, then maximal pages each anchored at the first\n * message of the window above it. `null` once even that exceeds the host's\n * interest cap — the reachable depth is `MAX_INTERESTS × MAX_MESSAGE_WINDOW`.\n */\nfunction coalescedInterestRequests(\n  chain: readonly DeepAgentInterestChainEntry[],\n  grow: number\n): DeepAgentInterestRequest[] | null {\n  const loaded = chain.length;\n  const requests: DeepAgentInterestRequest[] = [\n    { ns: DEEP_AGENT_MAIN_NS, before: Math.min(DEEP_AGENT_MAX_MESSAGE_WINDOW, loaded) },\n  ];\n  // `low` is the chain index of the lowest message the plan covers so far.\n  let low = loaded - (requests[0] as DeepAgentInterestRequest).before;\n  while (low > 0) {\n    const before = Math.min(DEEP_AGENT_MAX_MESSAGE_WINDOW, low + 1);\n    requests.push({\n      ns: DEEP_AGENT_MAIN_NS,\n      before,\n      at: (chain[low] as DeepAgentInterestChainEntry).id,\n    });\n    low = low + 1 - before;\n  }\n  // The plan covers the chain; the new messages extend its lowest request\n  // when that stays within one window, else they are a page of their own.\n  const last = requests[requests.length - 1] as DeepAgentInterestRequest;\n  if (last.before + grow <= DEEP_AGENT_MAX_MESSAGE_WINDOW) {\n    requests[requests.length - 1] = { ...last, before: last.before + grow };\n  } else {\n    requests.push({\n      ns: DEEP_AGENT_MAIN_NS,\n      before: grow + 1,\n      at: (chain[0] as DeepAgentInterestChainEntry).id,\n    });\n  }\n  return requests.length > DEEP_AGENT_MAX_INTERESTS ? null : requests;\n}\n\n/**\n * The requests to state for one more page of older history, or `null` when\n * there is nothing to load (no anchor, the beginning reached) or the host's\n * interest cap is spent. One page is appended, anchored at the window's first\n * message, while the cap allows; at the cap the loaded range is coalesced\n * into maximal requests first (see {@link coalescedInterestRequests}).\n */\nexport function pageInterestRequests(\n  window: DeepAgentMessageWindow,\n  size?: number\n): DeepAgentInterestRequest[] | null {\n  const anchor = window.chain[0]?.id;\n  if (anchor === undefined || !window.hasOlder) return null;\n  const page = pageInterestRequest(anchor, size);\n  const appended = [...window.requests, page];\n  if (appended.length <= DEEP_AGENT_MAX_INTERESTS) return appended;\n  // A page's `before` counts its anchor: it adds `before - 1` older messages.\n  return coalescedInterestRequests(window.chain, page.before - 1);\n}\n\n/**\n * Whether every `stated` request is carried by `carried` (the same anchor, at\n * least as wide): a reconnect's hello that carries what the client last\n * stated — the host remembers a client id's interests — needs no restatement.\n */\nexport function interestsCovered(\n  carried: readonly DeepAgentInterestRequest[],\n  stated: readonly DeepAgentInterestRequest[]\n): boolean {\n  return stated.every((request) =>\n    carried.some((have) => have.at === request.at && have.before >= request.before)\n  );\n}\n\n/**\n * The window off the mounted documents, or `null` while the head interest has\n * not resolved (pre-attach, before the first `set` answer, a wire without\n * documents). A page joins wherever its anchor sits in the chain so far — at\n * the first message it extends the view, inside the head (the head widened\n * past it) it adds nothing and is left out of `requests`; a page whose anchor\n * left the transcript (a rewind emptied it in place) is left out too. A\n * malformed document reads as absent.\n */\nexport function readDeepAgentMessageWindow(\n  mounted: ReadonlyMap<number, StatewireDocuments.Document> | undefined\n): DeepAgentMessageWindow | null {\n  if (mounted === undefined) return null;\n  let windows: DeepAgentInterestResolution[] | undefined;\n  const documents = new Map<string, LangChainMessage>();\n  for (const document of mounted.values()) {\n    const key = document.key;\n    if (document.protocol === DEEP_AGENT_INTEREST_PROTOCOL && document.main === true) {\n      windows = readDeepAgentInterestDocument(document.value)?.windows;\n    } else if (\n      document.protocol === DEEP_AGENT_STATEWIRE_PROTOCOL &&\n      key !== null &&\n      key[1] === DEEP_AGENT_MAIN_NS &&\n      (key.length === 3 || (key.length === 4 && key[3] === DEEP_AGENT_CUT_LANE_DOCUMENT))\n    ) {\n      documents.set(key[2] as string, document.value as LangChainMessage);\n    }\n  }\n  const head = windows?.[0];\n  if (windows === undefined || head === undefined) return null;\n  let chain = head.chain;\n  let hasOlder = head.more.before;\n  const requests = [interestRequest(head)];\n  const loaded = new Set(chain.map((entry) => entry.id));\n  let pages = 0;\n  for (const page of windows.slice(1)) {\n    if (page.at === undefined || !loaded.has(page.at)) continue;\n    const anchorAt = page.chain.findIndex((entry) => entry.id === page.at);\n    if (anchorAt < 0) continue;\n    // Both windows are ranges of one linear transcript sharing the anchor:\n    // what the page holds below it and the chain does not is a prefix.\n    const fresh = page.chain.slice(0, anchorAt).filter((entry) => !loaded.has(entry.id));\n    if (fresh.length === 0) {\n      // Nothing new, but a page that starts where the loaded chain starts\n      // has looked below it: its `more.before` is the answer for the edge.\n      if (page.chain[0]?.id === chain[0]?.id) hasOlder = page.more.before;\n      continue;\n    }\n    chain = [...fresh, ...chain];\n    for (const entry of fresh) loaded.add(entry.id);\n    hasOlder = page.more.before;\n    requests.push(interestRequest(page));\n    pages += 1;\n  }\n  const messages: LangChainMessage[] = [];\n  for (const entry of chain) {\n    const message = documents.get(entry.id);\n    if (message !== undefined) messages.push(message);\n  }\n  return { requests, chain, messages, hasOlder, pages };\n}\n\n/**\n * The windowed wire's read command on the statewire command union, so\n * `sendCommand({ type: 'history/tool_output', … })` typechecks wherever the\n * deep-agent host is driven (`DeepAgentThread.history_tool_output`, an\n * attach-ACL read).\n */\ndeclare module '@assistant-ui/core' {\n  namespace Assistant {\n    interface Commands {\n      /** A tool output the windowed wire carried cut, whole. */\n      historyToolOutput: ToolOutputCommand;\n      /** One stored message whole — the per-client shape, by command. */\n      historyMessage: HistoryMessageCommand;\n    }\n  }\n}\n\n/** The `history/tool_output` command: a truncated tool output, whole. */\nexport interface ToolOutputCommand {\n  readonly type: typeof DEEP_AGENT_HISTORY_TOOL_OUTPUT_COMMAND;\n  readonly messageId: string;\n}\n\nexport function buildToolOutputCommand({ messageId }: { messageId: string }): ToolOutputCommand {\n  return { type: DEEP_AGENT_HISTORY_TOOL_OUTPUT_COMMAND, messageId };\n}\n\nexport interface ToolOutputResult {\n  messageId: string;\n  content: unknown;\n}\n\nexport function readToolOutputResult(payload: unknown): ToolOutputResult | null {\n  if (typeof payload !== 'object' || payload === null) return null;\n  const { messageId, content } = payload as Record<string, unknown>;\n  if (typeof messageId !== 'string' || !messageId) return null;\n  return { messageId, content };\n}\n\n/**\n * The `history/message` command: a message as the host stores it — the\n * model-provenance blobs the wire strips and an untruncated tool output\n * included. Every client receives one projected wire form; this is the one\n * per-client shape, fetched on demand like `history/tool_output`.\n */\nexport interface HistoryMessageCommand {\n  readonly type: typeof DEEP_AGENT_HISTORY_MESSAGE_COMMAND;\n  readonly messageId: string;\n}\n\nexport function buildHistoryMessageCommand({\n  messageId,\n}: {\n  messageId: string;\n}): HistoryMessageCommand {\n  return { type: DEEP_AGENT_HISTORY_MESSAGE_COMMAND, messageId };\n}\n\nexport interface HistoryMessageResult {\n  messageId: string;\n  message: LangChainMessage;\n}\n\nexport function readHistoryMessageResult(payload: unknown): HistoryMessageResult | null {\n  if (typeof payload !== 'object' || payload === null) return null;\n  const { messageId, message } = payload as Record<string, unknown>;\n  if (typeof messageId !== 'string' || !messageId) return null;\n  if (typeof message !== 'object' || message === null) return null;\n  return { messageId, message: message as LangChainMessage };\n}\n\n/** The wire marker on a tool message whose content the host cut. */\nexport interface TruncatedToolContent {\n  /** Length of the full output, in characters. */\n  chars: number;\n}\n\n/**\n * The truncation marker off a raw wire message, or `null` when the message's\n * content is whole. Pure and selector-safe on the marker object.\n */\nexport function readTruncatedToolContent(message: unknown): TruncatedToolContent | null {\n  if (typeof message !== 'object' || message === null) return null;\n  const marker = (message as Record<string, unknown>)[DEEP_AGENT_TRUNCATED_CONTENT_KEY];\n  if (typeof marker !== 'object' || marker === null) return null;\n  const chars = (marker as Record<string, unknown>).chars;\n  return isIndex(chars) ? { chars } : null;\n}\n\n/**\n * `artifact` key the shared converter lifts a truncated tool output's marker\n * into (with the tool message id), so a tool card can ask for the full output\n * — the converter joins the ToolMessage into its tool-call part and would\n * otherwise drop the wire marker.\n */\nexport const ATHENA_TRUNCATED_TOOL_OUTPUT_ARTIFACT_KEY = '_athena_content_truncated';\n\nexport interface TruncatedToolOutputArtifact extends TruncatedToolContent {\n  /** The tool message to name in `history/tool_output`. */\n  messageId: string;\n}\n\n/**\n * The lifted truncation marker off a tool-call part's `artifact`, or `null`.\n */\nexport function readTruncatedToolOutputArtifact(\n  artifact: unknown\n): TruncatedToolOutputArtifact | null {\n  if (typeof artifact !== 'object' || artifact === null) return null;\n  const lifted = (artifact as Record<string, unknown>)[ATHENA_TRUNCATED_TOOL_OUTPUT_ARTIFACT_KEY];\n  if (typeof lifted !== 'object' || lifted === null) return null;\n  const { chars, messageId } = lifted as Record<string, unknown>;\n  if (!isIndex(chars) || typeof messageId !== 'string' || !messageId) return null;\n  return { chars, messageId };\n}\n"],"names":[],"mappings":";;AAqDA,MAAM,UAAU,CAAC,UACf,OAAO,UAAU,YAAY,OAAO,UAAU,KAAK,KAAK,SAAS;AAO5D,SAAS,uBAAuB,MAAuB;AAC5D,MAAI,SAAS,UAAa,CAAC,OAAO,SAAS,IAAI,EAAG,QAAO;AACzD,SAAO,KAAK;AAAA,IACV;AAAA,IACA,KAAK,IAAI,+BAA+B,KAAK,MAAM,IAAI,CAAC;AAAA,EAAA;AAE5D;AAeO,SAAS,oBAAoB,MAAyC;AAC3E,SAAO,EAAE,IAAI,oBAAoB,QAAQ,uBAAuB,IAAI,EAAA;AACtE;AAEO,SAAS,oBAAoB,IAAY,MAAyC;AACvF,SAAO,EAAE,IAAI,oBAAoB,QAAQ,uBAAuB,IAAI,GAAG,GAAA;AACzE;AAQO,SAAS,0BACd,SACA,cAAiD,IACzB;AACxB,MAAI,oCAAoC,QAAS,QAAO;AACxD,SAAO;AAAA,IACL,GAAG;AAAA,IACH,CAAC,gCAAgC,GAAG,OAAO,uBAAuB,YAAY,IAAI,CAAC;AAAA,EAAA;AAEvF;AAcO,SAAS,gCAAgC,OAAoD;AAClG,MAAI,OAAO,UAAU,YAAY,UAAU,KAAM,QAAO;AACxD,QAAM,EAAE,IAAI,QAAQ,IAAI,OAAO,OAAO,SAAS;AAC/C,MAAI,OAAO,sBAAsB,CAAC,QAAQ,MAAM,KAAK,CAAC,MAAM,QAAQ,KAAK,EAAG,QAAO;AACnF,MAAI,OAAO,UAAa,OAAO,OAAO,SAAU,QAAO;AACvD,MAAI,UAAU,UAAa,CAAC,QAAQ,KAAK,EAAG,QAAO;AACnD,MAAI,OAAO,SAAS,YAAY,SAAS,KAAM,QAAO;AACtD,QAAM,EAAE,QAAQ,YAAY,OAAO,cAAc;AACjD,MAAI,OAAO,eAAe,aAAa,OAAO,cAAc,UAAW,QAAO;AAC9E,QAAM,UAAyC,CAAA;AAC/C,aAAW,SAAS,OAAO;AACzB,UAAM,EAAE,IAAI,SAAA,IAAc,SAAS,CAAA;AACnC,QAAI,OAAO,OAAO,SAAU,QAAO;AACnC,YAAQ,KAAK;AAAA,MACX;AAAA,MACA,UAAU,MAAM,QAAQ,QAAQ,IAAI,SAAS,OAAO,CAAC,MAAM,OAAO,MAAM,QAAQ,IAAI,CAAC,EAAE;AAAA,IAAA,CACxF;AAAA,EACH;AACA,SAAO;AAAA,IACL,IAAI;AAAA,IACJ;AAAA,IACA,GAAI,OAAO,UAAa,EAAE,GAAA;AAAA,IAC1B,GAAI,UAAU,UAAa,EAAE,MAAA;AAAA,IAC7B,OAAO;AAAA,IACP,MAAM,EAAE,QAAQ,YAAY,OAAO,UAAA;AAAA,EAAU;AAEjD;AAOO,SAAS,8BAA8B,OAAkD;AAC9F,MAAI,OAAO,UAAU,YAAY,UAAU,KAAM,QAAO;AACxD,QAAM,EAAE,YAAY;AACpB,MAAI,CAAC,MAAM,QAAQ,OAAO,EAAG,QAAO;AACpC,QAAM,cAA6C,CAAA;AACnD,aAAW,UAAU,SAAS;AAC5B,UAAM,aAAa,gCAAgC,MAAM;AACzD,QAAI,eAAe,KAAM,QAAO;AAChC,gBAAY,KAAK,UAAU;AAAA,EAC7B;AACA,SAAO,EAAE,SAAS,YAAA;AACpB;AAEA,MAAM,kBAAkB,CAAC,EAAE,IAAI,QAAQ,IAAI,aAAiE;AAAA,EAC1G;AAAA,EACA;AAAA,EACA,GAAI,OAAO,UAAa,EAAE,GAAA;AAAA,EAC1B,GAAI,UAAU,UAAa,EAAE,MAAA;AAC/B;AAsBA,SAAS,0BACP,OACA,MACmC;AACnC,QAAM,SAAS,MAAM;AACrB,QAAM,WAAuC;AAAA,IAC3C,EAAE,IAAI,oBAAoB,QAAQ,KAAK,IAAI,+BAA+B,MAAM,EAAA;AAAA,EAAE;AAGpF,MAAI,MAAM,SAAU,SAAS,CAAC,EAA+B;AAC7D,SAAO,MAAM,GAAG;AACd,UAAM,SAAS,KAAK,IAAI,+BAA+B,MAAM,CAAC;AAC9D,aAAS,KAAK;AAAA,MACZ,IAAI;AAAA,MACJ;AAAA,MACA,IAAK,MAAM,GAAG,EAAkC;AAAA,IAAA,CACjD;AACD,UAAM,MAAM,IAAI;AAAA,EAClB;AAGA,QAAM,OAAO,SAAS,SAAS,SAAS,CAAC;AACzC,MAAI,KAAK,SAAS,QAAQ,+BAA+B;AACvD,aAAS,SAAS,SAAS,CAAC,IAAI,EAAE,GAAG,MAAM,QAAQ,KAAK,SAAS,KAAA;AAAA,EACnE,OAAO;AACL,aAAS,KAAK;AAAA,MACZ,IAAI;AAAA,MACJ,QAAQ,OAAO;AAAA,MACf,IAAK,MAAM,CAAC,EAAkC;AAAA,IAAA,CAC/C;AAAA,EACH;AACA,SAAO,SAAS,SAAS,2BAA2B,OAAO;AAC7D;AASO,SAAS,qBACd,QACA,MACmC;;AACnC,QAAM,UAAS,YAAO,MAAM,CAAC,MAAd,mBAAiB;AAChC,MAAI,WAAW,UAAa,CAAC,OAAO,SAAU,QAAO;AACrD,QAAM,OAAO,oBAAoB,QAAQ,IAAI;AAC7C,QAAM,WAAW,CAAC,GAAG,OAAO,UAAU,IAAI;AAC1C,MAAI,SAAS,UAAU,yBAA0B,QAAO;AAExD,SAAO,0BAA0B,OAAO,OAAO,KAAK,SAAS,CAAC;AAChE;AAOO,SAAS,iBACd,SACA,QACS;AACT,SAAO,OAAO;AAAA,IAAM,CAAC,YACnB,QAAQ,KAAK,CAAC,SAAS,KAAK,OAAO,QAAQ,MAAM,KAAK,UAAU,QAAQ,MAAM;AAAA,EAAA;AAElF;AAWO,SAAS,2BACd,SAC+B;;AAC/B,MAAI,YAAY,OAAW,QAAO;AAClC,MAAI;AACJ,QAAM,gCAAgB,IAAA;AACtB,aAAW,YAAY,QAAQ,UAAU;AACvC,UAAM,MAAM,SAAS;AACrB,QAAI,SAAS,aAAa,gCAAgC,SAAS,SAAS,MAAM;AAChF,iBAAU,mCAA8B,SAAS,KAAK,MAA5C,mBAA+C;AAAA,IAC3D,WACE,SAAS,aAAa,iCACtB,QAAQ,QACR,IAAI,CAAC,MAAM,uBACV,IAAI,WAAW,KAAM,IAAI,WAAW,KAAK,IAAI,CAAC,MAAM,+BACrD;AACA,gBAAU,IAAI,IAAI,CAAC,GAAa,SAAS,KAAyB;AAAA,IACpE;AAAA,EACF;AACA,QAAM,OAAO,mCAAU;AACvB,MAAI,YAAY,UAAa,SAAS,OAAW,QAAO;AACxD,MAAI,QAAQ,KAAK;AACjB,MAAI,WAAW,KAAK,KAAK;AACzB,QAAM,WAAW,CAAC,gBAAgB,IAAI,CAAC;AACvC,QAAM,SAAS,IAAI,IAAI,MAAM,IAAI,CAAC,UAAU,MAAM,EAAE,CAAC;AACrD,MAAI,QAAQ;AACZ,aAAW,QAAQ,QAAQ,MAAM,CAAC,GAAG;AACnC,QAAI,KAAK,OAAO,UAAa,CAAC,OAAO,IAAI,KAAK,EAAE,EAAG;AACnD,UAAM,WAAW,KAAK,MAAM,UAAU,CAAC,UAAU,MAAM,OAAO,KAAK,EAAE;AACrE,QAAI,WAAW,EAAG;AAGlB,UAAM,QAAQ,KAAK,MAAM,MAAM,GAAG,QAAQ,EAAE,OAAO,CAAC,UAAU,CAAC,OAAO,IAAI,MAAM,EAAE,CAAC;AACnF,QAAI,MAAM,WAAW,GAAG;AAGtB,YAAI,UAAK,MAAM,CAAC,MAAZ,mBAAe,UAAO,WAAM,CAAC,MAAP,mBAAU,IAAI,YAAW,KAAK,KAAK;AAC7D;AAAA,IACF;AACA,YAAQ,CAAC,GAAG,OAAO,GAAG,KAAK;AAC3B,eAAW,SAAS,MAAO,QAAO,IAAI,MAAM,EAAE;AAC9C,eAAW,KAAK,KAAK;AACrB,aAAS,KAAK,gBAAgB,IAAI,CAAC;AACnC,aAAS;AAAA,EACX;AACA,QAAM,WAA+B,CAAA;AACrC,aAAW,SAAS,OAAO;AACzB,UAAM,UAAU,UAAU,IAAI,MAAM,EAAE;AACtC,QAAI,YAAY,OAAW,UAAS,KAAK,OAAO;AAAA,EAClD;AACA,SAAO,EAAE,UAAU,OAAO,UAAU,UAAU,MAAA;AAChD;AAyBO,SAAS,uBAAuB,EAAE,aAAuD;AAC9F,SAAO,EAAE,MAAM,wCAAwC,UAAA;AACzD;AAOO,SAAS,qBAAqB,SAA2C;AAC9E,MAAI,OAAO,YAAY,YAAY,YAAY,KAAM,QAAO;AAC5D,QAAM,EAAE,WAAW,QAAA,IAAY;AAC/B,MAAI,OAAO,cAAc,YAAY,CAAC,UAAW,QAAO;AACxD,SAAO,EAAE,WAAW,QAAA;AACtB;AAaO,SAAS,2BAA2B;AAAA,EACzC;AACF,GAE0B;AACxB,SAAO,EAAE,MAAM,oCAAoC,UAAA;AACrD;AAOO,SAAS,yBAAyB,SAA+C;AACtF,MAAI,OAAO,YAAY,YAAY,YAAY,KAAM,QAAO;AAC5D,QAAM,EAAE,WAAW,QAAA,IAAY;AAC/B,MAAI,OAAO,cAAc,YAAY,CAAC,UAAW,QAAO;AACxD,MAAI,OAAO,YAAY,YAAY,YAAY,KAAM,QAAO;AAC5D,SAAO,EAAE,WAAW,QAAA;AACtB;AAYO,SAAS,yBAAyB,SAA+C;AACtF,MAAI,OAAO,YAAY,YAAY,YAAY,KAAM,QAAO;AAC5D,QAAM,SAAU,QAAoC,gCAAgC;AACpF,MAAI,OAAO,WAAW,YAAY,WAAW,KAAM,QAAO;AAC1D,QAAM,QAAS,OAAmC;AAClD,SAAO,QAAQ,KAAK,IAAI,EAAE,UAAU;AACtC;AAQO,MAAM,4CAA4C;AAUlD,SAAS,gCACd,UACoC;AACpC,MAAI,OAAO,aAAa,YAAY,aAAa,KAAM,QAAO;AAC9D,QAAM,SAAU,SAAqC,yCAAyC;AAC9F,MAAI,OAAO,WAAW,YAAY,WAAW,KAAM,QAAO;AAC1D,QAAM,EAAE,OAAO,UAAA,IAAc;AAC7B,MAAI,CAAC,QAAQ,KAAK,KAAK,OAAO,cAAc,YAAY,CAAC,UAAW,QAAO;AAC3E,SAAO,EAAE,OAAO,UAAA;AAClB;"}

package/dist/agent-runtime/src/messages.cjs

@@ -12,9 +12,90 @@ const composerMarkdown = require("./composer-markdown.cjs");
 const runconfigContract = require("./generated/runconfig-contract.cjs");
 const messageWindow = require("./message-window.cjs");
 const optimisticContent = require("./optimistic-content.cjs");
+const runningToolPlaceholder = require("./running-tool-placeholder.cjs");
 const statewireState = require("./statewire-state.cjs");
 const voiceTurnGrouping = require("./voice-turn-grouping.cjs");
-const messageConverter = react.createMessageConverter(reactLanggraph.convertLangChainMessages);
+const CACHE_EXCLUSIVE_ADAPTERS = /* @__PURE__ */ new Set([
+  "amazon_bedrock",
+  "aws_bedrock",
+  "bedrock"
+]);
+const tokens = (value, fallback = 0) => typeof value === "number" && Number.isFinite(value) ? Math.trunc(value) : typeof value === "string" && /^\d+$/.test(value) ? Number(value) : fallback;
+const CACHE_READ_KEYS = ["cache_read", "cached", "cached_tokens", "cache_read_tokens"];
+const CACHE_CREATION_KEYS = [
+  "cache_creation",
+  "cache_creation_tokens",
+  "cache_write",
+  "cache_write_tokens"
+];
+const cacheCount = (details, keys) => {
+  for (const key of keys) {
+    if (details[key] != null) return tokens(details[key]);
+  }
+  return 0;
+};
+function nativeStepUsage(message) {
+  const { usage_metadata: usage, response_metadata: response } = message;
+  if (typeof usage !== "object" || usage === null) return void 0;
+  const input = tokens(usage.input_tokens);
+  const output = tokens(usage.output_tokens);
+  const total = tokens(usage.total_tokens, input + output);
+  const details = typeof usage.input_token_details === "object" && usage.input_token_details !== null ? usage.input_token_details : {};
+  const cached = cacheCount(details, CACHE_READ_KEYS) + cacheCount(details, CACHE_CREATION_KEYS);
+  const adapter = response == null ? void 0 : response.model_provider;
+  const excluded = cached !== 0 && (typeof adapter === "string" && CACHE_EXCLUSIVE_ADAPTERS.has(adapter.trim().toLowerCase().replace(/-/g, "_")) || total < cached + output);
+  return { inputTokens: excluded ? input + cached : input, outputTokens: output };
+}
+function nativeTurnTiming(message) {
+  var _a;
+  if (message.type !== "ai") return void 0;
+  const kwargs = message.additional_kwargs;
+  const start = kwargs == null ? void 0 : kwargs.agent_start_time;
+  const duration = kwargs == null ? void 0 : kwargs.agent_duration_seconds;
+  if (typeof start !== "number" || typeof duration !== "number" || !(start > 0 && duration >= 0)) {
+    return void 0;
+  }
+  return {
+    streamStartTime: start * 1e3,
+    totalStreamTime: duration * 1e3,
+    totalChunks: 0,
+    toolCallCount: ((_a = message.tool_calls) == null ? void 0 : _a.length) ?? 0
+  };
+}
+const convertDeepAgentLangChainMessage = (message, metadata) => {
+  const converted = reactLanggraph.convertLangChainMessages(message, metadata);
+  if (message.type !== "ai" || Array.isArray(converted) || converted.role !== "assistant") {
+    return converted;
+  }
+  const usage = nativeStepUsage(message);
+  const timing = nativeTurnTiming(message);
+  if (usage === void 0 && timing === void 0) return converted;
+  return {
+    ...converted,
+    metadata: {
+      ...converted.metadata,
+      ...usage !== void 0 && { steps: [{ messageId: message.id, usage }] },
+      ...timing !== void 0 && { timing }
+    }
+  };
+};
+const messageConverter = react.createMessageConverter(convertDeepAgentLangChainMessage);
+const preliminaryMarkedTurns = /* @__PURE__ */ new WeakMap();
+function markPreliminaryToolResults(message) {
+  if (message.role !== "assistant") return message;
+  const cached = preliminaryMarkedTurns.get(message);
+  if (cached !== void 0) return cached;
+  let content = null;
+  for (const [index, part] of message.content.entries()) {
+    if (part.type !== "tool-call" || part.isPreliminary === true) continue;
+    if (!runningToolPlaceholder.isRunningToolPlaceholderArtifact(part.artifact)) continue;
+    content ?? (content = [...message.content]);
+    content[index] = { ...part, isPreliminary: true };
+  }
+  const marked = content === null ? message : { ...message, content };
+  preliminaryMarkedTurns.set(message, marked);
+  return marked;
+}
 class DeepAgentMessageConversionCache {
   constructor() {
     __publicField(this, "cache", react.createExternalMessageConversionCache());
@@ -41,11 +122,11 @@ class DeepAgentMessageConversionCache {
   convert(messages, isRunning) {
     const converted = react.convertExternalMessages(
       messages,
-      reactLanggraph.convertLangChainMessages,
+      convertDeepAgentLangChainMessage,
       isRunning,
       this.metadata,
       this.cache
-    );
+    ).map(markPreliminaryToolResults);
     const next = this.previous !== null && this.previousConverted !== null ? keepUnchangedParts({
       previous: this.previous,
       previousConverted: this.previousConverted,
@@ -633,7 +714,7 @@ function convertDeepAgentState(state, {
     ...laneTail
   ];
   return {
-    messages: conversionCache ? conversionCache.convert(sources, isRunning) : messageConverter.toThreadMessages(sources, isRunning),
+    messages: conversionCache ? conversionCache.convert(sources, isRunning) : messageConverter.toThreadMessages(sources, isRunning).map(markPreliminaryToolResults),
     isRunning
   };
 }
@@ -650,6 +731,7 @@ exports.ATHENA_STOP_SOURCE_SEND_NOW = ATHENA_STOP_SOURCE_SEND_NOW;
 exports.ATHENA_TERMINATION_REASON_KEY = ATHENA_TERMINATION_REASON_KEY;
 exports.DeepAgentMessageConversionCache = DeepAgentMessageConversionCache;
 exports.PENDING_ASSISTANT_MESSAGE_ID = PENDING_ASSISTANT_MESSAGE_ID;
+exports.convertDeepAgentLangChainMessage = convertDeepAgentLangChainMessage;
 exports.convertDeepAgentState = convertDeepAgentState;
 exports.createDeepAgentMaterialize = createDeepAgentMaterialize;
 exports.createDeepAgentMessageConversionCache = createDeepAgentMessageConversionCache;
@@ -670,7 +752,9 @@ exports.liftRunMarkerIntoMetadata = liftRunMarkerIntoMetadata;
 exports.liftTerminationIntoMetadata = liftTerminationIntoMetadata;
 exports.liftToolMessageMetadataIntoArtifact = liftToolMessageMetadataIntoArtifact;
 exports.liftTruncatedToolOutputIntoArtifact = liftTruncatedToolOutputIntoArtifact;
+exports.markPreliminaryToolResults = markPreliminaryToolResults;
 exports.messageConverter = messageConverter;
+exports.nativeStepUsage = nativeStepUsage;
 exports.normalizeStringContentParts = normalizeStringContentParts;
 exports.readRunMarkerFromSourceMessages = readRunMarkerFromSourceMessages;
 exports.sanitizeLangChainMessages = sanitizeLangChainMessages;