@assistant-ui/core 0.3.15 → 0.3.17

package/dist/runtime/utils/message-repository.js.map

@@ -1 +1 @@
-{"version":3,"file":"message-repository.js","names":["ThreadMessage","RunConfig","generateId","ThreadMessageLike","getAutoStatus","fromThreadMessageLike","ExportedMessageRepositoryItem","message","parentId","runConfig","ExportedMessageRepository","headId","messages","Array","fromArray","conv","map","m","undefined","idx","id","fromBranchableArray","items","options","fallbackStatus","Error","RepositoryParent","children","next","RepositoryMessage","prev","current","level","findHead","CachedValue","_value","T","func","constructor","value","dirty","MessageRepository","Map","head","root","updateLevels","newLevel","pending","length","pop","childId","childMessage","get","push","performOp","newParent","child","operation","parentOrRoot","newParentOrRoot","filter","fallbackId","at","fallback","_messages","canonicalHeadId","metadata","isOptimistic","getMessages","headMessage","addOrUpdateMessage","existingItem","newItem","set","getMessage","messageId","index","deleteMessage","replacementId","replacement","delete","getBranches","evictOffBranchOptimisticMessages","previousHead","currentHead","onHeadBranch","Set","add","stale","has","switchToBranch","prevOrRoot","resetHead","clear","descendantId","export","exportItems","import"],"sources":["../../../src/runtime/utils/message-repository.ts"],"sourcesContent":["import type { ThreadMessage } from \"../../types/message\";\nimport type { RunConfig } from \"../../types/message\";\nimport { generateId } from \"../../utils/id\";\nimport type { ThreadMessageLike } from \"./thread-message-like\";\nimport { getAutoStatus } from \"./auto-status\";\nimport { fromThreadMessageLike } from \"./thread-message-like\";\n\nexport type ExportedMessageRepositoryItem = {\n  message: ThreadMessage;\n  parentId: string | null;\n  runConfig?: RunConfig;\n};\n\nexport type ExportedMessageRepository = {\n  headId?: string | null;\n  messages: Array<{\n    message: ThreadMessage;\n    parentId: string | null;\n    runConfig?: RunConfig;\n  }>;\n};\n\nexport const ExportedMessageRepository = {\n  fromArray: (\n    messages: readonly ThreadMessageLike[],\n  ): ExportedMessageRepository => {\n    const conv = messages.map((m) =>\n      fromThreadMessageLike(\n        m,\n        generateId(),\n        getAutoStatus(false, false, false, false, undefined),\n      ),\n    );\n\n    return {\n      messages: conv.map((m, idx) => ({\n        parentId: idx > 0 ? conv[idx - 1]!.id : null,\n        message: m,\n      })),\n    };\n  },\n\n  fromBranchableArray: (\n    items: readonly {\n      message: ThreadMessageLike;\n      parentId: string | null;\n    }[],\n    options?: { headId?: string | null },\n  ): ExportedMessageRepository => {\n    const fallbackStatus = getAutoStatus(false, false, false, false, undefined);\n    return {\n      ...(options?.headId !== undefined\n        ? { headId: options.headId }\n        : undefined),\n      messages: items.map(({ message, parentId }) => {\n        if (!message.id) {\n          throw new Error(\n            \"ExportedMessageRepository.fromBranchableArray: Each message must have an 'id' field set.\",\n          );\n        }\n        return {\n          parentId,\n          message: fromThreadMessageLike(message, message.id, fallbackStatus),\n        };\n      }),\n    };\n  },\n};\n\ntype RepositoryParent = {\n  children: string[];\n  next: RepositoryMessage | null;\n};\n\ntype RepositoryMessage = RepositoryParent & {\n  prev: RepositoryMessage | null;\n  current: ThreadMessage;\n  level: number;\n};\n\nconst findHead = (\n  message: RepositoryMessage | RepositoryParent,\n): RepositoryMessage | null => {\n  let current = message;\n  while (current.next) current = current.next;\n  return \"current\" in current ? current : null;\n};\n\nclass CachedValue<T> {\n  private _value: T | null = null;\n\n  private func: () => T;\n\n  constructor(func: () => T) {\n    this.func = func;\n  }\n\n  get value() {\n    if (this._value === null) {\n      this._value = this.func();\n    }\n    return this._value;\n  }\n\n  dirty() {\n    this._value = null;\n  }\n}\n\nexport class MessageRepository {\n  private messages = new Map<string, RepositoryMessage>();\n  private head: RepositoryMessage | null = null;\n  private root: RepositoryParent = {\n    children: [],\n    next: null,\n  };\n\n  private updateLevels(message: RepositoryMessage, newLevel: number) {\n    const pending = [{ message, level: newLevel }];\n\n    while (pending.length > 0) {\n      const current = pending.pop()!;\n      current.message.level = current.level;\n\n      for (const childId of current.message.children) {\n        const childMessage = this.messages.get(childId);\n        if (childMessage) {\n          pending.push({ message: childMessage, level: current.level + 1 });\n        }\n      }\n    }\n  }\n\n  private performOp(\n    newParent: RepositoryMessage | null,\n    child: RepositoryMessage,\n    operation: \"cut\" | \"link\" | \"relink\",\n  ) {\n    const parentOrRoot = child.prev ?? this.root;\n    const newParentOrRoot = newParent ?? this.root;\n\n    if (operation === \"relink\" && parentOrRoot === newParentOrRoot) return;\n\n    // `link` receives a fresh ID from `addOrUpdateMessage`; only `relink` can introduce a cycle.\n    if (operation === \"relink\") {\n      for (\n        let current: RepositoryMessage | null = newParent;\n        current;\n        current = current.prev\n      ) {\n        if (current.current.id === child.current.id) {\n          throw new Error(\n            \"MessageRepository(performOp/relink): A message with the same id already exists in the parent tree. This error occurs if the same message id is found multiple times. This is likely an internal bug in assistant-ui.\",\n          );\n        }\n      }\n    }\n\n    if (operation !== \"link\") {\n      parentOrRoot.children = parentOrRoot.children.filter(\n        (m) => m !== child.current.id,\n      );\n\n      if (parentOrRoot.next === child) {\n        const fallbackId = parentOrRoot.children.at(-1);\n        const fallback = fallbackId ? this.messages.get(fallbackId) : null;\n        if (fallback === undefined) {\n          throw new Error(\n            \"MessageRepository(performOp/cut): Fallback sibling message not found. This is likely an internal bug in assistant-ui.\",\n          );\n        }\n        parentOrRoot.next = fallback;\n      }\n    }\n\n    if (operation !== \"cut\") {\n      newParentOrRoot.children = [\n        ...newParentOrRoot.children,\n        child.current.id,\n      ];\n\n      if (findHead(child) === this.head || newParentOrRoot.next === null) {\n        newParentOrRoot.next = child;\n      }\n\n      child.prev = newParent;\n\n      const newLevel = newParent ? newParent.level + 1 : 0;\n      this.updateLevels(child, newLevel);\n    }\n  }\n\n  private _messages = new CachedValue<readonly ThreadMessage[]>(() => {\n    const messages = new Array<ThreadMessage>((this.head?.level ?? -1) + 1);\n    for (let current = this.head; current; current = current.prev) {\n      messages[current.level] = current.current;\n    }\n    return messages;\n  });\n\n  get headId() {\n    return this.head?.current.id ?? null;\n  }\n\n  get canonicalHeadId() {\n    // Optimistic messages are ephemeral, so persisted callers need the nearest\n    // non-optimistic ancestor rather than the raw head.\n    let head = this.head;\n    while (head?.current.metadata?.isOptimistic) {\n      head = head.prev;\n    }\n    return head?.current.id ?? null;\n  }\n\n  getMessages(headId?: string) {\n    if (headId === undefined || headId === this.head?.current.id) {\n      return this._messages.value;\n    }\n\n    const headMessage = this.messages.get(headId);\n    if (!headMessage) {\n      throw new Error(\n        \"MessageRepository(getMessages): Head message not found. This is likely an internal bug in assistant-ui.\",\n      );\n    }\n\n    const messages = new Array<ThreadMessage>(headMessage.level + 1);\n    for (\n      let current: RepositoryMessage | null = headMessage;\n      current;\n      current = current.prev\n    ) {\n      messages[current.level] = current.current;\n    }\n    return messages;\n  }\n\n  addOrUpdateMessage(parentId: string | null, message: ThreadMessage) {\n    const existingItem = this.messages.get(message.id);\n    const prev = parentId ? this.messages.get(parentId) : null;\n    if (prev === undefined)\n      throw new Error(\n        \"MessageRepository(addOrUpdateMessage): Parent message not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    if (existingItem) {\n      existingItem.current = message;\n      this.performOp(prev, existingItem, \"relink\");\n      this._messages.dirty();\n      return;\n    }\n\n    const newItem: RepositoryMessage = {\n      prev,\n      current: message,\n      next: null,\n      children: [],\n      level: prev ? prev.level + 1 : 0,\n    };\n\n    this.messages.set(message.id, newItem);\n    this.performOp(prev, newItem, \"link\");\n\n    if (this.head === prev) {\n      this.head = newItem;\n    }\n\n    this._messages.dirty();\n  }\n\n  getMessage(messageId: string) {\n    const message = this.messages.get(messageId);\n    if (!message)\n      throw new Error(\n        \"MessageRepository(updateMessage): Message not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    return {\n      parentId: message.prev?.current.id ?? null,\n      message: message.current,\n      index: message.level,\n    };\n  }\n\n  deleteMessage(messageId: string, replacementId?: string | null | undefined) {\n    const message = this.messages.get(messageId);\n\n    if (!message)\n      throw new Error(\n        \"MessageRepository(deleteMessage): Message not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    const replacement =\n      replacementId === undefined\n        ? message.prev\n        : replacementId === null\n          ? null\n          : this.messages.get(replacementId);\n    if (replacement === undefined)\n      throw new Error(\n        \"MessageRepository(deleteMessage): Replacement not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    for (const child of message.children) {\n      const childMessage = this.messages.get(child);\n      if (!childMessage)\n        throw new Error(\n          \"MessageRepository(deleteMessage): Child message not found. This is likely an internal bug in assistant-ui.\",\n        );\n      this.performOp(replacement, childMessage, \"relink\");\n    }\n\n    this.performOp(null, message, \"cut\");\n    this.messages.delete(messageId);\n\n    if (this.head === message) {\n      this.head = findHead(replacement ?? this.root);\n    }\n\n    this._messages.dirty();\n  }\n\n  getBranches(messageId: string) {\n    const message = this.messages.get(messageId);\n    if (!message)\n      throw new Error(\n        \"MessageRepository(getBranches): Message not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    const { children } = message.prev ?? this.root;\n    return children;\n  }\n\n  /**\n   * Evicts optimistic messages (`metadata.isOptimistic`) the head just moved\n   * away from. Since eviction runs on every head move, the only optimistic\n   * messages in the repository live on the branch the head previously pointed\n   * at — so we walk just that branch rather than the whole repository. Keeps a\n   * client→server id swap from leaving a phantom sibling, and drops off-branch\n   * placeholders.\n   */\n  private evictOffBranchOptimisticMessages(\n    previousHead: RepositoryMessage | null,\n    currentHead: RepositoryMessage | null,\n  ) {\n    if (!previousHead) return;\n\n    const onHeadBranch = new Set<string>();\n    for (let current = currentHead; current; current = current.prev) {\n      onHeadBranch.add(current.current.id);\n    }\n\n    const stale: string[] = [];\n    for (\n      let current: RepositoryMessage | null = previousHead;\n      current;\n      current = current.prev\n    ) {\n      // Stop at the first node shared with the current head branch: every\n      // ancestor above it is shared too, so nothing further can be off-branch.\n      if (onHeadBranch.has(current.current.id)) break;\n      if (current.current.metadata?.isOptimistic) {\n        stale.push(current.current.id);\n      }\n    }\n\n    for (const id of stale) {\n      // A prior deletion may have already removed this node.\n      if (this.messages.has(id)) this.deleteMessage(id);\n    }\n  }\n\n  switchToBranch(messageId: string) {\n    const message = this.messages.get(messageId);\n    if (!message)\n      throw new Error(\n        \"MessageRepository(switchToBranch): Branch not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    const previousHead = this.head;\n    const prevOrRoot = message.prev ?? this.root;\n    prevOrRoot.next = message;\n\n    this.head = findHead(message);\n\n    this.evictOffBranchOptimisticMessages(previousHead, this.head);\n\n    this._messages.dirty();\n  }\n\n  resetHead(messageId: string | null) {\n    if (messageId === null) {\n      this.clear();\n      return;\n    }\n\n    const message = this.messages.get(messageId);\n    if (!message)\n      throw new Error(\n        \"MessageRepository(resetHead): Branch not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    const previousHead = this.head;\n\n    if (message.children.length > 0) {\n      const pending = [...message.children];\n      while (pending.length > 0) {\n        const childId = pending.pop()!;\n        const childMessage = this.messages.get(childId);\n        if (childMessage) {\n          for (const descendantId of childMessage.children) {\n            pending.push(descendantId);\n          }\n          this.messages.delete(childId);\n        }\n      }\n\n      message.children = [];\n      message.next = null;\n    }\n\n    this.head = message;\n    for (\n      let current: RepositoryMessage | null = message;\n      current;\n      current = current.prev\n    ) {\n      if (current.prev) {\n        current.prev.next = current;\n      } else {\n        this.root.next = current;\n      }\n    }\n\n    this.evictOffBranchOptimisticMessages(previousHead, this.head);\n\n    this._messages.dirty();\n  }\n\n  clear(): void {\n    this.messages.clear();\n    this.head = null;\n    this.root = {\n      children: [],\n      next: null,\n    };\n    this._messages.dirty();\n  }\n\n  export(): ExportedMessageRepository {\n    const exportItems: ExportedMessageRepository[\"messages\"] = [];\n\n    // Optimistic messages are ephemeral and never persisted. A persisted child\n    // of an optimistic node is re-parented onto its nearest persisted ancestor\n    // so the exported tree never references a skipped id.\n    for (const [, message] of this.messages) {\n      if (message.current.metadata?.isOptimistic) continue;\n      let prev = message.prev;\n      while (prev && prev.current.metadata?.isOptimistic) {\n        prev = prev.prev;\n      }\n      exportItems.push({\n        message: message.current,\n        parentId: prev?.current.id ?? null,\n      });\n    }\n\n    return {\n      headId: this.canonicalHeadId,\n      messages: exportItems,\n    };\n  }\n\n  import({ headId, messages }: ExportedMessageRepository) {\n    for (const { message, parentId } of messages) {\n      this.addOrUpdateMessage(parentId, message);\n    }\n\n    this.resetHead(headId ?? messages.at(-1)?.message.id ?? null);\n  }\n}\n"],"mappings":";;;;AAsBA,MAAaU,4BAA4B;CACvCI,YACEF,aAC8B;EAC9B,MAAMG,OAAOH,SAASI,KAAKC,MACzBZ,sBACEY,GACAf,WAAW,GACXE,cAAc,OAAO,OAAO,OAAO,OAAOc,KAAAA,CAAS,CACrD,CACF;EAEA,OAAO,EACLN,UAAUG,KAAKC,KAAKC,GAAGE,SAAS;GAC9BX,UAAUW,MAAM,IAAIJ,KAAKI,MAAM,EAAE,CAAEC,KAAK;GACxCb,SAASU;EACX,EAAE,EACJ;CACF;CAEAI,sBACEC,OAIAC,YAC8B;EAC9B,MAAMC,iBAAiBpB,cAAc,OAAO,OAAO,OAAO,OAAOc,KAAAA,CAAS;EAC1E,OAAO;GACL,GAAIK,SAASZ,WAAWO,KAAAA,IACpB,EAAEP,QAAQY,QAAQZ,OAAO,IACzBO,KAAAA;GACJN,UAAUU,MAAMN,KAAK,EAAET,SAASC,eAAe;IAC7C,IAAI,CAACD,QAAQa,IACX,MAAM,IAAIK,MACR,0FACF;IAEF,OAAO;KACLjB;KACAD,SAASF,sBAAsBE,SAASA,QAAQa,IAAII,cAAc;IACpE;GACF,CAAC;EACH;CACF;AACF;AAaA,MAAMS,YACJ1B,YAC6B;CAC7B,IAAIwB,UAAUxB;CACd,OAAOwB,QAAQH,MAAMG,UAAUA,QAAQH;CACvC,OAAO,aAAaG,UAAUA,UAAU;AAC1C;AAEA,IAAMG,cAAN,MAAqB;CACnB,SAA2B;CAE3B;CAEAI,YAAYD,MAAe;EACzB,KAAKA,OAAOA;CACd;CAEA,IAAIE,QAAQ;EACV,IAAI,KAAKJ,WAAW,MAClB,KAAKA,SAAS,KAAKE,KAAK;EAE1B,OAAO,KAAKF;CACd;CAEAK,QAAQ;EACN,KAAKL,SAAS;CAChB;AACF;AAEA,IAAaM,oBAAb,MAA+B;CAC7B,2BAAmB,IAAIC,IAA+B;CACtD,OAAyC;CACzC,OAAiC;EAC/Bf,UAAU,CAAA;EACVC,MAAM;CACR;CAEA,aAAqBrB,SAA4BuC,UAAkB;EACjE,MAAMC,UAAU,CAAC;GAAExC;GAASyB,OAAOc;EAAS,CAAC;EAE7C,OAAOC,QAAQC,SAAS,GAAG;GACzB,MAAMjB,UAAUgB,QAAQE,IAAI;GAC5BlB,QAAQxB,QAAQyB,QAAQD,QAAQC;GAEhC,KAAK,MAAMkB,WAAWnB,QAAQxB,QAAQoB,UAAU;IAC9C,MAAMwB,eAAe,KAAKvC,SAASwC,IAAIF,OAAO;IAC9C,IAAIC,cACFJ,QAAQM,KAAK;KAAE9C,SAAS4C;KAAcnB,OAAOD,QAAQC,QAAQ;IAAE,CAAC;GAEpE;EACF;CACF;CAEA,UACEuB,WACAC,OACAC,WACA;EACA,MAAMC,eAAeF,MAAM1B,QAAQ,KAAKc;EACxC,MAAMe,kBAAkBJ,aAAa,KAAKX;EAE1C,IAAIa,cAAc,YAAYC,iBAAiBC,iBAAiB;EAGhE,IAAIF,cAAc,UAEd;QAAA,IAAI1B,UAAoCwB,WACxCxB,SACAA,UAAUA,QAAQD,MAElB,IAAIC,QAAQA,QAAQX,OAAOoC,MAAMzB,QAAQX,IACvC,MAAM,IAAIK,MACR,sNACF;EAAA;EAKN,IAAIgC,cAAc,QAAQ;GACxBC,aAAa/B,WAAW+B,aAAa/B,SAASiC,QAC3C3C,MAAMA,MAAMuC,MAAMzB,QAAQX,EAC7B;GAEA,IAAIsC,aAAa9B,SAAS4B,OAAO;IAC/B,MAAMK,aAAaH,aAAa/B,SAASmC,GAAG,EAAE;IAC9C,MAAMC,WAAWF,aAAa,KAAKjD,SAASwC,IAAIS,UAAU,IAAI;IAC9D,IAAIE,aAAa7C,KAAAA,GACf,MAAM,IAAIO,MACR,uHACF;IAEFiC,aAAa9B,OAAOmC;GACtB;EACF;EAEA,IAAIN,cAAc,OAAO;GACvBE,gBAAgBhC,WAAW,CACzB,GAAGgC,gBAAgBhC,UACnB6B,MAAMzB,QAAQX,EAAE;GAGlB,IAAIa,SAASuB,KAAK,MAAM,KAAKb,QAAQgB,gBAAgB/B,SAAS,MAC5D+B,gBAAgB/B,OAAO4B;GAGzBA,MAAM1B,OAAOyB;GAEb,MAAMT,WAAWS,YAAYA,UAAUvB,QAAQ,IAAI;GACnD,KAAKa,aAAaW,OAAOV,QAAQ;EACnC;CACF;CAEA,YAAoB,IAAIZ,kBAA4C;EAClE,MAAMtB,WAAW,IAAIC,OAAsB,KAAK8B,MAAMX,SAAS,MAAM,CAAC;EACtE,KAAK,IAAID,UAAU,KAAKY,MAAMZ,SAASA,UAAUA,QAAQD,MACvDlB,SAASmB,QAAQC,SAASD,QAAQA;EAEpC,OAAOnB;CACT,CAAC;CAED,IAAID,SAAS;EACX,OAAO,KAAKgC,MAAMZ,QAAQX,MAAM;CAClC;CAEA,IAAI6C,kBAAkB;EAGpB,IAAItB,OAAO,KAAKA;EAChB,OAAOA,MAAMZ,QAAQmC,UAAUC,cAC7BxB,OAAOA,KAAKb;EAEd,OAAOa,MAAMZ,QAAQX,MAAM;CAC7B;CAEAgD,YAAYzD,QAAiB;EAC3B,IAAIA,WAAWO,KAAAA,KAAaP,WAAW,KAAKgC,MAAMZ,QAAQX,IACxD,OAAO,KAAK4C,UAAUzB;EAGxB,MAAM8B,cAAc,KAAKzD,SAASwC,IAAIzC,MAAM;EAC5C,IAAI,CAAC0D,aACH,MAAM,IAAI5C,MACR,yGACF;EAGF,MAAMb,WAAW,IAAIC,MAAqBwD,YAAYrC,QAAQ,CAAC;EAC/D,KACE,IAAID,UAAoCsC,aACxCtC,SACAA,UAAUA,QAAQD,MAElBlB,SAASmB,QAAQC,SAASD,QAAQA;EAEpC,OAAOnB;CACT;CAEA0D,mBAAmB9D,UAAyBD,SAAwB;EAClE,MAAMgE,eAAe,KAAK3D,SAASwC,IAAI7C,QAAQa,EAAE;EACjD,MAAMU,OAAOtB,WAAW,KAAKI,SAASwC,IAAI5C,QAAQ,IAAI;EACtD,IAAIsB,SAASZ,KAAAA,GACX,MAAM,IAAIO,MACR,kHACF;EAEF,IAAI8C,cAAc;GAChBA,aAAaxC,UAAUxB;GACvB,KAAK+C,UAAUxB,MAAMyC,cAAc,QAAQ;GAC3C,KAAKP,UAAUxB,MAAM;GACrB;EACF;EAEA,MAAMgC,UAA6B;GACjC1C;GACAC,SAASxB;GACTqB,MAAM;GACND,UAAU,CAAA;GACVK,OAAOF,OAAOA,KAAKE,QAAQ,IAAI;EACjC;EAEA,KAAKpB,SAAS6D,IAAIlE,QAAQa,IAAIoD,OAAO;EACrC,KAAKlB,UAAUxB,MAAM0C,SAAS,MAAM;EAEpC,IAAI,KAAK7B,SAASb,MAChB,KAAKa,OAAO6B;EAGd,KAAKR,UAAUxB,MAAM;CACvB;CAEAkC,WAAWC,WAAmB;EAC5B,MAAMpE,UAAU,KAAKK,SAASwC,IAAIuB,SAAS;EAC3C,IAAI,CAACpE,SACH,MAAM,IAAIkB,MACR,sGACF;EAEF,OAAO;GACLjB,UAAUD,QAAQuB,MAAMC,QAAQX,MAAM;GACtCb,SAASA,QAAQwB;GACjB6C,OAAOrE,QAAQyB;EACjB;CACF;CAEA6C,cAAcF,WAAmBG,eAA2C;EAC1E,MAAMvE,UAAU,KAAKK,SAASwC,IAAIuB,SAAS;EAE3C,IAAI,CAACpE,SACH,MAAM,IAAIkB,MACR,sGACF;EAEF,MAAMsD,cACJD,kBAAkB5D,KAAAA,IACdX,QAAQuB,OACRgD,kBAAkB,OAChB,OACA,KAAKlE,SAASwC,IAAI0B,aAAa;EACvC,IAAIC,gBAAgB7D,KAAAA,GAClB,MAAM,IAAIO,MACR,0GACF;EAEF,KAAK,MAAM+B,SAASjD,QAAQoB,UAAU;GACpC,MAAMwB,eAAe,KAAKvC,SAASwC,IAAII,KAAK;GAC5C,IAAI,CAACL,cACH,MAAM,IAAI1B,MACR,4GACF;GACF,KAAK6B,UAAUyB,aAAa5B,cAAc,QAAQ;EACpD;EAEA,KAAKG,UAAU,MAAM/C,SAAS,KAAK;EACnC,KAAKK,SAASoE,OAAOL,SAAS;EAE9B,IAAI,KAAKhC,SAASpC,SAChB,KAAKoC,OAAOV,SAAS8C,eAAe,KAAKnC,IAAI;EAG/C,KAAKoB,UAAUxB,MAAM;CACvB;CAEAyC,YAAYN,WAAmB;EAC7B,MAAMpE,UAAU,KAAKK,SAASwC,IAAIuB,SAAS;EAC3C,IAAI,CAACpE,SACH,MAAM,IAAIkB,MACR,oGACF;EAEF,MAAM,EAAEE,aAAapB,QAAQuB,QAAQ,KAAKc;EAC1C,OAAOjB;CACT;;;;;;;;;CAUA,iCACEwD,cACAC,aACA;EACA,IAAI,CAACD,cAAc;EAEnB,MAAME,+BAAe,IAAIC,IAAY;EACrC,KAAK,IAAIvD,UAAUqD,aAAarD,SAASA,UAAUA,QAAQD,MACzDuD,aAAaE,IAAIxD,QAAQA,QAAQX,EAAE;EAGrC,MAAMoE,QAAkB,CAAA;EACxB,KACE,IAAIzD,UAAoCoD,cACxCpD,SACAA,UAAUA,QAAQD,MAClB;GAGA,IAAIuD,aAAaI,IAAI1D,QAAQA,QAAQX,EAAE,GAAG;GAC1C,IAAIW,QAAQA,QAAQmC,UAAUC,cAC5BqB,MAAMnC,KAAKtB,QAAQA,QAAQX,EAAE;EAEjC;EAEA,KAAK,MAAMA,MAAMoE,OAEf,IAAI,KAAK5E,SAAS6E,IAAIrE,EAAE,GAAG,KAAKyD,cAAczD,EAAE;CAEpD;CAEAsE,eAAef,WAAmB;EAChC,MAAMpE,UAAU,KAAKK,SAASwC,IAAIuB,SAAS;EAC3C,IAAI,CAACpE,SACH,MAAM,IAAIkB,MACR,sGACF;EAEF,MAAM0D,eAAe,KAAKxC;EAC1B,MAAMgD,aAAapF,QAAQuB,QAAQ,KAAKc;EACxC+C,WAAW/D,OAAOrB;EAElB,KAAKoC,OAAOV,SAAS1B,OAAO;EAE5B,KAAK2E,iCAAiCC,cAAc,KAAKxC,IAAI;EAE7D,KAAKqB,UAAUxB,MAAM;CACvB;CAEAoD,UAAUjB,WAA0B;EAClC,IAAIA,cAAc,MAAM;GACtB,KAAKkB,MAAM;GACX;EACF;EAEA,MAAMtF,UAAU,KAAKK,SAASwC,IAAIuB,SAAS;EAC3C,IAAI,CAACpE,SACH,MAAM,IAAIkB,MACR,iGACF;EAEF,MAAM0D,eAAe,KAAKxC;EAE1B,IAAIpC,QAAQoB,SAASqB,SAAS,GAAG;GAC/B,MAAMD,UAAU,CAAC,GAAGxC,QAAQoB,QAAQ;GACpC,OAAOoB,QAAQC,SAAS,GAAG;IACzB,MAAME,UAAUH,QAAQE,IAAI;IAC5B,MAAME,eAAe,KAAKvC,SAASwC,IAAIF,OAAO;IAC9C,IAAIC,cAAc;KAChB,KAAK,MAAM2C,gBAAgB3C,aAAaxB,UACtCoB,QAAQM,KAAKyC,YAAY;KAE3B,KAAKlF,SAASoE,OAAO9B,OAAO;IAC9B;GACF;GAEA3C,QAAQoB,WAAW,CAAA;GACnBpB,QAAQqB,OAAO;EACjB;EAEA,KAAKe,OAAOpC;EACZ,KACE,IAAIwB,UAAoCxB,SACxCwB,SACAA,UAAUA,QAAQD,MAElB,IAAIC,QAAQD,MACVC,QAAQD,KAAKF,OAAOG;OAEpB,KAAKa,KAAKhB,OAAOG;EAIrB,KAAKmD,iCAAiCC,cAAc,KAAKxC,IAAI;EAE7D,KAAKqB,UAAUxB,MAAM;CACvB;CAEAqD,QAAc;EACZ,KAAKjF,SAASiF,MAAM;EACpB,KAAKlD,OAAO;EACZ,KAAKC,OAAO;GACVjB,UAAU,CAAA;GACVC,MAAM;EACR;EACA,KAAKoC,UAAUxB,MAAM;CACvB;CAEAuD,SAAoC;EAClC,MAAMC,cAAqD,CAAA;EAK3D,KAAK,MAAM,GAAGzF,YAAY,KAAKK,UAAU;GACvC,IAAIL,QAAQwB,QAAQmC,UAAUC,cAAc;GAC5C,IAAIrC,OAAOvB,QAAQuB;GACnB,OAAOA,QAAQA,KAAKC,QAAQmC,UAAUC,cACpCrC,OAAOA,KAAKA;GAEdkE,YAAY3C,KAAK;IACf9C,SAASA,QAAQwB;IACjBvB,UAAUsB,MAAMC,QAAQX,MAAM;GAChC,CAAC;EACH;EAEA,OAAO;GACLT,QAAQ,KAAKsD;GACbrD,UAAUoF;EACZ;CACF;CAEAC,OAAO,EAAEtF,QAAQC,YAAuC;EACtD,KAAK,MAAM,EAAEL,SAASC,cAAcI,UAClC,KAAK0D,mBAAmB9D,UAAUD,OAAO;EAG3C,KAAKqF,UAAUjF,UAAUC,SAASkD,GAAG,EAAE,CAAC,EAAEvD,QAAQa,MAAM,IAAI;CAC9D;AACF"}
+{"version":3,"file":"message-repository.js","names":["ThreadMessage","RunConfig","generateId","ThreadMessageLike","getContentAutoStatus","fromThreadMessageLike","ExportedMessageRepositoryItem","message","parentId","runConfig","ExportedMessageRepository","headId","messages","Array","fromArray","conv","map","m","content","idx","id","fromBranchableArray","items","options","undefined","Error","RepositoryParent","children","next","RepositoryMessage","prev","current","level","findHead","CachedValue","_value","T","func","constructor","value","dirty","MessageRepository","Map","head","root","updateLevels","newLevel","pending","length","pop","childId","childMessage","get","push","performOp","newParent","child","operation","parentOrRoot","newParentOrRoot","filter","fallbackId","at","fallback","_messages","canonicalHeadId","metadata","isOptimistic","getMessages","headMessage","addOrUpdateMessage","existingItem","newItem","set","getMessage","messageId","index","deleteMessage","replacementId","replacement","delete","getBranches","evictOffBranchOptimisticMessages","previousHead","currentHead","onHeadBranch","Set","add","stale","has","switchToBranch","prevOrRoot","resetHead","clear","descendantId","export","exportItems","import"],"sources":["../../../src/runtime/utils/message-repository.ts"],"sourcesContent":["import type { ThreadMessage } from \"../../types/message\";\nimport type { RunConfig } from \"../../types/message\";\nimport { generateId } from \"../../utils/id\";\nimport type { ThreadMessageLike } from \"./thread-message-like\";\nimport { getContentAutoStatus } from \"./auto-status\";\nimport { fromThreadMessageLike } from \"./thread-message-like\";\n\nexport type ExportedMessageRepositoryItem = {\n  message: ThreadMessage;\n  parentId: string | null;\n  runConfig?: RunConfig;\n};\n\nexport type ExportedMessageRepository = {\n  headId?: string | null;\n  messages: Array<{\n    message: ThreadMessage;\n    parentId: string | null;\n    runConfig?: RunConfig;\n  }>;\n};\n\nexport const ExportedMessageRepository = {\n  fromArray: (\n    messages: readonly ThreadMessageLike[],\n  ): ExportedMessageRepository => {\n    const conv = messages.map((m) =>\n      fromThreadMessageLike(\n        m,\n        generateId(),\n        getContentAutoStatus(m.content, false, false),\n      ),\n    );\n\n    return {\n      messages: conv.map((m, idx) => ({\n        parentId: idx > 0 ? conv[idx - 1]!.id : null,\n        message: m,\n      })),\n    };\n  },\n\n  fromBranchableArray: (\n    items: readonly {\n      message: ThreadMessageLike;\n      parentId: string | null;\n    }[],\n    options?: { headId?: string | null },\n  ): ExportedMessageRepository => {\n    return {\n      ...(options?.headId !== undefined\n        ? { headId: options.headId }\n        : undefined),\n      messages: items.map(({ message, parentId }) => {\n        if (!message.id) {\n          throw new Error(\n            \"ExportedMessageRepository.fromBranchableArray: Each message must have an 'id' field set.\",\n          );\n        }\n        return {\n          parentId,\n          message: fromThreadMessageLike(\n            message,\n            message.id,\n            getContentAutoStatus(message.content, false, false),\n          ),\n        };\n      }),\n    };\n  },\n};\n\ntype RepositoryParent = {\n  children: string[];\n  next: RepositoryMessage | null;\n};\n\ntype RepositoryMessage = RepositoryParent & {\n  prev: RepositoryMessage | null;\n  current: ThreadMessage;\n  level: number;\n};\n\nconst findHead = (\n  message: RepositoryMessage | RepositoryParent,\n): RepositoryMessage | null => {\n  let current = message;\n  while (current.next) current = current.next;\n  return \"current\" in current ? current : null;\n};\n\nclass CachedValue<T> {\n  private _value: T | null = null;\n\n  private func: () => T;\n\n  constructor(func: () => T) {\n    this.func = func;\n  }\n\n  get value() {\n    if (this._value === null) {\n      this._value = this.func();\n    }\n    return this._value;\n  }\n\n  dirty() {\n    this._value = null;\n  }\n}\n\nexport class MessageRepository {\n  private messages = new Map<string, RepositoryMessage>();\n  private head: RepositoryMessage | null = null;\n  private root: RepositoryParent = {\n    children: [],\n    next: null,\n  };\n\n  private updateLevels(message: RepositoryMessage, newLevel: number) {\n    const pending = [{ message, level: newLevel }];\n\n    while (pending.length > 0) {\n      const current = pending.pop()!;\n      current.message.level = current.level;\n\n      for (const childId of current.message.children) {\n        const childMessage = this.messages.get(childId);\n        if (childMessage) {\n          pending.push({ message: childMessage, level: current.level + 1 });\n        }\n      }\n    }\n  }\n\n  private performOp(\n    newParent: RepositoryMessage | null,\n    child: RepositoryMessage,\n    operation: \"cut\" | \"link\" | \"relink\",\n  ) {\n    const parentOrRoot = child.prev ?? this.root;\n    const newParentOrRoot = newParent ?? this.root;\n\n    if (operation === \"relink\" && parentOrRoot === newParentOrRoot) return;\n\n    // `link` receives a fresh ID from `addOrUpdateMessage`; only `relink` can introduce a cycle.\n    if (operation === \"relink\") {\n      for (\n        let current: RepositoryMessage | null = newParent;\n        current;\n        current = current.prev\n      ) {\n        if (current.current.id === child.current.id) {\n          throw new Error(\n            \"MessageRepository(performOp/relink): A message with the same id already exists in the parent tree. This error occurs if the same message id is found multiple times. This is likely an internal bug in assistant-ui.\",\n          );\n        }\n      }\n    }\n\n    if (operation !== \"link\") {\n      parentOrRoot.children = parentOrRoot.children.filter(\n        (m) => m !== child.current.id,\n      );\n\n      if (parentOrRoot.next === child) {\n        const fallbackId = parentOrRoot.children.at(-1);\n        const fallback = fallbackId ? this.messages.get(fallbackId) : null;\n        if (fallback === undefined) {\n          throw new Error(\n            \"MessageRepository(performOp/cut): Fallback sibling message not found. This is likely an internal bug in assistant-ui.\",\n          );\n        }\n        parentOrRoot.next = fallback;\n      }\n    }\n\n    if (operation !== \"cut\") {\n      newParentOrRoot.children = [\n        ...newParentOrRoot.children,\n        child.current.id,\n      ];\n\n      if (findHead(child) === this.head || newParentOrRoot.next === null) {\n        newParentOrRoot.next = child;\n      }\n\n      child.prev = newParent;\n\n      const newLevel = newParent ? newParent.level + 1 : 0;\n      this.updateLevels(child, newLevel);\n    }\n  }\n\n  private _messages = new CachedValue<readonly ThreadMessage[]>(() => {\n    const messages = new Array<ThreadMessage>((this.head?.level ?? -1) + 1);\n    for (let current = this.head; current; current = current.prev) {\n      messages[current.level] = current.current;\n    }\n    return messages;\n  });\n\n  get headId() {\n    return this.head?.current.id ?? null;\n  }\n\n  get canonicalHeadId() {\n    // Optimistic messages are ephemeral, so persisted callers need the nearest\n    // non-optimistic ancestor rather than the raw head.\n    let head = this.head;\n    while (head?.current.metadata?.isOptimistic) {\n      head = head.prev;\n    }\n    return head?.current.id ?? null;\n  }\n\n  getMessages(headId?: string) {\n    if (headId === undefined || headId === this.head?.current.id) {\n      return this._messages.value;\n    }\n\n    const headMessage = this.messages.get(headId);\n    if (!headMessage) {\n      throw new Error(\n        \"MessageRepository(getMessages): Head message not found. This is likely an internal bug in assistant-ui.\",\n      );\n    }\n\n    const messages = new Array<ThreadMessage>(headMessage.level + 1);\n    for (\n      let current: RepositoryMessage | null = headMessage;\n      current;\n      current = current.prev\n    ) {\n      messages[current.level] = current.current;\n    }\n    return messages;\n  }\n\n  addOrUpdateMessage(parentId: string | null, message: ThreadMessage) {\n    const existingItem = this.messages.get(message.id);\n    const prev = parentId ? this.messages.get(parentId) : null;\n    if (prev === undefined)\n      throw new Error(\n        \"MessageRepository(addOrUpdateMessage): Parent message not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    if (existingItem) {\n      existingItem.current = message;\n      this.performOp(prev, existingItem, \"relink\");\n      this._messages.dirty();\n      return;\n    }\n\n    const newItem: RepositoryMessage = {\n      prev,\n      current: message,\n      next: null,\n      children: [],\n      level: prev ? prev.level + 1 : 0,\n    };\n\n    this.messages.set(message.id, newItem);\n    this.performOp(prev, newItem, \"link\");\n\n    if (this.head === prev) {\n      this.head = newItem;\n    }\n\n    this._messages.dirty();\n  }\n\n  getMessage(messageId: string) {\n    const message = this.messages.get(messageId);\n    if (!message)\n      throw new Error(\n        \"MessageRepository(updateMessage): Message not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    return {\n      parentId: message.prev?.current.id ?? null,\n      message: message.current,\n      index: message.level,\n    };\n  }\n\n  deleteMessage(messageId: string, replacementId?: string | null | undefined) {\n    const message = this.messages.get(messageId);\n\n    if (!message)\n      throw new Error(\n        \"MessageRepository(deleteMessage): Message not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    const replacement =\n      replacementId === undefined\n        ? message.prev\n        : replacementId === null\n          ? null\n          : this.messages.get(replacementId);\n    if (replacement === undefined)\n      throw new Error(\n        \"MessageRepository(deleteMessage): Replacement not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    for (const child of message.children) {\n      const childMessage = this.messages.get(child);\n      if (!childMessage)\n        throw new Error(\n          \"MessageRepository(deleteMessage): Child message not found. This is likely an internal bug in assistant-ui.\",\n        );\n      this.performOp(replacement, childMessage, \"relink\");\n    }\n\n    this.performOp(null, message, \"cut\");\n    this.messages.delete(messageId);\n\n    if (this.head === message) {\n      this.head = findHead(replacement ?? this.root);\n    }\n\n    this._messages.dirty();\n  }\n\n  getBranches(messageId: string) {\n    const message = this.messages.get(messageId);\n    if (!message)\n      throw new Error(\n        \"MessageRepository(getBranches): Message not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    const { children } = message.prev ?? this.root;\n    return children;\n  }\n\n  /**\n   * Evicts optimistic messages (`metadata.isOptimistic`) the head just moved\n   * away from. Since eviction runs on every head move, the only optimistic\n   * messages in the repository live on the branch the head previously pointed\n   * at — so we walk just that branch rather than the whole repository. Keeps a\n   * client→server id swap from leaving a phantom sibling, and drops off-branch\n   * placeholders.\n   */\n  private evictOffBranchOptimisticMessages(\n    previousHead: RepositoryMessage | null,\n    currentHead: RepositoryMessage | null,\n  ) {\n    if (!previousHead) return;\n\n    const onHeadBranch = new Set<string>();\n    for (let current = currentHead; current; current = current.prev) {\n      onHeadBranch.add(current.current.id);\n    }\n\n    const stale: string[] = [];\n    for (\n      let current: RepositoryMessage | null = previousHead;\n      current;\n      current = current.prev\n    ) {\n      // Stop at the first node shared with the current head branch: every\n      // ancestor above it is shared too, so nothing further can be off-branch.\n      if (onHeadBranch.has(current.current.id)) break;\n      if (current.current.metadata?.isOptimistic) {\n        stale.push(current.current.id);\n      }\n    }\n\n    for (const id of stale) {\n      // A prior deletion may have already removed this node.\n      if (this.messages.has(id)) this.deleteMessage(id);\n    }\n  }\n\n  switchToBranch(messageId: string) {\n    const message = this.messages.get(messageId);\n    if (!message)\n      throw new Error(\n        \"MessageRepository(switchToBranch): Branch not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    const previousHead = this.head;\n    const prevOrRoot = message.prev ?? this.root;\n    prevOrRoot.next = message;\n\n    this.head = findHead(message);\n\n    this.evictOffBranchOptimisticMessages(previousHead, this.head);\n\n    this._messages.dirty();\n  }\n\n  resetHead(messageId: string | null) {\n    if (messageId === null) {\n      this.clear();\n      return;\n    }\n\n    const message = this.messages.get(messageId);\n    if (!message)\n      throw new Error(\n        \"MessageRepository(resetHead): Branch not found. This is likely an internal bug in assistant-ui.\",\n      );\n\n    const previousHead = this.head;\n\n    if (message.children.length > 0) {\n      const pending = [...message.children];\n      while (pending.length > 0) {\n        const childId = pending.pop()!;\n        const childMessage = this.messages.get(childId);\n        if (childMessage) {\n          for (const descendantId of childMessage.children) {\n            pending.push(descendantId);\n          }\n          this.messages.delete(childId);\n        }\n      }\n\n      message.children = [];\n      message.next = null;\n    }\n\n    this.head = message;\n    for (\n      let current: RepositoryMessage | null = message;\n      current;\n      current = current.prev\n    ) {\n      if (current.prev) {\n        current.prev.next = current;\n      } else {\n        this.root.next = current;\n      }\n    }\n\n    this.evictOffBranchOptimisticMessages(previousHead, this.head);\n\n    this._messages.dirty();\n  }\n\n  clear(): void {\n    this.messages.clear();\n    this.head = null;\n    this.root = {\n      children: [],\n      next: null,\n    };\n    this._messages.dirty();\n  }\n\n  export(): ExportedMessageRepository {\n    const exportItems: ExportedMessageRepository[\"messages\"] = [];\n\n    // Optimistic messages are ephemeral and never persisted. A persisted child\n    // of an optimistic node is re-parented onto its nearest persisted ancestor\n    // so the exported tree never references a skipped id.\n    for (const [, message] of this.messages) {\n      if (message.current.metadata?.isOptimistic) continue;\n      let prev = message.prev;\n      while (prev && prev.current.metadata?.isOptimistic) {\n        prev = prev.prev;\n      }\n      exportItems.push({\n        message: message.current,\n        parentId: prev?.current.id ?? null,\n      });\n    }\n\n    return {\n      headId: this.canonicalHeadId,\n      messages: exportItems,\n    };\n  }\n\n  import({ headId, messages }: ExportedMessageRepository) {\n    for (const { message, parentId } of messages) {\n      this.addOrUpdateMessage(parentId, message);\n    }\n\n    this.resetHead(headId ?? messages.at(-1)?.message.id ?? null);\n  }\n}\n"],"mappings":";;;;AAsBA,MAAaU,4BAA4B;CACvCI,YACEF,aAC8B;EAC9B,MAAMG,OAAOH,SAASI,KAAKC,MACzBZ,sBACEY,GACAf,WAAW,GACXE,qBAAqBa,EAAEC,SAAS,OAAO,KAAK,CAC9C,CACF;EAEA,OAAO,EACLN,UAAUG,KAAKC,KAAKC,GAAGE,SAAS;GAC9BX,UAAUW,MAAM,IAAIJ,KAAKI,MAAM,EAAE,CAAEC,KAAK;GACxCb,SAASU;EACX,EAAE,EACJ;CACF;CAEAI,sBACEC,OAIAC,YAC8B;EAC9B,OAAO;GACL,GAAIA,SAASZ,WAAWa,KAAAA,IACpB,EAAEb,QAAQY,QAAQZ,OAAO,IACzBa,KAAAA;GACJZ,UAAUU,MAAMN,KAAK,EAAET,SAASC,eAAe;IAC7C,IAAI,CAACD,QAAQa,IACX,MAAM,IAAIK,MACR,0FACF;IAEF,OAAO;KACLjB;KACAD,SAASF,sBACPE,SACAA,QAAQa,IACRhB,qBAAqBG,QAAQW,SAAS,OAAO,KAAK,CACpD;IACF;GACF,CAAC;EACH;CACF;AACF;AAaA,MAAMe,YACJ1B,YAC6B;CAC7B,IAAIwB,UAAUxB;CACd,OAAOwB,QAAQH,MAAMG,UAAUA,QAAQH;CACvC,OAAO,aAAaG,UAAUA,UAAU;AAC1C;AAEA,IAAMG,cAAN,MAAqB;CACnB,SAA2B;CAE3B;CAEAI,YAAYD,MAAe;EACzB,KAAKA,OAAOA;CACd;CAEA,IAAIE,QAAQ;EACV,IAAI,KAAKJ,WAAW,MAClB,KAAKA,SAAS,KAAKE,KAAK;EAE1B,OAAO,KAAKF;CACd;CAEAK,QAAQ;EACN,KAAKL,SAAS;CAChB;AACF;AAEA,IAAaM,oBAAb,MAA+B;CAC7B,2BAAmB,IAAIC,IAA+B;CACtD,OAAyC;CACzC,OAAiC;EAC/Bf,UAAU,CAAA;EACVC,MAAM;CACR;CAEA,aAAqBrB,SAA4BuC,UAAkB;EACjE,MAAMC,UAAU,CAAC;GAAExC;GAASyB,OAAOc;EAAS,CAAC;EAE7C,OAAOC,QAAQC,SAAS,GAAG;GACzB,MAAMjB,UAAUgB,QAAQE,IAAI;GAC5BlB,QAAQxB,QAAQyB,QAAQD,QAAQC;GAEhC,KAAK,MAAMkB,WAAWnB,QAAQxB,QAAQoB,UAAU;IAC9C,MAAMwB,eAAe,KAAKvC,SAASwC,IAAIF,OAAO;IAC9C,IAAIC,cACFJ,QAAQM,KAAK;KAAE9C,SAAS4C;KAAcnB,OAAOD,QAAQC,QAAQ;IAAE,CAAC;GAEpE;EACF;CACF;CAEA,UACEuB,WACAC,OACAC,WACA;EACA,MAAMC,eAAeF,MAAM1B,QAAQ,KAAKc;EACxC,MAAMe,kBAAkBJ,aAAa,KAAKX;EAE1C,IAAIa,cAAc,YAAYC,iBAAiBC,iBAAiB;EAGhE,IAAIF,cAAc,UAEd;QAAA,IAAI1B,UAAoCwB,WACxCxB,SACAA,UAAUA,QAAQD,MAElB,IAAIC,QAAQA,QAAQX,OAAOoC,MAAMzB,QAAQX,IACvC,MAAM,IAAIK,MACR,sNACF;EAAA;EAKN,IAAIgC,cAAc,QAAQ;GACxBC,aAAa/B,WAAW+B,aAAa/B,SAASiC,QAC3C3C,MAAMA,MAAMuC,MAAMzB,QAAQX,EAC7B;GAEA,IAAIsC,aAAa9B,SAAS4B,OAAO;IAC/B,MAAMK,aAAaH,aAAa/B,SAASmC,GAAG,EAAE;IAC9C,MAAMC,WAAWF,aAAa,KAAKjD,SAASwC,IAAIS,UAAU,IAAI;IAC9D,IAAIE,aAAavC,KAAAA,GACf,MAAM,IAAIC,MACR,uHACF;IAEFiC,aAAa9B,OAAOmC;GACtB;EACF;EAEA,IAAIN,cAAc,OAAO;GACvBE,gBAAgBhC,WAAW,CACzB,GAAGgC,gBAAgBhC,UACnB6B,MAAMzB,QAAQX,EAAE;GAGlB,IAAIa,SAASuB,KAAK,MAAM,KAAKb,QAAQgB,gBAAgB/B,SAAS,MAC5D+B,gBAAgB/B,OAAO4B;GAGzBA,MAAM1B,OAAOyB;GAEb,MAAMT,WAAWS,YAAYA,UAAUvB,QAAQ,IAAI;GACnD,KAAKa,aAAaW,OAAOV,QAAQ;EACnC;CACF;CAEA,YAAoB,IAAIZ,kBAA4C;EAClE,MAAMtB,WAAW,IAAIC,OAAsB,KAAK8B,MAAMX,SAAS,MAAM,CAAC;EACtE,KAAK,IAAID,UAAU,KAAKY,MAAMZ,SAASA,UAAUA,QAAQD,MACvDlB,SAASmB,QAAQC,SAASD,QAAQA;EAEpC,OAAOnB;CACT,CAAC;CAED,IAAID,SAAS;EACX,OAAO,KAAKgC,MAAMZ,QAAQX,MAAM;CAClC;CAEA,IAAI6C,kBAAkB;EAGpB,IAAItB,OAAO,KAAKA;EAChB,OAAOA,MAAMZ,QAAQmC,UAAUC,cAC7BxB,OAAOA,KAAKb;EAEd,OAAOa,MAAMZ,QAAQX,MAAM;CAC7B;CAEAgD,YAAYzD,QAAiB;EAC3B,IAAIA,WAAWa,KAAAA,KAAab,WAAW,KAAKgC,MAAMZ,QAAQX,IACxD,OAAO,KAAK4C,UAAUzB;EAGxB,MAAM8B,cAAc,KAAKzD,SAASwC,IAAIzC,MAAM;EAC5C,IAAI,CAAC0D,aACH,MAAM,IAAI5C,MACR,yGACF;EAGF,MAAMb,WAAW,IAAIC,MAAqBwD,YAAYrC,QAAQ,CAAC;EAC/D,KACE,IAAID,UAAoCsC,aACxCtC,SACAA,UAAUA,QAAQD,MAElBlB,SAASmB,QAAQC,SAASD,QAAQA;EAEpC,OAAOnB;CACT;CAEA0D,mBAAmB9D,UAAyBD,SAAwB;EAClE,MAAMgE,eAAe,KAAK3D,SAASwC,IAAI7C,QAAQa,EAAE;EACjD,MAAMU,OAAOtB,WAAW,KAAKI,SAASwC,IAAI5C,QAAQ,IAAI;EACtD,IAAIsB,SAASN,KAAAA,GACX,MAAM,IAAIC,MACR,kHACF;EAEF,IAAI8C,cAAc;GAChBA,aAAaxC,UAAUxB;GACvB,KAAK+C,UAAUxB,MAAMyC,cAAc,QAAQ;GAC3C,KAAKP,UAAUxB,MAAM;GACrB;EACF;EAEA,MAAMgC,UAA6B;GACjC1C;GACAC,SAASxB;GACTqB,MAAM;GACND,UAAU,CAAA;GACVK,OAAOF,OAAOA,KAAKE,QAAQ,IAAI;EACjC;EAEA,KAAKpB,SAAS6D,IAAIlE,QAAQa,IAAIoD,OAAO;EACrC,KAAKlB,UAAUxB,MAAM0C,SAAS,MAAM;EAEpC,IAAI,KAAK7B,SAASb,MAChB,KAAKa,OAAO6B;EAGd,KAAKR,UAAUxB,MAAM;CACvB;CAEAkC,WAAWC,WAAmB;EAC5B,MAAMpE,UAAU,KAAKK,SAASwC,IAAIuB,SAAS;EAC3C,IAAI,CAACpE,SACH,MAAM,IAAIkB,MACR,sGACF;EAEF,OAAO;GACLjB,UAAUD,QAAQuB,MAAMC,QAAQX,MAAM;GACtCb,SAASA,QAAQwB;GACjB6C,OAAOrE,QAAQyB;EACjB;CACF;CAEA6C,cAAcF,WAAmBG,eAA2C;EAC1E,MAAMvE,UAAU,KAAKK,SAASwC,IAAIuB,SAAS;EAE3C,IAAI,CAACpE,SACH,MAAM,IAAIkB,MACR,sGACF;EAEF,MAAMsD,cACJD,kBAAkBtD,KAAAA,IACdjB,QAAQuB,OACRgD,kBAAkB,OAChB,OACA,KAAKlE,SAASwC,IAAI0B,aAAa;EACvC,IAAIC,gBAAgBvD,KAAAA,GAClB,MAAM,IAAIC,MACR,0GACF;EAEF,KAAK,MAAM+B,SAASjD,QAAQoB,UAAU;GACpC,MAAMwB,eAAe,KAAKvC,SAASwC,IAAII,KAAK;GAC5C,IAAI,CAACL,cACH,MAAM,IAAI1B,MACR,4GACF;GACF,KAAK6B,UAAUyB,aAAa5B,cAAc,QAAQ;EACpD;EAEA,KAAKG,UAAU,MAAM/C,SAAS,KAAK;EACnC,KAAKK,SAASoE,OAAOL,SAAS;EAE9B,IAAI,KAAKhC,SAASpC,SAChB,KAAKoC,OAAOV,SAAS8C,eAAe,KAAKnC,IAAI;EAG/C,KAAKoB,UAAUxB,MAAM;CACvB;CAEAyC,YAAYN,WAAmB;EAC7B,MAAMpE,UAAU,KAAKK,SAASwC,IAAIuB,SAAS;EAC3C,IAAI,CAACpE,SACH,MAAM,IAAIkB,MACR,oGACF;EAEF,MAAM,EAAEE,aAAapB,QAAQuB,QAAQ,KAAKc;EAC1C,OAAOjB;CACT;;;;;;;;;CAUA,iCACEwD,cACAC,aACA;EACA,IAAI,CAACD,cAAc;EAEnB,MAAME,+BAAe,IAAIC,IAAY;EACrC,KAAK,IAAIvD,UAAUqD,aAAarD,SAASA,UAAUA,QAAQD,MACzDuD,aAAaE,IAAIxD,QAAQA,QAAQX,EAAE;EAGrC,MAAMoE,QAAkB,CAAA;EACxB,KACE,IAAIzD,UAAoCoD,cACxCpD,SACAA,UAAUA,QAAQD,MAClB;GAGA,IAAIuD,aAAaI,IAAI1D,QAAQA,QAAQX,EAAE,GAAG;GAC1C,IAAIW,QAAQA,QAAQmC,UAAUC,cAC5BqB,MAAMnC,KAAKtB,QAAQA,QAAQX,EAAE;EAEjC;EAEA,KAAK,MAAMA,MAAMoE,OAEf,IAAI,KAAK5E,SAAS6E,IAAIrE,EAAE,GAAG,KAAKyD,cAAczD,EAAE;CAEpD;CAEAsE,eAAef,WAAmB;EAChC,MAAMpE,UAAU,KAAKK,SAASwC,IAAIuB,SAAS;EAC3C,IAAI,CAACpE,SACH,MAAM,IAAIkB,MACR,sGACF;EAEF,MAAM0D,eAAe,KAAKxC;EAC1B,MAAMgD,aAAapF,QAAQuB,QAAQ,KAAKc;EACxC+C,WAAW/D,OAAOrB;EAElB,KAAKoC,OAAOV,SAAS1B,OAAO;EAE5B,KAAK2E,iCAAiCC,cAAc,KAAKxC,IAAI;EAE7D,KAAKqB,UAAUxB,MAAM;CACvB;CAEAoD,UAAUjB,WAA0B;EAClC,IAAIA,cAAc,MAAM;GACtB,KAAKkB,MAAM;GACX;EACF;EAEA,MAAMtF,UAAU,KAAKK,SAASwC,IAAIuB,SAAS;EAC3C,IAAI,CAACpE,SACH,MAAM,IAAIkB,MACR,iGACF;EAEF,MAAM0D,eAAe,KAAKxC;EAE1B,IAAIpC,QAAQoB,SAASqB,SAAS,GAAG;GAC/B,MAAMD,UAAU,CAAC,GAAGxC,QAAQoB,QAAQ;GACpC,OAAOoB,QAAQC,SAAS,GAAG;IACzB,MAAME,UAAUH,QAAQE,IAAI;IAC5B,MAAME,eAAe,KAAKvC,SAASwC,IAAIF,OAAO;IAC9C,IAAIC,cAAc;KAChB,KAAK,MAAM2C,gBAAgB3C,aAAaxB,UACtCoB,QAAQM,KAAKyC,YAAY;KAE3B,KAAKlF,SAASoE,OAAO9B,OAAO;IAC9B;GACF;GAEA3C,QAAQoB,WAAW,CAAA;GACnBpB,QAAQqB,OAAO;EACjB;EAEA,KAAKe,OAAOpC;EACZ,KACE,IAAIwB,UAAoCxB,SACxCwB,SACAA,UAAUA,QAAQD,MAElB,IAAIC,QAAQD,MACVC,QAAQD,KAAKF,OAAOG;OAEpB,KAAKa,KAAKhB,OAAOG;EAIrB,KAAKmD,iCAAiCC,cAAc,KAAKxC,IAAI;EAE7D,KAAKqB,UAAUxB,MAAM;CACvB;CAEAqD,QAAc;EACZ,KAAKjF,SAASiF,MAAM;EACpB,KAAKlD,OAAO;EACZ,KAAKC,OAAO;GACVjB,UAAU,CAAA;GACVC,MAAM;EACR;EACA,KAAKoC,UAAUxB,MAAM;CACvB;CAEAuD,SAAoC;EAClC,MAAMC,cAAqD,CAAA;EAK3D,KAAK,MAAM,GAAGzF,YAAY,KAAKK,UAAU;GACvC,IAAIL,QAAQwB,QAAQmC,UAAUC,cAAc;GAC5C,IAAIrC,OAAOvB,QAAQuB;GACnB,OAAOA,QAAQA,KAAKC,QAAQmC,UAAUC,cACpCrC,OAAOA,KAAKA;GAEdkE,YAAY3C,KAAK;IACf9C,SAASA,QAAQwB;IACjBvB,UAAUsB,MAAMC,QAAQX,MAAM;GAChC,CAAC;EACH;EAEA,OAAO;GACLT,QAAQ,KAAKsD;GACbrD,UAAUoF;EACZ;CACF;CAEAC,OAAO,EAAEtF,QAAQC,YAAuC;EACtD,KAAK,MAAM,EAAEL,SAASC,cAAcI,UAClC,KAAK0D,mBAAmB9D,UAAUD,OAAO;EAG3C,KAAKqF,UAAUjF,UAAUC,SAASkD,GAAG,EAAE,CAAC,EAAEvD,QAAQa,MAAM,IAAI;CAC9D;AACF"}

