@assistant-ui/ai-sdk 0.0.1

package/dist/runtime/useExternalHistory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useExternalHistory.js","names":["AssistantRuntime","ThreadHistoryAdapter","ThreadMessage","MessageFormatAdapter","MessageFormatRepository","ExportedMessageRepository","getExternalStoreMessages","MessageRepository","useAui","useRef","useEffect","useState","RefObject","useCallback","useMemo","toExportedMessageRepository","toThreadMessages","messages","TMessage","survivingIds","Set","survivors","flatMap","m","message","console","warn","parentId","has","add","id","headId","isAwaitingToolApproval","status","type","reason","snapshotExternalMessages","Map","map","useExternalHistory","runtimeRef","historyAdapter","storageFormatAdapter","onSetMessages","loadedRef","itemEpoch","setItemEpoch","aui","optionalThreadListItem","threadListItem","source","hasLoaded","setHasLoaded","historyIds","persistedInnerIds","deferredTelemetryIds","persistedExternalMessages","onSetMessagesRef","current","formatAdapter","undefined","withFormat","Error","isLoading","loadHistory","repo","load","length","getId","converted","thread","import","tempRepo","getMessages","error","pin","remoteId","getState","subscribe","n","threadState","isRunning","runStartRef","persistTimerRef","ReturnType","setTimeout","persistInFlightRef","Promise","resolve","stepBoundariesRef","wasRunningRef","toolCallCountRef","adapter","unsubscribe","wasRunning","lastMsg","at","role","currentToolCallCount","content","filter","p","push","Date","now","clearTimeout","persistSettled","ignoreRunning","latest","boundaries","durationMs","lastAssistant","findLast","tcCount","totalSteps","stepDur","i","Math","round","stepTimestamps","endMs","start_ms","end_ms","telemetryOptions","then","changedRunMessageIds","externalMessages","previous","get","some","item","index","lastInnerMessageId","failedUpdateIds","getLastInnerId","msgs","toBatchItems","msg","idx","innerMessages","isTerminal","isAwaitingToolCalls","isReady","update","isPersistedMessage","batchItems","innerId","append","delete","reportTelemetry","nextSnapshot","catch","deleteMessage","messageId","deleteMessages","bind","messageIndex","findIndex","previousInnerMessages","slice","itemsToDelete","deletion"],"sources":["../../src/runtime/useExternalHistory.ts"],"sourcesContent":["\"use client\";\n\nimport type {\n  AssistantRuntime,\n  ThreadHistoryAdapter,\n  ThreadMessage,\n  MessageFormatAdapter,\n  MessageFormatRepository,\n  ExportedMessageRepository,\n} from \"@assistant-ui/core\";\nimport { getExternalStoreMessages } from \"@assistant-ui/core\";\nimport { MessageRepository } from \"@assistant-ui/core/internal\";\nimport { useAui } from \"@assistant-ui/store\";\nimport {\n  useRef,\n  useEffect,\n  useState,\n  type RefObject,\n  useCallback,\n  useMemo,\n} from \"react\";\n\nexport const toExportedMessageRepository = <TMessage>(\n  toThreadMessages: (messages: TMessage[]) => ThreadMessage[],\n  messages: MessageFormatRepository<TMessage>,\n): ExportedMessageRepository => {\n  const survivingIds = new Set<string>();\n  const survivors = messages.messages.flatMap((m) => {\n    const message = toThreadMessages([m.message])[0];\n    if (!message) {\n      console.warn(\"Skipping a stored message that could not be loaded.\");\n      return [];\n    }\n    if (m.parentId && !survivingIds.has(m.parentId)) return [];\n    survivingIds.add(message.id);\n    return [{ ...m, message }];\n  });\n\n  return {\n    headId:\n      messages.headId && survivingIds.has(messages.headId)\n        ? messages.headId\n        : null,\n    messages: survivors,\n  };\n};\n\nconst isAwaitingToolApproval = (message: ThreadMessage) =>\n  message.status?.type === \"requires-action\" &&\n  message.status.reason === \"tool-calls\";\n\nconst snapshotExternalMessages = <TMessage>(\n  messages: readonly ThreadMessage[],\n) =>\n  new Map<string, TMessage[]>(\n    messages.map((message) => [\n      message.id,\n      [...getExternalStoreMessages<TMessage>(message)],\n    ]),\n  );\n\nexport const useExternalHistory = <TMessage>(\n  runtimeRef: RefObject<AssistantRuntime>,\n  historyAdapter: ThreadHistoryAdapter | undefined,\n  toThreadMessages: (messages: TMessage[]) => ThreadMessage[],\n  storageFormatAdapter: MessageFormatAdapter<TMessage, any>,\n  onSetMessages: (messages: TMessage[]) => void,\n) => {\n  const loadedRef = useRef(false);\n  const [itemEpoch, setItemEpoch] = useState(0);\n\n  const aui = useAui();\n  const optionalThreadListItem = useCallback(\n    () => (aui.threadListItem.source ? aui.threadListItem : null),\n    [aui],\n  );\n\n  const [hasLoaded, setHasLoaded] = useState(false);\n\n  const historyIds = useRef(new Set<string>());\n  const persistedInnerIds = useRef(new Set<string>());\n  const deferredTelemetryIds = useRef(new Set<string>());\n  const persistedExternalMessages = useRef(new Map<string, TMessage[]>());\n\n  const onSetMessagesRef = useRef(onSetMessages);\n  useEffect(() => {\n    onSetMessagesRef.current = onSetMessages;\n  });\n\n  const formatAdapter = useMemo(() => {\n    if (!historyAdapter) return undefined;\n    if (!historyAdapter.withFormat) {\n      throw new Error(\n        \"useAISDKRuntime: ThreadHistoryAdapter is missing the required `withFormat` method.\",\n      );\n    }\n    return historyAdapter.withFormat<TMessage, any>(storageFormatAdapter);\n  }, [historyAdapter, storageFormatAdapter]);\n\n  const isLoading = formatAdapter != null && !hasLoaded;\n\n  useEffect(() => {\n    if (!formatAdapter || loadedRef.current) return undefined;\n\n    const loadHistory = async () => {\n      try {\n        const repo = await formatAdapter.load();\n        if (repo && repo.messages.length > 0) {\n          for (const m of repo.messages) {\n            persistedInnerIds.current.add(\n              storageFormatAdapter.getId(m.message),\n            );\n          }\n          const converted = toExportedMessageRepository(toThreadMessages, repo);\n          runtimeRef.current.thread.import(converted);\n\n          const tempRepo = new MessageRepository();\n          tempRepo.import(converted);\n          const messages = tempRepo.getMessages();\n\n          onSetMessagesRef.current(\n            messages.flatMap(getExternalStoreMessages<TMessage>),\n          );\n\n          historyIds.current = new Set();\n          for (const m of converted.messages) {\n            historyIds.current.add(m.message.id);\n            if (isAwaitingToolApproval(m.message)) {\n              deferredTelemetryIds.current.add(m.message.id);\n            }\n          }\n          persistedExternalMessages.current =\n            snapshotExternalMessages<TMessage>(\n              converted.messages.map((m) => m.message),\n            );\n        }\n      } catch (error) {\n        console.error(\"Failed to load message history:\", error);\n      } finally {\n        setHasLoaded(true);\n      }\n    };\n\n    formatAdapter.pin?.();\n\n    const remoteId = optionalThreadListItem()?.getState().remoteId;\n    if (!remoteId) {\n      setHasLoaded(true);\n      return aui.subscribe(() => {\n        if (optionalThreadListItem()?.getState().remoteId) {\n          setItemEpoch((n) => n + 1);\n        }\n      });\n    }\n\n    const threadState = runtimeRef.current.thread.getState();\n    if (threadState.isRunning || threadState.messages.length > 0) {\n      loadedRef.current = true;\n      setHasLoaded(true);\n      return undefined;\n    }\n\n    loadedRef.current = true;\n    void loadHistory();\n    return undefined;\n  }, [\n    formatAdapter,\n    toThreadMessages,\n    runtimeRef,\n    optionalThreadListItem,\n    aui,\n    itemEpoch,\n    storageFormatAdapter,\n  ]);\n\n  const runStartRef = useRef<number | null>(null);\n  const persistTimerRef = useRef<ReturnType<typeof setTimeout> | null>(null);\n  const persistInFlightRef = useRef<Promise<void>>(Promise.resolve());\n  const stepBoundariesRef = useRef<number[]>([]);\n  const wasRunningRef = useRef(false);\n  const toolCallCountRef = useRef(0);\n\n  useEffect(() => {\n    if (!formatAdapter) return;\n    const adapter = formatAdapter;\n\n    const unsubscribe = runtimeRef.current.thread.subscribe(() => {\n      const threadState = runtimeRef.current.thread.getState();\n      const { isRunning } = threadState;\n      const wasRunning = wasRunningRef.current;\n      wasRunningRef.current = isRunning;\n\n      // Track step boundaries by content changes (more reliable than isRunning)\n      if (runStartRef.current != null) {\n        const lastMsg = threadState.messages.at(-1);\n        if (lastMsg?.role === \"assistant\") {\n          const currentToolCallCount = lastMsg.content.filter(\n            (p) => p.type === \"tool-call\",\n          ).length;\n          while (toolCallCountRef.current < currentToolCallCount) {\n            stepBoundariesRef.current.push(Date.now() - runStartRef.current);\n            toolCallCountRef.current++;\n          }\n        }\n      }\n\n      if (isRunning) {\n        if (runStartRef.current == null) {\n          runStartRef.current = Date.now();\n          stepBoundariesRef.current = [];\n          toolCallCountRef.current = 0;\n          adapter.pin?.();\n        }\n        // Cancel any pending persist — isRunning went back to true\n        if (persistTimerRef.current) {\n          clearTimeout(persistTimerRef.current);\n          persistTimerRef.current = null;\n        }\n        return;\n      }\n\n      // Only act on the true→false transition\n      if (!wasRunning) return;\n\n      // Record step boundary offset (synchronous for accuracy)\n      if (runStartRef.current != null) {\n        stepBoundariesRef.current.push(Date.now() - runStartRef.current);\n      }\n\n      // Debounce: wait one macrotask so agentic step flickers are absorbed\n      if (persistTimerRef.current) clearTimeout(persistTimerRef.current);\n      persistTimerRef.current = setTimeout(() => {\n        persistSettled(false);\n      }, 0);\n    });\n\n    function persistSettled(ignoreRunning: boolean) {\n      persistTimerRef.current = null;\n      const latest = runtimeRef.current.thread.getState();\n      if (!ignoreRunning && latest.isRunning) return;\n\n      const boundaries = stepBoundariesRef.current;\n      const durationMs = boundaries.length > 0 ? boundaries.at(-1) : undefined;\n\n      // Fallback: if only 1 boundary but message has multiple steps, distribute evenly\n      if (boundaries.length === 1 && durationMs != null) {\n        const lastAssistant = latest.messages.findLast(\n          (m) => m.role === \"assistant\",\n        );\n        if (lastAssistant) {\n          const tcCount = lastAssistant.content.filter(\n            (p) => p.type === \"tool-call\",\n          ).length;\n          if (tcCount > 0) {\n            const totalSteps = tcCount + 1;\n            const stepDur = durationMs / totalSteps;\n            boundaries.length = 0;\n            for (let i = 0; i < totalSteps; i++) {\n              boundaries.push(Math.round((i + 1) * stepDur));\n            }\n          }\n        }\n      }\n\n      // Build per-step timestamps when there are multiple steps\n      const stepTimestamps =\n        boundaries.length > 1\n          ? boundaries.map((endMs, i) => ({\n              start_ms: i === 0 ? 0 : boundaries[i - 1]!,\n              end_ms: endMs,\n            }))\n          : undefined;\n\n      runStartRef.current = null;\n      stepBoundariesRef.current = [];\n\n      const telemetryOptions = {\n        ...(durationMs != null ? { durationMs } : undefined),\n        ...(stepTimestamps != null ? { stepTimestamps } : undefined),\n      };\n\n      persistInFlightRef.current = persistInFlightRef.current\n        .then(async () => {\n          const changedRunMessageIds = new Set<string>();\n          for (const message of latest.messages) {\n            const externalMessages =\n              getExternalStoreMessages<TMessage>(message);\n            const previous = persistedExternalMessages.current.get(message.id);\n            if (\n              previous === undefined ||\n              previous.length !== externalMessages.length ||\n              externalMessages.some((item, index) => item !== previous[index])\n            ) {\n              changedRunMessageIds.add(message.id);\n            }\n          }\n\n          const { messages } = latest;\n          let lastInnerMessageId: string | null = null;\n          const failedUpdateIds = new Set<string>();\n\n          const getLastInnerId = (msgs: TMessage[]): string | null =>\n            msgs.length > 0 ? storageFormatAdapter.getId(msgs.at(-1)!) : null;\n\n          const toBatchItems = (msgs: TMessage[]) =>\n            msgs.map((msg, idx) => ({\n              parentId:\n                idx === 0\n                  ? lastInnerMessageId\n                  : storageFormatAdapter.getId(msgs[idx - 1]!),\n              message: msg,\n            }));\n\n          for (const message of messages) {\n            const innerMessages = getExternalStoreMessages<TMessage>(message);\n\n            const isTerminal =\n              message.status === undefined ||\n              message.status.type === \"complete\" ||\n              message.status.type === \"incomplete\";\n            const isAwaitingToolCalls = isAwaitingToolApproval(message);\n            // A paused message's later content can only reach storage via update, so it is persisted early only when the adapter supports update.\n            const isReady =\n              isTerminal ||\n              (isAwaitingToolCalls && adapter.update !== undefined);\n\n            if (!isReady) {\n              lastInnerMessageId =\n                getLastInnerId(innerMessages) ?? lastInnerMessageId;\n              continue;\n            }\n\n            const isPersistedMessage = historyIds.current.has(message.id);\n            if (isPersistedMessage && !changedRunMessageIds.has(message.id)) {\n              lastInnerMessageId =\n                getLastInnerId(innerMessages) ?? lastInnerMessageId;\n              continue;\n            }\n            if (!isPersistedMessage) {\n              historyIds.current.add(message.id);\n              deferredTelemetryIds.current.add(message.id);\n            }\n\n            const batchItems = toBatchItems(innerMessages);\n            for (const item of batchItems) {\n              const innerId = storageFormatAdapter.getId(item.message);\n              if (!persistedInnerIds.current.has(innerId)) {\n                await adapter.append(item);\n                persistedInnerIds.current.add(innerId);\n              } else if (durationMs !== undefined) {\n                try {\n                  await adapter.update?.(item, innerId);\n                } catch {\n                  // A failed update drops the message from the refreshed baseline so it retries on the next run stop.\n                  failedUpdateIds.add(message.id);\n                }\n              }\n            }\n\n            lastInnerMessageId =\n              getLastInnerId(innerMessages) ?? lastInnerMessageId;\n\n            if (deferredTelemetryIds.current.has(message.id) && isTerminal) {\n              deferredTelemetryIds.current.delete(message.id);\n              adapter.reportTelemetry?.(batchItems, telemetryOptions);\n            }\n          }\n\n          const nextSnapshot = snapshotExternalMessages<TMessage>(\n            latest.messages,\n          );\n          for (const id of failedUpdateIds) {\n            nextSnapshot.delete(id);\n          }\n          persistedExternalMessages.current = nextSnapshot;\n        })\n        .catch((error) => {\n          console.error(\"Failed to persist message history:\", error);\n        });\n    }\n\n    return () => {\n      unsubscribe();\n      if (persistTimerRef.current) {\n        clearTimeout(persistTimerRef.current);\n        persistTimerRef.current = null;\n        persistSettled(false);\n      }\n    };\n  }, [formatAdapter, storageFormatAdapter, runtimeRef]);\n\n  const deleteMessage = useCallback(\n    async (messageId: string) => {\n      const deleteMessages = formatAdapter?.delete?.bind(formatAdapter);\n      if (!deleteMessages) return;\n\n      const messages = runtimeRef.current.thread.getState().messages;\n      const messageIndex = messages.findIndex((m) => m.id === messageId);\n      if (messageIndex === -1) return;\n\n      const previousInnerMessages = messages\n        .slice(0, messageIndex)\n        .flatMap(getExternalStoreMessages<TMessage>);\n      let parentId = previousInnerMessages.at(-1)\n        ? storageFormatAdapter.getId(previousInnerMessages.at(-1)!)\n        : null;\n      const itemsToDelete = getExternalStoreMessages<TMessage>(\n        messages[messageIndex]!,\n      ).map((message) => {\n        const item = { parentId, message };\n        parentId = storageFormatAdapter.getId(message);\n        return item;\n      });\n\n      const deletion = persistInFlightRef.current.then(async () => {\n        await deleteMessages(itemsToDelete);\n\n        historyIds.current.delete(messageId);\n        deferredTelemetryIds.current.delete(messageId);\n        persistedExternalMessages.current.delete(messageId);\n        for (const item of itemsToDelete) {\n          persistedInnerIds.current.delete(\n            storageFormatAdapter.getId(item.message),\n          );\n        }\n      });\n\n      persistInFlightRef.current = deletion.catch(() => {});\n      await deletion;\n    },\n    [formatAdapter, runtimeRef, storageFormatAdapter],\n  );\n\n  return { isLoading, deleteMessage };\n};\n"],"mappings":";;;;;;AAsBA,MAAae,+BACXC,kBACAC,aAC8B;CAC9B,MAAME,+BAAe,IAAIC,IAAY;CACrC,MAAMC,YAAYJ,SAASA,SAASK,SAASC,MAAM;EACjD,MAAMC,UAAUR,iBAAiB,CAACO,EAAEC,OAAO,CAAC,CAAC,CAAC;EAC9C,IAAI,CAACA,SAAS;GACZC,QAAQC,KAAK,qDAAqD;GAClE,OAAO,CAAA;EACT;EACA,IAAIH,EAAEI,YAAY,CAACR,aAAaS,IAAIL,EAAEI,QAAQ,GAAG,OAAO,CAAA;EACxDR,aAAaU,IAAIL,QAAQM,EAAE;EAC3B,OAAO,CAAC;GAAE,GAAGP;GAAGC;EAAQ,CAAC;CAC3B,CAAC;CAED,OAAO;EACLO,QACEd,SAASc,UAAUZ,aAAaS,IAAIX,SAASc,MAAM,IAC/Cd,SAASc,SACT;EACNd,UAAUI;CACZ;AACF;AAEA,MAAMW,0BAA0BR,YAC9BA,QAAQS,QAAQC,SAAS,qBACzBV,QAAQS,OAAOE,WAAW;AAE5B,MAAMC,4BACJnB,aAEA,IAAIoB,IACFpB,SAASqB,KAAKd,YAAY,CACxBA,QAAQM,IACR,CAAC,GAAGxB,yBAAmCkB,OAAO,CAAC,CAAC,CACjD,CACH;AAEF,MAAae,sBACXC,YACAC,gBACAzB,kBACA0B,sBACAC,kBACG;CACH,MAAMC,YAAYnC,OAAO,KAAK;CAC9B,MAAM,CAACoC,WAAWC,gBAAgBnC,SAAS,CAAC;CAE5C,MAAMoC,MAAMvC,OAAO;CACnB,MAAMwC,yBAAyBnC,kBACtBkC,IAAIE,eAAeC,SAASH,IAAIE,iBAAiB,MACxD,CAACF,GAAG,CACN;CAEA,MAAM,CAACI,WAAWC,gBAAgBzC,SAAS,KAAK;CAEhD,MAAM0C,aAAa5C,uBAAO,IAAIW,IAAY,CAAC;CAC3C,MAAMkC,oBAAoB7C,uBAAO,IAAIW,IAAY,CAAC;CAClD,MAAMmC,uBAAuB9C,uBAAO,IAAIW,IAAY,CAAC;CACrD,MAAMoC,4BAA4B/C,uBAAO,IAAI4B,IAAwB,CAAC;CAEtE,MAAMoB,mBAAmBhD,OAAOkC,aAAa;CAC7CjC,gBAAgB;EACd+C,iBAAiBC,UAAUf;CAC7B,CAAC;CAED,MAAMgB,gBAAgB7C,cAAc;EAClC,IAAI,CAAC2B,gBAAgB,OAAOmB,KAAAA;EAC5B,IAAI,CAACnB,eAAeoB,YAClB,MAAM,IAAIC,MACR,oFACF;EAEF,OAAOrB,eAAeoB,WAA0BnB,oBAAoB;CACtE,GAAG,CAACD,gBAAgBC,oBAAoB,CAAC;CAEzC,MAAMqB,YAAYJ,iBAAiB,QAAQ,CAACR;CAE5CzC,gBAAgB;EACd,IAAI,CAACiD,iBAAiBf,UAAUc,SAAS,OAAOE,KAAAA;EAEhD,MAAMI,cAAc,YAAY;GAC9B,IAAI;IACF,MAAMC,OAAO,MAAMN,cAAcO,KAAK;IACtC,IAAID,QAAQA,KAAKhD,SAASkD,SAAS,GAAG;KACpC,KAAK,MAAM5C,KAAK0C,KAAKhD,UACnBqC,kBAAkBI,QAAQ7B,IACxBa,qBAAqB0B,MAAM7C,EAAEC,OAAO,CACtC;KAEF,MAAM6C,YAAYtD,4BAA4BC,kBAAkBiD,IAAI;KACpEzB,WAAWkB,QAAQY,OAAOC,OAAOF,SAAS;KAE1C,MAAMG,WAAW,IAAIjE,kBAAkB;KACvCiE,SAASD,OAAOF,SAAS;KACzB,MAAMpD,WAAWuD,SAASC,YAAY;KAEtChB,iBAAiBC,QACfzC,SAASK,QAAQhB,wBAAkC,CACrD;KAEA+C,WAAWK,0BAAU,IAAItC,IAAI;KAC7B,KAAK,MAAMG,OAAK8C,UAAUpD,UAAU;MAClCoC,WAAWK,QAAQ7B,IAAIN,IAAEC,QAAQM,EAAE;MACnC,IAAIE,uBAAuBT,IAAEC,OAAO,GAClC+B,qBAAqBG,QAAQ7B,IAAIN,IAAEC,QAAQM,EAAE;KAEjD;KACA0B,0BAA0BE,UACxBtB,yBACEiC,UAAUpD,SAASqB,KAAKf,QAAMA,IAAEC,OAAO,CACzC;IACJ;GACF,SAASkD,OAAO;IACdjD,QAAQiD,MAAM,mCAAmCA,KAAK;GACxD,UAAU;IACRtB,aAAa,IAAI;GACnB;EACF;EAEAO,cAAcgB,MAAM;EAGpB,IAAI,CADa3B,uBAAuB,CAAC,EAAE6B,SAAS,CAAC,CAACD,UACvC;GACbxB,aAAa,IAAI;GACjB,OAAOL,IAAI+B,gBAAgB;IACzB,IAAI9B,uBAAuB,CAAC,EAAE6B,SAAS,CAAC,CAACD,UACvC9B,cAAciC,MAAMA,IAAI,CAAC;GAE7B,CAAC;EACH;EAEA,MAAMC,cAAcxC,WAAWkB,QAAQY,OAAOO,SAAS;EACvD,IAAIG,YAAYC,aAAaD,YAAY/D,SAASkD,SAAS,GAAG;GAC5DvB,UAAUc,UAAU;GACpBN,aAAa,IAAI;GACjB;EACF;EAEAR,UAAUc,UAAU;EACpB,YAAiB;CAEnB,GAAG;EACDC;EACA3C;EACAwB;EACAQ;EACAD;EACAF;EACAH;CAAoB,CACrB;CAED,MAAMwC,cAAczE,OAAsB,IAAI;CAC9C,MAAM0E,kBAAkB1E,OAA6C,IAAI;CACzE,MAAM6E,qBAAqB7E,OAAsB8E,QAAQC,QAAQ,CAAC;CAClE,MAAMC,oBAAoBhF,OAAiB,CAAA,CAAE;CAC7C,MAAMiF,gBAAgBjF,OAAO,KAAK;CAClC,MAAMkF,mBAAmBlF,OAAO,CAAC;CAEjCC,gBAAgB;EACd,IAAI,CAACiD,eAAe;EACpB,MAAMiC,UAAUjC;EAEhB,MAAMkC,cAAcrD,WAAWkB,QAAQY,OAAOQ,gBAAgB;GAC5D,MAAME,gBAAcxC,WAAWkB,QAAQY,OAAOO,SAAS;GACvD,MAAM,EAAEI,cAAcD;GACtB,MAAMc,aAAaJ,cAAchC;GACjCgC,cAAchC,UAAUuB;GAGxB,IAAIC,YAAYxB,WAAW,MAAM;IAC/B,MAAMqC,UAAUf,cAAY/D,SAAS+E,GAAG,EAAE;IAC1C,IAAID,SAASE,SAAS,aAAa;KACjC,MAAMC,uBAAuBH,QAAQI,QAAQC,QAC1CC,MAAMA,EAAEnE,SAAS,WACpB,CAAC,CAACiC;KACF,OAAOwB,iBAAiBjC,UAAUwC,sBAAsB;MACtDT,kBAAkB/B,QAAQ4C,KAAKC,KAAKC,IAAI,IAAItB,YAAYxB,OAAO;MAC/DiC,iBAAiBjC;KACnB;IACF;GACF;GAEA,IAAIuB,WAAW;IACb,IAAIC,YAAYxB,WAAW,MAAM;KAC/BwB,YAAYxB,UAAU6C,KAAKC,IAAI;KAC/Bf,kBAAkB/B,UAAU,CAAA;KAC5BiC,iBAAiBjC,UAAU;KAC3BkC,QAAQjB,MAAM;IAChB;IAEA,IAAIQ,gBAAgBzB,SAAS;KAC3B+C,aAAatB,gBAAgBzB,OAAO;KACpCyB,gBAAgBzB,UAAU;IAC5B;IACA;GACF;GAGA,IAAI,CAACoC,YAAY;GAGjB,IAAIZ,YAAYxB,WAAW,MACzB+B,kBAAkB/B,QAAQ4C,KAAKC,KAAKC,IAAI,IAAItB,YAAYxB,OAAO;GAIjE,IAAIyB,gBAAgBzB,SAAS+C,aAAatB,gBAAgBzB,OAAO;GACjEyB,gBAAgBzB,UAAU2B,iBAAiB;IACzCqB,eAAe,KAAK;GACtB,GAAG,CAAC;EACN,CAAC;EAED,SAASA,eAAeC,eAAwB;GAC9CxB,gBAAgBzB,UAAU;GAC1B,MAAMkD,SAASpE,WAAWkB,QAAQY,OAAOO,SAAS;GAClD,IAAI,CAAC8B,iBAAiBC,OAAO3B,WAAW;GAExC,MAAM4B,aAAapB,kBAAkB/B;GACrC,MAAMoD,aAAaD,WAAW1C,SAAS,IAAI0C,WAAWb,GAAG,EAAE,IAAIpC,KAAAA;GAG/D,IAAIiD,WAAW1C,WAAW,KAAK2C,cAAc,MAAM;IACjD,MAAMC,gBAAgBH,OAAO3F,SAAS+F,UACnCzF,QAAMA,IAAE0E,SAAS,WACpB;IACA,IAAIc,eAAe;KACjB,MAAME,UAAUF,cAAcZ,QAAQC,QACnCC,QAAMA,IAAEnE,SAAS,WACpB,CAAC,CAACiC;KACF,IAAI8C,UAAU,GAAG;MACf,MAAMC,aAAaD,UAAU;MAC7B,MAAME,UAAUL,aAAaI;MAC7BL,WAAW1C,SAAS;MACpB,KAAK,IAAIiD,IAAI,GAAGA,IAAIF,YAAYE,KAC9BP,WAAWP,KAAKe,KAAKC,OAAOF,IAAI,KAAKD,OAAO,CAAC;KAEjD;IACF;GACF;GAGA,MAAMI,iBACJV,WAAW1C,SAAS,IAChB0C,WAAWvE,KAAKkF,OAAOJ,SAAO;IAC5BK,UAAUL,QAAM,IAAI,IAAIP,WAAWO,MAAI;IACvCM,QAAQF;GACV,EAAE,IACF5D,KAAAA;GAENsB,YAAYxB,UAAU;GACtB+B,kBAAkB/B,UAAU,CAAA;GAE5B,MAAMiE,mBAAmB;IACvB,GAAIb,cAAc,OAAO,EAAEA,WAAW,IAAIlD,KAAAA;IAC1C,GAAI2D,kBAAkB,OAAO,EAAEA,eAAe,IAAI3D,KAAAA;GACpD;GAEA0B,mBAAmB5B,UAAU4B,mBAAmB5B,QAC7CkE,KAAK,YAAY;IAChB,MAAMC,uCAAuB,IAAIzG,IAAY;IAC7C,KAAK,MAAMI,WAAWoF,OAAO3F,UAAU;KACrC,MAAM6G,mBACJxH,yBAAmCkB,OAAO;KAC5C,MAAMuG,WAAWvE,0BAA0BE,QAAQsE,IAAIxG,QAAQM,EAAE;KACjE,IACEiG,aAAanE,KAAAA,KACbmE,SAAS5D,WAAW2D,iBAAiB3D,UACrC2D,iBAAiBG,MAAMC,MAAMC,UAAUD,SAASH,SAASI,MAAM,GAE/DN,qBAAqBhG,IAAIL,QAAQM,EAAE;IAEvC;IAEA,MAAM,EAAEb,UAAAA,eAAa2F;IACrB,IAAIwB,qBAAoC;IACxC,MAAMC,kCAAkB,IAAIjH,IAAY;IAExC,MAAMkH,kBAAkBC,SACtBA,KAAKpE,SAAS,IAAIzB,qBAAqB0B,MAAMmE,KAAKvC,GAAG,EAAE,CAAE,IAAI;IAE/D,MAAMwC,gBAAgBD,WACpBA,OAAKjG,KAAKmG,KAAKC,SAAS;KACtB/G,UACE+G,QAAQ,IACJN,qBACA1F,qBAAqB0B,MAAMmE,OAAKG,MAAM,EAAG;KAC/ClH,SAASiH;IACX,EAAE;IAEJ,KAAK,MAAMjH,aAAWP,YAAU;KAC9B,MAAM0H,gBAAgBrI,yBAAmCkB,SAAO;KAEhE,MAAMoH,aACJpH,UAAQS,WAAW2B,KAAAA,KACnBpC,UAAQS,OAAOC,SAAS,cACxBV,UAAQS,OAAOC,SAAS;KAC1B,MAAM2G,sBAAsB7G,uBAAuBR,SAAO;KAM1D,IAAI,EAHFoH,cACCC,uBAAuBjD,QAAQmD,WAAWnF,KAAAA,IAE/B;MACZwE,qBACEE,eAAeK,aAAa,KAAKP;MACnC;KACF;KAEA,MAAMY,qBAAqB3F,WAAWK,QAAQ9B,IAAIJ,UAAQM,EAAE;KAC5D,IAAIkH,sBAAsB,CAACnB,qBAAqBjG,IAAIJ,UAAQM,EAAE,GAAG;MAC/DsG,qBACEE,eAAeK,aAAa,KAAKP;MACnC;KACF;KACA,IAAI,CAACY,oBAAoB;MACvB3F,WAAWK,QAAQ7B,IAAIL,UAAQM,EAAE;MACjCyB,qBAAqBG,QAAQ7B,IAAIL,UAAQM,EAAE;KAC7C;KAEA,MAAMmH,aAAaT,aAAaG,aAAa;KAC7C,KAAK,MAAMT,UAAQe,YAAY;MAC7B,MAAMC,UAAUxG,qBAAqB0B,MAAM8D,OAAK1G,OAAO;MACvD,IAAI,CAAC8B,kBAAkBI,QAAQ9B,IAAIsH,OAAO,GAAG;OAC3C,MAAMtD,QAAQuD,OAAOjB,MAAI;OACzB5E,kBAAkBI,QAAQ7B,IAAIqH,OAAO;MACvC,OAAO,IAAIpC,eAAelD,KAAAA,GACxB,IAAI;OACF,MAAMgC,QAAQmD,SAASb,QAAMgB,OAAO;MACtC,QAAQ;OAENb,gBAAgBxG,IAAIL,UAAQM,EAAE;MAChC;KAEJ;KAEAsG,qBACEE,eAAeK,aAAa,KAAKP;KAEnC,IAAI7E,qBAAqBG,QAAQ9B,IAAIJ,UAAQM,EAAE,KAAK8G,YAAY;MAC9DrF,qBAAqBG,QAAQ0F,OAAO5H,UAAQM,EAAE;MAC9C8D,QAAQyD,kBAAkBJ,YAAYtB,gBAAgB;KACxD;IACF;IAEA,MAAM2B,eAAelH,yBACnBwE,OAAO3F,QACT;IACA,KAAK,MAAMa,MAAMuG,iBACfiB,aAAaF,OAAOtH,EAAE;IAExB0B,0BAA0BE,UAAU4F;GACtC,CAAC,CAAC,CACDC,OAAO7E,YAAU;IAChBjD,QAAQiD,MAAM,sCAAsCA,OAAK;GAC3D,CAAC;EACL;EAEA,aAAa;GACXmB,YAAY;GACZ,IAAIV,gBAAgBzB,SAAS;IAC3B+C,aAAatB,gBAAgBzB,OAAO;IACpCyB,gBAAgBzB,UAAU;IAC1BgD,eAAe,KAAK;GACtB;EACF;CACF,GAAG;EAAC/C;EAAejB;EAAsBF;CAAU,CAAC;CA4CpD,OAAO;EAAEuB;EAAWyF,eA1CE3I,YACpB,OAAO4I,cAAsB;GAC3B,MAAMC,iBAAiB/F,eAAeyF,QAAQO,KAAKhG,aAAa;GAChE,IAAI,CAAC+F,gBAAgB;GAErB,MAAMzI,aAAWuB,WAAWkB,QAAQY,OAAOO,SAAS,CAAC,CAAC5D;GACtD,MAAM2I,eAAe3I,WAAS4I,WAAWtI,QAAMA,IAAEO,OAAO2H,SAAS;GACjE,IAAIG,iBAAiB,IAAI;GAEzB,MAAME,wBAAwB7I,WAC3B8I,MAAM,GAAGH,YAAY,CAAC,CACtBtI,QAAQhB,wBAAkC;GAC7C,IAAIqB,WAAWmI,sBAAsB9D,GAAG,EAAE,IACtCtD,qBAAqB0B,MAAM0F,sBAAsB9D,GAAG,EAAE,CAAE,IACxD;GACJ,MAAMgE,gBAAgB1J,yBACpBW,WAAS2I,aACX,CAAC,CAACtH,KAAKd,cAAY;IACjB,MAAM0G,SAAO;KAAEvG;KAAUH,SAAAA;IAAQ;IACjCG,WAAWe,qBAAqB0B,MAAM5C,SAAO;IAC7C,OAAO0G;GACT,CAAC;GAED,MAAM+B,WAAW3E,mBAAmB5B,QAAQkE,KAAK,YAAY;IAC3D,MAAM8B,eAAeM,aAAa;IAElC3G,WAAWK,QAAQ0F,OAAOK,SAAS;IACnClG,qBAAqBG,QAAQ0F,OAAOK,SAAS;IAC7CjG,0BAA0BE,QAAQ0F,OAAOK,SAAS;IAClD,KAAK,MAAMvB,UAAQ8B,eACjB1G,kBAAkBI,QAAQ0F,OACxB1G,qBAAqB0B,MAAM8D,OAAK1G,OAAO,CACzC;GAEJ,CAAC;GAED8D,mBAAmB5B,UAAUuG,SAASV,YAAY,CAAC,CAAC;GACpD,MAAMU;EACR,GACA;GAACtG;GAAenB;GAAYE;EAAoB,CAG9B8G;CAAc;AACpC"}

