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

package/dist/store/d1.mjs

@@ -0,0 +1,490 @@
+import { randomUUID } from "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(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(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(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(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(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
+export { createD1Store as d1Store, runD1Migrations };

package/dist/store/pg.cjs

@@ -1,7 +1,7 @@
-let __llmops_core = require("@llmops/core");
+const require_pg_store = require('../pg-store-uawkO2hJ.cjs');
 
 //#region src/store/pg/migrations/000001_2191b051.sql
-var _000001_2191b051_default = "BEGIN;\n\nCREATE TABLE llm_requests (\n  id UUID PRIMARY KEY,\n  \"requestId\" UUID NOT NULL,\n  \"configId\" UUID,\n  \"variantId\" UUID,\n  \"environmentId\" UUID,\n  \"providerConfigId\" UUID,\n  provider VARCHAR(255) NOT NULL,\n  model VARCHAR(255) 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 VARCHAR(255) NOT NULL,\n  \"statusCode\" INTEGER NOT NULL,\n  \"latencyMs\" INTEGER NOT NULL DEFAULT 0,\n  \"isStreaming\" BOOLEAN NOT NULL DEFAULT false,\n  \"userId\" VARCHAR(255),\n  tags JSONB NOT NULL DEFAULT '{}'::jsonb,\n  \"guardrailResults\" JSONB,\n  \"traceId\" VARCHAR(255),\n  \"spanId\" VARCHAR(255),\n  \"parentSpanId\" VARCHAR(255),\n  \"sessionId\" VARCHAR(255),\n  \"createdAt\" TIMESTAMP NOT NULL DEFAULT NOW(),\n  \"updatedAt\" TIMESTAMP NOT NULL DEFAULT NOW()\n);\n\nCREATE TABLE span_events (\n  id UUID PRIMARY KEY,\n  \"traceId\" VARCHAR(255) NOT NULL,\n  \"spanId\" VARCHAR(255) NOT NULL,\n  name VARCHAR(255) NOT NULL,\n  timestamp TIMESTAMP NOT NULL,\n  attributes JSONB NOT NULL DEFAULT '{}'::jsonb,\n  \"createdAt\" TIMESTAMP NOT NULL DEFAULT NOW()\n);\n\nCREATE TABLE spans (\n  id UUID PRIMARY KEY,\n  \"traceId\" VARCHAR(255) NOT NULL,\n  \"spanId\" VARCHAR(255) NOT NULL UNIQUE,\n  \"parentSpanId\" VARCHAR(255),\n  name VARCHAR(255) NOT NULL,\n  kind INTEGER NOT NULL DEFAULT 1,\n  status INTEGER NOT NULL DEFAULT 0,\n  \"statusMessage\" TEXT,\n  \"startTime\" TIMESTAMP NOT NULL,\n  \"endTime\" TIMESTAMP,\n  \"durationMs\" INTEGER,\n  provider VARCHAR(255),\n  model VARCHAR(255),\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\" UUID,\n  \"variantId\" UUID,\n  \"environmentId\" UUID,\n  \"providerConfigId\" UUID,\n  \"requestId\" UUID,\n  source VARCHAR(50) NOT NULL DEFAULT 'gateway',\n  input JSONB,\n  output JSONB,\n  attributes JSONB NOT NULL DEFAULT '{}'::jsonb,\n  \"createdAt\" TIMESTAMP NOT NULL DEFAULT NOW(),\n  \"updatedAt\" TIMESTAMP NOT NULL DEFAULT NOW()\n);\n\nCREATE TABLE traces (\n  id UUID PRIMARY KEY,\n  \"traceId\" VARCHAR(255) NOT NULL UNIQUE,\n  name VARCHAR(255),\n  \"sessionId\" VARCHAR(255),\n  \"userId\" VARCHAR(255),\n  status VARCHAR(50) NOT NULL DEFAULT 'unset',\n  \"startTime\" TIMESTAMP NOT NULL,\n  \"endTime\" TIMESTAMP,\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 JSONB NOT NULL DEFAULT '{}'::jsonb,\n  metadata JSONB NOT NULL DEFAULT '{}'::jsonb,\n  \"createdAt\" TIMESTAMP NOT NULL DEFAULT NOW(),\n  \"updatedAt\" TIMESTAMP NOT NULL DEFAULT NOW()\n);\n\nCOMMIT;\n";
+var _000001_2191b051_default = "BEGIN;\n\nCREATE TABLE IF NOT EXISTS llm_requests (\n  id UUID PRIMARY KEY,\n  \"requestId\" UUID NOT NULL,\n  \"configId\" UUID,\n  \"variantId\" UUID,\n  \"environmentId\" UUID,\n  \"providerConfigId\" UUID,\n  provider VARCHAR(255) NOT NULL,\n  model VARCHAR(255) 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 VARCHAR(255) NOT NULL,\n  \"statusCode\" INTEGER NOT NULL,\n  \"latencyMs\" INTEGER NOT NULL DEFAULT 0,\n  \"isStreaming\" BOOLEAN NOT NULL DEFAULT false,\n  \"userId\" VARCHAR(255),\n  tags JSONB NOT NULL DEFAULT '{}'::jsonb,\n  \"guardrailResults\" JSONB,\n  \"traceId\" VARCHAR(255),\n  \"spanId\" VARCHAR(255),\n  \"parentSpanId\" VARCHAR(255),\n  \"sessionId\" VARCHAR(255),\n  \"createdAt\" TIMESTAMP NOT NULL DEFAULT NOW(),\n  \"updatedAt\" TIMESTAMP NOT NULL DEFAULT NOW()\n);\n\nCREATE TABLE IF NOT EXISTS span_events (\n  id UUID PRIMARY KEY,\n  \"traceId\" VARCHAR(255) NOT NULL,\n  \"spanId\" VARCHAR(255) NOT NULL,\n  name VARCHAR(255) NOT NULL,\n  timestamp TIMESTAMP NOT NULL,\n  attributes JSONB NOT NULL DEFAULT '{}'::jsonb,\n  \"createdAt\" TIMESTAMP NOT NULL DEFAULT NOW()\n);\n\nCREATE TABLE IF NOT EXISTS spans (\n  id UUID PRIMARY KEY,\n  \"traceId\" VARCHAR(255) NOT NULL,\n  \"spanId\" VARCHAR(255) NOT NULL UNIQUE,\n  \"parentSpanId\" VARCHAR(255),\n  name VARCHAR(255) NOT NULL,\n  kind INTEGER NOT NULL DEFAULT 1,\n  status INTEGER NOT NULL DEFAULT 0,\n  \"statusMessage\" TEXT,\n  \"startTime\" TIMESTAMP NOT NULL,\n  \"endTime\" TIMESTAMP,\n  \"durationMs\" INTEGER,\n  provider VARCHAR(255),\n  model VARCHAR(255),\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\" UUID,\n  \"variantId\" UUID,\n  \"environmentId\" UUID,\n  \"providerConfigId\" UUID,\n  \"requestId\" UUID,\n  source VARCHAR(50) NOT NULL DEFAULT 'gateway',\n  input JSONB,\n  output JSONB,\n  attributes JSONB NOT NULL DEFAULT '{}'::jsonb,\n  \"createdAt\" TIMESTAMP NOT NULL DEFAULT NOW(),\n  \"updatedAt\" TIMESTAMP NOT NULL DEFAULT NOW()\n);\n\nCREATE TABLE IF NOT EXISTS traces (\n  id UUID PRIMARY KEY,\n  \"traceId\" VARCHAR(255) NOT NULL UNIQUE,\n  name VARCHAR(255),\n  \"sessionId\" VARCHAR(255),\n  \"userId\" VARCHAR(255),\n  status VARCHAR(50) NOT NULL DEFAULT 'unset',\n  \"startTime\" TIMESTAMP NOT NULL,\n  \"endTime\" TIMESTAMP,\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 JSONB NOT NULL DEFAULT '{}'::jsonb,\n  metadata JSONB NOT NULL DEFAULT '{}'::jsonb,\n  \"createdAt\" TIMESTAMP NOT NULL DEFAULT NOW(),\n  \"updatedAt\" TIMESTAMP NOT NULL DEFAULT NOW()\n);\n\nCOMMIT;\n";
 
 //#endregion
 //#region src/store/pg/migrations/index.ts