package/dist/runtime/utils/pending-tool-calls.d.ts

@@ -0,0 +1,20 @@
+//#region src/runtime/utils/pending-tool-calls.d.ts
+type PendingToolCallMessage<TToolCall> = {
+  toolCalls: readonly TToolCall[];
+} | {
+  toolCallId: string;
+} | undefined;
+declare const scanPendingToolCalls: <TMessage, TToolCall>(messages: readonly TMessage[], getMessage: (message: TMessage) => PendingToolCallMessage<TToolCall>, getToolCallId: (toolCall: TToolCall) => string) => TToolCall[];
+declare const createToolCallCancellationStub: (toolCall: {
+  readonly id: string;
+  readonly name: string;
+}) => {
+  type: "tool";
+  name: string;
+  tool_call_id: string;
+  content: string;
+  status: "error";
+};
+//#endregion
+export { createToolCallCancellationStub, scanPendingToolCalls };
+//# sourceMappingURL=pending-tool-calls.d.ts.map

package/dist/runtime/utils/pending-tool-calls.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pending-tool-calls.d.ts","names":[],"sources":["../../../src/runtime/utils/pending-tool-calls.ts"],"mappings":";KAAK,uBAAuB;EACtB,oBAAoB;;EACpB;;cAGO,uBAAwB,UAAU,WAC7C,mBAAmB,YACnB,aAAa,SAAS,aAAa,uBAAuB,YAC1D,gBAAgB,UAAU,yBACzB;cAgBU,iCAAkC;WACpC;WACA"}