package/dist/runtime/useResourceCleanup.d.ts

@@ -0,0 +1,5 @@
+//#region src/runtime/useResourceCleanup.d.ts
+declare const useResourceCleanup: (enabled: boolean, cleanup: () => void) => void;
+//#endregion
+export { useResourceCleanup };
+//# sourceMappingURL=useResourceCleanup.d.ts.map

package/dist/runtime/useResourceCleanup.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"useResourceCleanup.d.ts","names":[],"sources":["../../src/runtime/useResourceCleanup.ts"],"mappings":";cAGa,qBAAsB,kBAAkB"}

package/dist/runtime/useResourceCleanup.js

@@ -0,0 +1,47 @@
+import { useEffect, useRef } from "@assistant-ui/tap/react-shim";
+import { c } from "@assistant-ui/tap/react-shim/compiler-runtime";
+import { useAssistantClientDestroySignal } from "@assistant-ui/store/internal";
+//#region src/runtime/useResourceCleanup.ts
+const useResourceCleanup = (enabled, cleanup) => {
+	const $ = c(7);
+	const destroySignal = useAssistantClientDestroySignal();
+	const cleanupRef = useRef(cleanup);
+	const enabledRef = useRef(enabled);
+	const registeredSignalRef = useRef(void 0);
+	let t0;
+	if ($[0] !== cleanup || $[1] !== enabled) {
+		t0 = () => {
+			cleanupRef.current = cleanup;
+			enabledRef.current = enabled;
+		};
+		$[0] = cleanup;
+		$[1] = enabled;
+		$[2] = t0;
+	} else t0 = $[2];
+	useEffect(t0);
+	let t1;
+	let t2;
+	if ($[3] !== destroySignal || $[4] !== enabled) {
+		t1 = () => {
+			if (!enabled || !destroySignal) return;
+			if (registeredSignalRef.current === destroySignal) return;
+			registeredSignalRef.current = destroySignal;
+			destroySignal.addEventListener("abort", () => {
+				if (enabledRef.current) cleanupRef.current();
+			}, { once: true });
+		};
+		t2 = [destroySignal, enabled];
+		$[3] = destroySignal;
+		$[4] = enabled;
+		$[5] = t1;
+		$[6] = t2;
+	} else {
+		t1 = $[5];
+		t2 = $[6];
+	}
+	useEffect(t1, t2);
+};
+//#endregion
+export { useResourceCleanup };
+
+//# sourceMappingURL=useResourceCleanup.js.map

