@assistant-ui/react-ai-sdk 0.3.0 → 0.4.0

package/dist/index.d.mts

@@ -1,14 +1,14 @@
 import * as _assistant_ui_react from '@assistant-ui/react';
-import { ChatModelAdapter, ChatModelRunOptions, AssistantContentPart, AppendMessage, ModelConfigProvider, ReactThreadRuntime, ThreadMessage, Unsubscribe, INTERNAL } from '@assistant-ui/react';
+import { ChatModelAdapter, ChatModelRunOptions, CoreAssistantContentPart, AppendMessage, ReactThreadRuntime, ThreadMessage, Unsubscribe, ModelConfigProvider, INTERNAL, AddToolResultOptions } from '@assistant-ui/react';
 import { LanguageModel, Message } from 'ai';
 import { ReactNode } from 'react';
-import { useChat, UseAssistantHelpers } from '@ai-sdk/react';
+import { useChat, useAssistant } from 'ai/react';
 
 declare class VercelModelAdapter implements ChatModelAdapter {
     private readonly model;
     constructor(model: LanguageModel);
     run({ messages, abortSignal, config, onUpdate }: ChatModelRunOptions): Promise<{
-        content: AssistantContentPart[];
+        content: CoreAssistantContentPart[];
     }>;
 }
 
@@ -31,19 +31,15 @@ type VercelRSCAdapterBase<T> = {
 };
 type VercelRSCAdapter<T = VercelRSCMessage> = VercelRSCAdapterBase<T> & (T extends VercelRSCMessage ? object : RSCMessageConverter<T>);
 
-declare const BaseAssistantRuntime$2: typeof INTERNAL.BaseAssistantRuntime;
-declare class VercelRSCRuntime<T extends WeakKey = VercelRSCMessage> extends BaseAssistantRuntime$2<VercelRSCThreadRuntime<T>> {
-    private readonly _proxyConfigProvider;
-    constructor(adapter: VercelRSCAdapter<T>);
-    set adapter(adapter: VercelRSCAdapter<T>);
-    onAdapterUpdated(): void;
-    getModelConfig(): _assistant_ui_react.ModelConfig;
-    registerModelConfigProvider(provider: ModelConfigProvider): () => void;
-    switchToThread(): void;
-}
 declare class VercelRSCThreadRuntime<T extends WeakKey = VercelRSCMessage> implements ReactThreadRuntime {
     adapter: VercelRSCAdapter<T>;
     private useAdapter;
+    readonly capabilities: Readonly<{
+        edit: false;
+        reload: false;
+        cancel: false;
+        copy: false;
+    }>;
     private _subscriptions;
     isRunning: boolean;
     messages: ThreadMessage[];
@@ -61,6 +57,17 @@ declare class VercelRSCThreadRuntime<T extends WeakKey = VercelRSCMessage> imple
     addToolResult(): void;
 }
 
+declare const BaseAssistantRuntime$2: typeof INTERNAL.BaseAssistantRuntime;
+declare class VercelRSCRuntime<T extends WeakKey = VercelRSCMessage> extends BaseAssistantRuntime$2<VercelRSCThreadRuntime<T>> {
+    private readonly _proxyConfigProvider;
+    constructor(adapter: VercelRSCAdapter<T>);
+    set adapter(adapter: VercelRSCAdapter<T>);
+    onAdapterUpdated(): void;
+    getModelConfig(): _assistant_ui_react.ModelConfig;
+    registerModelConfigProvider(provider: ModelConfigProvider): () => void;
+    switchToThread(): void;
+}
+
 declare const useVercelRSCRuntime: <T extends WeakKey>(adapter: VercelRSCAdapter<T>) => VercelRSCRuntime<T>;
 
 declare const getVercelRSCMessage: <T>(message: ThreadMessage) => T | undefined;
