@assistant-ui/react 0.5.1 → 0.5.2

package/dist/index.d.mts

@@ -204,17 +204,33 @@ declare abstract class BaseAssistantRuntime<TThreadRuntime extends ReactThreadRu
     private subscriptionHandler;
 }
 
+type TextContentPartProps = {
+    part: TextContentPart;
+    status: ContentPartStatus;
+};
+type TextContentPartComponent = ComponentType<TextContentPartProps>;
+type ImageContentPartProps = {
+    part: ImageContentPart;
+    status: ContentPartStatus;
+};
+type ImageContentPartComponent = ComponentType<ImageContentPartProps>;
+type UIContentPartProps = {
+    part: UIContentPart;
+    status: ContentPartStatus;
+};
+type UIContentPartComponent = ComponentType<UIContentPartProps>;
+type ToolCallContentPartProps<TArgs extends Record<string, unknown> = any, TResult = unknown> = {
+    part: ToolCallContentPart<TArgs, TResult>;
+    status: ToolContentPartStatus;
+    addResult: (result: any) => void;
+};
+type ToolCallContentPartComponent<TArgs extends Record<string, unknown> = any, TResult = any> = ComponentType<ToolCallContentPartProps<TArgs, TResult>>;
+
 type LocalRuntimeOptions = {
     initialMessages?: readonly CoreMessage[] | undefined;
     maxToolRoundtrips?: number;
 };
-declare class LocalRuntime extends BaseAssistantRuntime<LocalThreadRuntime> {
-    private readonly _proxyConfigProvider;
-    constructor(adapter: ChatModelAdapter, options?: LocalRuntimeOptions);
-    set adapter(adapter: ChatModelAdapter);
-    registerModelConfigProvider(provider: ModelConfigProvider): () => void;
-    switchToThread(threadId: string | null): LocalThreadRuntime;
-}
+
 declare class LocalThreadRuntime implements ThreadRuntime {
     private configProvider;
     adapter: ChatModelAdapter;
@@ -242,32 +258,18 @@ declare class LocalThreadRuntime implements ThreadRuntime {
     addToolResult({ messageId, toolCallId, result }: AddToolResultOptions): void;
 }
 
+declare class LocalRuntime extends BaseAssistantRuntime<LocalThreadRuntime> {
+    private readonly _proxyConfigProvider;
+    constructor(adapter: ChatModelAdapter, options?: LocalRuntimeOptions);
+    set adapter(adapter: ChatModelAdapter);
+    registerModelConfigProvider(provider: ModelConfigProvider): () => void;
+    switchToThread(threadId: string | null): LocalThreadRuntime;
+}
+
 declare const useLocalRuntime: (adapter: ChatModelAdapter, options?: LocalRuntimeOptions) => LocalRuntime;
 
 declare function toLanguageModelMessages(message: readonly CoreMessage[] | readonly ThreadMessage[]): LanguageModelV1Message[];
 
-type TextContentPartProps = {
-    part: TextContentPart;
-    status: ContentPartStatus;
-};
-type TextContentPartComponent = ComponentType<TextContentPartProps>;
-type ImageContentPartProps = {
-    part: ImageContentPart;
-    status: ContentPartStatus;
-};
-type ImageContentPartComponent = ComponentType<ImageContentPartProps>;
-type UIContentPartProps = {
-    part: UIContentPart;
-    status: ContentPartStatus;
-};
-type UIContentPartComponent = ComponentType<UIContentPartProps>;
-type ToolCallContentPartProps<TArgs extends Record<string, unknown> = any, TResult = unknown> = {
-    part: ToolCallContentPart<TArgs, TResult>;
-    status: ToolContentPartStatus;
-    addResult: (result: any) => void;
-};
-type ToolCallContentPartComponent<TArgs extends Record<string, unknown> = any, TResult = any> = ComponentType<ToolCallContentPartProps<TArgs, TResult>>;
-
 type fromLanguageModelMessagesOptions = {
     mergeRoundtrips: boolean;
 };

package/dist/index.d.ts

@@ -204,17 +204,33 @@ declare abstract class BaseAssistantRuntime<TThreadRuntime extends ReactThreadRu
     private subscriptionHandler;
 }
 