package/dist/runtime/utils/pending-tool-calls.js

@@ -0,0 +1,22 @@
+//#region src/runtime/utils/pending-tool-calls.ts
+const scanPendingToolCalls = (messages, getMessage, getToolCallId) => {
+	const pending = /* @__PURE__ */ new Map();
+	for (const message of messages) {
+		const value = getMessage(message);
+		if (!value) continue;
+		if ("toolCalls" in value) for (const toolCall of value.toolCalls) pending.set(getToolCallId(toolCall), toolCall);
+		else pending.delete(value.toolCallId);
+	}
+	return [...pending.values()];
+};
+const createToolCallCancellationStub = (toolCall) => ({
+	type: "tool",
+	name: toolCall.name,
+	tool_call_id: toolCall.id,
+	content: JSON.stringify({ cancelled: true }),
+	status: "error"
+});
+//#endregion
+export { createToolCallCancellationStub, scanPendingToolCalls };
+
+//# sourceMappingURL=pending-tool-calls.js.map

package/dist/runtime/utils/pending-tool-calls.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pending-tool-calls.js","names":["PendingToolCallMessage","toolCalls","TToolCall","toolCallId","scanPendingToolCalls","messages","TMessage","getMessage","message","getToolCallId","toolCall","pending","Map","value","set","delete","values","createToolCallCancellationStub","id","name","type","const","tool_call_id","content","JSON","stringify","cancelled","status"],"sources":["../../../src/runtime/utils/pending-tool-calls.ts"],"sourcesContent":["type PendingToolCallMessage<TToolCall> =\n  | { toolCalls: readonly TToolCall[] }\n  | { toolCallId: string }\n  | undefined;\n\nexport const scanPendingToolCalls = <TMessage, TToolCall>(\n  messages: readonly TMessage[],\n  getMessage: (message: TMessage) => PendingToolCallMessage<TToolCall>,\n  getToolCallId: (toolCall: TToolCall) => string,\n): TToolCall[] => {\n  const pending = new Map<string, TToolCall>();\n  for (const message of messages) {\n    const value = getMessage(message);\n    if (!value) continue;\n    if (\"toolCalls\" in value) {\n      for (const toolCall of value.toolCalls) {\n        pending.set(getToolCallId(toolCall), toolCall);\n      }\n    } else {\n      pending.delete(value.toolCallId);\n    }\n  }\n  return [...pending.values()];\n};\n\nexport const createToolCallCancellationStub = (toolCall: {\n  readonly id: string;\n  readonly name: string;\n}) => ({\n  type: \"tool\" as const,\n  name: toolCall.name,\n  tool_call_id: toolCall.id,\n  content: JSON.stringify({ cancelled: true }),\n  status: \"error\" as const,\n});\n"],"mappings":";AAKA,MAAaI,wBACXC,UACAE,YACAE,kBACgB;CAChB,MAAME,0BAAU,IAAIC,IAAuB;CAC3C,KAAK,MAAMJ,WAAWH,UAAU;EAC9B,MAAMQ,QAAQN,WAAWC,OAAO;EAChC,IAAI,CAACK,OAAO;EACZ,IAAI,eAAeA,OACjB,KAAK,MAAMH,YAAYG,MAAMZ,WAC3BU,QAAQG,IAAIL,cAAcC,QAAQ,GAAGA,QAAQ;OAG/CC,QAAQI,OAAOF,MAAMV,UAAU;CAEnC;CACA,OAAO,CAAC,GAAGQ,QAAQK,OAAO,CAAC;AAC7B;AAEA,MAAaC,kCAAkCP,cAGxC;CACLU,MAAM;CACND,MAAMT,SAASS;CACfG,cAAcZ,SAASQ;CACvBK,SAASC,KAAKC,UAAU,EAAEC,WAAW,KAAK,CAAC;CAC3CC,QAAQ;AACV"}

package/dist/runtime/utils/resolveToolApprovalResponse.d.ts

@@ -1,12 +1,15 @@
-import { ToolApprovalOption, ToolApprovalResponse } from "../../types/message.js";
+import { ToolApprovalDisplay, ToolApprovalOption, ToolApprovalResponse } from "../../types/message.js";
 import { RespondToToolApprovalOptions } from "../interfaces/thread-runtime-core.js";
 //#region src/runtime/utils/resolveToolApprovalResponse.d.ts
 /**
- * Resolves a renderer-facing approval response (boolean or optionId form)
- * against the approval's option list into the runtime-facing decision shape.
+ * Resolves a renderer-facing approval response (boolean, optionId, or
+ * free-form answer) against the approval's request shape into the
+ * runtime-facing decision shape.
  */
 declare const resolveToolApprovalResponse: (approval: {
   readonly id: string;
+  readonly display?: ToolApprovalDisplay;
+  readonly allowFreeform?: boolean;
   readonly options?: readonly ToolApprovalOption[];
 }, response: ToolApprovalResponse) => RespondToToolApprovalOptions;
 //#endregion

package/dist/runtime/utils/resolveToolApprovalResponse.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"resolveToolApprovalResponse.d.ts","names":[],"sources":["../../../src/runtime/utils/resolveToolApprovalResponse.ts"],"mappings":";;;;;;;cAiBa,8BACX;WACW;WACA,mBAAmB;GAE9B,UAAU,yBACT"}
+{"version":3,"file":"resolveToolApprovalResponse.d.ts","names":[],"sources":["../../../src/runtime/utils/resolveToolApprovalResponse.ts"],"mappings":";;;;;;;;cAoBa,8BACX;WACW;WACA,UAAU;WACV;WACA,mBAAmB;GAE9B,UAAU,yBACT"}

