@liveblocks/core 3.8.0 → 3.9.0

package/dist/index.js

@@ -6,7 +6,7 @@ var __export = (target, all) => {
 
 // src/version.ts
 var PKG_NAME = "@liveblocks/core";
-var PKG_VERSION = "3.8.0";
+var PKG_VERSION = "3.9.0";
 var PKG_FORMAT = "esm";
 
 // src/dupe-detection.ts
@@ -703,6 +703,7 @@ var AiChatDB = class {
     this.signal = new MutableSignal(this);
   }
   getEvenIfDeleted(chatId) {
+    this.signal.get();
     return this.#byId.get(chatId);
   }
   markDeleted(chatId) {
@@ -2386,6 +2387,14 @@ function createApiClient({
   function getGroup(groupId) {
     return batchedGetGroups.get(groupId);
   }
+  async function getUrlMetadata(_url) {
+    const { metadata } = await httpClient.get(
+      url`/v2/c/urls/metadata`,
+      await authManager.getAuthValue({ requestedScope: "comments:read" }),
+      { url: _url }
+    );
+    return metadata;
+  }
   return {
     // Room threads
     getThreads,
@@ -2443,7 +2452,9 @@ function createApiClient({
     groupsStore,
     getGroup,
     // AI
-    executeContextualPrompt
+    executeContextualPrompt,
+    // URL metadata
+    getUrlMetadata
   };
 }
 function getBearerTokenFromAuthValue(authValue) {
@@ -3058,7 +3069,6 @@ var ServerMsgCode = /* @__PURE__ */ ((ServerMsgCode2) => {
   ServerMsgCode2[ServerMsgCode2["ROOM_STATE"] = 104] = "ROOM_STATE";
   ServerMsgCode2[ServerMsgCode2["INITIAL_STORAGE_STATE"] = 200] = "INITIAL_STORAGE_STATE";
   ServerMsgCode2[ServerMsgCode2["UPDATE_STORAGE"] = 201] = "UPDATE_STORAGE";
-  ServerMsgCode2[ServerMsgCode2["REJECT_STORAGE_OP"] = 299] = "REJECT_STORAGE_OP";
   ServerMsgCode2[ServerMsgCode2["UPDATE_YDOC"] = 300] = "UPDATE_YDOC";
   ServerMsgCode2[ServerMsgCode2["THREAD_CREATED"] = 400] = "THREAD_CREATED";
   ServerMsgCode2[ServerMsgCode2["THREAD_DELETED"] = 407] = "THREAD_DELETED";
@@ -3069,6 +3079,7 @@ var ServerMsgCode = /* @__PURE__ */ ((ServerMsgCode2) => {
   ServerMsgCode2[ServerMsgCode2["COMMENT_DELETED"] = 404] = "COMMENT_DELETED";
   ServerMsgCode2[ServerMsgCode2["COMMENT_REACTION_ADDED"] = 405] = "COMMENT_REACTION_ADDED";
   ServerMsgCode2[ServerMsgCode2["COMMENT_REACTION_REMOVED"] = 406] = "COMMENT_REACTION_REMOVED";
+  ServerMsgCode2[ServerMsgCode2["REJECT_STORAGE_OP"] = 299] = "REJECT_STORAGE_OP";
   return ServerMsgCode2;
 })(ServerMsgCode || {});
 
@@ -3696,240 +3707,6 @@ var ManagedSocket = class {
 // src/internal.ts
 var kInternal = Symbol();
 
-// src/lib/shallow.ts
-function shallowArray(xs, ys) {
-  if (xs.length !== ys.length) {
-    return false;
-  }
-  for (let i = 0; i < xs.length; i++) {
-    if (!Object.is(xs[i], ys[i])) {
-      return false;
-    }
-  }
-  return true;
-}
-function shallowObj(objA, objB) {
-  if (!isPlainObject(objA) || !isPlainObject(objB)) {
-    return false;
-  }
-  const keysA = Object.keys(objA);
-  if (keysA.length !== Object.keys(objB).length) {
-    return false;
-  }
-  return keysA.every(
-    (key) => Object.prototype.hasOwnProperty.call(objB, key) && Object.is(objA[key], objB[key])
-  );
-}
-function shallow(a, b) {
-  if (Object.is(a, b)) {
-    return true;
-  }
-  const isArrayA = Array.isArray(a);
-  const isArrayB = Array.isArray(b);
-  if (isArrayA || isArrayB) {
-    if (!isArrayA || !isArrayB) {
-      return false;
-    }
-    return shallowArray(a, b);
-  }
-  return shallowObj(a, b);
-}
-function shallow2(a, b) {
-  if (!isPlainObject(a) || !isPlainObject(b)) {
-    return shallow(a, b);
-  }
-  const keysA = Object.keys(a);
-  if (keysA.length !== Object.keys(b).length) {
-    return false;
-  }
-  return keysA.every(
-    (key) => Object.prototype.hasOwnProperty.call(b, key) && shallow(a[key], b[key])
-  );
-}
-
-// src/lib/TreePool.ts
-var TreePool = class {
-  #_items;
-  #_childrenOf;
-  #_sorted;
-  #_primaryKey;
-  #_parentKeyFn;
-  #_lt;
-  constructor(primaryKey, parentKey, lt) {
-    this.#_primaryKey = primaryKey;
-    this.#_parentKeyFn = parentKey;
-    this.#_lt = lt;
-    this.#_items = /* @__PURE__ */ new Map();
-    this.#_childrenOf = new DefaultMap(() => /* @__PURE__ */ new Set());
-    this.#_sorted = SortedList.with(lt);
-  }
-  get(id) {
-    return this.#_items.get(id);
-  }
-  getOrThrow(id) {
-    return this.get(id) ?? raise(`Item with id ${id} not found`);
-  }
-  get sorted() {
-    return this.#_sorted;
-  }
-  getParentId(id) {
-    const item = this.getOrThrow(id);
-    return this.#_parentKeyFn(item);
-  }
-  getParent(id) {
-    const parentId = this.getParentId(id);
-    return parentId ? this.getOrThrow(parentId) : null;
-  }
-  getChildren(id) {
-    const childIds = this.#_childrenOf.get(id);
-    if (!childIds) return [];
-    return Array.from(childIds).map(
-      (id2) => this.#_items.get(id2)
-      // eslint-disable-line no-restricted-syntax
-    );
-  }
-  *walkUp(id, predicate) {
-    const includeSelf = true;
-    let nodeId = id;
-    do {
-      const item = this.getOrThrow(nodeId);
-      if (includeSelf || nodeId !== id) {
-        if (!predicate || predicate(item)) {
-          yield item;
-        }
-      }
-      nodeId = this.#_parentKeyFn(item);
-    } while (nodeId !== null);
-  }
-  // XXXX Generalize
-  *walkLeft(id, predicate) {
-    const self = this.getOrThrow(id);
-    const siblings = SortedList.from(this.getSiblings(id), this.#_lt);
-    for (const sibling of siblings.iterReversed()) {
-      if (this.#_lt(self, sibling)) continue;
-      if (!predicate || predicate(sibling)) {
-        yield sibling;
-      }
-    }
-  }
-  // XXXX Generalize
-  *walkRight(id, predicate) {
-    const self = this.getOrThrow(id);
-    const siblings = SortedList.from(this.getSiblings(id), this.#_lt);
-    for (const sibling of siblings) {
-      if (this.#_lt(sibling, self)) continue;
-      if (!predicate || predicate(sibling)) {
-        yield sibling;
-      }
-    }
-  }
-  // XXXX Generalize
-  *walkDown(id, predicate) {
-    const children = SortedList.from(this.getChildren(id), this.#_lt).rawArray;
-    for (let i = children.length - 1; i >= 0; i--) {
-      const child = children[i];
-      yield* this.walkDown(
-        this.#_primaryKey(child),
-        predicate
-        // "depth-first",
-        // true
-      );
-      if (!predicate || predicate(child)) {
-        yield child;
-      }
-    }
-  }
-  /** Returns all siblings, not including the item itself. */
-  getSiblings(id) {
-    const self = this.getOrThrow(id);
-    const parentId = this.getParentId(id);
-    return this.getChildren(parentId).filter((item) => item !== self);
-  }
-  [Symbol.iterator]() {
-    return this.#_sorted[Symbol.iterator]();
-  }
-  upsert(item) {
-    const pk = this.#_primaryKey(item);
-    const existing = this.#_items.get(pk);
-    if (existing) {
-      if (this.#_parentKeyFn(existing) !== this.#_parentKeyFn(item)) {
-        throw new Error(
-          "Cannot upsert parent ID changes that change the tree structure. Remove the entry first, and recreate it"
-        );
-      }
-      this.#_sorted.remove(existing);
-    }
-    this.#_items.set(pk, item);
-    this.#_sorted.add(item);
-    const parentId = this.#_parentKeyFn(item);
-    this.#_childrenOf.getOrCreate(parentId).add(pk);
-  }
-  remove(pk) {
-    const item = this.#_items.get(pk);
-    if (!item) return false;
-    const childIds = this.#_childrenOf.get(pk);
-    if (childIds) {
-      throw new Error(
-        `Cannot remove item '${pk}' while it still has children. Remove children first.`
-      );
-    }
-    const parentId = this.#_parentKeyFn(item);
-    const siblings = this.#_childrenOf.get(parentId);
-    if (siblings) {
-      siblings.delete(pk);
-      if (siblings.size === 0) {
-        this.#_childrenOf.delete(parentId);
-      }
-    }
-    this.#_sorted.remove(item);
-    this.#_childrenOf.delete(pk);
-    this.#_items.delete(pk);
-    return true;
-  }
-  clear() {
-    if (this.#_items.size === 0) return false;
-    this.#_childrenOf.clear();
-    this.#_items.clear();
-    this.#_sorted.clear();
-    return true;
-  }
-};
-
-// src/protocol/AuthToken.ts
-var Permission = /* @__PURE__ */ ((Permission2) => {
-  Permission2["Read"] = "room:read";
-  Permission2["Write"] = "room:write";
-  Permission2["PresenceWrite"] = "room:presence:write";
-  Permission2["CommentsWrite"] = "comments:write";
-  Permission2["CommentsRead"] = "comments:read";
-  return Permission2;
-})(Permission || {});
-function canWriteStorage(scopes) {
-  return scopes.includes("room:write" /* Write */);
-}
-function canComment(scopes) {
-  return scopes.includes("comments:write" /* CommentsWrite */) || scopes.includes("room:write" /* Write */);
-}
-function isValidAuthTokenPayload(data) {
-  return isPlainObject(data) && (data.k === "acc" /* ACCESS_TOKEN */ || data.k === "id" /* ID_TOKEN */ || data.k === "sec-legacy" /* SECRET_LEGACY */);
-}
-function parseAuthToken(rawTokenString) {
-  const tokenParts = rawTokenString.split(".");
-  if (tokenParts.length !== 3) {
-    throw new Error("Authentication error: invalid JWT token");
-  }
-  const payload = tryParseJson(b64decode(tokenParts[1]));
-  if (!(payload && isValidAuthTokenPayload(payload))) {
-    throw new Error(
-      "Authentication error: expected a valid token but did not get one. Hint: if you are using a callback, ensure the room is passed when creating the token. For more information: https://liveblocks.io/docs/api-reference/liveblocks-client#createClientCallback"
-    );
-  }
-  return {
-    raw: rawTokenString,
-    parsed: payload
-  };
-}
-
 // src/lib/IncrementalJsonParser.ts
 var EMPTY_OBJECT = Object.freeze({});
 var NULL_KEYWORD_CHARS = Array.from(new Set("null"));
@@ -4099,95 +3876,237 @@ var IncrementalJsonParser = class {
   }
 };
 
-// src/types/ai.ts
-function replaceOrAppend(content, newItem, keyFn, now2) {
-  const existingIndex = findLastIndex(
-    content,
-    (item) => item.type === newItem.type && keyFn(item) === keyFn(newItem)
+// src/lib/shallow.ts
+function shallowArray(xs, ys) {
+  if (xs.length !== ys.length) {
+    return false;
+  }
+  for (let i = 0; i < xs.length; i++) {
+    if (!Object.is(xs[i], ys[i])) {
+      return false;
+    }
+  }
+  return true;
+}
+function shallowObj(objA, objB) {
+  if (!isPlainObject(objA) || !isPlainObject(objB)) {
+    return false;
+  }
+  const keysA = Object.keys(objA);
+  if (keysA.length !== Object.keys(objB).length) {
+    return false;
+  }
+  return keysA.every(
+    (key) => Object.prototype.hasOwnProperty.call(objB, key) && Object.is(objA[key], objB[key])
   );
-  if (existingIndex > -1) {
-    content[existingIndex] = newItem;
-  } else {
-    closePart(content[content.length - 1], now2);
-    content.push(newItem);
+}
+function shallow(a, b) {
+  if (Object.is(a, b)) {
+    return true;
+  }
+  const isArrayA = Array.isArray(a);
+  const isArrayB = Array.isArray(b);
+  if (isArrayA || isArrayB) {
+    if (!isArrayA || !isArrayB) {
+      return false;
+    }
+    return shallowArray(a, b);
   }
+  return shallowObj(a, b);
 }
-function closePart(prevPart, endedAt) {
-  if (prevPart?.type === "reasoning") {
-    prevPart.endedAt ??= endedAt;
+function shallow2(a, b) {
+  if (!isPlainObject(a) || !isPlainObject(b)) {
+    return shallow(a, b);
+  }
+  const keysA = Object.keys(a);
+  if (keysA.length !== Object.keys(b).length) {
+    return false;
   }
+  return keysA.every(
+    (key) => Object.prototype.hasOwnProperty.call(b, key) && shallow(a[key], b[key])
+  );
 }
-function patchContentWithDelta(content, delta) {
-  if (delta === null)
-    return;
-  const now2 = (/* @__PURE__ */ new Date()).toISOString();
-  const lastPart = content[content.length - 1];
-  switch (delta.type) {
-    case "text-delta":
-      if (lastPart?.type === "text") {
-        lastPart.text += delta.textDelta;
-      } else {
-        closePart(lastPart, now2);
-        content.push({ type: "text", text: delta.textDelta });
+
+// src/lib/TreePool.ts
+var TreePool = class {
+  #_items;
+  #_childrenOf;
+  #_sorted;
+  #_primaryKey;
+  #_parentKeyFn;
+  #_lt;
+  constructor(primaryKey, parentKey, lt) {
+    this.#_primaryKey = primaryKey;
+    this.#_parentKeyFn = parentKey;
+    this.#_lt = lt;
+    this.#_items = /* @__PURE__ */ new Map();
+    this.#_childrenOf = new DefaultMap(() => /* @__PURE__ */ new Set());
+    this.#_sorted = SortedList.with(lt);
+  }
+  get(id) {
+    return this.#_items.get(id);
+  }
+  getOrThrow(id) {
+    return this.get(id) ?? raise(`Item with id ${id} not found`);
+  }
+  get sorted() {
+    return this.#_sorted;
+  }
+  getParentId(id) {
+    const item = this.getOrThrow(id);
+    return this.#_parentKeyFn(item);
+  }
+  getParent(id) {
+    const parentId = this.getParentId(id);
+    return parentId ? this.getOrThrow(parentId) : null;
+  }
+  getChildren(id) {
+    const childIds = this.#_childrenOf.get(id);
+    if (!childIds) return [];
+    return Array.from(childIds).map(
+      (id2) => this.#_items.get(id2)
+      // eslint-disable-line no-restricted-syntax
+    );
+  }
+  *walkUp(id, predicate) {
+    const includeSelf = true;
+    let nodeId = id;
+    do {
+      const item = this.getOrThrow(nodeId);
+      if (includeSelf || nodeId !== id) {
+        if (!predicate || predicate(item)) {
+          yield item;
+        }
+      }
+      nodeId = this.#_parentKeyFn(item);
+    } while (nodeId !== null);
+  }
+  // XXXX Generalize
+  *walkLeft(id, predicate) {
+    const self = this.getOrThrow(id);
+    const siblings = SortedList.from(this.getSiblings(id), this.#_lt);
+    for (const sibling of siblings.iterReversed()) {
+      if (this.#_lt(self, sibling)) continue;
+      if (!predicate || predicate(sibling)) {
+        yield sibling;
+      }
+    }
+  }
+  // XXXX Generalize
+  *walkRight(id, predicate) {
+    const self = this.getOrThrow(id);
+    const siblings = SortedList.from(this.getSiblings(id), this.#_lt);
+    for (const sibling of siblings) {
+      if (this.#_lt(sibling, self)) continue;
+      if (!predicate || predicate(sibling)) {
+        yield sibling;
+      }
+    }
+  }
+  // XXXX Generalize
+  *walkDown(id, predicate) {
+    const children = SortedList.from(this.getChildren(id), this.#_lt).rawArray;
+    for (let i = children.length - 1; i >= 0; i--) {
+      const child = children[i];
+      yield* this.walkDown(
+        this.#_primaryKey(child),
+        predicate
+        // "depth-first",
+        // true
+      );
+      if (!predicate || predicate(child)) {
+        yield child;
       }
-      break;
-    case "reasoning-delta":
-      if (lastPart?.type === "reasoning") {
-        lastPart.text += delta.textDelta;
-      } else {
-        closePart(lastPart, now2);
-        content.push({
-          type: "reasoning",
-          text: delta.textDelta,
-          startedAt: now2
-        });
+    }
+  }
+  /** Returns all siblings, not including the item itself. */
+  getSiblings(id) {
+    const self = this.getOrThrow(id);
+    const parentId = this.getParentId(id);
+    return this.getChildren(parentId).filter((item) => item !== self);
+  }
+  [Symbol.iterator]() {
+    return this.#_sorted[Symbol.iterator]();
+  }
+  upsert(item) {
+    const pk = this.#_primaryKey(item);
+    const existing = this.#_items.get(pk);
+    if (existing) {
+      if (this.#_parentKeyFn(existing) !== this.#_parentKeyFn(item)) {
+        throw new Error(
+          "Cannot upsert parent ID changes that change the tree structure. Remove the entry first, and recreate it"
+        );
       }
-      break;
-    case "tool-stream": {
-      const toolInvocation = createReceivingToolInvocation(
-        delta.invocationId,
-        delta.name
+      this.#_sorted.remove(existing);
+    }
+    this.#_items.set(pk, item);
+    this.#_sorted.add(item);
+    const parentId = this.#_parentKeyFn(item);
+    this.#_childrenOf.getOrCreate(parentId).add(pk);
+  }
+  remove(pk) {
+    const item = this.#_items.get(pk);
+    if (!item) return false;
+    const childIds = this.#_childrenOf.get(pk);
+    if (childIds) {
+      throw new Error(
+        `Cannot remove item '${pk}' while it still has children. Remove children first.`
       );
-      content.push(toolInvocation);
-      break;
     }
-    case "tool-delta": {
-      if (lastPart?.type === "tool-invocation" && lastPart.stage === "receiving") {
-        lastPart.__appendDelta?.(delta.delta);
+    const parentId = this.#_parentKeyFn(item);
+    const siblings = this.#_childrenOf.get(parentId);
+    if (siblings) {
+      siblings.delete(pk);
+      if (siblings.size === 0) {
+        this.#_childrenOf.delete(parentId);
       }
-      break;
     }
-    case "tool-invocation":
-      replaceOrAppend(content, delta, (x) => x.invocationId, now2);
-      break;
-    case "retrieval":
-      replaceOrAppend(content, delta, (x) => x.id, now2);
-      break;
-    default:
-      return assertNever(delta, "Unhandled case");
+    this.#_sorted.remove(item);
+    this.#_childrenOf.delete(pk);
+    this.#_items.delete(pk);
+    return true;
+  }
+  clear() {
+    if (this.#_items.size === 0) return false;
+    this.#_childrenOf.clear();
+    this.#_items.clear();
+    this.#_sorted.clear();
+    return true;
   }
+};
+
+// src/protocol/AuthToken.ts
+var Permission = /* @__PURE__ */ ((Permission2) => {
+  Permission2["Read"] = "room:read";
+  Permission2["Write"] = "room:write";
+  Permission2["PresenceWrite"] = "room:presence:write";
+  Permission2["CommentsWrite"] = "comments:write";
+  Permission2["CommentsRead"] = "comments:read";
+  return Permission2;
+})(Permission || {});
+function canWriteStorage(scopes) {
+  return scopes.includes("room:write" /* Write */);
 }
-function createReceivingToolInvocation(invocationId, name, partialArgsText = "") {
-  const parser = new IncrementalJsonParser(partialArgsText);
+function canComment(scopes) {
+  return scopes.includes("comments:write" /* CommentsWrite */) || scopes.includes("room:write" /* Write */);
+}
+function isValidAuthTokenPayload(data) {
+  return isPlainObject(data) && (data.k === "acc" /* ACCESS_TOKEN */ || data.k === "id" /* ID_TOKEN */ || data.k === "sec-legacy" /* SECRET_LEGACY */);
+}
+function parseAuthToken(rawTokenString) {
+  const tokenParts = rawTokenString.split(".");
+  if (tokenParts.length !== 3) {
+    throw new Error("Authentication error: invalid JWT token");
+  }
+  const payload = tryParseJson(b64decode(tokenParts[1]));
+  if (!(payload && isValidAuthTokenPayload(payload))) {
+    throw new Error(
+      "Authentication error: expected a valid token but did not get one. Hint: if you are using a callback, ensure the room is passed when creating the token. For more information: https://liveblocks.io/docs/api-reference/liveblocks-client#createClientCallback"
+    );
+  }
   return {
-    type: "tool-invocation",
-    stage: "receiving",
-    invocationId,
-    name,
-    // --- Alternative implementation for FRONTEND only ------------------------
-    get partialArgsText() {
-      return parser.source;
-    },
-    // prettier-ignore
-    get partialArgs() {
-      return parser.json;
-    },
-    // prettier-ignore
-    __appendDelta(delta) {
-      parser.append(delta);
-    }
-    // prettier-ignore
-    // ------------------------------------------------------------------------
+    raw: rawTokenString,
+    parsed: payload
   };
 }
 
@@ -4255,6 +4174,23 @@ var KnowledgeStack = class {
     this.invalidate();
   }
 };
+function createStore_forKnowledge() {
+  const knowledgeByChatId = new DefaultMap(
+    (_chatId) => new KnowledgeStack()
+  );
+  function getKnowledgeStack(chatId) {
+    return knowledgeByChatId.getOrCreate(chatId ?? kWILDCARD);
+  }
+  function getKnowledgeForChat(chatId) {
+    const globalKnowledge = knowledgeByChatId.getOrCreate(kWILDCARD).get();
+    const scopedKnowledge = knowledgeByChatId.get(chatId)?.get() ?? [];
+    return [...globalKnowledge, ...scopedKnowledge];
+  }
+  return {
+    getKnowledgeStack,
+    getKnowledgeForChat
+  };
+}
 function now() {
   return (/* @__PURE__ */ new Date()).toISOString();
 }
@@ -4658,6 +4594,7 @@ function createAi(config) {
   );
   const chatsStore = createStore_forUserAiChats();
   const toolsStore = createStore_forTools();
+  const knowledgeStore = createStore_forKnowledge();
   const messagesStore = createStore_forChatMessages(toolsStore, setToolResult);
   const context = {
     staticSessionInfoSig: new Signal(null),
@@ -4666,8 +4603,9 @@ function createAi(config) {
     chatsStore,
     messagesStore,
     toolsStore,
-    knowledge: new KnowledgeStack()
+    knowledgeStore
   };
+  const status\u03A3 = new Signal("initial");
   const DELTA_THROTTLE = 25;
   let pendingDeltas = [];
   let deltaBatchTimer = null;
@@ -4691,7 +4629,7 @@ function createAi(config) {
     }
   }
   let lastTokenKey;
-  function onStatusDidChange(_newStatus) {
+  function onStatusDidChange(newStatus) {
     const authValue = managedSocket.authValue;
     if (authValue !== null) {
       const tokenKey = getBearerTokenFromAuthValue(authValue);
@@ -4711,6 +4649,7 @@ function createAi(config) {
         }
       }
     }
+    status\u03A3.set(newStatus);
   }
   let _connectionLossTimerId;
   let _hasLostConnection = false;
@@ -4895,17 +4834,8 @@ function createAi(config) {
       chatId
     });
   }
-  function registerKnowledgeLayer(uniqueLayerId) {
-    return context.knowledge.registerLayer(uniqueLayerId);
-  }
-  function deregisterKnowledgeLayer(layerKey) {
-    context.knowledge.deregisterLayer(layerKey);
-  }
-  function updateKnowledge(layerKey, data, key = nanoid()) {
-    context.knowledge.updateKnowledge(layerKey, key, data);
-  }
   async function setToolResult(chatId, messageId, invocationId, result, options) {
-    const knowledge = context.knowledge.get();
+    const knowledge = context.knowledgeStore.getKnowledgeForChat(chatId);
     const tools = context.toolsStore.getToolDescriptions(chatId);
     const resp = await sendClientMsgWithResponse({
       cmd: "set-tool-result",
@@ -4952,9 +4882,9 @@ function createAi(config) {
       deleteMessage: (chatId, messageId) => sendClientMsgWithResponse({ cmd: "delete-message", chatId, messageId }),
       clearChat: (chatId) => sendClientMsgWithResponse({ cmd: "clear-chat", chatId }),
       askUserMessageInChat: async (chatId, userMessage, targetMessageId, options) => {
-        const globalKnowledge = context.knowledge.get();
+        const knowledge = context.knowledgeStore.getKnowledgeForChat(chatId);
         const requestKnowledge = options?.knowledge || [];
-        const combinedKnowledge = [...globalKnowledge, ...requestKnowledge];
+        const combinedKnowledge = [...knowledge, ...requestKnowledge];
         const tools = context.toolsStore.getToolDescriptions(chatId);
         messagesStore.markMine(targetMessageId);
         const resp = await sendClientMsgWithResponse({
@@ -4978,14 +4908,24 @@ function createAi(config) {
       getStatus: () => managedSocket.getStatus(),
       signals: {
         getChatMessagesForBranch\u03A3: context.messagesStore.getChatMessagesForBranch\u03A3,
-        getTool\u03A3: context.toolsStore.getTool\u03A3
+        getTool\u03A3: context.toolsStore.getTool\u03A3,
+        status\u03A3
       },
       getChatById: context.chatsStore.getChatById,
       queryChats: context.chatsStore.findMany,
       getLastUsedCopilotId: context.messagesStore.getLastUsedCopilotId,
-      registerKnowledgeLayer,
-      deregisterKnowledgeLayer,
-      updateKnowledge,
+      registerKnowledgeLayer: (uniqueLayerId, chatId) => {
+        const stack = context.knowledgeStore.getKnowledgeStack(chatId);
+        const layerKey = stack.registerLayer(uniqueLayerId);
+        const deregister = () => stack.deregisterLayer(layerKey);
+        return {
+          layerKey,
+          deregister
+        };
+      },
+      updateKnowledge: (layerKey, data, key, chatId) => {
+        context.knowledgeStore.getKnowledgeStack(chatId).updateKnowledge(layerKey, key ?? nanoid(), data);
+      },
       registerTool: context.toolsStore.registerTool
     },
     kInternal,
@@ -5002,7 +4942,7 @@ function makeCreateSocketDelegateForAi(baseUrl, WebSocketPolyfill) {
     }
     const url2 = new URL(baseUrl);
     url2.protocol = url2.protocol === "http:" ? "ws" : "wss";
-    url2.pathname = "/ai/v6";
+    url2.pathname = "/ai/v7";
     if (authValue.type === "secret") {
       url2.searchParams.set("tok", authValue.token.raw);
     } else if (authValue.type === "public") {
@@ -5014,6 +4954,112 @@ function makeCreateSocketDelegateForAi(baseUrl, WebSocketPolyfill) {
     return new ws(url2.toString());
   };
 }
+function replaceOrAppend(content, newItem, keyFn, now2) {
+  const existingIndex = findLastIndex(
+    content,
+    (item) => item.type === newItem.type && keyFn(item) === keyFn(newItem)
+  );
+  if (existingIndex > -1) {
+    content[existingIndex] = newItem;
+  } else {
+    closePart(content[content.length - 1], now2);
+    content.push(newItem);
+  }
+}
+function closePart(prevPart, endedAt) {
+  if (prevPart?.type === "reasoning") {
+    prevPart.endedAt ??= endedAt;
+  }
+}
+function patchContentWithDelta(content, delta) {
+  if (delta === null)
+    return;
+  const parts = content.filter(
+    (part) => part.type !== "sources"
+  );
+  const sources = content.filter((part) => part.type === "sources").flatMap((part) => part.sources);
+  const now2 = (/* @__PURE__ */ new Date()).toISOString();
+  const lastPart = parts[parts.length - 1];
+  switch (delta.type) {
+    case "text-delta":
+      if (lastPart?.type === "text") {
+        lastPart.text += delta.textDelta;
+      } else {
+        closePart(lastPart, now2);
+        parts.push({ type: "text", text: delta.textDelta });
+      }
+      break;
+    case "reasoning-delta":
+      if (lastPart?.type === "reasoning") {
+        lastPart.text += delta.textDelta;
+      } else {
+        closePart(lastPart, now2);
+        parts.push({
+          type: "reasoning",
+          text: delta.textDelta,
+          startedAt: now2
+        });
+      }
+      break;
+    case "tool-stream": {
+      const toolInvocation = createReceivingToolInvocation(
+        delta.invocationId,
+        delta.name
+      );
+      parts.push(toolInvocation);
+      break;
+    }
+    case "tool-delta": {
+      if (lastPart?.type === "tool-invocation" && lastPart.stage === "receiving") {
+        lastPart.__appendDelta?.(delta.delta);
+      }
+      break;
+    }
+    case "tool-invocation":
+      replaceOrAppend(parts, delta, (x) => x.invocationId, now2);
+      break;
+    case "retrieval":
+      replaceOrAppend(parts, delta, (x) => x.id, now2);
+      break;
+    case "source": {
+      sources.push(delta);
+      break;
+    }
+    default:
+      return assertNever(delta, "Unhandled case");
+  }
+  if (sources.length > 0) {
+    parts.push({
+      type: "sources",
+      sources
+    });
+  }
+  content.length = 0;
+  content.push(...parts);
+}
+function createReceivingToolInvocation(invocationId, name, partialArgsText = "") {
+  const parser = new IncrementalJsonParser(partialArgsText);
+  return {
+    type: "tool-invocation",
+    stage: "receiving",
+    invocationId,
+    name,
+    // --- Alternative implementation for FRONTEND only ------------------------
+    get partialArgsText() {
+      return parser.source;
+    },
+    // prettier-ignore
+    get partialArgs() {
+      return parser.json;
+    },
+    // prettier-ignore
+    __appendDelta(delta) {
+      parser.append(delta);
+    }
+    // prettier-ignore
+    // ------------------------------------------------------------------------
+  };
+}
 
 // src/auth-manager.ts
 function createAuthManager(authOptions, onAuthenticate) {
@@ -8142,16 +8188,6 @@ function findNonSerializableValue(value, path = "") {
   return false;
 }
 
-// src/lib/debug.ts
-function captureStackTrace(msg, traceRoot) {
-  const errorLike = { name: msg };
-  if (typeof Error.captureStackTrace !== "function") {
-    return void 0;
-  }
-  Error.captureStackTrace(errorLike, traceRoot);
-  return errorLike.stack;
-}
-
 // src/lib/Deque.ts
 var Deque = class {
   #data;
@@ -8562,9 +8598,7 @@ function createRoom(options, config) {
     redoStack: [],
     pausedHistory: null,
     activeBatch: null,
-    unacknowledgedOps: /* @__PURE__ */ new Map(),
-    // Debug
-    opStackTraces: process.env.NODE_ENV !== "production" ? /* @__PURE__ */ new Map() : void 0
+    unacknowledgedOps: /* @__PURE__ */ new Map()
   };
   let lastTokenKey;
   function onStatusDidChange(newStatus) {
@@ -8648,16 +8682,6 @@ function createRoom(options, config) {
     }
   });
   function onDispatch(ops, reverse, storageUpdates) {
-    if (process.env.NODE_ENV !== "production") {
-      const stackTrace = captureStackTrace("Storage mutation", onDispatch);
-      if (stackTrace) {
-        for (const op of ops) {
-          if (op.opId) {
-            nn(context.opStackTraces).set(op.opId, stackTrace);
-          }
-        }
-      }
-    }
     if (context.activeBatch) {
       for (const op of ops) {
         context.activeBatch.ops.push(op);
@@ -8974,9 +8998,6 @@ function createRoom(options, config) {
           source = 0 /* UNDOREDO_RECONNECT */;
         } else {
           const opId = nn(op.opId);
-          if (process.env.NODE_ENV !== "production") {
-            nn(context.opStackTraces).delete(opId);
-          }
           const deleted = context.unacknowledgedOps.delete(opId);
           source = deleted ? 2 /* ACK */ : 1 /* REMOTE */;
         }
@@ -9262,38 +9283,6 @@ function createRoom(options, config) {
           }
           break;
         }
-        // Receiving a RejectedOps message in the client means that the server is no
-        // longer in sync with the client. Trying to synchronize the client again by
-        // rolling back particular Ops may be hard/impossible. It's fine to not try and
-        // accept the out-of-sync reality and throw an error. We look at this kind of bug
-        // as a developer-owned bug. In production, these errors are not expected to happen.
-        case 299 /* REJECT_STORAGE_OP */: {
-          errorWithTitle(
-            "Storage mutation rejection error",
-            message.reason
-          );
-          if (process.env.NODE_ENV !== "production") {
-            const traces = /* @__PURE__ */ new Set();
-            for (const opId of message.opIds) {
-              const trace = context.opStackTraces?.get(opId);
-              if (trace) {
-                traces.add(trace);
-              }
-            }
-            if (traces.size > 0) {
-              warnWithTitle(
-                "The following function calls caused the rejected storage mutations:",
-                `
-
-${Array.from(traces).join("\n\n")}`
-              );
-            }
-            throw new Error(
-              `Storage mutations rejected by server: ${message.reason}`
-            );
-          }
-          break;
-        }
         case 400 /* THREAD_CREATED */:
         case 407 /* THREAD_DELETED */:
         case 401 /* THREAD_METADATA_UPDATED */:
@@ -9306,6 +9295,8 @@ ${Array.from(traces).join("\n\n")}`
           eventHub.comments.notify(message);
           break;
         }
+        default:
+          break;
       }
     }
     notify(updates);