@assistant-ui/react 0.11.33 → 0.11.35

package/package.json

@@ -28,7 +28,7 @@
     "conversational-ui",
     "conversational-ai"
   ],
-  "version": "0.11.33",
+  "version": "0.11.35",
   "license": "MIT",
   "type": "module",
   "exports": {
@@ -48,8 +48,8 @@
   ],
   "sideEffects": false,
   "dependencies": {
-    "assistant-cloud": "^0.1.4",
-    "@assistant-ui/tap": "^0.1.3",
+    "assistant-cloud": "^0.1.5",
+    "@assistant-ui/tap": "^0.1.4",
     "@radix-ui/primitive": "^1.1.3",
     "@radix-ui/react-compose-refs": "^1.1.2",
     "@radix-ui/react-context": "^1.1.2",
@@ -59,7 +59,7 @@
     "@radix-ui/react-use-callback-ref": "^1.1.1",
     "@radix-ui/react-use-escape-keydown": "^1.1.1",
     "@standard-schema/spec": "^1.0.0",
-    "assistant-stream": "^0.2.36",
+    "assistant-stream": "^0.2.38",
     "json-schema": "^0.4.0",
     "nanoid": "5.1.6",
     "react-textarea-autosize": "^8.5.9",
@@ -84,11 +84,11 @@
     "@stryker-mutator/core": "^9.2.0",
     "@stryker-mutator/vitest-runner": "^9.2.0",
     "@types/json-schema": "^7.0.15",
-    "@types/node": "^24.8.1",
+    "@types/node": "^24.9.1",
     "eslint": "^9",
-    "eslint-config-next": "15.5.6",
+    "eslint-config-next": "16.0.0",
     "tsx": "^4.20.6",
-    "vitest": "^3.2.4",
+    "vitest": "^4.0.1",
     "@assistant-ui/x-buildutils": "0.0.1"
   },
   "publishConfig": {

package/src/augmentations.ts

@@ -11,6 +11,12 @@
  *         data: string;
  *       };
  *     }