package/dist/runtime/utils/resolveToolApprovalResponse.js

@@ -1,3 +1,4 @@
+import { toolApprovalAcceptsText } from "../../types/message.js";
 //#region src/runtime/utils/resolveToolApprovalResponse.ts
 const APPROVED_BY_KIND = {
 	"allow-once": true,
@@ -6,10 +7,13 @@ const APPROVED_BY_KIND = {
 	"reject-always": false
 };
 /**
-* Resolves a renderer-facing approval response (boolean or optionId form)
-* against the approval's option list into the runtime-facing decision shape.
+* Resolves a renderer-facing approval response (boolean, optionId, or
+* free-form answer) against the approval's request shape into the
+* runtime-facing decision shape.
 */
 const resolveToolApprovalResponse = (approval, response) => {
+	const text = response.text;
+	if (text !== void 0 && !toolApprovalAcceptsText(approval)) throw new Error(`Tool approval "${approval.id}" does not accept a free-form answer; the request must declare display "text" or allowFreeform`);
 	let approved;
 	let optionId;
 	if ("optionId" in response) {
@@ -21,11 +25,16 @@ const resolveToolApprovalResponse = (approval, response) => {
 			approved = APPROVED_BY_KIND[option.kind];
 		}
 		optionId = option.id;
-	} else approved = response.approved;
+	} else if ("approved" in response) approved = response.approved;
+	else {
+		if (approval.display !== "text" && approval.display !== "select") throw new Error(`Tool approval "${approval.id}" is a decision, not a question; respond with an explicit approved value, optionally alongside the answer`);
+		approved = true;
+	}
 	return {
 		approvalId: approval.id,
 		approved,
 		...optionId !== void 0 && { optionId },
+		...text !== void 0 && { text },
 		...response.reason != null && { reason: response.reason }
 	};
 };

package/dist/runtime/utils/resolveToolApprovalResponse.js.map

