@mastra/libsql 1.7.1 → 1.7.2-alpha.0

package/dist/index.js

@@ -2985,7 +2985,11 @@ var DatasetsLibSQL = class extends DatasetsStorage {
     await this.#db.createTable({ tableName: TABLE_DATASET_ITEMS, schema: DATASET_ITEMS_SCHEMA });
     await this.#db.createTable({ tableName: TABLE_DATASET_VERSIONS, schema: DATASET_VERSIONS_SCHEMA });
     await this.#addColumnIfNotExists(TABLE_DATASETS, "requestContextSchema", "TEXT");
+    await this.#addColumnIfNotExists(TABLE_DATASETS, "tags", "TEXT");
+    await this.#addColumnIfNotExists(TABLE_DATASETS, "targetType", "TEXT");
+    await this.#addColumnIfNotExists(TABLE_DATASETS, "targetIds", "TEXT");
     await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "requestContext", "TEXT");
+    await this.#addColumnIfNotExists(TABLE_DATASET_ITEMS, "source", "TEXT");
     await this.#client.execute({
       sql: `CREATE INDEX IF NOT EXISTS idx_dataset_items_dataset_validto ON "${TABLE_DATASET_ITEMS}" ("datasetId", "validTo")`,
       args: []
@@ -3028,6 +3032,9 @@ var DatasetsLibSQL = class extends DatasetsStorage {
       inputSchema: row.inputSchema ? safelyParseJSON(row.inputSchema) : void 0,
       groundTruthSchema: row.groundTruthSchema ? safelyParseJSON(row.groundTruthSchema) : void 0,
       requestContextSchema: row.requestContextSchema ? safelyParseJSON(row.requestContextSchema) : void 0,
+      tags: row.tags ? safelyParseJSON(row.tags) : void 0,
+      targetType: row.targetType || void 0,
+      targetIds: row.targetIds ? safelyParseJSON(row.targetIds) : void 0,
       version: row.version,
       createdAt: ensureDate(row.createdAt),
       updatedAt: ensureDate(row.updatedAt)
@@ -3042,6 +3049,7 @@ var DatasetsLibSQL = class extends DatasetsStorage {
       groundTruth: row.groundTruth ? safelyParseJSON(row.groundTruth) : void 0,
       requestContext: row.requestContext ? safelyParseJSON(row.requestContext) : void 0,
       metadata: row.metadata ? safelyParseJSON(row.metadata) : void 0,
+      source: row.source ? safelyParseJSON(row.source) : void 0,
       createdAt: ensureDate(row.createdAt),
       updatedAt: ensureDate(row.updatedAt)
     };
@@ -3057,6 +3065,7 @@ var DatasetsLibSQL = class extends DatasetsStorage {
       groundTruth: row.groundTruth ? safelyParseJSON(row.groundTruth) : void 0,
       requestContext: row.requestContext ? safelyParseJSON(row.requestContext) : void 0,
       metadata: row.metadata ? safelyParseJSON(row.metadata) : void 0,
+      source: row.source ? safelyParseJSON(row.source) : void 0,
       createdAt: ensureDate(row.createdAt),
       updatedAt: ensureDate(row.updatedAt)
     };
@@ -3085,6 +3094,8 @@ var DatasetsLibSQL = class extends DatasetsStorage {
           inputSchema: input.inputSchema ?? null,
           groundTruthSchema: input.groundTruthSchema ?? null,
           requestContextSchema: input.requestContextSchema ?? null,
+          targetType: input.targetType ?? null,
+          targetIds: input.targetIds ? JSON.stringify(input.targetIds) : null,
           version: 0,
           createdAt: nowIso,
           updatedAt: nowIso
@@ -3098,6 +3109,8 @@ var DatasetsLibSQL = class extends DatasetsStorage {
         inputSchema: input.inputSchema ?? void 0,
         groundTruthSchema: input.groundTruthSchema ?? void 0,
         requestContextSchema: input.requestContextSchema ?? void 0,
+        targetType: input.targetType ?? void 0,
+        targetIds: input.targetIds ?? void 0,
         version: 0,
         createdAt: now,
         updatedAt: now
@@ -3169,6 +3182,18 @@ var DatasetsLibSQL = class extends DatasetsStorage {
         updates.push("requestContextSchema = ?");
         values.push(args.requestContextSchema === null ? null : JSON.stringify(args.requestContextSchema));
       }
+      if (args.tags !== void 0) {
+        updates.push("tags = ?");
+        values.push(args.tags === null ? null : JSON.stringify(args.tags));
+      }
+      if (args.targetType !== void 0) {
+        updates.push("targetType = ?");
+        values.push(args.targetType === null ? null : args.targetType);
+      }
+      if (args.targetIds !== void 0) {
+        updates.push("targetIds = ?");
+        values.push(args.targetIds === null ? null : JSON.stringify(args.targetIds));
+      }
       values.push(args.id);
       await this.#client.execute({
         sql: `UPDATE ${TABLE_DATASETS} SET ${updates.join(", ")} WHERE id = ?`,
@@ -3182,6 +3207,9 @@ var DatasetsLibSQL = class extends DatasetsStorage {
         inputSchema: (args.inputSchema !== void 0 ? args.inputSchema : existing.inputSchema) ?? void 0,
         groundTruthSchema: (args.groundTruthSchema !== void 0 ? args.groundTruthSchema : existing.groundTruthSchema) ?? void 0,
         requestContextSchema: (args.requestContextSchema !== void 0 ? args.requestContextSchema : existing.requestContextSchema) ?? void 0,
+        tags: (args.tags !== void 0 ? args.tags : existing.tags) ?? void 0,
+        targetType: (args.targetType !== void 0 ? args.targetType : existing.targetType) ?? void 0,
+        targetIds: (args.targetIds !== void 0 ? args.targetIds : existing.targetIds) ?? void 0,
         updatedAt: new Date(now)
       };
     } catch (error) {
@@ -3287,7 +3315,7 @@ var DatasetsLibSQL = class extends DatasetsStorage {
             args: [args.datasetId]
           },
           {
-            sql: `INSERT INTO ${TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, validTo, isDeleted, input, groundTruth, requestContext, metadata, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${TABLE_DATASETS} WHERE id = ?), NULL, 0, jsonb(?), jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
+            sql: `INSERT INTO ${TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, validTo, isDeleted, input, groundTruth, requestContext, metadata, source, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${TABLE_DATASETS} WHERE id = ?), NULL, 0, jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
             args: [
               id,
               args.datasetId,
@@ -3296,6 +3324,7 @@ var DatasetsLibSQL = class extends DatasetsStorage {
               jsonbArg(args.groundTruth),
               jsonbArg(args.requestContext),
               jsonbArg(args.metadata),
+              jsonbArg(args.source),
               nowIso,
               nowIso
             ]
@@ -3316,6 +3345,7 @@ var DatasetsLibSQL = class extends DatasetsStorage {
         groundTruth: args.groundTruth,
         requestContext: args.requestContext,
         metadata: args.metadata,
+        source: args.source,
         createdAt: now,
         updatedAt: now
       };
@@ -3357,6 +3387,7 @@ var DatasetsLibSQL = class extends DatasetsStorage {
       const mergedGroundTruth = args.groundTruth ?? existing.groundTruth;
       const mergedRequestContext = args.requestContext ?? existing.requestContext;
       const mergedMetadata = args.metadata ?? existing.metadata;
+      const mergedSource = args.source ?? existing.source;
       const results = await this.#client.batch(
         [
           {
@@ -3368,7 +3399,7 @@ var DatasetsLibSQL = class extends DatasetsStorage {
             args: [args.datasetId, args.id]
           },
           {
-            sql: `INSERT INTO ${TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, validTo, isDeleted, input, groundTruth, requestContext, metadata, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${TABLE_DATASETS} WHERE id = ?), NULL, 0, jsonb(?), jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
+            sql: `INSERT INTO ${TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, validTo, isDeleted, input, groundTruth, requestContext, metadata, source, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${TABLE_DATASETS} WHERE id = ?), NULL, 0, jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
             args: [
               args.id,
               args.datasetId,
@@ -3377,6 +3408,7 @@ var DatasetsLibSQL = class extends DatasetsStorage {
               jsonbArg(mergedGroundTruth),
               jsonbArg(mergedRequestContext),
               jsonbArg(mergedMetadata),
+              jsonbArg(mergedSource),
               existing.createdAt.toISOString(),
               nowIso
             ]
@@ -3396,6 +3428,7 @@ var DatasetsLibSQL = class extends DatasetsStorage {
         groundTruth: mergedGroundTruth,
         requestContext: mergedRequestContext,
         metadata: mergedMetadata,
+        source: mergedSource,
         updatedAt: now
       };
     } catch (error) {
@@ -3435,7 +3468,7 @@ var DatasetsLibSQL = class extends DatasetsStorage {
             args: [datasetId, id]
           },
           {
-            sql: `INSERT INTO ${TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, validTo, isDeleted, input, groundTruth, requestContext, metadata, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${TABLE_DATASETS} WHERE id = ?), NULL, 1, jsonb(?), jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
+            sql: `INSERT INTO ${TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, validTo, isDeleted, input, groundTruth, requestContext, metadata, source, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${TABLE_DATASETS} WHERE id = ?), NULL, 1, jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
             args: [
               id,
               datasetId,
@@ -3444,6 +3477,7 @@ var DatasetsLibSQL = class extends DatasetsStorage {
               jsonbArg(existing.groundTruth),
               jsonbArg(existing.requestContext),
               jsonbArg(existing.metadata),
+              jsonbArg(existing.source),
               existing.createdAt.toISOString(),
               nowIso
             ]
@@ -3723,7 +3757,7 @@ var DatasetsLibSQL = class extends DatasetsStorage {
         const id = crypto.randomUUID();
         items.push({ id, input: itemInput });
         statements.push({
-          sql: `INSERT INTO ${TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, validTo, isDeleted, input, groundTruth, requestContext, metadata, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${TABLE_DATASETS} WHERE id = ?), NULL, 0, jsonb(?), jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
+          sql: `INSERT INTO ${TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, validTo, isDeleted, input, groundTruth, requestContext, metadata, source, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${TABLE_DATASETS} WHERE id = ?), NULL, 0, jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
           args: [
             id,
             input.datasetId,
@@ -3732,6 +3766,7 @@ var DatasetsLibSQL = class extends DatasetsStorage {
             jsonbArg(itemInput.groundTruth),
             jsonbArg(itemInput.requestContext),
             jsonbArg(itemInput.metadata),
+            jsonbArg(itemInput.source),
             nowIso,
             nowIso
           ]
@@ -3751,6 +3786,7 @@ var DatasetsLibSQL = class extends DatasetsStorage {
         groundTruth: itemInput.groundTruth,
         requestContext: itemInput.requestContext,
         metadata: itemInput.metadata,
+        source: itemInput.source,
         createdAt: now,
         updatedAt: now
       }));
@@ -3799,7 +3835,7 @@ var DatasetsLibSQL = class extends DatasetsStorage {
           args: [input.datasetId, item.id]
         });
         statements.push({
-          sql: `INSERT INTO ${TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, validTo, isDeleted, input, groundTruth, requestContext, metadata, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${TABLE_DATASETS} WHERE id = ?), NULL, 1, jsonb(?), jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
+          sql: `INSERT INTO ${TABLE_DATASET_ITEMS} (id, datasetId, datasetVersion, validTo, isDeleted, input, groundTruth, requestContext, metadata, source, createdAt, updatedAt) VALUES (?, ?, (SELECT version FROM ${TABLE_DATASETS} WHERE id = ?), NULL, 1, jsonb(?), jsonb(?), jsonb(?), jsonb(?), jsonb(?), ?, ?)`,
           args: [
             item.id,
             input.datasetId,
@@ -3808,6 +3844,7 @@ var DatasetsLibSQL = class extends DatasetsStorage {
             jsonbArg(item.groundTruth),
             jsonbArg(item.requestContext),
             jsonbArg(item.metadata),
+            jsonbArg(item.source),
             item.createdAt.toISOString(),
             nowIso
           ]
@@ -3900,6 +3937,8 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
       completedAt: ensureDate(row.completedAt),
       retryCount: row.retryCount,
       traceId: row.traceId ?? null,
+      status: row.status ?? null,
+      tags: row.tags ? safelyParseJSON(row.tags) : null,
       createdAt: ensureDate(row.createdAt)
     };
   }
@@ -4143,6 +4182,8 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
           completedAt: input.completedAt.toISOString(),
           retryCount: input.retryCount,
           traceId: input.traceId ?? null,
+          status: input.status ?? null,
+          tags: input.tags !== void 0 && input.tags !== null ? JSON.stringify(input.tags) : null,
           createdAt: nowIso
         }
       });
@@ -4159,6 +4200,8 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
         completedAt: input.completedAt,
         retryCount: input.retryCount,
         traceId: input.traceId ?? null,
+        status: input.status ?? null,
+        tags: input.tags ?? null,
         createdAt: now
       };
     } catch (error) {
@@ -4172,6 +4215,70 @@ var ExperimentsLibSQL = class extends ExperimentsStorage {
       );
     }
   }
+  async updateExperimentResult(input) {
+    try {
+      const setClauses = [];
+      const values = [];
+      if (input.status !== void 0) {
+        setClauses.push(`"status" = ?`);
+        values.push(input.status);
+      }
+      if (input.tags !== void 0) {
+        setClauses.push(`"tags" = ?`);
+        values.push(JSON.stringify(input.tags));
+      }
+      if (setClauses.length === 0) {
+        const existing = await this.getExperimentResultById({ id: input.id });
+        if (!existing) {
+          throw new MastraError({
+            id: createStorageErrorId("LIBSQL", "UPDATE_EXPERIMENT_RESULT", "NOT_FOUND"),
+            domain: ErrorDomain.STORAGE,
+            category: ErrorCategory.USER,
+            details: { resultId: input.id }
+          });
+        }
+        return existing;
+      }
+      values.push(input.id);
+      let whereClause = `"id" = ?`;
+      if (input.experimentId) {
+        values.push(input.experimentId);
+        whereClause += ` AND "experimentId" = ?`;
+      }
+      const updateResult = await this.#client.execute({
+        sql: `UPDATE ${TABLE_EXPERIMENT_RESULTS} SET ${setClauses.join(", ")} WHERE ${whereClause}`,
+        args: values
+      });
+      if (updateResult.rowsAffected === 0) {
+        throw new MastraError({
+          id: createStorageErrorId("LIBSQL", "UPDATE_EXPERIMENT_RESULT", "NOT_FOUND"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: { resultId: input.id, ...input.experimentId ? { experimentId: input.experimentId } : {} }
+        });
+      }
+      const result = await this.getExperimentResultById({ id: input.id });
+      if (!result) {
+        throw new MastraError({
+          id: createStorageErrorId("LIBSQL", "UPDATE_EXPERIMENT_RESULT", "NOT_FOUND"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: { resultId: input.id }
+        });
+      }
+      return result;
+    } catch (error) {
+      if (error instanceof MastraError) throw error;
+      throw new MastraError(
+        {
+          id: createStorageErrorId("LIBSQL", "UPDATE_EXPERIMENT_RESULT", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
+  }
   async getExperimentResultById(args) {
     try {
       const result = await this.#client.execute({