package/dist/runtime/useResourceCleanup.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useResourceCleanup.js","names":["useEffect","useRef","useAssistantClientDestroySignal","useResourceCleanup","enabled","cleanup","$","_c","destroySignal","cleanupRef","enabledRef","registeredSignalRef","undefined","t0","current","t1","t2","addEventListener","once"],"sources":["../../src/runtime/useResourceCleanup.ts"],"sourcesContent":["import { useEffect, useRef } from \"react\";\nimport { useAssistantClientDestroySignal } from \"@assistant-ui/store/internal\";\n\nexport const useResourceCleanup = (enabled: boolean, cleanup: () => void) => {\n  const destroySignal = useAssistantClientDestroySignal();\n  const cleanupRef = useRef(cleanup);\n  const enabledRef = useRef(enabled);\n  const registeredSignalRef = useRef<AbortSignal | undefined>(undefined);\n\n  useEffect(() => {\n    cleanupRef.current = cleanup;\n    enabledRef.current = enabled;\n  });\n\n  useEffect(() => {\n    if (!enabled || !destroySignal) return undefined;\n    if (registeredSignalRef.current === destroySignal) return undefined;\n\n    registeredSignalRef.current = destroySignal;\n    destroySignal.addEventListener(\n      \"abort\",\n      () => {\n        if (enabledRef.current) cleanupRef.current();\n      },\n      { once: true },\n    );\n\n    // The listener must survive standalone soft unmounts so a later permanent\n    // client destroy still cleans up the retained resource state.\n    return undefined;\n  }, [destroySignal, enabled]);\n};\n"],"mappings":";;;;AAGA,MAAaG,sBAAqBC,SAAAC,YAAA;CAAA,MAAAC,IAAAC,EAAA,CAAA;CAChC,MAAAC,gBAAsBN,gCAAgC;CACtD,MAAAO,aAAmBR,OAAOI,OAAO;CACjC,MAAAK,aAAmBT,OAAOG,OAAO;CACjC,MAAAO,sBAA4BV,OAAgCW,KAAAA,CAAS;CAAE,IAAAC;CAAA,IAAAP,EAAA,OAAAD,WAAAC,EAAA,OAAAF,SAAA;EAE7DS,WAAA;GACRJ,WAAUK,UAAWT;GACrBK,WAAUI,UAAWV;EAAH;EACnBE,EAAA,KAAAD;EAAAC,EAAA,KAAAF;EAAAE,EAAA,KAAAO;CAAA,OAAAA,KAAAP,EAAA;CAHDN,UAAUa,EAGT;CAAC,IAAAE;CAAA,IAAAC;CAAA,IAAAV,EAAA,OAAAE,iBAAAF,EAAA,OAAAF,SAAA;EAEQW,WAAA;GACR,IAAI,CAACX,WAAD,CAAaI,eAAa;GAC9B,IAAIG,oBAAmBG,YAAaN,eAAa;GAEjDG,oBAAmBG,UAAWN;GAC9BA,cAAaS,iBACX,eACA;IACE,IAAIP,WAAUI,SAAUL,WAAUK,QAAS;GAAE,GAE/C,EAAAI,MAAQ,KAAK,CACf;EAAC;EAKAF,KAAA,CAACR,eAAeJ,OAAO;EAACE,EAAA,KAAAE;EAAAF,EAAA,KAAAF;EAAAE,EAAA,KAAAS;EAAAT,EAAA,KAAAU;CAAA,OAAA;EAAAD,KAAAT,EAAA;EAAAU,KAAAV,EAAA;CAAA;CAhB3BN,UAAUe,IAgBPC,EAAwB;AAAC"}

