opencode-mastra-om 0.1.5 → 0.1.6

package/dist/index.js

@@ -180045,7 +180045,7 @@ ${OBSERVATION_CONTINUATION_HINT}`);
         }
       }),
       om_observe: tool5({
-        description: "Manually trigger an observation cycle right now, without waiting for the token threshold.",
+        description: "Manually trigger an observation cycle right now, without waiting for the token threshold. Automatically chunks large message sets if chunkTokens is configured.",
         args: {},
         async execute(_args, context2) {
           const threadId = context2.sessionID;
@@ -180056,8 +180056,54 @@ ${OBSERVATION_CONTINUATION_HINT}`);
               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.";
+            const chunkTokens = config2.chunkTokens;
+            if (!chunkTokens) {
+              await runObserve(threadId, mastraMessages);
+              return "Observation cycle triggered. Check om_status for results.";
+            }
+            const tokenCounter = new TokenCounter;
+            const chunks = [];
+            let currentChunk = [];
+            let currentTokens = 0;
+            for (const msg of mastraMessages) {
+              const msgTokens = tokenCounter.countMessages([msg]);
+              if (currentTokens + msgTokens > chunkTokens && currentChunk.length > 0) {
+                chunks.push(currentChunk);
+                currentChunk = [msg];
+                currentTokens = msgTokens;
+              } else {
+                currentChunk.push(msg);
+                currentTokens += msgTokens;
+              }
+            }
+            if (currentChunk.length > 0)
+              chunks.push(currentChunk);
+            if (chunks.length === 1) {
+              await runObserve(threadId, mastraMessages);
+              return "Observation cycle triggered (single chunk). Check om_status for results.";
+            }
+            omLog(`[observe] chunked into ${chunks.length} chunks of ~${chunkTokens} tokens each`);
+            ctx.client.tui.showToast({
+              body: { title: "Mastra OM", message: `Observing in ${chunks.length} chunks...`, variant: "info", duration: 5000 }
+            });
+            const refThreshold = resolveThreshold(omOptions.reflection?.observationTokens ?? 60000);
+            const reflectAt = Math.floor(refThreshold * 0.8);
+            for (let i = 0;i < chunks.length; i++) {
+              omLog(`[observe] processing chunk ${i + 1}/${chunks.length} (${chunks[i].length} messages)`);
+              await runObserve(threadId, chunks[i]);
+              if (i < chunks.length - 1) {
+                const record3 = await om.getRecord(threadId);
+                const obsTokens = record3?.observationTokenCount ?? 0;
+                if (obsTokens >= reflectAt) {
+                  omLog(`[observe] observations at ${obsTokens} tokens (>= ${reflectAt}), reflecting before next chunk`);
+                  ctx.client.tui.showToast({
+                    body: { title: "Mastra OM", message: `Reflecting between chunks (${i + 1}/${chunks.length})...`, variant: "info", duration: 5000 }
+                  });
+                  await om.reflect(threadId);
+                }
+              }
+            }
+            return `Observation complete — processed ${chunks.length} chunks, ${mastraMessages.length} messages total. Check om_status for results.`;
           } catch (err) {
             const msg = err instanceof Error ? err.message : String(err);
             lastError = msg;
@@ -180155,6 +180201,43 @@ ${OBSERVATION_CONTINUATION_HINT}`);
           }
         }
       }),
+      om_reset: tool5({
+        description: "Reset observational memory for this session to a clean slate. Backs up current state first so it can be restored via om_restore.",
+        args: {},
+        async execute(_args, context2) {
+          const threadId = context2.sessionID;
+          try {
+            const db = store.turso;
+            if (!db)
+              return "Raw DB access unavailable.";
+            await backupObservations(threadId, "pre-reset");
+            await db.execute({
+              sql: `UPDATE mastra_observational_memory SET
+                      activeObservations = '',
+                      generationCount = 0,
+                      observationTokenCount = 0,
+                      lastObservedAt = NULL,
+                      lastReflectionAt = NULL,
+                      pendingMessageTokens = 0,
+                      observedMessageIds = '[]',
+                      bufferedObservations = NULL,
+                      bufferedObservationTokens = 0,
+                      bufferedMessageIds = NULL,
+                      bufferedReflection = NULL,
+                      bufferedReflectionTokens = 0,
+                      bufferedReflectionInputTokens = 0,
+                      reflectedObservationLineCount = 0
+                    WHERE lookupKey = ?`,
+              args: [threadId]
+            });
+            omLog(`[reset] observations cleared for ${threadId}`);
+            return "✅ Observational memory reset. Previous state saved to backup slot 1 — use om_restore to recover if needed.";
+          } catch (err) {
+            const msg = err instanceof Error ? err.message : String(err);
+            return `Reset 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.5",
+  "version": "0.1.6",
   "type": "module",
   "license": "MIT",
   "description": "Enhanced Mastra Observational Memory plugin for OpenCode — persistent cross-session memory with observation, reflection, and manual trigger tools",