@assistant-ui/react 0.1.1 → 0.1.2

package/dist/index.mjs

@@ -1,22 +1,14 @@
-import {
-  MessageRepository,
-  ProxyConfigProvider,
-  generateId,
-  mergeModelConfigs
-} from "./chunk-NSBOH42A.mjs";
 import {
   AssistantContext,
   ContentPartContext,
   MessageContext,
   ThreadContext,
+  __export,
   useComposerContext,
   useContentPartContext,
   useMessageContext,
   useThreadContext
-} from "./chunk-CY4TTHR7.mjs";
-import {
-  __export
-} from "./chunk-J5LGTIGS.mjs";
+} from "./chunk-SCWPIDP6.mjs";
 
 // src/actions/useCopyMessage.tsx
 import { useCallback } from "react";
@@ -979,6 +971,185 @@ __export(contentPart_exports, {
 // src/runtime/local/useLocalRuntime.tsx
 import { useInsertionEffect, useState as useState3 } from "react";
 
+// src/utils/ModelConfigTypes.ts
+var mergeModelConfigs = (configs) => {
+  return configs.reduce((acc, config) => {
+    if (config.system) {
+      if (acc.system) {
+        acc.system += `
+
+${config.system}`;
+      } else {
+        acc.system = config.system;
+      }
+    }
+    if (config.tools) {
+      acc.tools = { ...acc.tools, ...config.tools };
+    }
+    return acc;
+  }, {});
+};
+
+// src/runtime/utils/idUtils.tsx
+import { customAlphabet } from "nanoid/non-secure";
+var generateId = customAlphabet(
+  "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz",
+  7
+);
+var optimisticPrefix = "__optimistic__";
+var generateOptimisticId = () => `${optimisticPrefix}${generateId()}`;
+
+// src/runtime/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;
+    }
+  }
+  appendOptimisticMessage(parentId, message) {
+    let optimisticId;
+    do {
+      optimisticId = generateOptimisticId();
+    } while (this.messages.has(optimisticId));
+    this.addOrUpdateMessage(parentId, {
+      ...message,
+      id: optimisticId,
+      createdAt: /* @__PURE__ */ new Date(),
+      ...message.role === "assistant" ? { status: "in_progress" } : void 0
+    });
+    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/runtime/local/LocalRuntime.tsx
 var LocalRuntime = class {
   constructor(adapter) {
@@ -1087,6 +1258,22 @@ import { useEffect as useEffect7, useInsertionEffect as useInsertionEffect3, use
 
 // src/context/stores/AssistantModelConfig.ts
 import { create as create4 } from "zustand";
+
+// src/utils/ProxyConfigProvider.ts
+var ProxyConfigProvider = class {
+  _providers = /* @__PURE__ */ new Set();
+  getModelConfig() {
+    return mergeModelConfigs([...this._providers].map((p) => p()));
+  }
+  registerModelConfigProvider(provider) {
+    this._providers.add(provider);
+    return () => {
+      this._providers.delete(provider);
+    };
+  }
+};
+
+// src/context/stores/AssistantModelConfig.ts
 var makeAssistantModelConfigStore = () => create4(() => {
   const proxy = new ProxyConfigProvider();
   return {
@@ -1227,12 +1414,20 @@ var AssistantRuntimeProviderImpl = ({ children, runtime }) => {
   return /* @__PURE__ */ jsx23(AssistantProvider, { runtime, children });
 };
 var AssistantRuntimeProvider = memo(AssistantRuntimeProviderImpl);
+
+// src/internal.ts
+var internal_exports = {};
+__export(internal_exports, {
+  MessageRepository: () => MessageRepository,
+  ProxyConfigProvider: () => ProxyConfigProvider
+});
 export {
   actionBar_exports as ActionBarPrimitive,
   AssistantRuntimeProvider,
   branchPicker_exports as BranchPickerPrimitive,
   composer_exports as ComposerPrimitive,
   contentPart_exports as ContentPartPrimitive,
+  internal_exports as INTERNAL,
   message_exports as MessagePrimitive,
   thread_exports as ThreadPrimitive,
   useBeginMessageEdit,