@@ -1 +1 @@
-{"version":3,"file":"resolveToolApprovalResponse.js","names":["ToolApprovalOption","ToolApprovalResponse","RespondToToolApprovalOptions","APPROVED_BY_KIND","Record","resolveToolApprovalResponse","approval","id","options","response","approved","optionId","option","find","o","Error","Object","hasOwn","kind","approvalId","undefined","reason"],"sources":["../../../src/runtime/utils/resolveToolApprovalResponse.ts"],"sourcesContent":["import type {\n  ToolApprovalOption,\n  ToolApprovalResponse,\n} from \"../../types/message\";\nimport type { RespondToToolApprovalOptions } from \"../interfaces/thread-runtime-core\";\n\nconst APPROVED_BY_KIND: Record<string, boolean> = {\n  \"allow-once\": true,\n  \"allow-always\": true,\n  \"reject-once\": false,\n  \"reject-always\": false,\n};\n\n/**\n * Resolves a renderer-facing approval response (boolean or optionId form)\n * against the approval's option list into the runtime-facing decision shape.\n */\nexport const resolveToolApprovalResponse = (\n  approval: {\n    readonly id: string;\n    readonly options?: readonly ToolApprovalOption[];\n  },\n  response: ToolApprovalResponse,\n): RespondToToolApprovalOptions => {\n  let approved: boolean;\n  let optionId: string | undefined;\n\n  if (\"optionId\" in response) {\n    const option = approval.options?.find((o) => o.id === response.optionId);\n    if (!option)\n      throw new Error(\n        `Tool approval has no option with id \"${response.optionId}\"`,\n      );\n\n    if (\"approved\" in response) {\n      approved = response.approved;\n    } else {\n      if (!Object.hasOwn(APPROVED_BY_KIND, option.kind))\n        throw new Error(\n          `Tool approval option \"${option.id}\" has a custom kind \"${option.kind}\"; respond with an explicit approved value instead`,\n        );\n      approved = APPROVED_BY_KIND[option.kind]!;\n    }\n    optionId = option.id;\n  } else {\n    approved = response.approved;\n  }\n\n  return {\n    approvalId: approval.id,\n    approved,\n    ...(optionId !== undefined && { optionId }),\n    ...(response.reason != null && { reason: response.reason }),\n  };\n};\n"],"mappings":";AAMA,MAAMG,mBAA4C;CAChD,cAAc;CACd,gBAAgB;CAChB,eAAe;CACf,iBAAiB;AACnB;;;;;AAMA,MAAaE,+BACXC,UAIAG,aACiC;CACjC,IAAIC;CACJ,IAAIC;CAEJ,IAAI,cAAcF,UAAU;EAC1B,MAAMG,SAASN,SAASE,SAASK,MAAMC,MAAMA,EAAEP,OAAOE,SAASE,QAAQ;EACvE,IAAI,CAACC,QACH,MAAM,IAAIG,MACR,wCAAwCN,SAASE,SAAQ,EAC3D;EAEF,IAAI,cAAcF,UAChBC,WAAWD,SAASC;OACf;GACL,IAAI,CAACM,OAAOC,OAAOd,kBAAkBS,OAAOM,IAAI,GAC9C,MAAM,IAAIH,MACR,yBAAyBH,OAAOL,GAAE,uBAAwBK,OAAOM,KAAI,mDACvE;GACFR,WAAWP,iBAAiBS,OAAOM;EACrC;EACAP,WAAWC,OAAOL;CACpB,OACEG,WAAWD,SAASC;CAGtB,OAAO;EACLS,YAAYb,SAASC;EACrBG;EACA,GAAIC,aAAaS,KAAAA,KAAa,EAAET,SAAS;EACzC,GAAIF,SAASY,UAAU,QAAQ,EAAEA,QAAQZ,SAASY,OAAO;CAC3D;AACF"}
+{"version":3,"file":"resolveToolApprovalResponse.js","names":["toolApprovalAcceptsText","ToolApprovalDisplay","ToolApprovalOption","ToolApprovalResponse","RespondToToolApprovalOptions","APPROVED_BY_KIND","Record","resolveToolApprovalResponse","approval","id","display","allowFreeform","options","response","text","undefined","Error","approved","optionId","option","find","o","Object","hasOwn","kind","approvalId","reason"],"sources":["../../../src/runtime/utils/resolveToolApprovalResponse.ts"],"sourcesContent":["import {\n  toolApprovalAcceptsText,\n  type ToolApprovalDisplay,\n  type ToolApprovalOption,\n  type ToolApprovalResponse,\n} from \"../../types/message\";\nimport type { RespondToToolApprovalOptions } from \"../interfaces/thread-runtime-core\";\n\nconst APPROVED_BY_KIND: Record<string, boolean> = {\n  \"allow-once\": true,\n  \"allow-always\": true,\n  \"reject-once\": false,\n  \"reject-always\": false,\n};\n\n/**\n * Resolves a renderer-facing approval response (boolean, optionId, or\n * free-form answer) against the approval's request shape into the\n * runtime-facing decision shape.\n */\nexport const resolveToolApprovalResponse = (\n  approval: {\n    readonly id: string;\n    readonly display?: ToolApprovalDisplay;\n    readonly allowFreeform?: boolean;\n    readonly options?: readonly ToolApprovalOption[];\n  },\n  response: ToolApprovalResponse,\n): RespondToToolApprovalOptions => {\n  const text = response.text;\n  if (text !== undefined && !toolApprovalAcceptsText(approval))\n    throw new Error(\n      `Tool approval \"${approval.id}\" does not accept a free-form answer; the request must declare display \"text\" or allowFreeform`,\n    );\n\n  let approved: boolean;\n  let optionId: string | undefined;\n\n  if (\"optionId\" in response) {\n    const option = approval.options?.find((o) => o.id === response.optionId);\n    if (!option)\n      throw new Error(\n        `Tool approval has no option with id \"${response.optionId}\"`,\n      );\n\n    if (\"approved\" in response) {\n      approved = response.approved;\n    } else {\n      if (!Object.hasOwn(APPROVED_BY_KIND, option.kind))\n        throw new Error(\n          `Tool approval option \"${option.id}\" has a custom kind \"${option.kind}\"; respond with an explicit approved value instead`,\n        );\n      approved = APPROVED_BY_KIND[option.kind]!;\n    }\n    optionId = option.id;\n  } else if (\"approved\" in response) {\n    approved = response.approved;\n  } else {\n    // A bare answer resolves a question, where answering is not refusing. On a\n    // decision the approval itself is the authorization, so inferring one from\n    // a typed note would let the note authorize the call.\n    if (approval.display !== \"text\" && approval.display !== \"select\")\n      throw new Error(\n        `Tool approval \"${approval.id}\" is a decision, not a question; respond with an explicit approved value, optionally alongside the answer`,\n      );\n    approved = true;\n  }\n\n  return {\n    approvalId: approval.id,\n    approved,\n    ...(optionId !== undefined && { optionId }),\n    ...(text !== undefined && { text }),\n    ...(response.reason != null && { reason: response.reason }),\n  };\n};\n"],"mappings":";;AAQA,MAAMK,mBAA4C;CAChD,cAAc;CACd,gBAAgB;CAChB,eAAe;CACf,iBAAiB;AACnB;;;;;;AAOA,MAAaE,+BACXC,UAMAK,aACiC;CACjC,MAAMC,OAAOD,SAASC;CACtB,IAAIA,SAASC,KAAAA,KAAa,CAACf,wBAAwBQ,QAAQ,GACzD,MAAM,IAAIQ,MACR,kBAAkBR,SAASC,GAAE,+FAC/B;CAEF,IAAIQ;CACJ,IAAIC;CAEJ,IAAI,cAAcL,UAAU;EAC1B,MAAMM,SAASX,SAASI,SAASQ,MAAMC,MAAMA,EAAEZ,OAAOI,SAASK,QAAQ;EACvE,IAAI,CAACC,QACH,MAAM,IAAIH,MACR,wCAAwCH,SAASK,SAAQ,EAC3D;EAEF,IAAI,cAAcL,UAChBI,WAAWJ,SAASI;OACf;GACL,IAAI,CAACK,OAAOC,OAAOlB,kBAAkBc,OAAOK,IAAI,GAC9C,MAAM,IAAIR,MACR,yBAAyBG,OAAOV,GAAE,uBAAwBU,OAAOK,KAAI,mDACvE;GACFP,WAAWZ,iBAAiBc,OAAOK;EACrC;EACAN,WAAWC,OAAOV;CACpB,OAAO,IAAI,cAAcI,UACvBI,WAAWJ,SAASI;MACf;EAIL,IAAIT,SAASE,YAAY,UAAUF,SAASE,YAAY,UACtD,MAAM,IAAIM,MACR,kBAAkBR,SAASC,GAAE,0GAC/B;EACFQ,WAAW;CACb;CAEA,OAAO;EACLQ,YAAYjB,SAASC;EACrBQ;EACA,GAAIC,aAAaH,KAAAA,KAAa,EAAEG,SAAS;EACzC,GAAIJ,SAASC,KAAAA,KAAa,EAAED,KAAK;EACjC,GAAID,SAASa,UAAU,QAAQ,EAAEA,QAAQb,SAASa,OAAO;CAC3D;AACF"}

package/dist/runtime/utils/thread-message-like.d.ts

@@ -1,4 +1,4 @@
-import { DataMessagePart, FileMessagePart, GenerativeUIMessagePart, ImageMessagePart, MessageStatus, MessageTiming, PartProviderMetadata, ReasoningMessagePart, SourceMessagePart, TextMessagePart, ThreadMessage, ThreadStep, ThreadUserMessagePart, ToolApprovalOption, ToolCallTiming, Unstable_AudioMessagePart } from "../../types/message.js";
+import { DataMessagePart, FileMessagePart, GenerativeUIMessagePart, ImageMessagePart, MessageStatus, MessageTiming, PartProviderMetadata, ReasoningMessagePart, SourceMessagePart, TextMessagePart, ThreadMessage, ThreadStep, ThreadUserMessagePart, ToolCallMessagePart, ToolCallMessagePartMcpMetadata, ToolCallTiming, Unstable_AudioMessagePart } from "../../types/message.js";
 import { CompleteAttachment } from "../../types/attachment.js";
 import { ReadonlyJSONObject, ReadonlyJSONValue } from "assistant-stream/utils";
 //#region src/runtime/utils/thread-message-like.d.ts
@@ -24,16 +24,9 @@ type ThreadMessageLike = {
       payload: unknown;
     };
     readonly timing?: ToolCallTiming;
+    readonly mcp?: ToolCallMessagePartMcpMetadata;
     readonly providerMetadata?: PartProviderMetadata;
-    readonly approval?: {
-      readonly id: string;
-      readonly approved?: boolean;
-      readonly reason?: string;
-      readonly isAutomatic?: boolean;
-      readonly options?: readonly ToolApprovalOption[];
-      readonly optionId?: string;
-      readonly resolution?: "cancelled" | "expired";
-    };
+    readonly approval?: NonNullable<ToolCallMessagePart["approval"]>;
   })[];
   readonly id?: string | undefined;
   readonly createdAt?: Date | undefined;

package/dist/runtime/utils/thread-message-like.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"thread-message-like.d.ts","names":[],"sources":["../../../src/runtime/utils/thread-message-like.ts"],"mappings":";;;;KAiCK;WACM;WACA;;KAGC;WACD;WACA,4BAGD,kBACA,uBACA,oBACA,mBACA,kBACA,kBACA,0BACA,4BACA;aAEW;aACA;aACA;aACA,OAAO;aACP;aACA;aACA;aACA;aACA;aACA,oBAAoB;aACpB;MAAc;MAAe;;aAC7B,SAAS;aACT,mBAAmB;aACnB;eACE;eACA;eACA;eACA;eACA,mBAAmB;eACnB;eACA;;;WAIZ;WACA,YAAY;WACZ,SAAS;WACT,wBACK,KAAK;aACJ,mBAAmB,wBAAwB;;WAGjD;aAEM,iBAAiB;aACjB,gCACI;aAEJ,yBAAyB;aACzB,iBAAiB;aACjB,SAAS;aACT;eAA+B;;aAC/B;aACA,SAAS;;;;;;cAgBb,wBACX,MAAM,mBACN,oBACA,gBAAgB,kBACf"}
+{"version":3,"file":"thread-message-like.d.ts","names":[],"sources":["../../../src/runtime/utils/thread-message-like.ts"],"mappings":";;;;KAkCK;WACM;WACA;;KAGC;WACD;WACA,4BAGD,kBACA,uBACA,oBACA,mBACA,kBACA,kBACA,0BACA,4BACA;aAEW;aACA;aACA;aACA,OAAO;aACP;aACA;aACA;aACA;aACA;aACA,oBAAoB;aACpB;MAAc;MAAe;;aAC7B,SAAS;aACT,MAAM;aACN,mBAAmB;aACnB,WAAW,YAAY;;WAGjC;WACA,YAAY;WACZ,SAAS;WACT,wBACK,KAAK;aACJ,mBAAmB,wBAAwB;;WAGjD;aAEM,iBAAiB;aACjB,gCACI;aAEJ,yBAAyB;aACzB,iBAAiB;aACjB,SAAS;aACT;eAA+B;;aAC/B;aACA,SAAS;;;;;;cAgBb,wBACX,MAAM,mBACN,oBACA,gBAAgB,kBACf"}

package/dist/runtime/utils/thread-message-like.js.map

