feat: add thread runtime threadId (#744)

Yonom · Sep 6, 2024 · fb8e58f · fb8e58f
1 parent d2b11a5
commit fb8e58f
Show file tree

Hide file tree

Showing 6 changed files with 23 additions and 2 deletions.
diff --git a/.changeset/mean-cameras-work.md b/.changeset/mean-cameras-work.md
@@ -0,0 +1,6 @@
+---
+"@assistant-ui/react-playground": patch
+"@assistant-ui/react": patch
+---
+
+feat: add thread runtime threadId
diff --git a/packages/react-playground/src/lib/playground-runtime.ts b/packages/react-playground/src/lib/playground-runtime.ts
@@ -97,6 +97,7 @@ export class PlaygroundThreadRuntime implements ReactThreadRuntime {
 
   public tools: Record<string, Tool<any, any>> = {};
 
+  public readonly threadId = generateId();
   public readonly isDisabled = false;
   public readonly capabilities = CAPABILITIES;
 

diff --git a/packages/react/src/runtimes/core/ThreadRuntime.tsx b/packages/react/src/runtimes/core/ThreadRuntime.tsx
@@ -7,6 +7,7 @@ export type ThreadRuntime = ThreadActionsState &
   Readonly<{
     composer: ThreadRuntime.Composer;
     capabilities: Readonly<RuntimeCapabilities>;
+    threadId: string;
     isDisabled: boolean;
     messages: readonly ThreadMessage[];
     subscribe: (callback: () => void) => Unsubscribe;

diff --git a/packages/react/src/runtimes/external-store/ExternalStoreRuntime.tsx b/packages/react/src/runtimes/external-store/ExternalStoreRuntime.tsx
@@ -10,6 +10,10 @@ export class ExternalStoreRuntime extends BaseAssistantRuntime<ExternalStoreThre
     super(new ExternalStoreThreadRuntime(store));
   }
 
+  public get store() {
+    return this.thread.store;
+  }
+
   public set store(store: ExternalStoreAdapter<any>) {
     this.thread.store = store;
   }

diff --git a/packages/react/src/runtimes/external-store/ExternalStoreThreadRuntime.tsx b/packages/react/src/runtimes/external-store/ExternalStoreThreadRuntime.tsx
@@ -12,6 +12,7 @@ import { getAutoStatus, isAutoStatus } from "./auto-status";
 import { fromThreadMessageLike } from "./ThreadMessageLike";
 import { RuntimeCapabilities } from "../../context/stores/Thread";
 import { getThreadMessageText } from "../../utils/getThreadMessageText";
+import { generateId } from "../../internal";
 
 export const hasUpcomingMessage = (
   isRunning: boolean,
@@ -38,8 +39,9 @@ export class ExternalStoreThreadRuntime implements ReactThreadRuntime {
     return this._capabilities;
   }
 
-  public messages: ThreadMessage[] = [];
-  public isDisabled = false;
+  public threadId!: string;
+  public messages!: ThreadMessage[];
+  public isDisabled!: boolean;
   public converter = new ThreadMessageConverter();
 
   private _store!: ExternalStoreAdapter<any>;
@@ -56,9 +58,14 @@ export class ExternalStoreThreadRuntime implements ReactThreadRuntime {
     this.store = store;
   }
 
+  public get store() {
+    return this._store;
+  }
+
   public set store(store: ExternalStoreAdapter<any>) {
     if (this._store === store) return;
 
+    this.threadId = store.threadId ?? this.threadId ?? generateId();
     const isRunning = store.isRunning ?? false;
     this.isDisabled = store.isDisabled ?? false;
 

diff --git a/packages/react/src/runtimes/local/LocalThreadRuntime.tsx b/packages/react/src/runtimes/local/LocalThreadRuntime.tsx
@@ -33,6 +33,7 @@ export class LocalThreadRuntime implements ThreadRuntime {
     speak: false,
   };
 
+  public readonly threadId: string;
   public readonly isDisabled = false;
 
   public get messages() {
@@ -52,6 +53,7 @@ export class LocalThreadRuntime implements ThreadRuntime {
     public adapter: ChatModelAdapter,
     { initialMessages, ...options }: LocalRuntimeOptions,
   ) {
+    this.threadId = generateId();
     this.options = options;
     if (initialMessages) {
       let parentId: string | null = null;