@memrosetta/core 0.2.22 → 0.4.0

package/dist/index.d.ts

@@ -2,6 +2,14 @@ import Database from 'better-sqlite3';
 import { Memory, MemoryState, MemoryInput, RelationType, MemoryRelation, SearchResult, SearchFilters, SearchQuery, SearchResponse, IMemoryEngine, CompressResult, MaintenanceResult, MemoryTier, MemoryQuality, TierConfig } from '@memrosetta/types';
 import { Embedder, ContradictionDetector } from '@memrosetta/embeddings';
 
+/**
+ * Thrown when a memory ID referenced in an operation does not exist.
+ */
+declare class MemoryNotFoundError extends Error {
+    readonly memoryId: string;
+    constructor(memoryId: string);
+}
+
 interface SchemaOptions {
     readonly vectorEnabled?: boolean;
     readonly embeddingDimension?: number;
@@ -342,4 +350,4 @@ declare function determineTier(memory: {
  */
 declare function estimateTokens(content: string): number;
 
-export { DEFAULT_TIER_CONFIG, type MemoryRow, type PreparedStatements, type RelationStatements, type SchemaOptions, type SearchSqlResult, type SqliteEngineOptions, SqliteMemoryEngine, type VectorSearchResult, applyKeywordBoost, applyThreeFactorReranking, bruteForceVectorSearch, buildFtsQuery, buildSearchSql, computeActivation, computeEbbinghaus, convexCombinationMerge, createEngine, createPreparedStatements, createRelation, createRelationStatements, deduplicateResults, deriveMemoryState, determineTier, ensureSchema, estimateTokens, extractQueryTokens, ftsSearch, generateMemoryId, getRelationsByMemory, keywordsToString, normalizeScores, nowIso, rowToMemory, rrfMerge, rrfMergeWeighted, searchMemories, serializeEmbedding, storeBatchAsync, storeBatchInTransaction, storeMemory, storeMemoryAsync, stringToKeywords, updateAccessTracking, vectorSearch };
+export { DEFAULT_TIER_CONFIG, MemoryNotFoundError, type MemoryRow, type PreparedStatements, type RelationStatements, type SchemaOptions, type SearchSqlResult, type SqliteEngineOptions, SqliteMemoryEngine, type VectorSearchResult, applyKeywordBoost, applyThreeFactorReranking, bruteForceVectorSearch, buildFtsQuery, buildSearchSql, computeActivation, computeEbbinghaus, convexCombinationMerge, createEngine, createPreparedStatements, createRelation, createRelationStatements, deduplicateResults, deriveMemoryState, determineTier, ensureSchema, estimateTokens, extractQueryTokens, ftsSearch, generateMemoryId, getRelationsByMemory, keywordsToString, normalizeScores, nowIso, rowToMemory, rrfMerge, rrfMergeWeighted, searchMemories, serializeEmbedding, storeBatchAsync, storeBatchInTransaction, storeMemory, storeMemoryAsync, stringToKeywords, updateAccessTracking, vectorSearch };

package/dist/index.js

@@ -1,3 +1,13 @@
+// src/errors.ts
+var MemoryNotFoundError = class extends Error {
+  memoryId;
+  constructor(memoryId) {
+    super(`Memory not found: ${memoryId}`);
+    this.name = "MemoryNotFoundError";
+    this.memoryId = memoryId;
+  }
+};
+
 // src/schema.ts
 var SCHEMA_V1 = `
 -- memories table
@@ -443,11 +453,11 @@ function createRelationStatements(db) {
 function createRelation(db, stmts, srcMemoryId, dstMemoryId, relationType, reason) {
   const srcExists = stmts.checkMemoryExists.get(srcMemoryId);
   if (!srcExists) {
-    throw new Error(`Source memory not found: ${srcMemoryId}`);
+    throw new MemoryNotFoundError(srcMemoryId);
   }
   const dstExists = stmts.checkMemoryExists.get(dstMemoryId);
   if (!dstExists) {
-    throw new Error(`Destination memory not found: ${dstMemoryId}`);
+    throw new MemoryNotFoundError(dstMemoryId);
   }
   const createdAt = nowIso();
   stmts.insertRelation.run(
@@ -792,7 +802,7 @@ function vectorSearch(db, queryVec, userId, limit, filters, useVecTable = true)
   if (!useVecTable) {
     return bruteForceVectorSearch(db, queryVec, userId, limit, filters);
   }
-  const candidateLimit = Math.min(limit * 5, 200);
+  const candidateLimit = Math.min(limit * 10, 500);
   const vecBuf = Buffer.from(queryVec.buffer, queryVec.byteOffset, queryVec.byteLength);
   let candidates;
   try {
@@ -802,7 +812,10 @@ function vectorSearch(db, queryVec, userId, limit, filters, useVecTable = true)
       WHERE embedding MATCH ?
       AND k = ?
     `).all(vecBuf, candidateLimit);
-  } catch {
+  } catch (err) {
+    const msg = err instanceof Error ? err.message : String(err);
+    process.stderr.write(`[memrosetta] vec_memories KNN unavailable, using brute-force fallback: ${msg}
+`);
     return bruteForceVectorSearch(db, queryVec, userId, limit, filters);
   }
   if (candidates.length === 0) return [];
@@ -863,6 +876,9 @@ function vectorSearch(db, queryVec, userId, limit, filters, useVecTable = true)
     params.push(filters.eventDateRange.end);
   }
   const rows = db.prepare(sql).all(...params);
+  if (rows.length < limit) {
+    return bruteForceVectorSearch(db, queryVec, userId, limit, filters);
+  }
   return rows.map((row) => ({
     memory: rowToMemory(row),
     distance: distanceMap.get(row.id) ?? Infinity
@@ -1726,6 +1742,7 @@ function createEngine(options) {
 }
 export {
   DEFAULT_TIER_CONFIG,
+  MemoryNotFoundError,
   SqliteMemoryEngine,
   applyKeywordBoost,
   applyThreeFactorReranking,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@memrosetta/core",
-  "version": "0.2.22",
+  "version": "0.4.0",
   "type": "module",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",
@@ -20,10 +20,10 @@
     "better-sqlite3": "^11.0.0",
     "nanoid": "^5.0.0",
     "sqlite-vec": "^0.1.7",
-    "@memrosetta/types": "0.2.21"
+    "@memrosetta/types": "0.4.0"
   },
   "optionalDependencies": {
-    "@memrosetta/embeddings": "0.2.21"
+    "@memrosetta/embeddings": "0.3.0"
   },
   "devDependencies": {
     "@types/better-sqlite3": "^7.6.0",