@llmops/sdk 1.0.0-beta.7 → 1.0.0-beta.8

package/dist/store/pg.cjs

@@ -1,5 +1,616 @@
-const require_pg_store = require('../pg-store-uawkO2hJ.cjs');
+const require_types = require('../types-uIsMdtuF.cjs');
+let __llmops_core = require("@llmops/core");
+let node_crypto = require("node:crypto");
 
+//#region src/telemetry/pg-store.ts
+function buildTagFilters(tags, paramOffset) {
+	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(`"tags"->>'${key}' = $${paramOffset + params.length + 1}`);
+			params.push(values[0]);
+		} else {
+			const placeholders = values.map((_, i) => `$${paramOffset + params.length + i + 1}`).join(", ");
+			conditions.push(`"tags"->>'${key}' IN (${placeholders})`);
+			params.push(...values);
+		}
+	}
+	return {
+		conditions,
+		params
+	};
+}
+function createLLMRequestsStore(pool) {
+	return {
+		batchInsertRequests: async (requests) => {
+			if (requests.length === 0) return { count: 0 };
+			const now = (/* @__PURE__ */ new Date()).toISOString();
+			const columns = [
+				"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"
+			];
+			const colNames = columns.map((c) => `"${c}"`).join(", ");
+			const params = [];
+			const valueRows = [];
+			for (const req of requests) {
+				const result = require_types.insertLLMRequestSchema.safeParse(req);
+				if (!result.success) {
+					__llmops_core.logger.warn(`[batchInsertRequests] Skipping invalid request: ${result.error.message}`);
+					continue;
+				}
+				const r = result.data;
+				const offset = params.length;
+				const placeholders = columns.map((_, i) => `$${offset + i + 1}`).join(", ");
+				valueRows.push(`(${placeholders})`);
+				params.push((0, node_crypto.randomUUID)(), r.requestId, r.configId ?? null, r.variantId ?? null, r.environmentId ?? null, r.providerConfigId ?? null, r.provider, r.model, r.promptTokens, r.completionTokens, r.totalTokens, r.cachedTokens, r.cacheCreationTokens, r.cost, r.cacheSavings, r.inputCost, r.outputCost, r.endpoint, r.statusCode, r.latencyMs, r.isStreaming, r.userId ?? null, JSON.stringify(r.tags), r.guardrailResults ? JSON.stringify(r.guardrailResults) : null, r.traceId ?? null, r.spanId ?? null, r.parentSpanId ?? null, r.sessionId ?? null, now, now);
+			}
+			if (valueRows.length === 0) return { count: 0 };
+			await pool.query(`INSERT INTO "llm_requests" (${colNames}) VALUES ${valueRows.join(", ")}`, params);
+			return { count: valueRows.length };
+		},
+		insertRequest: async (request) => {
+			const result = require_types.insertLLMRequestSchema.safeParse(request);
+			if (!result.success) throw new Error(`Invalid request data: ${result.error.message}`);
+			const r = result.data;
+			const now = (/* @__PURE__ */ new Date()).toISOString();
+			const { rows } = await pool.query(`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 (
+          $1,$2,$3,$4,$5,$6,$7,$8,$9,$10,$11,$12,$13,$14,$15,
+          $16,$17,$18,$19,$20,$21,$22,$23,$24,$25,$26,$27,$28,$29,$30
+        ) RETURNING *`, [
+				(0, node_crypto.randomUUID)(),
+				r.requestId,
+				r.configId ?? null,
+				r.variantId ?? null,
+				r.environmentId ?? null,
+				r.providerConfigId ?? null,
+				r.provider,
+				r.model,
+				r.promptTokens,
+				r.completionTokens,
+				r.totalTokens,
+				r.cachedTokens,
+				r.cacheCreationTokens,
+				r.cost,
+				r.cacheSavings,
+				r.inputCost,
+				r.outputCost,
+				r.endpoint,
+				r.statusCode,
+				r.latencyMs,
+				r.isStreaming,
+				r.userId ?? null,
+				JSON.stringify(r.tags),
+				r.guardrailResults ? JSON.stringify(r.guardrailResults) : null,
+				r.traceId ?? null,
+				r.spanId ?? null,
+				r.parentSpanId ?? null,
+				r.sessionId ?? null,
+				now,
+				now
+			]);
+			return rows[0] ?? null;
+		},
+		listRequests: async (params) => {
+			const { limit = 100, offset = 0, configId, variantId, environmentId, providerConfigId, provider, model, startDate, endDate, tags } = params ?? {};
+			const conditions = ["TRUE"];
+			const queryParams = [];
+			let idx = 0;
+			if (configId) {
+				conditions.push(`"configId" = $${++idx}`);
+				queryParams.push(configId);
+			}
+			if (variantId) {
+				conditions.push(`"variantId" = $${++idx}`);
+				queryParams.push(variantId);
+			}
+			if (environmentId) {
+				conditions.push(`"environmentId" = $${++idx}`);
+				queryParams.push(environmentId);
+			}
+			if (providerConfigId) {
+				conditions.push(`"providerConfigId" = $${++idx}`);
+				queryParams.push(providerConfigId);
+			}
+			if (provider) {
+				conditions.push(`"provider" = $${++idx}`);
+				queryParams.push(provider);
+			}
+			if (model) {
+				conditions.push(`"model" = $${++idx}`);
+				queryParams.push(model);
+			}
+			if (startDate) {
+				conditions.push(`"createdAt" >= $${++idx}`);
+				queryParams.push(startDate.toISOString());
+			}
+			if (endDate) {
+				conditions.push(`"createdAt" <= $${++idx}`);
+				queryParams.push(endDate.toISOString());
+			}
+			const tagFilter = buildTagFilters(tags, idx);
+			conditions.push(...tagFilter.conditions);
+			queryParams.push(...tagFilter.params);
+			idx += tagFilter.params.length;
+			const where = conditions.join(" AND ");
+			const total = (await pool.query(`SELECT COUNT(*)::int AS "total" FROM "llm_requests" WHERE ${where}`, queryParams)).rows[0]?.total ?? 0;
+			return {
+				data: (await pool.query(`SELECT * FROM "llm_requests" WHERE ${where} ORDER BY "createdAt" DESC LIMIT $${++idx} OFFSET $${++idx}`, [
+					...queryParams,
+					limit,
+					offset
+				])).rows,
+				total,
+				limit,
+				offset
+			};
+		},
+		getRequestByRequestId: async (requestId) => {
+			const { rows } = await pool.query(`SELECT * FROM "llm_requests" WHERE "requestId" = $1`, [requestId]);
+			return rows[0] ?? void 0;
+		},
+		getTotalCost: async (params) => {
+			const { startDate, endDate, configId, variantId, environmentId, tags } = params;
+			const conditions = [`"createdAt" >= $1`, `"createdAt" <= $2`];
+			const queryParams = [startDate.toISOString(), endDate.toISOString()];
+			let idx = 2;
+			if (configId) {
+				conditions.push(`"configId" = $${++idx}`);
+				queryParams.push(configId);
+			}
+			if (variantId) {
+				conditions.push(`"variantId" = $${++idx}`);
+				queryParams.push(variantId);
+			}
+			if (environmentId) {
+				conditions.push(`"environmentId" = $${++idx}`);
+				queryParams.push(environmentId);
+			}
+			const tagFilter = buildTagFilters(tags, idx);
+			conditions.push(...tagFilter.conditions);
+			queryParams.push(...tagFilter.params);
+			const where = conditions.join(" AND ");
+			const { rows } = await pool.query(`SELECT
+          COALESCE(SUM("cost"), 0)::int AS "totalCost",
+          COALESCE(SUM("inputCost"), 0)::int AS "totalInputCost",
+          COALESCE(SUM("outputCost"), 0)::int AS "totalOutputCost",
+          COALESCE(SUM("promptTokens"), 0)::int AS "totalPromptTokens",
+          COALESCE(SUM("completionTokens"), 0)::int AS "totalCompletionTokens",
+          COALESCE(SUM("totalTokens"), 0)::int AS "totalTokens",
+          COALESCE(SUM("cachedTokens"), 0)::int AS "totalCachedTokens",
+          COALESCE(SUM("cacheSavings"), 0)::int AS "totalCacheSavings",
+          COUNT(*)::int AS "requestCount"
+        FROM "llm_requests" WHERE ${where}`, queryParams);
+			return rows[0];
+		},
+		getCostByModel: async (params) => {
+			const { rows } = await pool.query(`SELECT "provider", "model",
+          COALESCE(SUM("cost"), 0)::int AS "totalCost",
+          COALESCE(SUM("inputCost"), 0)::int AS "totalInputCost",
+          COALESCE(SUM("outputCost"), 0)::int AS "totalOutputCost",
+          COALESCE(SUM("totalTokens"), 0)::int AS "totalTokens",
+          COUNT(*)::int AS "requestCount",
+          AVG("latencyMs") AS "avgLatencyMs"
+        FROM "llm_requests"
+        WHERE "createdAt" >= $1 AND "createdAt" <= $2
+        GROUP BY "provider", "model"
+        ORDER BY SUM("cost") DESC`, [params.startDate.toISOString(), params.endDate.toISOString()]);
+			return rows;
+		},
+		getCostByProvider: async (params) => {
+			const { rows } = await pool.query(`SELECT "provider",
+          COALESCE(SUM("cost"), 0)::int AS "totalCost",
+          COALESCE(SUM("inputCost"), 0)::int AS "totalInputCost",
+          COALESCE(SUM("outputCost"), 0)::int AS "totalOutputCost",
+          COALESCE(SUM("totalTokens"), 0)::int AS "totalTokens",
+          COUNT(*)::int AS "requestCount",
+          AVG("latencyMs") AS "avgLatencyMs"
+        FROM "llm_requests"
+        WHERE "createdAt" >= $1 AND "createdAt" <= $2
+        GROUP BY "provider"
+        ORDER BY SUM("cost") DESC`, [params.startDate.toISOString(), params.endDate.toISOString()]);
+			return rows;
+		},
+		getDailyCosts: async (params) => {
+			const { rows } = await pool.query(`SELECT DATE("createdAt")::text AS "date",
+          COALESCE(SUM("cost"), 0)::int AS "totalCost",
+          COALESCE(SUM("inputCost"), 0)::int AS "totalInputCost",
+          COALESCE(SUM("outputCost"), 0)::int AS "totalOutputCost",
+          COALESCE(SUM("totalTokens"), 0)::int AS "totalTokens",
+          COUNT(*)::int AS "requestCount"
+        FROM "llm_requests"
+        WHERE "createdAt" >= $1 AND "createdAt" <= $2
+        GROUP BY DATE("createdAt")
+        ORDER BY DATE("createdAt") ASC`, [params.startDate.toISOString(), params.endDate.toISOString()]);
+			return rows;
+		},
+		getCostSummary: async (params) => {
+			const { startDate, endDate, groupBy, configId, variantId, environmentId, tags, tagKeys } = params;
+			const baseParams = [
+				startDate.toISOString(),
+				endDate.toISOString(),
+				configId ?? null,
+				variantId ?? null,
+				environmentId ?? null
+			];
+			if (groupBy === "tags") {
+				const conditions = [
+					`"createdAt" >= $1`,
+					`"createdAt" <= $2`,
+					`($3::uuid IS NULL OR "configId" = $3)`,
+					`($4::uuid IS NULL OR "variantId" = $4)`,
+					`($5::uuid IS NULL OR "environmentId" = $5)`
+				];
+				const queryParams = [...baseParams];
+				let idx = 5;
+				const tagFilter = buildTagFilters(tags, idx);
+				conditions.push(...tagFilter.conditions);
+				queryParams.push(...tagFilter.params);
+				idx += tagFilter.params.length;
+				if (tagKeys && tagKeys.length > 0) {
+					const keyPlaceholders = tagKeys.map((_, i) => `$${idx + i + 1}`).join(", ");
+					conditions.push(`t.key IN (${keyPlaceholders})`);
+					queryParams.push(...tagKeys);
+				}
+				const where = conditions.join(" AND ");
+				const { rows: rows$1 } = await pool.query(`SELECT t.key || ':' || t.value AS "groupKey",
+                  COALESCE(SUM("cost"), 0)::int AS "totalCost",
+                  COUNT(*)::int AS "requestCount"
+           FROM "llm_requests", jsonb_each_text("tags") t
+           WHERE ${where}
+           GROUP BY t.key, t.value
+           ORDER BY SUM("cost") DESC`, queryParams);
+				return rows$1;
+			}
+			const sqlMap = {
+				day: `SELECT DATE("createdAt")::text AS "groupKey", COALESCE(SUM("cost"),0)::int AS "totalCost", COUNT(*)::int AS "requestCount", COALESCE(SUM("totalTokens"),0)::int AS "totalTokens" FROM "llm_requests" WHERE "createdAt">=$1 AND "createdAt"<=$2 AND ($3::uuid IS NULL OR "configId"=$3) AND ($4::uuid IS NULL OR "variantId"=$4) AND ($5::uuid IS NULL OR "environmentId"=$5) GROUP BY DATE("createdAt") ORDER BY DATE("createdAt") ASC`,
+				hour: `SELECT DATE_TRUNC('hour',"createdAt")::text AS "groupKey", COALESCE(SUM("cost"),0)::int AS "totalCost", COUNT(*)::int AS "requestCount", COALESCE(SUM("totalTokens"),0)::int AS "totalTokens" FROM "llm_requests" WHERE "createdAt">=$1 AND "createdAt"<=$2 AND ($3::uuid IS NULL OR "configId"=$3) AND ($4::uuid IS NULL OR "variantId"=$4) AND ($5::uuid IS NULL OR "environmentId"=$5) GROUP BY DATE_TRUNC('hour',"createdAt") ORDER BY DATE_TRUNC('hour',"createdAt") ASC`,
+				model: `SELECT "provider"||'/'||"model" AS "groupKey", COALESCE(SUM("cost"),0)::int AS "totalCost", COUNT(*)::int AS "requestCount" FROM "llm_requests" WHERE "createdAt">=$1 AND "createdAt"<=$2 AND ($3::uuid IS NULL OR "configId"=$3) AND ($4::uuid IS NULL OR "variantId"=$4) AND ($5::uuid IS NULL OR "environmentId"=$5) GROUP BY "provider","model" ORDER BY SUM("cost") DESC`,
+				provider: `SELECT "provider" AS "groupKey", COALESCE(SUM("cost"),0)::int AS "totalCost", COUNT(*)::int AS "requestCount" FROM "llm_requests" WHERE "createdAt">=$1 AND "createdAt"<=$2 AND ($3::uuid IS NULL OR "configId"=$3) AND ($4::uuid IS NULL OR "variantId"=$4) AND ($5::uuid IS NULL OR "environmentId"=$5) GROUP BY "provider" ORDER BY SUM("cost") DESC`,
+				endpoint: `SELECT COALESCE("endpoint",'unknown') AS "groupKey", COALESCE(SUM("cost"),0)::int AS "totalCost", COUNT(*)::int AS "requestCount" FROM "llm_requests" WHERE "createdAt">=$1 AND "createdAt"<=$2 AND ($3::uuid IS NULL OR "configId"=$3) AND ($4::uuid IS NULL OR "variantId"=$4) AND ($5::uuid IS NULL OR "environmentId"=$5) GROUP BY "endpoint" ORDER BY SUM("cost") DESC`
+			};
+			const totalSql = `SELECT 'total' AS "groupKey", COALESCE(SUM("cost"),0)::int AS "totalCost", COUNT(*)::int AS "requestCount" FROM "llm_requests" WHERE "createdAt">=$1 AND "createdAt"<=$2 AND ($3::uuid IS NULL OR "configId"=$3) AND ($4::uuid IS NULL OR "variantId"=$4) AND ($5::uuid IS NULL OR "environmentId"=$5)`;
+			const sql = groupBy ? sqlMap[groupBy] ?? totalSql : totalSql;
+			const { rows } = await pool.query(sql, baseParams);
+			return rows;
+		},
+		getRequestStats: async (params) => {
+			const { startDate, endDate, configId, variantId, environmentId, tags } = params;
+			const conditions = [`"createdAt" >= $1`, `"createdAt" <= $2`];
+			const queryParams = [startDate.toISOString(), endDate.toISOString()];
+			let idx = 2;
+			if (configId) {
+				conditions.push(`"configId" = $${++idx}`);
+				queryParams.push(configId);
+			}
+			if (variantId) {
+				conditions.push(`"variantId" = $${++idx}`);
+				queryParams.push(variantId);
+			}
+			if (environmentId) {
+				conditions.push(`"environmentId" = $${++idx}`);
+				queryParams.push(environmentId);
+			}
+			const tagFilter = buildTagFilters(tags, idx);
+			conditions.push(...tagFilter.conditions);
+			queryParams.push(...tagFilter.params);
+			const where = conditions.join(" AND ");
+			const { rows } = await pool.query(`SELECT
+          COUNT(*)::int AS "totalRequests",
+          COUNT(CASE WHEN "statusCode">=200 AND "statusCode"<300 THEN 1 END)::int AS "successfulRequests",
+          COUNT(CASE WHEN "statusCode">=400 THEN 1 END)::int AS "failedRequests",
+          COUNT(CASE WHEN "isStreaming"=true THEN 1 END)::int AS "streamingRequests",
+          AVG("latencyMs") AS "avgLatencyMs",
+          MAX("latencyMs")::int AS "maxLatencyMs",
+          MIN("latencyMs")::int AS "minLatencyMs"
+        FROM "llm_requests" WHERE ${where}`, queryParams);
+			return rows[0];
+		},
+		getDistinctTags: async () => {
+			const { rows } = await pool.query(`SELECT DISTINCT key, value
+         FROM "llm_requests", jsonb_each_text("tags") AS t(key, value)
+         WHERE "tags" != '{}'::jsonb
+         ORDER BY key, value`);
+			return rows;
+		}
+	};
+}
+function createTracesStore(pool) {
+	return {
+		upsertTrace: async (data) => {
+			const result = require_types.upsertTraceSchema.safeParse(data);
+			if (!result.success) throw new Error(`Invalid trace data: ${result.error.message}`);
+			const trace = result.data;
+			const now = (/* @__PURE__ */ new Date()).toISOString();
+			await pool.query(`INSERT INTO "traces" (
+          "id","traceId","name","sessionId","userId","status",
+          "startTime","endTime","durationMs","spanCount",
+          "totalInputTokens","totalOutputTokens","totalTokens","totalCost",
+          "tags","metadata","createdAt","updatedAt"
+        ) VALUES ($1,$2,$3,$4,$5,$6,$7,$8,$9,$10,$11,$12,$13,$14,$15::jsonb,$16::jsonb,$17,$17)
+        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" = LEAST("traces"."startTime", EXCLUDED."startTime"),
+          "endTime" = GREATEST(
+            COALESCE("traces"."endTime", EXCLUDED."endTime"),
+            COALESCE(EXCLUDED."endTime", "traces"."endTime")
+          ),
+          "durationMs" = EXTRACT(EPOCH FROM (
+            GREATEST(COALESCE("traces"."endTime",EXCLUDED."endTime"),COALESCE(EXCLUDED."endTime","traces"."endTime")) -
+            LEAST("traces"."startTime", EXCLUDED."startTime")
+          ))::integer * 1000,
+          "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" = "traces"."tags" || EXCLUDED."tags",
+          "metadata" = "traces"."metadata" || EXCLUDED."metadata",
+          "updatedAt" = $17`, [
+				(0, node_crypto.randomUUID)(),
+				trace.traceId,
+				trace.name ?? null,
+				trace.sessionId ?? null,
+				trace.userId ?? null,
+				trace.status,
+				trace.startTime.toISOString(),
+				trace.endTime?.toISOString() ?? null,
+				trace.durationMs ?? null,
+				trace.spanCount,
+				trace.totalInputTokens,
+				trace.totalOutputTokens,
+				trace.totalTokens,
+				trace.totalCost,
+				JSON.stringify(trace.tags),
+				JSON.stringify(trace.metadata),
+				now
+			]);
+		},
+		batchInsertSpans: async (spans) => {
+			if (spans.length === 0) return { count: 0 };
+			const now = (/* @__PURE__ */ new Date()).toISOString();
+			const columns = [
+				"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"
+			];
+			const colNames = columns.map((c) => `"${c}"`).join(", ");
+			const params = [];
+			const valueRows = [];
+			for (const span of spans) {
+				const result = require_types.insertSpanSchema.safeParse(span);
+				if (!result.success) {
+					__llmops_core.logger.warn(`[batchInsertSpans] Skipping invalid span ${span.spanId}: ${result.error.message}`);
+					continue;
+				}
+				const s = result.data;
+				const offset = params.length;
+				const placeholders = columns.map((_, i) => `$${offset + i + 1}`).join(", ");
+				valueRows.push(`(${placeholders})`);
+				params.push((0, node_crypto.randomUUID)(), s.traceId, s.spanId, s.parentSpanId ?? null, s.name, s.kind, s.status, s.statusMessage ?? null, s.startTime.toISOString(), s.endTime?.toISOString() ?? null, s.durationMs ?? null, s.provider ?? null, s.model ?? null, s.promptTokens, s.completionTokens, s.totalTokens, s.cost, s.configId ?? null, s.variantId ?? null, s.environmentId ?? null, s.providerConfigId ?? null, s.requestId ?? null, s.source, s.input != null ? JSON.stringify(s.input) : null, s.output != null ? JSON.stringify(s.output) : null, JSON.stringify(s.attributes), now, now);
+			}
+			if (valueRows.length === 0) return { count: 0 };
+			await pool.query(`INSERT INTO "spans" (${colNames}) VALUES ${valueRows.join(", ")} ON CONFLICT ("spanId") DO NOTHING`, params);
+			return { count: valueRows.length };
+		},
+		batchInsertSpanEvents: async (events) => {
+			if (events.length === 0) return { count: 0 };
+			const now = (/* @__PURE__ */ new Date()).toISOString();
+			const columns = [
+				"id",
+				"traceId",
+				"spanId",
+				"name",
+				"timestamp",
+				"attributes",
+				"createdAt"
+			];
+			const colNames = columns.map((c) => `"${c}"`).join(", ");
+			const params = [];
+			const valueRows = [];
+			for (const event of events) {
+				const result = require_types.insertSpanEventSchema.safeParse(event);
+				if (!result.success) {
+					__llmops_core.logger.warn(`[batchInsertSpanEvents] Skipping invalid event: ${result.error.message}`);
+					continue;
+				}
+				const e = result.data;
+				const offset = params.length;
+				const placeholders = columns.map((_, i) => `$${offset + i + 1}`).join(", ");
+				valueRows.push(`(${placeholders})`);
+				params.push((0, node_crypto.randomUUID)(), e.traceId, e.spanId, e.name, e.timestamp.toISOString(), JSON.stringify(e.attributes), now);
+			}
+			if (valueRows.length === 0) return { count: 0 };
+			await pool.query(`INSERT INTO "span_events" (${colNames}) VALUES ${valueRows.join(", ")}`, params);
+			return { count: valueRows.length };
+		},
+		listTraces: async (params) => {
+			const { limit = 50, offset = 0, sessionId, userId, status, name, startDate, endDate, tags } = params ?? {};
+			const conditions = ["TRUE"];
+			const queryParams = [];
+			let idx = 0;
+			if (sessionId) {
+				conditions.push(`"sessionId" = $${++idx}`);
+				queryParams.push(sessionId);
+			}
+			if (userId) {
+				conditions.push(`"userId" = $${++idx}`);
+				queryParams.push(userId);
+			}
+			if (status) {
+				conditions.push(`"status" = $${++idx}`);
+				queryParams.push(status);
+			}
+			if (name) {
+				conditions.push(`"name" ILIKE $${++idx}`);
+				queryParams.push(`%${name}%`);
+			}
+			if (startDate) {
+				conditions.push(`"startTime" >= $${++idx}`);
+				queryParams.push(startDate.toISOString());
+			}
+			if (endDate) {
+				conditions.push(`"startTime" <= $${++idx}`);
+				queryParams.push(endDate.toISOString());
+			}
+			const tagFilter = buildTagFilters(tags, idx);
+			conditions.push(...tagFilter.conditions);
+			queryParams.push(...tagFilter.params);
+			idx += tagFilter.params.length;
+			const where = conditions.join(" AND ");
+			const total = (await pool.query(`SELECT COUNT(*)::int AS "total" FROM "traces" WHERE ${where}`, queryParams)).rows[0]?.total ?? 0;
+			return {
+				data: (await pool.query(`SELECT * FROM "traces" WHERE ${where} ORDER BY "startTime" DESC LIMIT $${++idx} OFFSET $${++idx}`, [
+					...queryParams,
+					limit,
+					offset
+				])).rows,
+				total,
+				limit,
+				offset
+			};
+		},
+		getTraceWithSpans: async (traceId) => {
+			const trace = (await pool.query(`SELECT * FROM "traces" WHERE "traceId" = $1`, [traceId])).rows[0];
+			if (!trace) return void 0;
+			const [spanResult, eventResult] = await Promise.all([pool.query(`SELECT * FROM "spans" WHERE "traceId" = $1 ORDER BY "startTime" ASC`, [traceId]), pool.query(`SELECT * FROM "span_events" WHERE "traceId" = $1 ORDER BY "timestamp" ASC`, [traceId])]);
+			return {
+				trace,
+				spans: spanResult.rows,
+				events: eventResult.rows
+			};
+		},
+		getTraceStats: async (params) => {
+			const { rows } = await pool.query(`SELECT
+          COUNT(*)::int AS "totalTraces",
+          COALESCE(AVG("durationMs"), 0) AS "avgDurationMs",
+          COUNT(CASE WHEN "status" = 'error' THEN 1 END)::int AS "errorCount",
+          COALESCE(SUM("totalCost"), 0)::int AS "totalCost",
+          COALESCE(SUM("totalTokens"), 0)::int AS "totalTokens",
+          COALESCE(SUM("spanCount"), 0)::int AS "totalSpans"
+        FROM "traces"
+        WHERE "startTime" >= $1 AND "startTime" <= $2
+          AND ($3::varchar IS NULL OR "sessionId" = $3)
+          AND ($4::varchar IS NULL OR "userId" = $4)`, [
+				params.startDate.toISOString(),
+				params.endDate.toISOString(),
+				params.sessionId ?? null,
+				params.userId ?? null
+			]);
+			return rows[0];
+		}
+	};
+}
+const pgStoreOptionsSchema = require_types.zod_default.object({ schema: require_types.zod_default.string().default("llmops") });
+/**
+* Create a PostgreSQL-backed telemetry store.
+*
+* Usage:
+* ```ts
+* import { llmops } from '@llmops/sdk'
+* import { pgStore } from '@llmops/sdk/store/pg'
+*
+* const ops = llmops({
+*   telemetry: pgStore(process.env.DATABASE_URL),
+* })
+* ```
+*/
+function createPgStore(connectionString, options) {
+	const parsed = require_types.zod_default.string().url().safeParse(connectionString);
+	if (!parsed.success) throw new Error(`pgStore: invalid connection string — ${parsed.error.issues[0]?.message ?? "expected a postgres:// URL"}`);
+	const { schema } = pgStoreOptionsSchema.parse(options ?? {});
+	let pool;
+	try {
+		pool = new (require("pg")).Pool({ connectionString });
+	} catch {
+		throw new Error("pgStore requires the \"pg\" package. Install it with: pnpm add pg");
+	}
+	pool.on("connect", (client) => {
+		client.query(`SET search_path TO "${schema}"`);
+	});
+	__llmops_core.logger.debug(`pgStore: initialized with schema "${schema}"`);
+	return {
+		...createLLMRequestsStore(pool),
+		...createTracesStore(pool),
+		_pool: pool,
+		_schema: schema
+	};
+}
+
+//#endregion
 //#region src/store/pg/migrations/000001_2191b051.sql
 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";
 
@@ -43,5 +654,5 @@ async function runMigrations(pool, schema) {
 }
 
 //#endregion
-exports.pgStore = require_pg_store.createPgStore;
+exports.pgStore = createPgStore;
 exports.runMigrations = runMigrations;

package/dist/store/pg.d.cts

@@ -1,4 +1,5 @@
-import { a as PgStore, l as createPgStore } from "../interface-DMuF7YgM.cjs";
+import "../interface-SBcQ5H-b.cjs";
+import { n as createPgStore, t as PgStore } from "../pg-store-CQMQHDtm.cjs";
 import { Pool } from "pg";
 
 //#region src/store/pg/migrate.d.ts

package/dist/store/pg.d.mts

@@ -1,4 +1,5 @@
-import { a as PgStore, l as createPgStore } from "../interface-BrJMazBg.mjs";
+import "../interface-CNLmcugg.mjs";
+import { n as createPgStore, t as PgStore } from "../pg-store-Dr0F1Jq4.mjs";
 import { Pool } from "pg";
 
 //#region src/store/pg/migrate.d.ts