@assistant-ui/react 0.5.16 → 0.5.18

package/dist/index.d.mts

@@ -186,10 +186,13 @@ type ChatModelRunOptions = {
     messages: ThreadMessage[];
     abortSignal: AbortSignal;
     config: ModelConfig;
+    /**
+     * @deprecated Declare the run function as an AsyncGenerator instead. This method will be removed in v0.6
+     */
     onUpdate: (result: ChatModelRunUpdate) => void;
 };
 type ChatModelAdapter = {
-    run: (options: ChatModelRunOptions) => Promise<ChatModelRunResult>;
+    run: (options: ChatModelRunOptions) => Promise<ChatModelRunResult> | AsyncGenerator<ChatModelRunResult, void>;
 };
 
 declare abstract class BaseAssistantRuntime<TThreadRuntime extends ReactThreadRuntime> implements AssistantRuntime {
@@ -296,7 +299,7 @@ type EdgeChatAdapterOptions = {
 declare class EdgeChatAdapter implements ChatModelAdapter {
     private options;
     constructor(options: EdgeChatAdapterOptions);
-    run({ messages, abortSignal, config, onUpdate }: ChatModelRunOptions): Promise<ChatModelRunResult>;
+    run({ messages, abortSignal, config }: ChatModelRunOptions): AsyncGenerator<ChatModelRunResult, void, unknown>;
 }
 
 type EdgeRuntimeOptions = EdgeChatAdapterOptions & LocalRuntimeOptions;
@@ -658,7 +661,7 @@ declare class ExternalStoreThreadRuntime implements ReactThreadRuntime {
         cancel: boolean;
         copy: boolean;
     };
-    messages: any[];
+    messages: ThreadMessage[];
     isDisabled: boolean;
     isRunning: boolean;
     constructor(store: ExternalStoreAdapter<any>);

package/dist/index.d.ts

@@ -186,10 +186,13 @@ type ChatModelRunOptions = {
     messages: ThreadMessage[];
     abortSignal: AbortSignal;
     config: ModelConfig;
+    /**
+     * @deprecated Declare the run function as an AsyncGenerator instead. This method will be removed in v0.6
+     */
     onUpdate: (result: ChatModelRunUpdate) => void;
 };
 type ChatModelAdapter = {
-    run: (options: ChatModelRunOptions) => Promise<ChatModelRunResult>;
+    run: (options: ChatModelRunOptions) => Promise<ChatModelRunResult> | AsyncGenerator<ChatModelRunResult, void>;
 };
 
 declare abstract class BaseAssistantRuntime<TThreadRuntime extends ReactThreadRuntime> implements AssistantRuntime {
@@ -296,7 +299,7 @@ type EdgeChatAdapterOptions = {
 declare class EdgeChatAdapter implements ChatModelAdapter {
     private options;
     constructor(options: EdgeChatAdapterOptions);
-    run({ messages, abortSignal, config, onUpdate }: ChatModelRunOptions): Promise<ChatModelRunResult>;
+    run({ messages, abortSignal, config }: ChatModelRunOptions): AsyncGenerator<ChatModelRunResult, void, unknown>;
 }
 
 type EdgeRuntimeOptions = EdgeChatAdapterOptions & LocalRuntimeOptions;
@@ -658,7 +661,7 @@ declare class ExternalStoreThreadRuntime implements ReactThreadRuntime {
         cancel: boolean;
         copy: boolean;
     };
-    messages: any[];
+    messages: ThreadMessage[];
     isDisabled: boolean;
     isRunning: boolean;
     constructor(store: ExternalStoreAdapter<any>);

package/dist/index.js

@@ -3428,7 +3428,7 @@ var EdgeChatAdapter = class {
   constructor(options) {
     this.options = options;
   }
-  async run({ messages, abortSignal, config, onUpdate }) {
+  async *run({ messages, abortSignal, config }) {
     const result = await fetch(this.options.api, {
       method: "POST",
       headers: {
@@ -3449,11 +3449,10 @@ var EdgeChatAdapter = class {
     const stream = result.body.pipeThrough(new TextDecoderStream()).pipeThrough(chunkByLineStream()).pipeThrough(assistantDecoderStream()).pipeThrough(toolResultStream(config.tools)).pipeThrough(runResultStream());
     let update;
     for await (update of asAsyncIterable(stream)) {
-      onUpdate(update);
+      yield update;
     }
     if (update === void 0)
       throw new Error("No data received from Edge Runtime");
-    return update;
   }
 };
 
@@ -3572,21 +3571,27 @@ var LocalThreadRuntime = class {
       });
     }
     try {
-      const result = await this.adapter.run({
+      const promiseOrGenerator = this.adapter.run({
         messages,
         abortSignal: this.abortController.signal,
         config: this.configProvider.getModelConfig(),
         onUpdate: updateMessage
       });
-      if (result.status?.type === "running")
-        throw new Error(
-          "Unexpected running status returned from ChatModelAdapter"
-        );
+      if (Symbol.asyncIterator in promiseOrGenerator) {
+        for await (const r of promiseOrGenerator) {
+          updateMessage(r);
+        }
+      } else {
+        updateMessage(await promiseOrGenerator);
+      }
       this.abortController = null;
-      updateMessage({
-        status: { type: "complete", reason: "unknown" },
-        ...result
-      });
+      if (message.status.type === "running") {
+        updateMessage({
+          status: { type: "complete", reason: "unknown" }
+        });
+      } else {
+        this.notifySubscribers();
+      }
     } catch (e) {
       this.abortController = null;
       if (e instanceof Error && e.name === "AbortError") {
@@ -3827,9 +3832,11 @@ var hasUpcomingMessage = (isRunning, messages) => {
 var ExternalStoreThreadRuntime = class {
   constructor(store) {
     this.store = store;
-    this.isDisabled = store.isDisabled ?? false;
-    this.isRunning = store.isRunning ?? false;
-    this.messages = store.messages;
+    this.updateData(
+      store.isDisabled ?? false,
+      store.isRunning ?? false,
+      store.messages
+    );
     this.useStore = (0, import_zustand14.create)(() => ({
       store
     }));
@@ -3847,9 +3854,9 @@ var ExternalStoreThreadRuntime = class {
       copy: this.store.onCopy !== null
     };
   }
-  messages;
-  isDisabled;
-  isRunning;
+  messages = [];
+  isDisabled = false;
+  isRunning = false;
   getBranches(messageId) {
     return this.repository.getBranches(messageId);
   }