package/dist/runtime/useStreamingTiming.d.ts

@@ -0,0 +1,15 @@
+import { StreamingTimingAccessors } from "@assistant-ui/core/react";
+import { MessageTiming } from "@assistant-ui/core";
+import { UIMessage } from "@ai-sdk/react";
+//#region src/runtime/useStreamingTiming.d.ts
+declare const aiSdkStreamingTimingAccessors: StreamingTimingAccessors<UIMessage>;
+/**
+ * Tracks streaming timing for AI SDK messages client-side. Delegates to the
+ * shared `useStreamingTiming` primitive in `@assistant-ui/core/react`,
+ * adapted to the `UIMessage` shape. Timing is finalized when streaming ends
+ * and stored per message id.
+ */
+declare const useStreamingTiming: (messages: UIMessage[], isRunning: boolean) => Record<string, MessageTiming>;
+//#endregion
+export { aiSdkStreamingTimingAccessors, useStreamingTiming };
+//# sourceMappingURL=useStreamingTiming.d.ts.map

package/dist/runtime/useStreamingTiming.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"useStreamingTiming.d.ts","names":[],"sources":["../../src/runtime/useStreamingTiming.ts"],"mappings":";;;;cA8Ca,+BAA+B,yBAAyB;;;;;;;cAaxD,qBACX,UAAU,aACV,uBACC,eAAe"}