@@ -1 +1 @@
-{"version":3,"file":"thread-message-like.js","names":["parsePartialJsonObject","generateId","parseDataUrl","ReasoningMessagePart","SourceMessagePart","ThreadStep","MessageStatus","ImageMessagePart","ThreadMessage","ThreadAssistantMessagePart","ThreadAssistantMessage","ThreadUserMessagePart","ThreadUserMessage","ThreadSystemMessage","FileMessagePart","DataMessagePart","GenerativeUIMessagePart","Unstable_AudioMessagePart","CompleteAttachment","MessageTiming","PartProviderMetadata","TextMessagePart","ToolApprovalOption","ToolCallTiming","ReadonlyJSONObject","ReadonlyJSONValue","DataPrefixedPart","type","data","ThreadMessageLike","role","content","toolCallId","toolName","args","argsText","artifact","result","isError","parentId","messages","interrupt","payload","timing","providerMetadata","approval","id","approved","reason","isAutomatic","options","optionId","resolution","createdAt","Date","status","attachments","Omit","metadata","unstable_state","unstable_annotations","unstable_data","steps","submittedFeedback","isOptimistic","custom","Record","convertDataPrefixedPart","startsWith","undefined","name","substring","fromThreadMessageLike","like","fallbackId","fallbackStatus","common","const","text","sanitizeImageContent","image","rest","mimeType","test","console","warn","length","Error","map","part","trim","unstable_summary","basePart","commonProps","JSON","stringify","converted","filter","c","att","unsupportedRole"],"sources":["../../../src/runtime/utils/thread-message-like.ts"],"sourcesContent":["import { parsePartialJsonObject } from \"assistant-stream/utils\";\nimport { generateId } from \"../../utils/id\";\nimport { parseDataUrl } from \"../../utils/data-url\";\nimport type {\n  ReasoningMessagePart,\n  SourceMessagePart,\n  ThreadStep,\n  MessageStatus,\n  ImageMessagePart,\n  ThreadMessage,\n  ThreadAssistantMessagePart,\n  ThreadAssistantMessage,\n  ThreadUserMessagePart,\n  ThreadUserMessage,\n  ThreadSystemMessage,\n  FileMessagePart,\n  DataMessagePart,\n  GenerativeUIMessagePart,\n  Unstable_AudioMessagePart,\n} from \"../../types/message\";\nimport type { CompleteAttachment } from \"../../types/attachment\";\nimport type {\n  MessageTiming,\n  PartProviderMetadata,\n  TextMessagePart,\n  ToolApprovalOption,\n  ToolCallTiming,\n} from \"../../types/message\";\nimport type {\n  ReadonlyJSONObject,\n  ReadonlyJSONValue,\n} from \"assistant-stream/utils\";\n\ntype DataPrefixedPart = {\n  readonly type: `data-${string}`;\n  readonly data: any;\n};\n\nexport type ThreadMessageLike = {\n  readonly role: \"assistant\" | \"user\" | \"system\";\n  readonly content:\n    | string\n    | readonly (\n        | TextMessagePart\n        | ReasoningMessagePart\n        | SourceMessagePart\n        | ImageMessagePart\n        | FileMessagePart\n        | DataMessagePart\n        | GenerativeUIMessagePart\n        | Unstable_AudioMessagePart\n        | DataPrefixedPart\n        | {\n            readonly type: \"tool-call\";\n            readonly toolCallId?: string;\n            readonly toolName: string;\n            readonly args?: ReadonlyJSONObject;\n            readonly argsText?: string;\n            readonly artifact?: any;\n            readonly result?: any | undefined;\n            readonly isError?: boolean | undefined;\n            readonly parentId?: string | undefined;\n            readonly messages?: readonly ThreadMessage[] | undefined;\n            readonly interrupt?: { type: \"human\"; payload: unknown };\n            readonly timing?: ToolCallTiming;\n            readonly providerMetadata?: PartProviderMetadata;\n            readonly approval?: {\n              readonly id: string;\n              readonly approved?: boolean;\n              readonly reason?: string;\n              readonly isAutomatic?: boolean;\n              readonly options?: readonly ToolApprovalOption[];\n              readonly optionId?: string;\n              readonly resolution?: \"cancelled\" | \"expired\";\n            };\n          }\n      )[];\n  readonly id?: string | undefined;\n  readonly createdAt?: Date | undefined;\n  readonly status?: MessageStatus | undefined;\n  readonly attachments?:\n    | readonly (Omit<CompleteAttachment, \"content\"> & {\n        readonly content: readonly (ThreadUserMessagePart | DataPrefixedPart)[];\n      })[]\n    | undefined;\n  readonly metadata?:\n    | {\n        readonly unstable_state?: ReadonlyJSONValue;\n        readonly unstable_annotations?:\n          | readonly ReadonlyJSONValue[]\n          | undefined;\n        readonly unstable_data?: readonly ReadonlyJSONValue[] | undefined;\n        readonly steps?: readonly ThreadStep[] | undefined;\n        readonly timing?: MessageTiming | undefined;\n        readonly submittedFeedback?: { readonly type: \"positive\" | \"negative\" };\n        readonly isOptimistic?: boolean | undefined;\n        readonly custom?: Record<string, unknown> | undefined;\n      }\n    | undefined;\n};\n\nconst convertDataPrefixedPart = (\n  type: string,\n  data: unknown,\n): DataMessagePart | undefined => {\n  if (!type.startsWith(\"data-\")) return undefined;\n  return { type: \"data\", name: type.substring(5), data };\n};\n\n/**\n * @deprecated This API is experimental and may change without notice.\n */\nexport const fromThreadMessageLike = (\n  like: ThreadMessageLike,\n  fallbackId: string,\n  fallbackStatus: MessageStatus,\n): ThreadMessage => {\n  const { role, id, createdAt, attachments, status, metadata } = like;\n  const common = {\n    id: id ?? fallbackId,\n    createdAt: createdAt ?? new Date(),\n  };\n\n  const content =\n    typeof like.content === \"string\"\n      ? [{ type: \"text\" as const, text: like.content }]\n      : like.content;\n\n  const sanitizeImageContent = ({\n    image,\n    ...rest\n  }: ImageMessagePart): ImageMessagePart | null => {\n    if (typeof image !== \"string\") return null;\n    if (parseDataUrl(image)?.mimeType.startsWith(\"image/\")) {\n      return { ...rest, image };\n    }\n    if (/^(https:\\/\\/|blob:)/i.test(image)) {\n      return { ...rest, image };\n    }\n    console.warn(`Invalid image data format detected`);\n    return null;\n  };\n\n  if (role !== \"user\" && attachments?.length)\n    throw new Error(\"attachments are only supported for user messages\");\n\n  if (role !== \"assistant\" && status)\n    throw new Error(\"status is only supported for assistant messages\");\n\n  if (role !== \"assistant\" && metadata?.steps)\n    throw new Error(\"metadata.steps is only supported for assistant messages\");\n\n  switch (role) {\n    case \"assistant\":\n      return {\n        ...common,\n        role,\n        content: content\n          .map((part): ThreadAssistantMessagePart | null => {\n            const type = part.type;\n            switch (type) {\n              case \"text\":\n                if (!part.text?.trim()) return null;\n                return part;\n\n              case \"reasoning\":\n                if (!part.text?.trim() && !part.unstable_summary?.trim())\n                  return null;\n                return part;\n\n              case \"file\":\n              case \"source\":\n                return part;\n\n              case \"image\":\n                return sanitizeImageContent(part);\n\n              case \"data\":\n                return part;\n\n              case \"generative-ui\":\n                return part;\n\n              case \"tool-call\": {\n                const { parentId, messages, ...basePart } = part;\n                const commonProps = {\n                  ...basePart,\n                  toolCallId: part.toolCallId ?? `tool-${generateId()}`,\n                  ...(parentId !== undefined && { parentId }),\n                  ...(messages !== undefined && { messages }),\n                };\n\n                if (part.args) {\n                  return {\n                    ...commonProps,\n                    args: part.args,\n                    argsText: part.argsText ?? JSON.stringify(part.args),\n                  };\n                }\n                return {\n                  ...commonProps,\n                  args: parsePartialJsonObject(part.argsText ?? \"\") ?? {},\n                  argsText: part.argsText ?? \"\",\n                };\n              }\n\n              default: {\n                const converted = convertDataPrefixedPart(\n                  type,\n                  (part as DataPrefixedPart).data,\n                );\n                if (converted) return converted;\n                throw new Error(\n                  `Unsupported assistant message part type: ${type}`,\n                );\n              }\n            }\n          })\n          .filter((c) => !!c),\n        status: status ?? fallbackStatus,\n        metadata: {\n          unstable_state: metadata?.unstable_state ?? null,\n          unstable_annotations: metadata?.unstable_annotations ?? [],\n          unstable_data: metadata?.unstable_data ?? [],\n          custom: metadata?.custom ?? {},\n          steps: metadata?.steps ?? [],\n          ...(metadata?.timing && { timing: metadata.timing }),\n          ...(metadata?.submittedFeedback && {\n            submittedFeedback: metadata.submittedFeedback,\n          }),\n          ...(metadata?.isOptimistic && { isOptimistic: true }),\n        },\n      } satisfies ThreadAssistantMessage;\n\n    case \"user\":\n      return {\n        ...common,\n        role,\n        content: content.map((part): ThreadUserMessagePart => {\n          const type = part.type;\n          switch (type) {\n            case \"text\":\n            case \"image\":\n            case \"audio\":\n            case \"file\":\n            case \"data\":\n              return part;\n\n            default: {\n              const converted = convertDataPrefixedPart(\n                type,\n                (part as DataPrefixedPart).data,\n              );\n              if (converted) return converted;\n              throw new Error(`Unsupported user message part type: ${type}`);\n            }\n          }\n        }),\n        attachments: (attachments ?? []).map((att) => ({\n          ...att,\n          content: att.content.map((part): ThreadUserMessagePart => {\n            const converted = convertDataPrefixedPart(\n              part.type,\n              (part as DataPrefixedPart).data,\n            );\n            return converted ?? (part as ThreadUserMessagePart);\n          }),\n        })),\n        metadata: {\n          custom: metadata?.custom ?? {},\n          ...(metadata?.isOptimistic && { isOptimistic: true }),\n        },\n      } satisfies ThreadUserMessage;\n\n    case \"system\":\n      if (content.length !== 1 || content[0]!.type !== \"text\")\n        throw new Error(\n          \"System messages must have exactly one text message part.\",\n        );\n\n      return {\n        ...common,\n        role,\n        content: content as [TextMessagePart],\n        metadata: {\n          custom: metadata?.custom ?? {},\n        },\n      } satisfies ThreadSystemMessage;\n\n    default: {\n      const unsupportedRole: never = role;\n      throw new Error(`Unknown message role: ${unsupportedRole}`);\n    }\n  }\n};\n"],"mappings":";;;;AAqGA,MAAMmE,2BACJxC,MACAC,SACgC;CAChC,IAAI,CAACD,KAAKyC,WAAW,OAAO,GAAG,OAAOC,KAAAA;CACtC,OAAO;EAAE1C,MAAM;EAAQ2C,MAAM3C,KAAK4C,UAAU,CAAC;EAAG3C;CAAK;AACvD;;;;AAKA,MAAa4C,yBACXC,MACAC,YACAC,mBACkB;CAClB,MAAM,EAAE7C,MAAMgB,IAAIO,WAAWG,aAAaD,QAAQG,aAAae;CAC/D,MAAMG,SAAS;EACb9B,IAAIA,MAAM4B;EACVrB,WAAWA,6BAAa,IAAIC,KAAK;CACnC;CAEA,MAAMvB,UACJ,OAAO0C,KAAK1C,YAAY,WACpB,CAAC;EAAEJ,MAAM;EAAiBmD,MAAML,KAAK1C;CAAQ,CAAC,IAC9C0C,KAAK1C;CAEX,MAAMgD,wBAAwB,EAC5BC,OACA,GAAGC,WAC4C;EAC/C,IAAI,OAAOD,UAAU,UAAU,OAAO;EACtC,IAAI9E,aAAa8E,KAAK,CAAC,EAAEE,SAASd,WAAW,QAAQ,GACnD,OAAO;GAAE,GAAGa;GAAMD;EAAM;EAE1B,IAAI,uBAAuBG,KAAKH,KAAK,GACnC,OAAO;GAAE,GAAGC;GAAMD;EAAM;EAE1BI,QAAQC,KAAK,oCAAoC;EACjD,OAAO;CACT;CAEA,IAAIvD,SAAS,UAAU0B,aAAa8B,QAClC,MAAM,IAAIC,MAAM,kDAAkD;CAEpE,IAAIzD,SAAS,eAAeyB,QAC1B,MAAM,IAAIgC,MAAM,iDAAiD;CAEnE,IAAIzD,SAAS,eAAe4B,UAAUI,OACpC,MAAM,IAAIyB,MAAM,yDAAyD;CAE3E,QAAQzD,MAAR;EACE,KAAK,aACH,OAAO;GACL,GAAG8C;GACH9C;GACAC,SAASA,QACNyD,KAAKC,SAA4C;IAChD,MAAM9D,OAAO8D,KAAK9D;IAClB,QAAQA,MAAR;KACE,KAAK;MACH,IAAI,CAAC8D,KAAKX,MAAMY,KAAK,GAAG,OAAO;MAC/B,OAAOD;KAET,KAAK;MACH,IAAI,CAACA,KAAKX,MAAMY,KAAK,KAAK,CAACD,KAAKE,kBAAkBD,KAAK,GACrD,OAAO;MACT,OAAOD;KAET,KAAK;KACL,KAAK,UACH,OAAOA;KAET,KAAK,SACH,OAAOV,qBAAqBU,IAAI;KAElC,KAAK,QACH,OAAOA;KAET,KAAK,iBACH,OAAOA;KAET,KAAK,aAAa;MAChB,MAAM,EAAElD,UAAUC,UAAU,GAAGoD,aAAaH;MAC5C,MAAMI,cAAc;OAClB,GAAGD;OACH5D,YAAYyD,KAAKzD,cAAc,QAAQ/B,WAAW;OAClD,GAAIsC,aAAa8B,KAAAA,KAAa,EAAE9B,SAAS;OACzC,GAAIC,aAAa6B,KAAAA,KAAa,EAAE7B,SAAS;MAC3C;MAEA,IAAIiD,KAAKvD,MACP,OAAO;OACL,GAAG2D;OACH3D,MAAMuD,KAAKvD;OACXC,UAAUsD,KAAKtD,YAAY2D,KAAKC,UAAUN,KAAKvD,IAAI;MACrD;MAEF,OAAO;OACL,GAAG2D;OACH3D,MAAMlC,uBAAuByF,KAAKtD,YAAY,EAAE,KAAK,CAAC;OACtDA,UAAUsD,KAAKtD,YAAY;MAC7B;KACF;KAEA,SAAS;MACP,MAAM6D,YAAY7B,wBAChBxC,MACC8D,KAA0B7D,IAC7B;MACA,IAAIoE,WAAW,OAAOA;MACtB,MAAM,IAAIT,MACR,4CAA4C5D,MAC9C;KACF;IACF;GACF,CAAC,CAAC,CACDsE,QAAQC,MAAM,CAAC,CAACA,CAAC;GACpB3C,QAAQA,UAAUoB;GAClBjB,UAAU;IACRC,gBAAgBD,UAAUC,kBAAkB;IAC5CC,sBAAsBF,UAAUE,wBAAwB,CAAA;IACxDC,eAAeH,UAAUG,iBAAiB,CAAA;IAC1CI,QAAQP,UAAUO,UAAU,CAAC;IAC7BH,OAAOJ,UAAUI,SAAS,CAAA;IAC1B,GAAIJ,UAAUf,UAAU,EAAEA,QAAQe,SAASf,OAAO;IAClD,GAAIe,UAAUK,qBAAqB,EACjCA,mBAAmBL,SAASK,kBAC9B;IACA,GAAIL,UAAUM,gBAAgB,EAAEA,cAAc,KAAK;GACrD;EACF;EAEF,KAAK,QACH,OAAO;GACL,GAAGY;GACH9C;GACAC,SAASA,QAAQyD,KAAKC,SAAgC;IACpD,MAAM9D,OAAO8D,KAAK9D;IAClB,QAAQA,MAAR;KACE,KAAK;KACL,KAAK;KACL,KAAK;KACL,KAAK;KACL,KAAK,QACH,OAAO8D;KAET,SAAS;MACP,MAAMO,YAAY7B,wBAChBxC,MACC8D,KAA0B7D,IAC7B;MACA,IAAIoE,WAAW,OAAOA;MACtB,MAAM,IAAIT,MAAM,uCAAuC5D,MAAM;KAC/D;IACF;GACF,CAAC;GACD6B,cAAcA,eAAe,CAAA,EAAA,CAAIgC,KAAKW,SAAS;IAC7C,GAAGA;IACHpE,SAASoE,IAAIpE,QAAQyD,KAAKC,SAAgC;KAKxD,OAJkBtB,wBAChBsB,KAAK9D,MACJ8D,KAA0B7D,IAEtBoE,KAAcP;IACvB,CAAC;GACH,EAAE;GACF/B,UAAU;IACRO,QAAQP,UAAUO,UAAU,CAAC;IAC7B,GAAIP,UAAUM,gBAAgB,EAAEA,cAAc,KAAK;GACrD;EACF;EAEF,KAAK;GACH,IAAIjC,QAAQuD,WAAW,KAAKvD,QAAQ,EAAE,CAAEJ,SAAS,QAC/C,MAAM,IAAI4D,MACR,0DACF;GAEF,OAAO;IACL,GAAGX;IACH9C;IACSC;IACT2B,UAAU,EACRO,QAAQP,UAAUO,UAAU,CAAC,EAC/B;GACF;EAEF,SAEE,MAAM,IAAIsB,MAAM,yBAAyBa,MAAiB;CAE9D;AACF"}
+{"version":3,"file":"thread-message-like.js","names":["parsePartialJsonObject","generateId","parseDataUrl","ReasoningMessagePart","SourceMessagePart","ThreadStep","MessageStatus","ImageMessagePart","ThreadMessage","ThreadAssistantMessagePart","ThreadAssistantMessage","ThreadUserMessagePart","ThreadUserMessage","ThreadSystemMessage","FileMessagePart","DataMessagePart","GenerativeUIMessagePart","Unstable_AudioMessagePart","CompleteAttachment","MessageTiming","PartProviderMetadata","TextMessagePart","ToolCallTiming","ToolCallMessagePart","ToolCallMessagePartMcpMetadata","ReadonlyJSONObject","ReadonlyJSONValue","DataPrefixedPart","type","data","ThreadMessageLike","role","content","toolCallId","toolName","args","argsText","artifact","result","isError","parentId","messages","interrupt","payload","timing","mcp","providerMetadata","approval","NonNullable","id","createdAt","Date","status","attachments","Omit","metadata","unstable_state","unstable_annotations","unstable_data","steps","submittedFeedback","isOptimistic","custom","Record","convertDataPrefixedPart","startsWith","undefined","name","substring","fromThreadMessageLike","like","fallbackId","fallbackStatus","common","const","text","sanitizeImageContent","image","rest","mimeType","test","console","warn","length","Error","map","part","trim","unstable_summary","basePart","commonProps","JSON","stringify","converted","filter","c","att","unsupportedRole"],"sources":["../../../src/runtime/utils/thread-message-like.ts"],"sourcesContent":["import { parsePartialJsonObject } from \"assistant-stream/utils\";\nimport { generateId } from \"../../utils/id\";\nimport { parseDataUrl } from \"../../utils/data-url\";\nimport type {\n  ReasoningMessagePart,\n  SourceMessagePart,\n  ThreadStep,\n  MessageStatus,\n  ImageMessagePart,\n  ThreadMessage,\n  ThreadAssistantMessagePart,\n  ThreadAssistantMessage,\n  ThreadUserMessagePart,\n  ThreadUserMessage,\n  ThreadSystemMessage,\n  FileMessagePart,\n  DataMessagePart,\n  GenerativeUIMessagePart,\n  Unstable_AudioMessagePart,\n} from \"../../types/message\";\nimport type { CompleteAttachment } from \"../../types/attachment\";\nimport type {\n  MessageTiming,\n  PartProviderMetadata,\n  TextMessagePart,\n  ToolCallTiming,\n  ToolCallMessagePart,\n  ToolCallMessagePartMcpMetadata,\n} from \"../../types/message\";\nimport type {\n  ReadonlyJSONObject,\n  ReadonlyJSONValue,\n} from \"assistant-stream/utils\";\n\ntype DataPrefixedPart = {\n  readonly type: `data-${string}`;\n  readonly data: any;\n};\n\nexport type ThreadMessageLike = {\n  readonly role: \"assistant\" | \"user\" | \"system\";\n  readonly content:\n    | string\n    | readonly (\n        | TextMessagePart\n        | ReasoningMessagePart\n        | SourceMessagePart\n        | ImageMessagePart\n        | FileMessagePart\n        | DataMessagePart\n        | GenerativeUIMessagePart\n        | Unstable_AudioMessagePart\n        | DataPrefixedPart\n        | {\n            readonly type: \"tool-call\";\n            readonly toolCallId?: string;\n            readonly toolName: string;\n            readonly args?: ReadonlyJSONObject;\n            readonly argsText?: string;\n            readonly artifact?: any;\n            readonly result?: any | undefined;\n            readonly isError?: boolean | undefined;\n            readonly parentId?: string | undefined;\n            readonly messages?: readonly ThreadMessage[] | undefined;\n            readonly interrupt?: { type: \"human\"; payload: unknown };\n            readonly timing?: ToolCallTiming;\n            readonly mcp?: ToolCallMessagePartMcpMetadata;\n            readonly providerMetadata?: PartProviderMetadata;\n            readonly approval?: NonNullable<ToolCallMessagePart[\"approval\"]>;\n          }\n      )[];\n  readonly id?: string | undefined;\n  readonly createdAt?: Date | undefined;\n  readonly status?: MessageStatus | undefined;\n  readonly attachments?:\n    | readonly (Omit<CompleteAttachment, \"content\"> & {\n        readonly content: readonly (ThreadUserMessagePart | DataPrefixedPart)[];\n      })[]\n    | undefined;\n  readonly metadata?:\n    | {\n        readonly unstable_state?: ReadonlyJSONValue;\n        readonly unstable_annotations?:\n          | readonly ReadonlyJSONValue[]\n          | undefined;\n        readonly unstable_data?: readonly ReadonlyJSONValue[] | undefined;\n        readonly steps?: readonly ThreadStep[] | undefined;\n        readonly timing?: MessageTiming | undefined;\n        readonly submittedFeedback?: { readonly type: \"positive\" | \"negative\" };\n        readonly isOptimistic?: boolean | undefined;\n        readonly custom?: Record<string, unknown> | undefined;\n      }\n    | undefined;\n};\n\nconst convertDataPrefixedPart = (\n  type: string,\n  data: unknown,\n): DataMessagePart | undefined => {\n  if (!type.startsWith(\"data-\")) return undefined;\n  return { type: \"data\", name: type.substring(5), data };\n};\n\n/**\n * @deprecated This API is experimental and may change without notice.\n */\nexport const fromThreadMessageLike = (\n  like: ThreadMessageLike,\n  fallbackId: string,\n  fallbackStatus: MessageStatus,\n): ThreadMessage => {\n  const { role, id, createdAt, attachments, status, metadata } = like;\n  const common = {\n    id: id ?? fallbackId,\n    createdAt: createdAt ?? new Date(),\n  };\n\n  const content =\n    typeof like.content === \"string\"\n      ? [{ type: \"text\" as const, text: like.content }]\n      : like.content;\n\n  const sanitizeImageContent = ({\n    image,\n    ...rest\n  }: ImageMessagePart): ImageMessagePart | null => {\n    if (typeof image !== \"string\") return null;\n    if (parseDataUrl(image)?.mimeType.startsWith(\"image/\")) {\n      return { ...rest, image };\n    }\n    if (/^(https:\\/\\/|blob:)/i.test(image)) {\n      return { ...rest, image };\n    }\n    console.warn(`Invalid image data format detected`);\n    return null;\n  };\n\n  if (role !== \"user\" && attachments?.length)\n    throw new Error(\"attachments are only supported for user messages\");\n\n  if (role !== \"assistant\" && status)\n    throw new Error(\"status is only supported for assistant messages\");\n\n  if (role !== \"assistant\" && metadata?.steps)\n    throw new Error(\"metadata.steps is only supported for assistant messages\");\n\n  switch (role) {\n    case \"assistant\":\n      return {\n        ...common,\n        role,\n        content: content\n          .map((part): ThreadAssistantMessagePart | null => {\n            const type = part.type;\n            switch (type) {\n              case \"text\":\n                if (!part.text?.trim()) return null;\n                return part;\n\n              case \"reasoning\":\n                if (!part.text?.trim() && !part.unstable_summary?.trim())\n                  return null;\n                return part;\n\n              case \"file\":\n              case \"source\":\n                return part;\n\n              case \"image\":\n                return sanitizeImageContent(part);\n\n              case \"data\":\n                return part;\n\n              case \"generative-ui\":\n                return part;\n\n              case \"tool-call\": {\n                const { parentId, messages, ...basePart } = part;\n                const commonProps = {\n                  ...basePart,\n                  toolCallId: part.toolCallId ?? `tool-${generateId()}`,\n                  ...(parentId !== undefined && { parentId }),\n                  ...(messages !== undefined && { messages }),\n                };\n\n                if (part.args) {\n                  return {\n                    ...commonProps,\n                    args: part.args,\n                    argsText: part.argsText ?? JSON.stringify(part.args),\n                  };\n                }\n                return {\n                  ...commonProps,\n                  args: parsePartialJsonObject(part.argsText ?? \"\") ?? {},\n                  argsText: part.argsText ?? \"\",\n                };\n              }\n\n              default: {\n                const converted = convertDataPrefixedPart(\n                  type,\n                  (part as DataPrefixedPart).data,\n                );\n                if (converted) return converted;\n                throw new Error(\n                  `Unsupported assistant message part type: ${type}`,\n                );\n              }\n            }\n          })\n          .filter((c) => !!c),\n        status: status ?? fallbackStatus,\n        metadata: {\n          unstable_state: metadata?.unstable_state ?? null,\n          unstable_annotations: metadata?.unstable_annotations ?? [],\n          unstable_data: metadata?.unstable_data ?? [],\n          custom: metadata?.custom ?? {},\n          steps: metadata?.steps ?? [],\n          ...(metadata?.timing && { timing: metadata.timing }),\n          ...(metadata?.submittedFeedback && {\n            submittedFeedback: metadata.submittedFeedback,\n          }),\n          ...(metadata?.isOptimistic && { isOptimistic: true }),\n        },\n      } satisfies ThreadAssistantMessage;\n\n    case \"user\":\n      return {\n        ...common,\n        role,\n        content: content.map((part): ThreadUserMessagePart => {\n          const type = part.type;\n          switch (type) {\n            case \"text\":\n            case \"image\":\n            case \"audio\":\n            case \"file\":\n            case \"data\":\n              return part;\n\n            default: {\n              const converted = convertDataPrefixedPart(\n                type,\n                (part as DataPrefixedPart).data,\n              );\n              if (converted) return converted;\n              throw new Error(`Unsupported user message part type: ${type}`);\n            }\n          }\n        }),\n        attachments: (attachments ?? []).map((att) => ({\n          ...att,\n          content: att.content.map((part): ThreadUserMessagePart => {\n            const converted = convertDataPrefixedPart(\n              part.type,\n              (part as DataPrefixedPart).data,\n            );\n            return converted ?? (part as ThreadUserMessagePart);\n          }),\n        })),\n        metadata: {\n          custom: metadata?.custom ?? {},\n          ...(metadata?.isOptimistic && { isOptimistic: true }),\n        },\n      } satisfies ThreadUserMessage;\n\n    case \"system\":\n      if (content.length !== 1 || content[0]!.type !== \"text\")\n        throw new Error(\n          \"System messages must have exactly one text message part.\",\n        );\n\n      return {\n        ...common,\n        role,\n        content: content as [TextMessagePart],\n        metadata: {\n          custom: metadata?.custom ?? {},\n        },\n      } satisfies ThreadSystemMessage;\n\n    default: {\n      const unsupportedRole: never = role;\n      throw new Error(`Unknown message role: ${unsupportedRole}`);\n    }\n  }\n};\n"],"mappings":";;;;AA+FA,MAAMgE,2BACJpC,MACAC,SACgC;CAChC,IAAI,CAACD,KAAKqC,WAAW,OAAO,GAAG,OAAOC,KAAAA;CACtC,OAAO;EAAEtC,MAAM;EAAQuC,MAAMvC,KAAKwC,UAAU,CAAC;EAAGvC;CAAK;AACvD;;;;AAKA,MAAawC,yBACXC,MACAC,YACAC,mBACkB;CAClB,MAAM,EAAEzC,MAAMkB,IAAIC,WAAWG,aAAaD,QAAQG,aAAae;CAC/D,MAAMG,SAAS;EACbxB,IAAIA,MAAMsB;EACVrB,WAAWA,6BAAa,IAAIC,KAAK;CACnC;CAEA,MAAMnB,UACJ,OAAOsC,KAAKtC,YAAY,WACpB,CAAC;EAAEJ,MAAM;EAAiB+C,MAAML,KAAKtC;CAAQ,CAAC,IAC9CsC,KAAKtC;CAEX,MAAM4C,wBAAwB,EAC5BC,OACA,GAAGC,WAC4C;EAC/C,IAAI,OAAOD,UAAU,UAAU,OAAO;EACtC,IAAI3E,aAAa2E,KAAK,CAAC,EAAEE,SAASd,WAAW,QAAQ,GACnD,OAAO;GAAE,GAAGa;GAAMD;EAAM;EAE1B,IAAI,uBAAuBG,KAAKH,KAAK,GACnC,OAAO;GAAE,GAAGC;GAAMD;EAAM;EAE1BI,QAAQC,KAAK,oCAAoC;EACjD,OAAO;CACT;CAEA,IAAInD,SAAS,UAAUsB,aAAa8B,QAClC,MAAM,IAAIC,MAAM,kDAAkD;CAEpE,IAAIrD,SAAS,eAAeqB,QAC1B,MAAM,IAAIgC,MAAM,iDAAiD;CAEnE,IAAIrD,SAAS,eAAewB,UAAUI,OACpC,MAAM,IAAIyB,MAAM,yDAAyD;CAE3E,QAAQrD,MAAR;EACE,KAAK,aACH,OAAO;GACL,GAAG0C;GACH1C;GACAC,SAASA,QACNqD,KAAKC,SAA4C;IAChD,MAAM1D,OAAO0D,KAAK1D;IAClB,QAAQA,MAAR;KACE,KAAK;MACH,IAAI,CAAC0D,KAAKX,MAAMY,KAAK,GAAG,OAAO;MAC/B,OAAOD;KAET,KAAK;MACH,IAAI,CAACA,KAAKX,MAAMY,KAAK,KAAK,CAACD,KAAKE,kBAAkBD,KAAK,GACrD,OAAO;MACT,OAAOD;KAET,KAAK;KACL,KAAK,UACH,OAAOA;KAET,KAAK,SACH,OAAOV,qBAAqBU,IAAI;KAElC,KAAK,QACH,OAAOA;KAET,KAAK,iBACH,OAAOA;KAET,KAAK,aAAa;MAChB,MAAM,EAAE9C,UAAUC,UAAU,GAAGgD,aAAaH;MAC5C,MAAMI,cAAc;OAClB,GAAGD;OACHxD,YAAYqD,KAAKrD,cAAc,QAAQhC,WAAW;OAClD,GAAIuC,aAAa0B,KAAAA,KAAa,EAAE1B,SAAS;OACzC,GAAIC,aAAayB,KAAAA,KAAa,EAAEzB,SAAS;MAC3C;MAEA,IAAI6C,KAAKnD,MACP,OAAO;OACL,GAAGuD;OACHvD,MAAMmD,KAAKnD;OACXC,UAAUkD,KAAKlD,YAAYuD,KAAKC,UAAUN,KAAKnD,IAAI;MACrD;MAEF,OAAO;OACL,GAAGuD;OACHvD,MAAMnC,uBAAuBsF,KAAKlD,YAAY,EAAE,KAAK,CAAC;OACtDA,UAAUkD,KAAKlD,YAAY;MAC7B;KACF;KAEA,SAAS;MACP,MAAMyD,YAAY7B,wBAChBpC,MACC0D,KAA0BzD,IAC7B;MACA,IAAIgE,WAAW,OAAOA;MACtB,MAAM,IAAIT,MACR,4CAA4CxD,MAC9C;KACF;IACF;GACF,CAAC,CAAC,CACDkE,QAAQC,MAAM,CAAC,CAACA,CAAC;GACpB3C,QAAQA,UAAUoB;GAClBjB,UAAU;IACRC,gBAAgBD,UAAUC,kBAAkB;IAC5CC,sBAAsBF,UAAUE,wBAAwB,CAAA;IACxDC,eAAeH,UAAUG,iBAAiB,CAAA;IAC1CI,QAAQP,UAAUO,UAAU,CAAC;IAC7BH,OAAOJ,UAAUI,SAAS,CAAA;IAC1B,GAAIJ,UAAUX,UAAU,EAAEA,QAAQW,SAASX,OAAO;IAClD,GAAIW,UAAUK,qBAAqB,EACjCA,mBAAmBL,SAASK,kBAC9B;IACA,GAAIL,UAAUM,gBAAgB,EAAEA,cAAc,KAAK;GACrD;EACF;EAEF,KAAK,QACH,OAAO;GACL,GAAGY;GACH1C;GACAC,SAASA,QAAQqD,KAAKC,SAAgC;IACpD,MAAM1D,OAAO0D,KAAK1D;IAClB,QAAQA,MAAR;KACE,KAAK;KACL,KAAK;KACL,KAAK;KACL,KAAK;KACL,KAAK,QACH,OAAO0D;KAET,SAAS;MACP,MAAMO,YAAY7B,wBAChBpC,MACC0D,KAA0BzD,IAC7B;MACA,IAAIgE,WAAW,OAAOA;MACtB,MAAM,IAAIT,MAAM,uCAAuCxD,MAAM;KAC/D;IACF;GACF,CAAC;GACDyB,cAAcA,eAAe,CAAA,EAAA,CAAIgC,KAAKW,SAAS;IAC7C,GAAGA;IACHhE,SAASgE,IAAIhE,QAAQqD,KAAKC,SAAgC;KAKxD,OAJkBtB,wBAChBsB,KAAK1D,MACJ0D,KAA0BzD,IAEtBgE,KAAcP;IACvB,CAAC;GACH,EAAE;GACF/B,UAAU;IACRO,QAAQP,UAAUO,UAAU,CAAC;IAC7B,GAAIP,UAAUM,gBAAgB,EAAEA,cAAc,KAAK;GACrD;EACF;EAEF,KAAK;GACH,IAAI7B,QAAQmD,WAAW,KAAKnD,QAAQ,EAAE,CAAEJ,SAAS,QAC/C,MAAM,IAAIwD,MACR,0DACF;GAEF,OAAO;IACL,GAAGX;IACH1C;IACSC;IACTuB,UAAU,EACRO,QAAQP,UAAUO,UAAU,CAAC,EAC/B;GACF;EAEF,SAEE,MAAM,IAAIsB,MAAM,yBAAyBa,MAAiB;CAE9D;AACF"}

