@assistant-ui/react 0.5.21 → 0.5.22

package/dist/internal.js

@@ -1,620 +0,0 @@
-"use client";
-"use strict";
-var __create = Object.create;
-var __defProp = Object.defineProperty;
-var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
-var __getOwnPropNames = Object.getOwnPropertyNames;
-var __getProtoOf = Object.getPrototypeOf;
-var __hasOwnProp = Object.prototype.hasOwnProperty;
-var __export = (target, all) => {
-  for (var name in all)
-    __defProp(target, name, { get: all[name], enumerable: true });
-};
-var __copyProps = (to, from, except, desc) => {
-  if (from && typeof from === "object" || typeof from === "function") {
-    for (let key of __getOwnPropNames(from))
-      if (!__hasOwnProp.call(to, key) && key !== except)
-        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
-  }
-  return to;
-};
-var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
-  // If the importer is in node compatibility mode or this is not an ESM
-  // file that has been converted to a CommonJS file using a Babel-
-  // compatible transform (i.e. "__esModule" has not been set), then set
-  // "default" to the CommonJS "module.exports" for node compatibility.
-  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
-  mod
-));
-var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
-
-// src/internal.ts
-var internal_exports = {};
-__export(internal_exports, {
-  BaseAssistantRuntime: () => BaseAssistantRuntime,
-  MessageRepository: () => MessageRepository,
-  ProxyConfigProvider: () => ProxyConfigProvider,
-  TooltipIconButton: () => TooltipIconButton,
-  generateId: () => generateId,
-  useSmooth: () => useSmooth,
-  useSmoothStatus: () => useSmoothStatus,
-  withSmoothContextProvider: () => withSmoothContextProvider
-});
-module.exports = __toCommonJS(internal_exports);
-
-// src/types/ModelConfigTypes.ts
-var import_zod = require("zod");
-var LanguageModelV1CallSettingsSchema = import_zod.z.object({
-  maxTokens: import_zod.z.number().int().positive().optional(),
-  temperature: import_zod.z.number().optional(),
-  topP: import_zod.z.number().optional(),
-  presencePenalty: import_zod.z.number().optional(),
-  frequencyPenalty: import_zod.z.number().optional(),
-  seed: import_zod.z.number().int().optional(),
-  headers: import_zod.z.record(import_zod.z.string().optional()).optional()
-});
-var LanguageModelConfigSchema = import_zod.z.object({
-  apiKey: import_zod.z.string().optional(),
-  baseUrl: import_zod.z.string().optional(),
-  modelName: import_zod.z.string().optional()
-});
-var mergeModelConfigs = (configSet) => {
-  const configs = Array.from(configSet).map((c) => c.getModelConfig()).sort((a, b) => (b.priority ?? 0) - (a.priority ?? 0));
-  return configs.reduce((acc, config) => {
-    if (config.system) {
-      if (acc.system) {
-        acc.system += `
-
-${config.system}`;
-      } else {
-        acc.system = config.system;
-      }
-    }
-    if (config.tools) {
-      for (const [name, tool] of Object.entries(config.tools)) {
-        if (acc.tools?.[name]) {
-          throw new Error(
-            `You tried to define a tool with the name ${name}, but it already exists.`
-          );
-        }
-        if (!acc.tools) acc.tools = {};
-        acc.tools[name] = tool;
-      }
-    }
-    if (config.config) {
-      acc.config = {
-        ...acc.config,
-        ...config.config
-      };
-    }
-    if (config.callSettings) {
-      acc.callSettings = {
-        ...acc.callSettings,
-        ...config.callSettings
-      };
-    }
-    return acc;
-  }, {});
-};
-
-// src/utils/ProxyConfigProvider.ts
-var ProxyConfigProvider = class {
-  _providers = /* @__PURE__ */ new Set();
-  getModelConfig() {
-    return mergeModelConfigs(this._providers);
-  }
-  registerModelConfigProvider(provider) {
-    this._providers.add(provider);
-    return () => {
-      this._providers.delete(provider);
-    };
-  }
-};
-
-// 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/runtimes/edge/converters/fromCoreMessage.ts
-var fromCoreMessage = (message, {
-  id = generateId(),
-  status = { type: "complete", reason: "unknown" }
-} = {}) => {
-  const commonProps = {
-    id,
-    createdAt: /* @__PURE__ */ new Date()
-  };
-  const role = message.role;
-  switch (role) {
-    case "assistant":
-      return {
-        ...commonProps,
-        role,
-        content: message.content.map((part) => {
-          if (part.type === "tool-call") {
-            return {
-              ...part,
-              argsText: JSON.stringify(part.args)
-            };
-          }
-          return part;
-        }),
-        status
-      };
-    case "user":
-      return {
-        ...commonProps,
-        role,
-        content: message.content
-      };
-    case "system":
-      return {
-        ...commonProps,
-        role,
-        content: message.content
-      };
-    default: {
-      const unsupportedRole = role;
-      throw new Error(`Unknown message role: ${unsupportedRole}`);
-    }
-  }
-};
-
-// src/runtimes/utils/MessageRepository.tsx
-var findHead = (message) => {
-  if (message.next) return findHead(message.next);
-  return message;
-};
-var MessageRepository = class {
-  messages = /* @__PURE__ */ new Map();
-  // message_id -> item
-  head = null;
-  root = {
-    children: []
-  };
-  performOp(newParent, child, operation) {
-    const parentOrRoot = child.prev ?? this.root;
-    const newParentOrRoot = newParent ?? this.root;
-    if (operation === "relink" && parentOrRoot === newParentOrRoot) return;
-    if (operation !== "link") {
-      parentOrRoot.children = parentOrRoot.children.filter(
-        (m) => m !== child.current.id
-      );
-      if (child.prev?.next === child) {
-        const fallbackId = child.prev.children.at(-1);
-        const fallback = fallbackId ? this.messages.get(fallbackId) : null;
-        if (fallback === void 0) {
-          throw new Error(
-            "MessageRepository(performOp/cut): Fallback sibling message not found. This is likely an internal bug in assistant-ui."
-          );
-        }
-        child.prev.next = fallback;
-      }
-    }
-    if (operation !== "cut") {
-      newParentOrRoot.children = [
-        ...newParentOrRoot.children,
-        child.current.id
-      ];
-      if (newParent && (findHead(child) === this.head || newParent.next === null)) {
-        newParent.next = child;
-      }
-      child.prev = newParent;
-    }
-  }
-  getMessages() {
-    const messages = new Array(this.head?.level ?? 0);
-    for (let current = this.head; current; current = current.prev) {
-      messages[current.level] = current.current;
-    }
-    return messages;
-  }
-  addOrUpdateMessage(parentId, message) {
-    const existingItem = this.messages.get(message.id);
-    const prev = parentId ? this.messages.get(parentId) : null;
-    if (prev === void 0)
-      throw new Error(
-        "MessageRepository(addOrUpdateMessage): Parent message not found. This is likely an internal bug in assistant-ui."
-      );
-    if (existingItem) {
-      existingItem.current = message;
-      this.performOp(prev, existingItem, "relink");
-      return;
-    }
-    const newItem = {
-      prev,
-      current: message,
-      next: null,
-      children: [],
-      level: prev ? prev.level + 1 : 0
-    };
-    this.messages.set(message.id, newItem);
-    this.performOp(prev, newItem, "link");
-    if (this.head === prev) {
-      this.head = newItem;
-    }
-  }
-  getMessage(messageId) {
-    const message = this.messages.get(messageId);
-    if (!message)
-      throw new Error(
-        "MessageRepository(updateMessage): Message not found. This is likely an internal bug in assistant-ui."
-      );
-    return {
-      parentId: message.prev?.current.id ?? null,
-      message: message.current
-    };
-  }
-  appendOptimisticMessage(parentId, message) {
-    let optimisticId;
-    do {
-      optimisticId = generateOptimisticId();
-    } while (this.messages.has(optimisticId));
-    this.addOrUpdateMessage(
-      parentId,
-      fromCoreMessage(message, {
-        id: optimisticId,
-        status: { type: "running" }
-      })
-    );
-    return optimisticId;
-  }
-  deleteMessage(messageId, replacementId) {
-    const message = this.messages.get(messageId);
-    if (!message)
-      throw new Error(
-        "MessageRepository(deleteMessage): Optimistic message not found. This is likely an internal bug in assistant-ui."
-      );
-    const replacement = replacementId === void 0 ? message.prev : replacementId === null ? null : this.messages.get(replacementId);
-    if (replacement === void 0)
-      throw new Error(
-        "MessageRepository(deleteMessage): Replacement not found. This is likely an internal bug in assistant-ui."
-      );
-    for (const child of message.children) {
-      const childMessage = this.messages.get(child);
-      if (!childMessage)
-        throw new Error(
-          "MessageRepository(deleteMessage): Child message not found. This is likely an internal bug in assistant-ui."
-        );
-      this.performOp(replacement, childMessage, "relink");
-    }
-    this.performOp(null, message, "cut");
-    this.messages.delete(messageId);
-    if (this.head === message) {
-      this.head = replacement ? findHead(replacement) : null;
-    }
-  }
-  getBranches(messageId) {
-    const message = this.messages.get(messageId);
-    if (!message)
-      throw new Error(
-        "MessageRepository(getBranches): Message not found. This is likely an internal bug in assistant-ui."
-      );
-    const { children } = message.prev ?? this.root;
-    return children;
-  }
-  switchToBranch(messageId) {
-    const message = this.messages.get(messageId);
-    if (!message)
-      throw new Error(
-        "MessageRepository(switchToBranch): Branch not found. This is likely an internal bug in assistant-ui."
-      );
-    if (message.prev) {
-      message.prev.next = message;
-    }
-    this.head = findHead(message);
-  }
-  resetHead(messageId) {
-    if (messageId === null) {
-      this.head = null;
-      return;
-    }
-    const message = this.messages.get(messageId);
-    if (!message)
-      throw new Error(
-        "MessageRepository(resetHead): Branch not found. This is likely an internal bug in assistant-ui."
-      );
-    this.head = message;
-    for (let current = message; current; current = current.prev) {
-      if (current.prev) {
-        current.prev.next = current;
-      }
-    }
-  }
-};
-
-// src/runtimes/core/BaseAssistantRuntime.tsx
-var BaseAssistantRuntime = class {
-  constructor(_thread) {
-    this._thread = _thread;
-    this._thread = _thread;
-  }
-  get thread() {
-    return this._thread;
-  }
-  set thread(thread) {
-    this._thread = thread;
-    this.subscriptionHandler();
-  }
-  _subscriptions = /* @__PURE__ */ new Set();
-  subscribe(callback) {
-    this._subscriptions.add(callback);
-    return () => this._subscriptions.delete(callback);
-  }
-  subscriptionHandler = () => {
-    for (const callback of this._subscriptions) callback();
-  };
-};
-
-// src/utils/smooth/useSmooth.tsx
-var import_react4 = require("react");
-
-// src/context/react/MessageContext.ts
-var import_react = require("react");
-var MessageContext = (0, import_react.createContext)(null);
-function useMessageContext(options) {
-  const context = (0, import_react.useContext)(MessageContext);
-  if (!options?.optional && !context)
-    throw new Error(
-      "This component can only be used inside a component passed to <ThreadPrimitive.Messages components={...} />."
-    );
-  return context;
-}
-
-// src/context/react/ContentPartContext.ts
-var import_react2 = require("react");
-var ContentPartContext = (0, import_react2.createContext)(
-  null
-);
-function useContentPartContext(options) {
-  const context = (0, import_react2.useContext)(ContentPartContext);
-  if (!options?.optional && !context)
-    throw new Error(
-      "This component can only be used inside a component passed to <MessagePrimitive.Content components={...} >."
-    );
-  return context;
-}
-
-// src/utils/smooth/SmoothContext.tsx
-var import_react3 = require("react");
-var import_zustand = require("zustand");
-var import_jsx_runtime = require("react/jsx-runtime");
-var SmoothContext = (0, import_react3.createContext)(null);
-var makeSmoothContext = (initialState) => {
-  const useSmoothStatus2 = (0, import_zustand.create)(() => initialState);
-  return { useSmoothStatus: useSmoothStatus2 };
-};
-var SmoothContextProvider = ({ children }) => {
-  const outer = useSmoothContext({ optional: true });
-  const { useContentPart } = useContentPartContext();
-  const [context] = (0, import_react3.useState)(
-    () => makeSmoothContext(useContentPart.getState().status)
-  );
-  if (outer) return children;
-  return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(SmoothContext.Provider, { value: context, children });
-};
-var withSmoothContextProvider = (Component) => {
-  const Wrapped = (0, import_react3.forwardRef)((props, ref) => {
-    return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(SmoothContextProvider, { children: /* @__PURE__ */ (0, import_jsx_runtime.jsx)(Component, { ...props, ref }) });
-  });
-  Wrapped.displayName = Component.displayName;
-  return Wrapped;
-};
-function useSmoothContext(options) {
-  const context = (0, import_react3.useContext)(SmoothContext);
-  if (!options?.optional && !context)
-    throw new Error(
-      "This component must be used within a SmoothContextProvider."
-    );
-  return context;
-}
-var useSmoothStatus = () => {
-  const { useSmoothStatus: useSmoothStatus2 } = useSmoothContext();
-  return useSmoothStatus2();
-};
-
-// src/utils/smooth/useSmooth.tsx
-var import_react_use_callback_ref = require("@radix-ui/react-use-callback-ref");
-var TextStreamAnimator = class {
-  constructor(currentText, setText) {
-    this.currentText = currentText;
-    this.setText = setText;
-  }
-  animationFrameId = null;
-  lastUpdateTime = Date.now();
-  targetText = "";
-  start() {
-    if (this.animationFrameId !== null) return;
-    this.lastUpdateTime = Date.now();
-    this.animate();
-  }
-  stop() {
-    if (this.animationFrameId !== null) {
-      cancelAnimationFrame(this.animationFrameId);
-      this.animationFrameId = null;
-    }
-  }
-  animate = () => {
-    const currentTime = Date.now();
-    const deltaTime = currentTime - this.lastUpdateTime;
-    let timeToConsume = deltaTime;
-    const remainingChars = this.targetText.length - this.currentText.length;
-    const baseTimePerChar = Math.min(5, 250 / remainingChars);
-    let charsToAdd = 0;
-    while (timeToConsume >= baseTimePerChar && charsToAdd < remainingChars) {
-      charsToAdd++;
-      timeToConsume -= baseTimePerChar;
-    }
-    if (charsToAdd !== remainingChars) {
-      this.animationFrameId = requestAnimationFrame(this.animate);
-    } else {
-      this.animationFrameId = null;
-    }
-    if (charsToAdd === 0) return;
-    this.currentText = this.targetText.slice(
-      0,
-      this.currentText.length + charsToAdd
-    );
-    this.lastUpdateTime = currentTime - timeToConsume;
-    this.setText(this.currentText);
-  };
-};
-var SMOOTH_STATUS = Object.freeze({
-  type: "running"
-});
-var useSmooth = (state, smooth = false) => {
-  const { useSmoothStatus: useSmoothStatus2 } = useSmoothContext({ optional: true }) ?? {};
-  const {
-    part: { text }
-  } = state;
-  const { useMessage } = useMessageContext();
-  const id = useMessage((m) => m.message.id);
-  const idRef = (0, import_react4.useRef)(id);
-  const [displayedText, setDisplayedText] = (0, import_react4.useState)(text);
-  const setText = (0, import_react_use_callback_ref.useCallbackRef)((text2) => {
-    setDisplayedText(text2);
-    useSmoothStatus2?.setState(text2 !== state.part.text ? SMOOTH_STATUS : state.status);
-  });
-  const [animatorRef] = (0, import_react4.useState)(
-    new TextStreamAnimator(text, setText)
-  );
-  (0, import_react4.useEffect)(() => {
-    if (!smooth) {
-      animatorRef.stop();
-      return;
-    }
-    if (idRef.current !== id || !text.startsWith(animatorRef.targetText)) {
-      idRef.current = id;
-      setText(text);
-      animatorRef.currentText = text;
-      animatorRef.targetText = text;
-      animatorRef.stop();
-      return;
-    }
-    animatorRef.targetText = text;
-    animatorRef.start();
-  }, [setText, animatorRef, id, smooth, text]);
-  (0, import_react4.useEffect)(() => {
-    return () => {
-      animatorRef.stop();
-    };
-  }, [animatorRef]);
-  return (0, import_react4.useMemo)(
-    () => smooth ? {
-      part: { type: "text", text: displayedText },
-      status: text === displayedText ? state.status : SMOOTH_STATUS
-    } : state,
-    [smooth, displayedText, state, text]
-  );
-};
-
-// src/ui/base/tooltip-icon-button.tsx
-var import_react7 = require("react");
-
-// src/ui/base/tooltip.tsx
-var TooltipPrimitive = __toESM(require("@radix-ui/react-tooltip"));
-
-// src/ui/utils/withDefaults.tsx
-var import_react5 = require("react");
-var import_classnames = __toESM(require("classnames"));
-var import_jsx_runtime2 = 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_react5.forwardRef)(
-    (props, ref) => {
-      const ComponentAsAny = Component;
-      return /* @__PURE__ */ (0, import_jsx_runtime2.jsx)(ComponentAsAny, { ...getProps(props), ref });
-    }
-  );
-  WithDefaults.displayName = "withDefaults(" + (typeof Component === "string" ? Component : Component.displayName) + ")";
-  return WithDefaults;
-};
-
-// src/ui/base/tooltip.tsx
-var import_jsx_runtime3 = require("react/jsx-runtime");
-var Tooltip = (props) => {
-  return /* @__PURE__ */ (0, import_jsx_runtime3.jsx)(TooltipPrimitive.Provider, { children: /* @__PURE__ */ (0, import_jsx_runtime3.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_primitive = require("@radix-ui/react-primitive");
-var import_react6 = require("react");
-var import_jsx_runtime4 = 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_react6.forwardRef)(
-  ({ className, variant, size, ...props }, ref) => {
-    return /* @__PURE__ */ (0, import_jsx_runtime4.jsx)(
-      import_react_primitive.Primitive.button,
-      {
-        className: buttonVariants({ variant, size, className }),
-        ...props,
-        ref
-      }
-    );
-  }
-);
-Button.displayName = "Button";
-
-// src/ui/base/tooltip-icon-button.tsx
-var import_jsx_runtime5 = require("react/jsx-runtime");
-var TooltipIconButton = (0, import_react7.forwardRef)(({ children, tooltip, side = "bottom", ...rest }, ref) => {
-  return /* @__PURE__ */ (0, import_jsx_runtime5.jsxs)(Tooltip, { children: [
-    /* @__PURE__ */ (0, import_jsx_runtime5.jsx)(TooltipTrigger, { asChild: true, children: /* @__PURE__ */ (0, import_jsx_runtime5.jsxs)(Button, { variant: "ghost", size: "icon", ...rest, ref, children: [
-      children,
-      /* @__PURE__ */ (0, import_jsx_runtime5.jsx)("span", { className: "aui-sr-only", children: tooltip })
-    ] }) }),
-    /* @__PURE__ */ (0, import_jsx_runtime5.jsx)(TooltipContent, { side, children: tooltip })
-  ] });
-});
-TooltipIconButton.displayName = "TooltipIconButton";
-// Annotate the CommonJS export names for ESM import in node:
-0 && (module.exports = {
-  BaseAssistantRuntime,
-  MessageRepository,
-  ProxyConfigProvider,
-  TooltipIconButton,
-  generateId,
-  useSmooth,
-  useSmoothStatus,
-  withSmoothContextProvider
-});
-//# sourceMappingURL=internal.js.map