package/dist/runtime/useStreamingTiming.js

@@ -0,0 +1,38 @@
+"use client";
+import { useStreamingTiming as useStreamingTiming$1 } from "@assistant-ui/core/react";
+import { isToolUIPart } from "ai";
+//#region src/runtime/useStreamingTiming.ts
+const findAssistant = (messages, messageId) => messages.find((m) => m.role === "assistant" && m.id === messageId);
+const getTextLength = (messages, messageId) => {
+	const message = findAssistant(messages, messageId);
+	if (!message?.parts) return 0;
+	let len = 0;
+	for (const part of message.parts) if (part.type === "text") len += part.text.length;
+	return len;
+};
+const getToolCallCount = (messages, messageId) => {
+	const message = findAssistant(messages, messageId);
+	if (!message?.parts) return 0;
+	let count = 0;
+	for (const part of message.parts) if (isToolUIPart(part)) count++;
+	return count;
+};
+const getAssistantMessageId = (messages) => messages.findLast((m) => m.role === "assistant")?.id;
+const aiSdkStreamingTimingAccessors = {
+	getAssistantMessageId,
+	getTextLength,
+	getToolCallCount
+};
+/**
+* Tracks streaming timing for AI SDK messages client-side. Delegates to the
+* shared `useStreamingTiming` primitive in `@assistant-ui/core/react`,
+* adapted to the `UIMessage` shape. Timing is finalized when streaming ends
+* and stored per message id.
+*/
+const useStreamingTiming = (messages, isRunning) => {
+	return useStreamingTiming$1(messages, isRunning, aiSdkStreamingTimingAccessors);
+};
+//#endregion
+export { aiSdkStreamingTimingAccessors, useStreamingTiming };
+
+//# sourceMappingURL=useStreamingTiming.js.map

