@assistant-ui/react 0.7.44 → 0.7.45

package/dist/runtimes/remote-thread-list/RemoteThreadListThreadListRuntimeCore.mjs

@@ -122,7 +122,11 @@ var RemoteThreadListThreadListRuntimeCore = class extends BaseSubscribable {
               remoteId: thread.remoteId,
               externalId: thread.externalId,
               status: thread.status,
-              title: thread.title
+              title: thread.title,
+              initializeTask: Promise.resolve({
+                remoteId: thread.remoteId,
+                externalId: thread.externalId
+              })
             };
           }
           return {
@@ -243,8 +247,12 @@ var RemoteThreadListThreadListRuntimeCore = class extends BaseSubscribable {
     return this.switchToThread(threadId);
   }
   initialize = async (threadId) => {
-    if (this._state.value.newThreadId !== threadId)
-      throw new Error("The provided thread is already initialized");
+    if (this._state.value.newThreadId !== threadId) {
+      const data = this.getItemById(threadId);
+      if (!data) throw new Error("Thread not found");
+      if (data.status === "new") throw new Error("Unexpected new state");
+      return data.initializeTask;
+    }
     return this._state.optimisticUpdate({
       execute: () => {
         return this._adapter.initialize(threadId);
@@ -252,6 +260,19 @@ var RemoteThreadListThreadListRuntimeCore = class extends BaseSubscribable {
       optimistic: (state) => {
         return updateStatusReducer(state, threadId, "regular");
       },
+      loading: (state, task) => {
+        const mappingId = createThreadMappingId(threadId);
+        return {
+          ...state,
+          threadData: {
+            ...state.threadData,
+            [mappingId]: {
+              ...state.threadData[mappingId],
+              initializeTask: task
+            }
+          }
+        };
+      },
       then: (state, { remoteId, externalId }) => {
         const data = getThreadData(state, threadId);
         if (!data) return state;
@@ -264,8 +285,9 @@ var RemoteThreadListThreadListRuntimeCore = class extends BaseSubscribable {
           },
           threadData: {
             ...state.threadData,
-            [threadId]: {
+            [mappingId]: {
               ...data,
+              initializeTask: Promise.resolve({ remoteId, externalId }),
               remoteId,
               externalId
             }
@@ -300,8 +322,9 @@ var RemoteThreadListThreadListRuntimeCore = class extends BaseSubscribable {
     if (!data) throw new Error("Thread not found");
     if (data.status === "new") throw new Error("Thread is not yet initialized");
     return this._state.optimisticUpdate({
-      execute: () => {
-        return this._adapter.rename(data.remoteId, newTitle);
+      execute: async () => {
+        const { remoteId } = await data.initializeTask;
+        return this._adapter.rename(remoteId, newTitle);
       },
       optimistic: (state) => {
         const data2 = getThreadData(state, threadIdOrRemoteId);
@@ -337,7 +360,8 @@ var RemoteThreadListThreadListRuntimeCore = class extends BaseSubscribable {
     return this._state.optimisticUpdate({
       execute: async () => {
         await this._ensureThreadIsNotMain(data.threadId);
-        return this._adapter.archive(data.remoteId);
+        const { remoteId } = await data.initializeTask;
+        return this._adapter.archive(remoteId);
       },
       optimistic: (state) => {
         return updateStatusReducer(state, data.threadId, "archived");
@@ -351,7 +375,8 @@ var RemoteThreadListThreadListRuntimeCore = class extends BaseSubscribable {
     return this._state.optimisticUpdate({
       execute: async () => {
         try {
-          return await this._adapter.unarchive(data.remoteId);
+          const { remoteId } = await data.initializeTask;
+          return await this._adapter.unarchive(remoteId);
         } catch (error) {
           await this._ensureThreadIsNotMain(data.threadId);
           throw error;
@@ -370,7 +395,8 @@ var RemoteThreadListThreadListRuntimeCore = class extends BaseSubscribable {
     return this._state.optimisticUpdate({
       execute: async () => {
         await this._ensureThreadIsNotMain(data.threadId);
-        return await this._adapter.delete(data.remoteId);
+        const { remoteId } = await data.initializeTask;
+        return await this._adapter.delete(remoteId);
       },
       optimistic: (state) => {
         return updateStatusReducer(state, data.threadId, "deleted");

package/dist/runtimes/remote-thread-list/RemoteThreadListThreadListRuntimeCore.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/runtimes/remote-thread-list/RemoteThreadListThreadListRuntimeCore.tsx"],"sourcesContent":["\"use client\";\n\nimport { ThreadListRuntimeCore } from \"../core/ThreadListRuntimeCore\";\nimport { generateId } from \"../../internal\";\nimport { RemoteThreadListAdapter } from \"./types\";\nimport { RemoteThreadListHookInstanceManager } from \"./RemoteThreadListHookInstanceManager\";\nimport { BaseSubscribable } from \"./BaseSubscribable\";\nimport { EMPTY_THREAD_CORE } from \"./EMPTY_THREAD_CORE\";\nimport { OptimisticState } from \"./OptimisticState\";\nimport { FC, Fragment, useEffect, useId } from \"react\";\nimport { create } from \"zustand\";\nimport { AssistantStream, AssistantMessageStream } from \"assistant-stream\";\n\ntype RemoteThreadData =\n  | {\n      readonly threadId: string;\n      readonly remoteId?: undefined;\n      readonly externalId?: undefined;\n      readonly status: \"new\";\n      readonly title: undefined;\n    }\n  | {\n      readonly threadId: string;\n      readonly remoteId: string;\n      readonly externalId: string | undefined;\n      readonly status: \"regular\" | \"archived\";\n      readonly title?: string | undefined;\n    };\n\ntype THREAD_MAPPING_ID = string & { __brand: \"THREAD_MAPPING_ID\" };\nfunction createThreadMappingId(id: string): THREAD_MAPPING_ID {\n  return id as THREAD_MAPPING_ID;\n}\n\ntype RemoteThreadState = {\n  readonly isLoading: boolean;\n  readonly newThreadId: string | undefined;\n  readonly threadIds: readonly string[];\n  readonly archivedThreadIds: readonly string[];\n  readonly threadIdMap: Readonly<Record<string, THREAD_MAPPING_ID>>;\n  readonly threadData: Readonly<Record<THREAD_MAPPING_ID, RemoteThreadData>>;\n};\n\nconst getThreadData = (\n  state: RemoteThreadState,\n  threadIdOrRemoteId: string,\n) => {\n  const idx = state.threadIdMap[threadIdOrRemoteId];\n  if (idx === undefined) return undefined;\n  return state.threadData[idx];\n};\n\nconst updateStatusReducer = (\n  state: RemoteThreadState,\n  threadIdOrRemoteId: string,\n  newStatus: \"regular\" | \"archived\" | \"deleted\",\n) => {\n  const data = getThreadData(state, threadIdOrRemoteId);\n  if (!data) return state;\n\n  const { threadId, remoteId, status: lastStatus } = data;\n  if (lastStatus === newStatus) return state;\n\n  const newState = { ...state };\n\n  // lastStatus\n  switch (lastStatus) {\n    case \"new\":\n      newState.newThreadId = undefined;\n      break;\n    case \"regular\":\n      newState.threadIds = newState.threadIds.filter((t) => t !== threadId);\n      break;\n    case \"archived\":\n      newState.archivedThreadIds = newState.archivedThreadIds.filter(\n        (t) => t !== threadId,\n      );\n      break;\n\n    default: {\n      const _exhaustiveCheck: never = lastStatus;\n      throw new Error(`Unsupported state: ${_exhaustiveCheck}`);\n    }\n  }\n\n  // newStatus\n  switch (newStatus) {\n    case \"regular\":\n      newState.threadIds = [threadId, ...newState.threadIds];\n      break;\n\n    case \"archived\":\n      newState.archivedThreadIds = [threadId, ...newState.archivedThreadIds];\n      break;\n\n    case \"deleted\":\n      newState.threadData = Object.fromEntries(\n        Object.entries(newState.threadData).filter(([key]) => key !== threadId),\n      );\n      newState.threadIdMap = Object.fromEntries(\n        Object.entries(newState.threadIdMap).filter(\n          ([key]) => key !== threadId && key !== remoteId,\n        ),\n      );\n      break;\n\n    default: {\n      const _exhaustiveCheck: never = newStatus;\n      throw new Error(`Unsupported state: ${_exhaustiveCheck}`);\n    }\n  }\n\n  if (newStatus !== \"deleted\") {\n    newState.threadData = {\n      ...newState.threadData,\n      [threadId]: {\n        ...data,\n        status: newStatus,\n      },\n    };\n  }\n\n  return newState;\n};\n\nexport class RemoteThreadListThreadListRuntimeCore\n  extends BaseSubscribable\n  implements ThreadListRuntimeCore\n{\n  private _adapter!: RemoteThreadListAdapter;\n  private readonly _hookManager: RemoteThreadListHookInstanceManager;\n\n  private _loadThreadsPromise: Promise<void> | undefined;\n\n  private _mainThreadId!: string;\n  private readonly _state = new OptimisticState<RemoteThreadState>({\n    isLoading: false,\n    newThreadId: undefined,\n    threadIds: [],\n    archivedThreadIds: [],\n    threadIdMap: {},\n    threadData: {},\n  });\n\n  public getLoadThreadsPromise() {\n    // TODO this needs to be cached in case this promise is loaded during suspense\n    if (!this._loadThreadsPromise) {\n      this._loadThreadsPromise = this._state\n        .optimisticUpdate({\n          execute: () => this._adapter.list(),\n          loading: (state) => {\n            return {\n              ...state,\n              isLoading: true,\n            };\n          },\n          then: (state, l) => {\n            const newThreadIds = [];\n            const newArchivedThreadIds = [];\n            const newThreadIdMap = {} as Record<string, THREAD_MAPPING_ID>;\n            const newThreadData = {} as Record<\n              THREAD_MAPPING_ID,\n              RemoteThreadData\n            >;\n\n            for (const thread of l.threads) {\n              switch (thread.status) {\n                case \"regular\":\n                  newThreadIds.push(thread.remoteId);\n                  break;\n                case \"archived\":\n                  newArchivedThreadIds.push(thread.remoteId);\n                  break;\n                default: {\n                  const _exhaustiveCheck: never = thread.status;\n                  throw new Error(`Unsupported state: ${_exhaustiveCheck}`);\n                }\n              }\n\n              const mappingId = createThreadMappingId(thread.remoteId);\n              newThreadIdMap[thread.remoteId] = mappingId;\n              newThreadData[mappingId] = {\n                threadId: thread.remoteId,\n                remoteId: thread.remoteId,\n                externalId: thread.externalId,\n                status: thread.status,\n                title: thread.title,\n              };\n            }\n\n            return {\n              ...state,\n              threadIds: newThreadIds,\n              archivedThreadIds: newArchivedThreadIds,\n              threadIdMap: {\n                ...state.threadIdMap,\n                ...newThreadIdMap,\n              },\n              threadData: {\n                ...state.threadData,\n                ...newThreadData,\n              },\n            };\n          },\n        })\n        .then(() => {});\n    }\n\n    return this._loadThreadsPromise;\n  }\n\n  constructor(adapter: RemoteThreadListAdapter) {\n    super();\n\n    this._state.subscribe(() => this._notifySubscribers());\n    this._hookManager = new RemoteThreadListHookInstanceManager(\n      adapter.runtimeHook,\n    );\n    this.useProvider = create(() => ({\n      Provider: adapter.unstable_Provider ?? Fragment,\n    }));\n    this.__internal_setAdapter(adapter);\n\n    this.switchToNewThread();\n  }\n\n  private useProvider;\n\n  public __internal_setAdapter(adapter: RemoteThreadListAdapter) {\n    if (this._adapter === adapter) return;\n\n    this._adapter = adapter;\n\n    const Provider = adapter.unstable_Provider ?? Fragment;\n    if (Provider !== this.useProvider.getState().Provider) {\n      this.useProvider.setState({ Provider }, true);\n    }\n\n    this._hookManager.setRuntimeHook(adapter.runtimeHook);\n  }\n\n  public __internal_bindAdapter() {\n    this.getLoadThreadsPromise(); // begin loading on initial bind\n    return this._adapter.subscribe?.({\n      onGenerateTitle: this._onGenerateTitle,\n    });\n  }\n\n  public get threadIds() {\n    return this._state.value.threadIds;\n  }\n\n  public get archivedThreadIds() {\n    return this._state.value.archivedThreadIds;\n  }\n\n  public get newThreadId() {\n    return this._state.value.newThreadId;\n  }\n\n  public get mainThreadId(): string {\n    return this._mainThreadId;\n  }\n\n  public getMainThreadRuntimeCore() {\n    const result = this._hookManager.getThreadRuntimeCore(this._mainThreadId);\n    if (!result) return EMPTY_THREAD_CORE;\n    return result;\n  }\n\n  public getThreadRuntimeCore(threadIdOrRemoteId: string) {\n    const data = this.getItemById(threadIdOrRemoteId);\n    if (!data) throw new Error(\"Thread not found\");\n\n    const result = this._hookManager.getThreadRuntimeCore(data.threadId);\n    if (!result) throw new Error(\"Thread not found\");\n    return result;\n  }\n\n  public getItemById(threadIdOrRemoteId: string) {\n    return getThreadData(this._state.value, threadIdOrRemoteId);\n  }\n\n  public async switchToThread(threadIdOrRemoteId: string): Promise<void> {\n    const data = this.getItemById(threadIdOrRemoteId);\n    if (!data) throw new Error(\"Thread not found\");\n\n    if (this._mainThreadId === data.threadId) return;\n\n    const task = this._hookManager.startThreadRuntime(data.threadId);\n    if (this.mainThreadId !== undefined) {\n      await task;\n    } else {\n      task.then(() => this._notifySubscribers());\n    }\n\n    if (data.status === \"archived\") await this.unarchive(data.threadId);\n    this._mainThreadId = data.threadId;\n\n    this._notifySubscribers();\n  }\n\n  public async switchToNewThread(): Promise<void> {\n    // an initialization transaction is in progress, wait for it to settle\n    while (\n      this._state.baseValue.newThreadId !== undefined &&\n      this._state.value.newThreadId === undefined\n    ) {\n      await this._state.waitForUpdate();\n    }\n\n    const state = this._state.value;\n    let threadId: string | undefined = this._state.value.newThreadId;\n    if (threadId === undefined) {\n      do {\n        threadId = `__LOCALID_${generateId()}`;\n      } while (state.threadIdMap[threadId]);\n\n      const mappingId = createThreadMappingId(threadId);\n      this._state.update({\n        ...state,\n        newThreadId: threadId,\n        threadIdMap: {\n          ...state.threadIdMap,\n          [threadId]: mappingId,\n        },\n        threadData: {\n          ...state.threadData,\n          [threadId]: {\n            status: \"new\",\n            threadId,\n          },\n        },\n      });\n    }\n\n    return this.switchToThread(threadId);\n  }\n\n  public initialize = async (threadId: string) => {\n    if (this._state.value.newThreadId !== threadId)\n      throw new Error(\"The provided thread is already initialized\");\n\n    return this._state.optimisticUpdate({\n      execute: () => {\n        return this._adapter.initialize(threadId);\n      },\n      optimistic: (state) => {\n        return updateStatusReducer(state, threadId, \"regular\");\n      },\n      then: (state, { remoteId, externalId }) => {\n        const data = getThreadData(state, threadId);\n        if (!data) return state;\n\n        const mappingId = createThreadMappingId(threadId);\n        return {\n          ...state,\n          threadIdMap: {\n            ...state.threadIdMap,\n            [remoteId]: mappingId,\n          },\n          threadData: {\n            ...state.threadData,\n            [threadId]: {\n              ...data,\n              remoteId,\n              externalId,\n            },\n          },\n        };\n      },\n    });\n  };\n\n  private _onGenerateTitle = async (\n    remoteId: string,\n    begin: () => Promise<AssistantStream>,\n  ) => {\n    const data = this.getItemById(remoteId);\n    if (!data) throw new Error(\"Thread not found\");\n    if (data.status === \"new\") throw new Error(\"Thread is not yet initialized\");\n\n    const stream = await begin();\n    const messageStream = AssistantMessageStream.fromAssistantStream(stream);\n    for await (const result of messageStream) {\n      const newTitle =\n        result.content.filter((c) => c.type === \"text\")[0]?.text ??\n        \"New Thread\";\n      const state = this._state.baseValue;\n      this._state.update({\n        ...state,\n        threadData: {\n          ...state.threadData,\n          [data.threadId]: {\n            ...data,\n            title: newTitle,\n          },\n        },\n      });\n    }\n  };\n\n  public rename(threadIdOrRemoteId: string, newTitle: string): Promise<void> {\n    const data = this.getItemById(threadIdOrRemoteId);\n    if (!data) throw new Error(\"Thread not found\");\n    if (data.status === \"new\") throw new Error(\"Thread is not yet initialized\");\n\n    return this._state.optimisticUpdate({\n      execute: () => {\n        return this._adapter.rename(data.remoteId, newTitle);\n      },\n      optimistic: (state) => {\n        const data = getThreadData(state, threadIdOrRemoteId);\n        if (!data) return state;\n\n        return {\n          ...state,\n          threadData: {\n            ...state.threadData,\n            [data.threadId]: {\n              ...data,\n              title: newTitle,\n            },\n          },\n        };\n      },\n    });\n  }\n\n  private async _ensureThreadIsNotMain(threadId: string) {\n    // if thread is main thread, switch to another thread\n    if (threadId === this._mainThreadId) {\n      const lastThreadId = this._state.value.threadIds[0];\n      if (lastThreadId) {\n        await this.switchToThread(lastThreadId);\n      } else {\n        await this.switchToNewThread();\n      }\n    }\n  }\n\n  public async archive(threadIdOrRemoteId: string) {\n    const data = this.getItemById(threadIdOrRemoteId);\n    if (!data) throw new Error(\"Thread not found\");\n    if (data.status !== \"regular\")\n      throw new Error(\"Thread is not yet initialized or already archived\");\n\n    return this._state.optimisticUpdate({\n      execute: async () => {\n        await this._ensureThreadIsNotMain(data.threadId);\n        return this._adapter.archive(data.remoteId);\n      },\n      optimistic: (state) => {\n        return updateStatusReducer(state, data.threadId, \"archived\");\n      },\n    });\n  }\n\n  public unarchive(threadIdOrRemoteId: string): Promise<void> {\n    const data = this.getItemById(threadIdOrRemoteId);\n    if (!data) throw new Error(\"Thread not found\");\n    if (data.status !== \"archived\") throw new Error(\"Thread is not archived\");\n\n    return this._state.optimisticUpdate({\n      execute: async () => {\n        try {\n          return await this._adapter.unarchive(data.remoteId);\n        } catch (error) {\n          await this._ensureThreadIsNotMain(data.threadId);\n          throw error;\n        }\n      },\n      optimistic: (state) => {\n        return updateStatusReducer(state, data.threadId, \"regular\");\n      },\n    });\n  }\n\n  public async delete(threadIdOrRemoteId: string) {\n    const data = this.getItemById(threadIdOrRemoteId);\n    if (!data) throw new Error(\"Thread not found\");\n    if (data.status !== \"regular\" && data.status !== \"archived\")\n      throw new Error(\"Thread is not yet initialized\");\n\n    return this._state.optimisticUpdate({\n      execute: async () => {\n        await this._ensureThreadIsNotMain(data.threadId);\n        return await this._adapter.delete(data.remoteId);\n      },\n      optimistic: (state) => {\n        return updateStatusReducer(state, data.threadId, \"deleted\");\n      },\n    });\n  }\n\n  private useBoundIds = create<string[]>(() => []);\n\n  public __internal_RenderComponent: FC = () => {\n    const id = useId();\n    useEffect(() => {\n      this.useBoundIds.setState((s) => [...s, id], true);\n      return () => {\n        this.useBoundIds.setState((s) => s.filter((i) => i !== id), true);\n      };\n    }, []);\n\n    const boundIds = this.useBoundIds();\n    const { Provider } = this.useProvider();\n\n    return (\n      (boundIds.length === 0 || boundIds[0] === id) && (\n        // only render if the component is the first one mounted\n        <this._hookManager.__internal_RenderThreadRuntimes\n          provider={Provider}\n        />\n      )\n    );\n  };\n}\n"],"mappings":";;;AAGA,SAAS,kBAAkB;AAE3B,SAAS,2CAA2C;AACpD,SAAS,wBAAwB;AACjC,SAAS,yBAAyB;AAClC,SAAS,uBAAuB;AAChC,SAAa,UAAU,WAAW,aAAa;AAC/C,SAAS,cAAc;AACvB,SAA0B,8BAA8B;AAqfhD;AAleR,SAAS,sBAAsB,IAA+B;AAC5D,SAAO;AACT;AAWA,IAAM,gBAAgB,CACpB,OACA,uBACG;AACH,QAAM,MAAM,MAAM,YAAY,kBAAkB;AAChD,MAAI,QAAQ,OAAW,QAAO;AAC9B,SAAO,MAAM,WAAW,GAAG;AAC7B;AAEA,IAAM,sBAAsB,CAC1B,OACA,oBACA,cACG;AACH,QAAM,OAAO,cAAc,OAAO,kBAAkB;AACpD,MAAI,CAAC,KAAM,QAAO;AAElB,QAAM,EAAE,UAAU,UAAU,QAAQ,WAAW,IAAI;AACnD,MAAI,eAAe,UAAW,QAAO;AAErC,QAAM,WAAW,EAAE,GAAG,MAAM;AAG5B,UAAQ,YAAY;AAAA,IAClB,KAAK;AACH,eAAS,cAAc;AACvB;AAAA,IACF,KAAK;AACH,eAAS,YAAY,SAAS,UAAU,OAAO,CAAC,MAAM,MAAM,QAAQ;AACpE;AAAA,IACF,KAAK;AACH,eAAS,oBAAoB,SAAS,kBAAkB;AAAA,QACtD,CAAC,MAAM,MAAM;AAAA,MACf;AACA;AAAA,IAEF,SAAS;AACP,YAAM,mBAA0B;AAChC,YAAM,IAAI,MAAM,sBAAsB,gBAAgB,EAAE;AAAA,IAC1D;AAAA,EACF;AAGA,UAAQ,WAAW;AAAA,IACjB,KAAK;AACH,eAAS,YAAY,CAAC,UAAU,GAAG,SAAS,SAAS;AACrD;AAAA,IAEF,KAAK;AACH,eAAS,oBAAoB,CAAC,UAAU,GAAG,SAAS,iBAAiB;AACrE;AAAA,IAEF,KAAK;AACH,eAAS,aAAa,OAAO;AAAA,QAC3B,OAAO,QAAQ,SAAS,UAAU,EAAE,OAAO,CAAC,CAAC,GAAG,MAAM,QAAQ,QAAQ;AAAA,MACxE;AACA,eAAS,cAAc,OAAO;AAAA,QAC5B,OAAO,QAAQ,SAAS,WAAW,EAAE;AAAA,UACnC,CAAC,CAAC,GAAG,MAAM,QAAQ,YAAY,QAAQ;AAAA,QACzC;AAAA,MACF;AACA;AAAA,IAEF,SAAS;AACP,YAAM,mBAA0B;AAChC,YAAM,IAAI,MAAM,sBAAsB,gBAAgB,EAAE;AAAA,IAC1D;AAAA,EACF;AAEA,MAAI,cAAc,WAAW;AAC3B,aAAS,aAAa;AAAA,MACpB,GAAG,SAAS;AAAA,MACZ,CAAC,QAAQ,GAAG;AAAA,QACV,GAAG;AAAA,QACH,QAAQ;AAAA,MACV;AAAA,IACF;AAAA,EACF;AAEA,SAAO;AACT;AAEO,IAAM,wCAAN,cACG,iBAEV;AAAA,EACU;AAAA,EACS;AAAA,EAET;AAAA,EAEA;AAAA,EACS,SAAS,IAAI,gBAAmC;AAAA,IAC/D,WAAW;AAAA,IACX,aAAa;AAAA,IACb,WAAW,CAAC;AAAA,IACZ,mBAAmB,CAAC;AAAA,IACpB,aAAa,CAAC;AAAA,IACd,YAAY,CAAC;AAAA,EACf,CAAC;AAAA,EAEM,wBAAwB;AAE7B,QAAI,CAAC,KAAK,qBAAqB;AAC7B,WAAK,sBAAsB,KAAK,OAC7B,iBAAiB;AAAA,QAChB,SAAS,MAAM,KAAK,SAAS,KAAK;AAAA,QAClC,SAAS,CAAC,UAAU;AAClB,iBAAO;AAAA,YACL,GAAG;AAAA,YACH,WAAW;AAAA,UACb;AAAA,QACF;AAAA,QACA,MAAM,CAAC,OAAO,MAAM;AAClB,gBAAM,eAAe,CAAC;AACtB,gBAAM,uBAAuB,CAAC;AAC9B,gBAAM,iBAAiB,CAAC;AACxB,gBAAM,gBAAgB,CAAC;AAKvB,qBAAW,UAAU,EAAE,SAAS;AAC9B,oBAAQ,OAAO,QAAQ;AAAA,cACrB,KAAK;AACH,6BAAa,KAAK,OAAO,QAAQ;AACjC;AAAA,cACF,KAAK;AACH,qCAAqB,KAAK,OAAO,QAAQ;AACzC;AAAA,cACF,SAAS;AACP,sBAAM,mBAA0B,OAAO;AACvC,sBAAM,IAAI,MAAM,sBAAsB,gBAAgB,EAAE;AAAA,cAC1D;AAAA,YACF;AAEA,kBAAM,YAAY,sBAAsB,OAAO,QAAQ;AACvD,2BAAe,OAAO,QAAQ,IAAI;AAClC,0BAAc,SAAS,IAAI;AAAA,cACzB,UAAU,OAAO;AAAA,cACjB,UAAU,OAAO;AAAA,cACjB,YAAY,OAAO;AAAA,cACnB,QAAQ,OAAO;AAAA,cACf,OAAO,OAAO;AAAA,YAChB;AAAA,UACF;AAEA,iBAAO;AAAA,YACL,GAAG;AAAA,YACH,WAAW;AAAA,YACX,mBAAmB;AAAA,YACnB,aAAa;AAAA,cACX,GAAG,MAAM;AAAA,cACT,GAAG;AAAA,YACL;AAAA,YACA,YAAY;AAAA,cACV,GAAG,MAAM;AAAA,cACT,GAAG;AAAA,YACL;AAAA,UACF;AAAA,QACF;AAAA,MACF,CAAC,EACA,KAAK,MAAM;AAAA,MAAC,CAAC;AAAA,IAClB;AAEA,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,YAAY,SAAkC;AAC5C,UAAM;AAEN,SAAK,OAAO,UAAU,MAAM,KAAK,mBAAmB,CAAC;AACrD,SAAK,eAAe,IAAI;AAAA,MACtB,QAAQ;AAAA,IACV;AACA,SAAK,cAAc,OAAO,OAAO;AAAA,MAC/B,UAAU,QAAQ,qBAAqB;AAAA,IACzC,EAAE;AACF,SAAK,sBAAsB,OAAO;AAElC,SAAK,kBAAkB;AAAA,EACzB;AAAA,EAEQ;AAAA,EAED,sBAAsB,SAAkC;AAC7D,QAAI,KAAK,aAAa,QAAS;AAE/B,SAAK,WAAW;AAEhB,UAAM,WAAW,QAAQ,qBAAqB;AAC9C,QAAI,aAAa,KAAK,YAAY,SAAS,EAAE,UAAU;AACrD,WAAK,YAAY,SAAS,EAAE,SAAS,GAAG,IAAI;AAAA,IAC9C;AAEA,SAAK,aAAa,eAAe,QAAQ,WAAW;AAAA,EACtD;AAAA,EAEO,yBAAyB;AAC9B,SAAK,sBAAsB;AAC3B,WAAO,KAAK,SAAS,YAAY;AAAA,MAC/B,iBAAiB,KAAK;AAAA,IACxB,CAAC;AAAA,EACH;AAAA,EAEA,IAAW,YAAY;AACrB,WAAO,KAAK,OAAO,MAAM;AAAA,EAC3B;AAAA,EAEA,IAAW,oBAAoB;AAC7B,WAAO,KAAK,OAAO,MAAM;AAAA,EAC3B;AAAA,EAEA,IAAW,cAAc;AACvB,WAAO,KAAK,OAAO,MAAM;AAAA,EAC3B;AAAA,EAEA,IAAW,eAAuB;AAChC,WAAO,KAAK;AAAA,EACd;AAAA,EAEO,2BAA2B;AAChC,UAAM,SAAS,KAAK,aAAa,qBAAqB,KAAK,aAAa;AACxE,QAAI,CAAC,OAAQ,QAAO;AACpB,WAAO;AAAA,EACT;AAAA,EAEO,qBAAqB,oBAA4B;AACtD,UAAM,OAAO,KAAK,YAAY,kBAAkB;AAChD,QAAI,CAAC,KAAM,OAAM,IAAI,MAAM,kBAAkB;AAE7C,UAAM,SAAS,KAAK,aAAa,qBAAqB,KAAK,QAAQ;AACnE,QAAI,CAAC,OAAQ,OAAM,IAAI,MAAM,kBAAkB;AAC/C,WAAO;AAAA,EACT;AAAA,EAEO,YAAY,oBAA4B;AAC7C,WAAO,cAAc,KAAK,OAAO,OAAO,kBAAkB;AAAA,EAC5D;AAAA,EAEA,MAAa,eAAe,oBAA2C;AACrE,UAAM,OAAO,KAAK,YAAY,kBAAkB;AAChD,QAAI,CAAC,KAAM,OAAM,IAAI,MAAM,kBAAkB;AAE7C,QAAI,KAAK,kBAAkB,KAAK,SAAU;AAE1C,UAAM,OAAO,KAAK,aAAa,mBAAmB,KAAK,QAAQ;AAC/D,QAAI,KAAK,iBAAiB,QAAW;AACnC,YAAM;AAAA,IACR,OAAO;AACL,WAAK,KAAK,MAAM,KAAK,mBAAmB,CAAC;AAAA,IAC3C;AAEA,QAAI,KAAK,WAAW,WAAY,OAAM,KAAK,UAAU,KAAK,QAAQ;AAClE,SAAK,gBAAgB,KAAK;AAE1B,SAAK,mBAAmB;AAAA,EAC1B;AAAA,EAEA,MAAa,oBAAmC;AAE9C,WACE,KAAK,OAAO,UAAU,gBAAgB,UACtC,KAAK,OAAO,MAAM,gBAAgB,QAClC;AACA,YAAM,KAAK,OAAO,cAAc;AAAA,IAClC;AAEA,UAAM,QAAQ,KAAK,OAAO;AAC1B,QAAI,WAA+B,KAAK,OAAO,MAAM;AACrD,QAAI,aAAa,QAAW;AAC1B,SAAG;AACD,mBAAW,aAAa,WAAW,CAAC;AAAA,MACtC,SAAS,MAAM,YAAY,QAAQ;AAEnC,YAAM,YAAY,sBAAsB,QAAQ;AAChD,WAAK,OAAO,OAAO;AAAA,QACjB,GAAG;AAAA,QACH,aAAa;AAAA,QACb,aAAa;AAAA,UACX,GAAG,MAAM;AAAA,UACT,CAAC,QAAQ,GAAG;AAAA,QACd;AAAA,QACA,YAAY;AAAA,UACV,GAAG,MAAM;AAAA,UACT,CAAC,QAAQ,GAAG;AAAA,YACV,QAAQ;AAAA,YACR;AAAA,UACF;AAAA,QACF;AAAA,MACF,CAAC;AAAA,IACH;AAEA,WAAO,KAAK,eAAe,QAAQ;AAAA,EACrC;AAAA,EAEO,aAAa,OAAO,aAAqB;AAC9C,QAAI,KAAK,OAAO,MAAM,gBAAgB;AACpC,YAAM,IAAI,MAAM,4CAA4C;AAE9D,WAAO,KAAK,OAAO,iBAAiB;AAAA,MAClC,SAAS,MAAM;AACb,eAAO,KAAK,SAAS,WAAW,QAAQ;AAAA,MAC1C;AAAA,MACA,YAAY,CAAC,UAAU;AACrB,eAAO,oBAAoB,OAAO,UAAU,SAAS;AAAA,MACvD;AAAA,MACA,MAAM,CAAC,OAAO,EAAE,UAAU,WAAW,MAAM;AACzC,cAAM,OAAO,cAAc,OAAO,QAAQ;AAC1C,YAAI,CAAC,KAAM,QAAO;AAElB,cAAM,YAAY,sBAAsB,QAAQ;AAChD,eAAO;AAAA,UACL,GAAG;AAAA,UACH,aAAa;AAAA,YACX,GAAG,MAAM;AAAA,YACT,CAAC,QAAQ,GAAG;AAAA,UACd;AAAA,UACA,YAAY;AAAA,YACV,GAAG,MAAM;AAAA,YACT,CAAC,QAAQ,GAAG;AAAA,cACV,GAAG;AAAA,cACH;AAAA,cACA;AAAA,YACF;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEQ,mBAAmB,OACzB,UACA,UACG;AACH,UAAM,OAAO,KAAK,YAAY,QAAQ;AACtC,QAAI,CAAC,KAAM,OAAM,IAAI,MAAM,kBAAkB;AAC7C,QAAI,KAAK,WAAW,MAAO,OAAM,IAAI,MAAM,+BAA+B;AAE1E,UAAM,SAAS,MAAM,MAAM;AAC3B,UAAM,gBAAgB,uBAAuB,oBAAoB,MAAM;AACvE,qBAAiB,UAAU,eAAe;AACxC,YAAM,WACJ,OAAO,QAAQ,OAAO,CAAC,MAAM,EAAE,SAAS,MAAM,EAAE,CAAC,GAAG,QACpD;AACF,YAAM,QAAQ,KAAK,OAAO;AAC1B,WAAK,OAAO,OAAO;AAAA,QACjB,GAAG;AAAA,QACH,YAAY;AAAA,UACV,GAAG,MAAM;AAAA,UACT,CAAC,KAAK,QAAQ,GAAG;AAAA,YACf,GAAG;AAAA,YACH,OAAO;AAAA,UACT;AAAA,QACF;AAAA,MACF,CAAC;AAAA,IACH;AAAA,EACF;AAAA,EAEO,OAAO,oBAA4B,UAAiC;AACzE,UAAM,OAAO,KAAK,YAAY,kBAAkB;AAChD,QAAI,CAAC,KAAM,OAAM,IAAI,MAAM,kBAAkB;AAC7C,QAAI,KAAK,WAAW,MAAO,OAAM,IAAI,MAAM,+BAA+B;AAE1E,WAAO,KAAK,OAAO,iBAAiB;AAAA,MAClC,SAAS,MAAM;AACb,eAAO,KAAK,SAAS,OAAO,KAAK,UAAU,QAAQ;AAAA,MACrD;AAAA,MACA,YAAY,CAAC,UAAU;AACrB,cAAMA,QAAO,cAAc,OAAO,kBAAkB;AACpD,YAAI,CAACA,MAAM,QAAO;AAElB,eAAO;AAAA,UACL,GAAG;AAAA,UACH,YAAY;AAAA,YACV,GAAG,MAAM;AAAA,YACT,CAACA,MAAK,QAAQ,GAAG;AAAA,cACf,GAAGA;AAAA,cACH,OAAO;AAAA,YACT;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEA,MAAc,uBAAuB,UAAkB;AAErD,QAAI,aAAa,KAAK,eAAe;AACnC,YAAM,eAAe,KAAK,OAAO,MAAM,UAAU,CAAC;AAClD,UAAI,cAAc;AAChB,cAAM,KAAK,eAAe,YAAY;AAAA,MACxC,OAAO;AACL,cAAM,KAAK,kBAAkB;AAAA,MAC/B;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAa,QAAQ,oBAA4B;AAC/C,UAAM,OAAO,KAAK,YAAY,kBAAkB;AAChD,QAAI,CAAC,KAAM,OAAM,IAAI,MAAM,kBAAkB;AAC7C,QAAI,KAAK,WAAW;AAClB,YAAM,IAAI,MAAM,mDAAmD;AAErE,WAAO,KAAK,OAAO,iBAAiB;AAAA,MAClC,SAAS,YAAY;AACnB,cAAM,KAAK,uBAAuB,KAAK,QAAQ;AAC/C,eAAO,KAAK,SAAS,QAAQ,KAAK,QAAQ;AAAA,MAC5C;AAAA,MACA,YAAY,CAAC,UAAU;AACrB,eAAO,oBAAoB,OAAO,KAAK,UAAU,UAAU;AAAA,MAC7D;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEO,UAAU,oBAA2C;AAC1D,UAAM,OAAO,KAAK,YAAY,kBAAkB;AAChD,QAAI,CAAC,KAAM,OAAM,IAAI,MAAM,kBAAkB;AAC7C,QAAI,KAAK,WAAW,WAAY,OAAM,IAAI,MAAM,wBAAwB;AAExE,WAAO,KAAK,OAAO,iBAAiB;AAAA,MAClC,SAAS,YAAY;AACnB,YAAI;AACF,iBAAO,MAAM,KAAK,SAAS,UAAU,KAAK,QAAQ;AAAA,QACpD,SAAS,OAAO;AACd,gBAAM,KAAK,uBAAuB,KAAK,QAAQ;AAC/C,gBAAM;AAAA,QACR;AAAA,MACF;AAAA,MACA,YAAY,CAAC,UAAU;AACrB,eAAO,oBAAoB,OAAO,KAAK,UAAU,SAAS;AAAA,MAC5D;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEA,MAAa,OAAO,oBAA4B;AAC9C,UAAM,OAAO,KAAK,YAAY,kBAAkB;AAChD,QAAI,CAAC,KAAM,OAAM,IAAI,MAAM,kBAAkB;AAC7C,QAAI,KAAK,WAAW,aAAa,KAAK,WAAW;AAC/C,YAAM,IAAI,MAAM,+BAA+B;AAEjD,WAAO,KAAK,OAAO,iBAAiB;AAAA,MAClC,SAAS,YAAY;AACnB,cAAM,KAAK,uBAAuB,KAAK,QAAQ;AAC/C,eAAO,MAAM,KAAK,SAAS,OAAO,KAAK,QAAQ;AAAA,MACjD;AAAA,MACA,YAAY,CAAC,UAAU;AACrB,eAAO,oBAAoB,OAAO,KAAK,UAAU,SAAS;AAAA,MAC5D;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEQ,cAAc,OAAiB,MAAM,CAAC,CAAC;AAAA,EAExC,6BAAiC,MAAM;AAC5C,UAAM,KAAK,MAAM;AACjB,cAAU,MAAM;AACd,WAAK,YAAY,SAAS,CAAC,MAAM,CAAC,GAAG,GAAG,EAAE,GAAG,IAAI;AACjD,aAAO,MAAM;AACX,aAAK,YAAY,SAAS,CAAC,MAAM,EAAE,OAAO,CAAC,MAAM,MAAM,EAAE,GAAG,IAAI;AAAA,MAClE;AAAA,IACF,GAAG,CAAC,CAAC;AAEL,UAAM,WAAW,KAAK,YAAY;AAClC,UAAM,EAAE,SAAS,IAAI,KAAK,YAAY;AAEtC,YACG,SAAS,WAAW,KAAK,SAAS,CAAC,MAAM;AAAA,IAExC;AAAA,MAAC,KAAK,aAAa;AAAA,MAAlB;AAAA,QACC,UAAU;AAAA;AAAA,IACZ;AAAA,EAGN;AACF;","names":["data"]}
+{"version":3,"sources":["../../../src/runtimes/remote-thread-list/RemoteThreadListThreadListRuntimeCore.tsx"],"sourcesContent":["\"use client\";\n\nimport { ThreadListRuntimeCore } from \"../core/ThreadListRuntimeCore\";\nimport { generateId } from \"../../internal\";\nimport {\n  RemoteThreadInitializeResponse,\n  RemoteThreadListAdapter,\n} from \"./types\";\nimport { RemoteThreadListHookInstanceManager } from \"./RemoteThreadListHookInstanceManager\";\nimport { BaseSubscribable } from \"./BaseSubscribable\";\nimport { EMPTY_THREAD_CORE } from \"./EMPTY_THREAD_CORE\";\nimport { OptimisticState } from \"./OptimisticState\";\nimport { FC, Fragment, useEffect, useId } from \"react\";\nimport { create } from \"zustand\";\nimport { AssistantStream, AssistantMessageStream } from \"assistant-stream\";\n\ntype RemoteThreadData =\n  | {\n      readonly threadId: string;\n      readonly remoteId?: undefined;\n      readonly externalId?: undefined;\n      readonly status: \"new\";\n      readonly title: undefined;\n    }\n  | {\n      readonly threadId: string;\n      readonly initializeTask: Promise<RemoteThreadInitializeResponse>;\n      readonly remoteId?: undefined;\n      readonly externalId?: undefined;\n      readonly status: \"regular\" | \"archived\";\n      readonly title?: string | undefined;\n    }\n  | {\n      readonly threadId: string;\n      readonly initializeTask: Promise<RemoteThreadInitializeResponse>;\n      readonly remoteId: string;\n      readonly externalId: string | undefined;\n      readonly status: \"regular\" | \"archived\";\n      readonly title?: string | undefined;\n    };\n\ntype THREAD_MAPPING_ID = string & { __brand: \"THREAD_MAPPING_ID\" };\nfunction createThreadMappingId(id: string): THREAD_MAPPING_ID {\n  return id as THREAD_MAPPING_ID;\n}\n\ntype RemoteThreadState = {\n  readonly isLoading: boolean;\n  readonly newThreadId: string | undefined;\n  readonly threadIds: readonly string[];\n  readonly archivedThreadIds: readonly string[];\n  readonly threadIdMap: Readonly<Record<string, THREAD_MAPPING_ID>>;\n  readonly threadData: Readonly<Record<THREAD_MAPPING_ID, RemoteThreadData>>;\n};\n\nconst getThreadData = (\n  state: RemoteThreadState,\n  threadIdOrRemoteId: string,\n) => {\n  const idx = state.threadIdMap[threadIdOrRemoteId];\n  if (idx === undefined) return undefined;\n  return state.threadData[idx];\n};\n\nconst updateStatusReducer = (\n  state: RemoteThreadState,\n  threadIdOrRemoteId: string,\n  newStatus: \"regular\" | \"archived\" | \"deleted\",\n) => {\n  const data = getThreadData(state, threadIdOrRemoteId);\n  if (!data) return state;\n\n  const { threadId, remoteId, status: lastStatus } = data;\n  if (lastStatus === newStatus) return state;\n\n  const newState = { ...state };\n\n  // lastStatus\n  switch (lastStatus) {\n    case \"new\":\n      newState.newThreadId = undefined;\n      break;\n    case \"regular\":\n      newState.threadIds = newState.threadIds.filter((t) => t !== threadId);\n      break;\n    case \"archived\":\n      newState.archivedThreadIds = newState.archivedThreadIds.filter(\n        (t) => t !== threadId,\n      );\n      break;\n\n    default: {\n      const _exhaustiveCheck: never = lastStatus;\n      throw new Error(`Unsupported state: ${_exhaustiveCheck}`);\n    }\n  }\n\n  // newStatus\n  switch (newStatus) {\n    case \"regular\":\n      newState.threadIds = [threadId, ...newState.threadIds];\n      break;\n\n    case \"archived\":\n      newState.archivedThreadIds = [threadId, ...newState.archivedThreadIds];\n      break;\n\n    case \"deleted\":\n      newState.threadData = Object.fromEntries(\n        Object.entries(newState.threadData).filter(([key]) => key !== threadId),\n      );\n      newState.threadIdMap = Object.fromEntries(\n        Object.entries(newState.threadIdMap).filter(\n          ([key]) => key !== threadId && key !== remoteId,\n        ),\n      );\n      break;\n\n    default: {\n      const _exhaustiveCheck: never = newStatus;\n      throw new Error(`Unsupported state: ${_exhaustiveCheck}`);\n    }\n  }\n\n  if (newStatus !== \"deleted\") {\n    newState.threadData = {\n      ...newState.threadData,\n      [threadId]: {\n        ...data,\n        status: newStatus,\n      },\n    };\n  }\n\n  return newState;\n};\n\nexport class RemoteThreadListThreadListRuntimeCore\n  extends BaseSubscribable\n  implements ThreadListRuntimeCore\n{\n  private _adapter!: RemoteThreadListAdapter;\n  private readonly _hookManager: RemoteThreadListHookInstanceManager;\n\n  private _loadThreadsPromise: Promise<void> | undefined;\n\n  private _mainThreadId!: string;\n  private readonly _state = new OptimisticState<RemoteThreadState>({\n    isLoading: false,\n    newThreadId: undefined,\n    threadIds: [],\n    archivedThreadIds: [],\n    threadIdMap: {},\n    threadData: {},\n  });\n\n  public getLoadThreadsPromise() {\n    // TODO this needs to be cached in case this promise is loaded during suspense\n    if (!this._loadThreadsPromise) {\n      this._loadThreadsPromise = this._state\n        .optimisticUpdate({\n          execute: () => this._adapter.list(),\n          loading: (state) => {\n            return {\n              ...state,\n              isLoading: true,\n            };\n          },\n          then: (state, l) => {\n            const newThreadIds = [];\n            const newArchivedThreadIds = [];\n            const newThreadIdMap = {} as Record<string, THREAD_MAPPING_ID>;\n            const newThreadData = {} as Record<\n              THREAD_MAPPING_ID,\n              RemoteThreadData\n            >;\n\n            for (const thread of l.threads) {\n              switch (thread.status) {\n                case \"regular\":\n                  newThreadIds.push(thread.remoteId);\n                  break;\n                case \"archived\":\n                  newArchivedThreadIds.push(thread.remoteId);\n                  break;\n                default: {\n                  const _exhaustiveCheck: never = thread.status;\n                  throw new Error(`Unsupported state: ${_exhaustiveCheck}`);\n                }\n              }\n\n              const mappingId = createThreadMappingId(thread.remoteId);\n              newThreadIdMap[thread.remoteId] = mappingId;\n              newThreadData[mappingId] = {\n                threadId: thread.remoteId,\n                remoteId: thread.remoteId,\n                externalId: thread.externalId,\n                status: thread.status,\n                title: thread.title,\n                initializeTask: Promise.resolve({\n                  remoteId: thread.remoteId,\n                  externalId: thread.externalId,\n                }),\n              };\n            }\n\n            return {\n              ...state,\n              threadIds: newThreadIds,\n              archivedThreadIds: newArchivedThreadIds,\n              threadIdMap: {\n                ...state.threadIdMap,\n                ...newThreadIdMap,\n              },\n              threadData: {\n                ...state.threadData,\n                ...newThreadData,\n              },\n            };\n          },\n        })\n        .then(() => {});\n    }\n\n    return this._loadThreadsPromise;\n  }\n\n  constructor(adapter: RemoteThreadListAdapter) {\n    super();\n\n    this._state.subscribe(() => this._notifySubscribers());\n    this._hookManager = new RemoteThreadListHookInstanceManager(\n      adapter.runtimeHook,\n    );\n    this.useProvider = create(() => ({\n      Provider: adapter.unstable_Provider ?? Fragment,\n    }));\n    this.__internal_setAdapter(adapter);\n\n    this.switchToNewThread();\n  }\n\n  private useProvider;\n\n  public __internal_setAdapter(adapter: RemoteThreadListAdapter) {\n    if (this._adapter === adapter) return;\n\n    this._adapter = adapter;\n\n    const Provider = adapter.unstable_Provider ?? Fragment;\n    if (Provider !== this.useProvider.getState().Provider) {\n      this.useProvider.setState({ Provider }, true);\n    }\n\n    this._hookManager.setRuntimeHook(adapter.runtimeHook);\n  }\n\n  public __internal_bindAdapter() {\n    this.getLoadThreadsPromise(); // begin loading on initial bind\n    return this._adapter.subscribe?.({\n      onGenerateTitle: this._onGenerateTitle,\n    });\n  }\n\n  public get threadIds() {\n    return this._state.value.threadIds;\n  }\n\n  public get archivedThreadIds() {\n    return this._state.value.archivedThreadIds;\n  }\n\n  public get newThreadId() {\n    return this._state.value.newThreadId;\n  }\n\n  public get mainThreadId(): string {\n    return this._mainThreadId;\n  }\n\n  public getMainThreadRuntimeCore() {\n    const result = this._hookManager.getThreadRuntimeCore(this._mainThreadId);\n    if (!result) return EMPTY_THREAD_CORE;\n    return result;\n  }\n\n  public getThreadRuntimeCore(threadIdOrRemoteId: string) {\n    const data = this.getItemById(threadIdOrRemoteId);\n    if (!data) throw new Error(\"Thread not found\");\n\n    const result = this._hookManager.getThreadRuntimeCore(data.threadId);\n    if (!result) throw new Error(\"Thread not found\");\n    return result;\n  }\n\n  public getItemById(threadIdOrRemoteId: string) {\n    return getThreadData(this._state.value, threadIdOrRemoteId);\n  }\n\n  public async switchToThread(threadIdOrRemoteId: string): Promise<void> {\n    const data = this.getItemById(threadIdOrRemoteId);\n    if (!data) throw new Error(\"Thread not found\");\n\n    if (this._mainThreadId === data.threadId) return;\n\n    const task = this._hookManager.startThreadRuntime(data.threadId);\n    if (this.mainThreadId !== undefined) {\n      await task;\n    } else {\n      task.then(() => this._notifySubscribers());\n    }\n\n    if (data.status === \"archived\") await this.unarchive(data.threadId);\n    this._mainThreadId = data.threadId;\n\n    this._notifySubscribers();\n  }\n\n  public async switchToNewThread(): Promise<void> {\n    // an initialization transaction is in progress, wait for it to settle\n    while (\n      this._state.baseValue.newThreadId !== undefined &&\n      this._state.value.newThreadId === undefined\n    ) {\n      await this._state.waitForUpdate();\n    }\n\n    const state = this._state.value;\n    let threadId: string | undefined = this._state.value.newThreadId;\n    if (threadId === undefined) {\n      do {\n        threadId = `__LOCALID_${generateId()}`;\n      } while (state.threadIdMap[threadId]);\n\n      const mappingId = createThreadMappingId(threadId);\n      this._state.update({\n        ...state,\n        newThreadId: threadId,\n        threadIdMap: {\n          ...state.threadIdMap,\n          [threadId]: mappingId,\n        },\n        threadData: {\n          ...state.threadData,\n          [threadId]: {\n            status: \"new\",\n            threadId,\n          },\n        },\n      });\n    }\n\n    return this.switchToThread(threadId);\n  }\n\n  public initialize = async (threadId: string) => {\n    if (this._state.value.newThreadId !== threadId) {\n      const data = this.getItemById(threadId);\n      if (!data) throw new Error(\"Thread not found\");\n      if (data.status === \"new\") throw new Error(\"Unexpected new state\");\n      return data.initializeTask;\n    }\n\n    return this._state.optimisticUpdate({\n      execute: () => {\n        return this._adapter.initialize(threadId);\n      },\n      optimistic: (state) => {\n        return updateStatusReducer(state, threadId, \"regular\");\n      },\n      loading: (state, task) => {\n        const mappingId = createThreadMappingId(threadId);\n        return {\n          ...state,\n          threadData: {\n            ...state.threadData,\n            [mappingId]: {\n              ...state.threadData[mappingId],\n              initializeTask: task,\n            },\n          },\n        };\n      },\n      then: (state, { remoteId, externalId }) => {\n        const data = getThreadData(state, threadId);\n        if (!data) return state;\n\n        const mappingId = createThreadMappingId(threadId);\n        return {\n          ...state,\n          threadIdMap: {\n            ...state.threadIdMap,\n            [remoteId]: mappingId,\n          },\n          threadData: {\n            ...state.threadData,\n            [mappingId]: {\n              ...data,\n              initializeTask: Promise.resolve({ remoteId, externalId }),\n              remoteId,\n              externalId,\n            },\n          },\n        };\n      },\n    });\n  };\n\n  private _onGenerateTitle = async (\n    remoteId: string,\n    begin: () => Promise<AssistantStream>,\n  ) => {\n    const data = this.getItemById(remoteId);\n    if (!data) throw new Error(\"Thread not found\");\n    if (data.status === \"new\") throw new Error(\"Thread is not yet initialized\");\n\n    const stream = await begin();\n    const messageStream = AssistantMessageStream.fromAssistantStream(stream);\n    for await (const result of messageStream) {\n      const newTitle =\n        result.content.filter((c) => c.type === \"text\")[0]?.text ??\n        \"New Thread\";\n      const state = this._state.baseValue;\n      this._state.update({\n        ...state,\n        threadData: {\n          ...state.threadData,\n          [data.threadId]: {\n            ...data,\n            title: newTitle,\n          },\n        },\n      });\n    }\n  };\n\n  public rename(threadIdOrRemoteId: string, newTitle: string): Promise<void> {\n    const data = this.getItemById(threadIdOrRemoteId);\n    if (!data) throw new Error(\"Thread not found\");\n    if (data.status === \"new\") throw new Error(\"Thread is not yet initialized\");\n\n    return this._state.optimisticUpdate({\n      execute: async () => {\n        const { remoteId } = await data.initializeTask;\n        return this._adapter.rename(remoteId, newTitle);\n      },\n      optimistic: (state) => {\n        const data = getThreadData(state, threadIdOrRemoteId);\n        if (!data) return state;\n\n        return {\n          ...state,\n          threadData: {\n            ...state.threadData,\n            [data.threadId]: {\n              ...data,\n              title: newTitle,\n            },\n          },\n        };\n      },\n    });\n  }\n\n  private async _ensureThreadIsNotMain(threadId: string) {\n    // if thread is main thread, switch to another thread\n    if (threadId === this._mainThreadId) {\n      const lastThreadId = this._state.value.threadIds[0];\n      if (lastThreadId) {\n        await this.switchToThread(lastThreadId);\n      } else {\n        await this.switchToNewThread();\n      }\n    }\n  }\n\n  public async archive(threadIdOrRemoteId: string) {\n    const data = this.getItemById(threadIdOrRemoteId);\n    if (!data) throw new Error(\"Thread not found\");\n    if (data.status !== \"regular\")\n      throw new Error(\"Thread is not yet initialized or already archived\");\n\n    return this._state.optimisticUpdate({\n      execute: async () => {\n        await this._ensureThreadIsNotMain(data.threadId);\n        const { remoteId } = await data.initializeTask;\n        return this._adapter.archive(remoteId);\n      },\n      optimistic: (state) => {\n        return updateStatusReducer(state, data.threadId, \"archived\");\n      },\n    });\n  }\n\n  public unarchive(threadIdOrRemoteId: string): Promise<void> {\n    const data = this.getItemById(threadIdOrRemoteId);\n    if (!data) throw new Error(\"Thread not found\");\n    if (data.status !== \"archived\") throw new Error(\"Thread is not archived\");\n\n    return this._state.optimisticUpdate({\n      execute: async () => {\n        try {\n          const { remoteId } = await data.initializeTask;\n          return await this._adapter.unarchive(remoteId);\n        } catch (error) {\n          await this._ensureThreadIsNotMain(data.threadId);\n          throw error;\n        }\n      },\n      optimistic: (state) => {\n        return updateStatusReducer(state, data.threadId, \"regular\");\n      },\n    });\n  }\n\n  public async delete(threadIdOrRemoteId: string) {\n    const data = this.getItemById(threadIdOrRemoteId);\n    if (!data) throw new Error(\"Thread not found\");\n    if (data.status !== \"regular\" && data.status !== \"archived\")\n      throw new Error(\"Thread is not yet initialized\");\n\n    return this._state.optimisticUpdate({\n      execute: async () => {\n        await this._ensureThreadIsNotMain(data.threadId);\n        const { remoteId } = await data.initializeTask;\n        return await this._adapter.delete(remoteId);\n      },\n      optimistic: (state) => {\n        return updateStatusReducer(state, data.threadId, \"deleted\");\n      },\n    });\n  }\n\n  private useBoundIds = create<string[]>(() => []);\n\n  public __internal_RenderComponent: FC = () => {\n    const id = useId();\n    useEffect(() => {\n      this.useBoundIds.setState((s) => [...s, id], true);\n      return () => {\n        this.useBoundIds.setState((s) => s.filter((i) => i !== id), true);\n      };\n    }, []);\n\n    const boundIds = this.useBoundIds();\n    const { Provider } = this.useProvider();\n\n    return (\n      (boundIds.length === 0 || boundIds[0] === id) && (\n        // only render if the component is the first one mounted\n        <this._hookManager.__internal_RenderThreadRuntimes\n          provider={Provider}\n        />\n      )\n    );\n  };\n}\n"],"mappings":";;;AAGA,SAAS,kBAAkB;AAK3B,SAAS,2CAA2C;AACpD,SAAS,wBAAwB;AACjC,SAAS,yBAAyB;AAClC,SAAS,uBAAuB;AAChC,SAAa,UAAU,WAAW,aAAa;AAC/C,SAAS,cAAc;AACvB,SAA0B,8BAA8B;AAwhBhD;AA5fR,SAAS,sBAAsB,IAA+B;AAC5D,SAAO;AACT;AAWA,IAAM,gBAAgB,CACpB,OACA,uBACG;AACH,QAAM,MAAM,MAAM,YAAY,kBAAkB;AAChD,MAAI,QAAQ,OAAW,QAAO;AAC9B,SAAO,MAAM,WAAW,GAAG;AAC7B;AAEA,IAAM,sBAAsB,CAC1B,OACA,oBACA,cACG;AACH,QAAM,OAAO,cAAc,OAAO,kBAAkB;AACpD,MAAI,CAAC,KAAM,QAAO;AAElB,QAAM,EAAE,UAAU,UAAU,QAAQ,WAAW,IAAI;AACnD,MAAI,eAAe,UAAW,QAAO;AAErC,QAAM,WAAW,EAAE,GAAG,MAAM;AAG5B,UAAQ,YAAY;AAAA,IAClB,KAAK;AACH,eAAS,cAAc;AACvB;AAAA,IACF,KAAK;AACH,eAAS,YAAY,SAAS,UAAU,OAAO,CAAC,MAAM,MAAM,QAAQ;AACpE;AAAA,IACF,KAAK;AACH,eAAS,oBAAoB,SAAS,kBAAkB;AAAA,QACtD,CAAC,MAAM,MAAM;AAAA,MACf;AACA;AAAA,IAEF,SAAS;AACP,YAAM,mBAA0B;AAChC,YAAM,IAAI,MAAM,sBAAsB,gBAAgB,EAAE;AAAA,IAC1D;AAAA,EACF;AAGA,UAAQ,WAAW;AAAA,IACjB,KAAK;AACH,eAAS,YAAY,CAAC,UAAU,GAAG,SAAS,SAAS;AACrD;AAAA,IAEF,KAAK;AACH,eAAS,oBAAoB,CAAC,UAAU,GAAG,SAAS,iBAAiB;AACrE;AAAA,IAEF,KAAK;AACH,eAAS,aAAa,OAAO;AAAA,QAC3B,OAAO,QAAQ,SAAS,UAAU,EAAE,OAAO,CAAC,CAAC,GAAG,MAAM,QAAQ,QAAQ;AAAA,MACxE;AACA,eAAS,cAAc,OAAO;AAAA,QAC5B,OAAO,QAAQ,SAAS,WAAW,EAAE;AAAA,UACnC,CAAC,CAAC,GAAG,MAAM,QAAQ,YAAY,QAAQ;AAAA,QACzC;AAAA,MACF;AACA;AAAA,IAEF,SAAS;AACP,YAAM,mBAA0B;AAChC,YAAM,IAAI,MAAM,sBAAsB,gBAAgB,EAAE;AAAA,IAC1D;AAAA,EACF;AAEA,MAAI,cAAc,WAAW;AAC3B,aAAS,aAAa;AAAA,MACpB,GAAG,SAAS;AAAA,MACZ,CAAC,QAAQ,GAAG;AAAA,QACV,GAAG;AAAA,QACH,QAAQ;AAAA,MACV;AAAA,IACF;AAAA,EACF;AAEA,SAAO;AACT;AAEO,IAAM,wCAAN,cACG,iBAEV;AAAA,EACU;AAAA,EACS;AAAA,EAET;AAAA,EAEA;AAAA,EACS,SAAS,IAAI,gBAAmC;AAAA,IAC/D,WAAW;AAAA,IACX,aAAa;AAAA,IACb,WAAW,CAAC;AAAA,IACZ,mBAAmB,CAAC;AAAA,IACpB,aAAa,CAAC;AAAA,IACd,YAAY,CAAC;AAAA,EACf,CAAC;AAAA,EAEM,wBAAwB;AAE7B,QAAI,CAAC,KAAK,qBAAqB;AAC7B,WAAK,sBAAsB,KAAK,OAC7B,iBAAiB;AAAA,QAChB,SAAS,MAAM,KAAK,SAAS,KAAK;AAAA,QAClC,SAAS,CAAC,UAAU;AAClB,iBAAO;AAAA,YACL,GAAG;AAAA,YACH,WAAW;AAAA,UACb;AAAA,QACF;AAAA,QACA,MAAM,CAAC,OAAO,MAAM;AAClB,gBAAM,eAAe,CAAC;AACtB,gBAAM,uBAAuB,CAAC;AAC9B,gBAAM,iBAAiB,CAAC;AACxB,gBAAM,gBAAgB,CAAC;AAKvB,qBAAW,UAAU,EAAE,SAAS;AAC9B,oBAAQ,OAAO,QAAQ;AAAA,cACrB,KAAK;AACH,6BAAa,KAAK,OAAO,QAAQ;AACjC;AAAA,cACF,KAAK;AACH,qCAAqB,KAAK,OAAO,QAAQ;AACzC;AAAA,cACF,SAAS;AACP,sBAAM,mBAA0B,OAAO;AACvC,sBAAM,IAAI,MAAM,sBAAsB,gBAAgB,EAAE;AAAA,cAC1D;AAAA,YACF;AAEA,kBAAM,YAAY,sBAAsB,OAAO,QAAQ;AACvD,2BAAe,OAAO,QAAQ,IAAI;AAClC,0BAAc,SAAS,IAAI;AAAA,cACzB,UAAU,OAAO;AAAA,cACjB,UAAU,OAAO;AAAA,cACjB,YAAY,OAAO;AAAA,cACnB,QAAQ,OAAO;AAAA,cACf,OAAO,OAAO;AAAA,cACd,gBAAgB,QAAQ,QAAQ;AAAA,gBAC9B,UAAU,OAAO;AAAA,gBACjB,YAAY,OAAO;AAAA,cACrB,CAAC;AAAA,YACH;AAAA,UACF;AAEA,iBAAO;AAAA,YACL,GAAG;AAAA,YACH,WAAW;AAAA,YACX,mBAAmB;AAAA,YACnB,aAAa;AAAA,cACX,GAAG,MAAM;AAAA,cACT,GAAG;AAAA,YACL;AAAA,YACA,YAAY;AAAA,cACV,GAAG,MAAM;AAAA,cACT,GAAG;AAAA,YACL;AAAA,UACF;AAAA,QACF;AAAA,MACF,CAAC,EACA,KAAK,MAAM;AAAA,MAAC,CAAC;AAAA,IAClB;AAEA,WAAO,KAAK;AAAA,EACd;AAAA,EAEA,YAAY,SAAkC;AAC5C,UAAM;AAEN,SAAK,OAAO,UAAU,MAAM,KAAK,mBAAmB,CAAC;AACrD,SAAK,eAAe,IAAI;AAAA,MACtB,QAAQ;AAAA,IACV;AACA,SAAK,cAAc,OAAO,OAAO;AAAA,MAC/B,UAAU,QAAQ,qBAAqB;AAAA,IACzC,EAAE;AACF,SAAK,sBAAsB,OAAO;AAElC,SAAK,kBAAkB;AAAA,EACzB;AAAA,EAEQ;AAAA,EAED,sBAAsB,SAAkC;AAC7D,QAAI,KAAK,aAAa,QAAS;AAE/B,SAAK,WAAW;AAEhB,UAAM,WAAW,QAAQ,qBAAqB;AAC9C,QAAI,aAAa,KAAK,YAAY,SAAS,EAAE,UAAU;AACrD,WAAK,YAAY,SAAS,EAAE,SAAS,GAAG,IAAI;AAAA,IAC9C;AAEA,SAAK,aAAa,eAAe,QAAQ,WAAW;AAAA,EACtD;AAAA,EAEO,yBAAyB;AAC9B,SAAK,sBAAsB;AAC3B,WAAO,KAAK,SAAS,YAAY;AAAA,MAC/B,iBAAiB,KAAK;AAAA,IACxB,CAAC;AAAA,EACH;AAAA,EAEA,IAAW,YAAY;AACrB,WAAO,KAAK,OAAO,MAAM;AAAA,EAC3B;AAAA,EAEA,IAAW,oBAAoB;AAC7B,WAAO,KAAK,OAAO,MAAM;AAAA,EAC3B;AAAA,EAEA,IAAW,cAAc;AACvB,WAAO,KAAK,OAAO,MAAM;AAAA,EAC3B;AAAA,EAEA,IAAW,eAAuB;AAChC,WAAO,KAAK;AAAA,EACd;AAAA,EAEO,2BAA2B;AAChC,UAAM,SAAS,KAAK,aAAa,qBAAqB,KAAK,aAAa;AACxE,QAAI,CAAC,OAAQ,QAAO;AACpB,WAAO;AAAA,EACT;AAAA,EAEO,qBAAqB,oBAA4B;AACtD,UAAM,OAAO,KAAK,YAAY,kBAAkB;AAChD,QAAI,CAAC,KAAM,OAAM,IAAI,MAAM,kBAAkB;AAE7C,UAAM,SAAS,KAAK,aAAa,qBAAqB,KAAK,QAAQ;AACnE,QAAI,CAAC,OAAQ,OAAM,IAAI,MAAM,kBAAkB;AAC/C,WAAO;AAAA,EACT;AAAA,EAEO,YAAY,oBAA4B;AAC7C,WAAO,cAAc,KAAK,OAAO,OAAO,kBAAkB;AAAA,EAC5D;AAAA,EAEA,MAAa,eAAe,oBAA2C;AACrE,UAAM,OAAO,KAAK,YAAY,kBAAkB;AAChD,QAAI,CAAC,KAAM,OAAM,IAAI,MAAM,kBAAkB;AAE7C,QAAI,KAAK,kBAAkB,KAAK,SAAU;AAE1C,UAAM,OAAO,KAAK,aAAa,mBAAmB,KAAK,QAAQ;AAC/D,QAAI,KAAK,iBAAiB,QAAW;AACnC,YAAM;AAAA,IACR,OAAO;AACL,WAAK,KAAK,MAAM,KAAK,mBAAmB,CAAC;AAAA,IAC3C;AAEA,QAAI,KAAK,WAAW,WAAY,OAAM,KAAK,UAAU,KAAK,QAAQ;AAClE,SAAK,gBAAgB,KAAK;AAE1B,SAAK,mBAAmB;AAAA,EAC1B;AAAA,EAEA,MAAa,oBAAmC;AAE9C,WACE,KAAK,OAAO,UAAU,gBAAgB,UACtC,KAAK,OAAO,MAAM,gBAAgB,QAClC;AACA,YAAM,KAAK,OAAO,cAAc;AAAA,IAClC;AAEA,UAAM,QAAQ,KAAK,OAAO;AAC1B,QAAI,WAA+B,KAAK,OAAO,MAAM;AACrD,QAAI,aAAa,QAAW;AAC1B,SAAG;AACD,mBAAW,aAAa,WAAW,CAAC;AAAA,MACtC,SAAS,MAAM,YAAY,QAAQ;AAEnC,YAAM,YAAY,sBAAsB,QAAQ;AAChD,WAAK,OAAO,OAAO;AAAA,QACjB,GAAG;AAAA,QACH,aAAa;AAAA,QACb,aAAa;AAAA,UACX,GAAG,MAAM;AAAA,UACT,CAAC,QAAQ,GAAG;AAAA,QACd;AAAA,QACA,YAAY;AAAA,UACV,GAAG,MAAM;AAAA,UACT,CAAC,QAAQ,GAAG;AAAA,YACV,QAAQ;AAAA,YACR;AAAA,UACF;AAAA,QACF;AAAA,MACF,CAAC;AAAA,IACH;AAEA,WAAO,KAAK,eAAe,QAAQ;AAAA,EACrC;AAAA,EAEO,aAAa,OAAO,aAAqB;AAC9C,QAAI,KAAK,OAAO,MAAM,gBAAgB,UAAU;AAC9C,YAAM,OAAO,KAAK,YAAY,QAAQ;AACtC,UAAI,CAAC,KAAM,OAAM,IAAI,MAAM,kBAAkB;AAC7C,UAAI,KAAK,WAAW,MAAO,OAAM,IAAI,MAAM,sBAAsB;AACjE,aAAO,KAAK;AAAA,IACd;AAEA,WAAO,KAAK,OAAO,iBAAiB;AAAA,MAClC,SAAS,MAAM;AACb,eAAO,KAAK,SAAS,WAAW,QAAQ;AAAA,MAC1C;AAAA,MACA,YAAY,CAAC,UAAU;AACrB,eAAO,oBAAoB,OAAO,UAAU,SAAS;AAAA,MACvD;AAAA,MACA,SAAS,CAAC,OAAO,SAAS;AACxB,cAAM,YAAY,sBAAsB,QAAQ;AAChD,eAAO;AAAA,UACL,GAAG;AAAA,UACH,YAAY;AAAA,YACV,GAAG,MAAM;AAAA,YACT,CAAC,SAAS,GAAG;AAAA,cACX,GAAG,MAAM,WAAW,SAAS;AAAA,cAC7B,gBAAgB;AAAA,YAClB;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,MACA,MAAM,CAAC,OAAO,EAAE,UAAU,WAAW,MAAM;AACzC,cAAM,OAAO,cAAc,OAAO,QAAQ;AAC1C,YAAI,CAAC,KAAM,QAAO;AAElB,cAAM,YAAY,sBAAsB,QAAQ;AAChD,eAAO;AAAA,UACL,GAAG;AAAA,UACH,aAAa;AAAA,YACX,GAAG,MAAM;AAAA,YACT,CAAC,QAAQ,GAAG;AAAA,UACd;AAAA,UACA,YAAY;AAAA,YACV,GAAG,MAAM;AAAA,YACT,CAAC,SAAS,GAAG;AAAA,cACX,GAAG;AAAA,cACH,gBAAgB,QAAQ,QAAQ,EAAE,UAAU,WAAW,CAAC;AAAA,cACxD;AAAA,cACA;AAAA,YACF;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEQ,mBAAmB,OACzB,UACA,UACG;AACH,UAAM,OAAO,KAAK,YAAY,QAAQ;AACtC,QAAI,CAAC,KAAM,OAAM,IAAI,MAAM,kBAAkB;AAC7C,QAAI,KAAK,WAAW,MAAO,OAAM,IAAI,MAAM,+BAA+B;AAE1E,UAAM,SAAS,MAAM,MAAM;AAC3B,UAAM,gBAAgB,uBAAuB,oBAAoB,MAAM;AACvE,qBAAiB,UAAU,eAAe;AACxC,YAAM,WACJ,OAAO,QAAQ,OAAO,CAAC,MAAM,EAAE,SAAS,MAAM,EAAE,CAAC,GAAG,QACpD;AACF,YAAM,QAAQ,KAAK,OAAO;AAC1B,WAAK,OAAO,OAAO;AAAA,QACjB,GAAG;AAAA,QACH,YAAY;AAAA,UACV,GAAG,MAAM;AAAA,UACT,CAAC,KAAK,QAAQ,GAAG;AAAA,YACf,GAAG;AAAA,YACH,OAAO;AAAA,UACT;AAAA,QACF;AAAA,MACF,CAAC;AAAA,IACH;AAAA,EACF;AAAA,EAEO,OAAO,oBAA4B,UAAiC;AACzE,UAAM,OAAO,KAAK,YAAY,kBAAkB;AAChD,QAAI,CAAC,KAAM,OAAM,IAAI,MAAM,kBAAkB;AAC7C,QAAI,KAAK,WAAW,MAAO,OAAM,IAAI,MAAM,+BAA+B;AAE1E,WAAO,KAAK,OAAO,iBAAiB;AAAA,MAClC,SAAS,YAAY;AACnB,cAAM,EAAE,SAAS,IAAI,MAAM,KAAK;AAChC,eAAO,KAAK,SAAS,OAAO,UAAU,QAAQ;AAAA,MAChD;AAAA,MACA,YAAY,CAAC,UAAU;AACrB,cAAMA,QAAO,cAAc,OAAO,kBAAkB;AACpD,YAAI,CAACA,MAAM,QAAO;AAElB,eAAO;AAAA,UACL,GAAG;AAAA,UACH,YAAY;AAAA,YACV,GAAG,MAAM;AAAA,YACT,CAACA,MAAK,QAAQ,GAAG;AAAA,cACf,GAAGA;AAAA,cACH,OAAO;AAAA,YACT;AAAA,UACF;AAAA,QACF;AAAA,MACF;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEA,MAAc,uBAAuB,UAAkB;AAErD,QAAI,aAAa,KAAK,eAAe;AACnC,YAAM,eAAe,KAAK,OAAO,MAAM,UAAU,CAAC;AAClD,UAAI,cAAc;AAChB,cAAM,KAAK,eAAe,YAAY;AAAA,MACxC,OAAO;AACL,cAAM,KAAK,kBAAkB;AAAA,MAC/B;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAa,QAAQ,oBAA4B;AAC/C,UAAM,OAAO,KAAK,YAAY,kBAAkB;AAChD,QAAI,CAAC,KAAM,OAAM,IAAI,MAAM,kBAAkB;AAC7C,QAAI,KAAK,WAAW;AAClB,YAAM,IAAI,MAAM,mDAAmD;AAErE,WAAO,KAAK,OAAO,iBAAiB;AAAA,MAClC,SAAS,YAAY;AACnB,cAAM,KAAK,uBAAuB,KAAK,QAAQ;AAC/C,cAAM,EAAE,SAAS,IAAI,MAAM,KAAK;AAChC,eAAO,KAAK,SAAS,QAAQ,QAAQ;AAAA,MACvC;AAAA,MACA,YAAY,CAAC,UAAU;AACrB,eAAO,oBAAoB,OAAO,KAAK,UAAU,UAAU;AAAA,MAC7D;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEO,UAAU,oBAA2C;AAC1D,UAAM,OAAO,KAAK,YAAY,kBAAkB;AAChD,QAAI,CAAC,KAAM,OAAM,IAAI,MAAM,kBAAkB;AAC7C,QAAI,KAAK,WAAW,WAAY,OAAM,IAAI,MAAM,wBAAwB;AAExE,WAAO,KAAK,OAAO,iBAAiB;AAAA,MAClC,SAAS,YAAY;AACnB,YAAI;AACF,gBAAM,EAAE,SAAS,IAAI,MAAM,KAAK;AAChC,iBAAO,MAAM,KAAK,SAAS,UAAU,QAAQ;AAAA,QAC/C,SAAS,OAAO;AACd,gBAAM,KAAK,uBAAuB,KAAK,QAAQ;AAC/C,gBAAM;AAAA,QACR;AAAA,MACF;AAAA,MACA,YAAY,CAAC,UAAU;AACrB,eAAO,oBAAoB,OAAO,KAAK,UAAU,SAAS;AAAA,MAC5D;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEA,MAAa,OAAO,oBAA4B;AAC9C,UAAM,OAAO,KAAK,YAAY,kBAAkB;AAChD,QAAI,CAAC,KAAM,OAAM,IAAI,MAAM,kBAAkB;AAC7C,QAAI,KAAK,WAAW,aAAa,KAAK,WAAW;AAC/C,YAAM,IAAI,MAAM,+BAA+B;AAEjD,WAAO,KAAK,OAAO,iBAAiB;AAAA,MAClC,SAAS,YAAY;AACnB,cAAM,KAAK,uBAAuB,KAAK,QAAQ;AAC/C,cAAM,EAAE,SAAS,IAAI,MAAM,KAAK;AAChC,eAAO,MAAM,KAAK,SAAS,OAAO,QAAQ;AAAA,MAC5C;AAAA,MACA,YAAY,CAAC,UAAU;AACrB,eAAO,oBAAoB,OAAO,KAAK,UAAU,SAAS;AAAA,MAC5D;AAAA,IACF,CAAC;AAAA,EACH;AAAA,EAEQ,cAAc,OAAiB,MAAM,CAAC,CAAC;AAAA,EAExC,6BAAiC,MAAM;AAC5C,UAAM,KAAK,MAAM;AACjB,cAAU,MAAM;AACd,WAAK,YAAY,SAAS,CAAC,MAAM,CAAC,GAAG,GAAG,EAAE,GAAG,IAAI;AACjD,aAAO,MAAM;AACX,aAAK,YAAY,SAAS,CAAC,MAAM,EAAE,OAAO,CAAC,MAAM,MAAM,EAAE,GAAG,IAAI;AAAA,MAClE;AAAA,IACF,GAAG,CAAC,CAAC;AAEL,UAAM,WAAW,KAAK,YAAY;AAClC,UAAM,EAAE,SAAS,IAAI,KAAK,YAAY;AAEtC,YACG,SAAS,WAAW,KAAK,SAAS,CAAC,MAAM;AAAA,IAExC;AAAA,MAAC,KAAK,aAAa;AAAA,MAAlB;AAAA,QACC,UAAU;AAAA;AAAA,IACZ;AAAA,EAGN;AACF;","names":["data"]}

package/dist/ui/assistant-message.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"assistant-message.d.ts","sourceRoot":"","sources":["../../src/ui/assistant-message.tsx"],"names":[],"mappings":"AAEA,OAAO,EAAiD,KAAK,EAAE,EAAE,MAAM,OAAO,CAAC;AAC/E,OAAO,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AAQjD,QAAA,MAAM,gBAAgB,EAAE,EASvB,CAAC;AAgEF,QAAA,MAAM,OAAO;;;;;;CAIZ,CAAC;wBAKG,OAAO,gBAAgB,GAAG,OAAO,OAAO;AAH7C,wBAG8C"}
+{"version":3,"file":"assistant-message.d.ts","sourceRoot":"","sources":["../../src/ui/assistant-message.tsx"],"names":[],"mappings":"AAEA,OAAO,EAAiD,KAAK,EAAE,EAAE,MAAM,OAAO,CAAC;AAC/E,OAAO,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AAQjD,QAAA,MAAM,gBAAgB,EAAE,EASvB,CAAC;AAkEF,QAAA,MAAM,OAAO;;;;;;CAIZ,CAAC;wBAKG,OAAO,gBAAgB,GAAG,OAAO,OAAO;AAH7C,wBAG8C"}

package/dist/ui/assistant-message.js

@@ -78,17 +78,21 @@ var AssistantMessageContent = (0, import_react.forwardRef)(({ components: compon
     // eslint-disable-next-line react-hooks/exhaustive-deps
     [...tools ?? [], components.ToolFallback]
   );
-  return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(AssistantMessageContentWrapper, { ...rest, ref, children: /* @__PURE__ */ (0, import_jsx_runtime.jsx)(
-    import_primitives.MessagePrimitive.Content,
-    {
-      components: {
-        ...componentsProp,
-        Text: componentsProp?.Text ?? components.Text ?? import_content_part.default.Text,
-        Empty: componentsProp?.Empty ?? components.Empty,
-        tools: toolsComponents
+  const Footer = components.Footer;
+  return /* @__PURE__ */ (0, import_jsx_runtime.jsxs)(AssistantMessageContentWrapper, { ...rest, ref, children: [
+    /* @__PURE__ */ (0, import_jsx_runtime.jsx)(
+      import_primitives.MessagePrimitive.Content,
+      {
+        components: {
+          ...componentsProp,
+          Text: componentsProp?.Text ?? components.Text ?? import_content_part.default.Text,
+          Empty: componentsProp?.Empty ?? components.Empty,
+          tools: toolsComponents
+        }
       }
-    }
-  ) });
+    ),
+    Footer && /* @__PURE__ */ (0, import_jsx_runtime.jsx)(Footer, {})
+  ] });
 });
 AssistantMessageContent.displayName = "AssistantMessageContent";
 var exports2 = {

package/dist/ui/assistant-message.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/ui/assistant-message.tsx"],"sourcesContent":["\"use client\";\n\nimport { ComponentPropsWithoutRef, forwardRef, useMemo, type FC } from \"react\";\nimport { MessagePrimitive } from \"../primitives\";\nimport BranchPicker from \"./branch-picker\";\nimport { Avatar } from \"./base/avatar\";\nimport { withDefaults } from \"./utils/withDefaults\";\nimport { useThreadConfig } from \"./thread-config\";\nimport AssistantActionBar from \"./assistant-action-bar\";\nimport ContentPart from \"./content-part\";\n\nconst AssistantMessage: FC = () => {\n  return (\n    <AssistantMessageRoot>\n      <AssistantMessageAvatar />\n      <AssistantMessageContent />\n      <BranchPicker />\n      <AssistantActionBar />\n    </AssistantMessageRoot>\n  );\n};\n\nAssistantMessage.displayName = \"AssistantMessage\";\n\nconst AssistantMessageAvatar: FC = () => {\n  const { assistantAvatar: avatar = { fallback: \"A\" } } = useThreadConfig();\n  return <Avatar {...avatar} />;\n};\n\nconst AssistantMessageRoot = withDefaults(MessagePrimitive.Root, {\n  className: \"aui-assistant-message-root\",\n});\n\nAssistantMessageRoot.displayName = \"AssistantMessageRoot\";\n\nconst AssistantMessageContentWrapper = withDefaults(\"div\", {\n  className: \"aui-assistant-message-content\",\n});\n\nnamespace AssistantMessageContent {\n  export type Element = HTMLDivElement;\n  export type Props = MessagePrimitive.Content.Props &\n    ComponentPropsWithoutRef<\"div\">;\n}\n\nconst AssistantMessageContent = forwardRef<\n  AssistantMessageContent.Element,\n  AssistantMessageContent.Props\n>(({ components: componentsProp, ...rest }, ref) => {\n  const { tools, assistantMessage: { components = {} } = {} } =\n    useThreadConfig();\n\n  const toolsComponents = useMemo(\n    () => ({\n      by_name: !tools\n        ? undefined\n        : Object.fromEntries(\n            tools.map((t) => [\n              t.unstable_tool.toolName,\n              t.unstable_tool.render,\n            ]),\n          ),\n      Fallback: components.ToolFallback,\n    }),\n    // eslint-disable-next-line react-hooks/exhaustive-deps\n    [...(tools ?? []), components.ToolFallback],\n  );\n\n  return (\n    <AssistantMessageContentWrapper {...rest} ref={ref}>\n      <MessagePrimitive.Content\n        components={{\n          ...componentsProp,\n          Text: componentsProp?.Text ?? components.Text ?? ContentPart.Text,\n          Empty: componentsProp?.Empty ?? components.Empty,\n          tools: toolsComponents,\n        }}\n      />\n    </AssistantMessageContentWrapper>\n  );\n});\n\nAssistantMessageContent.displayName = \"AssistantMessageContent\";\n\nconst exports = {\n  Root: AssistantMessageRoot,\n  Avatar: AssistantMessageAvatar,\n  Content: AssistantMessageContent,\n};\n\nexport default Object.assign(\n  AssistantMessage,\n  exports,\n) as typeof AssistantMessage & typeof exports;\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAEA,mBAAuE;AACvE,wBAAiC;AACjC,2BAAyB;AACzB,oBAAuB;AACvB,0BAA6B;AAC7B,2BAAgC;AAChC,kCAA+B;AAC/B,0BAAwB;AAIpB;AAFJ,IAAM,mBAAuB,MAAM;AACjC,SACE,6CAAC,wBACC;AAAA,gDAAC,0BAAuB;AAAA,IACxB,4CAAC,2BAAwB;AAAA,IACzB,4CAAC,qBAAAA,SAAA,EAAa;AAAA,IACd,4CAAC,4BAAAC,SAAA,EAAmB;AAAA,KACtB;AAEJ;AAEA,iBAAiB,cAAc;AAE/B,IAAM,yBAA6B,MAAM;AACvC,QAAM,EAAE,iBAAiB,SAAS,EAAE,UAAU,IAAI,EAAE,QAAI,sCAAgB;AACxE,SAAO,4CAAC,wBAAQ,GAAG,QAAQ;AAC7B;AAEA,IAAM,2BAAuB,kCAAa,mCAAiB,MAAM;AAAA,EAC/D,WAAW;AACb,CAAC;AAED,qBAAqB,cAAc;AAEnC,IAAM,qCAAiC,kCAAa,OAAO;AAAA,EACzD,WAAW;AACb,CAAC;AAQD,IAAM,8BAA0B,yBAG9B,CAAC,EAAE,YAAY,gBAAgB,GAAG,KAAK,GAAG,QAAQ;AAClD,QAAM,EAAE,OAAO,kBAAkB,EAAE,aAAa,CAAC,EAAE,IAAI,CAAC,EAAE,QACxD,sCAAgB;AAElB,QAAM,sBAAkB;AAAA,IACtB,OAAO;AAAA,MACL,SAAS,CAAC,QACN,SACA,OAAO;AAAA,QACL,MAAM,IAAI,CAAC,MAAM;AAAA,UACf,EAAE,cAAc;AAAA,UAChB,EAAE,cAAc;AAAA,QAClB,CAAC;AAAA,MACH;AAAA,MACJ,UAAU,WAAW;AAAA,IACvB;AAAA;AAAA,IAEA,CAAC,GAAI,SAAS,CAAC,GAAI,WAAW,YAAY;AAAA,EAC5C;AAEA,SACE,4CAAC,kCAAgC,GAAG,MAAM,KACxC;AAAA,IAAC,mCAAiB;AAAA,IAAjB;AAAA,MACC,YAAY;AAAA,QACV,GAAG;AAAA,QACH,MAAM,gBAAgB,QAAQ,WAAW,QAAQ,oBAAAC,QAAY;AAAA,QAC7D,OAAO,gBAAgB,SAAS,WAAW;AAAA,QAC3C,OAAO;AAAA,MACT;AAAA;AAAA,EACF,GACF;AAEJ,CAAC;AAED,wBAAwB,cAAc;AAEtC,IAAMC,WAAU;AAAA,EACd,MAAM;AAAA,EACN,QAAQ;AAAA,EACR,SAAS;AACX;AAEA,IAAO,4BAAQ,OAAO;AAAA,EACpB;AAAA,EACAA;AACF;","names":["BranchPicker","AssistantActionBar","ContentPart","exports"]}
+{"version":3,"sources":["../../src/ui/assistant-message.tsx"],"sourcesContent":["\"use client\";\n\nimport { ComponentPropsWithoutRef, forwardRef, useMemo, type FC } from \"react\";\nimport { MessagePrimitive } from \"../primitives\";\nimport BranchPicker from \"./branch-picker\";\nimport { Avatar } from \"./base/avatar\";\nimport { withDefaults } from \"./utils/withDefaults\";\nimport { useThreadConfig } from \"./thread-config\";\nimport AssistantActionBar from \"./assistant-action-bar\";\nimport ContentPart from \"./content-part\";\n\nconst AssistantMessage: FC = () => {\n  return (\n    <AssistantMessageRoot>\n      <AssistantMessageAvatar />\n      <AssistantMessageContent />\n      <BranchPicker />\n      <AssistantActionBar />\n    </AssistantMessageRoot>\n  );\n};\n\nAssistantMessage.displayName = \"AssistantMessage\";\n\nconst AssistantMessageAvatar: FC = () => {\n  const { assistantAvatar: avatar = { fallback: \"A\" } } = useThreadConfig();\n  return <Avatar {...avatar} />;\n};\n\nconst AssistantMessageRoot = withDefaults(MessagePrimitive.Root, {\n  className: \"aui-assistant-message-root\",\n});\n\nAssistantMessageRoot.displayName = \"AssistantMessageRoot\";\n\nconst AssistantMessageContentWrapper = withDefaults(\"div\", {\n  className: \"aui-assistant-message-content\",\n});\n\nnamespace AssistantMessageContent {\n  export type Element = HTMLDivElement;\n  export type Props = MessagePrimitive.Content.Props &\n    ComponentPropsWithoutRef<\"div\">;\n}\n\nconst AssistantMessageContent = forwardRef<\n  AssistantMessageContent.Element,\n  AssistantMessageContent.Props\n>(({ components: componentsProp, ...rest }, ref) => {\n  const { tools, assistantMessage: { components = {} } = {} } =\n    useThreadConfig();\n\n  const toolsComponents = useMemo(\n    () => ({\n      by_name: !tools\n        ? undefined\n        : Object.fromEntries(\n            tools.map((t) => [\n              t.unstable_tool.toolName,\n              t.unstable_tool.render,\n            ]),\n          ),\n      Fallback: components.ToolFallback,\n    }),\n    // eslint-disable-next-line react-hooks/exhaustive-deps\n    [...(tools ?? []), components.ToolFallback],\n  );\n  const Footer = components.Footer;\n\n  return (\n    <AssistantMessageContentWrapper {...rest} ref={ref}>\n      <MessagePrimitive.Content\n        components={{\n          ...componentsProp,\n          Text: componentsProp?.Text ?? components.Text ?? ContentPart.Text,\n          Empty: componentsProp?.Empty ?? components.Empty,\n          tools: toolsComponents,\n        }}\n      />\n      {Footer && <Footer />}\n    </AssistantMessageContentWrapper>\n  );\n});\n\nAssistantMessageContent.displayName = \"AssistantMessageContent\";\n\nconst exports = {\n  Root: AssistantMessageRoot,\n  Avatar: AssistantMessageAvatar,\n  Content: AssistantMessageContent,\n};\n\nexport default Object.assign(\n  AssistantMessage,\n  exports,\n) as typeof AssistantMessage & typeof exports;\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAEA,mBAAuE;AACvE,wBAAiC;AACjC,2BAAyB;AACzB,oBAAuB;AACvB,0BAA6B;AAC7B,2BAAgC;AAChC,kCAA+B;AAC/B,0BAAwB;AAIpB;AAFJ,IAAM,mBAAuB,MAAM;AACjC,SACE,6CAAC,wBACC;AAAA,gDAAC,0BAAuB;AAAA,IACxB,4CAAC,2BAAwB;AAAA,IACzB,4CAAC,qBAAAA,SAAA,EAAa;AAAA,IACd,4CAAC,4BAAAC,SAAA,EAAmB;AAAA,KACtB;AAEJ;AAEA,iBAAiB,cAAc;AAE/B,IAAM,yBAA6B,MAAM;AACvC,QAAM,EAAE,iBAAiB,SAAS,EAAE,UAAU,IAAI,EAAE,QAAI,sCAAgB;AACxE,SAAO,4CAAC,wBAAQ,GAAG,QAAQ;AAC7B;AAEA,IAAM,2BAAuB,kCAAa,mCAAiB,MAAM;AAAA,EAC/D,WAAW;AACb,CAAC;AAED,qBAAqB,cAAc;AAEnC,IAAM,qCAAiC,kCAAa,OAAO;AAAA,EACzD,WAAW;AACb,CAAC;AAQD,IAAM,8BAA0B,yBAG9B,CAAC,EAAE,YAAY,gBAAgB,GAAG,KAAK,GAAG,QAAQ;AAClD,QAAM,EAAE,OAAO,kBAAkB,EAAE,aAAa,CAAC,EAAE,IAAI,CAAC,EAAE,QACxD,sCAAgB;AAElB,QAAM,sBAAkB;AAAA,IACtB,OAAO;AAAA,MACL,SAAS,CAAC,QACN,SACA,OAAO;AAAA,QACL,MAAM,IAAI,CAAC,MAAM;AAAA,UACf,EAAE,cAAc;AAAA,UAChB,EAAE,cAAc;AAAA,QAClB,CAAC;AAAA,MACH;AAAA,MACJ,UAAU,WAAW;AAAA,IACvB;AAAA;AAAA,IAEA,CAAC,GAAI,SAAS,CAAC,GAAI,WAAW,YAAY;AAAA,EAC5C;AACA,QAAM,SAAS,WAAW;AAE1B,SACE,6CAAC,kCAAgC,GAAG,MAAM,KACxC;AAAA;AAAA,MAAC,mCAAiB;AAAA,MAAjB;AAAA,QACC,YAAY;AAAA,UACV,GAAG;AAAA,UACH,MAAM,gBAAgB,QAAQ,WAAW,QAAQ,oBAAAC,QAAY;AAAA,UAC7D,OAAO,gBAAgB,SAAS,WAAW;AAAA,UAC3C,OAAO;AAAA,QACT;AAAA;AAAA,IACF;AAAA,IACC,UAAU,4CAAC,UAAO;AAAA,KACrB;AAEJ,CAAC;AAED,wBAAwB,cAAc;AAEtC,IAAMC,WAAU;AAAA,EACd,MAAM;AAAA,EACN,QAAQ;AAAA,EACR,SAAS;AACX;AAEA,IAAO,4BAAQ,OAAO;AAAA,EACpB;AAAA,EACAA;AACF;","names":["BranchPicker","AssistantActionBar","ContentPart","exports"]}

package/dist/ui/assistant-message.mjs

@@ -45,17 +45,21 @@ var AssistantMessageContent = forwardRef(({ components: componentsProp, ...rest
     // eslint-disable-next-line react-hooks/exhaustive-deps
     [...tools ?? [], components.ToolFallback]
   );
-  return /* @__PURE__ */ jsx(AssistantMessageContentWrapper, { ...rest, ref, children: /* @__PURE__ */ jsx(
-    MessagePrimitive.Content,
-    {
-      components: {
-        ...componentsProp,
-        Text: componentsProp?.Text ?? components.Text ?? ContentPart.Text,
-        Empty: componentsProp?.Empty ?? components.Empty,
-        tools: toolsComponents
+  const Footer = components.Footer;
+  return /* @__PURE__ */ jsxs(AssistantMessageContentWrapper, { ...rest, ref, children: [
+    /* @__PURE__ */ jsx(
+      MessagePrimitive.Content,
+      {
+        components: {
+          ...componentsProp,
+          Text: componentsProp?.Text ?? components.Text ?? ContentPart.Text,
+          Empty: componentsProp?.Empty ?? components.Empty,
+          tools: toolsComponents
+        }
       }
-    }
-  ) });
+    ),
+    Footer && /* @__PURE__ */ jsx(Footer, {})
+  ] });
 });
 AssistantMessageContent.displayName = "AssistantMessageContent";
 var exports = {

package/dist/ui/assistant-message.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/ui/assistant-message.tsx"],"sourcesContent":["\"use client\";\n\nimport { ComponentPropsWithoutRef, forwardRef, useMemo, type FC } from \"react\";\nimport { MessagePrimitive } from \"../primitives\";\nimport BranchPicker from \"./branch-picker\";\nimport { Avatar } from \"./base/avatar\";\nimport { withDefaults } from \"./utils/withDefaults\";\nimport { useThreadConfig } from \"./thread-config\";\nimport AssistantActionBar from \"./assistant-action-bar\";\nimport ContentPart from \"./content-part\";\n\nconst AssistantMessage: FC = () => {\n  return (\n    <AssistantMessageRoot>\n      <AssistantMessageAvatar />\n      <AssistantMessageContent />\n      <BranchPicker />\n      <AssistantActionBar />\n    </AssistantMessageRoot>\n  );\n};\n\nAssistantMessage.displayName = \"AssistantMessage\";\n\nconst AssistantMessageAvatar: FC = () => {\n  const { assistantAvatar: avatar = { fallback: \"A\" } } = useThreadConfig();\n  return <Avatar {...avatar} />;\n};\n\nconst AssistantMessageRoot = withDefaults(MessagePrimitive.Root, {\n  className: \"aui-assistant-message-root\",\n});\n\nAssistantMessageRoot.displayName = \"AssistantMessageRoot\";\n\nconst AssistantMessageContentWrapper = withDefaults(\"div\", {\n  className: \"aui-assistant-message-content\",\n});\n\nnamespace AssistantMessageContent {\n  export type Element = HTMLDivElement;\n  export type Props = MessagePrimitive.Content.Props &\n    ComponentPropsWithoutRef<\"div\">;\n}\n\nconst AssistantMessageContent = forwardRef<\n  AssistantMessageContent.Element,\n  AssistantMessageContent.Props\n>(({ components: componentsProp, ...rest }, ref) => {\n  const { tools, assistantMessage: { components = {} } = {} } =\n    useThreadConfig();\n\n  const toolsComponents = useMemo(\n    () => ({\n      by_name: !tools\n        ? undefined\n        : Object.fromEntries(\n            tools.map((t) => [\n              t.unstable_tool.toolName,\n              t.unstable_tool.render,\n            ]),\n          ),\n      Fallback: components.ToolFallback,\n    }),\n    // eslint-disable-next-line react-hooks/exhaustive-deps\n    [...(tools ?? []), components.ToolFallback],\n  );\n\n  return (\n    <AssistantMessageContentWrapper {...rest} ref={ref}>\n      <MessagePrimitive.Content\n        components={{\n          ...componentsProp,\n          Text: componentsProp?.Text ?? components.Text ?? ContentPart.Text,\n          Empty: componentsProp?.Empty ?? components.Empty,\n          tools: toolsComponents,\n        }}\n      />\n    </AssistantMessageContentWrapper>\n  );\n});\n\nAssistantMessageContent.displayName = \"AssistantMessageContent\";\n\nconst exports = {\n  Root: AssistantMessageRoot,\n  Avatar: AssistantMessageAvatar,\n  Content: AssistantMessageContent,\n};\n\nexport default Object.assign(\n  AssistantMessage,\n  exports,\n) as typeof AssistantMessage & typeof exports;\n"],"mappings":";;;AAEA,SAAmC,YAAY,eAAwB;AACvE,SAAS,wBAAwB;AACjC,OAAO,kBAAkB;AACzB,SAAS,cAAc;AACvB,SAAS,oBAAoB;AAC7B,SAAS,uBAAuB;AAChC,OAAO,wBAAwB;AAC/B,OAAO,iBAAiB;AAIpB,SACE,KADF;AAFJ,IAAM,mBAAuB,MAAM;AACjC,SACE,qBAAC,wBACC;AAAA,wBAAC,0BAAuB;AAAA,IACxB,oBAAC,2BAAwB;AAAA,IACzB,oBAAC,gBAAa;AAAA,IACd,oBAAC,sBAAmB;AAAA,KACtB;AAEJ;AAEA,iBAAiB,cAAc;AAE/B,IAAM,yBAA6B,MAAM;AACvC,QAAM,EAAE,iBAAiB,SAAS,EAAE,UAAU,IAAI,EAAE,IAAI,gBAAgB;AACxE,SAAO,oBAAC,UAAQ,GAAG,QAAQ;AAC7B;AAEA,IAAM,uBAAuB,aAAa,iBAAiB,MAAM;AAAA,EAC/D,WAAW;AACb,CAAC;AAED,qBAAqB,cAAc;AAEnC,IAAM,iCAAiC,aAAa,OAAO;AAAA,EACzD,WAAW;AACb,CAAC;AAQD,IAAM,0BAA0B,WAG9B,CAAC,EAAE,YAAY,gBAAgB,GAAG,KAAK,GAAG,QAAQ;AAClD,QAAM,EAAE,OAAO,kBAAkB,EAAE,aAAa,CAAC,EAAE,IAAI,CAAC,EAAE,IACxD,gBAAgB;AAElB,QAAM,kBAAkB;AAAA,IACtB,OAAO;AAAA,MACL,SAAS,CAAC,QACN,SACA,OAAO;AAAA,QACL,MAAM,IAAI,CAAC,MAAM;AAAA,UACf,EAAE,cAAc;AAAA,UAChB,EAAE,cAAc;AAAA,QAClB,CAAC;AAAA,MACH;AAAA,MACJ,UAAU,WAAW;AAAA,IACvB;AAAA;AAAA,IAEA,CAAC,GAAI,SAAS,CAAC,GAAI,WAAW,YAAY;AAAA,EAC5C;AAEA,SACE,oBAAC,kCAAgC,GAAG,MAAM,KACxC;AAAA,IAAC,iBAAiB;AAAA,IAAjB;AAAA,MACC,YAAY;AAAA,QACV,GAAG;AAAA,QACH,MAAM,gBAAgB,QAAQ,WAAW,QAAQ,YAAY;AAAA,QAC7D,OAAO,gBAAgB,SAAS,WAAW;AAAA,QAC3C,OAAO;AAAA,MACT;AAAA;AAAA,EACF,GACF;AAEJ,CAAC;AAED,wBAAwB,cAAc;AAEtC,IAAM,UAAU;AAAA,EACd,MAAM;AAAA,EACN,QAAQ;AAAA,EACR,SAAS;AACX;AAEA,IAAO,4BAAQ,OAAO;AAAA,EACpB;AAAA,EACA;AACF;","names":[]}
+{"version":3,"sources":["../../src/ui/assistant-message.tsx"],"sourcesContent":["\"use client\";\n\nimport { ComponentPropsWithoutRef, forwardRef, useMemo, type FC } from \"react\";\nimport { MessagePrimitive } from \"../primitives\";\nimport BranchPicker from \"./branch-picker\";\nimport { Avatar } from \"./base/avatar\";\nimport { withDefaults } from \"./utils/withDefaults\";\nimport { useThreadConfig } from \"./thread-config\";\nimport AssistantActionBar from \"./assistant-action-bar\";\nimport ContentPart from \"./content-part\";\n\nconst AssistantMessage: FC = () => {\n  return (\n    <AssistantMessageRoot>\n      <AssistantMessageAvatar />\n      <AssistantMessageContent />\n      <BranchPicker />\n      <AssistantActionBar />\n    </AssistantMessageRoot>\n  );\n};\n\nAssistantMessage.displayName = \"AssistantMessage\";\n\nconst AssistantMessageAvatar: FC = () => {\n  const { assistantAvatar: avatar = { fallback: \"A\" } } = useThreadConfig();\n  return <Avatar {...avatar} />;\n};\n\nconst AssistantMessageRoot = withDefaults(MessagePrimitive.Root, {\n  className: \"aui-assistant-message-root\",\n});\n\nAssistantMessageRoot.displayName = \"AssistantMessageRoot\";\n\nconst AssistantMessageContentWrapper = withDefaults(\"div\", {\n  className: \"aui-assistant-message-content\",\n});\n\nnamespace AssistantMessageContent {\n  export type Element = HTMLDivElement;\n  export type Props = MessagePrimitive.Content.Props &\n    ComponentPropsWithoutRef<\"div\">;\n}\n\nconst AssistantMessageContent = forwardRef<\n  AssistantMessageContent.Element,\n  AssistantMessageContent.Props\n>(({ components: componentsProp, ...rest }, ref) => {\n  const { tools, assistantMessage: { components = {} } = {} } =\n    useThreadConfig();\n\n  const toolsComponents = useMemo(\n    () => ({\n      by_name: !tools\n        ? undefined\n        : Object.fromEntries(\n            tools.map((t) => [\n              t.unstable_tool.toolName,\n              t.unstable_tool.render,\n            ]),\n          ),\n      Fallback: components.ToolFallback,\n    }),\n    // eslint-disable-next-line react-hooks/exhaustive-deps\n    [...(tools ?? []), components.ToolFallback],\n  );\n  const Footer = components.Footer;\n\n  return (\n    <AssistantMessageContentWrapper {...rest} ref={ref}>\n      <MessagePrimitive.Content\n        components={{\n          ...componentsProp,\n          Text: componentsProp?.Text ?? components.Text ?? ContentPart.Text,\n          Empty: componentsProp?.Empty ?? components.Empty,\n          tools: toolsComponents,\n        }}\n      />\n      {Footer && <Footer />}\n    </AssistantMessageContentWrapper>\n  );\n});\n\nAssistantMessageContent.displayName = \"AssistantMessageContent\";\n\nconst exports = {\n  Root: AssistantMessageRoot,\n  Avatar: AssistantMessageAvatar,\n  Content: AssistantMessageContent,\n};\n\nexport default Object.assign(\n  AssistantMessage,\n  exports,\n) as typeof AssistantMessage & typeof exports;\n"],"mappings":";;;AAEA,SAAmC,YAAY,eAAwB;AACvE,SAAS,wBAAwB;AACjC,OAAO,kBAAkB;AACzB,SAAS,cAAc;AACvB,SAAS,oBAAoB;AAC7B,SAAS,uBAAuB;AAChC,OAAO,wBAAwB;AAC/B,OAAO,iBAAiB;AAIpB,SACE,KADF;AAFJ,IAAM,mBAAuB,MAAM;AACjC,SACE,qBAAC,wBACC;AAAA,wBAAC,0BAAuB;AAAA,IACxB,oBAAC,2BAAwB;AAAA,IACzB,oBAAC,gBAAa;AAAA,IACd,oBAAC,sBAAmB;AAAA,KACtB;AAEJ;AAEA,iBAAiB,cAAc;AAE/B,IAAM,yBAA6B,MAAM;AACvC,QAAM,EAAE,iBAAiB,SAAS,EAAE,UAAU,IAAI,EAAE,IAAI,gBAAgB;AACxE,SAAO,oBAAC,UAAQ,GAAG,QAAQ;AAC7B;AAEA,IAAM,uBAAuB,aAAa,iBAAiB,MAAM;AAAA,EAC/D,WAAW;AACb,CAAC;AAED,qBAAqB,cAAc;AAEnC,IAAM,iCAAiC,aAAa,OAAO;AAAA,EACzD,WAAW;AACb,CAAC;AAQD,IAAM,0BAA0B,WAG9B,CAAC,EAAE,YAAY,gBAAgB,GAAG,KAAK,GAAG,QAAQ;AAClD,QAAM,EAAE,OAAO,kBAAkB,EAAE,aAAa,CAAC,EAAE,IAAI,CAAC,EAAE,IACxD,gBAAgB;AAElB,QAAM,kBAAkB;AAAA,IACtB,OAAO;AAAA,MACL,SAAS,CAAC,QACN,SACA,OAAO;AAAA,QACL,MAAM,IAAI,CAAC,MAAM;AAAA,UACf,EAAE,cAAc;AAAA,UAChB,EAAE,cAAc;AAAA,QAClB,CAAC;AAAA,MACH;AAAA,MACJ,UAAU,WAAW;AAAA,IACvB;AAAA;AAAA,IAEA,CAAC,GAAI,SAAS,CAAC,GAAI,WAAW,YAAY;AAAA,EAC5C;AACA,QAAM,SAAS,WAAW;AAE1B,SACE,qBAAC,kCAAgC,GAAG,MAAM,KACxC;AAAA;AAAA,MAAC,iBAAiB;AAAA,MAAjB;AAAA,QACC,YAAY;AAAA,UACV,GAAG;AAAA,UACH,MAAM,gBAAgB,QAAQ,WAAW,QAAQ,YAAY;AAAA,UAC7D,OAAO,gBAAgB,SAAS,WAAW;AAAA,UAC3C,OAAO;AAAA,QACT;AAAA;AAAA,IACF;AAAA,IACC,UAAU,oBAAC,UAAO;AAAA,KACrB;AAEJ,CAAC;AAED,wBAAwB,cAAc;AAEtC,IAAM,UAAU;AAAA,EACd,MAAM;AAAA,EACN,QAAQ;AAAA,EACR,SAAS;AACX;AAEA,IAAO,4BAAQ,OAAO;AAAA,EACpB;AAAA,EACA;AACF;","names":[]}

package/dist/ui/thread-config.d.ts

@@ -24,6 +24,7 @@ export type AssistantMessageConfig = {
         Text?: TextContentPartComponent | undefined;
         Empty?: EmptyContentPartComponent | undefined;
         ToolFallback?: ComponentType<ToolCallContentPartProps> | undefined;
+        Footer?: ComponentType | undefined;
     } | undefined;
 };
 export type BranchPickerConfig = {

package/dist/ui/thread-config.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"thread-config.d.ts","sourceRoot":"","sources":["../../src/ui/thread-config.tsx"],"names":[],"mappings":"AAEA,OAAO,EACL,aAAa,EACb,EAAE,EACF,iBAAiB,EACjB,SAAS,EAGV,MAAM,OAAO,CAAC;AAEf,OAAO,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAC3D,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAC5C,OAAO,EACL,yBAAyB,EACzB,wBAAwB,EACxB,wBAAwB,EACzB,MAAM,UAAU,CAAC;AAElB,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAGlD,MAAM,MAAM,gBAAgB,GAAG;IAC7B,IAAI,CAAC,EAAE,SAAS,GAAG,SAAS,CAAC;IAC7B,MAAM,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAAG;IAChC,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,GAAG,SAAS,CAAC;IACpC,WAAW,CAAC,EAAE,gBAAgB,EAAE,GAAG,SAAS,CAAC;CAC9C,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG;IAC9B,SAAS,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;CACjC,CAAC;AAEF,MAAM,MAAM,sBAAsB,GAAG;IACnC,WAAW,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IAClC,SAAS,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IAChC,UAAU,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACjC,qBAAqB,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IAC5C,qBAAqB,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IAC5C,UAAU,CAAC,EACP;QACE,IAAI,CAAC,EAAE,wBAAwB,GAAG,SAAS,CAAC;QAC5C,KAAK,CAAC,EAAE,yBAAyB,GAAG,SAAS,CAAC;QAC9C,YAAY,CAAC,EAAE,aAAa,CAAC,wBAAwB,CAAC,GAAG,SAAS,CAAC;KACpE,GACD,SAAS,CAAC;CACf,CAAC;AAEF,MAAM,MAAM,kBAAkB,GAAG;IAC/B,iBAAiB,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;CACzC,CAAC;AAEF,MAAM,MAAM,cAAc,GAAG;IAC3B,gBAAgB,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;CACxC,CAAC;AAEF,MAAM,MAAM,aAAa,GAAG;IAC1B,cAAc,CAAC,EAAE;QACf,IAAI,EAAE;YACJ,MAAM,EAAE;gBACN,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;aAC9B,CAAC;SACH,CAAC;QACF,MAAM,EAAE;YACN,MAAM,EAAE;gBACN,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;aAC9B,CAAC;SACH,CAAC;KACH,CAAC;IACF,UAAU,CAAC,EAAE;QACX,GAAG,CAAC,EAAE;YACJ,KAAK,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAC5B,CAAC;QACF,IAAI,CAAC,EAAE;YACL,KAAK,CAAC,EAAE;gBACN,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;aAC/B,CAAC;YACF,OAAO,CAAC,EAAE;gBACR,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;aAC9B,CAAC;SACH,CAAC;KACH,CAAC;IACF,MAAM,CAAC,EAAE;QACP,cAAc,CAAC,EAAE;YACf,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAC9B,CAAC;KACH,CAAC;IACF,OAAO,CAAC,EAAE;QACR,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;KAC9B,CAAC;IACF,WAAW,CAAC,EAAE;QACZ,IAAI,CAAC,EAAE;YACL,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAC9B,CAAC;KACH,CAAC;IACF,gBAAgB,CAAC,EAAE;QACjB,MAAM,CAAC,EAAE;YACP,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAC9B,CAAC;QACF,IAAI,CAAC,EAAE;YACL,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAC9B,CAAC;QACF,KAAK,CAAC,EAAE;YACN,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;YAC7B,IAAI,CAAC,EAAE;gBACL,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;aAC9B,CAAC;SACH,CAAC;QACF,QAAQ,CAAC,EAAE;YACT,QAAQ,CAAC,EAAE;gBACT,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;aAC9B,CAAC;YACF,QAAQ,CAAC,EAAE;gBACT,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;aAC9B,CAAC;SACH,CAAC;KACH,CAAC;IACF,YAAY,CAAC,EAAE;QACb,QAAQ,CAAC,EAAE;YACT,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAC9B,CAAC;QACF,IAAI,CAAC,EAAE;YACL,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAC9B,CAAC;KACH,CAAC;IACF,QAAQ,CAAC,EAAE;QACT,IAAI,CAAC,EACD;YACE,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAC9B,GACD,SAAS,CAAC;QACd,MAAM,CAAC,EACH;YACE,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAC9B,GACD,SAAS,CAAC;QACd,aAAa,CAAC,EACV;YACE,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAC9B,GACD,SAAS,CAAC;QACd,gBAAgB,CAAC,EAAE;YACjB,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAC9B,CAAC;QACF,KAAK,CAAC,EAAE;YACN,WAAW,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAClC,CAAC;KACH,CAAC;IACF,YAAY,CAAC,EAAE;QACb,IAAI,CAAC,EAAE;YACL,KAAK,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAC5B,CAAC;QACF,MAAM,CAAC,EAAE;YACP,KAAK,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAC5B,CAAC;KACH,CAAC;IACF,IAAI,CAAC,EAAE;QACL,MAAM,CAAC,EAAE;YACP,IAAI,CAAC,EAAE;gBACL,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;aAC9B,CAAC;SACH,CAAC;KACH,CAAC;CACH,CAAC;AAIF,MAAM,MAAM,YAAY,GAAG;IACzB,OAAO,CAAC,EAAE,gBAAgB,GAAG,SAAS,CAAC;IAEvC,eAAe,CAAC,EAAE,WAAW,GAAG,SAAS,CAAC;IAE1C,OAAO,CAAC,EAAE,mBAAmB,GAAG,SAAS,CAAC;IAC1C,gBAAgB,CAAC,EAAE,sBAAsB,GAAG,SAAS,CAAC;IACtD,WAAW,CAAC,EAAE,iBAAiB,GAAG,SAAS,CAAC;IAE5C,YAAY,CAAC,EAAE,kBAAkB,GAAG,SAAS,CAAC;IAE9C,QAAQ,CAAC,EAAE,cAAc,GAAG,SAAS,CAAC;IAEtC,OAAO,CAAC,EAAE,aAAa,GAAG,SAAS,CAAC;IAEpC,KAAK,CAAC,EAAE,eAAe,EAAE,GAAG,SAAS,CAAC;IAEtC,UAAU,CAAC,EACP;QACE,WAAW,CAAC,EAAE,aAAa,GAAG,SAAS,CAAC;QACxC,gBAAgB,CAAC,EAAE,aAAa,GAAG,SAAS,CAAC;QAC7C,YAAY,CAAC,EAAE,aAAa,GAAG,SAAS,CAAC;QACzC,QAAQ,CAAC,EAAE,aAAa,GAAG,SAAS,CAAC;QACrC,aAAa,CAAC,EAAE,aAAa,GAAG,SAAS,CAAC;KAC3C,GACD,SAAS,CAAC;CACf,CAAC;AAEF,eAAO,MAAM,eAAe,QAAO,IAAI,CAAC,YAAY,EAAE,SAAS,CAE9D,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG,iBAAiB,CAAC;IACxD,MAAM,CAAC,EAAE,YAAY,GAAG,SAAS,CAAC;CACnC,CAAC,CAAC;AAEH,eAAO,MAAM,oBAAoB,EAAE,EAAE,CAAC,yBAAyB,CAmC9D,CAAC"}
+{"version":3,"file":"thread-config.d.ts","sourceRoot":"","sources":["../../src/ui/thread-config.tsx"],"names":[],"mappings":"AAEA,OAAO,EACL,aAAa,EACb,EAAE,EACF,iBAAiB,EACjB,SAAS,EAGV,MAAM,OAAO,CAAC;AAEf,OAAO,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAC3D,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAC5C,OAAO,EACL,yBAAyB,EACzB,wBAAwB,EACxB,wBAAwB,EACzB,MAAM,UAAU,CAAC;AAElB,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAGlD,MAAM,MAAM,gBAAgB,GAAG;IAC7B,IAAI,CAAC,EAAE,SAAS,GAAG,SAAS,CAAC;IAC7B,MAAM,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF,MAAM,MAAM,mBAAmB,GAAG;IAChC,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,GAAG,SAAS,CAAC;IACpC,WAAW,CAAC,EAAE,gBAAgB,EAAE,GAAG,SAAS,CAAC;CAC9C,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG;IAC9B,SAAS,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;CACjC,CAAC;AAEF,MAAM,MAAM,sBAAsB,GAAG;IACnC,WAAW,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IAClC,SAAS,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IAChC,UAAU,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IACjC,qBAAqB,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IAC5C,qBAAqB,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IAC5C,UAAU,CAAC,EACP;QACE,IAAI,CAAC,EAAE,wBAAwB,GAAG,SAAS,CAAC;QAC5C,KAAK,CAAC,EAAE,yBAAyB,GAAG,SAAS,CAAC;QAC9C,YAAY,CAAC,EAAE,aAAa,CAAC,wBAAwB,CAAC,GAAG,SAAS,CAAC;QACnE,MAAM,CAAC,EAAE,aAAa,GAAG,SAAS,CAAC;KACpC,GACD,SAAS,CAAC;CACf,CAAC;AAEF,MAAM,MAAM,kBAAkB,GAAG;IAC/B,iBAAiB,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;CACzC,CAAC;AAEF,MAAM,MAAM,cAAc,GAAG;IAC3B,gBAAgB,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;CACxC,CAAC;AAEF,MAAM,MAAM,aAAa,GAAG;IAC1B,cAAc,CAAC,EAAE;QACf,IAAI,EAAE;YACJ,MAAM,EAAE;gBACN,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;aAC9B,CAAC;SACH,CAAC;QACF,MAAM,EAAE;YACN,MAAM,EAAE;gBACN,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;aAC9B,CAAC;SACH,CAAC;KACH,CAAC;IACF,UAAU,CAAC,EAAE;QACX,GAAG,CAAC,EAAE;YACJ,KAAK,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAC5B,CAAC;QACF,IAAI,CAAC,EAAE;YACL,KAAK,CAAC,EAAE;gBACN,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;aAC/B,CAAC;YACF,OAAO,CAAC,EAAE;gBACR,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;aAC9B,CAAC;SACH,CAAC;KACH,CAAC;IACF,MAAM,CAAC,EAAE;QACP,cAAc,CAAC,EAAE;YACf,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAC9B,CAAC;KACH,CAAC;IACF,OAAO,CAAC,EAAE;QACR,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;KAC9B,CAAC;IACF,WAAW,CAAC,EAAE;QACZ,IAAI,CAAC,EAAE;YACL,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAC9B,CAAC;KACH,CAAC;IACF,gBAAgB,CAAC,EAAE;QACjB,MAAM,CAAC,EAAE;YACP,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAC9B,CAAC;QACF,IAAI,CAAC,EAAE;YACL,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAC9B,CAAC;QACF,KAAK,CAAC,EAAE;YACN,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;YAC7B,IAAI,CAAC,EAAE;gBACL,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;aAC9B,CAAC;SACH,CAAC;QACF,QAAQ,CAAC,EAAE;YACT,QAAQ,CAAC,EAAE;gBACT,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;aAC9B,CAAC;YACF,QAAQ,CAAC,EAAE;gBACT,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;aAC9B,CAAC;SACH,CAAC;KACH,CAAC;IACF,YAAY,CAAC,EAAE;QACb,QAAQ,CAAC,EAAE;YACT,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAC9B,CAAC;QACF,IAAI,CAAC,EAAE;YACL,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAC9B,CAAC;KACH,CAAC;IACF,QAAQ,CAAC,EAAE;QACT,IAAI,CAAC,EACD;YACE,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAC9B,GACD,SAAS,CAAC;QACd,MAAM,CAAC,EACH;YACE,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAC9B,GACD,SAAS,CAAC;QACd,aAAa,CAAC,EACV;YACE,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAC9B,GACD,SAAS,CAAC;QACd,gBAAgB,CAAC,EAAE;YACjB,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAC9B,CAAC;QACF,KAAK,CAAC,EAAE;YACN,WAAW,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAClC,CAAC;KACH,CAAC;IACF,YAAY,CAAC,EAAE;QACb,IAAI,CAAC,EAAE;YACL,KAAK,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAC5B,CAAC;QACF,MAAM,CAAC,EAAE;YACP,KAAK,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;SAC5B,CAAC;KACH,CAAC;IACF,IAAI,CAAC,EAAE;QACL,MAAM,CAAC,EAAE;YACP,IAAI,CAAC,EAAE;gBACL,OAAO,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;aAC9B,CAAC;SACH,CAAC;KACH,CAAC;CACH,CAAC;AAIF,MAAM,MAAM,YAAY,GAAG;IACzB,OAAO,CAAC,EAAE,gBAAgB,GAAG,SAAS,CAAC;IAEvC,eAAe,CAAC,EAAE,WAAW,GAAG,SAAS,CAAC;IAE1C,OAAO,CAAC,EAAE,mBAAmB,GAAG,SAAS,CAAC;IAC1C,gBAAgB,CAAC,EAAE,sBAAsB,GAAG,SAAS,CAAC;IACtD,WAAW,CAAC,EAAE,iBAAiB,GAAG,SAAS,CAAC;IAE5C,YAAY,CAAC,EAAE,kBAAkB,GAAG,SAAS,CAAC;IAE9C,QAAQ,CAAC,EAAE,cAAc,GAAG,SAAS,CAAC;IAEtC,OAAO,CAAC,EAAE,aAAa,GAAG,SAAS,CAAC;IAEpC,KAAK,CAAC,EAAE,eAAe,EAAE,GAAG,SAAS,CAAC;IAEtC,UAAU,CAAC,EACP;QACE,WAAW,CAAC,EAAE,aAAa,GAAG,SAAS,CAAC;QACxC,gBAAgB,CAAC,EAAE,aAAa,GAAG,SAAS,CAAC;QAC7C,YAAY,CAAC,EAAE,aAAa,GAAG,SAAS,CAAC;QACzC,QAAQ,CAAC,EAAE,aAAa,GAAG,SAAS,CAAC;QACrC,aAAa,CAAC,EAAE,aAAa,GAAG,SAAS,CAAC;KAC3C,GACD,SAAS,CAAC;CACf,CAAC;AAEF,eAAO,MAAM,eAAe,QAAO,IAAI,CAAC,YAAY,EAAE,SAAS,CAE9D,CAAC;AAEF,MAAM,MAAM,yBAAyB,GAAG,iBAAiB,CAAC;IACxD,MAAM,CAAC,EAAE,YAAY,GAAG,SAAS,CAAC;CACnC,CAAC,CAAC;AAEH,eAAO,MAAM,oBAAoB,EAAE,EAAE,CAAC,yBAAyB,CAmC9D,CAAC"}

package/dist/ui/thread-config.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/ui/thread-config.tsx"],"sourcesContent":["\"use client\";\n\nimport {\n  ComponentType,\n  FC,\n  PropsWithChildren,\n  ReactNode,\n  createContext,\n  useContext,\n} from \"react\";\n\nimport { AssistantRuntime } from \"../api/AssistantRuntime\";\nimport { AvatarProps } from \"./base/avatar\";\nimport {\n  EmptyContentPartComponent,\n  TextContentPartComponent,\n  ToolCallContentPartProps,\n} from \"../types\";\nimport { AssistantRuntimeProvider } from \"../context\";\nimport { AssistantToolUI } from \"../model-config\";\nimport { useAssistantRuntime } from \"../context/react/AssistantContext\";\n\nexport type SuggestionConfig = {\n  text?: ReactNode | undefined;\n  prompt: string;\n};\n\nexport type ThreadWelcomeConfig = {\n  message?: string | null | undefined;\n  suggestions?: SuggestionConfig[] | undefined;\n};\n\nexport type UserMessageConfig = {\n  allowEdit?: boolean | undefined;\n};\n\nexport type AssistantMessageConfig = {\n  allowReload?: boolean | undefined;\n  allowCopy?: boolean | undefined;\n  allowSpeak?: boolean | undefined;\n  allowFeedbackPositive?: boolean | undefined;\n  allowFeedbackNegative?: boolean | undefined;\n  components?:\n    | {\n        Text?: TextContentPartComponent | undefined;\n        Empty?: EmptyContentPartComponent | undefined;\n        ToolFallback?: ComponentType<ToolCallContentPartProps> | undefined;\n      }\n    | undefined;\n};\n\nexport type BranchPickerConfig = {\n  allowBranchPicker?: boolean | undefined;\n};\n\nexport type ComposerConfig = {\n  allowAttachments?: boolean | undefined;\n};\n\nexport type StringsConfig = {\n  assistantModal?: {\n    open: {\n      button: {\n        tooltip?: string | undefined;\n      };\n    };\n    closed: {\n      button: {\n        tooltip?: string | undefined;\n      };\n    };\n  };\n  threadList?: {\n    new?: {\n      label?: string | undefined;\n    };\n    item?: {\n      title?: {\n        fallback?: string | undefined;\n      };\n      archive?: {\n        tooltip?: string | undefined;\n      };\n    };\n  };\n  thread?: {\n    scrollToBottom?: {\n      tooltip?: string | undefined;\n    };\n  };\n  welcome?: {\n    message?: string | undefined;\n  };\n  userMessage?: {\n    edit?: {\n      tooltip?: string | undefined;\n    };\n  };\n  assistantMessage?: {\n    reload?: {\n      tooltip?: string | undefined;\n    };\n    copy?: {\n      tooltip?: string | undefined;\n    };\n    speak?: {\n      tooltip?: string | undefined;\n      stop?: {\n        tooltip?: string | undefined;\n      };\n    };\n    feedback?: {\n      positive?: {\n        tooltip?: string | undefined;\n      };\n      negative?: {\n        tooltip?: string | undefined;\n      };\n    };\n  };\n  branchPicker?: {\n    previous?: {\n      tooltip?: string | undefined;\n    };\n    next?: {\n      tooltip?: string | undefined;\n    };\n  };\n  composer?: {\n    send?:\n      | {\n          tooltip?: string | undefined;\n        }\n      | undefined;\n    cancel?:\n      | {\n          tooltip?: string | undefined;\n        }\n      | undefined;\n    addAttachment?:\n      | {\n          tooltip?: string | undefined;\n        }\n      | undefined;\n    removeAttachment?: {\n      tooltip?: string | undefined;\n    };\n    input?: {\n      placeholder?: string | undefined;\n    };\n  };\n  editComposer?: {\n    send?: {\n      label?: string | undefined;\n    };\n    cancel?: {\n      label?: string | undefined;\n    };\n  };\n  code?: {\n    header?: {\n      copy?: {\n        tooltip?: string | undefined;\n      };\n    };\n  };\n};\n\nconst ThreadConfigContext = createContext<ThreadConfig>({});\n\nexport type ThreadConfig = {\n  runtime?: AssistantRuntime | undefined;\n\n  assistantAvatar?: AvatarProps | undefined;\n\n  welcome?: ThreadWelcomeConfig | undefined;\n  assistantMessage?: AssistantMessageConfig | undefined;\n  userMessage?: UserMessageConfig | undefined;\n\n  branchPicker?: BranchPickerConfig | undefined;\n\n  composer?: ComposerConfig | undefined;\n\n  strings?: StringsConfig | undefined;\n\n  tools?: AssistantToolUI[] | undefined; // TODO add AssistantTool support\n\n  components?:\n    | {\n        UserMessage?: ComponentType | undefined;\n        AssistantMessage?: ComponentType | undefined;\n        EditComposer?: ComponentType | undefined;\n        Composer?: ComponentType | undefined;\n        ThreadWelcome?: ComponentType | undefined;\n      }\n    | undefined;\n};\n\nexport const useThreadConfig = (): Omit<ThreadConfig, \"runtime\"> => {\n  return useContext(ThreadConfigContext);\n};\n\nexport type ThreadConfigProviderProps = PropsWithChildren<{\n  config?: ThreadConfig | undefined;\n}>;\n\nexport const ThreadConfigProvider: FC<ThreadConfigProviderProps> = ({\n  children,\n  config,\n}) => {\n  const hasAssistant = !!useAssistantRuntime({ optional: true });\n\n  const hasConfig = config && Object.keys(config).length > 0;\n  const outerConfig = useThreadConfig();\n\n  if (hasConfig && Object.keys(outerConfig).length > 0) {\n    throw new Error(\n      \"You are providing ThreadConfig to several nested components. Please provide all configuration to the same component.\",\n    );\n  }\n\n  const configProvider = hasConfig ? (\n    <ThreadConfigContext.Provider value={config}>\n      {children}\n    </ThreadConfigContext.Provider>\n  ) : (\n    <>{children}</>\n  );\n\n  if (!config?.runtime) return configProvider;\n\n  if (hasAssistant) {\n    throw new Error(\n      \"You provided a runtime to <Thread> while simulataneously using <AssistantRuntimeProvider>. This is not allowed.\",\n    );\n  }\n  return (\n    <AssistantRuntimeProvider runtime={config.runtime}>\n      {configProvider}\n    </AssistantRuntimeProvider>\n  );\n};\n\nThreadConfigProvider.displayName = \"ThreadConfigProvider\";\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAEA,mBAOO;AASP,qBAAyC;AAEzC,8BAAoC;AA0MhC;AAtDJ,IAAM,0BAAsB,4BAA4B,CAAC,CAAC;AA8BnD,IAAM,kBAAkB,MAAqC;AAClE,aAAO,yBAAW,mBAAmB;AACvC;AAMO,IAAM,uBAAsD,CAAC;AAAA,EAClE;AAAA,EACA;AACF,MAAM;AACJ,QAAM,eAAe,CAAC,KAAC,6CAAoB,EAAE,UAAU,KAAK,CAAC;AAE7D,QAAM,YAAY,UAAU,OAAO,KAAK,MAAM,EAAE,SAAS;AACzD,QAAM,cAAc,gBAAgB;AAEpC,MAAI,aAAa,OAAO,KAAK,WAAW,EAAE,SAAS,GAAG;AACpD,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AAEA,QAAM,iBAAiB,YACrB,4CAAC,oBAAoB,UAApB,EAA6B,OAAO,QAClC,UACH,IAEA,2EAAG,UAAS;AAGd,MAAI,CAAC,QAAQ,QAAS,QAAO;AAE7B,MAAI,cAAc;AAChB,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AACA,SACE,4CAAC,2CAAyB,SAAS,OAAO,SACvC,0BACH;AAEJ;AAEA,qBAAqB,cAAc;","names":[]}
+{"version":3,"sources":["../../src/ui/thread-config.tsx"],"sourcesContent":["\"use client\";\n\nimport {\n  ComponentType,\n  FC,\n  PropsWithChildren,\n  ReactNode,\n  createContext,\n  useContext,\n} from \"react\";\n\nimport { AssistantRuntime } from \"../api/AssistantRuntime\";\nimport { AvatarProps } from \"./base/avatar\";\nimport {\n  EmptyContentPartComponent,\n  TextContentPartComponent,\n  ToolCallContentPartProps,\n} from \"../types\";\nimport { AssistantRuntimeProvider } from \"../context\";\nimport { AssistantToolUI } from \"../model-config\";\nimport { useAssistantRuntime } from \"../context/react/AssistantContext\";\n\nexport type SuggestionConfig = {\n  text?: ReactNode | undefined;\n  prompt: string;\n};\n\nexport type ThreadWelcomeConfig = {\n  message?: string | null | undefined;\n  suggestions?: SuggestionConfig[] | undefined;\n};\n\nexport type UserMessageConfig = {\n  allowEdit?: boolean | undefined;\n};\n\nexport type AssistantMessageConfig = {\n  allowReload?: boolean | undefined;\n  allowCopy?: boolean | undefined;\n  allowSpeak?: boolean | undefined;\n  allowFeedbackPositive?: boolean | undefined;\n  allowFeedbackNegative?: boolean | undefined;\n  components?:\n    | {\n        Text?: TextContentPartComponent | undefined;\n        Empty?: EmptyContentPartComponent | undefined;\n        ToolFallback?: ComponentType<ToolCallContentPartProps> | undefined;\n        Footer?: ComponentType | undefined;\n      }\n    | undefined;\n};\n\nexport type BranchPickerConfig = {\n  allowBranchPicker?: boolean | undefined;\n};\n\nexport type ComposerConfig = {\n  allowAttachments?: boolean | undefined;\n};\n\nexport type StringsConfig = {\n  assistantModal?: {\n    open: {\n      button: {\n        tooltip?: string | undefined;\n      };\n    };\n    closed: {\n      button: {\n        tooltip?: string | undefined;\n      };\n    };\n  };\n  threadList?: {\n    new?: {\n      label?: string | undefined;\n    };\n    item?: {\n      title?: {\n        fallback?: string | undefined;\n      };\n      archive?: {\n        tooltip?: string | undefined;\n      };\n    };\n  };\n  thread?: {\n    scrollToBottom?: {\n      tooltip?: string | undefined;\n    };\n  };\n  welcome?: {\n    message?: string | undefined;\n  };\n  userMessage?: {\n    edit?: {\n      tooltip?: string | undefined;\n    };\n  };\n  assistantMessage?: {\n    reload?: {\n      tooltip?: string | undefined;\n    };\n    copy?: {\n      tooltip?: string | undefined;\n    };\n    speak?: {\n      tooltip?: string | undefined;\n      stop?: {\n        tooltip?: string | undefined;\n      };\n    };\n    feedback?: {\n      positive?: {\n        tooltip?: string | undefined;\n      };\n      negative?: {\n        tooltip?: string | undefined;\n      };\n    };\n  };\n  branchPicker?: {\n    previous?: {\n      tooltip?: string | undefined;\n    };\n    next?: {\n      tooltip?: string | undefined;\n    };\n  };\n  composer?: {\n    send?:\n      | {\n          tooltip?: string | undefined;\n        }\n      | undefined;\n    cancel?:\n      | {\n          tooltip?: string | undefined;\n        }\n      | undefined;\n    addAttachment?:\n      | {\n          tooltip?: string | undefined;\n        }\n      | undefined;\n    removeAttachment?: {\n      tooltip?: string | undefined;\n    };\n    input?: {\n      placeholder?: string | undefined;\n    };\n  };\n  editComposer?: {\n    send?: {\n      label?: string | undefined;\n    };\n    cancel?: {\n      label?: string | undefined;\n    };\n  };\n  code?: {\n    header?: {\n      copy?: {\n        tooltip?: string | undefined;\n      };\n    };\n  };\n};\n\nconst ThreadConfigContext = createContext<ThreadConfig>({});\n\nexport type ThreadConfig = {\n  runtime?: AssistantRuntime | undefined;\n\n  assistantAvatar?: AvatarProps | undefined;\n\n  welcome?: ThreadWelcomeConfig | undefined;\n  assistantMessage?: AssistantMessageConfig | undefined;\n  userMessage?: UserMessageConfig | undefined;\n\n  branchPicker?: BranchPickerConfig | undefined;\n\n  composer?: ComposerConfig | undefined;\n\n  strings?: StringsConfig | undefined;\n\n  tools?: AssistantToolUI[] | undefined; // TODO add AssistantTool support\n\n  components?:\n    | {\n        UserMessage?: ComponentType | undefined;\n        AssistantMessage?: ComponentType | undefined;\n        EditComposer?: ComponentType | undefined;\n        Composer?: ComponentType | undefined;\n        ThreadWelcome?: ComponentType | undefined;\n      }\n    | undefined;\n};\n\nexport const useThreadConfig = (): Omit<ThreadConfig, \"runtime\"> => {\n  return useContext(ThreadConfigContext);\n};\n\nexport type ThreadConfigProviderProps = PropsWithChildren<{\n  config?: ThreadConfig | undefined;\n}>;\n\nexport const ThreadConfigProvider: FC<ThreadConfigProviderProps> = ({\n  children,\n  config,\n}) => {\n  const hasAssistant = !!useAssistantRuntime({ optional: true });\n\n  const hasConfig = config && Object.keys(config).length > 0;\n  const outerConfig = useThreadConfig();\n\n  if (hasConfig && Object.keys(outerConfig).length > 0) {\n    throw new Error(\n      \"You are providing ThreadConfig to several nested components. Please provide all configuration to the same component.\",\n    );\n  }\n\n  const configProvider = hasConfig ? (\n    <ThreadConfigContext.Provider value={config}>\n      {children}\n    </ThreadConfigContext.Provider>\n  ) : (\n    <>{children}</>\n  );\n\n  if (!config?.runtime) return configProvider;\n\n  if (hasAssistant) {\n    throw new Error(\n      \"You provided a runtime to <Thread> while simulataneously using <AssistantRuntimeProvider>. This is not allowed.\",\n    );\n  }\n  return (\n    <AssistantRuntimeProvider runtime={config.runtime}>\n      {configProvider}\n    </AssistantRuntimeProvider>\n  );\n};\n\nThreadConfigProvider.displayName = \"ThreadConfigProvider\";\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAEA,mBAOO;AASP,qBAAyC;AAEzC,8BAAoC;AA2MhC;AAtDJ,IAAM,0BAAsB,4BAA4B,CAAC,CAAC;AA8BnD,IAAM,kBAAkB,MAAqC;AAClE,aAAO,yBAAW,mBAAmB;AACvC;AAMO,IAAM,uBAAsD,CAAC;AAAA,EAClE;AAAA,EACA;AACF,MAAM;AACJ,QAAM,eAAe,CAAC,KAAC,6CAAoB,EAAE,UAAU,KAAK,CAAC;AAE7D,QAAM,YAAY,UAAU,OAAO,KAAK,MAAM,EAAE,SAAS;AACzD,QAAM,cAAc,gBAAgB;AAEpC,MAAI,aAAa,OAAO,KAAK,WAAW,EAAE,SAAS,GAAG;AACpD,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AAEA,QAAM,iBAAiB,YACrB,4CAAC,oBAAoB,UAApB,EAA6B,OAAO,QAClC,UACH,IAEA,2EAAG,UAAS;AAGd,MAAI,CAAC,QAAQ,QAAS,QAAO;AAE7B,MAAI,cAAc;AAChB,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AACA,SACE,4CAAC,2CAAyB,SAAS,OAAO,SACvC,0BACH;AAEJ;AAEA,qBAAqB,cAAc;","names":[]}

package/dist/ui/thread-config.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/ui/thread-config.tsx"],"sourcesContent":["\"use client\";\n\nimport {\n  ComponentType,\n  FC,\n  PropsWithChildren,\n  ReactNode,\n  createContext,\n  useContext,\n} from \"react\";\n\nimport { AssistantRuntime } from \"../api/AssistantRuntime\";\nimport { AvatarProps } from \"./base/avatar\";\nimport {\n  EmptyContentPartComponent,\n  TextContentPartComponent,\n  ToolCallContentPartProps,\n} from \"../types\";\nimport { AssistantRuntimeProvider } from \"../context\";\nimport { AssistantToolUI } from \"../model-config\";\nimport { useAssistantRuntime } from \"../context/react/AssistantContext\";\n\nexport type SuggestionConfig = {\n  text?: ReactNode | undefined;\n  prompt: string;\n};\n\nexport type ThreadWelcomeConfig = {\n  message?: string | null | undefined;\n  suggestions?: SuggestionConfig[] | undefined;\n};\n\nexport type UserMessageConfig = {\n  allowEdit?: boolean | undefined;\n};\n\nexport type AssistantMessageConfig = {\n  allowReload?: boolean | undefined;\n  allowCopy?: boolean | undefined;\n  allowSpeak?: boolean | undefined;\n  allowFeedbackPositive?: boolean | undefined;\n  allowFeedbackNegative?: boolean | undefined;\n  components?:\n    | {\n        Text?: TextContentPartComponent | undefined;\n        Empty?: EmptyContentPartComponent | undefined;\n        ToolFallback?: ComponentType<ToolCallContentPartProps> | undefined;\n      }\n    | undefined;\n};\n\nexport type BranchPickerConfig = {\n  allowBranchPicker?: boolean | undefined;\n};\n\nexport type ComposerConfig = {\n  allowAttachments?: boolean | undefined;\n};\n\nexport type StringsConfig = {\n  assistantModal?: {\n    open: {\n      button: {\n        tooltip?: string | undefined;\n      };\n    };\n    closed: {\n      button: {\n        tooltip?: string | undefined;\n      };\n    };\n  };\n  threadList?: {\n    new?: {\n      label?: string | undefined;\n    };\n    item?: {\n      title?: {\n        fallback?: string | undefined;\n      };\n      archive?: {\n        tooltip?: string | undefined;\n      };\n    };\n  };\n  thread?: {\n    scrollToBottom?: {\n      tooltip?: string | undefined;\n    };\n  };\n  welcome?: {\n    message?: string | undefined;\n  };\n  userMessage?: {\n    edit?: {\n      tooltip?: string | undefined;\n    };\n  };\n  assistantMessage?: {\n    reload?: {\n      tooltip?: string | undefined;\n    };\n    copy?: {\n      tooltip?: string | undefined;\n    };\n    speak?: {\n      tooltip?: string | undefined;\n      stop?: {\n        tooltip?: string | undefined;\n      };\n    };\n    feedback?: {\n      positive?: {\n        tooltip?: string | undefined;\n      };\n      negative?: {\n        tooltip?: string | undefined;\n      };\n    };\n  };\n  branchPicker?: {\n    previous?: {\n      tooltip?: string | undefined;\n    };\n    next?: {\n      tooltip?: string | undefined;\n    };\n  };\n  composer?: {\n    send?:\n      | {\n          tooltip?: string | undefined;\n        }\n      | undefined;\n    cancel?:\n      | {\n          tooltip?: string | undefined;\n        }\n      | undefined;\n    addAttachment?:\n      | {\n          tooltip?: string | undefined;\n        }\n      | undefined;\n    removeAttachment?: {\n      tooltip?: string | undefined;\n    };\n    input?: {\n      placeholder?: string | undefined;\n    };\n  };\n  editComposer?: {\n    send?: {\n      label?: string | undefined;\n    };\n    cancel?: {\n      label?: string | undefined;\n    };\n  };\n  code?: {\n    header?: {\n      copy?: {\n        tooltip?: string | undefined;\n      };\n    };\n  };\n};\n\nconst ThreadConfigContext = createContext<ThreadConfig>({});\n\nexport type ThreadConfig = {\n  runtime?: AssistantRuntime | undefined;\n\n  assistantAvatar?: AvatarProps | undefined;\n\n  welcome?: ThreadWelcomeConfig | undefined;\n  assistantMessage?: AssistantMessageConfig | undefined;\n  userMessage?: UserMessageConfig | undefined;\n\n  branchPicker?: BranchPickerConfig | undefined;\n\n  composer?: ComposerConfig | undefined;\n\n  strings?: StringsConfig | undefined;\n\n  tools?: AssistantToolUI[] | undefined; // TODO add AssistantTool support\n\n  components?:\n    | {\n        UserMessage?: ComponentType | undefined;\n        AssistantMessage?: ComponentType | undefined;\n        EditComposer?: ComponentType | undefined;\n        Composer?: ComponentType | undefined;\n        ThreadWelcome?: ComponentType | undefined;\n      }\n    | undefined;\n};\n\nexport const useThreadConfig = (): Omit<ThreadConfig, \"runtime\"> => {\n  return useContext(ThreadConfigContext);\n};\n\nexport type ThreadConfigProviderProps = PropsWithChildren<{\n  config?: ThreadConfig | undefined;\n}>;\n\nexport const ThreadConfigProvider: FC<ThreadConfigProviderProps> = ({\n  children,\n  config,\n}) => {\n  const hasAssistant = !!useAssistantRuntime({ optional: true });\n\n  const hasConfig = config && Object.keys(config).length > 0;\n  const outerConfig = useThreadConfig();\n\n  if (hasConfig && Object.keys(outerConfig).length > 0) {\n    throw new Error(\n      \"You are providing ThreadConfig to several nested components. Please provide all configuration to the same component.\",\n    );\n  }\n\n  const configProvider = hasConfig ? (\n    <ThreadConfigContext.Provider value={config}>\n      {children}\n    </ThreadConfigContext.Provider>\n  ) : (\n    <>{children}</>\n  );\n\n  if (!config?.runtime) return configProvider;\n\n  if (hasAssistant) {\n    throw new Error(\n      \"You provided a runtime to <Thread> while simulataneously using <AssistantRuntimeProvider>. This is not allowed.\",\n    );\n  }\n  return (\n    <AssistantRuntimeProvider runtime={config.runtime}>\n      {configProvider}\n    </AssistantRuntimeProvider>\n  );\n};\n\nThreadConfigProvider.displayName = \"ThreadConfigProvider\";\n"],"mappings":";;;AAEA;AAAA,EAKE;AAAA,EACA;AAAA,OACK;AASP,SAAS,gCAAgC;AAEzC,SAAS,2BAA2B;AA0MhC,SAIA,UAJA;AAtDJ,IAAM,sBAAsB,cAA4B,CAAC,CAAC;AA8BnD,IAAM,kBAAkB,MAAqC;AAClE,SAAO,WAAW,mBAAmB;AACvC;AAMO,IAAM,uBAAsD,CAAC;AAAA,EAClE;AAAA,EACA;AACF,MAAM;AACJ,QAAM,eAAe,CAAC,CAAC,oBAAoB,EAAE,UAAU,KAAK,CAAC;AAE7D,QAAM,YAAY,UAAU,OAAO,KAAK,MAAM,EAAE,SAAS;AACzD,QAAM,cAAc,gBAAgB;AAEpC,MAAI,aAAa,OAAO,KAAK,WAAW,EAAE,SAAS,GAAG;AACpD,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AAEA,QAAM,iBAAiB,YACrB,oBAAC,oBAAoB,UAApB,EAA6B,OAAO,QAClC,UACH,IAEA,gCAAG,UAAS;AAGd,MAAI,CAAC,QAAQ,QAAS,QAAO;AAE7B,MAAI,cAAc;AAChB,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AACA,SACE,oBAAC,4BAAyB,SAAS,OAAO,SACvC,0BACH;AAEJ;AAEA,qBAAqB,cAAc;","names":[]}
+{"version":3,"sources":["../../src/ui/thread-config.tsx"],"sourcesContent":["\"use client\";\n\nimport {\n  ComponentType,\n  FC,\n  PropsWithChildren,\n  ReactNode,\n  createContext,\n  useContext,\n} from \"react\";\n\nimport { AssistantRuntime } from \"../api/AssistantRuntime\";\nimport { AvatarProps } from \"./base/avatar\";\nimport {\n  EmptyContentPartComponent,\n  TextContentPartComponent,\n  ToolCallContentPartProps,\n} from \"../types\";\nimport { AssistantRuntimeProvider } from \"../context\";\nimport { AssistantToolUI } from \"../model-config\";\nimport { useAssistantRuntime } from \"../context/react/AssistantContext\";\n\nexport type SuggestionConfig = {\n  text?: ReactNode | undefined;\n  prompt: string;\n};\n\nexport type ThreadWelcomeConfig = {\n  message?: string | null | undefined;\n  suggestions?: SuggestionConfig[] | undefined;\n};\n\nexport type UserMessageConfig = {\n  allowEdit?: boolean | undefined;\n};\n\nexport type AssistantMessageConfig = {\n  allowReload?: boolean | undefined;\n  allowCopy?: boolean | undefined;\n  allowSpeak?: boolean | undefined;\n  allowFeedbackPositive?: boolean | undefined;\n  allowFeedbackNegative?: boolean | undefined;\n  components?:\n    | {\n        Text?: TextContentPartComponent | undefined;\n        Empty?: EmptyContentPartComponent | undefined;\n        ToolFallback?: ComponentType<ToolCallContentPartProps> | undefined;\n        Footer?: ComponentType | undefined;\n      }\n    | undefined;\n};\n\nexport type BranchPickerConfig = {\n  allowBranchPicker?: boolean | undefined;\n};\n\nexport type ComposerConfig = {\n  allowAttachments?: boolean | undefined;\n};\n\nexport type StringsConfig = {\n  assistantModal?: {\n    open: {\n      button: {\n        tooltip?: string | undefined;\n      };\n    };\n    closed: {\n      button: {\n        tooltip?: string | undefined;\n      };\n    };\n  };\n  threadList?: {\n    new?: {\n      label?: string | undefined;\n    };\n    item?: {\n      title?: {\n        fallback?: string | undefined;\n      };\n      archive?: {\n        tooltip?: string | undefined;\n      };\n    };\n  };\n  thread?: {\n    scrollToBottom?: {\n      tooltip?: string | undefined;\n    };\n  };\n  welcome?: {\n    message?: string | undefined;\n  };\n  userMessage?: {\n    edit?: {\n      tooltip?: string | undefined;\n    };\n  };\n  assistantMessage?: {\n    reload?: {\n      tooltip?: string | undefined;\n    };\n    copy?: {\n      tooltip?: string | undefined;\n    };\n    speak?: {\n      tooltip?: string | undefined;\n      stop?: {\n        tooltip?: string | undefined;\n      };\n    };\n    feedback?: {\n      positive?: {\n        tooltip?: string | undefined;\n      };\n      negative?: {\n        tooltip?: string | undefined;\n      };\n    };\n  };\n  branchPicker?: {\n    previous?: {\n      tooltip?: string | undefined;\n    };\n    next?: {\n      tooltip?: string | undefined;\n    };\n  };\n  composer?: {\n    send?:\n      | {\n          tooltip?: string | undefined;\n        }\n      | undefined;\n    cancel?:\n      | {\n          tooltip?: string | undefined;\n        }\n      | undefined;\n    addAttachment?:\n      | {\n          tooltip?: string | undefined;\n        }\n      | undefined;\n    removeAttachment?: {\n      tooltip?: string | undefined;\n    };\n    input?: {\n      placeholder?: string | undefined;\n    };\n  };\n  editComposer?: {\n    send?: {\n      label?: string | undefined;\n    };\n    cancel?: {\n      label?: string | undefined;\n    };\n  };\n  code?: {\n    header?: {\n      copy?: {\n        tooltip?: string | undefined;\n      };\n    };\n  };\n};\n\nconst ThreadConfigContext = createContext<ThreadConfig>({});\n\nexport type ThreadConfig = {\n  runtime?: AssistantRuntime | undefined;\n\n  assistantAvatar?: AvatarProps | undefined;\n\n  welcome?: ThreadWelcomeConfig | undefined;\n  assistantMessage?: AssistantMessageConfig | undefined;\n  userMessage?: UserMessageConfig | undefined;\n\n  branchPicker?: BranchPickerConfig | undefined;\n\n  composer?: ComposerConfig | undefined;\n\n  strings?: StringsConfig | undefined;\n\n  tools?: AssistantToolUI[] | undefined; // TODO add AssistantTool support\n\n  components?:\n    | {\n        UserMessage?: ComponentType | undefined;\n        AssistantMessage?: ComponentType | undefined;\n        EditComposer?: ComponentType | undefined;\n        Composer?: ComponentType | undefined;\n        ThreadWelcome?: ComponentType | undefined;\n      }\n    | undefined;\n};\n\nexport const useThreadConfig = (): Omit<ThreadConfig, \"runtime\"> => {\n  return useContext(ThreadConfigContext);\n};\n\nexport type ThreadConfigProviderProps = PropsWithChildren<{\n  config?: ThreadConfig | undefined;\n}>;\n\nexport const ThreadConfigProvider: FC<ThreadConfigProviderProps> = ({\n  children,\n  config,\n}) => {\n  const hasAssistant = !!useAssistantRuntime({ optional: true });\n\n  const hasConfig = config && Object.keys(config).length > 0;\n  const outerConfig = useThreadConfig();\n\n  if (hasConfig && Object.keys(outerConfig).length > 0) {\n    throw new Error(\n      \"You are providing ThreadConfig to several nested components. Please provide all configuration to the same component.\",\n    );\n  }\n\n  const configProvider = hasConfig ? (\n    <ThreadConfigContext.Provider value={config}>\n      {children}\n    </ThreadConfigContext.Provider>\n  ) : (\n    <>{children}</>\n  );\n\n  if (!config?.runtime) return configProvider;\n\n  if (hasAssistant) {\n    throw new Error(\n      \"You provided a runtime to <Thread> while simulataneously using <AssistantRuntimeProvider>. This is not allowed.\",\n    );\n  }\n  return (\n    <AssistantRuntimeProvider runtime={config.runtime}>\n      {configProvider}\n    </AssistantRuntimeProvider>\n  );\n};\n\nThreadConfigProvider.displayName = \"ThreadConfigProvider\";\n"],"mappings":";;;AAEA;AAAA,EAKE;AAAA,EACA;AAAA,OACK;AASP,SAAS,gCAAgC;AAEzC,SAAS,2BAA2B;AA2MhC,SAIA,UAJA;AAtDJ,IAAM,sBAAsB,cAA4B,CAAC,CAAC;AA8BnD,IAAM,kBAAkB,MAAqC;AAClE,SAAO,WAAW,mBAAmB;AACvC;AAMO,IAAM,uBAAsD,CAAC;AAAA,EAClE;AAAA,EACA;AACF,MAAM;AACJ,QAAM,eAAe,CAAC,CAAC,oBAAoB,EAAE,UAAU,KAAK,CAAC;AAE7D,QAAM,YAAY,UAAU,OAAO,KAAK,MAAM,EAAE,SAAS;AACzD,QAAM,cAAc,gBAAgB;AAEpC,MAAI,aAAa,OAAO,KAAK,WAAW,EAAE,SAAS,GAAG;AACpD,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AAEA,QAAM,iBAAiB,YACrB,oBAAC,oBAAoB,UAApB,EAA6B,OAAO,QAClC,UACH,IAEA,gCAAG,UAAS;AAGd,MAAI,CAAC,QAAQ,QAAS,QAAO;AAE7B,MAAI,cAAc;AAChB,UAAM,IAAI;AAAA,MACR;AAAA,IACF;AAAA,EACF;AACA,SACE,oBAAC,4BAAyB,SAAS,OAAO,SACvC,0BACH;AAEJ;AAEA,qBAAqB,cAAc;","names":[]}

package/package.json

@@ -29,7 +29,7 @@
     "conversational-ui",
     "conversational-ai"
   ],
-  "version": "0.7.44",
+  "version": "0.7.45",
   "license": "MIT",
   "exports": {
     ".": {

package/src/runtimes/remote-thread-list/OptimisticState.ts

@@ -10,7 +10,11 @@ type Transform<TState, TResult> = {
   optimistic?: (state: TState) => TState;
 
   /** transform the state only while loading */
-  loading?: (state: TState) => TState;
+  loading?: (state: TState, task: Promise<TResult>) => TState;
+};
+
+type PendingTransform<TState, TResult> = Transform<TState, TResult> & {
+  task: Promise<TResult>;
 };
 
 const pipeTransforms = <TState, TExtra>(
@@ -24,7 +28,8 @@ const pipeTransforms = <TState, TExtra>(
 };
 
 export class OptimisticState<TState> extends BaseSubscribable {
-  private readonly _pendingTransforms: Array<Transform<TState, any>> = [];
+  private readonly _pendingTransforms: Array<PendingTransform<TState, any>> =
+    [];
   private _baseValue: TState;
   private _cachedValue: TState;
 
@@ -36,7 +41,7 @@ export class OptimisticState<TState> extends BaseSubscribable {
 
   private _updateState(): void {
     this._cachedValue = this._pendingTransforms.reduce((state, transform) => {
-      return pipeTransforms(state, undefined, [
+      return pipeTransforms(state, transform.task, [
         transform.loading,
         transform.optimistic,
       ]);
@@ -61,18 +66,20 @@ export class OptimisticState<TState> extends BaseSubscribable {
   public async optimisticUpdate<TResult>(
     transform: Transform<TState, TResult>,
   ): Promise<TResult> {
-    this._pendingTransforms.push(transform);
-    this._updateState();
-
+    const task = transform.execute();
+    const pendingTransform = { ...transform, task };
     try {
-      const result = await transform.execute();
+      this._pendingTransforms.push(pendingTransform);
+      this._updateState();
+
+      const result = await task;
       this._baseValue = pipeTransforms(this._baseValue, result, [
         transform.optimistic,
         transform.then,
       ]);
       return result;
     } finally {
-      const index = this._pendingTransforms.indexOf(transform);
+      const index = this._pendingTransforms.indexOf(pendingTransform);
       if (index > -1) {
         this._pendingTransforms.splice(index, 1);
       }

package/src/runtimes/remote-thread-list/RemoteThreadListHookInstanceManager.tsx

@@ -13,7 +13,10 @@ import {
 import { UseBoundStore, StoreApi, create } from "zustand";
 import { useAssistantRuntime } from "../../context";
 import { ThreadListItemRuntimeProvider } from "../../context/providers/ThreadListItemRuntimeProvider";
-import { useThreadListItem } from "../../context/react/ThreadListItemContext";
+import {
+  useThreadListItem,
+  useThreadListItemRuntime,
+} from "../../context/react/ThreadListItemContext";
 import { ThreadRuntimeCore, ThreadRuntimeImpl } from "../../internal";
 import { BaseSubscribable } from "./BaseSubscribable";
 import { AssistantRuntime } from "../../api";
@@ -112,6 +115,16 @@ export class RemoteThreadListHookInstanceManager extends BaseSubscribable {
       return threadBinding.outerSubscribe(updateRuntime);
     }, [threadBinding]);
 
+    // auto initialize thread
+    const threadListItemRuntime = useThreadListItemRuntime();
+    useEffect(() => {
+      return runtime.threads.main.unstable_on("initialize", () => {
+        if (threadListItemRuntime.getState().status === "new") {
+          threadListItemRuntime.initialize();
+        }
+      });
+    }, [runtime, threadListItemRuntime]);
+
     return null;
   };
 
@@ -121,7 +134,7 @@ export class RemoteThreadListHookInstanceManager extends BaseSubscribable {
   }> = memo(({ threadId, provider: Provider }) => {
     const assistantRuntime = useAssistantRuntime();
     const threadListItemRuntime = useMemo(
-      () => assistantRuntime.threadList.getItemById(threadId),
+      () => assistantRuntime.threads.getItemById(threadId),
       [assistantRuntime, threadId],
     );