@assistant-ui/react 0.7.43 → 0.7.44

package/src/api/AssistantRuntime.ts

@@ -1,5 +1,4 @@
 import { AssistantRuntimeCore } from "../runtimes/core/AssistantRuntimeCore";
-import { NestedSubscriptionSubject } from "./subscribable/NestedSubscriptionSubject";
 import { ModelConfigProvider } from "../types/ModelConfigTypes";
 import {
   ThreadListItemRuntimeBinding,
@@ -12,17 +11,24 @@ import { ThreadListRuntime, ThreadListRuntimeImpl } from "./ThreadListRuntime";
 
 export type AssistantRuntime = {
   /**
-   * The currently selected main thread.
+   * The threads in this assistant.
+   */
+  readonly threads: ThreadListRuntime;
+
+  /**
+   * The currently selected main thread. Equivalent to `threads.main`.
    */
   readonly thread: ThreadRuntime;
 
   /**
-   * The thread manager, to rename, archive and delete threads.
+   * @deprecated This field was renamed to `threads.main`.
    */
   readonly threadList: ThreadListRuntime;
 
   /**
    * Switch to a new thread.
+   *
+   * @deprecated This field was moved to `threads.switchToNewThread`.
    */
   switchToNewThread(): void;
 
@@ -30,6 +36,7 @@ export type AssistantRuntime = {
    * Switch to a thread.
    *
    * @param threadId The thread ID to switch to.
+   * @deprecated This field was moved to `threads.switchToThread`.
    */
   switchToThread(threadId: string): void;
 
@@ -42,7 +49,11 @@ export type AssistantRuntime = {
 };
 
 export class AssistantRuntimeImpl implements AssistantRuntime {
-  public readonly threadList;
+  public readonly threads;
+  public get threadList() {
+    return this.threads;
+  }
+
   public readonly _thread: ThreadRuntime;
 
   protected constructor(
@@ -52,18 +63,8 @@ export class AssistantRuntimeImpl implements AssistantRuntime {
       threadListItemBinding: ThreadListItemRuntimeBinding,
     ) => ThreadRuntime = ThreadRuntimeImpl,
   ) {
-    this.threadList = new ThreadListRuntimeImpl(_core.threadList);
-    this._thread = new runtimeFactory(
-      new NestedSubscriptionSubject({
-        path: {
-          ref: "threads.main",
-          threadSelector: { type: "main" },
-        },
-        getState: () => _core.threadList.getMainThreadRuntimeCore(),
-        subscribe: (callback) => _core.threadList.subscribe(callback),
-      }),
-      this.threadList.mainItem, // TODO capture "main" threadListItem from context around useLocalRuntime / useExternalStoreRuntime
-    );
+    this.threads = new ThreadListRuntimeImpl(_core.threads, runtimeFactory);
+    this._thread = this.threadList.main;
   }
 
   public get thread() {
@@ -71,11 +72,11 @@ export class AssistantRuntimeImpl implements AssistantRuntime {
   }
 
   public switchToNewThread() {
-    return this._core.threadList.switchToNewThread();
+    return this._core.threads.switchToNewThread();
   }
 
   public switchToThread(threadId: string) {
-    return this._core.threadList.switchToThread(threadId);
+    return this._core.threads.switchToThread(threadId);
   }
 
   public registerModelConfigProvider(provider: ModelConfigProvider) {

package/src/api/RuntimePathTypes.ts

@@ -9,7 +9,12 @@ export type ThreadListItemRuntimePath = {
 
 export type ThreadRuntimePath = {
   readonly ref: string;
-  readonly threadSelector: { type: "main" };
+  readonly threadSelector:
+    | { readonly type: "main" }
+    | {
+        readonly type: "threadId";
+        readonly threadId: string;
+      };
 };
 
 export type MessageRuntimePath = ThreadRuntimePath & {

package/src/api/ThreadListItemRuntime.ts

@@ -25,6 +25,7 @@ export type ThreadListItemRuntime = {
   readonly path: ThreadListItemRuntimePath;
   getState(): ThreadListItemState;
 
+  initialize(): Promise<{ remoteId: string; externalId: string | undefined }>;
   switchTo(): Promise<void>;
   rename(newTitle: string): Promise<void>;
   archive(): Promise<void>;
@@ -87,6 +88,14 @@ export class ThreadListItemRuntimeImpl implements ThreadListItemRuntime {
     return this._threadListBinding.delete(state.id);
   }
 
+  public initialize(): Promise<{
+    remoteId: string;
+    externalId: string | undefined;
+  }> {
+    const state = this._core.getState();
+    return this._threadListBinding.initialize(state.id);
+  }
+
   public unstable_on(event: ThreadListItemEventType, callback: () => void) {
     let prevIsMain = this._core.getState().isMain;
     return this.subscribe(() => {

package/src/api/ThreadListRuntime.ts

@@ -8,6 +8,13 @@ import {
 } from "./ThreadListItemRuntime";
 import { SKIP_UPDATE } from "./subscribable/SKIP_UPDATE";
 import { ShallowMemoizeSubject } from "./subscribable/ShallowMemoizeSubject";
+import {
+  ThreadListItemRuntimeBinding,
+  ThreadRuntime,
+  ThreadRuntimeCoreBinding,
+  ThreadRuntimeImpl,
+} from "./ThreadRuntime";
+import { NestedSubscriptionSubject } from "./subscribable/NestedSubscriptionSubject";
 
 export type ThreadListState = {
   readonly mainThreadId: string;
@@ -21,10 +28,16 @@ export type ThreadListRuntime = {
 
   subscribe(callback: () => void): Unsubscribe;
 
+  readonly main: ThreadRuntime;
+  getById(threadId: string): ThreadRuntime;
+
   readonly mainItem: ThreadListItemRuntime;
   getItemById(threadId: string): ThreadListItemRuntime;
   getItemByIndex(idx: number): ThreadListItemRuntime;
   getArchivedItemByIndex(idx: number): ThreadListItemRuntime;
+
+  switchToThread(threadId: string): Promise<void>;
+  switchToNewThread(): Promise<void>;
 };
 
 const getThreadListState = (
@@ -61,7 +74,13 @@ export type ThreadListRuntimeCoreBinding = ThreadListRuntimeCore;
 
 export class ThreadListRuntimeImpl implements ThreadListRuntime {
   private _getState;
-  constructor(private _core: ThreadListRuntimeCoreBinding) {
+  constructor(
+    private _core: ThreadListRuntimeCoreBinding,
+    private _runtimeFactory: new (
+      binding: ThreadRuntimeCoreBinding,
+      threadListItemBinding: ThreadListItemRuntimeBinding,
+    ) => ThreadRuntime = ThreadRuntimeImpl,
+  ) {
     const stateBinding = new LazyMemoizeSubject({
       path: {},
       getState: () => getThreadListState(_core),
@@ -83,6 +102,26 @@ export class ThreadListRuntimeImpl implements ThreadListRuntime {
       }),
       this._core,
     );
+
+    this.main = new _runtimeFactory(
+      new NestedSubscriptionSubject({
+        path: {
+          ref: "threads.main",
+          threadSelector: { type: "main" },
+        },
+        getState: () => _core.getMainThreadRuntimeCore(),
+        subscribe: (callback) => _core.subscribe(callback),
+      }),
+      this._mainThreadListItemRuntime, // TODO capture "main" threadListItem from context around useLocalRuntime / useExternalStoreRuntime
+    );
+  }
+
+  public switchToThread(threadId: string): Promise<void> {
+    return this._core.switchToThread(threadId);
+  }
+
+  public switchToNewThread(): Promise<void> {
+    return this._core.switchToNewThread();
   }
 
   public getState(): ThreadListState {
@@ -95,10 +134,26 @@ export class ThreadListRuntimeImpl implements ThreadListRuntime {
 
   private _mainThreadListItemRuntime;
 
+  public readonly main: ThreadRuntime;
+
   public get mainItem() {
     return this._mainThreadListItemRuntime;
   }
 
+  public getById(threadId: string) {
+    return new this._runtimeFactory(
+      new NestedSubscriptionSubject({
+        path: {
+          ref: "threads[threadId=" + JSON.stringify(threadId) + "]",
+          threadSelector: { type: "threadId", threadId },
+        },
+        getState: () => this._core.getThreadRuntimeCore(threadId),
+        subscribe: (callback) => this._core.subscribe(callback),
+      }),
+      this.mainItem, // TODO capture "main" threadListItem from context around useLocalRuntime / useExternalStoreRuntime
+    );
+  }
+
   public getItemByIndex(idx: number) {
     return new ThreadListItemRuntimeImpl(
       new ShallowMemoizeSubject({

package/src/runtimes/adapters/RuntimeAdapterProvider.tsx

@@ -0,0 +1,31 @@
+import { createContext, FC, useContext } from "react";
+import { ThreadHistoryAdapter } from "./thread-history/ThreadHistoryAdapter";
+
+export type RuntimeAdapters = {
+  history: ThreadHistoryAdapter;
+};
+
+const RuntimeAdaptersContext = createContext<RuntimeAdapters | null>(null);
+
+namespace RuntimeAdapterProvider {
+  export type Props = {
+    adapters: RuntimeAdapters;
+    children: React.ReactNode;
+  };
+}
+
+export const RuntimeAdapterProvider: FC<RuntimeAdapterProvider.Props> = ({
+  adapters,
+  children,
+}) => {
+  return (
+    <RuntimeAdaptersContext.Provider value={adapters}>
+      {children}
+    </RuntimeAdaptersContext.Provider>
+  );
+};
+
+export const useRuntimeAdapters = () => {
+  const adapters = useContext(RuntimeAdaptersContext);
+  return adapters;
+};

package/src/runtimes/{attachment → adapters/attachment}/AttachmentAdapter.ts

@@ -2,7 +2,7 @@ import {
   Attachment,
   PendingAttachment,
   CompleteAttachment,
-} from "../../types/AttachmentTypes";
+} from "../../../types/AttachmentTypes";
 
 export type AttachmentAdapter = {
   accept: string;

package/src/runtimes/{attachment → adapters/attachment}/CompositeAttachmentAdapter.ts

@@ -1,4 +1,4 @@
-import { Attachment, PendingAttachment } from "../../types/AttachmentTypes";
+import { Attachment, PendingAttachment } from "../../../types/AttachmentTypes";
 import { AttachmentAdapter } from "./AttachmentAdapter";
 
 function fileMatchesAccept(

package/src/runtimes/{attachment → adapters/attachment}/SimpleImageAttachmentAdapter.ts

@@ -1,7 +1,7 @@
 import {
   PendingAttachment,
   CompleteAttachment,
-} from "../../types/AttachmentTypes";
+} from "../../../types/AttachmentTypes";
 import { AttachmentAdapter } from "./AttachmentAdapter";
 
 export class SimpleImageAttachmentAdapter implements AttachmentAdapter {

package/src/runtimes/{attachment → adapters/attachment}/SimpleTextAttachmentAdapter.ts

@@ -1,7 +1,7 @@
 import {
   CompleteAttachment,
   PendingAttachment,
-} from "../../types/AttachmentTypes";
+} from "../../../types/AttachmentTypes";
 import { AttachmentAdapter } from "./AttachmentAdapter";
 
 export class SimpleTextAttachmentAdapter implements AttachmentAdapter {

package/src/runtimes/{feedback → adapters/feedback}/FeedbackAdapter.ts

@@ -1,4 +1,4 @@
-import { ThreadMessage } from "../../types/AssistantTypes";
+import { ThreadMessage } from "../../../types/AssistantTypes";
 
 type FeedbackAdapterFeedback = {
   message: ThreadMessage;

package/src/runtimes/adapters/index.ts

@@ -0,0 +1,3 @@
+export * from "./attachment";
+export * from "./feedback";
+export * from "./speech";

package/src/runtimes/{speech → adapters/speech}/SpeechAdapterTypes.ts

@@ -1,4 +1,4 @@
-import { Unsubscribe } from "../../types";
+import { Unsubscribe } from "../../../types";
 
 export namespace SpeechSynthesisAdapter {
   export type Status =

package/src/runtimes/adapters/thread-history/ThreadHistoryAdapter.ts

@@ -0,0 +1,9 @@
+import {
+  ExportedMessageRepository,
+  ExportedMessageRepositoryItem,
+} from "../../utils/MessageRepository";
+
+export type ThreadHistoryAdapter = {
+  load(): Promise<ExportedMessageRepository>;
+  append(item: ExportedMessageRepositoryItem): Promise<void>;
+};

package/src/runtimes/composer/BaseComposerRuntimeCore.tsx

@@ -4,7 +4,7 @@ import {
   PendingAttachment,
 } from "../../types/AttachmentTypes";
 import { AppendMessage } from "../../types";
-import { AttachmentAdapter } from "../attachment";
+import { AttachmentAdapter } from "../adapters/attachment";
 import {
   ComposerRuntimeCore,
   ComposerRuntimeEventType,

package/src/runtimes/composer/DefaultEditComposerRuntimeCore.tsx

@@ -1,6 +1,6 @@
 import { AppendMessage, ThreadMessage } from "../../types";
 import { getThreadMessageText } from "../../utils/getThreadMessageText";
-import { AttachmentAdapter } from "../attachment";
+import { AttachmentAdapter } from "../adapters/attachment";
 import { ThreadRuntimeCore } from "../core/ThreadRuntimeCore";
 import { BaseComposerRuntimeCore } from "./BaseComposerRuntimeCore";
 

package/src/runtimes/composer/DefaultThreadComposerRuntimeCore.tsx

@@ -1,5 +1,5 @@
 import { AppendMessage, PendingAttachment } from "../../types";
-import { AttachmentAdapter } from "../attachment";
+import { AttachmentAdapter } from "../adapters/attachment";
 import { ThreadComposerRuntimeCore } from "../core/ComposerRuntimeCore";
 import { ThreadRuntimeCore } from "../core/ThreadRuntimeCore";
 import { BaseComposerRuntimeCore } from "./BaseComposerRuntimeCore";

package/src/runtimes/core/AssistantRuntimeCore.tsx

@@ -4,7 +4,7 @@ import type { Unsubscribe } from "../../types/Unsubscribe";
 import { ThreadListRuntimeCore } from "./ThreadListRuntimeCore";
 
 export type AssistantRuntimeCore = {
-  readonly threadList: ThreadListRuntimeCore;
+  readonly threads: ThreadListRuntimeCore;
 
   registerModelConfigProvider: (provider: ModelConfigProvider) => Unsubscribe;
 

package/src/runtimes/core/BaseAssistantRuntimeCore.tsx

@@ -6,7 +6,7 @@ import { ThreadListRuntimeCore } from "./ThreadListRuntimeCore";
 
 export abstract class BaseAssistantRuntimeCore implements AssistantRuntimeCore {
   protected readonly _proxyConfigProvider = new ProxyConfigProvider();
-  public abstract get threadList(): ThreadListRuntimeCore;
+  public abstract get threads(): ThreadListRuntimeCore;
 
   constructor() {}
 

package/src/runtimes/core/BaseThreadRuntimeCore.tsx

@@ -20,9 +20,9 @@ import {
   StartRunConfig,
 } from "../core/ThreadRuntimeCore";
 import { DefaultEditComposerRuntimeCore } from "../composer/DefaultEditComposerRuntimeCore";
-import { SpeechSynthesisAdapter } from "../speech/SpeechAdapterTypes";
-import { FeedbackAdapter } from "../feedback/FeedbackAdapter";
-import { AttachmentAdapter } from "../attachment";
+import { SpeechSynthesisAdapter } from "../adapters/speech/SpeechAdapterTypes";
+import { FeedbackAdapter } from "../adapters/feedback/FeedbackAdapter";
+import { AttachmentAdapter } from "../adapters/attachment";
 import { getThreadMessageText } from "../../utils/getThreadMessageText";
 
 type BaseThreadAdapters = {

package/src/runtimes/core/ThreadListRuntimeCore.tsx

@@ -20,6 +20,8 @@ export type ThreadListRuntimeCore = {
   archivedThreadIds: readonly string[];
 
   getMainThreadRuntimeCore(): ThreadRuntimeCore;
+  getThreadRuntimeCore(threadId: string): ThreadRuntimeCore;
+
   getItemById(threadId: string): ThreadListItemCoreState | undefined;
 
   switchToThread(threadId: string): Promise<void>;
@@ -33,5 +35,9 @@ export type ThreadListRuntimeCore = {
   unarchive(threadId: string): Promise<void>;
   delete(threadId: string): Promise<void>;
 
+  initialize(
+    threadId: string,
+  ): Promise<{ remoteId: string; externalId: string | undefined }>;
+
   subscribe(callback: () => void): Unsubscribe;
 };

package/src/runtimes/core/ThreadRuntimeCore.tsx

@@ -1,7 +1,7 @@
 import { AppendMessage, ModelConfig, ThreadMessage } from "../../types";
 import { RunConfig } from "../../types/AssistantTypes";
 import type { Unsubscribe } from "../../types/Unsubscribe";
-import { SpeechSynthesisAdapter } from "../speech/SpeechAdapterTypes";
+import { SpeechSynthesisAdapter } from "../adapters/speech/SpeechAdapterTypes";
 import { ExportedMessageRepository } from "../utils/MessageRepository";
 import {
   ComposerRuntimeCore,
@@ -46,6 +46,7 @@ export type SubmittedFeedback = {
 
 export type ThreadRuntimeEventType =
   | "run-start"
+  | "run-end"
   | "initialize"
   | "model-config-update";
 

package/src/runtimes/external-store/ExternalStoreAdapter.tsx

@@ -1,12 +1,12 @@
 import { AppendMessage, ThreadMessage } from "../../types";
-import { AttachmentAdapter } from "../attachment";
+import { AttachmentAdapter } from "../adapters/attachment";
 import {
   AddToolResultOptions,
   StartRunConfig,
   ThreadSuggestion,
 } from "../core/ThreadRuntimeCore";
-import { FeedbackAdapter } from "../feedback/FeedbackAdapter";
-import { SpeechSynthesisAdapter } from "../speech/SpeechAdapterTypes";
+import { FeedbackAdapter } from "../adapters/feedback/FeedbackAdapter";
+import { SpeechSynthesisAdapter } from "../adapters/speech/SpeechAdapterTypes";
 import { ThreadMessageLike } from "./ThreadMessageLike";
 
 export type ExternalStoreThreadData<TState extends "regular" | "archived"> = {

package/src/runtimes/external-store/ExternalStoreRuntimeCore.tsx

@@ -10,11 +10,11 @@ const getThreadListAdapter = (store: ExternalStoreAdapter<any>) => {
 };
 
 export class ExternalStoreRuntimeCore extends BaseAssistantRuntimeCore {
-  public readonly threadList;
+  public readonly threads;
 
   constructor(adapter: ExternalStoreAdapter<any>) {
     super();
-    this.threadList = new ExternalStoreThreadListRuntimeCore(
+    this.threads = new ExternalStoreThreadListRuntimeCore(
       getThreadListAdapter(adapter),
       () =>
         new ExternalStoreThreadRuntimeCore(this._proxyConfigProvider, adapter),
@@ -23,7 +23,7 @@ export class ExternalStoreRuntimeCore extends BaseAssistantRuntimeCore {
 
   public setAdapter(adapter: ExternalStoreAdapter<any>) {
     // Update the thread list adapter and propagate store changes to the main thread
-    this.threadList.__internal_setAdapter(getThreadListAdapter(adapter));
-    this.threadList.getMainThreadRuntimeCore().__internal_setAdapter(adapter);
+    this.threads.__internal_setAdapter(getThreadListAdapter(adapter));
+    this.threads.getMainThreadRuntimeCore().__internal_setAdapter(adapter);
   }
 }

package/src/runtimes/external-store/ExternalStoreThreadListRuntimeCore.tsx

@@ -57,6 +57,10 @@ export class ExternalStoreThreadListRuntimeCore
     return this._mainThread;
   }
 
+  public getThreadRuntimeCore(): never {
+    throw new Error("Method not implemented.");
+  }
+
   public getItemById(threadId: string) {
     for (const thread of this.adapter.threads ?? []) {
       if (thread.threadId === threadId) return thread;
@@ -159,6 +163,10 @@ export class ExternalStoreThreadListRuntimeCore
     onDelete(threadId);
   }
 
+  public initialize(): never {
+    throw new Error("Method not implemented.");
+  }
+
   private _subscriptions = new Set<() => void>();
 
   public subscribe(callback: () => void): Unsubscribe {

package/src/runtimes/external-store/ExternalStoreThreadRuntimeCore.tsx

@@ -146,6 +146,14 @@ export class ExternalStoreThreadRuntimeCore
 
     if (messages.length > 0) this.ensureInitialized();
 
+    if (oldStore?.isRunning ?? false !== store.isRunning ?? false) {
+      if (store.isRunning) {
+        this._notifyEventSubscribers("run-start");
+      } else {
+        this._notifyEventSubscribers("run-end");
+      }
+    }
+
     for (let i = 0; i < messages.length; i++) {
       const message = messages[i]!;
       const parent = messages[i - 1];

package/src/runtimes/index.ts

@@ -1,9 +1,7 @@
-export * from "./attachment";
+export * from "./adapters";
 export * from "./core";
 export * from "./dangerous-in-browser";
 export * from "./edge";
 export * from "./external-store";
-export * from "./feedback";
 export * from "./local";
 export * from "./remote-thread-list";
-export * from "./speech";

package/src/runtimes/local/LocalRuntimeCore.tsx

@@ -25,7 +25,7 @@ const getExportFromInitialMessages = (
 };
 
 export class LocalRuntimeCore extends BaseAssistantRuntimeCore {
-  public readonly threadList;
+  public readonly threads;
   public readonly Provider = undefined;
 
   private _options: LocalRuntimeOptionsBase;
@@ -38,7 +38,7 @@ export class LocalRuntimeCore extends BaseAssistantRuntimeCore {
 
     this._options = options;
 
-    this.threadList = new LocalThreadListRuntimeCore(() => {
+    this.threads = new LocalThreadListRuntimeCore(() => {
       return new LocalThreadRuntimeCore(
         this._proxyConfigProvider,
         this._options,
@@ -46,7 +46,7 @@ export class LocalRuntimeCore extends BaseAssistantRuntimeCore {
     });
 
     if (initialMessages) {
-      this.threadList
+      this.threads
         .getMainThreadRuntimeCore()
         .import(getExportFromInitialMessages(initialMessages));
     }
@@ -57,10 +57,10 @@ export class LocalRuntimeCore extends BaseAssistantRuntimeCore {
   }: {
     initialMessages?: readonly CoreMessage[] | undefined;
   } = {}) {
-    this.threadList.switchToNewThread();
+    this.threads.switchToNewThread();
     if (!initialMessages) return;
 
-    this.threadList
+    this.threads
       .getMainThreadRuntimeCore()
       .import(getExportFromInitialMessages(initialMessages));
   }

package/src/runtimes/local/LocalRuntimeOptions.tsx

@@ -1,13 +1,15 @@
-import { AttachmentAdapter } from "../attachment/AttachmentAdapter";
+import { ThreadHistoryAdapter } from "../adapters/thread-history/ThreadHistoryAdapter";
+import { AttachmentAdapter } from "../adapters/attachment/AttachmentAdapter";
 import { ThreadMessageLike } from "../external-store";
-import { FeedbackAdapter } from "../feedback/FeedbackAdapter";
-import { SpeechSynthesisAdapter } from "../speech/SpeechAdapterTypes";
+import { FeedbackAdapter } from "../adapters/feedback/FeedbackAdapter";
+import { SpeechSynthesisAdapter } from "../adapters/speech/SpeechAdapterTypes";
 import { ChatModelAdapter } from "./ChatModelAdapter";
 
 export type LocalRuntimeOptionsBase = {
   maxSteps?: number | undefined;
   adapters: {
     chatModel: ChatModelAdapter;
+    history?: ThreadHistoryAdapter | undefined;
     attachments?: AttachmentAdapter | undefined;
     speech?: SpeechSynthesisAdapter | undefined;
     feedback?: FeedbackAdapter | undefined;

package/src/runtimes/local/LocalThreadListRuntimeCore.tsx

@@ -52,6 +52,12 @@ export class LocalThreadListRuntimeCore implements ThreadListRuntimeCore {
     return result;
   }
 
+  public getThreadRuntimeCore(threadId: string) {
+    const result = this._threadData.get(threadId)?.runtime;
+    if (!result) throw new Error("Thread not found.");
+    return result;
+  }
+
   public getLoadThreadsPromise(): Promise<void> {
     return RESOLVED_PROMISE;
   }
@@ -212,6 +218,10 @@ export class LocalThreadListRuntimeCore implements ThreadListRuntimeCore {
     return Promise.resolve();
   }
 
+  public initialize(): never {
+    throw new Error("Method not implemented.");
+  }
+
   private _subscriptions = new Set<() => void>();
 
   public subscribe(callback: () => void): Unsubscribe {

package/src/runtimes/local/LocalThreadRuntimeCore.tsx

@@ -127,9 +127,13 @@ export class LocalThreadRuntimeCore
 
     this._notifyEventSubscribers("run-start");
 
-    do {
-      message = await this.performRoundtrip(parentId, message, runConfig);
-    } while (shouldContinue(message));
+    try {
+      do {
+        message = await this.performRoundtrip(parentId, message, runConfig);
+      } while (shouldContinue(message));
+    } finally {
+      this._notifyEventSubscribers("run-end");
+    }
   }
 
   private async performRoundtrip(

package/src/runtimes/local/useLocalRuntime.tsx

@@ -43,7 +43,7 @@ export const useLocalRuntime = (
   const [runtime] = useState(() => new LocalRuntimeCore(opt, initialMessages));
 
   useEffect(() => {
-    runtime.threadList.getMainThreadRuntimeCore().__internal_setOptions(opt);
+    runtime.threads.getMainThreadRuntimeCore().__internal_setOptions(opt);
   }, [runtime, opt]);
 
   return useMemo(() => LocalRuntimeImpl.create(runtime), [runtime]);