@memstack/core 0.4.0 → 0.5.0

package/dist/index.cjs

@@ -360,7 +360,6 @@ var ContextCompiler = class {
 };
 
 // src/memory/MemoryStore.ts
-var EMBEDDING_BATCH_SIZE = 2048;
 var MemoryStore = class {
   storage;
   embedding;
@@ -449,8 +448,9 @@ var MemoryStore = class {
     let stored;
     if (this.embedding && this.embedOnStore) {
       const enrichedResults = [];
-      for (let i = 0; i < normalInputs.length; i += EMBEDDING_BATCH_SIZE) {
-        const batch = normalInputs.slice(i, i + EMBEDDING_BATCH_SIZE);
+      const batchSize = this.embedding.maxBatchSize ?? 2048;
+      for (let i = 0; i < normalInputs.length; i += batchSize) {
+        const batch = normalInputs.slice(i, i + batchSize);
         const texts = batch.map((inp) => inp.content);
         const embeddings = await this.embedding.embed(texts);
         const enriched = batch.map((input, j) => ({ ...input, embedding: embeddings[j] }));
@@ -3438,6 +3438,264 @@ var MongoDBStorageAdapter = class {
   }
 };
 
+// src/adapters/storage/sqlite.ts
+function cosineSimilarity(a, b) {
+  if (a.length === 0 || b.length === 0) return 0;
+  const dims = Math.min(a.length, b.length);
+  let dot = 0;
+  let magA = 0;
+  let magB = 0;
+  for (let i = 0; i < dims; i++) {
+    dot += a[i] * b[i];
+    magA += a[i] * a[i];
+    magB += b[i] * b[i];
+  }
+  const mag = Math.sqrt(magA) * Math.sqrt(magB);
+  return mag === 0 ? 0 : dot / mag;
+}
+var SQLiteStorageAdapter = class {
+  db;
+  table;
+  vectorDimensions;
+  constructor(config) {
+    if (!config.db) {
+      throw configError("SQLiteStorageAdapter requires a better-sqlite3 database instance");
+    }
+    this.db = config.db;
+    this.table = config.tableName ?? "memstack_memories";
+    if (!/^[a-zA-Z0-9_]+$/.test(this.table)) {
+      throw configError(`Invalid table name: "${this.table}". Use only alphanumeric characters and underscores.`);
+    }
+    this.vectorDimensions = config.vectorDimensions ?? 1536;
+  }
+  async initialize() {
+    this.db.exec(`
+      CREATE TABLE IF NOT EXISTS ${this.table} (
+        id TEXT PRIMARY KEY,
+        actor_id TEXT NOT NULL,
+        memory_type TEXT NOT NULL DEFAULT 'interaction',
+        content TEXT NOT NULL,
+        importance REAL NOT NULL DEFAULT 0.5,
+        emotional_valence REAL NOT NULL DEFAULT 0,
+        tags TEXT NOT NULL DEFAULT '[]',
+        embedding TEXT,
+        source_id TEXT,
+        metadata TEXT NOT NULL DEFAULT '{}',
+        expires_at TEXT,
+        created_at TEXT NOT NULL,
+        touched_at TEXT NOT NULL
+      )
+    `);
+    this.db.exec(`CREATE INDEX IF NOT EXISTS idx_${this.table}_actor ON ${this.table}(actor_id)`);
+    this.db.exec(`CREATE INDEX IF NOT EXISTS idx_${this.table}_created ON ${this.table}(created_at DESC)`);
+    this.db.exec(`CREATE INDEX IF NOT EXISTS idx_${this.table}_importance ON ${this.table}(importance DESC)`);
+  }
+  generateId() {
+    return `mem_${Date.now().toString(36)}_${crypto.randomUUID().slice(0, 8)}`;
+  }
+  async store(input) {
+    const now = (/* @__PURE__ */ new Date()).toISOString();
+    const id = input.id ?? this.generateId();
+    this.db.prepare(
+      `INSERT INTO ${this.table} (id, actor_id, memory_type, content, importance, emotional_valence, tags, embedding, source_id, metadata, expires_at, created_at, touched_at)
+       VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+       ON CONFLICT(id) DO UPDATE SET
+         content = excluded.content,
+         importance = excluded.importance,
+         emotional_valence = excluded.emotional_valence,
+         tags = excluded.tags,
+         embedding = excluded.embedding,
+         source_id = excluded.source_id,
+         metadata = excluded.metadata,
+         expires_at = excluded.expires_at,
+         touched_at = excluded.touched_at`
+    ).run(
+      id,
+      input.actorId,
+      input.memoryType ?? "interaction",
+      input.content,
+      input.importance ?? 0.5,
+      input.emotionalValence ?? 0,
+      JSON.stringify(input.tags ?? []),
+      input.embedding ? JSON.stringify(input.embedding) : null,
+      input.sourceId ?? null,
+      JSON.stringify(input.metadata ?? {}),
+      input.expiresAt?.toISOString() ?? null,
+      now,
+      now
+    );
+    return {
+      id,
+      actorId: input.actorId,
+      memoryType: input.memoryType ?? "interaction",
+      content: input.content,
+      importance: input.importance ?? 0.5,
+      emotionalValence: input.emotionalValence ?? 0,
+      tags: input.tags ?? [],
+      embedding: input.embedding,
+      sourceId: input.sourceId,
+      metadata: input.metadata ?? {},
+      expiresAt: input.expiresAt,
+      createdAt: new Date(now)
+    };
+  }
+  async storeBatch(inputs) {
+    const results = [];
+    for (const input of inputs) {
+      results.push(await this.store(input));
+    }
+    return results;
+  }
+  async get(id) {
+    const row = this.db.prepare(
+      `SELECT * FROM ${this.table} WHERE id = ? AND (expires_at IS NULL OR expires_at > ?)`
+    ).get(id, (/* @__PURE__ */ new Date()).toISOString());
+    if (!row) return null;
+    return this._rowToMemory(row);
+  }
+  async delete(id) {
+    const row = this.db.prepare(`SELECT id FROM ${this.table} WHERE id = ?`).get(id);
+    if (!row) throw notFound("Memory", id);
+    this.db.prepare(`DELETE FROM ${this.table} WHERE id = ?`).run(id);
+  }
+  async deleteMany(ids) {
+    const placeholders = ids.map(() => "?").join(", ");
+    const result = this.db.prepare(
+      `DELETE FROM ${this.table} WHERE id IN (${placeholders})`
+    ).run(...ids);
+    return result.changes;
+  }
+  async touch(id) {
+    const row = this.db.prepare(`SELECT id FROM ${this.table} WHERE id = ?`).get(id);
+    if (!row) throw notFound("Memory", id);
+    this.db.prepare(`UPDATE ${this.table} SET touched_at = ? WHERE id = ?`).run((/* @__PURE__ */ new Date()).toISOString(), id);
+  }
+  async retrieve(query, embedding) {
+    const conditions = ["(expires_at IS NULL OR expires_at > ?)"];
+    const params = [(/* @__PURE__ */ new Date()).toISOString()];
+    if (query.actorId) {
+      conditions.push("actor_id = ?");
+      params.push(query.actorId);
+    }
+    if (query.memoryTypes && query.memoryTypes.length > 0) {
+      const placeholders = query.memoryTypes.map(() => "?").join(", ");
+      conditions.push(`memory_type IN (${placeholders})`);
+      params.push(...query.memoryTypes);
+    }
+    if (query.createdAfter) {
+      conditions.push("created_at >= ?");
+      params.push(query.createdAfter.toISOString());
+    }
+    if (query.createdBefore) {
+      conditions.push("created_at <= ?");
+      params.push(query.createdBefore.toISOString());
+    }
+    const where = `WHERE ${conditions.join(" AND ")}`;
+    const hasSemantic = (query.strategy === "semantic" || query.strategy === "hybrid") && embedding && embedding.length > 0;
+    let orderBy;
+    switch (query.strategy) {
+      case "recent":
+        orderBy = "touched_at DESC";
+        break;
+      case "important":
+        orderBy = "importance DESC";
+        break;
+      case "hybrid":
+      case "semantic":
+      default:
+        if (hasSemantic) {
+          orderBy = "touched_at DESC";
+        } else {
+          orderBy = "importance DESC, touched_at DESC";
+        }
+        break;
+    }
+    const limit = query.limit ?? 10;
+    const sql = `SELECT * FROM ${this.table} ${where} ORDER BY ${orderBy}`;
+    const rows = this.db.prepare(sql).all(...params);
+    let results = rows;
+    if (query.query) {
+      const q = query.query.toLowerCase();
+      results = results.filter((r) => r.content.toLowerCase().includes(q));
+    }
+    if (query.tags && query.tags.length > 0) {
+      results = results.filter((r) => {
+        const tags = JSON.parse(r.tags);
+        return tags.some((t) => query.tags.includes(t));
+      });
+    }
+    if (hasSemantic && embedding) {
+      const scored = results.filter((r) => r.embedding !== null).map((r) => {
+        const emb = JSON.parse(r.embedding);
+        const score = cosineSimilarity(embedding, emb);
+        return { row: r, score };
+      }).sort((a, b) => b.score - a.score);
+      if (query.strategy === "hybrid") {
+        scored.sort((a, b) => b.score + b.row.importance - (a.score + a.row.importance));
+      }
+      const topK = scored.slice(0, limit);
+      for (const { row } of topK) {
+        this.db.prepare(`UPDATE ${this.table} SET touched_at = ? WHERE id = ?`).run((/* @__PURE__ */ new Date()).toISOString(), row.id);
+      }
+      return topK.map(({ row }) => this._rowToMemory(row));
+    }
+    const now = (/* @__PURE__ */ new Date()).toISOString();
+    const limited = results.slice(0, limit);
+    for (const row of limited) {
+      this.db.prepare(`UPDATE ${this.table} SET touched_at = ? WHERE id = ?`).run(now, row.id);
+    }
+    return limited.map((r) => this._rowToMemory(r));
+  }
+  async count(filter) {
+    const conditions = ["(expires_at IS NULL OR expires_at > ?)"];
+    const params = [(/* @__PURE__ */ new Date()).toISOString()];
+    if (filter?.actorId) {
+      conditions.push("actor_id = ?");
+      params.push(filter.actorId);
+    }
+    if (filter?.memoryType) {
+      conditions.push("memory_type = ?");
+      params.push(filter.memoryType);
+    }
+    if (filter?.minImportance !== void 0) {
+      conditions.push("importance >= ?");
+      params.push(filter.minImportance);
+    }
+    if (filter?.createdAfter) {
+      conditions.push("created_at >= ?");
+      params.push(filter.createdAfter.toISOString());
+    }
+    if (filter?.createdBefore) {
+      conditions.push("created_at <= ?");
+      params.push(filter.createdBefore.toISOString());
+    }
+    const where = conditions.join(" AND ");
+    const row = this.db.prepare(
+      `SELECT COUNT(*) as count FROM ${this.table} WHERE ${where}`
+    ).get(...params);
+    return row?.count ?? 0;
+  }
+  async close() {
+    this.db.close();
+  }
+  _rowToMemory(row) {
+    return {
+      id: row.id,
+      actorId: row.actor_id,
+      memoryType: row.memory_type,
+      content: row.content,
+      importance: row.importance,
+      emotionalValence: row.emotional_valence,
+      tags: JSON.parse(row.tags),
+      embedding: row.embedding ? JSON.parse(row.embedding) : void 0,
+      sourceId: row.source_id ?? void 0,
+      metadata: JSON.parse(row.metadata),
+      expiresAt: row.expires_at ? new Date(row.expires_at) : void 0,
+      createdAt: new Date(row.created_at)
+    };
+  }
+};
+
 // src/adapters/llm/openai.ts
 var OpenAILLMAdapter = class {
   apiKey;
@@ -3624,6 +3882,7 @@ var OpenAIEmbeddingAdapter = class {
   baseURL;
   model;
   dimensions;
+  maxBatchSize = 2048;
   constructor(config) {
     this.apiKey = config.apiKey;
     this.baseURL = config.baseURL ?? "https://api.openai.com/v1";
@@ -3632,8 +3891,8 @@ var OpenAIEmbeddingAdapter = class {
   }
   async embed(texts) {
     if (texts.length === 0) return [];
-    if (texts.length > 2048) {
-      throw new MemStackError("VALIDATION_ERROR", `Max 2048 texts per batch, got ${texts.length}`);
+    if (texts.length > this.maxBatchSize) {
+      throw new MemStackError("VALIDATION_ERROR", `Max ${this.maxBatchSize} texts per batch, got ${texts.length}`);
     }
     try {
       const response = await fetch(`${this.baseURL}/embeddings`, {
@@ -3680,6 +3939,7 @@ var CohereEmbeddingAdapter = class {
   baseURL;
   model;
   dimensions;
+  maxBatchSize = 96;
   constructor(config) {
     this.apiKey = config.apiKey;
     this.baseURL = config.baseURL ?? "https://api.cohere.ai/v1";
@@ -3688,8 +3948,8 @@ var CohereEmbeddingAdapter = class {
   }
   async embed(texts) {
     if (texts.length === 0) return [];
-    if (texts.length > 96) {
-      throw new MemStackError("VALIDATION_ERROR", `Cohere max 96 texts per batch, got ${texts.length}`);
+    if (texts.length > this.maxBatchSize) {
+      throw new MemStackError("VALIDATION_ERROR", `Cohere max ${this.maxBatchSize} texts per batch, got ${texts.length}`);
     }
     try {
       const response = await fetch(`${this.baseURL}/embed`, {
@@ -3932,6 +4192,7 @@ exports.OpenAILLMAdapter = OpenAILLMAdapter;
 exports.PostgresStorageAdapter = PostgresStorageAdapter;
 exports.QdrantStorageAdapter = QdrantStorageAdapter;
 exports.RedisStorageAdapter = RedisStorageAdapter;
+exports.SQLiteStorageAdapter = SQLiteStorageAdapter;
 exports.WeaviateStorageAdapter = WeaviateStorageAdapter;
 exports.configError = configError;
 exports.notFound = notFound;