@mastra/pg 1.9.2-alpha.0 → 1.9.2

package/dist/index.js

@@ -10153,6 +10153,51 @@ var ObservabilityPG = class _ObservabilityPG extends ObservabilityStorage {
       );
     }
   }
+  async getTraceLight(args) {
+    const { traceId } = args;
+    try {
+      const tableName = getTableName2({
+        indexName: TABLE_SPANS,
+        schemaName: getSchemaName2(this.#schema)
+      });
+      const spans = await this.#db.client.manyOrNone(
+        `SELECT
+          "traceId", "spanId", "parentSpanId", "name",
+          "entityType", "entityId", "entityName",
+          "spanType", "error", "isEvent",
+          "startedAtZ" as "startedAt", "endedAtZ" as "endedAt",
+          "createdAtZ" as "createdAt", "updatedAtZ" as "updatedAt"
+        FROM ${tableName}
+        WHERE "traceId" = $1
+        ORDER BY "startedAtZ" ASC`,
+        [traceId]
+      );
+      if (!spans || spans.length === 0) {
+        return null;
+      }
+      return {
+        traceId,
+        spans: spans.map(
+          (span) => transformFromSqlRow({
+            tableName: TABLE_SPANS,
+            sqlRow: span
+          })
+        )
+      };
+    } catch (error) {
+      throw new MastraError(
+        {
+          id: createStorageErrorId("PG", "GET_TRACE_LIGHT", "FAILED"),
+          domain: ErrorDomain.STORAGE,
+          category: ErrorCategory.USER,
+          details: {
+            traceId
+          }
+        },
+        error
+      );
+    }
+  }
   async updateSpan(args) {
     const { traceId, spanId, updates } = args;
     try {