@assistant-ui/react 0.7.36 → 0.7.37

package/src/runtimes/composer/DefaultEditComposerRuntimeCore.tsx

@@ -16,6 +16,7 @@ export class DefaultEditComposerRuntimeCore extends BaseComposerRuntimeCore {
   private _nonTextParts;
   private _previousText;
   private _parentId;
+  private _sourceId;
   constructor(
     private runtime: Omit<ThreadRuntimeCore, "composer"> & {
       adapters?: { attachments?: AttachmentAdapter | undefined } | undefined;
@@ -25,6 +26,7 @@ export class DefaultEditComposerRuntimeCore extends BaseComposerRuntimeCore {
   ) {
     super();
     this._parentId = parentId;
+    this._sourceId = message.id;
     this._previousText = getThreadMessageText(message);
     this.setText(this._previousText);
 
@@ -36,13 +38,16 @@ export class DefaultEditComposerRuntimeCore extends BaseComposerRuntimeCore {
     );
   }
 
-  public async handleSend(message: Omit<AppendMessage, "parentId">) {
+  public async handleSend(
+    message: Omit<AppendMessage, "parentId" | "sourceId">,
+  ) {
     const text = getThreadMessageText(message as AppendMessage);
     if (text !== this._previousText) {
       this.runtime.append({
-        ...(message as AppendMessage),
+        ...message,
         content: [...message.content, ...this._nonTextParts] as any,
         parentId: this._parentId,
+        sourceId: this._sourceId,
       });
     }
 

package/src/runtimes/composer/DefaultThreadComposerRuntimeCore.tsx

@@ -39,10 +39,13 @@ export class DefaultThreadComposerRuntimeCore
     });
   }
 
-  public async handleSend(message: Omit<AppendMessage, "parentId">) {
+  public async handleSend(
+    message: Omit<AppendMessage, "parentId" | "sourceId">,
+  ) {
     this.runtime.append({
       ...(message as AppendMessage),
       parentId: this.runtime.messages.at(-1)?.id ?? null,
+      sourceId: null,
     });
   }
 

package/src/runtimes/core/ThreadRuntimeCore.tsx

@@ -51,7 +51,8 @@ export type ThreadRuntimeEventType =
 
 export type StartRunConfig = {
   parentId: string | null;
-  runConfig?: RunConfig | undefined;
+  sourceId: string | null;
+  runConfig: RunConfig;
 };
 
 export type ThreadRuntimeCore = Readonly<{

package/src/runtimes/local/LocalThreadRuntimeCore.tsx

@@ -97,6 +97,7 @@ export class LocalThreadRuntimeCore
     if (startRun) {
       await this.startRun({
         parentId: newMessage.id,
+        sourceId: message.sourceId,
         runConfig: message.runConfig ?? {},
       });
     } else {

package/src/types/AssistantTypes.ts

@@ -198,10 +198,13 @@ export type RunConfig = {
 
 export type AppendMessage = CoreMessage & {
   parentId: string | null;
+
+  /** The ID of the message that was edited or undefined. */
+  sourceId: string | null;
+  runConfig: RunConfig | undefined;
+
   attachments: readonly CompleteAttachment[];
   startRun?: boolean | undefined;
-  /** TODO: make required in 0.8.0 */
-  runConfig?: RunConfig | undefined;
 };
 
 type BaseThreadMessage = {