package/dist/runtime/useStreamingTiming.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"useStreamingTiming.js","names":["UIMessage","isToolUIPart","MessageTiming","useStreamingTiming","useStreamingTimingPrimitive","StreamingTimingAccessors","findAssistant","messages","messageId","find","m","role","id","getTextLength","message","parts","len","part","type","text","length","getToolCallCount","count","getAssistantMessageId","findLast","aiSdkStreamingTimingAccessors","isRunning"],"sources":["../../src/runtime/useStreamingTiming.ts"],"sourcesContent":["\"use client\";\n\nimport type { UIMessage } from \"@ai-sdk/react\";\nimport { isToolUIPart } from \"ai\";\nimport type { MessageTiming } from \"@assistant-ui/core\";\nimport {\n  useStreamingTiming as useStreamingTimingPrimitive,\n  type StreamingTimingAccessors,\n} from \"@assistant-ui/core/react\";\n\nconst findAssistant = (\n  messages: readonly UIMessage[],\n  messageId: string,\n): UIMessage | undefined =>\n  messages.find((m) => m.role === \"assistant\" && m.id === messageId);\n\nconst getTextLength = (\n  messages: readonly UIMessage[],\n  messageId: string,\n): number => {\n  const message = findAssistant(messages, messageId);\n  if (!message?.parts) return 0;\n  let len = 0;\n  for (const part of message.parts) {\n    if (part.type === \"text\") len += part.text.length;\n  }\n  return len;\n};\n\nconst getToolCallCount = (\n  messages: readonly UIMessage[],\n  messageId: string,\n): number => {\n  const message = findAssistant(messages, messageId);\n  if (!message?.parts) return 0;\n  let count = 0;\n  for (const part of message.parts) {\n    if (isToolUIPart(part)) count++;\n  }\n  return count;\n};\n\nconst getAssistantMessageId = (\n  messages: readonly UIMessage[],\n): string | undefined => messages.findLast((m) => m.role === \"assistant\")?.id;\n\nexport const aiSdkStreamingTimingAccessors: StreamingTimingAccessors<UIMessage> =\n  {\n    getAssistantMessageId,\n    getTextLength,\n    getToolCallCount,\n  };\n\n/**\n * Tracks streaming timing for AI SDK messages client-side. Delegates to the\n * shared `useStreamingTiming` primitive in `@assistant-ui/core/react`,\n * adapted to the `UIMessage` shape. Timing is finalized when streaming ends\n * and stored per message id.\n */\nexport const useStreamingTiming = (\n  messages: UIMessage[],\n  isRunning: boolean,\n): Record<string, MessageTiming> =>\n  useStreamingTimingPrimitive(\n    messages,\n    isRunning,\n    aiSdkStreamingTimingAccessors,\n  );\n"],"mappings":";;;;AAUA,MAAMM,iBACJC,UACAC,cAEAD,SAASE,MAAMC,MAAMA,EAAEC,SAAS,eAAeD,EAAEE,OAAOJ,SAAS;AAEnE,MAAMK,iBACJN,UACAC,cACW;CACX,MAAMM,UAAUR,cAAcC,UAAUC,SAAS;CACjD,IAAI,CAACM,SAASC,OAAO,OAAO;CAC5B,IAAIC,MAAM;CACV,KAAK,MAAMC,QAAQH,QAAQC,OACzB,IAAIE,KAAKC,SAAS,QAAQF,OAAOC,KAAKE,KAAKC;CAE7C,OAAOJ;AACT;AAEA,MAAMK,oBACJd,UACAC,cACW;CACX,MAAMM,UAAUR,cAAcC,UAAUC,SAAS;CACjD,IAAI,CAACM,SAASC,OAAO,OAAO;CAC5B,IAAIO,QAAQ;CACZ,KAAK,MAAML,QAAQH,QAAQC,OACzB,IAAId,aAAagB,IAAI,GAAGK;CAE1B,OAAOA;AACT;AAEA,MAAMC,yBACJhB,aACuBA,SAASiB,UAAUd,MAAMA,EAAEC,SAAS,WAAW,CAAC,EAAEC;AAE3E,MAAaa,gCACX;CACEF;CACAV;CACAQ;AACF;;;;;;;AAQF,MAAalB,sBAAqBI,UAAAmB,cAAA;CAAA,OAIhCtB,qBACEG,UACAmB,WACAD,6BACF;AAAC"}

package/dist/tools/frontendTools.d.ts

@@ -0,0 +1,33 @@
+import { ToolSet } from "ai";
+import { ToolJSONSchema } from "assistant-stream";
+//#region src/tools/frontendTools.d.ts
+/** Frontend tool definitions uploaded by AssistantChatTransport. */
+type FrontendTools = Record<string, ToolJSONSchema>;
+declare const defaultToModelOutput: ({ output }: {
+  output: unknown;
+}) => {
+  type: "content";
+  value: ({
+    type: "text";
+    text: string;
+  } | {
+    filename?: string;
+    type: "file";
+    data: {
+      type: "data";
+      data: string;
+    };
+    mediaType: string;
+    text?: never;
+  })[];
+} | {
+  type: "text";
+  value: string;
+} | {
+  type: "json";
+  value: import("ai").JSONValue;
+};
+declare const frontendTools: (tools: FrontendTools) => ToolSet;
+//#endregion
+export { FrontendTools, defaultToModelOutput, frontendTools };
+//# sourceMappingURL=frontendTools.d.ts.map

package/dist/tools/frontendTools.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"frontendTools.d.ts","names":[],"sources":["../../src/tools/frontendTools.ts"],"mappings":";;;;KASY,gBAAgB,eAAe;cAE9B,yBAAwB;EAAc;;;;;;;;;;;;;;;;;;;;;;;cAuDtC,gBAAiB,OAAO,kBAAgB"}

package/dist/tools/frontendTools.js

@@ -0,0 +1,33 @@
+import { unwrapModelContentEnvelope } from "../converters/modelContentEnvelope.js";
+import { toAISDKContent, toAISDKDefaultOutput } from "../converters/toolOutputConversion.js";
+import { jsonSchema } from "ai";
+//#region src/tools/frontendTools.ts
+const defaultToModelOutput = ({ output }) => {
+	const { result, modelContent } = unwrapModelContentEnvelope(output);
+	if (modelContent !== void 0) return toAISDKContent(modelContent);
+	return toAISDKDefaultOutput(result);
+};
+const isPlainObject = (value) => typeof value === "object" && value !== null && !Array.isArray(value);
+function validateFrontendTool(name, tool) {
+	if (!isPlainObject(tool)) throw new Error(`frontendTools() expected tool "${name}" to be an object with a JSON Schema parameters object.`);
+	if (!isPlainObject(tool.parameters)) throw new Error(`frontendTools() expected tool "${name}" to include a JSON Schema parameters object.`);
+	if (tool.description !== void 0 && typeof tool.description !== "string") throw new Error(`frontendTools() expected tool "${name}" description to be a string.`);
+	if (tool.providerOptions !== void 0 && !isPlainObject(tool.providerOptions)) throw new Error(`frontendTools() expected tool "${name}" providerOptions to be an object.`);
+}
+function validateFrontendTools(tools) {
+	if (!isPlainObject(tools)) throw new Error("frontendTools() expected tools to be an object keyed by tool name.");
+	for (const [name, tool] of Object.entries(tools)) validateFrontendTool(name, tool);
+}
+const frontendTools = (tools) => {
+	validateFrontendTools(tools);
+	return Object.fromEntries(Object.entries(tools).map(([name, tool]) => [name, {
+		...tool.description !== void 0 && { description: tool.description },
+		inputSchema: jsonSchema(tool.parameters),
+		toModelOutput: defaultToModelOutput,
+		...tool.providerOptions && { providerOptions: tool.providerOptions }
+	}]));
+};
+//#endregion
+export { defaultToModelOutput, frontendTools };
+
+//# sourceMappingURL=frontendTools.js.map