package/dist/runtime/utils/tool-call-tree.d.ts

@@ -0,0 +1,39 @@
+import { ThreadAssistantMessagePart, ThreadMessage, ToolCallMessagePart } from "../../types/message.js";
+//#region src/runtime/utils/tool-call-tree.d.ts
+type ToolCallTreeEntry = {
+  readonly part: ToolCallMessagePart;
+  /** Id of the message whose content directly holds `part`. */
+  readonly messageId: string;
+};
+/**
+ * Walk every tool-call part reachable from `messages`. A tool call projects a
+ * child run as nested messages on `ToolCallMessagePart.messages`, so a
+ * subagent's own calls live below the top-level content and a seam that only
+ * scans the top level cannot reach them.
+ *
+ * Parts arrive in document order, each one ahead of its own descendants.
+ * `messageId` names the message that directly holds the part, which for a
+ * nested call is the child run's message rather than the top-level message the
+ * tree hangs from. Only assistant messages are descended, the same rule
+ * {@link mapToolCallPartsDeep} rewrites under, so a part this reports is always
+ * a part that can be written back.
+ */
+declare function walkToolCallTree(messages: readonly ThreadMessage[]): Generator<ToolCallTreeEntry>;
+/**
+ * Content-level form of {@link walkToolCallTree}, for a caller that already
+ * holds one assistant message's content and does not need the owning id.
+ */
+declare function iterateToolCallParts(content: readonly ThreadAssistantMessagePart[]): Generator<ToolCallMessagePart>;
+/**
+ * Rebuild `content` with `fn` applied to every tool-call part in the tree.
+ * Arrays and nested messages are reused wherever `fn` returned the part it was
+ * given, so an unchanged subtree keeps its identity and `changed` reports
+ * whether anything moved.
+ */
+declare function mapToolCallPartsDeep(content: readonly ThreadAssistantMessagePart[], fn: (part: ToolCallMessagePart) => ToolCallMessagePart): {
+  content: readonly ThreadAssistantMessagePart[];
+  changed: boolean;
+};
+//#endregion
+export { ToolCallTreeEntry, iterateToolCallParts, mapToolCallPartsDeep, walkToolCallTree };
+//# sourceMappingURL=tool-call-tree.d.ts.map

package/dist/runtime/utils/tool-call-tree.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-call-tree.d.ts","names":[],"sources":["../../../src/runtime/utils/tool-call-tree.ts"],"mappings":";;KAOY;WACD,MAAM;;WAEN;;;;;;;;;;;;;;;iBAgBM,iBACf,mBAAmB,kBAClB,UAAU;;;;;iBAiBI,qBACf,kBAAkB,+BACjB,UAAU;;;;;;;iBAkBG,qBACd,kBAAkB,8BAClB,KAAK,MAAM,wBAAwB;EAChC,kBAAkB;EAA8B"}

package/dist/runtime/utils/tool-call-tree.js

@@ -0,0 +1,85 @@
+//#region src/runtime/utils/tool-call-tree.ts
+/**
+* Walk every tool-call part reachable from `messages`. A tool call projects a
+* child run as nested messages on `ToolCallMessagePart.messages`, so a
+* subagent's own calls live below the top-level content and a seam that only
+* scans the top level cannot reach them.
+*
+* Parts arrive in document order, each one ahead of its own descendants.
+* `messageId` names the message that directly holds the part, which for a
+* nested call is the child run's message rather than the top-level message the
+* tree hangs from. Only assistant messages are descended, the same rule
+* {@link mapToolCallPartsDeep} rewrites under, so a part this reports is always
+* a part that can be written back.
+*/
+function* walkToolCallTree(messages) {
+	for (const message of messages) {
+		if (message?.role !== "assistant" || !Array.isArray(message.content)) continue;
+		for (const part of message.content) {
+			if (!part || part.type !== "tool-call") continue;
+			yield {
+				part,
+				messageId: message.id
+			};
+			if (part.messages?.length) yield* walkToolCallTree(part.messages);
+		}
+	}
+}
+/**
+* Content-level form of {@link walkToolCallTree}, for a caller that already
+* holds one assistant message's content and does not need the owning id.
+*/
+function* iterateToolCallParts(content) {
+	for (const part of content) {
+		if (!part || part.type !== "tool-call") continue;
+		yield part;
+		if (part.messages?.length) for (const entry of walkToolCallTree(part.messages)) yield entry.part;
+	}
+}
+/**
+* Rebuild `content` with `fn` applied to every tool-call part in the tree.
+* Arrays and nested messages are reused wherever `fn` returned the part it was
+* given, so an unchanged subtree keeps its identity and `changed` reports
+* whether anything moved.
+*/
+function mapToolCallPartsDeep(content, fn) {
+	let changed = false;
+	const next = content.map((part) => {
+		if (part.type !== "tool-call") return part;
+		let mapped = fn(part);
+		if (mapped.messages !== void 0) {
+			let nestedChanged = false;
+			const nestedMessages = mapped.messages.map((nested) => {
+				if (nested.role !== "assistant" || !Array.isArray(nested.content)) return nested;
+				const assistant = nested;
+				const result = mapToolCallPartsDeep(assistant.content, fn);
+				if (!result.changed) return nested;
+				nestedChanged = true;
+				return {
+					...assistant,
+					content: result.content
+				};
+			});
+			if (nestedChanged) mapped = mapped === part ? {
+				...part,
+				messages: nestedMessages
+			} : {
+				...mapped,
+				messages: nestedMessages
+			};
+		}
+		if (mapped !== part) changed = true;
+		return mapped;
+	});
+	return changed ? {
+		content: next,
+		changed
+	} : {
+		content,
+		changed
+	};
+}
+//#endregion
+export { iterateToolCallParts, mapToolCallPartsDeep, walkToolCallTree };
+
+//# sourceMappingURL=tool-call-tree.js.map

