@llmops/sdk 1.0.0-beta.4 → 1.0.0-beta.6

package/dist/store/d1.cjs

@@ -0,0 +1,491 @@
+let node_crypto = require("node:crypto");
+
+//#region src/store/d1/d1-store.ts
+function buildTagFilters(tags) {
+	const conditions = [];
+	const params = [];
+	if (!tags) return {
+		conditions,
+		params
+	};
+	for (const [key, values] of Object.entries(tags)) {
+		if (values.length === 0) continue;
+		if (values.length === 1) {
+			conditions.push(`json_extract("tags", '$.' || ?) = ?`);
+			params.push(key, values[0]);
+		} else {
+			const placeholders = values.map(() => "?").join(", ");
+			conditions.push(`json_extract("tags", '$.' || ?) IN (${placeholders})`);
+			params.push(key, ...values);
+		}
+	}
+	return {
+		conditions,
+		params
+	};
+}
+const D1_BATCH_LIMIT = 100;
+function createD1LLMRequestsStore(db) {
+	return {
+		batchInsertRequests: async (requests) => {
+			if (requests.length === 0) return { count: 0 };
+			const now = (/* @__PURE__ */ new Date()).toISOString();
+			const stmts = requests.map((req) => db.prepare(`
+          INSERT INTO "llm_requests" (
+            "id","requestId","configId","variantId","environmentId",
+            "providerConfigId","provider","model","promptTokens",
+            "completionTokens","totalTokens","cachedTokens",
+            "cacheCreationTokens","cost","cacheSavings","inputCost",
+            "outputCost","endpoint","statusCode","latencyMs","isStreaming",
+            "userId","tags","guardrailResults","traceId","spanId",
+            "parentSpanId","sessionId","createdAt","updatedAt"
+          ) VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?)
+        `).bind((0, node_crypto.randomUUID)(), req.requestId, req.configId ?? null, req.variantId ?? null, req.environmentId ?? null, req.providerConfigId ?? null, req.provider, req.model, req.promptTokens ?? 0, req.completionTokens ?? 0, req.totalTokens ?? 0, req.cachedTokens ?? 0, req.cacheCreationTokens ?? 0, req.cost ?? 0, req.cacheSavings ?? 0, req.inputCost ?? 0, req.outputCost ?? 0, req.endpoint, req.statusCode, req.latencyMs ?? 0, req.isStreaming ? 1 : 0, req.userId ?? null, JSON.stringify(req.tags ?? {}), req.guardrailResults ? JSON.stringify(req.guardrailResults) : null, req.traceId ?? null, req.spanId ?? null, req.parentSpanId ?? null, req.sessionId ?? null, now, now));
+			for (let i = 0; i < stmts.length; i += D1_BATCH_LIMIT) await db.batch(stmts.slice(i, i + D1_BATCH_LIMIT));
+			return { count: requests.length };
+		},
+		insertRequest: async (req) => {
+			const now = (/* @__PURE__ */ new Date()).toISOString();
+			return db.prepare(`
+        INSERT INTO "llm_requests" (
+          "id","requestId","configId","variantId","environmentId",
+          "providerConfigId","provider","model","promptTokens",
+          "completionTokens","totalTokens","cachedTokens",
+          "cacheCreationTokens","cost","cacheSavings","inputCost",
+          "outputCost","endpoint","statusCode","latencyMs","isStreaming",
+          "userId","tags","guardrailResults","traceId","spanId",
+          "parentSpanId","sessionId","createdAt","updatedAt"
+        ) VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?)
+        RETURNING *
+      `).bind((0, node_crypto.randomUUID)(), req.requestId, req.configId ?? null, req.variantId ?? null, req.environmentId ?? null, req.providerConfigId ?? null, req.provider, req.model, req.promptTokens ?? 0, req.completionTokens ?? 0, req.totalTokens ?? 0, req.cachedTokens ?? 0, req.cacheCreationTokens ?? 0, req.cost ?? 0, req.cacheSavings ?? 0, req.inputCost ?? 0, req.outputCost ?? 0, req.endpoint, req.statusCode, req.latencyMs ?? 0, req.isStreaming ? 1 : 0, req.userId ?? null, JSON.stringify(req.tags ?? {}), req.guardrailResults ? JSON.stringify(req.guardrailResults) : null, req.traceId ?? null, req.spanId ?? null, req.parentSpanId ?? null, req.sessionId ?? null, now, now).first();
+		},
+		listRequests: async (params) => {
+			const { limit = 100, offset = 0, configId, variantId, environmentId, providerConfigId, provider, model, startDate, endDate, tags } = params ?? {};
+			const conditions = ["1=1"];
+			const queryParams = [];
+			if (configId) {
+				conditions.push(`"configId" = ?`);
+				queryParams.push(configId);
+			}
+			if (variantId) {
+				conditions.push(`"variantId" = ?`);
+				queryParams.push(variantId);
+			}
+			if (environmentId) {
+				conditions.push(`"environmentId" = ?`);
+				queryParams.push(environmentId);
+			}
+			if (providerConfigId) {
+				conditions.push(`"providerConfigId" = ?`);
+				queryParams.push(providerConfigId);
+			}
+			if (provider) {
+				conditions.push(`"provider" = ?`);
+				queryParams.push(provider);
+			}
+			if (model) {
+				conditions.push(`"model" = ?`);
+				queryParams.push(model);
+			}
+			if (startDate) {
+				conditions.push(`"createdAt" >= ?`);
+				queryParams.push(startDate.toISOString());
+			}
+			if (endDate) {
+				conditions.push(`"createdAt" <= ?`);
+				queryParams.push(endDate.toISOString());
+			}
+			const tagFilter = buildTagFilters(tags);
+			conditions.push(...tagFilter.conditions);
+			queryParams.push(...tagFilter.params);
+			const where = conditions.join(" AND ");
+			const total = (await db.prepare(`SELECT COUNT(*) AS "total" FROM "llm_requests" WHERE ${where}`).bind(...queryParams).first())?.total ?? 0;
+			const { results: data } = await db.prepare(`SELECT * FROM "llm_requests" WHERE ${where} ORDER BY "createdAt" DESC LIMIT ? OFFSET ?`).bind(...queryParams, limit, offset).all();
+			return {
+				data,
+				total,
+				limit,
+				offset
+			};
+		},
+		getRequestByRequestId: async (requestId) => {
+			return db.prepare(`SELECT * FROM "llm_requests" WHERE "requestId" = ?`).bind(requestId).first();
+		},
+		getTotalCost: async (params) => {
+			const conditions = [`"createdAt" >= ?`, `"createdAt" <= ?`];
+			const queryParams = [params.startDate.toISOString(), params.endDate.toISOString()];
+			if (params.configId) {
+				conditions.push(`"configId" = ?`);
+				queryParams.push(params.configId);
+			}
+			if (params.variantId) {
+				conditions.push(`"variantId" = ?`);
+				queryParams.push(params.variantId);
+			}
+			if (params.environmentId) {
+				conditions.push(`"environmentId" = ?`);
+				queryParams.push(params.environmentId);
+			}
+			const tagFilter = buildTagFilters(params.tags);
+			conditions.push(...tagFilter.conditions);
+			queryParams.push(...tagFilter.params);
+			const where = conditions.join(" AND ");
+			return db.prepare(`
+        SELECT
+          COALESCE(SUM("cost"), 0) AS "totalCost",
+          COALESCE(SUM("inputCost"), 0) AS "totalInputCost",
+          COALESCE(SUM("outputCost"), 0) AS "totalOutputCost",
+          COALESCE(SUM("promptTokens"), 0) AS "totalPromptTokens",
+          COALESCE(SUM("completionTokens"), 0) AS "totalCompletionTokens",
+          COALESCE(SUM("totalTokens"), 0) AS "totalTokens",
+          COALESCE(SUM("cachedTokens"), 0) AS "totalCachedTokens",
+          COALESCE(SUM("cacheSavings"), 0) AS "totalCacheSavings",
+          COUNT(*) AS "requestCount"
+        FROM "llm_requests" WHERE ${where}
+      `).bind(...queryParams).first();
+		},
+		getCostByModel: async (params) => {
+			const { results } = await db.prepare(`
+        SELECT "provider", "model",
+          COALESCE(SUM("cost"), 0) AS "totalCost",
+          COALESCE(SUM("inputCost"), 0) AS "totalInputCost",
+          COALESCE(SUM("outputCost"), 0) AS "totalOutputCost",
+          COALESCE(SUM("totalTokens"), 0) AS "totalTokens",
+          COUNT(*) AS "requestCount",
+          AVG("latencyMs") AS "avgLatencyMs"
+        FROM "llm_requests"
+        WHERE "createdAt" >= ? AND "createdAt" <= ?
+        GROUP BY "provider", "model"
+        ORDER BY SUM("cost") DESC
+      `).bind(params.startDate.toISOString(), params.endDate.toISOString()).all();
+			return results;
+		},
+		getCostByProvider: async (params) => {
+			const { results } = await db.prepare(`
+        SELECT "provider",
+          COALESCE(SUM("cost"), 0) AS "totalCost",
+          COALESCE(SUM("inputCost"), 0) AS "totalInputCost",
+          COALESCE(SUM("outputCost"), 0) AS "totalOutputCost",
+          COALESCE(SUM("totalTokens"), 0) AS "totalTokens",
+          COUNT(*) AS "requestCount",
+          AVG("latencyMs") AS "avgLatencyMs"
+        FROM "llm_requests"
+        WHERE "createdAt" >= ? AND "createdAt" <= ?
+        GROUP BY "provider"
+        ORDER BY SUM("cost") DESC
+      `).bind(params.startDate.toISOString(), params.endDate.toISOString()).all();
+			return results;
+		},
+		getDailyCosts: async (params) => {
+			const { results } = await db.prepare(`
+        SELECT date("createdAt") AS "date",
+          COALESCE(SUM("cost"), 0) AS "totalCost",
+          COALESCE(SUM("inputCost"), 0) AS "totalInputCost",
+          COALESCE(SUM("outputCost"), 0) AS "totalOutputCost",
+          COALESCE(SUM("totalTokens"), 0) AS "totalTokens",
+          COUNT(*) AS "requestCount"
+        FROM "llm_requests"
+        WHERE "createdAt" >= ? AND "createdAt" <= ?
+        GROUP BY date("createdAt")
+        ORDER BY date("createdAt") ASC
+      `).bind(params.startDate.toISOString(), params.endDate.toISOString()).all();
+			return results;
+		},
+		getCostSummary: async (params) => {
+			const { startDate, endDate, groupBy, configId, variantId, environmentId, tags, tagKeys } = params;
+			const conditions = [`"createdAt" >= ?`, `"createdAt" <= ?`];
+			const queryParams = [startDate.toISOString(), endDate.toISOString()];
+			if (configId) {
+				conditions.push(`"configId" = ?`);
+				queryParams.push(configId);
+			}
+			if (variantId) {
+				conditions.push(`"variantId" = ?`);
+				queryParams.push(variantId);
+			}
+			if (environmentId) {
+				conditions.push(`"environmentId" = ?`);
+				queryParams.push(environmentId);
+			}
+			const tagFilter = buildTagFilters(tags);
+			conditions.push(...tagFilter.conditions);
+			queryParams.push(...tagFilter.params);
+			const where = conditions.join(" AND ");
+			if (groupBy === "tags") {
+				const tagConditions = [...conditions];
+				const tagParams = [...queryParams];
+				if (tagKeys && tagKeys.length > 0) {
+					tagConditions.push(`json_each.key IN (${tagKeys.map(() => "?").join(",")})`);
+					tagParams.push(...tagKeys);
+				}
+				const tagWhere = tagConditions.join(" AND ");
+				const { results: results$1 } = await db.prepare(`
+          SELECT json_each.key || ':' || json_each.value AS "groupKey",
+                 COALESCE(SUM("cost"), 0) AS "totalCost",
+                 COUNT(*) AS "requestCount"
+          FROM "llm_requests", json_each("tags")
+          WHERE ${tagWhere}
+          GROUP BY json_each.key, json_each.value
+          ORDER BY SUM("cost") DESC
+        `).bind(...tagParams).all();
+				return results$1;
+			}
+			const sqlMap = {
+				day: `SELECT date("createdAt") AS "groupKey", COALESCE(SUM("cost"),0) AS "totalCost", COUNT(*) AS "requestCount", COALESCE(SUM("totalTokens"),0) AS "totalTokens" FROM "llm_requests" WHERE ${where} GROUP BY date("createdAt") ORDER BY date("createdAt") ASC`,
+				hour: `SELECT strftime('%Y-%m-%d %H:00:00',"createdAt") AS "groupKey", COALESCE(SUM("cost"),0) AS "totalCost", COUNT(*) AS "requestCount", COALESCE(SUM("totalTokens"),0) AS "totalTokens" FROM "llm_requests" WHERE ${where} GROUP BY strftime('%Y-%m-%d %H:00:00',"createdAt") ORDER BY strftime('%Y-%m-%d %H:00:00',"createdAt") ASC`,
+				model: `SELECT "provider"||'/'||"model" AS "groupKey", COALESCE(SUM("cost"),0) AS "totalCost", COUNT(*) AS "requestCount" FROM "llm_requests" WHERE ${where} GROUP BY "provider","model" ORDER BY SUM("cost") DESC`,
+				provider: `SELECT "provider" AS "groupKey", COALESCE(SUM("cost"),0) AS "totalCost", COUNT(*) AS "requestCount" FROM "llm_requests" WHERE ${where} GROUP BY "provider" ORDER BY SUM("cost") DESC`,
+				endpoint: `SELECT COALESCE("endpoint",'unknown') AS "groupKey", COALESCE(SUM("cost"),0) AS "totalCost", COUNT(*) AS "requestCount" FROM "llm_requests" WHERE ${where} GROUP BY "endpoint" ORDER BY SUM("cost") DESC`
+			};
+			const totalSql = `SELECT 'total' AS "groupKey", COALESCE(SUM("cost"),0) AS "totalCost", COUNT(*) AS "requestCount" FROM "llm_requests" WHERE ${where}`;
+			const sql = groupBy ? sqlMap[groupBy] ?? totalSql : totalSql;
+			const { results } = await db.prepare(sql).bind(...queryParams).all();
+			return results;
+		},
+		getRequestStats: async (params) => {
+			const conditions = [`"createdAt" >= ?`, `"createdAt" <= ?`];
+			const queryParams = [params.startDate.toISOString(), params.endDate.toISOString()];
+			if (params.configId) {
+				conditions.push(`"configId" = ?`);
+				queryParams.push(params.configId);
+			}
+			if (params.variantId) {
+				conditions.push(`"variantId" = ?`);
+				queryParams.push(params.variantId);
+			}
+			if (params.environmentId) {
+				conditions.push(`"environmentId" = ?`);
+				queryParams.push(params.environmentId);
+			}
+			const tagFilter = buildTagFilters(params.tags);
+			conditions.push(...tagFilter.conditions);
+			queryParams.push(...tagFilter.params);
+			const where = conditions.join(" AND ");
+			return db.prepare(`
+        SELECT
+          COUNT(*) AS "totalRequests",
+          COUNT(CASE WHEN "statusCode">=200 AND "statusCode"<300 THEN 1 END) AS "successfulRequests",
+          COUNT(CASE WHEN "statusCode">=400 THEN 1 END) AS "failedRequests",
+          COUNT(CASE WHEN "isStreaming"=1 THEN 1 END) AS "streamingRequests",
+          AVG("latencyMs") AS "avgLatencyMs",
+          MAX("latencyMs") AS "maxLatencyMs",
+          MIN("latencyMs") AS "minLatencyMs"
+        FROM "llm_requests" WHERE ${where}
+      `).bind(...queryParams).first();
+		},
+		getDistinctTags: async () => {
+			const { results } = await db.prepare(`
+        SELECT DISTINCT json_each.key AS key, json_each.value AS value
+        FROM "llm_requests", json_each("tags")
+        WHERE "tags" != '{}'
+        ORDER BY json_each.key, json_each.value
+      `).all();
+			return results;
+		}
+	};
+}
+function createD1TracesStore(db) {
+	return {
+		upsertTrace: async (data) => {
+			const now = (/* @__PURE__ */ new Date()).toISOString();
+			await db.prepare(`
+        INSERT INTO "traces" (
+          "id","traceId","name","sessionId","userId","status",
+          "startTime","endTime","durationMs","spanCount",
+          "totalInputTokens","totalOutputTokens","totalTokens","totalCost",
+          "tags","metadata","createdAt","updatedAt"
+        ) VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?)
+        ON CONFLICT ("traceId") DO UPDATE SET
+          "name" = COALESCE(EXCLUDED."name", "traces"."name"),
+          "sessionId" = COALESCE(EXCLUDED."sessionId", "traces"."sessionId"),
+          "userId" = COALESCE(EXCLUDED."userId", "traces"."userId"),
+          "status" = CASE
+            WHEN EXCLUDED."status" = 'error' THEN 'error'
+            WHEN EXCLUDED."status" = 'ok' AND "traces"."status" != 'error' THEN 'ok'
+            ELSE "traces"."status"
+          END,
+          "startTime" = MIN("traces"."startTime", EXCLUDED."startTime"),
+          "endTime" = MAX(
+            COALESCE("traces"."endTime", EXCLUDED."endTime"),
+            COALESCE(EXCLUDED."endTime", "traces"."endTime")
+          ),
+          "durationMs" = CAST(
+            (julianday(MAX(
+              COALESCE("traces"."endTime", EXCLUDED."endTime"),
+              COALESCE(EXCLUDED."endTime", "traces"."endTime")
+            )) - julianday(MIN("traces"."startTime", EXCLUDED."startTime")))
+            * 86400000 AS INTEGER
+          ),
+          "spanCount" = "traces"."spanCount" + EXCLUDED."spanCount",
+          "totalInputTokens" = "traces"."totalInputTokens" + EXCLUDED."totalInputTokens",
+          "totalOutputTokens" = "traces"."totalOutputTokens" + EXCLUDED."totalOutputTokens",
+          "totalTokens" = "traces"."totalTokens" + EXCLUDED."totalTokens",
+          "totalCost" = "traces"."totalCost" + EXCLUDED."totalCost",
+          "tags" = json_patch("traces"."tags", EXCLUDED."tags"),
+          "metadata" = json_patch("traces"."metadata", EXCLUDED."metadata"),
+          "updatedAt" = ?
+      `).bind((0, node_crypto.randomUUID)(), data.traceId, data.name ?? null, data.sessionId ?? null, data.userId ?? null, data.status ?? "unset", data.startTime.toISOString(), data.endTime?.toISOString() ?? null, data.durationMs ?? null, data.spanCount ?? 1, data.totalInputTokens ?? 0, data.totalOutputTokens ?? 0, data.totalTokens ?? 0, data.totalCost ?? 0, JSON.stringify(data.tags ?? {}), JSON.stringify(data.metadata ?? {}), now, now, now).run();
+		},
+		batchInsertSpans: async (spans) => {
+			if (spans.length === 0) return { count: 0 };
+			const now = (/* @__PURE__ */ new Date()).toISOString();
+			const stmts = spans.map((s) => db.prepare(`
+          INSERT OR IGNORE INTO "spans" (
+            "id","traceId","spanId","parentSpanId","name","kind",
+            "status","statusMessage","startTime","endTime","durationMs",
+            "provider","model","promptTokens","completionTokens",
+            "totalTokens","cost","configId","variantId","environmentId",
+            "providerConfigId","requestId","source","input","output",
+            "attributes","createdAt","updatedAt"
+          ) VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?)
+        `).bind((0, node_crypto.randomUUID)(), s.traceId, s.spanId, s.parentSpanId ?? null, s.name, s.kind ?? 1, s.status ?? 0, s.statusMessage ?? null, s.startTime.toISOString(), s.endTime?.toISOString() ?? null, s.durationMs ?? null, s.provider ?? null, s.model ?? null, s.promptTokens ?? 0, s.completionTokens ?? 0, s.totalTokens ?? 0, s.cost ?? 0, s.configId ?? null, s.variantId ?? null, s.environmentId ?? null, s.providerConfigId ?? null, s.requestId ?? null, s.source ?? "gateway", s.input != null ? JSON.stringify(s.input) : null, s.output != null ? JSON.stringify(s.output) : null, JSON.stringify(s.attributes ?? {}), now, now));
+			for (let i = 0; i < stmts.length; i += D1_BATCH_LIMIT) await db.batch(stmts.slice(i, i + D1_BATCH_LIMIT));
+			return { count: spans.length };
+		},
+		batchInsertSpanEvents: async (events) => {
+			if (events.length === 0) return { count: 0 };
+			const now = (/* @__PURE__ */ new Date()).toISOString();
+			const stmts = events.map((e) => db.prepare(`
+          INSERT INTO "span_events" (
+            "id","traceId","spanId","name","timestamp","attributes","createdAt"
+          ) VALUES (?,?,?,?,?,?,?)
+        `).bind((0, node_crypto.randomUUID)(), e.traceId, e.spanId, e.name, e.timestamp.toISOString(), JSON.stringify(e.attributes ?? {}), now));
+			for (let i = 0; i < stmts.length; i += D1_BATCH_LIMIT) await db.batch(stmts.slice(i, i + D1_BATCH_LIMIT));
+			return { count: events.length };
+		},
+		listTraces: async (params) => {
+			const { limit = 50, offset = 0, sessionId, userId, status, name, startDate, endDate, tags } = params ?? {};
+			const conditions = ["1=1"];
+			const queryParams = [];
+			if (sessionId) {
+				conditions.push(`"sessionId" = ?`);
+				queryParams.push(sessionId);
+			}
+			if (userId) {
+				conditions.push(`"userId" = ?`);
+				queryParams.push(userId);
+			}
+			if (status) {
+				conditions.push(`"status" = ?`);
+				queryParams.push(status);
+			}
+			if (name) {
+				conditions.push(`"name" LIKE ?`);
+				queryParams.push(`%${name}%`);
+			}
+			if (startDate) {
+				conditions.push(`"startTime" >= ?`);
+				queryParams.push(startDate.toISOString());
+			}
+			if (endDate) {
+				conditions.push(`"startTime" <= ?`);
+				queryParams.push(endDate.toISOString());
+			}
+			const tagFilter = buildTagFilters(tags);
+			conditions.push(...tagFilter.conditions);
+			queryParams.push(...tagFilter.params);
+			const where = conditions.join(" AND ");
+			const total = (await db.prepare(`SELECT COUNT(*) AS "total" FROM "traces" WHERE ${where}`).bind(...queryParams).first())?.total ?? 0;
+			const { results: data } = await db.prepare(`SELECT * FROM "traces" WHERE ${where} ORDER BY "startTime" DESC LIMIT ? OFFSET ?`).bind(...queryParams, limit, offset).all();
+			return {
+				data,
+				total,
+				limit,
+				offset
+			};
+		},
+		getTraceWithSpans: async (traceId) => {
+			const trace = await db.prepare(`SELECT * FROM "traces" WHERE "traceId" = ?`).bind(traceId).first();
+			if (!trace) return void 0;
+			const [spanResult, eventResult] = await db.batch([db.prepare(`SELECT * FROM "spans" WHERE "traceId" = ? ORDER BY "startTime" ASC`).bind(traceId), db.prepare(`SELECT * FROM "span_events" WHERE "traceId" = ? ORDER BY "timestamp" ASC`).bind(traceId)]);
+			return {
+				trace,
+				spans: spanResult.results ?? [],
+				events: eventResult.results ?? []
+			};
+		},
+		getTraceStats: async (params) => {
+			const conditions = [`"startTime" >= ?`, `"startTime" <= ?`];
+			const queryParams = [params.startDate.toISOString(), params.endDate.toISOString()];
+			if (params.sessionId) {
+				conditions.push(`"sessionId" = ?`);
+				queryParams.push(params.sessionId);
+			}
+			if (params.userId) {
+				conditions.push(`"userId" = ?`);
+				queryParams.push(params.userId);
+			}
+			const where = conditions.join(" AND ");
+			return db.prepare(`
+        SELECT
+          COUNT(*) AS "totalTraces",
+          COALESCE(AVG("durationMs"), 0) AS "avgDurationMs",
+          COUNT(CASE WHEN "status" = 'error' THEN 1 END) AS "errorCount",
+          COALESCE(SUM("totalCost"), 0) AS "totalCost",
+          COALESCE(SUM("totalTokens"), 0) AS "totalTokens",
+          COALESCE(SUM("spanCount"), 0) AS "totalSpans"
+        FROM "traces" WHERE ${where}
+      `).bind(...queryParams).first();
+		}
+	};
+}
+/**
+* Create a Cloudflare D1-backed telemetry store.
+*
+* Usage:
+* ```ts
+* import { d1Store } from '@llmops/sdk/store/d1'
+*
+* export default {
+*   async fetch(request, env) {
+*     const ops = llmops({
+*       telemetry: d1Store(env.DB),
+*     })
+*   }
+* }
+* ```
+*/
+function createD1Store(db) {
+	return {
+		...createD1LLMRequestsStore(db),
+		...createD1TracesStore(db),
+		_db: db
+	};
+}
+
+//#endregion
+//#region src/store/d1/migrations/000001_e44e1c4f.sql
+var _000001_e44e1c4f_default = "CREATE TABLE IF NOT EXISTS llm_requests (\n  id TEXT PRIMARY KEY,\n  \"requestId\" TEXT NOT NULL,\n  \"configId\" TEXT,\n  \"variantId\" TEXT,\n  \"environmentId\" TEXT,\n  \"providerConfigId\" TEXT,\n  provider TEXT NOT NULL,\n  model TEXT NOT NULL,\n  \"promptTokens\" INTEGER NOT NULL DEFAULT 0,\n  \"completionTokens\" INTEGER NOT NULL DEFAULT 0,\n  \"totalTokens\" INTEGER NOT NULL DEFAULT 0,\n  \"cachedTokens\" INTEGER NOT NULL DEFAULT 0,\n  \"cacheCreationTokens\" INTEGER NOT NULL DEFAULT 0,\n  cost INTEGER NOT NULL DEFAULT 0,\n  \"cacheSavings\" INTEGER NOT NULL DEFAULT 0,\n  \"inputCost\" INTEGER NOT NULL DEFAULT 0,\n  \"outputCost\" INTEGER NOT NULL DEFAULT 0,\n  endpoint TEXT NOT NULL,\n  \"statusCode\" INTEGER NOT NULL,\n  \"latencyMs\" INTEGER NOT NULL DEFAULT 0,\n  \"isStreaming\" INTEGER NOT NULL DEFAULT 0,\n  \"userId\" TEXT,\n  tags TEXT NOT NULL DEFAULT '{}',\n  \"guardrailResults\" TEXT,\n  \"traceId\" TEXT,\n  \"spanId\" TEXT,\n  \"parentSpanId\" TEXT,\n  \"sessionId\" TEXT,\n  \"createdAt\" TEXT NOT NULL DEFAULT (datetime('now')),\n  \"updatedAt\" TEXT NOT NULL DEFAULT (datetime('now'))\n);\n\nCREATE TABLE IF NOT EXISTS span_events (\n  id TEXT PRIMARY KEY,\n  \"traceId\" TEXT NOT NULL,\n  \"spanId\" TEXT NOT NULL,\n  name TEXT NOT NULL,\n  timestamp TEXT NOT NULL,\n  attributes TEXT NOT NULL DEFAULT '{}',\n  \"createdAt\" TEXT NOT NULL DEFAULT (datetime('now'))\n);\n\nCREATE TABLE IF NOT EXISTS spans (\n  id TEXT PRIMARY KEY,\n  \"traceId\" TEXT NOT NULL,\n  \"spanId\" TEXT NOT NULL UNIQUE,\n  \"parentSpanId\" TEXT,\n  name TEXT NOT NULL,\n  kind INTEGER NOT NULL DEFAULT 1,\n  status INTEGER NOT NULL DEFAULT 0,\n  \"statusMessage\" TEXT,\n  \"startTime\" TEXT NOT NULL,\n  \"endTime\" TEXT,\n  \"durationMs\" INTEGER,\n  provider TEXT,\n  model TEXT,\n  \"promptTokens\" INTEGER NOT NULL DEFAULT 0,\n  \"completionTokens\" INTEGER NOT NULL DEFAULT 0,\n  \"totalTokens\" INTEGER NOT NULL DEFAULT 0,\n  cost INTEGER NOT NULL DEFAULT 0,\n  \"configId\" TEXT,\n  \"variantId\" TEXT,\n  \"environmentId\" TEXT,\n  \"providerConfigId\" TEXT,\n  \"requestId\" TEXT,\n  source TEXT NOT NULL DEFAULT 'gateway',\n  input TEXT,\n  output TEXT,\n  attributes TEXT NOT NULL DEFAULT '{}',\n  \"createdAt\" TEXT NOT NULL DEFAULT (datetime('now')),\n  \"updatedAt\" TEXT NOT NULL DEFAULT (datetime('now'))\n);\n\nCREATE TABLE IF NOT EXISTS traces (\n  id TEXT PRIMARY KEY,\n  \"traceId\" TEXT NOT NULL UNIQUE,\n  name TEXT,\n  \"sessionId\" TEXT,\n  \"userId\" TEXT,\n  status TEXT NOT NULL DEFAULT 'unset',\n  \"startTime\" TEXT NOT NULL,\n  \"endTime\" TEXT,\n  \"durationMs\" INTEGER,\n  \"spanCount\" INTEGER NOT NULL DEFAULT 0,\n  \"totalInputTokens\" INTEGER NOT NULL DEFAULT 0,\n  \"totalOutputTokens\" INTEGER NOT NULL DEFAULT 0,\n  \"totalTokens\" INTEGER NOT NULL DEFAULT 0,\n  \"totalCost\" INTEGER NOT NULL DEFAULT 0,\n  tags TEXT NOT NULL DEFAULT '{}',\n  metadata TEXT NOT NULL DEFAULT '{}',\n  \"createdAt\" TEXT NOT NULL DEFAULT (datetime('now')),\n  \"updatedAt\" TEXT NOT NULL DEFAULT (datetime('now'))\n);\n";
+
+//#endregion
+//#region src/store/d1/migrations/index.ts
+const migrations = [["000001_e44e1c4f", _000001_e44e1c4f_default]];
+
+//#endregion
+//#region src/store/d1/migrate.ts
+/**
+* Run pending migrations against a D1 database.
+*/
+async function runD1Migrations(db) {
+	await db.prepare(`
+    CREATE TABLE IF NOT EXISTS _llmops_migrations (
+      name TEXT PRIMARY KEY,
+      applied_at TEXT NOT NULL DEFAULT (datetime('now'))
+    )
+  `).run();
+	const { results } = await db.prepare("SELECT name FROM _llmops_migrations ORDER BY name").all();
+	const applied = new Set(results.map((r) => r.name));
+	const newlyApplied = [];
+	for (const [name, sql] of migrations) {
+		if (applied.has(name)) continue;
+		const statements = sql.split(";").map((s) => s.trim()).filter(Boolean).map((s) => db.prepare(s));
+		statements.push(db.prepare("INSERT INTO _llmops_migrations (name) VALUES (?)").bind(name));
+		await db.batch(statements);
+		newlyApplied.push(name);
+	}
+	return { applied: newlyApplied };
+}
+
+//#endregion
+exports.d1Store = createD1Store;
+exports.runD1Migrations = runD1Migrations;

package/dist/store/d1.d.cts

@@ -0,0 +1,12 @@
+import "../interface-DMuF7YgM.cjs";
+import { n as createD1Store, r as D1Database, t as D1Store } from "../d1-store-D664VLg0.cjs";
+
+//#region src/store/d1/migrate.d.ts
+/**
+ * Run pending migrations against a D1 database.
+ */
+declare function runD1Migrations(db: D1Database): Promise<{
+  applied: string[];
+}>;
+//#endregion
+export { type D1Database, type D1Store, createD1Store as d1Store, runD1Migrations };

package/dist/store/d1.d.mts

@@ -0,0 +1,12 @@
+import "../interface-BrJMazBg.mjs";
+import { n as createD1Store, r as D1Database, t as D1Store } from "../d1-store-DBf99oU8.mjs";
+
+//#region src/store/d1/migrate.d.ts
+/**
+ * Run pending migrations against a D1 database.
+ */
+declare function runD1Migrations(db: D1Database): Promise<{
+  applied: string[];
+}>;
+//#endregion
+export { type D1Database, type D1Store, createD1Store as d1Store, runD1Migrations };