package/dist/tools/frontendTools.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"frontendTools.js","names":["jsonSchema","ToolSet","ToolJSONSchema","unwrapModelContentEnvelope","toAISDKContent","toAISDKDefaultOutput","FrontendTools","Record","defaultToModelOutput","output","result","modelContent","undefined","isPlainObject","value","Array","isArray","validateFrontendTool","name","tool","Error","parameters","description","providerOptions","validateFrontendTools","tools","Object","entries","frontendTools","fromEntries","map","inputSchema","toModelOutput"],"sources":["../../src/tools/frontendTools.ts"],"sourcesContent":["import { jsonSchema, type ToolSet } from \"ai\";\nimport type { ToolJSONSchema } from \"assistant-stream\";\nimport { unwrapModelContentEnvelope } from \"../converters/modelContentEnvelope\";\nimport {\n  toAISDKContent,\n  toAISDKDefaultOutput,\n} from \"../converters/toolOutputConversion\";\n\n/** Frontend tool definitions uploaded by AssistantChatTransport. */\nexport type FrontendTools = Record<string, ToolJSONSchema>;\n\nexport const defaultToModelOutput = ({ output }: { output: unknown }) => {\n  const { result, modelContent } = unwrapModelContentEnvelope(output);\n  if (modelContent !== undefined) {\n    return toAISDKContent(modelContent);\n  }\n  return toAISDKDefaultOutput(result);\n};\n\nconst isPlainObject = (value: unknown): value is Record<string, unknown> =>\n  typeof value === \"object\" && value !== null && !Array.isArray(value);\n\nfunction validateFrontendTool(\n  name: string,\n  tool: unknown,\n): asserts tool is ToolJSONSchema {\n  if (!isPlainObject(tool)) {\n    throw new Error(\n      `frontendTools() expected tool \"${name}\" to be an object with a JSON Schema parameters object.`,\n    );\n  }\n\n  if (!isPlainObject(tool.parameters)) {\n    throw new Error(\n      `frontendTools() expected tool \"${name}\" to include a JSON Schema parameters object.`,\n    );\n  }\n\n  if (tool.description !== undefined && typeof tool.description !== \"string\") {\n    throw new Error(\n      `frontendTools() expected tool \"${name}\" description to be a string.`,\n    );\n  }\n\n  if (\n    tool.providerOptions !== undefined &&\n    !isPlainObject(tool.providerOptions)\n  ) {\n    throw new Error(\n      `frontendTools() expected tool \"${name}\" providerOptions to be an object.`,\n    );\n  }\n}\n\nfunction validateFrontendTools(tools: unknown): asserts tools is FrontendTools {\n  if (!isPlainObject(tools)) {\n    throw new Error(\n      \"frontendTools() expected tools to be an object keyed by tool name.\",\n    );\n  }\n\n  for (const [name, tool] of Object.entries(tools)) {\n    validateFrontendTool(name, tool);\n  }\n}\n\nexport const frontendTools = (tools: FrontendTools): ToolSet => {\n  validateFrontendTools(tools);\n\n  return Object.fromEntries(\n    Object.entries(tools).map(([name, tool]) => [\n      name,\n      {\n        ...(tool.description !== undefined && {\n          description: tool.description,\n        }),\n        inputSchema: jsonSchema(tool.parameters),\n        toModelOutput: defaultToModelOutput,\n        ...(tool.providerOptions && { providerOptions: tool.providerOptions }),\n      },\n    ]),\n  ) as ToolSet;\n};\n"],"mappings":";;;;AAWA,MAAaQ,wBAAwB,EAAEC,aAAkC;CACvE,MAAM,EAAEC,QAAQC,iBAAiBR,2BAA2BM,MAAM;CAClE,IAAIE,iBAAiBC,KAAAA,GACnB,OAAOR,eAAeO,YAAY;CAEpC,OAAON,qBAAqBK,MAAM;AACpC;AAEA,MAAMG,iBAAiBC,UACrB,OAAOA,UAAU,YAAYA,UAAU,QAAQ,CAACC,MAAMC,QAAQF,KAAK;AAErE,SAASG,qBACPC,MACAC,MACgC;CAChC,IAAI,CAACN,cAAcM,IAAI,GACrB,MAAM,IAAIC,MACR,kCAAkCF,KAAI,wDACxC;CAGF,IAAI,CAACL,cAAcM,KAAKE,UAAU,GAChC,MAAM,IAAID,MACR,kCAAkCF,KAAI,8CACxC;CAGF,IAAIC,KAAKG,gBAAgBV,KAAAA,KAAa,OAAOO,KAAKG,gBAAgB,UAChE,MAAM,IAAIF,MACR,kCAAkCF,KAAI,8BACxC;CAGF,IACEC,KAAKI,oBAAoBX,KAAAA,KACzB,CAACC,cAAcM,KAAKI,eAAe,GAEnC,MAAM,IAAIH,MACR,kCAAkCF,KAAI,mCACxC;AAEJ;AAEA,SAASM,sBAAsBC,OAAgD;CAC7E,IAAI,CAACZ,cAAcY,KAAK,GACtB,MAAM,IAAIL,MACR,oEACF;CAGF,KAAK,MAAM,CAACF,MAAMC,SAASO,OAAOC,QAAQF,KAAK,GAC7CR,qBAAqBC,MAAMC,IAAI;AAEnC;AAEA,MAAaS,iBAAiBH,UAAkC;CAC9DD,sBAAsBC,KAAK;CAE3B,OAAOC,OAAOG,YACZH,OAAOC,QAAQF,KAAK,CAAC,CAACK,KAAK,CAACZ,MAAMC,UAAU,CAC1CD,MACA;EACE,GAAIC,KAAKG,gBAAgBV,KAAAA,KAAa,EACpCU,aAAaH,KAAKG,YACpB;EACAS,aAAa/B,WAAWmB,KAAKE,UAAU;EACvCW,eAAexB;EACf,GAAIW,KAAKI,mBAAmB,EAAEA,iBAAiBJ,KAAKI,gBAAgB;CACtE,CAAC,CACF,CACH;AACF"}

package/dist/tools/generativeTools.d.ts

@@ -0,0 +1,73 @@
+import { FrontendTools } from "./frontendTools.js";
+import { Toolkit } from "@assistant-ui/core/react";
+import { ToolSet } from "ai";
+//#region src/tools/generativeTools.d.ts
+/**
+ * @deprecated Options for the deprecated {@link generativeTools}. Use
+ * {@link AISDKToolkit} with {@link AISDKToolkitOptions} /
+ * {@link AISDKToolkitToolsOptions} instead.
+ */
+interface GenerativeToolsOptions {
+  /**
+   * The server build of a generative toolkit (schema + server `execute`). Typed
+   * as the canonical {@link Toolkit} so callers don't need to cast; the server
+   * build carries `execute`, recovered internally as {@link ToolkitDefinition}.
+   */
+  toolkit: Toolkit;
+  /**
+   * Tools uploaded by the frontend (the request body's `tools`). Merged in
+   * alongside the `toolkit`; a server `execute` from `toolkit` takes precedence
+   * over an uploaded entry of the same name.
+   */
+  frontendTools?: FrontendTools;
+}
+type AISDKToolkitOptions = {
+  toolkit: Toolkit;
+};
+type AISDKToolkitToolsOptions = {
+  /**
+   * Tools uploaded by the frontend request body.
+   */
+  frontend?: FrontendTools;
+};
+/**
+ * Builds an AI SDK `ToolSet` for server-side use with `streamText` /
+ * `generateText` from a generative `toolkit` and the frontend-uploaded tools.
+ *
+ * Each toolkit tool's `execute` runs on the server. Pair this with the
+ * `"use generative"` compiler: import the toolkit in a server route (where it
+ * resolves to the server build — schema + `execute`, with `render` stripped) and
+ * pass it here. Tools without an `execute` are still exposed to the model but
+ * left for the client to fulfill. `frontendTools` lets the client contribute
+ * tools that aren't in the static toolkit.
+ *
+ * @deprecated Use {@link AISDKToolkit} instead:
+ * `new AISDKToolkit({ toolkit }).tools({ frontend })`. It is a strict superset
+ * (it also opens MCP server connections), so it replaces `generativeTools`
+ * everywhere. The `frontendTools` option is named `frontend` on `.tools()`, and
+ * `.tools()` is async. `generativeTools` will be removed in a future version.
+ *
+ * @example
+ * ```ts
+ * // Define once at module scope so any MCP connections pool across requests.
+ * const aiToolkit = new AISDKToolkit({ toolkit: docsToolkit });
+ *
+ * // In your route handler:
+ * const { tools } = await req.json();
+ * streamText({
+ *   model,
+ *   messages,
+ *   tools: await aiToolkit.tools({ frontend: tools }),
+ * });
+ * ```
+ */
+declare const generativeTools: (options: GenerativeToolsOptions) => ToolSet;
+declare class AISDKToolkit {
+  #private;
+  constructor(options: AISDKToolkitOptions);
+  tools(options?: AISDKToolkitToolsOptions): Promise<ToolSet>;
+  close(): Promise<void>;
+}
+//#endregion
+export { AISDKToolkit, AISDKToolkitOptions, AISDKToolkitToolsOptions, GenerativeToolsOptions, generativeTools };
+//# sourceMappingURL=generativeTools.d.ts.map

package/dist/tools/generativeTools.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"generativeTools.d.ts","names":[],"sources":["../../src/tools/generativeTools.ts"],"mappings":";;;;;;;;;UAsFiB;;;;;;EAMf,SAAS;;;;;;EAMT,gBAAgB;;KAGN;EACV,SAAS;;KAGC;;;;EAIV,WAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAkCA,kBAAmB,SAAS,2BAAyB;cAYrD;;EAIC,YAAA,SAAS;EAIf,MAAM,UAAS,2BAAgC,QAAQ;EAsBvD,SAAS"}

package/dist/tools/generativeTools.js