@@ -43,10 +43,5 @@ async function runMigrations(pool, schema) {
 }
 
 //#endregion
-Object.defineProperty(exports, 'pgStore', {
-  enumerable: true,
-  get: function () {
-    return __llmops_core.createPgStore;
-  }
-});
+exports.pgStore = require_pg_store.createPgStore;
 exports.runMigrations = runMigrations;

package/dist/store/pg.d.cts

@@ -1,4 +1,4 @@
-import { PgStore, createPgStore as pgStore } from "@llmops/core";
+import { a as PgStore, l as createPgStore } from "../interface-DMuF7YgM.cjs";
 import { Pool } from "pg";
 
 //#region src/store/pg/migrate.d.ts
@@ -14,4 +14,4 @@ declare function runMigrations(pool: Pool, schema: string): Promise<{
   applied: string[];
 }>;
 //#endregion
-export { type PgStore, pgStore, runMigrations };
+export { type PgStore, createPgStore as pgStore, runMigrations };

package/dist/store/pg.d.mts

@@ -1,4 +1,4 @@
-import { PgStore, createPgStore as pgStore } from "@llmops/core";
+import { a as PgStore, l as createPgStore } from "../interface-BrJMazBg.mjs";
 import { Pool } from "pg";
 
 //#region src/store/pg/migrate.d.ts
