@assistant-ui/react 0.11.34 → 0.11.35

package/package.json

@@ -28,7 +28,7 @@
     "conversational-ui",
     "conversational-ai"
   ],
-  "version": "0.11.34",
+  "version": "0.11.35",
   "license": "MIT",
   "type": "module",
   "exports": {
@@ -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.37",
+    "assistant-stream": "^0.2.38",
     "json-schema": "^0.4.0",
     "nanoid": "5.1.6",
     "react-textarea-autosize": "^8.5.9",

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/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/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"));