@@ -0,0 +1,245 @@
+import { unwrapModelContentEnvelope } from "../converters/modelContentEnvelope.js";
+import { toAISDKContent, toAISDKDefaultOutput } from "../converters/toolOutputConversion.js";
+import { frontendTools } from "./frontendTools.js";
+import { jsonSchema } from "ai";
+import { toJSONSchema } from "assistant-stream";
+import { createMCPClient } from "@ai-sdk/mcp";
+import { Experimental_StdioMCPTransport } from "#mcp-stdio";
+//#region src/tools/generativeTools.ts
+const EMPTY_SCHEMA = {
+	type: "object",
+	properties: {}
+};
+const humanNotSupported = () => {
+	throw new Error("`human()` is not available during server-side tool execution.");
+};
+const neverAbort = new AbortController().signal;
+var MCPConnectionTimeoutError = class extends Error {};
+const createMcpConnectionTimeoutError = (name, phase, timeoutMs) => new MCPConnectionTimeoutError(`MCP toolkit entry "${name}" timed out while ${phase} after ${timeoutMs}ms.`);
+const withMcpConnectionTimeout = async (promise, options) => {
+	const timeoutMs = options.config.connectionTimeout;
+	if (timeoutMs === void 0) return await promise;
+	const remainingMs = timeoutMs - (Date.now() - options.startedAt);
+	const timeoutError = () => createMcpConnectionTimeoutError(options.name, options.phase, timeoutMs);
+	if (remainingMs <= 0) throw timeoutError();
+	let timeout;
+	try {
+		return await Promise.race([promise, new Promise((_, reject) => {
+			timeout = setTimeout(() => reject(timeoutError()), remainingMs);
+		})]);
+	} finally {
+		if (timeout !== void 0) clearTimeout(timeout);
+	}
+};
+const parametersToInputSchema = (parameters) => jsonSchema(parameters ? toJSONSchema(parameters) : EMPTY_SCHEMA);
+/**
+* Builds an AI SDK `ToolSet` for server-side use with `streamText` /
+* `generateText` from a generative `toolkit` and the frontend-uploaded tools.
+*
+* Each toolkit tool's `execute` runs on the server. Pair this with the
+* `"use generative"` compiler: import the toolkit in a server route (where it
+* resolves to the server build — schema + `execute`, with `render` stripped) and
+* pass it here. Tools without an `execute` are still exposed to the model but
+* left for the client to fulfill. `frontendTools` lets the client contribute
+* tools that aren't in the static toolkit.
+*
+* @deprecated Use {@link AISDKToolkit} instead:
+* `new AISDKToolkit({ toolkit }).tools({ frontend })`. It is a strict superset
+* (it also opens MCP server connections), so it replaces `generativeTools`
+* everywhere. The `frontendTools` option is named `frontend` on `.tools()`, and
+* `.tools()` is async. `generativeTools` will be removed in a future version.
+*
+* @example
+* ```ts
+* // Define once at module scope so any MCP connections pool across requests.
+* const aiToolkit = new AISDKToolkit({ toolkit: docsToolkit });
+*
+* // In your route handler:
+* const { tools } = await req.json();
+* streamText({
+*   model,
+*   messages,
+*   tools: await aiToolkit.tools({ frontend: tools }),
+* });
+* ```
+*/
+const generativeTools = (options) => {
+	assertNoMcpToolkitTools(options.toolkit);
+	return {
+		...options.frontendTools ? frontendTools(options.frontendTools) : {},
+		...toProviderToolSet(options.toolkit),
+		...toServerToolSet(options.toolkit)
+	};
+};
+var AISDKToolkit = class {
+	#toolkit;
+	#mcpClients = /* @__PURE__ */ new Map();
+	constructor(options) {
+		this.#toolkit = options.toolkit;
+	}
+	async tools(options = {}) {
+		const frontendToolSet = options.frontend ? frontendTools(options.frontend) : {};
+		const mcpToolSet = await this.#mcpTools();
+		const providerToolSet = toProviderToolSet(this.#toolkit);
+		const serverToolSet = toServerToolSet(this.#toolkit);
+		assertNoMcpToolNameCollisions(mcpToolSet, [
+			{
+				source: "frontend",
+				tools: frontendToolSet
+			},
+			{
+				source: "provider",
+				tools: providerToolSet
+			},
+			{
+				source: "toolkit",
+				tools: serverToolSet
+			}
+		]);
+		return {
+			...frontendToolSet,
+			...mcpToolSet.tools,
+			...providerToolSet,
+			...serverToolSet
+		};
+	}
+	async close() {
+		const clientEntries = [...this.#mcpClients.entries()];
+		const clientNames = clientEntries.map(([name]) => name);
+		this.#mcpClients.clear();
+		const clientResults = await Promise.allSettled(clientEntries.map(([, clientPromise]) => clientPromise));
+		const clients = clientResults.flatMap((result, index) => result.status === "fulfilled" ? [[clientNames[index], result.value]] : []);
+		const closeResults = await Promise.allSettled(clients.map(([, client]) => client.close()));
+		const errors = [...clientResults.flatMap((result, index) => result.status === "rejected" ? [toMcpToolkitError(clientNames[index], "connect", result.reason)] : []), ...closeResults.flatMap((result, index) => result.status === "rejected" ? [toMcpToolkitError(clients[index][0], "close", result.reason)] : [])];
+		if (errors.length === 1) throw errors[0];
+		if (errors.length > 1) throw new AggregateError(errors, "Failed to close one or more MCP clients");
+	}
+	async #mcpTools() {
+		const toolSets = await Promise.all(Object.entries(this.#toolkit).filter((entry) => isMcpToolkitTool(entry[1])).map(async ([name, tool]) => {
+			const startedAt = Date.now();
+			const client = await this.#mcpClient(name, tool.server, startedAt).catch((error) => {
+				if (error instanceof MCPConnectionTimeoutError) throw error;
+				throw toMcpToolkitError(name, "connect", error);
+			});
+			try {
+				return [
+					name,
+					tool,
+					await withMcpConnectionTimeout(client.tools(), {
+						name,
+						config: tool.server,
+						phase: "listing tools",
+						startedAt
+					})
+				];
+			} catch (error) {
+				if (error instanceof MCPConnectionTimeoutError) {
+					this.#mcpClients.delete(name);
+					client.close().catch(() => {});
+					throw error;
+				}
+				throw toMcpToolkitError(name, "list tools", error);
+			}
+		}));
+		const tools = {};
+		const toolSources = /* @__PURE__ */ new Map();
+		for (const [serverName, mcpTool, toolSet] of toolSets) for (const [toolName, tool] of Object.entries(toolSet)) {
+			if (isDisabledMcpTool(mcpTool.tools?.[toolName])) continue;
+			const exposedName = `${mcpTool.prefix ?? ""}${toolName}`;
+			const existingServerName = toolSources.get(exposedName);
+			if (existingServerName) throw new Error(`MCP tool name collision: "${exposedName}" is exposed by both "${existingServerName}" and "${serverName}". Rename one of the toolkit entries or expose distinct MCP tool names.`);
+			toolSources.set(exposedName, serverName);
+			tools[exposedName] = tool;
+		}
+		return {
+			tools,
+			sources: toolSources
+		};
+	}
+	#mcpClient(name, config, startedAt) {
+		const existing = this.#mcpClients.get(name);
+		if (existing) return existing;
+		const createPromise = createMCPClient(toMCPClientConfig(config));
+		let next;
+		next = withMcpConnectionTimeout(createPromise, {
+			name,
+			config,
+			phase: "connecting",
+			startedAt
+		}).catch((error) => {
+			if (this.#mcpClients.get(name) === next) this.#mcpClients.delete(name);
+			if (error instanceof MCPConnectionTimeoutError) createPromise.then((client) => client.close()).catch(() => {});
+			throw error;
+		});
+		this.#mcpClients.set(name, next);
+		return next;
+	}
+};
+const toMCPClientConfig = (config) => {
+	if (config.type === "stdio") return { transport: new Experimental_StdioMCPTransport({
+		command: config.command,
+		...config.args && { args: [...config.args] },
+		...config.env && { env: config.env },
+		...config.cwd && { cwd: config.cwd }
+	}) };
+	return { transport: {
+		type: config.type,
+		url: config.url,
+		...config.headers && { headers: config.headers },
+		...config.redirect && { redirect: config.redirect }
+	} };
+};
+const assertNoMcpToolNameCollisions = (mcp, toolSets) => {
+	for (const [toolName, serverName] of mcp.sources) for (const { source, tools } of toolSets) {
+		if (!Object.prototype.hasOwnProperty.call(tools, toolName)) continue;
+		throw new Error(`MCP tool "${toolName}" from "${serverName}" conflicts with ${source} tool "${toolName}". Rename one of the tools so each model-visible tool name is unique.`);
+	}
+};
+const isMcpToolkitTool = (tool) => tool.type === "mcp" && !tool.disabled;
+const getErrorMessage = (error) => error instanceof Error ? error.message || error.name : String(error);
+const toMcpToolkitError = (entryName, action, error) => {
+	return new Error(`MCP toolkit entry "${entryName}" failed to ${action}: ${getErrorMessage(error)}`, { cause: error });
+};
+const isDisabledMcpTool = (config) => config?.disabled === true;
+const assertNoMcpToolkitTools = (toolkit) => {
+	const mcpToolName = Object.entries(toolkit).find(([, tool]) => isMcpToolkitTool(tool))?.[0];
+	if (!mcpToolName) return;
+	throw new Error(`MCP toolkit entry "${mcpToolName}" requires AISDKToolkit. Use new AISDKToolkit({ toolkit }).tools(...) instead of generativeTools(...).`);
+};
+const toAISDKToModelOutput = (toModelOutput) => async (options) => {
+	const { result, modelContent } = unwrapModelContentEnvelope(options.output);
+	if (modelContent !== void 0) return toAISDKContent(modelContent);
+	if (!toModelOutput) return toAISDKDefaultOutput(result);
+	const parts = await toModelOutput({
+		...options,
+		output: result
+	});
+	return toAISDKContent(parts);
+};
+const toServerToolSet = (toolkit) => Object.fromEntries(Object.entries(toolkit).filter(([, t]) => t.type !== "mcp" && t.type !== "provider" && !t.disabled).map(([name, t]) => {
+	const execute = t.execute;
+	return [name, {
+		...t.description !== void 0 && { description: t.description },
+		inputSchema: parametersToInputSchema(t.parameters),
+		toModelOutput: toAISDKToModelOutput(t.toModelOutput),
+		...t.providerOptions && { providerOptions: t.providerOptions },
+		...execute && { execute: (args, callOptions) => execute(args, {
+			toolCallId: callOptions.toolCallId,
+			abortSignal: callOptions.abortSignal ?? neverAbort,
+			human: humanNotSupported
+		}) }
+	}];
+}));
+const toProviderToolSet = (toolkit) => Object.fromEntries(Object.entries(toolkit).filter((entry) => isProviderToolkitTool(entry[1])).map(([name, t]) => [name, {
+	type: "provider",
+	id: t.providerId,
+	args: t.args,
+	...t.parameters && { inputSchema: parametersToInputSchema(t.parameters) },
+	...t.providerOptions && { providerOptions: t.providerOptions },
+	...t.supportsDeferredResults !== void 0 && { supportsDeferredResults: t.supportsDeferredResults }
+}]));
+const isProviderToolkitTool = (tool) => tool.type === "provider" && !tool.disabled;
+//#endregion
+export { AISDKToolkit, generativeTools };
+
+//# sourceMappingURL=generativeTools.js.map