@mastra/libsql 1.6.2 → 1.6.3

package/dist/index.js

@@ -5166,7 +5166,8 @@ var MemoryLibSQL = class extends MemoryStorage {
           "isBufferingObservation",
           "isBufferingReflection",
           "lastBufferedAtTokens",
-          "lastBufferedAtTime"
+          "lastBufferedAtTime",
+          "metadata"
         ]
       });
     }
@@ -6136,9 +6137,11 @@ var MemoryLibSQL = class extends MemoryStorage {
           ]
         });
         const clonedMessages = [];
+        const messageIdMap = {};
         const targetResourceId = resourceId || sourceThread.resourceId;
         for (const sourceMsg of sourceMessages) {
           const newMessageId = crypto.randomUUID();
+          messageIdMap[sourceMsg.id] = newMessageId;
           const contentStr = sourceMsg.content;
           let parsedContent;
           try {
@@ -6172,7 +6175,8 @@ var MemoryLibSQL = class extends MemoryStorage {
         await tx.commit();
         return {
           thread: newThread,
-          clonedMessages
+          clonedMessages,
+          messageIdMap
         };
       } catch (error) {
         await tx.rollback();
@@ -6362,6 +6366,68 @@ var MemoryLibSQL = class extends MemoryStorage {
       );
     }
   }
+  async insertObservationalMemoryRecord(record) {
+    try {
+      const lookupKey = this.getOMKey(record.threadId, record.resourceId);
+      await this.#client.execute({
+        sql: `INSERT INTO "${OM_TABLE}" (
+          id, "lookupKey", scope, "resourceId", "threadId",
+          "activeObservations", "activeObservationsPendingUpdate",
+          "originType", config, "generationCount", "lastObservedAt", "lastReflectionAt",
+          "pendingMessageTokens", "totalTokensObserved", "observationTokenCount",
+          "observedMessageIds", "bufferedObservationChunks",
+          "bufferedReflection", "bufferedReflectionTokens", "bufferedReflectionInputTokens",
+          "reflectedObservationLineCount",
+          "isObserving", "isReflecting", "isBufferingObservation", "isBufferingReflection",
+          "lastBufferedAtTokens", "lastBufferedAtTime",
+          "observedTimezone", metadata, "createdAt", "updatedAt"
+        ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
+        args: [
+          record.id,
+          lookupKey,
+          record.scope,
+          record.resourceId,
+          record.threadId || null,
+          record.activeObservations || "",
+          null,
+          record.originType || "initial",
+          record.config ? JSON.stringify(record.config) : null,
+          record.generationCount || 0,
+          record.lastObservedAt ? record.lastObservedAt.toISOString() : null,
+          null,
+          record.pendingMessageTokens || 0,
+          record.totalTokensObserved || 0,
+          record.observationTokenCount || 0,
+          record.observedMessageIds ? JSON.stringify(record.observedMessageIds) : null,
+          record.bufferedObservationChunks ? JSON.stringify(record.bufferedObservationChunks) : null,
+          record.bufferedReflection || null,
+          record.bufferedReflectionTokens ?? null,
+          record.bufferedReflectionInputTokens ?? null,
+          record.reflectedObservationLineCount ?? null,
+          record.isObserving || false,
+          record.isReflecting || false,
+          record.isBufferingObservation || false,
+          record.isBufferingReflection || false,
+          record.lastBufferedAtTokens || 0,
+          record.lastBufferedAtTime ? record.lastBufferedAtTime.toISOString() : null,
+          record.observedTimezone || null,
+          record.metadata ? JSON.stringify(record.metadata) : null,
+          record.createdAt.toISOString(),
+          record.updatedAt.toISOString()
+        ]
+      });
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("LIBSQL", "INSERT_OBSERVATIONAL_MEMORY_RECORD", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { id: record.id, threadId: record.threadId, resourceId: record.resourceId }
+        },
+        error
+      );
+    }
+  }
   async updateActiveObservations(input) {
     try {
       const now = /* @__PURE__ */ new Date();
@@ -6446,8 +6512,8 @@ var MemoryLibSQL = class extends MemoryStorage {
           "originType", config, "generationCount", "lastObservedAt", "lastReflectionAt",
           "pendingMessageTokens", "totalTokensObserved", "observationTokenCount",
           "isObserving", "isReflecting", "isBufferingObservation", "isBufferingReflection", "lastBufferedAtTokens", "lastBufferedAtTime",
-          "observedTimezone", "createdAt", "updatedAt"
-        ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
+          "observedTimezone", metadata, "createdAt", "updatedAt"
+        ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
         args: [
           id,
           lookupKey,
@@ -6477,6 +6543,7 @@ var MemoryLibSQL = class extends MemoryStorage {
           null,
           // lastBufferedAtTime
           record.observedTimezone || null,
+          record.metadata ? JSON.stringify(record.metadata) : null,
           now.toISOString(),
           now.toISOString()
         ]
@@ -9274,6 +9341,9 @@ var WorkflowsLibSQL = class extends WorkflowsStorage {
       (err) => this.logger.warn("LibSQL Workflows: Failed to setup PRAGMA settings.", err)
     );
   }
+  supportsConcurrentUpdates() {
+    return true;
+  }
   parseWorkflowRun(row) {
     let parsedSnapshot = row.snapshot;
     if (typeof parsedSnapshot === "string") {