+ *
+ *     interface ExternalState {
+ *       myCustomState: {
+ *         foo: string;
+ *       };
+ *     }
  *   }
  * }
  * ```
@@ -18,6 +24,12 @@
 export namespace Assistant {
   // eslint-disable-next-line @typescript-eslint/no-empty-object-type
   export interface Commands {}
+
+  // eslint-disable-next-line @typescript-eslint/no-empty-object-type
+  export interface ExternalState {}
 }
 
 export type UserCommands = Assistant.Commands[keyof Assistant.Commands];
+export type UserExternalState = keyof Assistant.ExternalState extends never
+  ? Record<string, unknown>
+  : Assistant.ExternalState[keyof Assistant.ExternalState];

package/src/client/ThreadMessageClient.tsx

@@ -106,10 +106,7 @@ export const ThreadMessageClient = resource(
         branchNumber,
         branchCount,
         speech: undefined,
-        submittedFeedback:
-          message.role === "assistant"
-            ? message.metadata.submittedFeedback
-            : undefined,
+        submittedFeedback: message.metadata.submittedFeedback,
         isCopied: isCopiedState,
         isHovering: isHoveringState,
       };

package/src/legacy-runtime/runtime/ThreadRuntime.ts

@@ -485,10 +485,7 @@ export class ThreadRuntimeImpl implements ThreadRuntime {
           const thread = this._threadBinding.getState();
 
           const branches = thread.getBranches(message.id);
-          const submittedFeedback =
-            message.role === "assistant"
-              ? message.metadata.submittedFeedback
-              : undefined;
+          const submittedFeedback = message.metadata.submittedFeedback;
 
           return {
             ...message,

package/src/legacy-runtime/runtime-cores/assistant-transport/index.ts

@@ -1,8 +1,10 @@
 export {
   useAssistantTransportRuntime,
   useAssistantTransportSendCommand,
+  useAssistantTransportState,
 } from "./useAssistantTransportRuntime";
 export type {
   AssistantTransportConnectionMetadata,
   AssistantTransportCommand,
+  AssistantTransportProtocol,
 } from "./types";

package/src/legacy-runtime/runtime-cores/assistant-transport/types.ts

@@ -79,10 +79,13 @@ export type QueuedCommand = AssistantTransportCommand;
 // Options types
 export type HeadersValue = Record<string, string> | Headers;
 
+export type AssistantTransportProtocol = "data-stream" | "assistant-transport";
+
 export type AssistantTransportOptions<T> = {
   initialState: T;
   api: string;
   resumeApi?: string;
+  protocol?: AssistantTransportProtocol;
   converter: AssistantTransportStateConverter<T>;
   headers: HeadersValue | (() => Promise<HeadersValue>);
   body?: object;

package/src/legacy-runtime/runtime-cores/assistant-transport/useAssistantTransportRuntime.tsx

@@ -13,6 +13,7 @@ import { useState, useRef, useMemo } from "react";
 import {
   AssistantMessageAccumulator,
   DataStreamDecoder,
+  AssistantTransportDecoder,
   unstable_createInitialMessage as createInitialMessage,
 } from "assistant-stream";
 import {
@@ -30,12 +31,14 @@ import { ToolExecutionStatus, useToolInvocations } from "./useToolInvocations";
 import { toAISDKTools, getEnabledTools, createRequestHeaders } from "./utils";
 import { useRemoteThreadListRuntime } from "../remote-thread-list/useRemoteThreadListRuntime";
 import { InMemoryThreadListAdapter } from "../remote-thread-list/adapter/in-memory";
-import { useAssistantApi } from "../../../context/react";
+import { useAssistantApi, useAssistantState } from "../../../context/react";
+import { UserExternalState } from "../../../augmentations";
 
 const symbolAssistantTransportExtras = Symbol("assistant-transport-extras");
 type AssistantTransportExtras = {
   [symbolAssistantTransportExtras]: true;
   sendCommand: (command: AssistantTransportCommand) => void;
+  state: UserExternalState;
 };
 
 const asAssistantTransportExtras = (
@@ -63,6 +66,18 @@ export const useAssistantTransportSendCommand = () => {
   };
 };
 
+export function useAssistantTransportState(): UserExternalState;
+export function useAssistantTransportState<T>(
+  selector: (state: UserExternalState) => T,
+): T;
+export function useAssistantTransportState<T>(
+  selector: (state: UserExternalState) => T = (t) => t as T,
+): T | UserExternalState {
+  return useAssistantState(({ thread }) =>
+    selector(asAssistantTransportExtras(thread.extras).state),
+  );
+}
+
 const useAssistantTransportThreadRuntime = <T,>(
   options: AssistantTransportOptions<T>,
 ): AssistantRuntime => {
@@ -114,21 +129,26 @@ const useAssistantTransportThreadRuntime = <T,>(
         throw new Error("Response body is null");
       }
 
+      // Select decoder based on protocol option
+      const protocol = options.protocol ?? "data-stream";
+      const decoder =
+        protocol === "assistant-transport"
+          ? new AssistantTransportDecoder()
+          : new DataStreamDecoder();
+
       let err: string | undefined;
-      const stream = response.body
-        .pipeThrough(new DataStreamDecoder())
-        .pipeThrough(
-          new AssistantMessageAccumulator({
-            initialMessage: createInitialMessage({
-              unstable_state:
-                (agentStateRef.current as ReadonlyJSONValue) ?? null,
-            }),
-            throttle: isResume,
-            onError: (error) => {
-              err = error;
-            },
+      const stream = response.body.pipeThrough(decoder).pipeThrough(
+        new AssistantMessageAccumulator({
+          initialMessage: createInitialMessage({
+            unstable_state:
+              (agentStateRef.current as ReadonlyJSONValue) ?? null,
           }),
-        );
+          throttle: isResume,
+          onError: (error) => {
+            err = error;
+          },
+        }),
+      );
 
       let markedDelivered = false;
 
@@ -206,6 +226,7 @@ const useAssistantTransportThreadRuntime = <T,>(
       sendCommand: (command: AssistantTransportCommand) => {
         commandQueue.enqueue(command);
       },
+      state: agentStateRef.current as UserExternalState,
     } satisfies AssistantTransportExtras,
     onNew: async (message: AppendMessage): Promise<void> => {
       if (message.role !== "user")

package/src/legacy-runtime/runtime-cores/core/BaseThreadRuntimeCore.tsx

@@ -126,14 +126,6 @@ export abstract class BaseThreadRuntimeCore implements ThreadRuntimeCore {
     return () => this._subscriptions.delete(callback);
   }
 
-  public getSubmittedFeedback(messageId: string) {
-    const messageData = this.repository.getMessage(messageId);
-    if (messageData.message.role === "assistant") {
-      return messageData.message.metadata.submittedFeedback;
-    }
-    return undefined;
-  }
-
   public submitFeedback({ messageId, type }: SubmitFeedbackOptions) {
     const adapter = this.adapters?.feedback;
     if (!adapter) throw new Error("Feedback adapter not configured");

package/src/legacy-runtime/runtime-cores/core/ThreadRuntimeCore.tsx

@@ -96,7 +96,6 @@ export type ThreadRuntimeCore = Readonly<{
   speak: (messageId: string) => void;
   stopSpeaking: () => void;
 
-  getSubmittedFeedback: (messageId: string) => SubmittedFeedback | undefined;
   submitFeedback: (feedback: SubmitFeedbackOptions) => void;
 
   getModelContext: () => ModelContext;

package/src/legacy-runtime/runtime-cores/external-store/external-message-converter.tsx

@@ -155,6 +155,11 @@ const joinExternalMessages = (
                   ...output.metadata.custom,
                 };
               }
+
+              if (output.metadata.submittedFeedback) {
+                assistantMessage.metadata.submittedFeedback =
+                  output.metadata.submittedFeedback;
+              }
             }
             // TODO keep this in sync
           }

package/src/legacy-runtime/runtime-cores/remote-thread-list/EMPTY_THREAD_CORE.tsx

@@ -48,10 +48,6 @@ export const EMPTY_THREAD_CORE: ThreadRuntimeCore = {
     throw EMPTY_THREAD_ERROR;
   },
 
-  getSubmittedFeedback() {
-    return undefined;
-  },
-
   submitFeedback() {
     throw EMPTY_THREAD_ERROR;
   },

package/src/tests/setup.ts

@@ -1,11 +1,5 @@
 // This file contains setup code for tests
 import { vi } from "vitest";
 
-// Set up global mocks if needed
-// Using a fixed date to avoid recursive calls
-const OriginalDate = global.Date;
-const fixedDate = new OriginalDate("2023-01-01");
-global.Date = vi.fn(() => fixedDate) as any;
-global.Date.now = vi.fn(() => fixedDate.getTime());
-
-// Add any other global setup needed for tests
+// Use Vitest's built-in to mock Date
+vi.setSystemTime(new Date("2023-01-01"));