@mastra/cloudflare 0.0.3-alpha.0 → 0.0.3-alpha.1

package/dist/_tsup-dts-rollup.d.cts

@@ -151,13 +151,14 @@ declare class CloudflareStore extends MastraStorage {
         tableName: T;
         records: Partial<RecordTypes[T]>[];
     }): Promise<void>;
-    getTraces(_input: {
+    getTraces({ name, scope, page, perPage, attributes, }: {
         name?: string;
         scope?: string;
         page: number;
         perPage: number;
         attributes?: Record<string, string>;
     }): Promise<any[]>;
+    private parseJSON;
     getEvalsByAgentName(_agentName: string, _type?: 'test' | 'live'): Promise<EvalRow[]>;
     getWorkflowRuns(_args?: {
         namespace?: string;
@@ -189,6 +190,23 @@ export declare interface CloudflareWorkersConfig {
     keyPrefix?: string;
 }
 
+export declare const createSampleTrace: (name: string, scope?: string, attributes?: Record<string, string>) => {
+    id: string;
+    parentSpanId: string;
+    traceId: string;
+    name: string;
+    scope: string | undefined;
+    kind: string;
+    status: string;
+    events: string;
+    links: string;
+    attributes: string | undefined;
+    startTime: string;
+    endTime: string;
+    other: string;
+    createdAt: string;
+};
+
 /**
  * Helper to determine if a config is using Workers bindings
  */

package/dist/_tsup-dts-rollup.d.ts

@@ -151,13 +151,14 @@ declare class CloudflareStore extends MastraStorage {
         tableName: T;
         records: Partial<RecordTypes[T]>[];
     }): Promise<void>;
-    getTraces(_input: {
+    getTraces({ name, scope, page, perPage, attributes, }: {
         name?: string;
         scope?: string;
         page: number;
         perPage: number;
         attributes?: Record<string, string>;
     }): Promise<any[]>;
+    private parseJSON;
     getEvalsByAgentName(_agentName: string, _type?: 'test' | 'live'): Promise<EvalRow[]>;
     getWorkflowRuns(_args?: {
         namespace?: string;
@@ -189,6 +190,23 @@ export declare interface CloudflareWorkersConfig {
     keyPrefix?: string;
 }
 
+export declare const createSampleTrace: (name: string, scope?: string, attributes?: Record<string, string>) => {
+    id: string;
+    parentSpanId: string;
+    traceId: string;
+    name: string;
+    scope: string | undefined;
+    kind: string;
+    status: string;
+    events: string;
+    links: string;
+    attributes: string | undefined;
+    startTime: string;
+    endTime: string;
+    other: string;
+    createdAt: string;
+};
+
 /**
  * Helper to determine if a config is using Workers bindings
  */

package/dist/index.cjs

@@ -349,7 +349,7 @@ var CloudflareStore = class extends storage.MastraStorage {
       const threadMessagesKey = this.getThreadMessagesKey(threadId);
       const latestIds = await this.getLastN(threadMessagesKey, limit);
       latestIds.forEach((id) => messageIds.add(id));
-    } catch (error) {
+    } catch {
       console.log(`No message order found for thread ${threadId}, skipping latest messages`);
     }
   }
@@ -438,6 +438,9 @@ var CloudflareStore = class extends storage.MastraStorage {
           throw new Error("Namespace, workflow name, and run ID are required");
         }
         return `${prefix}${tableName}:${record.namespace}:${record.workflow_name}:${record.run_id}`;
+      case storage.TABLE_TRACES:
+        if (!record.id) throw new Error("Trace ID is required");
+        return `${prefix}${tableName}:${record.id}`;
       default:
         throw new Error(`Unsupported table: ${tableName}`);
     }
@@ -527,6 +530,11 @@ var CloudflareStore = class extends storage.MastraStorage {
             throw new Error("Workflow record missing required fields");
           }
           break;
+        case storage.TABLE_TRACES:
+          if (!("id" in recordTyped)) {
+            throw new Error("Trace record missing required fields");
+          }
+          break;
         default:
           throw new Error(`Unknown table type: ${tableName}`);
       }
@@ -907,8 +915,88 @@ var CloudflareStore = class extends storage.MastraStorage {
       throw error;
     }
   }
-  getTraces(_input) {
-    throw new Error("Method not implemented.");
+  async getTraces({
+    name,
+    scope,
+    page = 0,
+    perPage = 100,
+    attributes
+  }) {
+    try {
+      let keys;
+      if (this.bindings) {
+        keys = (await this.listKV(storage.TABLE_TRACES))?.map((k) => k.name) || [];
+      } else {
+        const namespaceId = await this.getNamespaceId(storage.TABLE_TRACES);
+        const result = await this.client.kv.namespaces.keys.list(namespaceId, {
+          prefix: "",
+          limit: 1e3,
+          account_id: this.accountId
+        });
+        keys = result.result?.map((k) => k.name) || [];
+      }
+      const traceRecords = await Promise.all(
+        keys.map(async (key) => {
+          const record = await this.getKV(storage.TABLE_TRACES, key);
+          if (!record) return null;
+          return record;
+        })
+      );
+      let filteredTraces = traceRecords.filter(
+        (record) => record !== null && typeof record === "object"
+      );
+      if (name) {
+        filteredTraces = filteredTraces.filter((record) => record.name?.toLowerCase().startsWith(name.toLowerCase()));
+      }
+      if (scope) {
+        filteredTraces = filteredTraces.filter((record) => record.scope === scope);
+      }
+      if (attributes) {
+        filteredTraces = filteredTraces.filter((record) => {
+          if (!record.attributes) return false;
+          const recordAttrs = this.parseJSON(record.attributes);
+          if (!recordAttrs) return false;
+          return Object.entries(attributes).every(([key, value]) => recordAttrs[key] === value);
+        });
+      }
+      filteredTraces.sort((a, b) => {
+        const dateA = new Date(a.createdAt).getTime();
+        const dateB = new Date(b.createdAt).getTime();
+        return dateB - dateA;
+      });
+      const start = page * perPage;
+      const end = start + perPage;
+      const paginatedTraces = filteredTraces.slice(start, end);
+      return paginatedTraces.map((record) => ({
+        id: record.id,
+        parentSpanId: record.parentSpanId,
+        traceId: record.traceId,
+        name: record.name,
+        scope: record.scope,
+        kind: record.kind,
+        status: this.parseJSON(record.status),
+        events: this.parseJSON(record.events) || [],
+        links: this.parseJSON(record.links) || [],
+        attributes: this.parseJSON(record?.attributes) || {},
+        startTime: record.startTime,
+        endTime: record.endTime,
+        other: this.parseJSON(record.other) || {},
+        createdAt: record.createdAt
+      }));
+    } catch (error) {
+      this.logger.error("Failed to get traces:", { message: error instanceof Error ? error.message : String(error) });
+      return [];
+    }
+  }
+  parseJSON(value) {
+    if (typeof value === "string") {
+      try {
+        return JSON.parse(value);
+      } catch {
+        return value;
+      }
+    }
+    return value;
   }
   getEvalsByAgentName(_agentName, _type) {
     throw new Error("Method not implemented.");

package/dist/index.js

@@ -343,7 +343,7 @@ var CloudflareStore = class extends MastraStorage {
       const threadMessagesKey = this.getThreadMessagesKey(threadId);
       const latestIds = await this.getLastN(threadMessagesKey, limit);
       latestIds.forEach((id) => messageIds.add(id));
-    } catch (error) {
+    } catch {
       console.log(`No message order found for thread ${threadId}, skipping latest messages`);
     }
   }
@@ -432,6 +432,9 @@ var CloudflareStore = class extends MastraStorage {
           throw new Error("Namespace, workflow name, and run ID are required");
         }
         return `${prefix}${tableName}:${record.namespace}:${record.workflow_name}:${record.run_id}`;
+      case TABLE_TRACES:
+        if (!record.id) throw new Error("Trace ID is required");
+        return `${prefix}${tableName}:${record.id}`;
       default:
         throw new Error(`Unsupported table: ${tableName}`);
     }
@@ -521,6 +524,11 @@ var CloudflareStore = class extends MastraStorage {
             throw new Error("Workflow record missing required fields");
           }
           break;
+        case TABLE_TRACES:
+          if (!("id" in recordTyped)) {
+            throw new Error("Trace record missing required fields");
+          }
+          break;
         default:
           throw new Error(`Unknown table type: ${tableName}`);
       }
@@ -901,8 +909,88 @@ var CloudflareStore = class extends MastraStorage {
       throw error;
     }
   }
-  getTraces(_input) {
-    throw new Error("Method not implemented.");
+  async getTraces({
+    name,
+    scope,
+    page = 0,
+    perPage = 100,
+    attributes
+  }) {
+    try {
+      let keys;
+      if (this.bindings) {
+        keys = (await this.listKV(TABLE_TRACES))?.map((k) => k.name) || [];
+      } else {
+        const namespaceId = await this.getNamespaceId(TABLE_TRACES);
+        const result = await this.client.kv.namespaces.keys.list(namespaceId, {
+          prefix: "",
+          limit: 1e3,
+          account_id: this.accountId
+        });
+        keys = result.result?.map((k) => k.name) || [];
+      }
+      const traceRecords = await Promise.all(
+        keys.map(async (key) => {
+          const record = await this.getKV(TABLE_TRACES, key);
+          if (!record) return null;
+          return record;
+        })
+      );
+      let filteredTraces = traceRecords.filter(
+        (record) => record !== null && typeof record === "object"
+      );
+      if (name) {
+        filteredTraces = filteredTraces.filter((record) => record.name?.toLowerCase().startsWith(name.toLowerCase()));
+      }
+      if (scope) {
+        filteredTraces = filteredTraces.filter((record) => record.scope === scope);
+      }
+      if (attributes) {
+        filteredTraces = filteredTraces.filter((record) => {
+          if (!record.attributes) return false;
+          const recordAttrs = this.parseJSON(record.attributes);
+          if (!recordAttrs) return false;
+          return Object.entries(attributes).every(([key, value]) => recordAttrs[key] === value);
+        });
+      }
+      filteredTraces.sort((a, b) => {
+        const dateA = new Date(a.createdAt).getTime();
+        const dateB = new Date(b.createdAt).getTime();
+        return dateB - dateA;
+      });
+      const start = page * perPage;
+      const end = start + perPage;
+      const paginatedTraces = filteredTraces.slice(start, end);
+      return paginatedTraces.map((record) => ({
+        id: record.id,
+        parentSpanId: record.parentSpanId,
+        traceId: record.traceId,
+        name: record.name,
+        scope: record.scope,
+        kind: record.kind,
+        status: this.parseJSON(record.status),
+        events: this.parseJSON(record.events) || [],
+        links: this.parseJSON(record.links) || [],
+        attributes: this.parseJSON(record?.attributes) || {},
+        startTime: record.startTime,
+        endTime: record.endTime,
+        other: this.parseJSON(record.other) || {},
+        createdAt: record.createdAt
+      }));
+    } catch (error) {
+      this.logger.error("Failed to get traces:", { message: error instanceof Error ? error.message : String(error) });
+      return [];
+    }
+  }
+  parseJSON(value) {
+    if (typeof value === "string") {
+      try {
+        return JSON.parse(value);
+      } catch {
+        return value;
+      }
+    }
+    return value;
   }
   getEvalsByAgentName(_agentName, _type) {
     throw new Error("Method not implemented.");

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mastra/cloudflare",
-  "version": "0.0.3-alpha.0",
+  "version": "0.0.3-alpha.1",
   "description": "Cloudflare provider for Mastra - includes db storage capabilities",
   "type": "module",
   "files": [