@assistant-ui/react 0.7.36 → 0.7.38

package/src/api/ThreadRuntime.ts

@@ -36,11 +36,27 @@ import {
   ThreadRuntimePath,
 } from "./RuntimePathTypes";
 import { ThreadListItemState } from "./ThreadListItemRuntime";
+import { RunConfig } from "../types/AssistantTypes";
+
+export type CreateStartRunConfig = {
+  parentId: string | null;
+  sourceId?: string | null | undefined;
+  runConfig?: RunConfig | undefined;
+};
+
+const toStartRunConfig = (message: CreateStartRunConfig): StartRunConfig => {
+  return {
+    parentId: message.parentId ?? null,
+    sourceId: message.sourceId ?? null,
+    runConfig: message.runConfig ?? {},
+  };
+};
 
 export type CreateAppendMessage =
   | string
   | {
       parentId?: string | null | undefined;
+      sourceId?: string | null | undefined;
       role?: AppendMessage["role"] | undefined;
       content: AppendMessage["content"];
       attachments?: AppendMessage["attachments"] | undefined;
@@ -54,6 +70,8 @@ const toAppendMessage = (
   if (typeof message === "string") {
     return {
       parentId: messages.at(-1)?.id ?? null,
+      sourceId: null,
+      runConfig: {},
       role: "user",
       content: [{ type: "text", text: message }],
       attachments: [],
@@ -67,6 +85,7 @@ const toAppendMessage = (
   return {
     ...message,
     parentId: message.parentId ?? messages.at(-1)?.id ?? null,
+    sourceId: message.sourceId ?? null,
     role: message.role ?? "user",
     attachments: message.attachments ?? [],
   } as AppendMessage;
@@ -193,7 +212,11 @@ export type ThreadRuntime = {
    * @deprecated pass an object with `parentId` instead. This will be removed in 0.8.0.
    */
   startRun(parentId: string | null): void;
-  startRun(config: StartRunConfig): void;
+  /**
+   * Start a new run with the given configuration.
+   * @param config The configuration for starting the run
+   */
+  startRun(config: CreateStartRunConfig): void;
   subscribe(callback: () => void): Unsubscribe;
   cancelRun(): void;
   getModelConfig(): ModelConfig;
@@ -287,12 +310,12 @@ export class ThreadRuntimeImpl implements ThreadRuntime {
     return this._threadBinding.getState().getModelConfig();
   }
 
-  public startRun(configOrParentId: string | null | StartRunConfig) {
+  public startRun(configOrParentId: string | null | CreateStartRunConfig) {
     const config =
       configOrParentId === null || typeof configOrParentId === "string"
         ? { parentId: configOrParentId }
         : configOrParentId;
-    return this._threadBinding.getState().startRun(config);
+    return this._threadBinding.getState().startRun(toStartRunConfig(config));
   }
 
   public cancelRun() {

package/src/api/index.ts

@@ -1,5 +1,10 @@
 export type { AssistantRuntime } from "./AssistantRuntime";
-export type { ThreadRuntime, ThreadState } from "./ThreadRuntime";
+export type {
+  ThreadRuntime,
+  ThreadState,
+  CreateAppendMessage,
+  CreateStartRunConfig,
+} from "./ThreadRuntime";
 export type { MessageRuntime, MessageState } from "./MessageRuntime";
 export type {
   ContentPartRuntime,

package/src/internal.ts

@@ -14,5 +14,6 @@ export {
 } from "./api/ThreadRuntime";
 export { fromThreadMessageLike } from "./runtimes/external-store/ThreadMessageLike";
 export { getAutoStatus } from "./runtimes/external-store/auto-status";
+export { EdgeRuntimeRequestOptionsSchema } from "./runtimes/edge/EdgeRuntimeRequestOptions";
 
 export * from "./utils/smooth";

package/src/runtimes/composer/BaseComposerRuntimeCore.tsx

@@ -133,7 +133,7 @@ export abstract class BaseComposerRuntimeCore
           )
         : [];
 
-    const message: Omit<AppendMessage, "parentId"> = {
+    const message: Omit<AppendMessage, "parentId" | "sourceId"> = {
       role: this.role,
       content: this.text ? [{ type: "text", text: this.text }] : [],
       attachments,
@@ -149,7 +149,9 @@ export abstract class BaseComposerRuntimeCore
     this.handleCancel();
   }
 
-  protected abstract handleSend(message: Omit<AppendMessage, "parentId">): void;
+  protected abstract handleSend(
+    message: Omit<AppendMessage, "parentId" | "sourceId">,
+  ): void;
   protected abstract handleCancel(): void;
 
   async addAttachment(file: File) {

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 = {