@mastra/pg 1.8.6 → 1.9.0

package/dist/index.js

@@ -8463,17 +8463,34 @@ var MemoryPG = class _MemoryPG extends MemoryStorage {
       );
     }
   }
-  async getObservationalMemoryHistory(threadId, resourceId, limit = 10) {
+  async getObservationalMemoryHistory(threadId, resourceId, limit = 10, options) {
     try {
       const lookupKey = this.getOMKey(threadId, resourceId);
       const tableName = getTableName3({
         indexName: OM_TABLE,
         schemaName: getSchemaName3(this.#schema)
       });
-      const result = await this.#db.client.manyOrNone(
-        `SELECT * FROM ${tableName} WHERE "lookupKey" = $1 ORDER BY "generationCount" DESC LIMIT $2`,
-        [lookupKey, limit]
-      );
+      const conditions = [`"lookupKey" = $1`];
+      const params = [lookupKey];
+      let paramIndex = 2;
+      if (options?.from) {
+        conditions.push(`"createdAtZ" >= $${paramIndex}`);
+        params.push(options.from.toISOString());
+        paramIndex++;
+      }
+      if (options?.to) {
+        conditions.push(`"createdAtZ" <= $${paramIndex}`);
+        params.push(options.to.toISOString());
+        paramIndex++;
+      }
+      params.push(limit);
+      let sql = `SELECT * FROM ${tableName} WHERE ${conditions.join(" AND ")} ORDER BY "generationCount" DESC LIMIT $${paramIndex}`;
+      paramIndex++;
+      if (options?.offset != null) {
+        params.push(options.offset);
+        sql += ` OFFSET $${paramIndex}`;
+      }
+      const result = await this.#db.client.manyOrNone(sql, params);
       if (!result) return [];
       return result.map((row) => this.parseOMRow(row));
     } catch (error) {
@@ -9026,6 +9043,45 @@ var MemoryPG = class _MemoryPG extends MemoryStorage {
       );
     }
   }
+  async updateObservationalMemoryConfig(input) {
+    try {
+      const tableName = getTableName3({
+        indexName: OM_TABLE,
+        schemaName: getSchemaName3(this.#schema)
+      });
+      const selectResult = await this.#db.client.query(`SELECT config FROM ${tableName} WHERE id = $1`, [input.id]);
+      if (selectResult.rowCount === 0) {
+        throw new MastraError({
+          id: createStorageErrorId("PG", "UPDATE_OM_CONFIG", "NOT_FOUND"),
+          text: `Observational memory record not found: ${input.id}`,
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { id: input.id }
+        });
+      }
+      const row = selectResult.rows[0];
+      const existing = row.config ? typeof row.config === "string" ? JSON.parse(row.config) : row.config : {};
+      const merged = this.deepMergeConfig(existing, input.config);
+      const nowStr = (/* @__PURE__ */ new Date()).toISOString();
+      await this.#db.client.query(
+        `UPDATE ${tableName} SET config = $1, "updatedAt" = $2, "updatedAtZ" = $3 WHERE id = $4`,
+        [JSON.stringify(merged), nowStr, nowStr, input.id]
+      );
+    } catch (error) {
+      if (error instanceof MastraError) {
+        throw error;
+      }
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "UPDATE_OM_CONFIG", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY,
+          details: { id: input.id }
+        },
+        error
+      );
+    }
+  }
   // ============================================
   // Async Buffering Methods
   // ============================================