package/dist/runtime/utils/tool-call-tree.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool-call-tree.js","names":["ThreadAssistantMessage","ThreadAssistantMessagePart","ThreadMessage","ToolCallMessagePart","ToolCallTreeEntry","part","messageId","walkToolCallTree","messages","Generator","message","role","Array","isArray","content","type","id","length","iterateToolCallParts","entry","mapToolCallPartsDeep","fn","changed","next","map","mapped","undefined","nestedChanged","nestedMessages","nested","assistant","result"],"sources":["../../../src/runtime/utils/tool-call-tree.ts"],"sourcesContent":["import type {\n  ThreadAssistantMessage,\n  ThreadAssistantMessagePart,\n  ThreadMessage,\n  ToolCallMessagePart,\n} from \"../../types/message\";\n\nexport type ToolCallTreeEntry = {\n  readonly part: ToolCallMessagePart;\n  /** Id of the message whose content directly holds `part`. */\n  readonly messageId: string;\n};\n\n/**\n * Walk every tool-call part reachable from `messages`. A tool call projects a\n * child run as nested messages on `ToolCallMessagePart.messages`, so a\n * subagent's own calls live below the top-level content and a seam that only\n * scans the top level cannot reach them.\n *\n * Parts arrive in document order, each one ahead of its own descendants.\n * `messageId` names the message that directly holds the part, which for a\n * nested call is the child run's message rather than the top-level message the\n * tree hangs from. Only assistant messages are descended, the same rule\n * {@link mapToolCallPartsDeep} rewrites under, so a part this reports is always\n * a part that can be written back.\n */\nexport function* walkToolCallTree(\n  messages: readonly ThreadMessage[],\n): Generator<ToolCallTreeEntry> {\n  for (const message of messages) {\n    if (message?.role !== \"assistant\" || !Array.isArray(message.content)) {\n      continue;\n    }\n    for (const part of message.content) {\n      if (!part || part.type !== \"tool-call\") continue;\n      yield { part, messageId: message.id };\n      if (part.messages?.length) yield* walkToolCallTree(part.messages);\n    }\n  }\n}\n\n/**\n * Content-level form of {@link walkToolCallTree}, for a caller that already\n * holds one assistant message's content and does not need the owning id.\n */\nexport function* iterateToolCallParts(\n  content: readonly ThreadAssistantMessagePart[],\n): Generator<ToolCallMessagePart> {\n  for (const part of content) {\n    if (!part || part.type !== \"tool-call\") continue;\n    yield part;\n    if (part.messages?.length) {\n      for (const entry of walkToolCallTree(part.messages)) {\n        yield entry.part;\n      }\n    }\n  }\n}\n\n/**\n * Rebuild `content` with `fn` applied to every tool-call part in the tree.\n * Arrays and nested messages are reused wherever `fn` returned the part it was\n * given, so an unchanged subtree keeps its identity and `changed` reports\n * whether anything moved.\n */\nexport function mapToolCallPartsDeep(\n  content: readonly ThreadAssistantMessagePart[],\n  fn: (part: ToolCallMessagePart) => ToolCallMessagePart,\n): { content: readonly ThreadAssistantMessagePart[]; changed: boolean } {\n  let changed = false;\n  const next = content.map((part): ThreadAssistantMessagePart => {\n    if (part.type !== \"tool-call\") return part;\n    let mapped = fn(part);\n    if (mapped.messages !== undefined) {\n      let nestedChanged = false;\n      const nestedMessages = mapped.messages.map((nested) => {\n        if (nested.role !== \"assistant\" || !Array.isArray(nested.content)) {\n          return nested;\n        }\n        const assistant = nested as ThreadAssistantMessage;\n        const result = mapToolCallPartsDeep(assistant.content, fn);\n        if (!result.changed) return nested;\n        nestedChanged = true;\n        return { ...assistant, content: result.content };\n      });\n      if (nestedChanged) {\n        mapped =\n          mapped === part\n            ? { ...part, messages: nestedMessages }\n            : { ...mapped, messages: nestedMessages };\n      }\n    }\n    if (mapped !== part) changed = true;\n    return mapped;\n  });\n  return changed ? { content: next, changed } : { content, changed };\n}\n"],"mappings":";;;;;;;;;;;;;;AA0BA,UAAiBO,iBACfC,UAC8B;CAC9B,KAAK,MAAME,WAAWF,UAAU;EAC9B,IAAIE,SAASC,SAAS,eAAe,CAACC,MAAMC,QAAQH,QAAQI,OAAO,GACjE;EAEF,KAAK,MAAMT,QAAQK,QAAQI,SAAS;GAClC,IAAI,CAACT,QAAQA,KAAKU,SAAS,aAAa;GACxC,MAAM;IAAEV;IAAMC,WAAWI,QAAQM;GAAG;GACpC,IAAIX,KAAKG,UAAUS,QAAQ,OAAOV,iBAAiBF,KAAKG,QAAQ;EAClE;CACF;AACF;;;;;AAMA,UAAiBU,qBACfJ,SACgC;CAChC,KAAK,MAAMT,QAAQS,SAAS;EAC1B,IAAI,CAACT,QAAQA,KAAKU,SAAS,aAAa;EACxC,MAAMV;EACN,IAAIA,KAAKG,UAAUS,QACjB,KAAK,MAAME,SAASZ,iBAAiBF,KAAKG,QAAQ,GAChD,MAAMW,MAAMd;CAGlB;AACF;;;;;;;AAQA,SAAgBe,qBACdN,SACAO,IACsE;CACtE,IAAIC,UAAU;CACd,MAAMC,OAAOT,QAAQU,KAAKnB,SAAqC;EAC7D,IAAIA,KAAKU,SAAS,aAAa,OAAOV;EACtC,IAAIoB,SAASJ,GAAGhB,IAAI;EACpB,IAAIoB,OAAOjB,aAAakB,KAAAA,GAAW;GACjC,IAAIC,gBAAgB;GACpB,MAAMC,iBAAiBH,OAAOjB,SAASgB,KAAKK,WAAW;IACrD,IAAIA,OAAOlB,SAAS,eAAe,CAACC,MAAMC,QAAQgB,OAAOf,OAAO,GAC9D,OAAOe;IAET,MAAMC,YAAYD;IAClB,MAAME,SAASX,qBAAqBU,UAAUhB,SAASO,EAAE;IACzD,IAAI,CAACU,OAAOT,SAAS,OAAOO;IAC5BF,gBAAgB;IAChB,OAAO;KAAE,GAAGG;KAAWhB,SAASiB,OAAOjB;IAAQ;GACjD,CAAC;GACD,IAAIa,eACFF,SACEA,WAAWpB,OACP;IAAE,GAAGA;IAAMG,UAAUoB;GAAe,IACpC;IAAE,GAAGH;IAAQjB,UAAUoB;GAAe;EAEhD;EACA,IAAIH,WAAWpB,MAAMiB,UAAU;EAC/B,OAAOG;CACT,CAAC;CACD,OAAOH,UAAU;EAAER,SAASS;EAAMD;CAAQ,IAAI;EAAER;EAASQ;CAAQ;AACnE"}

package/dist/runtimes/external-store/external-store-adapter.d.ts

@@ -1,10 +1,10 @@
-import { AppendMessage, ThreadMessage } from "../../types/message.js";
+import { AppendMessage, ThreadMessage, ToolCallMessagePart } from "../../types/message.js";
 import { AttachmentAdapter } from "../../adapters/attachment.js";
 import { FeedbackAdapter } from "../../adapters/feedback.js";
 import { DictationAdapter, SpeechSynthesisAdapter } from "../../adapters/speech.js";
 import { RealtimeVoiceAdapter } from "../../adapters/voice.js";
 import { ThreadMessageLike } from "../../runtime/utils/thread-message-like.js";
-import { ExportedMessageRepository } from "../../runtime/utils/message-repository.js";
+import { ExportedMessageRepository, MessageRepository } from "../../runtime/utils/message-repository.js";
 import { ExternalThreadQueueAdapter } from "../../runtime/queue/external-thread-queue-adapter.js";
 import { AddToolResultOptions, RespondToToolApprovalOptions, ResumeRunConfig, StartRunConfig, ThreadSuggestion } from "../../runtime/interfaces/thread-runtime-core.js";
 import { ToolExecutionStatus } from "../tool-invocations/ToolInvocationTracker.js";
@@ -80,6 +80,15 @@ type ExternalStoreAdapterBase<T> = {
   isLoading?: boolean | undefined;
   messages?: readonly T[];
   messageRepository?: ExportedMessageRepository;
+  /**
+   * An externally owned message repository instance. When provided, the
+   * thread runtime adopts it as its branch store and swaps to it atomically
+   * whenever a different instance is passed, so hosts that route multiple
+   * conversations through one runtime keep each conversation's history and
+   * branches isolated in its own instance. Omit it to keep the runtime's own
+   * repository.
+   */
+  unstable_messageRepositoryInstance?: MessageRepository | undefined;
   suggestions?: readonly ThreadSuggestion[] | undefined;
   state?: ReadonlyJSONValue | undefined;
   extras?: unknown;
@@ -170,6 +179,24 @@ type ExternalStoreAdapterBase<T> = {
    * `modelContent` populated when present.
    */
   unstable_enableToolInvocations?: boolean | undefined;
+  /**
+   * Decides whether a tool call's result is produced on the client. Only
+   * consulted when `unstable_enableToolInvocations` is `true`.
+   *
+   * A provider that runs tools itself answers its own calls, and its result
+   * arrives one or more snapshots after the call's arguments complete. In
+   * that window the call is complete and result-less, so a registered tool
+   * of the same name would otherwise execute locally and produce a result
+   * the provider never asked for. An adapter that can tell the two apart
+   * supplies this predicate; it is read once per tool call, when the call is
+   * first observed live.
+   *
+   * The predicate is also what licenses running a frontend tool while the
+   * provider's run is still open. Without it, ownership is unknown until the
+   * run ends, so a registered tool executes only once the run's outcome is
+   * known and cannot fire on a call the provider was about to answer or gate.
+   */
+  unstable_isClientToolCall?: ((toolCall: ToolCallMessagePart) => boolean) | undefined;
   /**
    * Receives the current per-tool-call execution status map whenever it
    * changes. Only invoked when `unstable_enableToolInvocations` is `true`

package/dist/runtimes/external-store/external-store-adapter.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"external-store-adapter.d.ts","names":[],"sources":["../../../src/runtimes/external-store/external-store-adapter.ts"],"mappings":";;;;;;;;;;;;KAqBY,wBAAwB;EAClC,QAAQ;EACR;EACA;EACA;EACA;EACA,SAAS;;KAGC;;;;EAIV;EACA;EACA,mBAAmB;EACnB,2BAA2B;;;;EAI3B,6BAA6B;;;;EAI7B,qBAAqB,qBAAqB;EAC1C,YACE,kBACA,sBACI;EACN,mBAEM,kBACA,QAAQ,wCACL;EAET,cAAc,qBAAqB;EACnC,gBAAgB,qBAAqB;EACrC,aAAa,qBAAqB;;KAGxB,8BAA8B,MACxC,SAAS,GACT,gBACG;;;;KAKO;EACV;EACA;;KAGG,qCAAqC;EACxC,gBAAgB,8BAA8B;;KAG3C,yBAAyB;;;;;;;EAO5B;;;;;;;;;EASA;;;;;;;;;EASA;EACA;EACA,oBAAoB;EACpB,oBAAoB;EACpB,uBAAuB;EACvB,QAAQ;EACR;;;;;;;;EASA,gBAAgB,mBAAmB;;;;;;;;;;;;;;;;;;EAkBnC,4BACM,OAAO;EAEb,aAAa,mBAAmB;EAChC;EACA,wBAAwB;EACxB,QAAQ,SAAS,kBAAkB;;EAEnC,QAAQ;EACR,WAAW,SAAS,kBAAkB;EACtC,aAAa,sBAAsB;EACnC,aACM,yBAAyB,QAAQ,mBAAmB;EAE1D,aAAa,QAAQ,oBAAoB;EACzC,kBAAkB;;;;;;;;;EASlB,yBAAyB;EACzB,oBACM,SAAS,yBAAyB;EAExC,qBACM;IAAW;IAAoB;;EAErC,4BACM,SAAS,iCAAiC;EAEhD,iBAAiB,8BAA8B;EAC/C;IAEM,cAAc;IACd,SAAS;IACT,YAAY;IACZ,QAAQ;IACR,WAAW;;;;IAIX,aAAa;;EAGnB;IAEM;;;;;;;;;;;;;;;;;;;;;EAsBN;;;;;;;;;;EAUA,oBACM,UAAU,eAAe;;KAIrB,qBAAqB,IAAI,iBACnC,yBAAyB,MACtB,UAAU,yBAEP,qCAAqC"}
+{"version":3,"file":"external-store-adapter.d.ts","names":[],"sources":["../../../src/runtimes/external-store/external-store-adapter.ts"],"mappings":";;;;;;;;;;;;KA4BY,wBAAwB;EAClC,QAAQ;EACR;EACA;EACA;EACA;EACA,SAAS;;KAGC;;;;EAIV;EACA;EACA,mBAAmB;EACnB,2BAA2B;;;;EAI3B,6BAA6B;;;;EAI7B,qBAAqB,qBAAqB;EAC1C,YACE,kBACA,sBACI;EACN,mBAEM,kBACA,QAAQ,wCACL;EAET,cAAc,qBAAqB;EACnC,gBAAgB,qBAAqB;EACrC,aAAa,qBAAqB;;KAGxB,8BAA8B,MACxC,SAAS,GACT,gBACG;;;;KAKO;EACV;EACA;;KAGG,qCAAqC;EACxC,gBAAgB,8BAA8B;;KAG3C,yBAAyB;;;;;;;EAO5B;;;;;;;;;EASA;;;;;;;;;EASA;EACA;EACA,oBAAoB;EACpB,oBAAoB;;;;;;;;;EASpB,qCAAqC;EACrC,uBAAuB;EACvB,QAAQ;EACR;;;;;;;;EASA,gBAAgB,mBAAmB;;;;;;;;;;;;;;;;;;EAkBnC,4BACM,OAAO;EAEb,aAAa,mBAAmB;EAChC;EACA,wBAAwB;EACxB,QAAQ,SAAS,kBAAkB;;EAEnC,QAAQ;EACR,WAAW,SAAS,kBAAkB;EACtC,aAAa,sBAAsB;EACnC,aACM,yBAAyB,QAAQ,mBAAmB;EAE1D,aAAa,QAAQ,oBAAoB;EACzC,kBAAkB;;;;;;;;;EASlB,yBAAyB;EACzB,oBACM,SAAS,yBAAyB;EAExC,qBACM;IAAW;IAAoB;;EAErC,4BACM,SAAS,iCAAiC;EAEhD,iBAAiB,8BAA8B;EAC/C;IAEM,cAAc;IACd,SAAS;IACT,YAAY;IACZ,QAAQ;IACR,WAAW;;;;IAIX,aAAa;;EAGnB;IAEM;;;;;;;;;;;;;;;;;;;;;EAsBN;;;;;;;;;;;;;;;;;;EAkBA,8BACM,UAAU;;;;;;;;;;EAWhB,oBACM,UAAU,eAAe;;KAIrB,qBAAqB,IAAI,iBACnC,yBAAyB,MACtB,UAAU,yBAEP,qCAAqC"}

package/dist/runtimes/external-store/external-store-thread-list-runtime-core.d.ts

@@ -1,10 +1,10 @@
-import { Unsubscribe } from "../../types/unsubscribe.js";
 import { ThreadListItemCoreState, ThreadListRuntimeCore } from "../../runtime/interfaces/thread-list-runtime-core.js";
+import { BaseSubscribable } from "../../subscribable/subscribable.js";
 import { ExternalStoreThreadListAdapter } from "./external-store-adapter.js";
 import { ExternalStoreThreadRuntimeCore } from "./external-store-thread-runtime-core.js";
 //#region src/runtimes/external-store/external-store-thread-list-runtime-core.d.ts
 type ExternalStoreThreadFactory = () => ExternalStoreThreadRuntimeCore;
-declare class ExternalStoreThreadListRuntimeCore implements ThreadListRuntimeCore {
+declare class ExternalStoreThreadListRuntimeCore extends BaseSubscribable implements ThreadListRuntimeCore {
   private _mainThreadId;
   private _threads;
   private _archivedThreads;
@@ -40,9 +40,6 @@ declare class ExternalStoreThreadListRuntimeCore implements ThreadListRuntimeCor
     externalId: string | undefined;
   }>;
   generateTitle(): never;
-  private _subscriptions;
-  subscribe(callback: () => void): Unsubscribe;
-  private _notifySubscribers;
 }
 //#endregion
 export { ExternalStoreThreadFactory, ExternalStoreThreadListRuntimeCore };

package/dist/runtimes/external-store/external-store-thread-list-runtime-core.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"external-store-thread-list-runtime-core.d.ts","names":[],"sources":["../../../src/runtimes/external-store/external-store-thread-list-runtime-core.ts"],"mappings":";;;;;KAUY,mCAAmC;cAgBlC,8CAA8C;UACjD;UACA;UACA;UACA;UAEA;MAEG;MAIA;MAIA;MAIA;MAIA,eAAW,SAAA,eAAA;EAIf,yBAAqB;UAIpB;MAEG;UAIH;EAGN,YAAA,SAAS,4CACT,eAAe;EAMV,4BAAwB;EAIxB;EAIA,YAAY,mBAAgB;EAI5B,sBACL,SAAS,gCACT;EAsFW,oBAAoB;EASpB,eACX,kBACA;IAAa;MACZ;EAUU,qBAAqB;EAUrB,OAAO,kBAAkB,mBAAmB;EAQ5C,aACX,kBACA,QAAQ,sCACP;EAUU,UAAU;EAIV,QAAQ,mBAAmB;EAQ3B,UAAU,mBAAmB;EAQ7B,OAAO,mBAAmB;EAQhC,WACL,mBACC;IAAU;IAAkB;;EAIxB;UAIC;EAED,UAAU,uBAAuB;UAKhC"}
+{"version":3,"file":"external-store-thread-list-runtime-core.d.ts","names":[],"sources":["../../../src/runtimes/external-store/external-store-thread-list-runtime-core.ts"],"mappings":";;;;;KASY,mCAAmC;cAgBlC,2CACH,4BACG;UAEH;UACA;UACA;UACA;UAEA;MAEG;MAIA;MAIA;MAIA;MAIA,eAAW,SAAA,eAAA;EAIf,yBAAqB;UAIpB;MAEG;UAIH;EAGN,YAAA,SAAS,4CACT,eAAe;EAOV,4BAAwB;EAIxB;EAIA,YAAY,mBAAgB;EAI5B,sBACL,SAAS,gCACT;EAsFW,oBAAoB;EASpB,eACX,kBACA;IAAa;MACZ;EAUU,qBAAqB;EAUrB,OAAO,kBAAkB,mBAAmB;EAQ5C,aACX,kBACA,QAAQ,sCACP;EAUU,UAAU;EAIV,QAAQ,mBAAmB;EAQ3B,UAAU,mBAAmB;EAQ7B,OAAO,mBAAmB;EAQhC,WACL,mBACC;IAAU;IAAkB;;EAIxB"}