@assistant-ui/react 0.1.0 → 0.1.2

package/dist/index.js

@@ -35,12 +35,14 @@ __export(src_exports, {
   BranchPickerPrimitive: () => branchPicker_exports,
   ComposerPrimitive: () => composer_exports,
   ContentPartPrimitive: () => contentPart_exports,
+  INTERNAL: () => internal_exports,
   MessagePrimitive: () => message_exports,
   ThreadPrimitive: () => thread_exports,
   useBeginMessageEdit: () => useBeginMessageEdit,
   useCopyMessage: () => useCopyMessage,
   useGoToNextBranch: () => useGoToNextBranch,
   useGoToPreviousBranch: () => useGoToPreviousBranch,
+  useLocalRuntime: () => useLocalRuntime,
   useReloadMessage: () => useReloadMessage
 });
 module.exports = __toCommonJS(src_exports);
@@ -1028,20 +1030,8 @@ __export(contentPart_exports, {
   InProgressIndicator: () => ContentPartInProgressIndicator
 });
 
-// src/context/providers/AssistantRuntimeProvider.tsx
-var import_react32 = require("react");
-
-// src/context/providers/AssistantProvider.tsx
-var import_react31 = require("react");
-
-// src/context/AssistantContext.ts
+// src/runtime/local/useLocalRuntime.tsx
 var import_react29 = require("react");
-var AssistantContext = (0, import_react29.createContext)(
-  null
-);
-
-// src/context/stores/AssistantModelConfig.ts
-var import_zustand4 = require("zustand");
 
 // src/utils/ModelConfigTypes.ts
 var mergeModelConfigs = (configs) => {
@@ -1062,6 +1052,281 @@ ${config.system}`;
   }, {});
 };
 
+// src/runtime/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/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] = (0, import_react29.useState)(() => new LocalRuntime(adapter));
+  (0, import_react29.useInsertionEffect)(() => {
+    runtime.adapter = adapter;
+  });
+  return runtime;
+};
+
+// src/context/providers/AssistantRuntimeProvider.tsx
+var import_react33 = require("react");
+
+// src/context/providers/AssistantProvider.tsx
+var import_react32 = require("react");
+
+// src/context/AssistantContext.ts
+var import_react30 = require("react");
+var AssistantContext = (0, import_react30.createContext)(
+  null
+);
+
+// src/context/stores/AssistantModelConfig.ts
+var import_zustand4 = require("zustand");
+
 // src/utils/ProxyConfigProvider.ts
 var ProxyConfigProvider = class {
   _providers = /* @__PURE__ */ new Set();
@@ -1090,7 +1355,7 @@ var makeAssistantModelConfigStore = () => (0, import_zustand4.create)(() => {
 });
 
 // src/context/providers/ThreadProvider.tsx
-var import_react30 = require("react");
+var import_react31 = require("react");
 
 // src/context/stores/Composer.ts
 var import_zustand5 = require("zustand");
@@ -1165,11 +1430,11 @@ var ThreadProvider = ({
   children,
   runtime
 }) => {
-  const runtimeRef = (0, import_react30.useRef)(runtime);
-  (0, import_react30.useInsertionEffect)(() => {
+  const runtimeRef = (0, import_react31.useRef)(runtime);
+  (0, import_react31.useInsertionEffect)(() => {
     runtimeRef.current = runtime;
   });
-  const [{ context, onRuntimeUpdate }] = (0, import_react30.useState)(() => {
+  const [{ context, onRuntimeUpdate }] = (0, import_react31.useState)(() => {
     const { useThread, onRuntimeUpdate: onRuntimeUpdate2 } = makeThreadStore(runtimeRef);
     const useViewport = makeThreadViewportStore();
     const useComposer = makeComposerStore(useThread);
@@ -1182,7 +1447,7 @@ var ThreadProvider = ({
       onRuntimeUpdate: onRuntimeUpdate2
     };
   });
-  (0, import_react30.useEffect)(() => {
+  (0, import_react31.useEffect)(() => {
     onRuntimeUpdate();
     return runtime.subscribe(onRuntimeUpdate);
   }, [onRuntimeUpdate, runtime]);
@@ -1196,16 +1461,16 @@ var ThreadProvider = ({
 // src/context/providers/AssistantProvider.tsx
 var import_jsx_runtime22 = require("react/jsx-runtime");
 var AssistantProvider = ({ children, runtime }) => {
-  const runtimeRef = (0, import_react31.useRef)(runtime);
-  (0, import_react31.useInsertionEffect)(() => {
+  const runtimeRef = (0, import_react32.useRef)(runtime);
+  (0, import_react32.useInsertionEffect)(() => {
     runtimeRef.current = runtime;
   });
-  const [context] = (0, import_react31.useState)(() => {
+  const [context] = (0, import_react32.useState)(() => {
     const useModelConfig = makeAssistantModelConfigStore();
     return { useModelConfig };
   });
   const getModelCOnfig = context.useModelConfig((c) => c.getModelConfig);
-  (0, import_react31.useEffect)(() => {
+  (0, import_react32.useEffect)(() => {
     return runtime.registerModelConfigProvider(getModelCOnfig);
   }, [runtime, getModelCOnfig]);
   return /* @__PURE__ */ (0, import_jsx_runtime22.jsx)(AssistantContext.Provider, { value: context, children: /* @__PURE__ */ (0, import_jsx_runtime22.jsx)(ThreadProvider, { runtime, children }) });
@@ -1216,7 +1481,14 @@ var import_jsx_runtime23 = require("react/jsx-runtime");
 var AssistantRuntimeProviderImpl = ({ children, runtime }) => {
   return /* @__PURE__ */ (0, import_jsx_runtime23.jsx)(AssistantProvider, { runtime, children });
 };
-var AssistantRuntimeProvider = (0, import_react32.memo)(AssistantRuntimeProviderImpl);
+var AssistantRuntimeProvider = (0, import_react33.memo)(AssistantRuntimeProviderImpl);
+
+// src/internal.ts
+var internal_exports = {};
+__export(internal_exports, {
+  MessageRepository: () => MessageRepository,
+  ProxyConfigProvider: () => ProxyConfigProvider
+});
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   ActionBarPrimitive,
@@ -1224,12 +1496,14 @@ var AssistantRuntimeProvider = (0, import_react32.memo)(AssistantRuntimeProvider
   BranchPickerPrimitive,
   ComposerPrimitive,
   ContentPartPrimitive,
+  INTERNAL,
   MessagePrimitive,
   ThreadPrimitive,
   useBeginMessageEdit,
   useCopyMessage,
   useGoToNextBranch,
   useGoToPreviousBranch,
+  useLocalRuntime,
   useReloadMessage
 });
 //# sourceMappingURL=index.js.map