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

package/dist/{pg-store-uawkO2hJ.cjs → types-uIsMdtuF.cjs}

@@ -15,8 +15,6 @@ var __export = (all, symbols) => {
 };
 
 //#endregion
-let __llmops_core = require("@llmops/core");
-let node_crypto = require("node:crypto");
 
 //#region ../../node_modules/.pnpm/zod@4.3.5/node_modules/zod/v4/core/core.js
 /** A special constant with type `never` */
@@ -12872,7 +12870,7 @@ config(en_default());
 var zod_default = external_exports;
 
 //#endregion
-//#region src/telemetry/pg-store.ts
+//#region src/telemetry/types.ts
 const guardrailResultSchema = zod_default.object({
 	checkId: zod_default.string(),
 	functionId: zod_default.string(),
@@ -12985,612 +12983,6 @@ const insertSpanEventSchema = zod_default.object({
 	timestamp: zod_default.date(),
 	attributes: zod_default.record(zod_default.string(), zod_default.unknown()).default({})
 });
-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 = 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 = 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 = 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 = 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 = 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 = zod_default.object({ schema: 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 = 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
 Object.defineProperty(exports, 'COST_SUMMARY_GROUP_BY', {
@@ -13599,9 +12991,33 @@ Object.defineProperty(exports, 'COST_SUMMARY_GROUP_BY', {
     return COST_SUMMARY_GROUP_BY;
   }
 });
-Object.defineProperty(exports, 'createPgStore', {
+Object.defineProperty(exports, 'insertLLMRequestSchema', {
+  enumerable: true,
+  get: function () {
+    return insertLLMRequestSchema;
+  }
+});
+Object.defineProperty(exports, 'insertSpanEventSchema', {
+  enumerable: true,
+  get: function () {
+    return insertSpanEventSchema;
+  }
+});
+Object.defineProperty(exports, 'insertSpanSchema', {
+  enumerable: true,
+  get: function () {
+    return insertSpanSchema;
+  }
+});
+Object.defineProperty(exports, 'upsertTraceSchema', {
+  enumerable: true,
+  get: function () {
+    return upsertTraceSchema;
+  }
+});
+Object.defineProperty(exports, 'zod_default', {
   enumerable: true,
   get: function () {
-    return createPgStore;
+    return zod_default;
   }
 });

package/dist/types.d.cts

@@ -1,3 +1,4 @@
-import { a as PgStore, t as TelemetryStore } from "./interface-DMuF7YgM.cjs";
-import { t as D1Store } from "./d1-store-D664VLg0.cjs";
+import { t as TelemetryStore } from "./interface-SBcQ5H-b.cjs";
+import { t as D1Store } from "./d1-store-Pbuwjx6D.cjs";
+import { t as PgStore } from "./pg-store-CQMQHDtm.cjs";
 export { type D1Store, type PgStore, type TelemetryStore };

package/dist/types.d.mts

@@ -1,3 +1,4 @@
-import { a as PgStore, t as TelemetryStore } from "./interface-BrJMazBg.mjs";
-import { t as D1Store } from "./d1-store-DBf99oU8.mjs";
+import { t as TelemetryStore } from "./interface-CNLmcugg.mjs";
+import { t as D1Store } from "./d1-store-DC1FMAzk.mjs";
+import { t as PgStore } from "./pg-store-Dr0F1Jq4.mjs";
 export { type D1Store, type PgStore, type TelemetryStore };