@assistant-ui/react 0.5.56 → 0.5.58

package/dist/index.d.mts

@@ -603,7 +603,8 @@ type AssistantRuntime = ThreadRuntimeWithSubscribe & {
     /**
      * @deprecated Use `switchToNewThread` instead. This will be removed in 0.6.0.
      */
-    switchToThread: (threadId: string | null) => void;
+    switchToThread(threadId: null): void;
+    switchToThread(threadId: string): void;
     registerModelConfigProvider: (provider: ModelConfigProvider) => Unsubscribe;
 };
 
@@ -614,6 +615,7 @@ declare const AssistantRuntimeProvider: react.NamedExoticComponent<PropsWithChil
 
 type TextContentPartProviderProps = {
     text: string;
+    isRunning?: boolean | undefined;
 };
 declare const TextContentPartProvider: FC<PropsWithChildren<TextContentPartProviderProps>>;
 
@@ -621,7 +623,8 @@ type AssistantActionsState = Readonly<{
     /**
      * @deprecated Use `switchToNewThread` instead. This will be removed in 0.6.0.
      */
-    switchToThread: (threadId: string | null) => void;
+    switchToThread(threadId: null): void;
+    switchToThread(threadId: string): void;
     switchToNewThread: () => void;
     registerModelConfigProvider: (provider: ModelConfigProvider) => Unsubscribe;
 }>;