+type TextContentPartProps = {
+    part: TextContentPart;
+    status: ContentPartStatus;
+};
+type TextContentPartComponent = ComponentType<TextContentPartProps>;
+type ImageContentPartProps = {
+    part: ImageContentPart;
+    status: ContentPartStatus;
+};
+type ImageContentPartComponent = ComponentType<ImageContentPartProps>;
+type UIContentPartProps = {
+    part: UIContentPart;
+    status: ContentPartStatus;
+};
+type UIContentPartComponent = ComponentType<UIContentPartProps>;
+type ToolCallContentPartProps<TArgs extends Record<string, unknown> = any, TResult = unknown> = {
+    part: ToolCallContentPart<TArgs, TResult>;
+    status: ToolContentPartStatus;
+    addResult: (result: any) => void;
+};
+type ToolCallContentPartComponent<TArgs extends Record<string, unknown> = any, TResult = any> = ComponentType<ToolCallContentPartProps<TArgs, TResult>>;
+
 type LocalRuntimeOptions = {
     initialMessages?: readonly CoreMessage[] | undefined;
     maxToolRoundtrips?: number;
 };
-declare class LocalRuntime extends BaseAssistantRuntime<LocalThreadRuntime> {
-    private readonly _proxyConfigProvider;
-    constructor(adapter: ChatModelAdapter, options?: LocalRuntimeOptions);
-    set adapter(adapter: ChatModelAdapter);
-    registerModelConfigProvider(provider: ModelConfigProvider): () => void;
-    switchToThread(threadId: string | null): LocalThreadRuntime;
-}
+
 declare class LocalThreadRuntime implements ThreadRuntime {
     private configProvider;
     adapter: ChatModelAdapter;
@@ -242,32 +258,18 @@ declare class LocalThreadRuntime implements ThreadRuntime {
     addToolResult({ messageId, toolCallId, result }: AddToolResultOptions): void;
 }
 
+declare class LocalRuntime extends BaseAssistantRuntime<LocalThreadRuntime> {
+    private readonly _proxyConfigProvider;
+    constructor(adapter: ChatModelAdapter, options?: LocalRuntimeOptions);
+    set adapter(adapter: ChatModelAdapter);
+    registerModelConfigProvider(provider: ModelConfigProvider): () => void;
+    switchToThread(threadId: string | null): LocalThreadRuntime;
+}
+
 declare const useLocalRuntime: (adapter: ChatModelAdapter, options?: LocalRuntimeOptions) => LocalRuntime;
 
 declare function toLanguageModelMessages(message: readonly CoreMessage[] | readonly ThreadMessage[]): LanguageModelV1Message[];
 
-type TextContentPartProps = {
-    part: TextContentPart;
-    status: ContentPartStatus;
-};
-type TextContentPartComponent = ComponentType<TextContentPartProps>;
-type ImageContentPartProps = {
-    part: ImageContentPart;
-    status: ContentPartStatus;
-};
-type ImageContentPartComponent = ComponentType<ImageContentPartProps>;
-type UIContentPartProps = {
-    part: UIContentPart;
-    status: ContentPartStatus;
-};
-type UIContentPartComponent = ComponentType<UIContentPartProps>;
-type ToolCallContentPartProps<TArgs extends Record<string, unknown> = any, TResult = unknown> = {
-    part: ToolCallContentPart<TArgs, TResult>;
-    status: ToolContentPartStatus;
-    addResult: (result: any) => void;
-};
-type ToolCallContentPartComponent<TArgs extends Record<string, unknown> = any, TResult = any> = ComponentType<ToolCallContentPartProps<TArgs, TResult>>;
-
 type fromLanguageModelMessagesOptions = {
     mergeRoundtrips: boolean;
 };

package/dist/index.js

@@ -1267,7 +1267,7 @@ var COMPLETE_STATUS = {
 };
 var toContentPartStatus = (message, partIndex, part) => {
   if (message.role !== "assistant") return COMPLETE_STATUS;
-  const isLastPart = partIndex === message.content.length - 1;
+  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(
@@ -1280,9 +1280,16 @@ var toContentPartStatus = (message, partIndex, part) => {
   }
   return message.status;
 };
+var EMPTY_CONTENT = Object.freeze({ type: "text", text: "" });
 var syncContentPart = ({ message }, useContentPart, partIndex) => {
-  const part = message.content[partIndex];
-  if (!part) return;
+  let part = message.content[partIndex];
+  if (!part) {
+    if (message.content.length === 0 && partIndex === 0) {
+      part = EMPTY_CONTENT;
+    } else {
+      return;
+    }
+  }
   const status = toContentPartStatus(message, partIndex, part);
   const currentState = useContentPart.getState();
   if (currentState.part === part && currentState.status === status) return;
@@ -1408,7 +1415,7 @@ var ContentPartPrimitiveText = (0, import_react36.forwardRef)(({ smooth = true,
     part: { text }
   } = useContentPartText();
   const smoothText = useSmooth(text, smooth);
-  return /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(import_react_primitive4.Primitive.span, { "data-status": status, ...rest, ref: forwardedRef, children: smoothText });
+  return /* @__PURE__ */ (0, import_jsx_runtime14.jsx)(import_react_primitive4.Primitive.span, { "data-status": status.type, ...rest, ref: forwardedRef, children: smoothText });
 });
 ContentPartPrimitiveText.displayName = "ContentPartPrimitive.Text";
 
@@ -1514,7 +1521,7 @@ var MessagePrimitiveContent = ({
   components
 }) => {
   const { useMessage } = useMessageContext();
-  const contentLength = useMessage((s) => s.message.content.length);
+  const contentLength = useMessage((s) => s.message.content.length) || 1;
   return new Array(contentLength).fill(null).map((_, idx) => {
     const partIndex = idx;
     return /* @__PURE__ */ (0, import_jsx_runtime16.jsx)(
@@ -1811,7 +1818,7 @@ var useThreadViewportAutoScroll = ({
     const div = divRef.current;
     if (!div) return;
     const isAtBottom = useViewport.getState().isAtBottom;
-    const newIsAtBottom = div.scrollHeight - div.scrollTop <= div.clientHeight;
+    const newIsAtBottom = div.scrollHeight - div.scrollTop <= div.clientHeight + 1;
     if (!newIsAtBottom && lastScrollTop.current < div.scrollTop) {
     } else {
       if (newIsAtBottom) {
@@ -2047,26 +2054,6 @@ var ThreadPrimitiveSuggestion = createActionButton(
 // src/runtimes/local/useLocalRuntime.tsx
 var import_react54 = require("react");
 
-// src/utils/idUtils.tsx
-var import_non_secure = require("nanoid/non-secure");
-var generateId = (0, import_non_secure.customAlphabet)(
-  "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz",
-  7
-);
-var optimisticPrefix = "__optimistic__";
-var generateOptimisticId = () => `${optimisticPrefix}${generateId()}`;
-
-// src/internal.ts
-var internal_exports = {};
-__export(internal_exports, {
-  BaseAssistantRuntime: () => BaseAssistantRuntime,
-  MessageRepository: () => MessageRepository,
-  ProxyConfigProvider: () => ProxyConfigProvider,
-  TooltipIconButton: () => TooltipIconButton,
-  generateId: () => generateId,
-  useSmooth: () => useSmooth
-});
-
 // src/runtimes/core/BaseAssistantRuntime.tsx
 var BaseAssistantRuntime = class {
   constructor(_thread) {
@@ -2090,99 +2077,25 @@ var BaseAssistantRuntime = class {
   };
 };
 
-// src/ui/base/tooltip-icon-button.tsx
-var import_react52 = require("react");
-
-// src/ui/base/tooltip.tsx
-var TooltipPrimitive = __toESM(require("@radix-ui/react-tooltip"));
-
-// src/ui/utils/withDefaults.tsx
-var import_react50 = require("react");
-var import_classnames = __toESM(require("classnames"));
-var import_jsx_runtime25 = require("react/jsx-runtime");
-var withDefaultProps = ({
-  className,
-  ...defaultProps
-}) => ({ className: classNameProp, ...props }) => {
-  return {
-    className: (0, import_classnames.default)(className, classNameProp),
-    ...defaultProps,
-    ...props
-  };
-};
-var withDefaults = (Component, defaultProps) => {
-  const getProps = withDefaultProps(defaultProps);
-  const WithDefaults = (0, import_react50.forwardRef)(
-    (props, ref) => {
-      const ComponentAsAny = Component;
-      return /* @__PURE__ */ (0, import_jsx_runtime25.jsx)(ComponentAsAny, { ...getProps(props), ref });
-    }
-  );
-  WithDefaults.displayName = "withDefaults(" + (typeof Component === "string" ? Component : Component.displayName) + ")";
-  return WithDefaults;
-};
-
-// src/ui/base/tooltip.tsx
-var import_jsx_runtime26 = require("react/jsx-runtime");
-var Tooltip = (props) => {
-  return /* @__PURE__ */ (0, import_jsx_runtime26.jsx)(TooltipPrimitive.Provider, { children: /* @__PURE__ */ (0, import_jsx_runtime26.jsx)(TooltipPrimitive.Root, { ...props }) });
-};
-Tooltip.displayName = "Tooltip";
-var TooltipTrigger = TooltipPrimitive.Trigger;
-var TooltipContent = withDefaults(TooltipPrimitive.Content, {
-  sideOffset: 4,
-  className: "aui-tooltip-content"
+// src/internal.ts
+var internal_exports = {};
+__export(internal_exports, {
+  BaseAssistantRuntime: () => BaseAssistantRuntime,
+  MessageRepository: () => MessageRepository,
+  ProxyConfigProvider: () => ProxyConfigProvider,
+  TooltipIconButton: () => TooltipIconButton,
+  generateId: () => generateId,
+  useSmooth: () => useSmooth
 });
-TooltipContent.displayName = "TooltipContent";
 
-// src/ui/base/button.tsx
-var import_class_variance_authority = require("class-variance-authority");
-var import_react_primitive11 = require("@radix-ui/react-primitive");
-var import_react51 = require("react");
-var import_jsx_runtime27 = require("react/jsx-runtime");
-var buttonVariants = (0, import_class_variance_authority.cva)("aui-button", {
-  variants: {
-    variant: {
-      default: "aui-button-primary",
-      outline: "aui-button-outline",
-      ghost: "aui-button-ghost"
-    },
-    size: {
-      default: "aui-button-medium",
-      icon: "aui-button-icon"
-    }
-  },
-  defaultVariants: {
-    variant: "default",
-    size: "default"
-  }
-});
-var Button = (0, import_react51.forwardRef)(
-  ({ className, variant, size, ...props }, ref) => {
-    return /* @__PURE__ */ (0, import_jsx_runtime27.jsx)(
-      import_react_primitive11.Primitive.button,
-      {
-        className: buttonVariants({ variant, size, className }),
-        ...props,
-        ref
-      }
-    );
-  }
+// src/utils/idUtils.tsx
+var import_non_secure = require("nanoid/non-secure");
+var generateId = (0, import_non_secure.customAlphabet)(
+  "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz",
+  7
 );
-Button.displayName = "Button";
-
-// src/ui/base/tooltip-icon-button.tsx
-var import_jsx_runtime28 = require("react/jsx-runtime");
-var TooltipIconButton = (0, import_react52.forwardRef)(({ children, tooltip, side = "bottom", ...rest }, ref) => {
-  return /* @__PURE__ */ (0, import_jsx_runtime28.jsxs)(Tooltip, { children: [
-    /* @__PURE__ */ (0, import_jsx_runtime28.jsx)(TooltipTrigger, { asChild: true, children: /* @__PURE__ */ (0, import_jsx_runtime28.jsxs)(Button, { variant: "ghost", size: "icon", ...rest, ref, children: [
-      children,
-      /* @__PURE__ */ (0, import_jsx_runtime28.jsx)("span", { className: "aui-sr-only", children: tooltip })
-    ] }) }),
-    /* @__PURE__ */ (0, import_jsx_runtime28.jsx)(TooltipContent, { side, children: tooltip })
-  ] });
-});
-TooltipIconButton.displayName = "TooltipIconButton";
+var optimisticPrefix = "__optimistic__";
+var generateOptimisticId = () => `${optimisticPrefix}${generateId()}`;
 
 // src/runtimes/edge/converters/fromCoreMessage.ts
 var fromCoreMessages = (message) => {
@@ -2395,6 +2308,100 @@ var MessageRepository = class {
   }
 };
 
+// src/ui/base/tooltip-icon-button.tsx
+var import_react52 = require("react");
+
+// src/ui/base/tooltip.tsx
+var TooltipPrimitive = __toESM(require("@radix-ui/react-tooltip"));
+
+// src/ui/utils/withDefaults.tsx
+var import_react50 = require("react");
+var import_classnames = __toESM(require("classnames"));
+var import_jsx_runtime25 = require("react/jsx-runtime");
+var withDefaultProps = ({
+  className,
+  ...defaultProps
+}) => ({ className: classNameProp, ...props }) => {
+  return {
+    className: (0, import_classnames.default)(className, classNameProp),
+    ...defaultProps,
+    ...props
+  };
+};
+var withDefaults = (Component, defaultProps) => {
+  const getProps = withDefaultProps(defaultProps);
+  const WithDefaults = (0, import_react50.forwardRef)(
+    (props, ref) => {
+      const ComponentAsAny = Component;
+      return /* @__PURE__ */ (0, import_jsx_runtime25.jsx)(ComponentAsAny, { ...getProps(props), ref });
+    }
+  );
+  WithDefaults.displayName = "withDefaults(" + (typeof Component === "string" ? Component : Component.displayName) + ")";
+  return WithDefaults;
+};
+
+// src/ui/base/tooltip.tsx
+var import_jsx_runtime26 = require("react/jsx-runtime");
+var Tooltip = (props) => {
+  return /* @__PURE__ */ (0, import_jsx_runtime26.jsx)(TooltipPrimitive.Provider, { children: /* @__PURE__ */ (0, import_jsx_runtime26.jsx)(TooltipPrimitive.Root, { ...props }) });
+};
+Tooltip.displayName = "Tooltip";
+var TooltipTrigger = TooltipPrimitive.Trigger;
+var TooltipContent = withDefaults(TooltipPrimitive.Content, {
+  sideOffset: 4,
+  className: "aui-tooltip-content"
+});
+TooltipContent.displayName = "TooltipContent";
+
+// src/ui/base/button.tsx
+var import_class_variance_authority = require("class-variance-authority");
+var import_react_primitive11 = require("@radix-ui/react-primitive");
+var import_react51 = require("react");
+var import_jsx_runtime27 = require("react/jsx-runtime");
+var buttonVariants = (0, import_class_variance_authority.cva)("aui-button", {
+  variants: {
+    variant: {
+      default: "aui-button-primary",
+      outline: "aui-button-outline",
+      ghost: "aui-button-ghost"
+    },
+    size: {
+      default: "aui-button-medium",
+      icon: "aui-button-icon"
+    }
+  },
+  defaultVariants: {
+    variant: "default",
+    size: "default"
+  }
+});
+var Button = (0, import_react51.forwardRef)(
+  ({ className, variant, size, ...props }, ref) => {
+    return /* @__PURE__ */ (0, import_jsx_runtime27.jsx)(
+      import_react_primitive11.Primitive.button,
+      {
+        className: buttonVariants({ variant, size, className }),
+        ...props,
+        ref
+      }
+    );
+  }
+);
+Button.displayName = "Button";
+
+// src/ui/base/tooltip-icon-button.tsx
+var import_jsx_runtime28 = require("react/jsx-runtime");
+var TooltipIconButton = (0, import_react52.forwardRef)(({ children, tooltip, side = "bottom", ...rest }, ref) => {
+  return /* @__PURE__ */ (0, import_jsx_runtime28.jsxs)(Tooltip, { children: [
+    /* @__PURE__ */ (0, import_jsx_runtime28.jsx)(TooltipTrigger, { asChild: true, children: /* @__PURE__ */ (0, import_jsx_runtime28.jsxs)(Button, { variant: "ghost", size: "icon", ...rest, ref, children: [
+      children,
+      /* @__PURE__ */ (0, import_jsx_runtime28.jsx)("span", { className: "aui-sr-only", children: tooltip })
+    ] }) }),
+    /* @__PURE__ */ (0, import_jsx_runtime28.jsx)(TooltipContent, { side, children: tooltip })
+  ] });
+});
+TooltipIconButton.displayName = "TooltipIconButton";
+
 // src/runtimes/edge/converters/toLanguageModelMessages.ts
 var assistantMessageSplitter = () => {
   const stash = [];
@@ -3427,31 +3434,10 @@ var useEdgeRuntime = ({
   return useLocalRuntime(adapter, { initialMessages });
 };
 
-// src/runtimes/local/LocalRuntime.tsx
+// src/runtimes/local/shouldContinue.tsx
 var shouldContinue = (result) => result.status?.type === "requires-action" && result.status.reason === "tool-calls" && result.content.every((c) => c.type !== "tool-call" || !!c.result);
-var LocalRuntime = class extends BaseAssistantRuntime {
-  _proxyConfigProvider;
-  constructor(adapter, options) {
-    const proxyConfigProvider = new ProxyConfigProvider();
-    super(new LocalThreadRuntime(proxyConfigProvider, adapter, options));
-    this._proxyConfigProvider = proxyConfigProvider;
-  }
-  set adapter(adapter) {
-    this.thread.adapter = adapter;
-  }
-  registerModelConfigProvider(provider) {
-    return this._proxyConfigProvider.registerModelConfigProvider(provider);
-  }
-  switchToThread(threadId) {
-    if (threadId) {
-      throw new Error("LocalRuntime does not yet support switching threads");
-    }
-    return this.thread = new LocalThreadRuntime(
-      this._proxyConfigProvider,
-      this.thread.adapter
-    );
-  }
-};
+
+// src/runtimes/local/LocalThreadRuntime.tsx
 var CAPABILITIES = Object.freeze({
   edit: true,
   reload: true,
@@ -3511,7 +3497,7 @@ var LocalThreadRuntime = class {
       id,
       role: "assistant",
       status: { type: "running" },
-      content: [{ type: "text", text: "" }],
+      content: [],
       createdAt: /* @__PURE__ */ new Date()
     };
     do {
@@ -3623,6 +3609,31 @@ var LocalThreadRuntime = class {
   }
 };
 
+// src/runtimes/local/LocalRuntime.tsx
+var LocalRuntime = class extends BaseAssistantRuntime {
+  _proxyConfigProvider;
+  constructor(adapter, options) {
+    const proxyConfigProvider = new ProxyConfigProvider();
+    super(new LocalThreadRuntime(proxyConfigProvider, adapter, options));
+    this._proxyConfigProvider = proxyConfigProvider;
+  }
+  set adapter(adapter) {
+    this.thread.adapter = adapter;
+  }
+  registerModelConfigProvider(provider) {
+    return this._proxyConfigProvider.registerModelConfigProvider(provider);
+  }
+  switchToThread(threadId) {
+    if (threadId) {
+      throw new Error("LocalRuntime does not yet support switching threads");
+    }
+    return this.thread = new LocalThreadRuntime(
+      this._proxyConfigProvider,
+      this.thread.adapter
+    );
+  }
+};
+
 // src/runtimes/local/useLocalRuntime.tsx
 var useLocalRuntime = (adapter, options) => {
   const [runtime] = (0, import_react54.useState)(() => new LocalRuntime(adapter, options));