@mastra/libsql 0.14.3-alpha.0 → 0.15.0-alpha.0

package/dist/index.js

@@ -5,6 +5,7 @@ import { MastraVector } from '@mastra/core/vector';
 import { BaseFilterTranslator } from '@mastra/core/vector/filter';
 import { MastraStorage, StoreOperations, TABLE_WORKFLOW_SNAPSHOT, TABLE_AI_SPANS, ScoresStorage, TABLE_SCORERS, safelyParseJSON, TracesStorage, TABLE_TRACES, WorkflowsStorage, MemoryStorage, resolveMessageLimit, TABLE_MESSAGES, TABLE_THREADS, TABLE_RESOURCES, LegacyEvalsStorage, TABLE_EVALS, ObservabilityStorage, TABLE_SCHEMAS, AI_SPAN_SCHEMA } from '@mastra/core/storage';
 import { MessageList } from '@mastra/core/agent';
+import { saveScorePayloadSchema } from '@mastra/core/scores';
 
 // src/vector/index.ts
 var LibSQLFilterTranslator = class extends BaseFilterTranslator {
@@ -1949,7 +1950,13 @@ var ObservabilityLibSQL = class extends ObservabilityStorage {
   }
   async createAISpan(span) {
     try {
-      return this.operations.insert({ tableName: TABLE_AI_SPANS, record: span });
+      const now = (/* @__PURE__ */ new Date()).toISOString();
+      const record = {
+        ...span,
+        createdAt: now,
+        updatedAt: now
+      };
+      return this.operations.insert({ tableName: TABLE_AI_SPANS, record });
     } catch (error) {
       throw new MastraError(
         {
@@ -2123,12 +2130,13 @@ var ObservabilityLibSQL = class extends ObservabilityStorage {
   }
   async batchCreateAISpans(args) {
     try {
+      const now = (/* @__PURE__ */ new Date()).toISOString();
       return this.operations.batchInsert({
         tableName: TABLE_AI_SPANS,
         records: args.records.map((record) => ({
           ...record,
-          createdAt: (/* @__PURE__ */ new Date()).toISOString(),
-          updatedAt: (/* @__PURE__ */ new Date()).toISOString()
+          createdAt: now,
+          updatedAt: now
         }))
       });
     } catch (error) {
@@ -2669,6 +2677,7 @@ var ScoresLibSQL = class extends ScoresStorage {
     return {
       id: row.id,
       traceId: row.traceId,
+      spanId: row.spanId,
       runId: row.runId,
       scorer: scorerValue,
       score: row.score,
@@ -2703,6 +2712,26 @@ var ScoresLibSQL = class extends ScoresStorage {
     return result.rows?.[0] ? this.transformScoreRow(result.rows[0]) : null;
   }
   async saveScore(score) {
+    let parsedScore;
+    try {
+      parsedScore = saveScorePayloadSchema.parse(score);
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "LIBSQL_STORE_SAVE_SCORE_FAILED_INVALID_SCORE_PAYLOAD",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: {
+            scorer: score.scorer.name,
+            entityId: score.entityId,
+            entityType: score.entityType,
+            traceId: score.traceId || "",
+            spanId: score.spanId || ""
+          }
+        },
+        error
+      );
+    }
     try {
       const id = crypto.randomUUID();
       await this.operations.insert({
@@ -2711,7 +2740,7 @@ var ScoresLibSQL = class extends ScoresStorage {
           id,
           createdAt: (/* @__PURE__ */ new Date()).toISOString(),
           updatedAt: (/* @__PURE__ */ new Date()).toISOString(),
-          ...score
+          ...parsedScore
         }
       });
       const scoreFromDb = await this.getScoreById({ id });
@@ -2757,6 +2786,43 @@ var ScoresLibSQL = class extends ScoresStorage {
       );
     }
   }
+  async getScoresBySpan({
+    traceId,
+    spanId,
+    pagination
+  }) {
+    try {
+      const countSQLResult = await this.client.execute({
+        sql: `SELECT COUNT(*) as count FROM ${TABLE_SCORERS} WHERE traceId = ? AND spanId = ?`,
+        args: [traceId, spanId]
+      });
+      const total = Number(countSQLResult.rows?.[0]?.count ?? 0);
+      const result = await this.client.execute({
+        sql: `SELECT * FROM ${TABLE_SCORERS} WHERE traceId = ? AND spanId = ? ORDER BY createdAt DESC LIMIT ? OFFSET ?`,
+        args: [traceId, spanId, pagination.perPage + 1, pagination.page * pagination.perPage]
+      });
+      const hasMore = result.rows?.length > pagination.perPage;
+      const scores = result.rows?.slice(0, pagination.perPage).map((row) => this.transformScoreRow(row)) ?? [];
+      return {
+        scores,
+        pagination: {
+          total,
+          page: pagination.page,
+          perPage: pagination.perPage,
+          hasMore
+        }
+      };
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: "LIBSQL_STORE_GET_SCORES_BY_SPAN_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.THIRD_PARTY
+        },
+        error
+      );
+    }
+  }
 };
 var TracesLibSQL = class extends TracesStorage {
   client;
@@ -3246,7 +3312,8 @@ var LibSQLStore = class extends MastraStorage {
       hasColumn: true,
       createTable: true,
       deleteMessages: true,
-      aiTracing: true
+      aiTracing: true,
+      getScoresBySpan: true
     };
   }
   async createTable({
@@ -3459,6 +3526,13 @@ var LibSQLStore = class extends MastraStorage {
   async getAITracesPaginated(args) {
     return this.stores.observability.getAITracesPaginated(args);
   }
+  async getScoresBySpan({
+    traceId,
+    spanId,
+    pagination
+  }) {
+    return this.stores.scores.getScoresBySpan({ traceId, spanId, pagination });
+  }
   async batchCreateAISpans(args) {
     return this.stores.observability.batchCreateAISpans(args);
   }