@@ -71,6 +78,12 @@ declare class VercelUseChatThreadRuntime implements ReactThreadRuntime {
     private repository;
     private assistantOptimisticId;
     private useVercel;
+    readonly capabilities: Readonly<{
+        edit: true;
+        reload: true;
+        cancel: true;
+        copy: true;
+    }>;
     messages: ThreadMessage[];
     isRunning: boolean;
     constructor(vercel: ReturnType<typeof useChat>);
@@ -84,7 +97,7 @@ declare class VercelUseChatThreadRuntime implements ReactThreadRuntime {
     onVercelUpdated(): void;
     private updateData;
     unstable_synchronizer: () => null;
-    addToolResult(toolCallId: string, result: any): void;
+    addToolResult({ toolCallId, result }: AddToolResultOptions): void;
 }
 
 declare const BaseAssistantRuntime$1: typeof INTERNAL.BaseAssistantRuntime;
@@ -101,12 +114,18 @@ declare class VercelUseChatRuntime extends BaseAssistantRuntime$1<VercelUseChatT
 declare const useVercelUseChatRuntime: (chatHelpers: ReturnType<typeof useChat>) => VercelUseChatRuntime;
 
 declare class VercelUseAssistantThreadRuntime implements ReactThreadRuntime {
-    vercel: UseAssistantHelpers;
+    vercel: ReturnType<typeof useAssistant>;
     private _subscriptions;
+    readonly capabilities: Readonly<{
+        edit: false;
+        reload: false;
+        cancel: false;
+        copy: true;
+    }>;
     private useVercel;
     messages: readonly ThreadMessage[];
     isRunning: boolean;
-    constructor(vercel: UseAssistantHelpers);
+    constructor(vercel: ReturnType<typeof useAssistant>);
     getBranches(): readonly string[];
     switchToBranch(): void;
     append(message: AppendMessage): Promise<void>;
@@ -122,15 +141,15 @@ declare class VercelUseAssistantThreadRuntime implements ReactThreadRuntime {
 declare const BaseAssistantRuntime: typeof INTERNAL.BaseAssistantRuntime;
 declare class VercelUseAssistantRuntime extends BaseAssistantRuntime<VercelUseAssistantThreadRuntime> {
     private readonly _proxyConfigProvider;
-    constructor(vercel: UseAssistantHelpers);
-    set vercel(vercel: UseAssistantHelpers);
+    constructor(vercel: ReturnType<typeof useAssistant>);
+    set vercel(vercel: ReturnType<typeof useAssistant>);
     onVercelUpdated(): void;
     getModelConfig(): _assistant_ui_react.ModelConfig;
     registerModelConfigProvider(provider: ModelConfigProvider): () => void;
     switchToThread(threadId: string | null): void;
 }
 
-declare const useVercelUseAssistantRuntime: (assistantHelpers: UseAssistantHelpers) => VercelUseAssistantRuntime;
+declare const useVercelUseAssistantRuntime: (assistantHelpers: ReturnType<typeof useAssistant>) => VercelUseAssistantRuntime;
 
 declare const getVercelAIMessage: (message: ThreadMessage) => Message[] | undefined;
 

package/dist/index.d.ts

@@ -1,14 +1,14 @@
 import * as _assistant_ui_react from '@assistant-ui/react';
-import { ChatModelAdapter, ChatModelRunOptions, AssistantContentPart, AppendMessage, ModelConfigProvider, ReactThreadRuntime, ThreadMessage, Unsubscribe, INTERNAL } from '@assistant-ui/react';
+import { ChatModelAdapter, ChatModelRunOptions, CoreAssistantContentPart, AppendMessage, ReactThreadRuntime, ThreadMessage, Unsubscribe, ModelConfigProvider, INTERNAL, AddToolResultOptions } from '@assistant-ui/react';
 import { LanguageModel, Message } from 'ai';
 import { ReactNode } from 'react';
-import { useChat, UseAssistantHelpers } from '@ai-sdk/react';
+import { useChat, useAssistant } from 'ai/react';
 
 declare class VercelModelAdapter implements ChatModelAdapter {
     private readonly model;
     constructor(model: LanguageModel);
     run({ messages, abortSignal, config, onUpdate }: ChatModelRunOptions): Promise<{
-        content: AssistantContentPart[];
+        content: CoreAssistantContentPart[];
     }>;
 }
 
@@ -31,19 +31,15 @@ type VercelRSCAdapterBase<T> = {
 };
 type VercelRSCAdapter<T = VercelRSCMessage> = VercelRSCAdapterBase<T> & (T extends VercelRSCMessage ? object : RSCMessageConverter<T>);
 
-declare const BaseAssistantRuntime$2: typeof INTERNAL.BaseAssistantRuntime;
-declare class VercelRSCRuntime<T extends WeakKey = VercelRSCMessage> extends BaseAssistantRuntime$2<VercelRSCThreadRuntime<T>> {
-    private readonly _proxyConfigProvider;
-    constructor(adapter: VercelRSCAdapter<T>);
-    set adapter(adapter: VercelRSCAdapter<T>);
-    onAdapterUpdated(): void;
-    getModelConfig(): _assistant_ui_react.ModelConfig;
-    registerModelConfigProvider(provider: ModelConfigProvider): () => void;
-    switchToThread(): void;
-}
 declare class VercelRSCThreadRuntime<T extends WeakKey = VercelRSCMessage> implements ReactThreadRuntime {
     adapter: VercelRSCAdapter<T>;
     private useAdapter;
+    readonly capabilities: Readonly<{
+        edit: false;
+        reload: false;
+        cancel: false;
+        copy: false;
+    }>;
     private _subscriptions;
     isRunning: boolean;
     messages: ThreadMessage[];
@@ -61,6 +57,17 @@ declare class VercelRSCThreadRuntime<T extends WeakKey = VercelRSCMessage> imple
     addToolResult(): void;
 }
 
+declare const BaseAssistantRuntime$2: typeof INTERNAL.BaseAssistantRuntime;
+declare class VercelRSCRuntime<T extends WeakKey = VercelRSCMessage> extends BaseAssistantRuntime$2<VercelRSCThreadRuntime<T>> {
+    private readonly _proxyConfigProvider;
+    constructor(adapter: VercelRSCAdapter<T>);
+    set adapter(adapter: VercelRSCAdapter<T>);
+    onAdapterUpdated(): void;
+    getModelConfig(): _assistant_ui_react.ModelConfig;
+    registerModelConfigProvider(provider: ModelConfigProvider): () => void;
+    switchToThread(): void;
+}
+
 declare const useVercelRSCRuntime: <T extends WeakKey>(adapter: VercelRSCAdapter<T>) => VercelRSCRuntime<T>;
 
 declare const getVercelRSCMessage: <T>(message: ThreadMessage) => T | undefined;
@@ -71,6 +78,12 @@ declare class VercelUseChatThreadRuntime implements ReactThreadRuntime {
     private repository;
     private assistantOptimisticId;
     private useVercel;
+    readonly capabilities: Readonly<{
+        edit: true;
+        reload: true;
+        cancel: true;
+        copy: true;
+    }>;
     messages: ThreadMessage[];
     isRunning: boolean;
     constructor(vercel: ReturnType<typeof useChat>);
@@ -84,7 +97,7 @@ declare class VercelUseChatThreadRuntime implements ReactThreadRuntime {
     onVercelUpdated(): void;
     private updateData;
     unstable_synchronizer: () => null;
-    addToolResult(toolCallId: string, result: any): void;
+    addToolResult({ toolCallId, result }: AddToolResultOptions): void;
 }
 
 declare const BaseAssistantRuntime$1: typeof INTERNAL.BaseAssistantRuntime;
@@ -101,12 +114,18 @@ declare class VercelUseChatRuntime extends BaseAssistantRuntime$1<VercelUseChatT
 declare const useVercelUseChatRuntime: (chatHelpers: ReturnType<typeof useChat>) => VercelUseChatRuntime;
 
 declare class VercelUseAssistantThreadRuntime implements ReactThreadRuntime {
-    vercel: UseAssistantHelpers;
+    vercel: ReturnType<typeof useAssistant>;
     private _subscriptions;
+    readonly capabilities: Readonly<{
+        edit: false;
+        reload: false;
+        cancel: false;
+        copy: true;
+    }>;
     private useVercel;
     messages: readonly ThreadMessage[];
     isRunning: boolean;
-    constructor(vercel: UseAssistantHelpers);
+    constructor(vercel: ReturnType<typeof useAssistant>);
     getBranches(): readonly string[];
     switchToBranch(): void;
     append(message: AppendMessage): Promise<void>;
@@ -122,15 +141,15 @@ declare class VercelUseAssistantThreadRuntime implements ReactThreadRuntime {
 declare const BaseAssistantRuntime: typeof INTERNAL.BaseAssistantRuntime;
 declare class VercelUseAssistantRuntime extends BaseAssistantRuntime<VercelUseAssistantThreadRuntime> {
     private readonly _proxyConfigProvider;
-    constructor(vercel: UseAssistantHelpers);
-    set vercel(vercel: UseAssistantHelpers);
+    constructor(vercel: ReturnType<typeof useAssistant>);
+    set vercel(vercel: ReturnType<typeof useAssistant>);
     onVercelUpdated(): void;
     getModelConfig(): _assistant_ui_react.ModelConfig;
     registerModelConfigProvider(provider: ModelConfigProvider): () => void;
     switchToThread(threadId: string | null): void;
 }
 
-declare const useVercelUseAssistantRuntime: (assistantHelpers: UseAssistantHelpers) => VercelUseAssistantRuntime;
+declare const useVercelUseAssistantRuntime: (assistantHelpers: ReturnType<typeof useAssistant>) => VercelUseAssistantRuntime;
 
 declare const getVercelAIMessage: (message: ThreadMessage) => Message[] | undefined;
 

package/dist/index.js

@@ -34,6 +34,9 @@ var import_ai = require("ai");
 
 // src/core/convertToCoreMessage.ts
 var convertToCoreMessage = (message) => {
+  if (message.role === "system") {
+    return [{ role: "system", content: message.content[0].text }];
+  }
   const expandedMessages = [
     {
       role: message.role,
@@ -92,7 +95,8 @@ var VercelModelAdapter = class {
     });
     const content = [];
     for await (const aiPart of fullStream) {
-      switch (aiPart.type) {
+      const partType = aiPart.type;
+      switch (partType) {
         case "text-delta": {
           let part = content.at(-1);
           if (!part || part.type !== "text") {
@@ -117,19 +121,26 @@ var VercelModelAdapter = class {
         }
         case "tool-result": {
           const toolCall = content.findIndex(
+            // @ts-expect-error
             (c) => c.type === "tool-call" && c.toolCallId === aiPart.toolCallId
           );
           if (toolCall === -1) {
             throw new Error(
+              // @ts-expect-error
               `Tool call ${aiPart.toolCallId} not found in the content stream. This is likely an internal bug in assistant-ui.`
             );
           }
           content[toolCall] = {
             ...content[toolCall],
+            // @ts-expect-error
             result: aiPart.result
           };
           break;
         }
+        default: {
+          const unhandledType = partType;
+          throw new Error(`Unknown content part type: ${unhandledType}`);
+        }
       }
       onUpdate({ content });
     }
@@ -138,9 +149,12 @@ var VercelModelAdapter = class {
 };
 
 // src/rsc/useVercelRSCRuntime.tsx
-var import_react3 = require("react");
+var import_react4 = require("react");
 
 // src/rsc/VercelRSCRuntime.tsx
+var import_react3 = require("@assistant-ui/react");
+
+// src/rsc/VercelRSCThreadRuntime.tsx
 var import_react2 = require("@assistant-ui/react");
 var import_zustand = require("zustand");
 
@@ -175,7 +189,7 @@ var vercelToThreadMessage = (converter, rawMessage) => {
     role: message.role,
     content: [{ type: "ui", display: message.display }],
     createdAt: message.createdAt ?? /* @__PURE__ */ new Date(),
-    ...{ status: "done" },
+    ...{ status: { type: "done" } },
     [symbolInnerRSCMessage]: rawMessage
   };
 };
@@ -193,30 +207,15 @@ var useVercelRSCSync = (adapter, updateData) => {
   }, [updateData, converter, convertCallback, adapter.messages]);
 };
 
-// src/rsc/VercelRSCRuntime.tsx
-var { ProxyConfigProvider, BaseAssistantRuntime } = import_react2.INTERNAL;
+// src/rsc/VercelRSCThreadRuntime.tsx
+var { ProxyConfigProvider } = import_react2.INTERNAL;
 var EMPTY_BRANCHES = Object.freeze([]);
-var VercelRSCRuntime = class extends BaseAssistantRuntime {
-  _proxyConfigProvider = new ProxyConfigProvider();
-  constructor(adapter) {
-    super(new VercelRSCThreadRuntime(adapter));
-  }
-  set adapter(adapter) {
-    this.thread.adapter = adapter;
-  }
-  onAdapterUpdated() {
-    return this.thread.onAdapterUpdated();
-  }
-  getModelConfig() {
-    return this._proxyConfigProvider.getModelConfig();
-  }
-  registerModelConfigProvider(provider) {
-    return this._proxyConfigProvider.registerModelConfigProvider(provider);
-  }
-  switchToThread() {
-    throw new Error("VercelRSCRuntime does not support switching threads");
-  }
-};
+var CAPABILITIES = Object.freeze({
+  edit: false,
+  reload: false,
+  cancel: false,
+  copy: false
+});
 var VercelRSCThreadRuntime = class {
   constructor(adapter) {
     this.adapter = adapter;
@@ -225,6 +224,7 @@ var VercelRSCThreadRuntime = class {
     }));
   }
   useAdapter;
+  capabilities = CAPABILITIES;
   _subscriptions = /* @__PURE__ */ new Set();
   isRunning = false;
   messages = [];
@@ -261,11 +261,7 @@ var VercelRSCThreadRuntime = class {
     await this.withRunning(this.adapter.reload(parentId));
   }
   cancelRun() {
-    if (process.env["NODE_ENV"] === "development") {
-      console.warn(
-        "Run cancellation is not supported by VercelRSCAssistantProvider."
-      );
-    }
+    throw new Error("VercelRSCRuntime does not support cancelling runs.");
   }
   subscribe(callback) {
     this._subscriptions.add(callback);
@@ -290,26 +286,50 @@ var VercelRSCThreadRuntime = class {
   }
 };
 
+// src/rsc/VercelRSCRuntime.tsx
+var { BaseAssistantRuntime } = import_react3.INTERNAL;
+var VercelRSCRuntime = class extends BaseAssistantRuntime {
+  _proxyConfigProvider = new ProxyConfigProvider();
+  constructor(adapter) {
+    super(new VercelRSCThreadRuntime(adapter));
+  }
+  set adapter(adapter) {
+    this.thread.adapter = adapter;
+  }
+  onAdapterUpdated() {
+    return this.thread.onAdapterUpdated();
+  }
+  getModelConfig() {
+    return this._proxyConfigProvider.getModelConfig();
+  }
+  registerModelConfigProvider(provider) {
+    return this._proxyConfigProvider.registerModelConfigProvider(provider);
+  }
+  switchToThread() {
+    throw new Error("VercelRSCRuntime does not support switching threads");
+  }
+};
+
 // src/rsc/useVercelRSCRuntime.tsx
 var useVercelRSCRuntime = (adapter) => {
-  const [runtime] = (0, import_react3.useState)(() => new VercelRSCRuntime(adapter));
-  (0, import_react3.useInsertionEffect)(() => {
+  const [runtime] = (0, import_react4.useState)(() => new VercelRSCRuntime(adapter));
+  (0, import_react4.useInsertionEffect)(() => {
     runtime.adapter = adapter;
   });
-  (0, import_react3.useEffect)(() => {
+  (0, import_react4.useEffect)(() => {
     runtime.onAdapterUpdated();
   });
   return runtime;
 };
 
 // src/ui/use-chat/useVercelUseChatRuntime.tsx
-var import_react9 = require("react");
+var import_react10 = require("react");
 
 // src/ui/use-chat/VercelUseChatRuntime.tsx
-var import_react8 = require("@assistant-ui/react");
+var import_react9 = require("@assistant-ui/react");
 
 // src/ui/use-chat/VercelUseChatThreadRuntime.tsx
-var import_react7 = require("@assistant-ui/react");
+var import_react8 = require("@assistant-ui/react");
 var import_zustand2 = require("zustand");
 
 // src/ui/getVercelAIMessage.tsx
@@ -333,11 +353,11 @@ var sliceMessagesUntil = (messages, messageId) => {
 };
 
 // src/ui/utils/useVercelAIComposerSync.tsx
-var import_react4 = require("@assistant-ui/react");
-var import_react5 = require("react");
+var import_react5 = require("@assistant-ui/react");
+var import_react6 = require("react");
 var useVercelAIComposerSync = (vercel) => {
-  const { useComposer } = (0, import_react4.useThreadContext)();
-  (0, import_react5.useEffect)(() => {
+  const { useComposer } = (0, import_react5.useThreadContext)();
+  (0, import_react6.useEffect)(() => {
     useComposer.setState({
       value: vercel.input,
       setValue: vercel.setInput
@@ -346,7 +366,7 @@ var useVercelAIComposerSync = (vercel) => {
 };
 
 // src/ui/utils/useVercelAIThreadSync.tsx
-var import_react6 = require("react");
+var import_react7 = require("react");
 var getIsRunning = (vercel) => {
   if ("isLoading" in vercel) return vercel.isLoading;
   return vercel.status === "in_progress";
@@ -371,6 +391,12 @@ var vercelToThreadMessage2 = (messages, status) => {
         role: "user",
         content: [{ type: "text", text: firstMessage.content }]
       };
+    case "system":
+      return {
+        ...common,
+        role: "system",
+        content: [{ type: "text", text: firstMessage.content }]
+      };
     case "data":
     case "assistant": {
       const res = {
@@ -406,8 +432,9 @@ var vercelToThreadMessage2 = (messages, status) => {
       return res;
     }
     default:
+      const _unsupported = firstMessage.role;
       throw new Error(
-        `123 You have a message with an unsupported role. The role ${firstMessage.role} is not supported.`
+        `You have a message with an unsupported role. The role ${_unsupported} is not supported.`
       );
   }
 };
@@ -440,12 +467,14 @@ var shallowArrayEqual = (a, b) => {
 };
 var useVercelAIThreadSync = (vercel, updateData) => {
   const isRunning = getIsRunning(vercel);
-  const converter = (0, import_react6.useMemo)(() => new ThreadMessageConverter(), []);
-  (0, import_react6.useEffect)(() => {
+  const converter = (0, import_react7.useMemo)(() => new ThreadMessageConverter(), []);
+  (0, import_react7.useEffect)(() => {
     const lastMessageId = vercel.messages.at(-1)?.id;
     const convertCallback = (messages2, cache) => {
-      const status = lastMessageId === messages2[0].id && isRunning ? "in_progress" : "done";
-      if (cache && shallowArrayEqual(cache.content, messages2) && (cache.role === "user" || cache.status === status))
+      const status = {
+        type: lastMessageId === messages2[0].id && isRunning ? "in_progress" : "done"
+      };
+      if (cache && shallowArrayEqual(cache.content, messages2) && (cache.role !== "assistant" || cache.status.type === status.type))
         return cache;
       return vercelToThreadMessage2(messages2, status);
     };
@@ -459,10 +488,16 @@ var useVercelAIThreadSync = (vercel, updateData) => {
 };
 
 // src/ui/use-chat/VercelUseChatThreadRuntime.tsx
-var { MessageRepository } = import_react7.INTERNAL;
+var { MessageRepository } = import_react8.INTERNAL;
 var hasUpcomingMessage = (isRunning, messages) => {
   return isRunning && messages[messages.length - 1]?.role !== "assistant";
 };
+var CAPABILITIES2 = Object.freeze({
+  edit: true,
+  reload: true,
+  cancel: true,
+  copy: true
+});
 var VercelUseChatThreadRuntime = class {
   constructor(vercel) {
     this.vercel = vercel;
@@ -474,6 +509,7 @@ var VercelUseChatThreadRuntime = class {
   repository = new MessageRepository();
   assistantOptimisticId = null;
   useVercel;
+  capabilities = CAPABILITIES2;
   messages = [];
   isRunning = false;
   getBranches(messageId) {
@@ -566,13 +602,13 @@ var VercelUseChatThreadRuntime = class {
     useVercelAIComposerSync(vercel);
     return null;
   };
-  addToolResult(toolCallId, result) {
+  addToolResult({ toolCallId, result }) {
     this.vercel.addToolResult({ toolCallId, result });
   }
 };
 
 // src/ui/use-chat/VercelUseChatRuntime.tsx
-var { ProxyConfigProvider: ProxyConfigProvider2, BaseAssistantRuntime: BaseAssistantRuntime2 } = import_react8.INTERNAL;
+var { ProxyConfigProvider: ProxyConfigProvider2, BaseAssistantRuntime: BaseAssistantRuntime2 } = import_react9.INTERNAL;
 var VercelUseChatRuntime = class extends BaseAssistantRuntime2 {
   _proxyConfigProvider = new ProxyConfigProvider2();
   constructor(vercel) {
@@ -606,25 +642,31 @@ var VercelUseChatRuntime = class extends BaseAssistantRuntime2 {
 
 // src/ui/use-chat/useVercelUseChatRuntime.tsx
 var useVercelUseChatRuntime = (chatHelpers) => {
-  const [runtime] = (0, import_react9.useState)(() => new VercelUseChatRuntime(chatHelpers));
-  (0, import_react9.useInsertionEffect)(() => {
+  const [runtime] = (0, import_react10.useState)(() => new VercelUseChatRuntime(chatHelpers));
+  (0, import_react10.useInsertionEffect)(() => {
     runtime.vercel = chatHelpers;
   });
-  (0, import_react9.useEffect)(() => {
+  (0, import_react10.useEffect)(() => {
     runtime.onVercelUpdated();
   });
   return runtime;
 };
 
 // src/ui/use-assistant/useVercelUseAssistantRuntime.tsx
-var import_react11 = require("react");
+var import_react12 = require("react");
 
 // src/ui/use-assistant/VercelUseAssistantRuntime.tsx
-var import_react10 = require("@assistant-ui/react");
+var import_react11 = require("@assistant-ui/react");
 
 // src/ui/use-assistant/VercelUseAssistantThreadRuntime.tsx
 var import_zustand3 = require("zustand");
 var EMPTY_BRANCHES2 = Object.freeze([]);
+var CAPABILITIES3 = Object.freeze({
+  edit: false,
+  reload: false,
+  cancel: false,
+  copy: true
+});
 var VercelUseAssistantThreadRuntime = class {
   constructor(vercel) {
     this.vercel = vercel;
@@ -633,6 +675,7 @@ var VercelUseAssistantThreadRuntime = class {
     }));
   }
   _subscriptions = /* @__PURE__ */ new Set();
+  capabilities = CAPABILITIES3;
   useVercel;
   messages = [];
   isRunning = false;
@@ -680,7 +723,7 @@ var VercelUseAssistantThreadRuntime = class {
       vm.push({
         id: "__optimistic__result",
         createdAt: /* @__PURE__ */ new Date(),
-        status: "in_progress",
+        status: { type: "in_progress" },
         role: "assistant",
         content: [{ type: "text", text: "" }]
       });
@@ -703,7 +746,7 @@ var VercelUseAssistantThreadRuntime = class {
 };
 
 // src/ui/use-assistant/VercelUseAssistantRuntime.tsx
-var { ProxyConfigProvider: ProxyConfigProvider3, BaseAssistantRuntime: BaseAssistantRuntime3 } = import_react10.INTERNAL;
+var { ProxyConfigProvider: ProxyConfigProvider3, BaseAssistantRuntime: BaseAssistantRuntime3 } = import_react11.INTERNAL;
 var hasUpcomingMessage2 = (isRunning, messages) => {
   return isRunning && messages[messages.length - 1]?.role !== "assistant";
 };
@@ -738,13 +781,13 @@ var VercelUseAssistantRuntime = class extends BaseAssistantRuntime3 {
 
 // src/ui/use-assistant/useVercelUseAssistantRuntime.tsx
 var useVercelUseAssistantRuntime = (assistantHelpers) => {
-  const [runtime] = (0, import_react11.useState)(
+  const [runtime] = (0, import_react12.useState)(
     () => new VercelUseAssistantRuntime(assistantHelpers)
   );
-  (0, import_react11.useInsertionEffect)(() => {
+  (0, import_react12.useInsertionEffect)(() => {
     runtime.vercel = assistantHelpers;
   });
-  (0, import_react11.useEffect)(() => {
+  (0, import_react12.useEffect)(() => {
     runtime.onVercelUpdated();
   });
   return runtime;