opencode-mastra-om 0.1.3 → 0.1.5

package/dist/index.js

@@ -179825,6 +179825,79 @@ var MastraPlugin = async (ctx) => {
   }
   const om = new ObservationalMemory(omOptions);
   omLog(`[init] ObservationalMemory created, model=${config2.model ?? "default"}`);
+  const backupObservations = async (threadId, trigger) => {
+    try {
+      const record3 = await om.getRecord(threadId);
+      const observations = record3?.activeObservations;
+      if (!observations)
+        return;
+      const db = store.turso;
+      if (!db)
+        return;
+      const snap = {
+        lookupKey: threadId,
+        generationCount: record3.generationCount ?? 0,
+        observations,
+        observationTokenCount: record3.observationTokenCount ?? 0,
+        lastObservedAt: record3.lastObservedAt ?? null,
+        lastReflectionAt: record3.lastReflectionAt ?? null,
+        pendingMessageTokens: record3.pendingMessageTokens ?? 0,
+        observedMessageIds: record3.observedMessageIds ?? "[]",
+        trigger,
+        savedAt: new Date().toISOString()
+      };
+      await db.execute({
+        sql: `INSERT INTO mastra_om_backups
+                (id, lookupKey, slot, generationCount, observations, observationTokenCount,
+                 lastObservedAt, lastReflectionAt, pendingMessageTokens, observedMessageIds, trigger, savedAt)
+              SELECT hex(randomblob(16)), lookupKey, 2, generationCount, observations, observationTokenCount,
+                     lastObservedAt, lastReflectionAt, pendingMessageTokens, observedMessageIds, trigger, savedAt
+              FROM mastra_om_backups WHERE lookupKey = ? AND slot = 1
+              ON CONFLICT(lookupKey, slot) DO UPDATE SET
+                generationCount = excluded.generationCount,
+                observations = excluded.observations,
+                observationTokenCount = excluded.observationTokenCount,
+                lastObservedAt = excluded.lastObservedAt,
+                lastReflectionAt = excluded.lastReflectionAt,
+                pendingMessageTokens = excluded.pendingMessageTokens,
+                observedMessageIds = excluded.observedMessageIds,
+                trigger = excluded.trigger,
+                savedAt = excluded.savedAt`,
+        args: [threadId]
+      });
+      await db.execute({
+        sql: `INSERT INTO mastra_om_backups
+                (id, lookupKey, slot, generationCount, observations, observationTokenCount,
+                 lastObservedAt, lastReflectionAt, pendingMessageTokens, observedMessageIds, trigger, savedAt)
+              VALUES (hex(randomblob(16)), ?, 1, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+              ON CONFLICT(lookupKey, slot) DO UPDATE SET
+                generationCount = excluded.generationCount,
+                observations = excluded.observations,
+                observationTokenCount = excluded.observationTokenCount,
+                lastObservedAt = excluded.lastObservedAt,
+                lastReflectionAt = excluded.lastReflectionAt,
+                pendingMessageTokens = excluded.pendingMessageTokens,
+                observedMessageIds = excluded.observedMessageIds,
+                trigger = excluded.trigger,
+                savedAt = excluded.savedAt`,
+        args: [
+          threadId,
+          snap.generationCount,
+          snap.observations,
+          snap.observationTokenCount,
+          snap.lastObservedAt,
+          snap.lastReflectionAt,
+          snap.pendingMessageTokens,
+          snap.observedMessageIds,
+          snap.trigger,
+          snap.savedAt
+        ]
+      });
+      omLog(`[backup] ${trigger} — saved gen ${snap.generationCount} to slot 1, rotated old slot 1 → slot 2`);
+    } catch (err) {
+      omLog(`[backup] failed: ${err instanceof Error ? err.message : String(err)}`);
+    }
+  };
   setTimeout(() => {
     ctx.client.tui.showToast({
       body: { title: "Mastra OM", message: "Observational Memory active", variant: "success", duration: 3000 }
@@ -179875,6 +179948,7 @@ var MastraPlugin = async (ctx) => {
       try {
         const mastraMessages = convertMessages2(output.messages, sessionId);
         if (mastraMessages.length > 0) {
+          await backupObservations(sessionId, "pre-auto-observe");
           await runObserve(sessionId, mastraMessages);
         }
         const record3 = await om.getRecord(sessionId);
@@ -179981,6 +180055,7 @@ ${OBSERVATION_CONTINUATION_HINT}`);
             if (!resp.data || resp.data.length === 0)
               return "No messages to observe.";
             const mastraMessages = convertMessages2(resp.data, threadId);
+            await backupObservations(threadId, "pre-observe");
             await runObserve(threadId, mastraMessages);
             return "Observation cycle triggered. Check memory_status for results.";
           } catch (err) {
@@ -179997,6 +180072,7 @@ ${OBSERVATION_CONTINUATION_HINT}`);
           const threadId = context2.sessionID;
           await resolveCredentials();
           try {
+            await backupObservations(threadId, "pre-reflect");
             await om.reflect(threadId);
             return "Reflection cycle triggered. Check memory_observations for results.";
           } catch (err) {
@@ -180024,6 +180100,61 @@ ${OBSERVATION_CONTINUATION_HINT}`);
           }
         }
       }),
+      om_restore: tool5({
+        description: "Restore observational memory from backup slot 1 (most recent) or slot 2 (one generation older).",
+        args: { slot: { type: "number", description: "1 = most recent backup, 2 = one generation older" } },
+        async execute(args, context2) {
+          const threadId = context2.sessionID;
+          const slot = args.slot === 2 ? 2 : 1;
+          try {
+            const db = store.turso;
+            if (!db)
+              return "Raw DB access unavailable.";
+            const result = await db.execute({
+              sql: `SELECT * FROM mastra_om_backups WHERE lookupKey = ? AND slot = ?`,
+              args: [threadId, slot]
+            });
+            const row = result.rows?.[0];
+            if (!row)
+              return `No backup found in slot ${slot}.`;
+            await db.execute({
+              sql: `UPDATE mastra_observational_memory SET
+                      activeObservations = ?,
+                      generationCount = ?,
+                      observationTokenCount = ?,
+                      lastObservedAt = ?,
+                      lastReflectionAt = ?,
+                      pendingMessageTokens = ?,
+                      observedMessageIds = ?
+                    WHERE lookupKey = ?`,
+              args: [
+                row.observations,
+                row.generationCount,
+                row.observationTokenCount,
+                row.lastObservedAt,
+                row.lastReflectionAt,
+                row.pendingMessageTokens,
+                row.observedMessageIds,
+                threadId
+              ]
+            });
+            omLog(`[restore] restored slot ${slot} — gen ${row.generationCount}, saved at ${row.savedAt}, trigger=${row.trigger}`);
+            return [
+              `✅ Restored from slot ${slot}`,
+              `  Generation: ${row.generationCount}`,
+              `  Saved at: ${row.savedAt}`,
+              `  Trigger: ${row.trigger}`,
+              `  Observation tokens: ${row.observationTokenCount}`,
+              `  Last observed: ${row.lastObservedAt ?? "never"}`,
+              `  Last reflection: ${row.lastReflectionAt ?? "never"}`
+            ].join(`
+`);
+          } catch (err) {
+            const msg = err instanceof Error ? err.message : String(err);
+            return `Restore failed: ${msg}`;
+          }
+        }
+      }),
       om_config: tool5({
         description: "Show the current Mastra Observational Memory configuration.",
         args: {},

package/package.json

@@ -1,7 +1,7 @@
 {
   "$schema": "https://json.schemastore.org/package.json",
   "name": "opencode-mastra-om",
-  "version": "0.1.3",
+  "version": "0.1.5",
   "type": "module",
   "license": "MIT",
   "description": "Enhanced Mastra Observational Memory plugin for OpenCode — persistent cross-session memory with observation, reflection, and manual trigger tools",
@@ -38,8 +38,8 @@
   },
   "dependencies": {
     "@mastra/core": "^1.21.0",
-    "@mastra/libsql": "^1.7.3",
-    "@mastra/memory": "^1.13.0",
+    "@mastra/libsql": "^1.7.4",
+    "@mastra/memory": "^1.13.1",
     "@opencode-ai/plugin": "^1.3.11",
     "@opencode-ai/sdk": "^1.3.11"
   },