@@ -14,4 +14,4 @@ declare function runMigrations(pool: Pool, schema: string): Promise<{
   applied: string[];
 }>;
 //#endregion
-export { type PgStore, pgStore, runMigrations };
+export { type PgStore, createPgStore as pgStore, runMigrations };

package/dist/store/pg.mjs

@@ -1,7 +1,7 @@
-import { createPgStore as pgStore } from "@llmops/core";
+import { n as createPgStore } from "../pg-store-sIMdF_Pc.mjs";
 
 //#region src/store/pg/migrations/000001_2191b051.sql
-var _000001_2191b051_default = "BEGIN;\n\nCREATE TABLE llm_requests (\n  id UUID PRIMARY KEY,\n  \"requestId\" UUID NOT NULL,\n  \"configId\" UUID,\n  \"variantId\" UUID,\n  \"environmentId\" UUID,\n  \"providerConfigId\" UUID,\n  provider VARCHAR(255) NOT NULL,\n  model VARCHAR(255) 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 VARCHAR(255) NOT NULL,\n  \"statusCode\" INTEGER NOT NULL,\n  \"latencyMs\" INTEGER NOT NULL DEFAULT 0,\n  \"isStreaming\" BOOLEAN NOT NULL DEFAULT false,\n  \"userId\" VARCHAR(255),\n  tags JSONB NOT NULL DEFAULT '{}'::jsonb,\n  \"guardrailResults\" JSONB,\n  \"traceId\" VARCHAR(255),\n  \"spanId\" VARCHAR(255),\n  \"parentSpanId\" VARCHAR(255),\n  \"sessionId\" VARCHAR(255),\n  \"createdAt\" TIMESTAMP NOT NULL DEFAULT NOW(),\n  \"updatedAt\" TIMESTAMP NOT NULL DEFAULT NOW()\n);\n\nCREATE TABLE span_events (\n  id UUID PRIMARY KEY,\n  \"traceId\" VARCHAR(255) NOT NULL,\n  \"spanId\" VARCHAR(255) NOT NULL,\n  name VARCHAR(255) NOT NULL,\n  timestamp TIMESTAMP NOT NULL,\n  attributes JSONB NOT NULL DEFAULT '{}'::jsonb,\n  \"createdAt\" TIMESTAMP NOT NULL DEFAULT NOW()\n);\n\nCREATE TABLE spans (\n  id UUID PRIMARY KEY,\n  \"traceId\" VARCHAR(255) NOT NULL,\n  \"spanId\" VARCHAR(255) NOT NULL UNIQUE,\n  \"parentSpanId\" VARCHAR(255),\n  name VARCHAR(255) NOT NULL,\n  kind INTEGER NOT NULL DEFAULT 1,\n  status INTEGER NOT NULL DEFAULT 0,\n  \"statusMessage\" TEXT,\n  \"startTime\" TIMESTAMP NOT NULL,\n  \"endTime\" TIMESTAMP,\n  \"durationMs\" INTEGER,\n  provider VARCHAR(255),\n  model VARCHAR(255),\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\" UUID,\n  \"variantId\" UUID,\n  \"environmentId\" UUID,\n  \"providerConfigId\" UUID,\n  \"requestId\" UUID,\n  source VARCHAR(50) NOT NULL DEFAULT 'gateway',\n  input JSONB,\n  output JSONB,\n  attributes JSONB NOT NULL DEFAULT '{}'::jsonb,\n  \"createdAt\" TIMESTAMP NOT NULL DEFAULT NOW(),\n  \"updatedAt\" TIMESTAMP NOT NULL DEFAULT NOW()\n);\n\nCREATE TABLE traces (\n  id UUID PRIMARY KEY,\n  \"traceId\" VARCHAR(255) NOT NULL UNIQUE,\n  name VARCHAR(255),\n  \"sessionId\" VARCHAR(255),\n  \"userId\" VARCHAR(255),\n  status VARCHAR(50) NOT NULL DEFAULT 'unset',\n  \"startTime\" TIMESTAMP NOT NULL,\n  \"endTime\" TIMESTAMP,\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 JSONB NOT NULL DEFAULT '{}'::jsonb,\n  metadata JSONB NOT NULL DEFAULT '{}'::jsonb,\n  \"createdAt\" TIMESTAMP NOT NULL DEFAULT NOW(),\n  \"updatedAt\" TIMESTAMP NOT NULL DEFAULT NOW()\n);\n\nCOMMIT;\n";
+var _000001_2191b051_default = "BEGIN;\n\nCREATE TABLE IF NOT EXISTS llm_requests (\n  id UUID PRIMARY KEY,\n  \"requestId\" UUID NOT NULL,\n  \"configId\" UUID,\n  \"variantId\" UUID,\n  \"environmentId\" UUID,\n  \"providerConfigId\" UUID,\n  provider VARCHAR(255) NOT NULL,\n  model VARCHAR(255) 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 VARCHAR(255) NOT NULL,\n  \"statusCode\" INTEGER NOT NULL,\n  \"latencyMs\" INTEGER NOT NULL DEFAULT 0,\n  \"isStreaming\" BOOLEAN NOT NULL DEFAULT false,\n  \"userId\" VARCHAR(255),\n  tags JSONB NOT NULL DEFAULT '{}'::jsonb,\n  \"guardrailResults\" JSONB,\n  \"traceId\" VARCHAR(255),\n  \"spanId\" VARCHAR(255),\n  \"parentSpanId\" VARCHAR(255),\n  \"sessionId\" VARCHAR(255),\n  \"createdAt\" TIMESTAMP NOT NULL DEFAULT NOW(),\n  \"updatedAt\" TIMESTAMP NOT NULL DEFAULT NOW()\n);\n\nCREATE TABLE IF NOT EXISTS span_events (\n  id UUID PRIMARY KEY,\n  \"traceId\" VARCHAR(255) NOT NULL,\n  \"spanId\" VARCHAR(255) NOT NULL,\n  name VARCHAR(255) NOT NULL,\n  timestamp TIMESTAMP NOT NULL,\n  attributes JSONB NOT NULL DEFAULT '{}'::jsonb,\n  \"createdAt\" TIMESTAMP NOT NULL DEFAULT NOW()\n);\n\nCREATE TABLE IF NOT EXISTS spans (\n  id UUID PRIMARY KEY,\n  \"traceId\" VARCHAR(255) NOT NULL,\n  \"spanId\" VARCHAR(255) NOT NULL UNIQUE,\n  \"parentSpanId\" VARCHAR(255),\n  name VARCHAR(255) NOT NULL,\n  kind INTEGER NOT NULL DEFAULT 1,\n  status INTEGER NOT NULL DEFAULT 0,\n  \"statusMessage\" TEXT,\n  \"startTime\" TIMESTAMP NOT NULL,\n  \"endTime\" TIMESTAMP,\n  \"durationMs\" INTEGER,\n  provider VARCHAR(255),\n  model VARCHAR(255),\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\" UUID,\n  \"variantId\" UUID,\n  \"environmentId\" UUID,\n  \"providerConfigId\" UUID,\n  \"requestId\" UUID,\n  source VARCHAR(50) NOT NULL DEFAULT 'gateway',\n  input JSONB,\n  output JSONB,\n  attributes JSONB NOT NULL DEFAULT '{}'::jsonb,\n  \"createdAt\" TIMESTAMP NOT NULL DEFAULT NOW(),\n  \"updatedAt\" TIMESTAMP NOT NULL DEFAULT NOW()\n);\n\nCREATE TABLE IF NOT EXISTS traces (\n  id UUID PRIMARY KEY,\n  \"traceId\" VARCHAR(255) NOT NULL UNIQUE,\n  name VARCHAR(255),\n  \"sessionId\" VARCHAR(255),\n  \"userId\" VARCHAR(255),\n  status VARCHAR(50) NOT NULL DEFAULT 'unset',\n  \"startTime\" TIMESTAMP NOT NULL,\n  \"endTime\" TIMESTAMP,\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 JSONB NOT NULL DEFAULT '{}'::jsonb,\n  metadata JSONB NOT NULL DEFAULT '{}'::jsonb,\n  \"createdAt\" TIMESTAMP NOT NULL DEFAULT NOW(),\n  \"updatedAt\" TIMESTAMP NOT NULL DEFAULT NOW()\n);\n\nCOMMIT;\n";
 
 //#endregion
 //#region src/store/pg/migrations/index.ts
