mindcache 3.4.0 → 3.4.2

package/dist/index.mjs

@@ -46,7 +46,7 @@ var init_IndexedDBAdapter = __esm({
         await this.initDB();
         await this.load();
         const listener = () => {
-          if (this.mindcache && !this.mindcache.isRemoteUpdate()) {
+          if (this.mindcache) {
             this.scheduleSave();
           }
         };
@@ -426,15 +426,21 @@ var init_CloudAdapter = __esm({
 var DEFAULT_KEY_ATTRIBUTES = {
   type: "text",
   contentTags: [],
-  systemTags: ["SystemPrompt", "LLMWrite"],
-  // visible in system prompt and writable by LLM by default
-  zIndex: 0,
-  // Legacy - derived from systemTags
-  readonly: false,
-  visible: true,
-  hardcoded: false,
-  template: false,
-  tags: []
+  systemTags: [],
+  // Keys are private by default - explicitly add SystemPrompt/LLMRead/LLMWrite to enable LLM access
+  zIndex: 0
+};
+var SystemTagHelpers = {
+  /** Check if key is writable by LLM */
+  isLLMWritable: (attrs) => attrs.systemTags.includes("LLMWrite"),
+  /** Check if key is readable by LLM (in context or via tools) */
+  isLLMReadable: (attrs) => attrs.systemTags.includes("SystemPrompt") || attrs.systemTags.includes("LLMRead"),
+  /** Check if key is included in system prompt */
+  isInSystemPrompt: (attrs) => attrs.systemTags.includes("SystemPrompt"),
+  /** Check if key is protected from deletion */
+  isProtected: (attrs) => attrs.systemTags.includes("protected"),
+  /** Check if key uses template injection */
+  hasTemplateInjection: (attrs) => attrs.systemTags.includes("ApplyTemplate")
 };
 
 // src/core/MindCache.ts
@@ -450,38 +456,16 @@ var MindCache = class {
   version = "3.3.2";
   // Internal flag to prevent sync loops when receiving remote updates
   // (Less critical with Yjs but kept for API compat)
-  _isRemoteUpdate = false;
   normalizeSystemTags(tags) {
     const normalized = [];
-    let hasSystemPrompt = false;
-    let hasLLMRead = false;
-    let hasLLMWrite = false;
-    let hasReadonly = false;
+    const seen = /* @__PURE__ */ new Set();
     for (const tag of tags) {
-      if (tag === "SystemPrompt" || tag === "prompt") {
-        hasSystemPrompt = true;
-      } else if (tag === "LLMRead") {
-        hasLLMRead = true;
-      } else if (tag === "LLMWrite") {
-        hasLLMWrite = true;
-      } else if (tag === "readonly") {
-        hasReadonly = true;
-      } else if (tag === "protected") {
-        normalized.push(tag);
-      } else if (tag === "ApplyTemplate" || tag === "template") {
-        normalized.push("ApplyTemplate");
-      }
-    }
-    if (hasSystemPrompt) {
-      normalized.push("SystemPrompt");
-    }
-    if (hasLLMRead) {
-      normalized.push("LLMRead");
-    }
-    if (hasReadonly) {
-      normalized.push("readonly");
-    } else if (hasLLMWrite) {
-      normalized.push("LLMWrite");
+      if (["SystemPrompt", "LLMRead", "LLMWrite", "protected", "ApplyTemplate"].includes(tag)) {
+        if (!seen.has(tag)) {
+          seen.add(tag);
+          normalized.push(tag);
+        }
+      }
     }
     return normalized;
   }
@@ -872,10 +856,6 @@ var MindCache = class {
       this._idbProvider = null;
     }
   }
-  // Legacy bridge
-  isRemoteUpdate() {
-    return false;
-  }
   // Serialize state
   serialize() {
     const json = {};
@@ -912,16 +892,10 @@ var MindCache = class {
         const attrs = entry.attributes || {};
         const normalizedAttrs = {
           type: attrs.type || "text",
+          contentType: attrs.contentType,
           contentTags: attrs.contentTags || [],
-          systemTags: attrs.systemTags || this.normalizeSystemTags(attrs.visible !== false ? ["prompt"] : []),
-          zIndex: attrs.zIndex ?? 0,
-          // Legacy fields
-          readonly: attrs.readonly ?? false,
-          visible: attrs.visible ?? true,
-          hardcoded: attrs.hardcoded ?? false,
-          template: attrs.template ?? false,
-          tags: attrs.tags || [],
-          contentType: attrs.contentType
+          systemTags: this.normalizeSystemTags(attrs.systemTags || []),
+          zIndex: attrs.zIndex ?? 0
         };
         entryMap.set("attributes", normalizedAttrs);
       }
@@ -1026,7 +1000,7 @@ var MindCache = class {
     if (_processingStack && _processingStack.has(key)) {
       return `{{${key}}}`;
     }
-    if (attributes?.systemTags?.includes("ApplyTemplate") || attributes?.systemTags?.includes("template") || attributes?.template) {
+    if (attributes?.systemTags?.includes("ApplyTemplate")) {
       if (typeof value === "string") {
         const stack = _processingStack || /* @__PURE__ */ new Set();
         stack.add(key);
@@ -1040,13 +1014,8 @@ var MindCache = class {
       return {
         type: "text",
         contentTags: [],
-        systemTags: ["prompt", "readonly", "protected"],
-        zIndex: 999999,
-        readonly: true,
-        visible: true,
-        hardcoded: true,
-        template: false,
-        tags: []
+        systemTags: ["SystemPrompt", "protected"],
+        zIndex: 999999
       };
     }
     const entryMap = this.rootMap.get(key);
@@ -1071,6 +1040,14 @@ var MindCache = class {
         mergedAttrs.systemTags = this.normalizeSystemTags(mergedAttrs.systemTags);
       }
       entryMap.set("attributes", mergedAttrs);
+      const currentValue = entryMap.get("value");
+      if (mergedAttrs.type === "document" && !(currentValue instanceof Y.Text)) {
+        const strValue = typeof currentValue === "string" ? currentValue : String(currentValue ?? "");
+        entryMap.set("value", new Y.Text(strValue));
+        this.getUndoManager(key);
+      } else if (mergedAttrs.type !== "document" && currentValue instanceof Y.Text) {
+        entryMap.set("value", currentValue.toString());
+      }
     });
   }
   set_value(key, value, attributes) {
@@ -1081,10 +1058,15 @@ var MindCache = class {
     if (existingEntry) {
       const existingAttrs = existingEntry.get("attributes");
       if (existingAttrs?.type === "document") {
-        if (typeof value === "string") {
-          this._replaceDocumentText(key, value);
+        if (!attributes?.type || attributes.type === "document") {
+          if (typeof value === "string") {
+            this._replaceDocumentText(key, value);
+          }
+          if (attributes) {
+            this.set_attributes(key, attributes);
+          }
+          return;
         }
-        return;
       }
     }
     if (!existingEntry && attributes?.type === "document") {
@@ -1100,26 +1082,55 @@ var MindCache = class {
     this.getUndoManager(key);
     this.doc.transact(() => {
       const oldAttributes = isNewEntry ? {
-        ...DEFAULT_KEY_ATTRIBUTES,
-        contentTags: [],
-        systemTags: ["SystemPrompt", "LLMWrite"],
-        tags: [],
-        zIndex: 0
+        ...DEFAULT_KEY_ATTRIBUTES
       } : entryMap.get("attributes");
       const finalAttributes = attributes ? { ...oldAttributes, ...attributes } : oldAttributes;
       let normalizedAttributes = { ...finalAttributes };
       if (finalAttributes.systemTags) {
         normalizedAttributes.systemTags = this.normalizeSystemTags(finalAttributes.systemTags);
       }
-      if (finalAttributes.template) {
-        if (!normalizedAttributes.systemTags.includes("template")) {
-          normalizedAttributes.systemTags.push("template");
-        }
+      let valueToSet = value;
+      if (normalizedAttributes.type === "document" && !(valueToSet instanceof Y.Text)) {
+        valueToSet = new Y.Text(typeof value === "string" ? value : String(value ?? ""));
+      } else if (normalizedAttributes.type !== "document" && valueToSet instanceof Y.Text) {
+        valueToSet = valueToSet.toString();
       }
-      entryMap.set("value", value);
+      entryMap.set("value", valueToSet);
       entryMap.set("attributes", normalizedAttributes);
     });
   }
+  /**
+   * LLM-safe method to write a value to a key.
+   * This method:
+   * - Only updates the value, never modifies attributes/systemTags
+   * - Checks LLMWrite permission before writing
+   * - Returns false if key doesn't exist or lacks LLMWrite permission
+   *
+   * Used by create_vercel_ai_tools() to prevent LLMs from escalating privileges.
+   */
+  llm_set_key(key, value) {
+    if (key === "$date" || key === "$time" || key === "$version") {
+      return false;
+    }
+    const entryMap = this.rootMap.get(key);
+    if (!entryMap) {
+      return false;
+    }
+    const attributes = entryMap.get("attributes");
+    if (!attributes?.systemTags?.includes("LLMWrite")) {
+      return false;
+    }
+    if (attributes.type === "document") {
+      if (typeof value === "string") {
+        this._replaceDocumentText(key, value);
+      }
+      return true;
+    }
+    this.doc.transact(() => {
+      entryMap.set("value", value);
+    });
+    return true;
+  }
   delete_key(key) {
     if (key === "$date" || key === "$time") {
       return;
@@ -1564,15 +1575,13 @@ var MindCache = class {
       }
       const attributes = entryMap.get("attributes");
       const value = entryMap.get("value");
-      if (attributes?.hardcoded) {
+      if (attributes?.systemTags?.includes("protected")) {
         return;
       }
       lines.push(`### ${key}`);
       const entryType = attributes?.type || "text";
       lines.push(`- **Type**: \`${entryType}\``);
-      lines.push(`- **Readonly**: \`${attributes?.readonly ?? false}\``);
-      lines.push(`- **Visible**: \`${attributes?.visible ?? true}\``);
-      lines.push(`- **Template**: \`${attributes?.template ?? false}\``);
+      lines.push(`- **System Tags**: \`${attributes?.systemTags?.join(", ") || "none"}\``);
       lines.push(`- **Z-Index**: \`${attributes?.zIndex ?? 0}\``);
       if (attributes?.contentTags && attributes.contentTags.length > 0) {
         lines.push(`- **Tags**: \`${attributes.contentTags.join("`, `")}\``);
@@ -1658,16 +1667,11 @@ var MindCache = class {
         }
         continue;
       }
-      if (line.startsWith("- **Readonly**:")) {
-        currentAttributes.readonly = line.includes("`true`");
-        continue;
-      }
-      if (line.startsWith("- **Visible**:")) {
-        currentAttributes.visible = line.includes("`true`");
-        continue;
-      }
-      if (line.startsWith("- **Template**:")) {
-        currentAttributes.template = line.includes("`true`");
+      if (line.startsWith("- **System Tags**:")) {
+        const tagsStr = line.match(/`([^`]+)`/)?.[1] || "";
+        if (tagsStr !== "none") {
+          currentAttributes.systemTags = tagsStr.split(", ").filter((t) => t);
+        }
         continue;
       }
       if (line.startsWith("- **Z-Index**:")) {
@@ -1678,7 +1682,6 @@ var MindCache = class {
       if (line.startsWith("- **Tags**:")) {
         const tags = line.match(/`([^`]+)`/g)?.map((t) => t.slice(1, -1)) || [];
         currentAttributes.contentTags = tags;
-        currentAttributes.tags = tags;
         continue;
       }
       if (line.startsWith("- **Content Type**:")) {
@@ -1934,8 +1937,12 @@ var MindCache = class {
   /**
    * Generate Vercel AI SDK compatible tools for writable keys.
    * For document type keys, generates additional tools: append_, insert_, edit_
+   *
+   * Security: All tools use llm_set_key internally which:
+   * - Only modifies VALUES, never attributes/systemTags
+   * - Prevents LLMs from escalating privileges
    */
-  get_aisdk_tools() {
+  create_vercel_ai_tools() {
     const tools = {};
     for (const [key, val] of this.rootMap) {
       if (key.startsWith("$")) {
@@ -1943,7 +1950,7 @@ var MindCache = class {
       }
       const entryMap = val;
       const attributes = entryMap.get("attributes");
-      const isWritable = !attributes?.readonly && (attributes?.systemTags?.includes("LLMWrite") || !attributes?.systemTags);
+      const isWritable = attributes?.systemTags?.includes("LLMWrite");
       if (!isWritable) {
         continue;
       }
@@ -1959,15 +1966,18 @@ var MindCache = class {
           required: ["value"]
         },
         execute: async ({ value }) => {
-          if (isDocument) {
-            this._replaceDocumentText(key, value);
-          } else {
-            this.set_value(key, value);
+          const success = this.llm_set_key(key, value);
+          if (success) {
+            return {
+              result: `Successfully wrote "${value}" to ${key}`,
+              key,
+              value
+            };
           }
           return {
-            result: `Successfully wrote "${value}" to ${key}`,
+            result: `Failed to write to ${key} - permission denied or key not found`,
             key,
-            value
+            error: true
           };
         }
       };
@@ -1982,6 +1992,9 @@ var MindCache = class {
             required: ["text"]
           },
           execute: async ({ text }) => {
+            if (!attributes?.systemTags?.includes("LLMWrite")) {
+              return { result: `Permission denied for ${key}`, key, error: true };
+            }
             const yText = this.get_document(key);
             if (yText) {
               yText.insert(yText.length, text);
@@ -2005,6 +2018,9 @@ var MindCache = class {
             required: ["index", "text"]
           },
           execute: async ({ index, text }) => {
+            if (!attributes?.systemTags?.includes("LLMWrite")) {
+              return { result: `Permission denied for ${key}`, key, error: true };
+            }
             this.insert_text(key, index, text);
             return {
               result: `Successfully inserted text at position ${index} in ${key}`,
@@ -2025,6 +2041,9 @@ var MindCache = class {
             required: ["find", "replace"]
           },
           execute: async ({ find, replace }) => {
+            if (!attributes?.systemTags?.includes("LLMWrite")) {
+              return { result: `Permission denied for ${key}`, key, error: true };
+            }
             const yText = this.get_document(key);
             if (yText) {
               const text = yText.toString();
@@ -2049,6 +2068,12 @@ var MindCache = class {
     }
     return tools;
   }
+  /**
+   * @deprecated Use create_vercel_ai_tools() instead
+   */
+  get_aisdk_tools() {
+    return this.create_vercel_ai_tools();
+  }
   /**
    * Generate a system prompt containing all visible STM keys and their values.
    * Indicates which tools can be used to modify writable keys.
@@ -2061,13 +2086,13 @@ var MindCache = class {
       }
       const entryMap = val;
       const attributes = entryMap.get("attributes");
-      const isVisible = attributes?.visible !== false && (attributes?.systemTags?.includes("prompt") || attributes?.systemTags?.includes("SystemPrompt") || !attributes?.systemTags);
+      const isVisible = attributes?.systemTags?.includes("SystemPrompt") || attributes?.systemTags?.includes("LLMRead");
       if (!isVisible) {
         continue;
       }
       const value = this.get_value(key);
       const displayValue = typeof value === "object" ? JSON.stringify(value) : value;
-      const isWritable = !attributes?.readonly && (attributes?.systemTags?.includes("LLMWrite") || !attributes?.systemTags);
+      const isWritable = attributes?.systemTags?.includes("LLMWrite");
       const isDocument = attributes?.type === "document";
       const sanitizedKey = this.sanitizeKeyForTool(key);
       if (isWritable) {
@@ -2108,7 +2133,7 @@ var MindCache = class {
       return null;
     }
     const attributes = entryMap.get("attributes");
-    const isWritable = !attributes?.readonly && (attributes?.systemTags?.includes("LLMWrite") || !attributes?.systemTags);
+    const isWritable = attributes?.systemTags?.includes("LLMWrite");
     if (!isWritable) {
       return null;
     }
@@ -2223,6 +2248,6 @@ function useMindCache(options) {
 // src/index.ts
 var mindcache = new MindCache();
 
-export { CloudAdapter, DEFAULT_KEY_ATTRIBUTES, IndexedDBAdapter, MindCache, mindcache, useMindCache };
+export { CloudAdapter, DEFAULT_KEY_ATTRIBUTES, IndexedDBAdapter, MindCache, SystemTagHelpers, mindcache, useMindCache };
 //# sourceMappingURL=index.mjs.map
 //# sourceMappingURL=index.mjs.map