@assistant-ui/react 0.1.0 → 0.1.2

package/dist/index.mjs

@@ -1,19 +1,14 @@
-import {
-  ProxyConfigProvider
-} from "./chunk-DKAWDNW5.mjs";
 import {
   AssistantContext,
   ContentPartContext,
   MessageContext,
   ThreadContext,
+  __export,
   useComposerContext,
   useContentPartContext,
   useMessageContext,
   useThreadContext
-} from "./chunk-CY4TTHR7.mjs";
-import {
-  __export
-} from "./chunk-X4HBDEFP.mjs";
+} from "./chunk-SCWPIDP6.mjs";
 
 // src/actions/useCopyMessage.tsx
 import { useCallback } from "react";
@@ -973,14 +968,312 @@ __export(contentPart_exports, {
   InProgressIndicator: () => ContentPartInProgressIndicator
 });
 
+// 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) {
+    this.adapter = adapter;
+  }
+  _subscriptions = /* @__PURE__ */ new Set();
+  _configProviders = /* @__PURE__ */ new Set();
+  abortController = null;
+  repository = new MessageRepository();
+  get messages() {
+    return this.repository.getMessages();
+  }
+  get isRunning() {
+    return this.abortController != null;
+  }
+  getBranches(messageId) {
+    return this.repository.getBranches(messageId);
+  }
+  switchToBranch(branchId) {
+    this.repository.switchToBranch(branchId);
+    this.notifySubscribers();
+  }
+  async append(message) {
+    const userMessageId = generateId();
+    const userMessage = {
+      id: userMessageId,
+      role: "user",
+      content: message.content,
+      createdAt: /* @__PURE__ */ new Date()
+    };
+    this.repository.addOrUpdateMessage(message.parentId, userMessage);
+    await this.startRun(userMessageId);
+  }
+  async startRun(parentId) {
+    const id = generateId();
+    this.repository.resetHead(parentId);
+    const messages = this.repository.getMessages();
+    const message = {
+      id,
+      role: "assistant",
+      status: "in_progress",
+      content: [{ type: "text", text: "" }],
+      createdAt: /* @__PURE__ */ new Date()
+    };
+    this.repository.addOrUpdateMessage(parentId, { ...message });
+    this.abortController?.abort();
+    this.abortController = new AbortController();
+    this.notifySubscribers();
+    try {
+      const updateHandler = ({ content }) => {
+        message.content = content;
+        this.repository.addOrUpdateMessage(parentId, { ...message });
+        this.notifySubscribers();
+      };
+      const result = await this.adapter.run({
+        messages,
+        abortSignal: this.abortController.signal,
+        config: mergeModelConfigs([...this._configProviders].map((p) => p())),
+        onUpdate: updateHandler
+      });
+      updateHandler(result);
+      message.status = "done";
+      this.repository.addOrUpdateMessage(parentId, { ...message });
+    } catch (e) {
+      message.status = "error";
+      this.repository.addOrUpdateMessage(parentId, { ...message });
+      console.error(e);
+    } finally {
+      this.abortController = null;
+      this.notifySubscribers();
+    }
+  }
+  cancelRun() {
+    if (!this.abortController) return;
+    this.abortController.abort();
+    this.abortController = null;
+    this.notifySubscribers();
+  }
+  notifySubscribers() {
+    for (const callback of this._subscriptions) callback();
+  }
+  subscribe(callback) {
+    this._subscriptions.add(callback);
+    return () => this._subscriptions.delete(callback);
+  }
+  registerModelConfigProvider(provider) {
+    this._configProviders.add(provider);
+    return () => this._configProviders.delete(provider);
+  }
+};
+
+// src/runtime/local/useLocalRuntime.tsx
+var useLocalRuntime = (adapter) => {
+  const [runtime] = useState3(() => new LocalRuntime(adapter));
+  useInsertionEffect(() => {
+    runtime.adapter = adapter;
+  });
+  return runtime;
+};
+
 // src/context/providers/AssistantRuntimeProvider.tsx
 import { memo } from "react";
 
 // src/context/providers/AssistantProvider.tsx
-import { useEffect as useEffect7, useInsertionEffect as useInsertionEffect2, useRef as useRef5, useState as useState4 } from "react";
+import { useEffect as useEffect7, useInsertionEffect as useInsertionEffect3, useRef as useRef5, useState as useState5 } from "react";
 
 // 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 {
@@ -994,7 +1287,7 @@ var makeAssistantModelConfigStore = () => create4(() => {
 });
 
 // src/context/providers/ThreadProvider.tsx
-import { useEffect as useEffect6, useInsertionEffect, useRef as useRef4, useState as useState3 } from "react";
+import { useEffect as useEffect6, useInsertionEffect as useInsertionEffect2, useRef as useRef4, useState as useState4 } from "react";
 
 // src/context/stores/Composer.ts
 import { create as create5 } from "zustand";
@@ -1070,10 +1363,10 @@ var ThreadProvider = ({
   runtime
 }) => {
   const runtimeRef = useRef4(runtime);
-  useInsertionEffect(() => {
+  useInsertionEffect2(() => {
     runtimeRef.current = runtime;
   });
-  const [{ context, onRuntimeUpdate }] = useState3(() => {
+  const [{ context, onRuntimeUpdate }] = useState4(() => {
     const { useThread, onRuntimeUpdate: onRuntimeUpdate2 } = makeThreadStore(runtimeRef);
     const useViewport = makeThreadViewportStore();
     const useComposer = makeComposerStore(useThread);
@@ -1101,10 +1394,10 @@ var ThreadProvider = ({
 import { jsx as jsx22 } from "react/jsx-runtime";
 var AssistantProvider = ({ children, runtime }) => {
   const runtimeRef = useRef5(runtime);
-  useInsertionEffect2(() => {
+  useInsertionEffect3(() => {
     runtimeRef.current = runtime;
   });
-  const [context] = useState4(() => {
+  const [context] = useState5(() => {
     const useModelConfig = makeAssistantModelConfigStore();
     return { useModelConfig };
   });
@@ -1121,18 +1414,27 @@ 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,
   useCopyMessage,
   useGoToNextBranch,
   useGoToPreviousBranch,
+  useLocalRuntime,
   useReloadMessage
 };
 //# sourceMappingURL=index.mjs.map