@assistant-ui/react 0.8.11 → 0.8.12

package/src/api/ThreadRuntime.ts

@@ -5,6 +5,7 @@ import {
   SpeechState,
   ThreadRuntimeEventType,
   StartRunConfig,
+  ResumeRunConfig,
 } from "../runtimes/core/ThreadRuntimeCore";
 import { ExportedMessageRepository } from "../runtimes/utils/MessageRepository";
 import { AppendMessage, ThreadMessage, Unsubscribe } from "../types";
@@ -31,6 +32,7 @@ import { RunConfig } from "../types/AssistantTypes";
 import { EventSubscriptionSubject } from "./subscribable/EventSubscriptionSubject";
 import { symbolInnerMessage } from "../runtimes/external-store/getExternalStoreMessage";
 import { ModelContext } from "../model-context";
+import { ChatModelRunResult } from "../runtimes";
 
 export type CreateStartRunConfig = {
   parentId: string | null;
@@ -38,6 +40,19 @@ export type CreateStartRunConfig = {
   runConfig?: RunConfig | undefined;
 };
 
+export type CreateResumeRunConfig = CreateStartRunConfig & {
+  stream: AsyncGenerator<ChatModelRunResult, void, unknown>;
+};
+
+const toResumeRunConfig = (message: CreateResumeRunConfig): ResumeRunConfig => {
+  return {
+    parentId: message.parentId ?? null,
+    sourceId: message.sourceId ?? null,
+    runConfig: message.runConfig ?? {},
+    stream: message.stream,
+  };
+};
+
 const toStartRunConfig = (message: CreateStartRunConfig): StartRunConfig => {
   return {
     parentId: message.parentId ?? null,
@@ -211,6 +226,13 @@ export type ThreadRuntime = {
    * @param config The configuration for starting the run
    */
   startRun(config: CreateStartRunConfig): void;
+
+  /**
+   * Resume a run with the given configuration.
+   * @param config The configuration for resuming the run
+   **/
+  unstable_resumeRun(config: CreateResumeRunConfig): void;
+
   subscribe(callback: () => void): Unsubscribe;
   cancelRun(): void;
   getModelContext(): ModelContext;
@@ -338,6 +360,10 @@ export class ThreadRuntimeImpl implements ThreadRuntime {
     return this._threadBinding.getState().startRun(toStartRunConfig(config));
   }
 
+  public unstable_resumeRun(config: CreateResumeRunConfig) {
+    return this._threadBinding.getState().resumeRun(toResumeRunConfig(config));
+  }
+
   public cancelRun() {
     this._threadBinding.getState().cancelRun();
   }

package/src/runtimes/core/BaseThreadRuntimeCore.tsx

@@ -14,6 +14,7 @@ import {
   SubmittedFeedback,
   ThreadRuntimeEventType,
   StartRunConfig,
+  ResumeRunConfig,
 } from "../core/ThreadRuntimeCore";
 import { DefaultEditComposerRuntimeCore } from "../composer/DefaultEditComposerRuntimeCore";
 import { SpeechSynthesisAdapter } from "../adapters/speech/SpeechAdapterTypes";
@@ -41,6 +42,7 @@ export abstract class BaseThreadRuntimeCore implements ThreadRuntimeCore {
   public abstract get capabilities(): RuntimeCapabilities;
   public abstract append(message: AppendMessage): void;
   public abstract startRun(config: StartRunConfig): void;
+  public abstract resumeRun(config: ResumeRunConfig): void;
   public abstract addToolResult(options: AddToolResultOptions): void;
   public abstract cancelRun(): void;
 

package/src/runtimes/core/ThreadRuntimeCore.tsx

@@ -3,6 +3,7 @@ import { AppendMessage, ThreadMessage } from "../../types";
 import { RunConfig } from "../../types/AssistantTypes";
 import type { Unsubscribe } from "../../types/Unsubscribe";
 import { SpeechSynthesisAdapter } from "../adapters/speech/SpeechAdapterTypes";
+import { ChatModelRunResult } from "../local";
 import { ExportedMessageRepository } from "../utils/MessageRepository";
 import {
   ComposerRuntimeCore,
@@ -57,6 +58,10 @@ export type StartRunConfig = {
   runConfig: RunConfig;
 };
 
+export type ResumeRunConfig = StartRunConfig & {
+  stream: AsyncGenerator<ChatModelRunResult, void, unknown>;
+};
+
 export type ThreadRuntimeCore = Readonly<{
   getMessageById: (messageId: string) =>
     | {
@@ -70,6 +75,7 @@ export type ThreadRuntimeCore = Readonly<{
 
   append: (message: AppendMessage) => void;
   startRun: (config: StartRunConfig) => void;
+  resumeRun: (config: ResumeRunConfig) => void;
   cancelRun: () => void;
 
   addToolResult: (options: AddToolResultOptions) => void;

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

@@ -209,6 +209,10 @@ export class ExternalStoreThreadRuntimeCore
     await this._store.onReload(config.parentId, config);
   }
 
+  public async resumeRun(): Promise<void> {
+    throw new Error("Runtime does not support resuming runs.");
+  }
+
   public cancelRun(): void {
     if (!this._store.onCancel)
       throw new Error("Runtime does not support cancelling runs.");

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

@@ -41,6 +41,7 @@ export type ThreadMessageLike = {
             readonly toolName: string;
             readonly args?: ReadonlyJSONObject;
             readonly argsText?: string;
+            readonly artifact?: any;
             readonly result?: any | undefined;
             readonly isError?: boolean | undefined;
           }

package/src/runtimes/local/LocalThreadRuntimeCore.tsx

@@ -9,6 +9,7 @@ import {
   ThreadSuggestion,
   ThreadRuntimeCore,
   StartRunConfig,
+  ResumeRunConfig,
 } from "../core/ThreadRuntimeCore";
 import { BaseThreadRuntimeCore } from "../core/BaseThreadRuntimeCore";
 import { RunConfig } from "../../types/AssistantTypes";
@@ -139,6 +140,10 @@ export class LocalThreadRuntimeCore
     }
   }
 
+  public resumeRun({ stream, ...startConfig }: ResumeRunConfig): Promise<void> {
+    return this.startRun(startConfig, () => stream);
+  }
+
   public async startRun(
     { parentId, runConfig }: StartRunConfig,
     runCallback?: ChatModelAdapter["run"],

package/src/types/AssistantTypes.ts

@@ -99,6 +99,7 @@ export type ToolCallContentPart<
   TResult = unknown,
 > = CoreToolCallContentPart<TArgs, TResult> & {
   readonly argsText: string;
+  readonly artifact?: ReadonlyJSONValue;
 };
 
 export type ThreadUserContentPart =