@@ -43,4 +43,4 @@ async function runMigrations(pool, schema) {
 }
 
 //#endregion
-export { pgStore, runMigrations };
+export { createPgStore as pgStore, runMigrations };

package/dist/types.d.cts

@@ -1,2 +1,3 @@
-import { PgStore, TelemetryStore } from "@llmops/core";
-export { type PgStore, type TelemetryStore };
+import { a as PgStore, t as TelemetryStore } from "./interface-DMuF7YgM.cjs";
+import { t as D1Store } from "./d1-store-D664VLg0.cjs";
+export { type D1Store, type PgStore, type TelemetryStore };

package/dist/types.d.mts

@@ -1,2 +1,3 @@
-import { PgStore, TelemetryStore } from "@llmops/core";
-export { type PgStore, type TelemetryStore };
+import { a as PgStore, t as TelemetryStore } from "./interface-BrJMazBg.mjs";
+import { t as D1Store } from "./d1-store-DBf99oU8.mjs";
+export { type D1Store, type PgStore, type TelemetryStore };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@llmops/sdk",
-  "version": "1.0.0-beta.4",
+  "version": "1.0.0-beta.6",
   "description": "An LLMOps toolkit for TypeScript applications",
   "type": "module",
   "license": "Apache-2.0",
@@ -88,6 +88,16 @@
         "default": "./dist/store/pg.cjs"
       }
     },
+    "./store/d1": {
+      "import": {
+        "types": "./dist/store/d1.d.mts",
+        "default": "./dist/store/d1.mjs"
+      },
+      "require": {
+        "types": "./dist/store/d1.d.cts",
+        "default": "./dist/store/d1.cjs"
+      }
+    },
     "./types": {
       "import": {
         "types": "./dist/types.d.mts",
@@ -104,8 +114,8 @@
     "access": "public"
   },
   "dependencies": {
-    "@llmops/app": "^1.0.0-beta.4",
-    "@llmops/core": "^1.0.0-beta.4"
+    "@llmops/app": "^1.0.0-beta.6",
+    "@llmops/core": "^1.0.0-beta.6"
   },
   "peerDependencies": {
     "pg": "*"
@@ -116,8 +126,8 @@
     }
   },
   "devDependencies": {
-    "@tsqx/cli": "^0.0.4",
-    "@tsqx/kit": "^0.0.4",
+    "@tsqx/cli": "^0.0.5",
+    "@tsqx/kit": "^0.0.5",
     "@types/express": "^5.0.6",
     "@types/pg": "^8.15.6",
     "hono": "^4.7.0"