@mastra/libsql 0.14.0 → 0.14.1-alpha.0

package/dist/index.js

@@ -2027,17 +2027,47 @@ var ObservabilityLibSQL = class extends ObservabilityStorage {
   }) {
     const page = pagination?.page ?? 0;
     const perPage = pagination?.perPage ?? 10;
+    const { entityId, entityType, ...actualFilters } = filters || {};
     const filtersWithDateRange = {
-      ...filters,
-      ...buildDateRangeFilter(pagination?.dateRange, "startedAt")
+      ...actualFilters,
+      ...buildDateRangeFilter(pagination?.dateRange, "startedAt"),
+      parentSpanId: null
     };
     const whereClause = prepareWhereClause(filtersWithDateRange, AI_SPAN_SCHEMA);
+    let actualWhereClause = whereClause.sql || "";
+    if (entityId && entityType) {
+      const statement = `name = ?`;
+      let name = "";
+      if (entityType === "workflow") {
+        name = `workflow run: '${entityId}'`;
+      } else if (entityType === "agent") {
+        name = `agent run: '${entityId}'`;
+      } else {
+        const error = new MastraError({
+          id: "LIBSQL_STORE_GET_AI_TRACES_PAGINATED_FAILED",
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: {
+            entityType
+          },
+          text: `Cannot filter by entity type: ${entityType}`
+        });
+        this.logger?.trackException(error);
+        throw error;
+      }
+      whereClause.args.push(name);
+      if (actualWhereClause) {
+        actualWhereClause += ` AND ${statement}`;
+      } else {
+        actualWhereClause += `WHERE ${statement}`;
+      }
+    }
     const orderBy = "startedAt DESC";
     let count = 0;
     try {
       count = await this.operations.loadTotalCount({
         tableName: TABLE_AI_SPANS,
-        whereClause: { sql: whereClause.sql, args: whereClause.args }
+        whereClause: { sql: actualWhereClause, args: whereClause.args }
       });
     } catch (error) {
       throw new MastraError(
@@ -2063,7 +2093,10 @@ var ObservabilityLibSQL = class extends ObservabilityStorage {
     try {
       const spans = await this.operations.loadMany({
         tableName: TABLE_AI_SPANS,
-        whereClause,
+        whereClause: {
+          sql: actualWhereClause,
+          args: whereClause.args
+        },
         orderBy,
         offset: page * perPage,
         limit: perPage