claude-memory-layer 1.0.34 → 1.0.36

package/dist/server/index.js

@@ -3590,6 +3590,16 @@ var SQLiteEventStore = class {
   getDatabase() {
     return this.db;
   }
+  hasTableColumn(tableName, columnName) {
+    if (!/^[A-Za-z_][A-Za-z0-9_]*$/.test(tableName))
+      return false;
+    try {
+      const rows = sqliteAll(this.db, `PRAGMA table_info("${tableName}")`, []);
+      return rows.some((row) => row.name === columnName);
+    } catch {
+      return false;
+    }
+  }
   async recordRetrievalTrace(input) {
     await this.initialize();
     const traceId = randomUUID5();
@@ -3655,17 +3665,18 @@ var SQLiteEventStore = class {
   async getRetrievalTraceStats() {
     await this.initialize();
     try {
+      const rewrittenQueryRewriteKindSql = this.hasTableColumn("retrieval_traces", "query_rewrite_kind") ? REWRITTEN_QUERY_REWRITE_KIND_SQL : "0";
       const row = sqliteGet(
         this.db,
         `SELECT
           COUNT(*) as total_queries,
           AVG(candidate_count) as avg_candidate_count,
           AVG(selected_count) as avg_selected_count,
-          SUM(CASE WHEN ${REWRITTEN_QUERY_REWRITE_KIND_SQL} THEN 1 ELSE 0 END) as rewritten_queries,
-          SUM(CASE WHEN ${REWRITTEN_QUERY_REWRITE_KIND_SQL} AND selected_count > 0 THEN 1 ELSE 0 END) as rewritten_queries_with_selection,
-          SUM(CASE WHEN NOT (${REWRITTEN_QUERY_REWRITE_KIND_SQL}) AND selected_count > 0 THEN 1 ELSE 0 END) as raw_queries_with_selection,
-          AVG(CASE WHEN ${REWRITTEN_QUERY_REWRITE_KIND_SQL} THEN selected_count END) as avg_selected_count_for_rewritten_queries,
-          AVG(CASE WHEN NOT (${REWRITTEN_QUERY_REWRITE_KIND_SQL}) THEN selected_count END) as avg_selected_count_for_raw_queries,
+          SUM(CASE WHEN ${rewrittenQueryRewriteKindSql} THEN 1 ELSE 0 END) as rewritten_queries,
+          SUM(CASE WHEN ${rewrittenQueryRewriteKindSql} AND selected_count > 0 THEN 1 ELSE 0 END) as rewritten_queries_with_selection,
+          SUM(CASE WHEN NOT (${rewrittenQueryRewriteKindSql}) AND selected_count > 0 THEN 1 ELSE 0 END) as raw_queries_with_selection,
+          AVG(CASE WHEN ${rewrittenQueryRewriteKindSql} THEN selected_count END) as avg_selected_count_for_rewritten_queries,
+          AVG(CASE WHEN NOT (${rewrittenQueryRewriteKindSql}) THEN selected_count END) as avg_selected_count_for_raw_queries,
           CASE
             WHEN SUM(candidate_count) > 0 THEN (SUM(selected_count) * 1.0 / SUM(candidate_count))
             ELSE 0
@@ -8068,7 +8079,7 @@ var sessionsRouter = new Hono();
 sessionsRouter.get("/", async (c) => {
   const page = parseInt(c.req.query("page") || "1", 10);
   const pageSize = parseInt(c.req.query("pageSize") || "20", 10);
-  const memoryService = getServiceFromQuery(c);
+  const memoryService = getLightweightServiceFromQuery(c);
   try {
     await memoryService.initialize();
     const recentEvents = await memoryService.getRecentEvents(1e3);
@@ -8112,7 +8123,7 @@ sessionsRouter.get("/", async (c) => {
 });
 sessionsRouter.get("/:id", async (c) => {
   const { id } = c.req.param();
-  const memoryService = getServiceFromQuery(c);
+  const memoryService = getLightweightServiceFromQuery(c);
   try {
     await memoryService.initialize();
     const events = await memoryService.getSessionHistory(id);
@@ -8158,7 +8169,7 @@ eventsRouter.get("/", async (c) => {
   const q = (c.req.query("q") || "").trim().toLowerCase();
   const limit = parseInt(c.req.query("limit") || "100", 10);
   const offset = parseInt(c.req.query("offset") || "0", 10);
-  const memoryService = getServiceFromQuery(c);
+  const memoryService = getLightweightServiceFromQuery(c);
   try {
     await memoryService.initialize();
     let events;
@@ -8214,7 +8225,7 @@ eventsRouter.get("/", async (c) => {
 });
 eventsRouter.get("/:id", async (c) => {
   const { id } = c.req.param();
-  const memoryService = getServiceFromQuery(c);
+  const memoryService = getLightweightServiceFromQuery(c);
   try {
     await memoryService.initialize();
     const recentEvents = await memoryService.getRecentEvents(1e4);
@@ -8253,6 +8264,10 @@ eventsRouter.get("/:id", async (c) => {
 // src/apps/server/api/search.ts
 import { Hono as Hono3 } from "hono";
 var searchRouter = new Hono3();
+function isEmbeddingBackendUnavailable(error) {
+  const message = error instanceof Error ? error.message : String(error);
+  return /model file path or buffer|onnxruntime|transformers|embedding backend/i.test(message);
+}
 searchRouter.post("/", async (c) => {
   const memoryService = getServiceFromQuery(c);
   try {
@@ -8294,15 +8309,41 @@ searchRouter.post("/", async (c) => {
 });
 searchRouter.post("/disclosure", async (c) => {
   let memoryService;
+  let body;
   try {
-    const body = await c.req.json();
+    body = await c.req.json();
     if (!body.query) {
       return c.json({ error: "Query is required" }, 400);
     }
-    memoryService = body.options?.strategy === "fast" ? getLightweightServiceFromQuery(c) : getServiceFromQuery(c);
-    await memoryService.initialize();
-    const result = await memoryService.searchDisclosure(body.query, body.options);
-    return c.json(result);
+    const useFastStrategy = body.options?.strategy === "fast";
+    memoryService = useFastStrategy ? getLightweightServiceFromQuery(c) : getServiceFromQuery(c);
+    try {
+      await memoryService.initialize();
+      const result = await memoryService.searchDisclosure(body.query, body.options);
+      return c.json(result);
+    } catch (error) {
+      if (!useFastStrategy && isEmbeddingBackendUnavailable(error)) {
+        await memoryService.shutdown();
+        memoryService = getLightweightServiceFromQuery(c);
+        await memoryService.initialize();
+        const result = await memoryService.searchDisclosure(body.query, {
+          ...body.options,
+          strategy: "fast"
+        });
+        return c.json({
+          ...result,
+          meta: {
+            ...result.meta,
+            fallbackApplied: true,
+            fallbackTrace: [
+              ...result.meta.fallbackTrace || [],
+              "fallback:embedding-backend-unavailable:fast"
+            ]
+          }
+        });
+      }
+      throw error;
+    }
   } catch (error) {
     return c.json({ error: error.message }, 500);
   } finally {
@@ -8310,7 +8351,7 @@ searchRouter.post("/disclosure", async (c) => {
   }
 });
 searchRouter.get("/disclosure/:resultId/expand", async (c) => {
-  const memoryService = getServiceFromQuery(c);
+  const memoryService = getLightweightServiceFromQuery(c);
   try {
     const resultId = c.req.param("resultId");
     const rawWindowSize = c.req.query("windowSize");
@@ -8330,7 +8371,7 @@ searchRouter.get("/disclosure/:resultId/expand", async (c) => {
   }
 });
 searchRouter.get("/disclosure/:resultId/source", async (c) => {
-  const memoryService = getServiceFromQuery(c);
+  const memoryService = getLightweightServiceFromQuery(c);
   try {
     const resultId = c.req.param("resultId");
     const result = await memoryService.sourceDisclosure(resultId);