@@ -698,26 +701,30 @@ declare const useToolUIsStore: {
 };
 declare const useAssistantActions: {
     (): Readonly<{
-        switchToThread: (threadId: string | null) => void;
+        switchToThread(threadId: null): void;
+        switchToThread(threadId: string): void;
         switchToNewThread: () => void;
         registerModelConfigProvider: (provider: ModelConfigProvider) => Unsubscribe;
     }>;
     <TSelected>(selector: (state: Readonly<{
-        switchToThread: (threadId: string | null) => void;
+        switchToThread(threadId: null): void;
+        switchToThread(threadId: string): void;
         switchToNewThread: () => void;
         registerModelConfigProvider: (provider: ModelConfigProvider) => Unsubscribe;
     }>) => TSelected): TSelected;
     (options: {
         optional: true;
     }): Readonly<{
-        switchToThread: (threadId: string | null) => void;
+        switchToThread(threadId: null): void;
+        switchToThread(threadId: string): void;
         switchToNewThread: () => void;
         registerModelConfigProvider: (provider: ModelConfigProvider) => Unsubscribe;
     }> | null;
     <TSelected>(options: {
         optional: true;
         selector?: (state: Readonly<{
-            switchToThread: (threadId: string | null) => void;
+            switchToThread(threadId: null): void;
+            switchToThread(threadId: string): void;
             switchToNewThread: () => void;
             registerModelConfigProvider: (provider: ModelConfigProvider) => Unsubscribe;
         }>) => TSelected;
@@ -725,14 +732,16 @@ declare const useAssistantActions: {
 };
 declare const useAssistantActionsStore: {
     (): ReadonlyStore<Readonly<{
-        switchToThread: (threadId: string | null) => void;
+        switchToThread(threadId: null): void;
+        switchToThread(threadId: string): void;
         switchToNewThread: () => void;
         registerModelConfigProvider: (provider: ModelConfigProvider) => Unsubscribe;
     }>>;
     (options: {
         optional: true;
     }): ReadonlyStore<Readonly<{
-        switchToThread: (threadId: string | null) => void;
+        switchToThread(threadId: null): void;
+        switchToThread(threadId: string): void;
         switchToNewThread: () => void;
         registerModelConfigProvider: (provider: ModelConfigProvider) => Unsubscribe;
     }>> | null;

package/dist/index.d.ts

@@ -603,7 +603,8 @@ type AssistantRuntime = ThreadRuntimeWithSubscribe & {
     /**
      * @deprecated Use `switchToNewThread` instead. This will be removed in 0.6.0.
      */
-    switchToThread: (threadId: string | null) => void;
+    switchToThread(threadId: null): void;
+    switchToThread(threadId: string): void;
     registerModelConfigProvider: (provider: ModelConfigProvider) => Unsubscribe;
 };
 
@@ -614,6 +615,7 @@ declare const AssistantRuntimeProvider: react.NamedExoticComponent<PropsWithChil
 
 type TextContentPartProviderProps = {
     text: string;
+    isRunning?: boolean | undefined;
 };
 declare const TextContentPartProvider: FC<PropsWithChildren<TextContentPartProviderProps>>;
 
@@ -621,7 +623,8 @@ type AssistantActionsState = Readonly<{
     /**
      * @deprecated Use `switchToNewThread` instead. This will be removed in 0.6.0.
      */
-    switchToThread: (threadId: string | null) => void;
+    switchToThread(threadId: null): void;
+    switchToThread(threadId: string): void;
     switchToNewThread: () => void;
     registerModelConfigProvider: (provider: ModelConfigProvider) => Unsubscribe;
 }>;
@@ -698,26 +701,30 @@ declare const useToolUIsStore: {
 };
 declare const useAssistantActions: {
     (): Readonly<{
-        switchToThread: (threadId: string | null) => void;
+        switchToThread(threadId: null): void;
+        switchToThread(threadId: string): void;
         switchToNewThread: () => void;
         registerModelConfigProvider: (provider: ModelConfigProvider) => Unsubscribe;
     }>;
     <TSelected>(selector: (state: Readonly<{
-        switchToThread: (threadId: string | null) => void;
+        switchToThread(threadId: null): void;
+        switchToThread(threadId: string): void;
         switchToNewThread: () => void;
         registerModelConfigProvider: (provider: ModelConfigProvider) => Unsubscribe;
     }>) => TSelected): TSelected;
     (options: {
         optional: true;
     }): Readonly<{
-        switchToThread: (threadId: string | null) => void;
+        switchToThread(threadId: null): void;
+        switchToThread(threadId: string): void;
         switchToNewThread: () => void;
         registerModelConfigProvider: (provider: ModelConfigProvider) => Unsubscribe;
     }> | null;
     <TSelected>(options: {
         optional: true;
         selector?: (state: Readonly<{
-            switchToThread: (threadId: string | null) => void;
+            switchToThread(threadId: null): void;
+            switchToThread(threadId: string): void;
             switchToNewThread: () => void;
             registerModelConfigProvider: (provider: ModelConfigProvider) => Unsubscribe;
         }>) => TSelected;
@@ -725,14 +732,16 @@ declare const useAssistantActions: {
 };
 declare const useAssistantActionsStore: {
     (): ReadonlyStore<Readonly<{
-        switchToThread: (threadId: string | null) => void;
+        switchToThread(threadId: null): void;
+        switchToThread(threadId: string): void;
         switchToNewThread: () => void;
         registerModelConfigProvider: (provider: ModelConfigProvider) => Unsubscribe;
     }>>;
     (options: {
         optional: true;
     }): ReadonlyStore<Readonly<{
-        switchToThread: (threadId: string | null) => void;
+        switchToThread(threadId: null): void;
+        switchToThread(threadId: string): void;
         switchToNewThread: () => void;
         registerModelConfigProvider: (provider: ModelConfigProvider) => Unsubscribe;
     }>> | null;

package/dist/index.js

@@ -1839,8 +1839,8 @@ var ExternalStoreRuntime = (_class9 = class extends BaseAssistantRuntime {
     if (!this.store.onSwitchToNewThread)
       throw new Error("Runtime does not support switching to new threads.");
     this.thread = new ExternalStoreThreadRuntime({
-      messages: [],
-      onNew: this.store.onNew
+      ...this.store,
+      messages: []
     });
     await this.store.onSwitchToNewThread();
   }
@@ -1849,8 +1849,9 @@ var ExternalStoreRuntime = (_class9 = class extends BaseAssistantRuntime {
       if (!this.store.onSwitchToThread)
         throw new Error("Runtime does not support switching threads.");
       this.thread = new ExternalStoreThreadRuntime({
-        messages: [],
-        onNew: this.store.onNew
+        ...this.store,
+        messages: []
+        // ignore messages until rerender
       });
       this.store.onSwitchToThread(threadId);
     } else {
@@ -2347,16 +2348,37 @@ var AssistantRuntimeProvider = _react.memo.call(void 0, AssistantRuntimeProvider
 
 
 
-var TextContentPartProvider = ({ children, text }) => {
+var COMPLETE_STATUS = {
+  type: "complete"
+};
+var RUNNING_STATUS = {
+  type: "running"
+};
+var TextContentPartProvider = ({ children, text, isRunning }) => {
   const [context] = _react.useState.call(void 0, () => {
     const useContentPart2 = _zustand.create.call(void 0, () => ({
-      status: { type: "complete" },
+      status: isRunning ? RUNNING_STATUS : COMPLETE_STATUS,
       part: { type: "text", text }
     }));
     return {
       useContentPart: useContentPart2
     };
   });
+  _react.useEffect.call(void 0, () => {
+    const state = context.useContentPart.getState();
+    const textUpdated = state.part.text !== text;
+    const targetTextPart = textUpdated ? { type: "text", text } : state.part;
+    const targetStatus = isRunning ? RUNNING_STATUS : COMPLETE_STATUS;
+    const statusUpdated = state.status !== targetStatus;
+    if (!textUpdated && !statusUpdated) return;
+    writableStore(context.useContentPart).setState(
+      {
+        part: targetTextPart,
+        status: targetStatus
+      },
+      true
+    );
+  }, [context, isRunning, text]);
   return /* @__PURE__ */ _jsxruntime.jsx.call(void 0, ContentPartContext.Provider, { value: context, children });
 };
 
@@ -3283,21 +3305,21 @@ MessagePrimitiveIf.displayName = "MessagePrimitive.If";
 
 
 
-var COMPLETE_STATUS = {
+var COMPLETE_STATUS2 = {
   type: "complete"
 };
 var toContentPartStatus = (message, partIndex, part) => {
-  if (message.role !== "assistant") return COMPLETE_STATUS;
+  if (message.role !== "assistant") return COMPLETE_STATUS2;
   const isLastPart = partIndex === Math.max(0, message.content.length - 1);
   if (part.type !== "tool-call") {
     if ("reason" in message.status && message.status.reason === "tool-calls" && isLastPart)
       throw new Error(
         "Encountered unexpected requires-action status. This is likely an internal bug in assistant-ui."
       );
-    return isLastPart ? message.status : COMPLETE_STATUS;
+    return isLastPart ? message.status : COMPLETE_STATUS2;
   }
   if (!!part.result) {
-    return COMPLETE_STATUS;
+    return COMPLETE_STATUS2;
   }
   return message.status;
 };