@llmops/core 1.0.0-beta.2 → 1.0.0-beta.3

package/dist/index.cjs

@@ -1387,468 +1387,54 @@ const createDatasetsDataLayer = (db) => {
 };
 
 //#endregion
-//#region src/datalayer/llmRequests.ts
-/**
-* Schema for individual guardrail result in telemetry
-*/
-const guardrailResultSchema = require_db.zod_default.object({
-	checkId: require_db.zod_default.string(),
-	functionId: require_db.zod_default.string(),
-	hookType: require_db.zod_default.enum(["beforeRequestHook", "afterRequestHook"]),
-	verdict: require_db.zod_default.boolean(),
-	latencyMs: require_db.zod_default.number()
-});
-/**
-* Schema for guardrail results aggregate
-*/
-const guardrailResultsSchema = require_db.zod_default.object({
-	results: require_db.zod_default.array(guardrailResultSchema),
-	action: require_db.zod_default.enum([
-		"allowed",
-		"blocked",
-		"logged"
-	]),
-	totalLatencyMs: require_db.zod_default.number()
-});
-/**
-* Schema for inserting a new LLM request log
-*/
-const insertLLMRequestSchema = require_db.zod_default.object({
-	requestId: require_db.zod_default.string().uuid(),
-	configId: require_db.zod_default.string().uuid().nullable().optional(),
-	variantId: require_db.zod_default.string().uuid().nullable().optional(),
-	environmentId: require_db.zod_default.string().uuid().nullable().optional(),
-	providerConfigId: require_db.zod_default.string().uuid().nullable().optional(),
-	provider: require_db.zod_default.string(),
-	model: require_db.zod_default.string(),
-	promptTokens: require_db.zod_default.number().int().default(0),
-	completionTokens: require_db.zod_default.number().int().default(0),
-	totalTokens: require_db.zod_default.number().int().default(0),
-	cachedTokens: require_db.zod_default.number().int().default(0),
-	cacheCreationTokens: require_db.zod_default.number().int().default(0),
-	cost: require_db.zod_default.number().int().default(0),
-	cacheSavings: require_db.zod_default.number().int().default(0),
-	inputCost: require_db.zod_default.number().int().default(0),
-	outputCost: require_db.zod_default.number().int().default(0),
-	endpoint: require_db.zod_default.string(),
-	statusCode: require_db.zod_default.number().int(),
-	latencyMs: require_db.zod_default.number().int().default(0),
-	isStreaming: require_db.zod_default.boolean().default(false),
-	userId: require_db.zod_default.string().nullable().optional(),
-	tags: require_db.zod_default.record(require_db.zod_default.string(), require_db.zod_default.string()).default({}),
-	guardrailResults: guardrailResultsSchema.nullable().optional(),
-	traceId: require_db.zod_default.string().nullable().optional(),
-	spanId: require_db.zod_default.string().nullable().optional(),
-	parentSpanId: require_db.zod_default.string().nullable().optional(),
-	sessionId: require_db.zod_default.string().nullable().optional()
-});
-/**
-* Schema for listing LLM requests
-*/
-const listRequestsSchema = require_db.zod_default.object({
-	limit: require_db.zod_default.number().int().positive().max(1e3).default(100),
-	offset: require_db.zod_default.number().int().nonnegative().default(0),
-	configId: require_db.zod_default.string().uuid().optional(),
-	variantId: require_db.zod_default.string().uuid().optional(),
-	environmentId: require_db.zod_default.string().uuid().optional(),
-	providerConfigId: require_db.zod_default.string().uuid().optional(),
-	provider: require_db.zod_default.string().optional(),
-	model: require_db.zod_default.string().optional(),
-	startDate: require_db.zod_default.date().optional(),
-	endDate: require_db.zod_default.date().optional(),
-	tags: require_db.zod_default.record(require_db.zod_default.string(), require_db.zod_default.array(require_db.zod_default.string())).optional()
+//#region src/datalayer/playgrounds.ts
+const createNewPlayground = require_db.zod_default.object({
+	name: require_db.zod_default.string(),
+	datasetId: require_db.zod_default.string().uuid().nullable().optional(),
+	columns: require_db.zod_default.array(require_db.playgroundColumnSchema).nullable().optional()
 });
-/**
-* Schema for date range queries with optional filters
-*/
-const dateRangeSchema = require_db.zod_default.object({
-	startDate: require_db.zod_default.date(),
-	endDate: require_db.zod_default.date(),
-	configId: require_db.zod_default.string().uuid().optional(),
-	variantId: require_db.zod_default.string().uuid().optional(),
-	environmentId: require_db.zod_default.string().uuid().optional(),
-	tags: require_db.zod_default.record(require_db.zod_default.string(), require_db.zod_default.array(require_db.zod_default.string())).optional()
+const updatePlayground = require_db.zod_default.object({
+	playgroundId: require_db.zod_default.uuidv4(),
+	name: require_db.zod_default.string().optional(),
+	datasetId: require_db.zod_default.string().uuid().nullable().optional(),
+	columns: require_db.zod_default.array(require_db.playgroundColumnSchema).nullable().optional()
 });
-/**
-* Valid groupBy values for cost summary queries
-*/
-const COST_SUMMARY_GROUP_BY = [
-	"day",
-	"hour",
-	"model",
-	"provider",
-	"endpoint",
-	"tags"
-];
-/**
-* Schema for cost summary with grouping
-*/
-const costSummarySchema = require_db.zod_default.object({
-	startDate: require_db.zod_default.date(),
-	endDate: require_db.zod_default.date(),
-	configId: require_db.zod_default.string().uuid().optional(),
-	variantId: require_db.zod_default.string().uuid().optional(),
-	environmentId: require_db.zod_default.string().uuid().optional(),
-	tags: require_db.zod_default.record(require_db.zod_default.string(), require_db.zod_default.array(require_db.zod_default.string())).optional(),
-	groupBy: require_db.zod_default.enum(COST_SUMMARY_GROUP_BY).optional(),
-	tagKeys: require_db.zod_default.array(require_db.zod_default.string()).optional()
+const getPlaygroundById = require_db.zod_default.object({ playgroundId: require_db.zod_default.uuidv4() });
+const deletePlayground = require_db.zod_default.object({ playgroundId: require_db.zod_default.uuidv4() });
+const listPlaygrounds = require_db.zod_default.object({
+	limit: require_db.zod_default.number().int().positive().optional(),
+	offset: require_db.zod_default.number().int().nonnegative().optional()
 });
-/**
-* Helper to create column reference for SQL
-* Uses sql.ref() to properly quote column names for the database
-*/
-const col$1 = (name) => kysely.sql.ref(name);
-const createLLMRequestsDataLayer = (db) => {
+const createPlaygroundDataLayer = (db) => {
 	return {
-		batchInsertRequests: async (requests) => {
-			if (requests.length === 0) return { count: 0 };
-			const validatedRequests = await Promise.all(requests.map(async (req) => {
-				const result = await insertLLMRequestSchema.safeParseAsync(req);
-				if (!result.success) throw new LLMOpsError(`Invalid request data: ${result.error.message}`);
-				return result.data;
-			}));
-			const now = (/* @__PURE__ */ new Date()).toISOString();
-			const values = validatedRequests.map((req) => ({
-				id: (0, node_crypto.randomUUID)(),
-				requestId: req.requestId,
-				configId: req.configId ?? null,
-				variantId: req.variantId ?? null,
-				environmentId: req.environmentId ?? null,
-				providerConfigId: req.providerConfigId ?? null,
-				provider: req.provider,
-				model: req.model,
-				promptTokens: req.promptTokens,
-				completionTokens: req.completionTokens,
-				totalTokens: req.totalTokens,
-				cachedTokens: req.cachedTokens,
-				cacheCreationTokens: req.cacheCreationTokens,
-				cost: req.cost,
-				cacheSavings: req.cacheSavings,
-				inputCost: req.inputCost,
-				outputCost: req.outputCost,
-				endpoint: req.endpoint,
-				statusCode: req.statusCode,
-				latencyMs: req.latencyMs,
-				isStreaming: req.isStreaming,
-				userId: req.userId ?? null,
-				tags: JSON.stringify(req.tags),
-				guardrailResults: req.guardrailResults ? JSON.stringify(req.guardrailResults) : null,
-				traceId: req.traceId ?? null,
-				spanId: req.spanId ?? null,
-				parentSpanId: req.parentSpanId ?? null,
-				sessionId: req.sessionId ?? null,
-				createdAt: now,
-				updatedAt: now
-			}));
-			await db.insertInto("llm_requests").values(values).execute();
-			return { count: values.length };
-		},
-		insertRequest: async (request) => {
-			const result = await insertLLMRequestSchema.safeParseAsync(request);
-			if (!result.success) throw new LLMOpsError(`Invalid request data: ${result.error.message}`);
-			const req = result.data;
-			const now = (/* @__PURE__ */ new Date()).toISOString();
-			return db.insertInto("llm_requests").values({
+		createNewPlayground: async (params) => {
+			const value = await createNewPlayground.safeParseAsync(params);
+			if (!value.success) throw new LLMOpsError(`Invalid parameters: ${value.error.message}`);
+			const { name, datasetId, columns } = value.data;
+			return db.insertInto("playgrounds").values({
 				id: (0, node_crypto.randomUUID)(),
-				requestId: req.requestId,
-				configId: req.configId ?? null,
-				variantId: req.variantId ?? null,
-				environmentId: req.environmentId ?? null,
-				providerConfigId: req.providerConfigId ?? null,
-				provider: req.provider,
-				model: req.model,
-				promptTokens: req.promptTokens,
-				completionTokens: req.completionTokens,
-				totalTokens: req.totalTokens,
-				cachedTokens: req.cachedTokens,
-				cacheCreationTokens: req.cacheCreationTokens,
-				cost: req.cost,
-				cacheSavings: req.cacheSavings,
-				inputCost: req.inputCost,
-				outputCost: req.outputCost,
-				endpoint: req.endpoint,
-				statusCode: req.statusCode,
-				latencyMs: req.latencyMs,
-				isStreaming: req.isStreaming,
-				userId: req.userId ?? null,
-				tags: JSON.stringify(req.tags),
-				guardrailResults: req.guardrailResults ? JSON.stringify(req.guardrailResults) : null,
-				traceId: req.traceId ?? null,
-				spanId: req.spanId ?? null,
-				parentSpanId: req.parentSpanId ?? null,
-				sessionId: req.sessionId ?? null,
-				createdAt: now,
-				updatedAt: now
+				name,
+				datasetId: datasetId ?? null,
+				columns: columns ? JSON.stringify(columns) : null,
+				createdAt: (/* @__PURE__ */ new Date()).toISOString(),
+				updatedAt: (/* @__PURE__ */ new Date()).toISOString()
 			}).returningAll().executeTakeFirst();
 		},
-		listRequests: async (params) => {
-			const result = await listRequestsSchema.safeParseAsync(params || {});
-			if (!result.success) throw new LLMOpsError(`Invalid parameters: ${result.error.message}`);
-			const { limit, offset, configId, variantId, environmentId, providerConfigId, provider, model, startDate, endDate, tags } = result.data;
-			let baseQuery = db.selectFrom("llm_requests");
-			if (configId) baseQuery = baseQuery.where("configId", "=", configId);
-			if (variantId) baseQuery = baseQuery.where("variantId", "=", variantId);
-			if (environmentId) baseQuery = baseQuery.where("environmentId", "=", environmentId);
-			if (providerConfigId) baseQuery = baseQuery.where("providerConfigId", "=", providerConfigId);
-			if (provider) baseQuery = baseQuery.where("provider", "=", provider);
-			if (model) baseQuery = baseQuery.where("model", "=", model);
-			if (startDate) baseQuery = baseQuery.where(kysely.sql`${col$1("createdAt")} >= ${startDate.toISOString()}`);
-			if (endDate) baseQuery = baseQuery.where(kysely.sql`${col$1("createdAt")} <= ${endDate.toISOString()}`);
-			if (tags && Object.keys(tags).length > 0) for (const [key, values] of Object.entries(tags)) {
-				if (values.length === 0) continue;
-				if (values.length === 1) baseQuery = baseQuery.where(kysely.sql`${col$1("tags")}->>${key} = ${values[0]}`);
-				else {
-					const valueList = kysely.sql.join(values.map((v) => kysely.sql`${v}`));
-					baseQuery = baseQuery.where(kysely.sql`${col$1("tags")}->>${key} IN (${valueList})`);
-				}
-			}
-			const countResult = await baseQuery.select(kysely.sql`COUNT(*)`.as("total")).executeTakeFirst();
-			const total = Number(countResult?.total ?? 0);
-			return {
-				data: await baseQuery.selectAll().orderBy("createdAt", "desc").limit(limit).offset(offset).execute(),
-				total,
-				limit,
-				offset
-			};
+		updatePlayground: async (params) => {
+			const value = await updatePlayground.safeParseAsync(params);
+			if (!value.success) throw new LLMOpsError(`Invalid parameters: ${value.error.message}`);
+			const { playgroundId, name, datasetId, columns } = value.data;
+			const updateData = { updatedAt: (/* @__PURE__ */ new Date()).toISOString() };
+			if (name !== void 0) updateData.name = name;
+			if (datasetId !== void 0) updateData.datasetId = datasetId;
+			if (columns !== void 0) updateData.columns = columns ? JSON.stringify(columns) : null;
+			return db.updateTable("playgrounds").set(updateData).where("id", "=", playgroundId).returningAll().executeTakeFirst();
 		},
-		getRequestByRequestId: async (requestId) => {
-			return db.selectFrom("llm_requests").selectAll().where("requestId", "=", requestId).executeTakeFirst();
-		},
-		getTotalCost: async (params) => {
-			const result = await dateRangeSchema.safeParseAsync(params);
-			if (!result.success) throw new LLMOpsError(`Invalid parameters: ${result.error.message}`);
-			const { startDate, endDate, configId, variantId, environmentId, tags } = result.data;
-			let query = db.selectFrom("llm_requests").select([
-				kysely.sql`COALESCE(SUM(${col$1("cost")}), 0)`.as("totalCost"),
-				kysely.sql`COALESCE(SUM(${col$1("inputCost")}), 0)`.as("totalInputCost"),
-				kysely.sql`COALESCE(SUM(${col$1("outputCost")}), 0)`.as("totalOutputCost"),
-				kysely.sql`COALESCE(SUM(${col$1("promptTokens")}), 0)`.as("totalPromptTokens"),
-				kysely.sql`COALESCE(SUM(${col$1("completionTokens")}), 0)`.as("totalCompletionTokens"),
-				kysely.sql`COALESCE(SUM(${col$1("totalTokens")}), 0)`.as("totalTokens"),
-				kysely.sql`COALESCE(SUM(${col$1("cachedTokens")}), 0)`.as("totalCachedTokens"),
-				kysely.sql`COALESCE(SUM(${col$1("cacheSavings")}), 0)`.as("totalCacheSavings"),
-				kysely.sql`COUNT(*)`.as("requestCount")
-			]).where(kysely.sql`${col$1("createdAt")} >= ${startDate.toISOString()}`).where(kysely.sql`${col$1("createdAt")} <= ${endDate.toISOString()}`);
-			if (configId) query = query.where("configId", "=", configId);
-			if (variantId) query = query.where("variantId", "=", variantId);
-			if (environmentId) query = query.where("environmentId", "=", environmentId);
-			if (tags && Object.keys(tags).length > 0) for (const [key, values] of Object.entries(tags)) {
-				if (values.length === 0) continue;
-				if (values.length === 1) query = query.where(kysely.sql`${col$1("tags")}->>${key} = ${values[0]}`);
-				else {
-					const valueList = kysely.sql.join(values.map((v) => kysely.sql`${v}`));
-					query = query.where(kysely.sql`${col$1("tags")}->>${key} IN (${valueList})`);
-				}
-			}
-			return await query.executeTakeFirst();
-		},
-		getCostByModel: async (params) => {
-			const result = await dateRangeSchema.safeParseAsync(params);
-			if (!result.success) throw new LLMOpsError(`Invalid parameters: ${result.error.message}`);
-			const { startDate, endDate } = result.data;
-			return db.selectFrom("llm_requests").select([
-				"provider",
-				"model",
-				kysely.sql`COALESCE(SUM(${col$1("cost")}), 0)`.as("totalCost"),
-				kysely.sql`COALESCE(SUM(${col$1("inputCost")}), 0)`.as("totalInputCost"),
-				kysely.sql`COALESCE(SUM(${col$1("outputCost")}), 0)`.as("totalOutputCost"),
-				kysely.sql`COALESCE(SUM(${col$1("totalTokens")}), 0)`.as("totalTokens"),
-				kysely.sql`COUNT(*)`.as("requestCount"),
-				kysely.sql`AVG(${col$1("latencyMs")})`.as("avgLatencyMs")
-			]).where(kysely.sql`${col$1("createdAt")} >= ${startDate.toISOString()}`).where(kysely.sql`${col$1("createdAt")} <= ${endDate.toISOString()}`).groupBy(["provider", "model"]).orderBy(kysely.sql`SUM(${col$1("cost")})`, "desc").execute();
-		},
-		getCostByProvider: async (params) => {
-			const result = await dateRangeSchema.safeParseAsync(params);
-			if (!result.success) throw new LLMOpsError(`Invalid parameters: ${result.error.message}`);
-			const { startDate, endDate } = result.data;
-			return db.selectFrom("llm_requests").select([
-				"provider",
-				kysely.sql`COALESCE(SUM(${col$1("cost")}), 0)`.as("totalCost"),
-				kysely.sql`COALESCE(SUM(${col$1("inputCost")}), 0)`.as("totalInputCost"),
-				kysely.sql`COALESCE(SUM(${col$1("outputCost")}), 0)`.as("totalOutputCost"),
-				kysely.sql`COALESCE(SUM(${col$1("totalTokens")}), 0)`.as("totalTokens"),
-				kysely.sql`COUNT(*)`.as("requestCount"),
-				kysely.sql`AVG(${col$1("latencyMs")})`.as("avgLatencyMs")
-			]).where(kysely.sql`${col$1("createdAt")} >= ${startDate.toISOString()}`).where(kysely.sql`${col$1("createdAt")} <= ${endDate.toISOString()}`).groupBy("provider").orderBy(kysely.sql`SUM(${col$1("cost")})`, "desc").execute();
-		},
-		getDailyCosts: async (params) => {
-			const result = await dateRangeSchema.safeParseAsync(params);
-			if (!result.success) throw new LLMOpsError(`Invalid parameters: ${result.error.message}`);
-			const { startDate, endDate } = result.data;
-			return db.selectFrom("llm_requests").select([
-				kysely.sql`DATE(${col$1("createdAt")})`.as("date"),
-				kysely.sql`COALESCE(SUM(${col$1("cost")}), 0)`.as("totalCost"),
-				kysely.sql`COALESCE(SUM(${col$1("inputCost")}), 0)`.as("totalInputCost"),
-				kysely.sql`COALESCE(SUM(${col$1("outputCost")}), 0)`.as("totalOutputCost"),
-				kysely.sql`COALESCE(SUM(${col$1("totalTokens")}), 0)`.as("totalTokens"),
-				kysely.sql`COUNT(*)`.as("requestCount")
-			]).where(kysely.sql`${col$1("createdAt")} >= ${startDate.toISOString()}`).where(kysely.sql`${col$1("createdAt")} <= ${endDate.toISOString()}`).groupBy(kysely.sql`DATE(${col$1("createdAt")})`).orderBy(kysely.sql`DATE(${col$1("createdAt")})`, "asc").execute();
-		},
-		getCostSummary: async (params) => {
-			const result = await costSummarySchema.safeParseAsync(params);
-			if (!result.success) throw new LLMOpsError(`Invalid parameters: ${result.error.message}`);
-			const { startDate, endDate, groupBy, configId, variantId, environmentId, tags, tagKeys } = result.data;
-			let baseQuery = db.selectFrom("llm_requests").where(kysely.sql`${col$1("createdAt")} >= ${startDate.toISOString()}`).where(kysely.sql`${col$1("createdAt")} <= ${endDate.toISOString()}`);
-			if (configId) baseQuery = baseQuery.where("configId", "=", configId);
-			if (variantId) baseQuery = baseQuery.where("variantId", "=", variantId);
-			if (environmentId) baseQuery = baseQuery.where("environmentId", "=", environmentId);
-			if (tags && Object.keys(tags).length > 0) for (const [key, values] of Object.entries(tags)) {
-				if (values.length === 0) continue;
-				if (values.length === 1) baseQuery = baseQuery.where(kysely.sql`${col$1("tags")}->>${key} = ${values[0]}`);
-				else {
-					const valueList = kysely.sql.join(values.map((v) => kysely.sql`${v}`));
-					baseQuery = baseQuery.where(kysely.sql`${col$1("tags")}->>${key} IN (${valueList})`);
-				}
-			}
-			switch (groupBy) {
-				case "day": return baseQuery.select([
-					kysely.sql`DATE(${col$1("createdAt")})`.as("groupKey"),
-					kysely.sql`COALESCE(SUM(${col$1("cost")}), 0)`.as("totalCost"),
-					kysely.sql`COUNT(*)`.as("requestCount"),
-					kysely.sql`COALESCE(SUM(${col$1("totalTokens")}), 0)`.as("totalTokens")
-				]).groupBy(kysely.sql`DATE(${col$1("createdAt")})`).orderBy(kysely.sql`DATE(${col$1("createdAt")})`, "asc").execute();
-				case "hour": return baseQuery.select([
-					kysely.sql`DATE_TRUNC('hour', ${col$1("createdAt")})`.as("groupKey"),
-					kysely.sql`COALESCE(SUM(${col$1("cost")}), 0)`.as("totalCost"),
-					kysely.sql`COUNT(*)`.as("requestCount"),
-					kysely.sql`COALESCE(SUM(${col$1("totalTokens")}), 0)`.as("totalTokens")
-				]).groupBy(kysely.sql`DATE_TRUNC('hour', ${col$1("createdAt")})`).orderBy(kysely.sql`DATE_TRUNC('hour', ${col$1("createdAt")})`, "asc").execute();
-				case "model": return baseQuery.select([
-					kysely.sql`${col$1("provider")} || '/' || ${col$1("model")}`.as("groupKey"),
-					kysely.sql`COALESCE(SUM(${col$1("cost")}), 0)`.as("totalCost"),
-					kysely.sql`COUNT(*)`.as("requestCount")
-				]).groupBy(["provider", "model"]).orderBy(kysely.sql`SUM(${col$1("cost")})`, "desc").execute();
-				case "provider": return baseQuery.select([
-					kysely.sql`${col$1("provider")}`.as("groupKey"),
-					kysely.sql`COALESCE(SUM(${col$1("cost")}), 0)`.as("totalCost"),
-					kysely.sql`COUNT(*)`.as("requestCount")
-				]).groupBy("provider").orderBy(kysely.sql`SUM(${col$1("cost")})`, "desc").execute();
-				case "endpoint": return baseQuery.select([
-					kysely.sql`COALESCE(${col$1("endpoint")}, 'unknown')`.as("groupKey"),
-					kysely.sql`COALESCE(SUM(${col$1("cost")}), 0)`.as("totalCost"),
-					kysely.sql`COUNT(*)`.as("requestCount")
-				]).groupBy("endpoint").orderBy(kysely.sql`SUM(${col$1("cost")})`, "desc").execute();
-				case "tags": {
-					const conditions = [kysely.sql`${col$1("createdAt")} >= ${startDate.toISOString()}`, kysely.sql`${col$1("createdAt")} <= ${endDate.toISOString()}`];
-					if (configId) conditions.push(kysely.sql`${col$1("configId")} = ${configId}`);
-					if (variantId) conditions.push(kysely.sql`${col$1("variantId")} = ${variantId}`);
-					if (environmentId) conditions.push(kysely.sql`${col$1("environmentId")} = ${environmentId}`);
-					if (tags && Object.keys(tags).length > 0) for (const [key, values] of Object.entries(tags)) {
-						if (values.length === 0) continue;
-						if (values.length === 1) conditions.push(kysely.sql`${col$1("tags")}->>${key} = ${values[0]}`);
-						else {
-							const valueList = kysely.sql.join(values.map((v) => kysely.sql`${v}`));
-							conditions.push(kysely.sql`${col$1("tags")}->>${key} IN (${valueList})`);
-						}
-					}
-					if (tagKeys && tagKeys.length > 0) {
-						const tagKeyList = kysely.sql.join(tagKeys.map((k) => kysely.sql`${k}`), kysely.sql`, `);
-						conditions.push(kysely.sql`t.key IN (${tagKeyList})`);
-					}
-					const whereClause = kysely.sql.join(conditions, kysely.sql` AND `);
-					return (await kysely.sql`
-            SELECT t.key || ':' || t.value as "groupKey",
-                   COALESCE(SUM(${col$1("cost")}), 0) as "totalCost",
-                   COUNT(*) as "requestCount"
-            FROM "llm_requests", jsonb_each_text(${col$1("tags")}) t
-            WHERE ${whereClause}
-            GROUP BY t.key, t.value
-            ORDER BY SUM(${col$1("cost")}) DESC
-          `.execute(db)).rows;
-				}
-				default: return baseQuery.select([
-					kysely.sql`'total'`.as("groupKey"),
-					kysely.sql`COALESCE(SUM(${col$1("cost")}), 0)`.as("totalCost"),
-					kysely.sql`COUNT(*)`.as("requestCount")
-				]).execute();
-			}
-		},
-		getRequestStats: async (params) => {
-			const result = await dateRangeSchema.safeParseAsync(params);
-			if (!result.success) throw new LLMOpsError(`Invalid parameters: ${result.error.message}`);
-			const { startDate, endDate, configId, variantId, environmentId, tags } = result.data;
-			let query = db.selectFrom("llm_requests").select([
-				kysely.sql`COUNT(*)`.as("totalRequests"),
-				kysely.sql`COUNT(CASE WHEN ${col$1("statusCode")} >= 200 AND ${col$1("statusCode")} < 300 THEN 1 END)`.as("successfulRequests"),
-				kysely.sql`COUNT(CASE WHEN ${col$1("statusCode")} >= 400 THEN 1 END)`.as("failedRequests"),
-				kysely.sql`COUNT(CASE WHEN ${col$1("isStreaming")} = true THEN 1 END)`.as("streamingRequests"),
-				kysely.sql`AVG(${col$1("latencyMs")})`.as("avgLatencyMs"),
-				kysely.sql`MAX(${col$1("latencyMs")})`.as("maxLatencyMs"),
-				kysely.sql`MIN(${col$1("latencyMs")})`.as("minLatencyMs")
-			]).where(kysely.sql`${col$1("createdAt")} >= ${startDate.toISOString()}`).where(kysely.sql`${col$1("createdAt")} <= ${endDate.toISOString()}`);
-			if (configId) query = query.where("configId", "=", configId);
-			if (variantId) query = query.where("variantId", "=", variantId);
-			if (environmentId) query = query.where("environmentId", "=", environmentId);
-			if (tags && Object.keys(tags).length > 0) for (const [key, values] of Object.entries(tags)) {
-				if (values.length === 0) continue;
-				if (values.length === 1) query = query.where(kysely.sql`${col$1("tags")}->>${key} = ${values[0]}`);
-				else {
-					const valueList = kysely.sql.join(values.map((v) => kysely.sql`${v}`));
-					query = query.where(kysely.sql`${col$1("tags")}->>${key} IN (${valueList})`);
-				}
-			}
-			return await query.executeTakeFirst();
-		},
-		getDistinctTags: async () => {
-			return (await kysely.sql`
-        SELECT DISTINCT key, value
-        FROM llm_requests, jsonb_each_text(tags) AS t(key, value)
-        WHERE tags != '{}'::jsonb
-        ORDER BY key, value
-      `.execute(db)).rows;
-		}
-	};
-};
-
-//#endregion
-//#region src/datalayer/playgrounds.ts
-const createNewPlayground = require_db.zod_default.object({
-	name: require_db.zod_default.string(),
-	datasetId: require_db.zod_default.string().uuid().nullable().optional(),
-	columns: require_db.zod_default.array(require_db.playgroundColumnSchema).nullable().optional()
-});
-const updatePlayground = require_db.zod_default.object({
-	playgroundId: require_db.zod_default.uuidv4(),
-	name: require_db.zod_default.string().optional(),
-	datasetId: require_db.zod_default.string().uuid().nullable().optional(),
-	columns: require_db.zod_default.array(require_db.playgroundColumnSchema).nullable().optional()
-});
-const getPlaygroundById = require_db.zod_default.object({ playgroundId: require_db.zod_default.uuidv4() });
-const deletePlayground = require_db.zod_default.object({ playgroundId: require_db.zod_default.uuidv4() });
-const listPlaygrounds = require_db.zod_default.object({
-	limit: require_db.zod_default.number().int().positive().optional(),
-	offset: require_db.zod_default.number().int().nonnegative().optional()
-});
-const createPlaygroundDataLayer = (db) => {
-	return {
-		createNewPlayground: async (params) => {
-			const value = await createNewPlayground.safeParseAsync(params);
-			if (!value.success) throw new LLMOpsError(`Invalid parameters: ${value.error.message}`);
-			const { name, datasetId, columns } = value.data;
-			return db.insertInto("playgrounds").values({
-				id: (0, node_crypto.randomUUID)(),
-				name,
-				datasetId: datasetId ?? null,
-				columns: columns ? JSON.stringify(columns) : null,
-				createdAt: (/* @__PURE__ */ new Date()).toISOString(),
-				updatedAt: (/* @__PURE__ */ new Date()).toISOString()
-			}).returningAll().executeTakeFirst();
-		},
-		updatePlayground: async (params) => {
-			const value = await updatePlayground.safeParseAsync(params);
-			if (!value.success) throw new LLMOpsError(`Invalid parameters: ${value.error.message}`);
-			const { playgroundId, name, datasetId, columns } = value.data;
-			const updateData = { updatedAt: (/* @__PURE__ */ new Date()).toISOString() };
-			if (name !== void 0) updateData.name = name;
-			if (datasetId !== void 0) updateData.datasetId = datasetId;
-			if (columns !== void 0) updateData.columns = columns ? JSON.stringify(columns) : null;
-			return db.updateTable("playgrounds").set(updateData).where("id", "=", playgroundId).returningAll().executeTakeFirst();
-		},
-		getPlaygroundById: async (params) => {
-			const value = await getPlaygroundById.safeParseAsync(params);
-			if (!value.success) throw new LLMOpsError(`Invalid parameters: ${value.error.message}`);
-			const { playgroundId } = value.data;
-			return db.selectFrom("playgrounds").selectAll().where("id", "=", playgroundId).executeTakeFirst();
+		getPlaygroundById: async (params) => {
+			const value = await getPlaygroundById.safeParseAsync(params);
+			if (!value.success) throw new LLMOpsError(`Invalid parameters: ${value.error.message}`);
+			const { playgroundId } = value.data;
+			return db.selectFrom("playgrounds").selectAll().where("id", "=", playgroundId).executeTakeFirst();
 		},
 		deletePlayground: async (params) => {
 			const value = await deletePlayground.safeParseAsync(params);
@@ -2095,292 +1681,13 @@ const createPlaygroundRunsDataLayer = (db) => {
 };
 
 //#endregion
-//#region src/datalayer/traces.ts
-const col = (name) => kysely.sql.ref(name);
-/**
-* Schema for upserting a trace
-*/
-const upsertTraceSchema = require_db.zod_default.object({
-	traceId: require_db.zod_default.string(),
+//#region src/datalayer/workspaceSettings.ts
+const updateWorkspaceSettings = require_db.zod_default.object({
 	name: require_db.zod_default.string().nullable().optional(),
-	sessionId: require_db.zod_default.string().nullable().optional(),
-	userId: require_db.zod_default.string().nullable().optional(),
-	status: require_db.zod_default.enum([
-		"unset",
-		"ok",
-		"error"
-	]).default("unset"),
-	startTime: require_db.zod_default.date(),
-	endTime: require_db.zod_default.date().nullable().optional(),
-	durationMs: require_db.zod_default.number().int().nullable().optional(),
-	spanCount: require_db.zod_default.number().int().default(1),
-	totalInputTokens: require_db.zod_default.number().int().default(0),
-	totalOutputTokens: require_db.zod_default.number().int().default(0),
-	totalTokens: require_db.zod_default.number().int().default(0),
-	totalCost: require_db.zod_default.number().int().default(0),
-	tags: require_db.zod_default.record(require_db.zod_default.string(), require_db.zod_default.string()).default({}),
-	metadata: require_db.zod_default.record(require_db.zod_default.string(), require_db.zod_default.unknown()).default({})
-});
-/**
-* Schema for inserting spans
-*/
-const insertSpanSchema = require_db.zod_default.object({
-	traceId: require_db.zod_default.string(),
-	spanId: require_db.zod_default.string(),
-	parentSpanId: require_db.zod_default.string().nullable().optional(),
-	name: require_db.zod_default.string(),
-	kind: require_db.zod_default.number().int().default(1),
-	status: require_db.zod_default.number().int().default(0),
-	statusMessage: require_db.zod_default.string().nullable().optional(),
-	startTime: require_db.zod_default.date(),
-	endTime: require_db.zod_default.date().nullable().optional(),
-	durationMs: require_db.zod_default.number().int().nullable().optional(),
-	provider: require_db.zod_default.string().nullable().optional(),
-	model: require_db.zod_default.string().nullable().optional(),
-	promptTokens: require_db.zod_default.number().int().default(0),
-	completionTokens: require_db.zod_default.number().int().default(0),
-	totalTokens: require_db.zod_default.number().int().default(0),
-	cost: require_db.zod_default.number().int().default(0),
-	configId: require_db.zod_default.string().uuid().nullable().optional(),
-	variantId: require_db.zod_default.string().uuid().nullable().optional(),
-	environmentId: require_db.zod_default.string().uuid().nullable().optional(),
-	providerConfigId: require_db.zod_default.string().uuid().nullable().optional(),
-	requestId: require_db.zod_default.string().uuid().nullable().optional(),
-	source: require_db.zod_default.enum([
-		"gateway",
-		"otlp",
-		"langsmith"
-	]).default("gateway"),
-	input: require_db.zod_default.unknown().nullable().optional(),
-	output: require_db.zod_default.unknown().nullable().optional(),
-	attributes: require_db.zod_default.record(require_db.zod_default.string(), require_db.zod_default.unknown()).default({})
-});
-/**
-* Schema for inserting span events
-*/
-const insertSpanEventSchema = require_db.zod_default.object({
-	traceId: require_db.zod_default.string(),
-	spanId: require_db.zod_default.string(),
-	name: require_db.zod_default.string(),
-	timestamp: require_db.zod_default.date(),
-	attributes: require_db.zod_default.record(require_db.zod_default.string(), require_db.zod_default.unknown()).default({})
-});
-/**
-* Schema for listing traces
-*/
-const listTracesSchema = require_db.zod_default.object({
-	limit: require_db.zod_default.number().int().positive().max(1e3).default(50),
-	offset: require_db.zod_default.number().int().nonnegative().default(0),
-	sessionId: require_db.zod_default.string().optional(),
-	userId: require_db.zod_default.string().optional(),
-	status: require_db.zod_default.enum([
-		"unset",
-		"ok",
-		"error"
-	]).optional(),
-	name: require_db.zod_default.string().optional(),
-	startDate: require_db.zod_default.date().optional(),
-	endDate: require_db.zod_default.date().optional(),
-	tags: require_db.zod_default.record(require_db.zod_default.string(), require_db.zod_default.array(require_db.zod_default.string())).optional()
-});
-/**
-* Schema for trace stats query
-*/
-const traceStatsSchema = require_db.zod_default.object({
-	startDate: require_db.zod_default.date(),
-	endDate: require_db.zod_default.date(),
-	sessionId: require_db.zod_default.string().optional(),
-	userId: require_db.zod_default.string().optional()
+	setupComplete: require_db.zod_default.boolean().optional(),
+	superAdminId: require_db.zod_default.string().nullable().optional()
 });
-const createTracesDataLayer = (db) => {
-	return {
-		upsertTrace: async (data) => {
-			const result = await upsertTraceSchema.safeParseAsync(data);
-			if (!result.success) throw new LLMOpsError(`Invalid trace data: ${result.error.message}`);
-			const trace = result.data;
-			const now = (/* @__PURE__ */ new Date()).toISOString();
-			await kysely.sql`
-        INSERT INTO "traces" (
-          "id", "traceId", "name", "sessionId", "userId", "status",
-          "startTime", "endTime", "durationMs", "spanCount",
-          "totalInputTokens", "totalOutputTokens", "totalTokens", "totalCost",
-          "tags", "metadata", "createdAt", "updatedAt"
-        ) VALUES (
-          ${(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)}::jsonb, ${JSON.stringify(trace.metadata)}::jsonb,
-          ${now}, ${now}
-        )
-        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" = ${now}
-      `.execute(db);
-		},
-		batchInsertSpans: async (spans) => {
-			if (spans.length === 0) return { count: 0 };
-			const validatedSpans = [];
-			for (const span of spans) {
-				const result = await insertSpanSchema.safeParseAsync(span);
-				if (!result.success) {
-					require_db.logger.warn(`[batchInsertSpans] Skipping invalid span ${span.spanId}: ${result.error.message}`);
-					continue;
-				}
-				validatedSpans.push(result.data);
-			}
-			if (validatedSpans.length === 0) return { count: 0 };
-			const now = (/* @__PURE__ */ new Date()).toISOString();
-			const values = validatedSpans.map((span) => ({
-				id: (0, node_crypto.randomUUID)(),
-				traceId: span.traceId,
-				spanId: span.spanId,
-				parentSpanId: span.parentSpanId ?? null,
-				name: span.name,
-				kind: span.kind,
-				status: span.status,
-				statusMessage: span.statusMessage ?? null,
-				startTime: span.startTime.toISOString(),
-				endTime: span.endTime?.toISOString() ?? null,
-				durationMs: span.durationMs ?? null,
-				provider: span.provider ?? null,
-				model: span.model ?? null,
-				promptTokens: span.promptTokens,
-				completionTokens: span.completionTokens,
-				totalTokens: span.totalTokens,
-				cost: span.cost,
-				configId: span.configId ?? null,
-				variantId: span.variantId ?? null,
-				environmentId: span.environmentId ?? null,
-				providerConfigId: span.providerConfigId ?? null,
-				requestId: span.requestId ?? null,
-				source: span.source,
-				input: span.input != null ? JSON.stringify(span.input) : null,
-				output: span.output != null ? JSON.stringify(span.output) : null,
-				attributes: JSON.stringify(span.attributes),
-				createdAt: now,
-				updatedAt: now
-			}));
-			await db.insertInto("spans").values(values).onConflict((oc) => oc.column("spanId").doNothing()).execute();
-			return { count: values.length };
-		},
-		batchInsertSpanEvents: async (events) => {
-			if (events.length === 0) return { count: 0 };
-			const validatedEvents = [];
-			for (const event of events) {
-				const result = await insertSpanEventSchema.safeParseAsync(event);
-				if (!result.success) {
-					require_db.logger.warn(`[batchInsertSpanEvents] Skipping invalid event: ${result.error.message}`);
-					continue;
-				}
-				validatedEvents.push(result.data);
-			}
-			if (validatedEvents.length === 0) return { count: 0 };
-			const now = (/* @__PURE__ */ new Date()).toISOString();
-			const values = validatedEvents.map((event) => ({
-				id: (0, node_crypto.randomUUID)(),
-				traceId: event.traceId,
-				spanId: event.spanId,
-				name: event.name,
-				timestamp: event.timestamp.toISOString(),
-				attributes: JSON.stringify(event.attributes),
-				createdAt: now
-			}));
-			await db.insertInto("span_events").values(values).execute();
-			return { count: values.length };
-		},
-		listTraces: async (params) => {
-			const result = await listTracesSchema.safeParseAsync(params || {});
-			if (!result.success) throw new LLMOpsError(`Invalid parameters: ${result.error.message}`);
-			const { limit, offset, sessionId, userId, status, name, startDate, endDate, tags } = result.data;
-			let baseQuery = db.selectFrom("traces");
-			if (sessionId) baseQuery = baseQuery.where("sessionId", "=", sessionId);
-			if (userId) baseQuery = baseQuery.where("userId", "=", userId);
-			if (status) baseQuery = baseQuery.where("status", "=", status);
-			if (name) baseQuery = baseQuery.where(kysely.sql`${col("name")} ILIKE ${"%" + name + "%"}`);
-			if (startDate) baseQuery = baseQuery.where(kysely.sql`${col("startTime")} >= ${startDate.toISOString()}`);
-			if (endDate) baseQuery = baseQuery.where(kysely.sql`${col("startTime")} <= ${endDate.toISOString()}`);
-			if (tags && Object.keys(tags).length > 0) for (const [key, values] of Object.entries(tags)) {
-				if (values.length === 0) continue;
-				if (values.length === 1) baseQuery = baseQuery.where(kysely.sql`${col("tags")}->>${key} = ${values[0]}`);
-				else {
-					const valueList = kysely.sql.join(values.map((v) => kysely.sql`${v}`));
-					baseQuery = baseQuery.where(kysely.sql`${col("tags")}->>${key} IN (${valueList})`);
-				}
-			}
-			const countResult = await baseQuery.select(kysely.sql`COUNT(*)`.as("total")).executeTakeFirst();
-			const total = Number(countResult?.total ?? 0);
-			return {
-				data: await baseQuery.selectAll().orderBy("startTime", "desc").limit(limit).offset(offset).execute(),
-				total,
-				limit,
-				offset
-			};
-		},
-		getTraceWithSpans: async (traceId) => {
-			const trace = await db.selectFrom("traces").selectAll().where("traceId", "=", traceId).executeTakeFirst();
-			if (!trace) return void 0;
-			return {
-				trace,
-				spans: await db.selectFrom("spans").selectAll().where("traceId", "=", traceId).orderBy("startTime", "asc").execute(),
-				events: await db.selectFrom("span_events").selectAll().where("traceId", "=", traceId).orderBy("timestamp", "asc").execute()
-			};
-		},
-		getTraceStats: async (params) => {
-			const result = await traceStatsSchema.safeParseAsync(params);
-			if (!result.success) throw new LLMOpsError(`Invalid parameters: ${result.error.message}`);
-			const { startDate, endDate, sessionId, userId } = result.data;
-			let query = db.selectFrom("traces").select([
-				kysely.sql`COUNT(*)`.as("totalTraces"),
-				kysely.sql`COALESCE(AVG(${col("durationMs")}), 0)`.as("avgDurationMs"),
-				kysely.sql`COUNT(CASE WHEN ${col("status")} = 'error' THEN 1 END)`.as("errorCount"),
-				kysely.sql`COALESCE(SUM(${col("totalCost")}), 0)`.as("totalCost"),
-				kysely.sql`COALESCE(SUM(${col("totalTokens")}), 0)`.as("totalTokens"),
-				kysely.sql`COALESCE(SUM(${col("spanCount")}), 0)`.as("totalSpans")
-			]).where(kysely.sql`${col("startTime")} >= ${startDate.toISOString()}`).where(kysely.sql`${col("startTime")} <= ${endDate.toISOString()}`);
-			if (sessionId) query = query.where("sessionId", "=", sessionId);
-			if (userId) query = query.where("userId", "=", userId);
-			return query.executeTakeFirst();
-		}
-	};
-};
-
-//#endregion
-//#region src/datalayer/workspaceSettings.ts
-const updateWorkspaceSettings = require_db.zod_default.object({
-	name: require_db.zod_default.string().nullable().optional(),
-	setupComplete: require_db.zod_default.boolean().optional(),
-	superAdminId: require_db.zod_default.string().nullable().optional()
-});
-const createWorkspaceSettingsDataLayer = (db) => {
+const createWorkspaceSettingsDataLayer = (db) => {
 	return {
 		getWorkspaceSettings: async () => {
 			let settings = await db.selectFrom("workspace_settings").selectAll().executeTakeFirst();
@@ -2466,11 +1773,9 @@ const createWorkspaceSettingsDataLayer = (db) => {
 function createDataLayer(db) {
 	return {
 		...createDatasetsDataLayer(db),
-		...createLLMRequestsDataLayer(db),
 		...createPlaygroundDataLayer(db),
 		...createPlaygroundResultsDataLayer(db),
 		...createPlaygroundRunsDataLayer(db),
-		...createTracesDataLayer(db),
 		...createWorkspaceSettingsDataLayer(db)
 	};
 }
@@ -2750,12 +2055,659 @@ function getDefaultPricingProvider() {
 
 //#endregion
 //#region src/telemetry/postgres.ts
+const guardrailResultSchema = require_db.zod_default.object({
+	checkId: require_db.zod_default.string(),
+	functionId: require_db.zod_default.string(),
+	hookType: require_db.zod_default.enum(["beforeRequestHook", "afterRequestHook"]),
+	verdict: require_db.zod_default.boolean(),
+	latencyMs: require_db.zod_default.number()
+});
+const guardrailResultsSchema = require_db.zod_default.object({
+	results: require_db.zod_default.array(guardrailResultSchema),
+	action: require_db.zod_default.enum([
+		"allowed",
+		"blocked",
+		"logged"
+	]),
+	totalLatencyMs: require_db.zod_default.number()
+});
+const insertLLMRequestSchema = require_db.zod_default.object({
+	requestId: require_db.zod_default.string().uuid(),
+	configId: require_db.zod_default.string().uuid().nullable().optional(),
+	variantId: require_db.zod_default.string().uuid().nullable().optional(),
+	environmentId: require_db.zod_default.string().uuid().nullable().optional(),
+	providerConfigId: require_db.zod_default.string().uuid().nullable().optional(),
+	provider: require_db.zod_default.string(),
+	model: require_db.zod_default.string(),
+	promptTokens: require_db.zod_default.number().int().default(0),
+	completionTokens: require_db.zod_default.number().int().default(0),
+	totalTokens: require_db.zod_default.number().int().default(0),
+	cachedTokens: require_db.zod_default.number().int().default(0),
+	cacheCreationTokens: require_db.zod_default.number().int().default(0),
+	cost: require_db.zod_default.number().int().default(0),
+	cacheSavings: require_db.zod_default.number().int().default(0),
+	inputCost: require_db.zod_default.number().int().default(0),
+	outputCost: require_db.zod_default.number().int().default(0),
+	endpoint: require_db.zod_default.string(),
+	statusCode: require_db.zod_default.number().int(),
+	latencyMs: require_db.zod_default.number().int().default(0),
+	isStreaming: require_db.zod_default.boolean().default(false),
+	userId: require_db.zod_default.string().nullable().optional(),
+	tags: require_db.zod_default.record(require_db.zod_default.string(), require_db.zod_default.string()).default({}),
+	guardrailResults: guardrailResultsSchema.nullable().optional(),
+	traceId: require_db.zod_default.string().nullable().optional(),
+	spanId: require_db.zod_default.string().nullable().optional(),
+	parentSpanId: require_db.zod_default.string().nullable().optional(),
+	sessionId: require_db.zod_default.string().nullable().optional()
+});
+const listRequestsSchema = require_db.zod_default.object({
+	limit: require_db.zod_default.number().int().positive().max(1e3).default(100),
+	offset: require_db.zod_default.number().int().nonnegative().default(0),
+	configId: require_db.zod_default.string().uuid().optional(),
+	variantId: require_db.zod_default.string().uuid().optional(),
+	environmentId: require_db.zod_default.string().uuid().optional(),
+	providerConfigId: require_db.zod_default.string().uuid().optional(),
+	provider: require_db.zod_default.string().optional(),
+	model: require_db.zod_default.string().optional(),
+	startDate: require_db.zod_default.date().optional(),
+	endDate: require_db.zod_default.date().optional(),
+	tags: require_db.zod_default.record(require_db.zod_default.string(), require_db.zod_default.array(require_db.zod_default.string())).optional()
+});
+const dateRangeSchema = require_db.zod_default.object({
+	startDate: require_db.zod_default.date(),
+	endDate: require_db.zod_default.date(),
+	configId: require_db.zod_default.string().uuid().optional(),
+	variantId: require_db.zod_default.string().uuid().optional(),
+	environmentId: require_db.zod_default.string().uuid().optional(),
+	tags: require_db.zod_default.record(require_db.zod_default.string(), require_db.zod_default.array(require_db.zod_default.string())).optional()
+});
+const COST_SUMMARY_GROUP_BY = [
+	"day",
+	"hour",
+	"model",
+	"provider",
+	"endpoint",
+	"tags"
+];
+const costSummarySchema = require_db.zod_default.object({
+	startDate: require_db.zod_default.date(),
+	endDate: require_db.zod_default.date(),
+	configId: require_db.zod_default.string().uuid().optional(),
+	variantId: require_db.zod_default.string().uuid().optional(),
+	environmentId: require_db.zod_default.string().uuid().optional(),
+	tags: require_db.zod_default.record(require_db.zod_default.string(), require_db.zod_default.array(require_db.zod_default.string())).optional(),
+	groupBy: require_db.zod_default.enum(COST_SUMMARY_GROUP_BY).optional(),
+	tagKeys: require_db.zod_default.array(require_db.zod_default.string()).optional()
+});
+const upsertTraceSchema = require_db.zod_default.object({
+	traceId: require_db.zod_default.string(),
+	name: require_db.zod_default.string().nullable().optional(),
+	sessionId: require_db.zod_default.string().nullable().optional(),
+	userId: require_db.zod_default.string().nullable().optional(),
+	status: require_db.zod_default.enum([
+		"unset",
+		"ok",
+		"error"
+	]).default("unset"),
+	startTime: require_db.zod_default.date(),
+	endTime: require_db.zod_default.date().nullable().optional(),
+	durationMs: require_db.zod_default.number().int().nullable().optional(),
+	spanCount: require_db.zod_default.number().int().default(1),
+	totalInputTokens: require_db.zod_default.number().int().default(0),
+	totalOutputTokens: require_db.zod_default.number().int().default(0),
+	totalTokens: require_db.zod_default.number().int().default(0),
+	totalCost: require_db.zod_default.number().int().default(0),
+	tags: require_db.zod_default.record(require_db.zod_default.string(), require_db.zod_default.string()).default({}),
+	metadata: require_db.zod_default.record(require_db.zod_default.string(), require_db.zod_default.unknown()).default({})
+});
+const insertSpanSchema = require_db.zod_default.object({
+	traceId: require_db.zod_default.string(),
+	spanId: require_db.zod_default.string(),
+	parentSpanId: require_db.zod_default.string().nullable().optional(),
+	name: require_db.zod_default.string(),
+	kind: require_db.zod_default.number().int().default(1),
+	status: require_db.zod_default.number().int().default(0),
+	statusMessage: require_db.zod_default.string().nullable().optional(),
+	startTime: require_db.zod_default.date(),
+	endTime: require_db.zod_default.date().nullable().optional(),
+	durationMs: require_db.zod_default.number().int().nullable().optional(),
+	provider: require_db.zod_default.string().nullable().optional(),
+	model: require_db.zod_default.string().nullable().optional(),
+	promptTokens: require_db.zod_default.number().int().default(0),
+	completionTokens: require_db.zod_default.number().int().default(0),
+	totalTokens: require_db.zod_default.number().int().default(0),
+	cost: require_db.zod_default.number().int().default(0),
+	configId: require_db.zod_default.string().uuid().nullable().optional(),
+	variantId: require_db.zod_default.string().uuid().nullable().optional(),
+	environmentId: require_db.zod_default.string().uuid().nullable().optional(),
+	providerConfigId: require_db.zod_default.string().uuid().nullable().optional(),
+	requestId: require_db.zod_default.string().uuid().nullable().optional(),
+	source: require_db.zod_default.enum([
+		"gateway",
+		"otlp",
+		"langsmith"
+	]).default("gateway"),
+	input: require_db.zod_default.unknown().nullable().optional(),
+	output: require_db.zod_default.unknown().nullable().optional(),
+	attributes: require_db.zod_default.record(require_db.zod_default.string(), require_db.zod_default.unknown()).default({})
+});
+const insertSpanEventSchema = require_db.zod_default.object({
+	traceId: require_db.zod_default.string(),
+	spanId: require_db.zod_default.string(),
+	name: require_db.zod_default.string(),
+	timestamp: require_db.zod_default.date(),
+	attributes: require_db.zod_default.record(require_db.zod_default.string(), require_db.zod_default.unknown()).default({})
+});
+const listTracesSchema = require_db.zod_default.object({
+	limit: require_db.zod_default.number().int().positive().max(1e3).default(50),
+	offset: require_db.zod_default.number().int().nonnegative().default(0),
+	sessionId: require_db.zod_default.string().optional(),
+	userId: require_db.zod_default.string().optional(),
+	status: require_db.zod_default.enum([
+		"unset",
+		"ok",
+		"error"
+	]).optional(),
+	name: require_db.zod_default.string().optional(),
+	startDate: require_db.zod_default.date().optional(),
+	endDate: require_db.zod_default.date().optional(),
+	tags: require_db.zod_default.record(require_db.zod_default.string(), require_db.zod_default.array(require_db.zod_default.string())).optional()
+});
+const traceStatsSchema = require_db.zod_default.object({
+	startDate: require_db.zod_default.date(),
+	endDate: require_db.zod_default.date(),
+	sessionId: require_db.zod_default.string().optional(),
+	userId: require_db.zod_default.string().optional()
+});
+const col = (name) => kysely.sql.ref(name);
+function createLLMRequestsStore(db) {
+	return {
+		batchInsertRequests: async (requests) => {
+			if (requests.length === 0) return { count: 0 };
+			const validatedRequests = await Promise.all(requests.map(async (req) => {
+				const result = await insertLLMRequestSchema.safeParseAsync(req);
+				if (!result.success) throw new LLMOpsError(`Invalid request data: ${result.error.message}`);
+				return result.data;
+			}));
+			const now = (/* @__PURE__ */ new Date()).toISOString();
+			const values = validatedRequests.map((req) => ({
+				id: (0, node_crypto.randomUUID)(),
+				requestId: req.requestId,
+				configId: req.configId ?? null,
+				variantId: req.variantId ?? null,
+				environmentId: req.environmentId ?? null,
+				providerConfigId: req.providerConfigId ?? null,
+				provider: req.provider,
+				model: req.model,
+				promptTokens: req.promptTokens,
+				completionTokens: req.completionTokens,
+				totalTokens: req.totalTokens,
+				cachedTokens: req.cachedTokens,
+				cacheCreationTokens: req.cacheCreationTokens,
+				cost: req.cost,
+				cacheSavings: req.cacheSavings,
+				inputCost: req.inputCost,
+				outputCost: req.outputCost,
+				endpoint: req.endpoint,
+				statusCode: req.statusCode,
+				latencyMs: req.latencyMs,
+				isStreaming: req.isStreaming,
+				userId: req.userId ?? null,
+				tags: JSON.stringify(req.tags),
+				guardrailResults: req.guardrailResults ? JSON.stringify(req.guardrailResults) : null,
+				traceId: req.traceId ?? null,
+				spanId: req.spanId ?? null,
+				parentSpanId: req.parentSpanId ?? null,
+				sessionId: req.sessionId ?? null,
+				createdAt: now,
+				updatedAt: now
+			}));
+			await db.insertInto("llm_requests").values(values).execute();
+			return { count: values.length };
+		},
+		insertRequest: async (request) => {
+			const result = await insertLLMRequestSchema.safeParseAsync(request);
+			if (!result.success) throw new LLMOpsError(`Invalid request data: ${result.error.message}`);
+			const req = result.data;
+			const now = (/* @__PURE__ */ new Date()).toISOString();
+			return db.insertInto("llm_requests").values({
+				id: (0, node_crypto.randomUUID)(),
+				requestId: req.requestId,
+				configId: req.configId ?? null,
+				variantId: req.variantId ?? null,
+				environmentId: req.environmentId ?? null,
+				providerConfigId: req.providerConfigId ?? null,
+				provider: req.provider,
+				model: req.model,
+				promptTokens: req.promptTokens,
+				completionTokens: req.completionTokens,
+				totalTokens: req.totalTokens,
+				cachedTokens: req.cachedTokens,
+				cacheCreationTokens: req.cacheCreationTokens,
+				cost: req.cost,
+				cacheSavings: req.cacheSavings,
+				inputCost: req.inputCost,
+				outputCost: req.outputCost,
+				endpoint: req.endpoint,
+				statusCode: req.statusCode,
+				latencyMs: req.latencyMs,
+				isStreaming: req.isStreaming,
+				userId: req.userId ?? null,
+				tags: JSON.stringify(req.tags),
+				guardrailResults: req.guardrailResults ? JSON.stringify(req.guardrailResults) : null,
+				traceId: req.traceId ?? null,
+				spanId: req.spanId ?? null,
+				parentSpanId: req.parentSpanId ?? null,
+				sessionId: req.sessionId ?? null,
+				createdAt: now,
+				updatedAt: now
+			}).returningAll().executeTakeFirst();
+		},
+		listRequests: async (params) => {
+			const result = await listRequestsSchema.safeParseAsync(params || {});
+			if (!result.success) throw new LLMOpsError(`Invalid parameters: ${result.error.message}`);
+			const { limit, offset, configId, variantId, environmentId, providerConfigId, provider, model, startDate, endDate, tags } = result.data;
+			let baseQuery = db.selectFrom("llm_requests");
+			if (configId) baseQuery = baseQuery.where("configId", "=", configId);
+			if (variantId) baseQuery = baseQuery.where("variantId", "=", variantId);
+			if (environmentId) baseQuery = baseQuery.where("environmentId", "=", environmentId);
+			if (providerConfigId) baseQuery = baseQuery.where("providerConfigId", "=", providerConfigId);
+			if (provider) baseQuery = baseQuery.where("provider", "=", provider);
+			if (model) baseQuery = baseQuery.where("model", "=", model);
+			if (startDate) baseQuery = baseQuery.where(kysely.sql`${col("createdAt")} >= ${startDate.toISOString()}`);
+			if (endDate) baseQuery = baseQuery.where(kysely.sql`${col("createdAt")} <= ${endDate.toISOString()}`);
+			if (tags && Object.keys(tags).length > 0) for (const [key, values] of Object.entries(tags)) {
+				if (values.length === 0) continue;
+				if (values.length === 1) baseQuery = baseQuery.where(kysely.sql`${col("tags")}->>${key} = ${values[0]}`);
+				else {
+					const valueList = kysely.sql.join(values.map((v) => kysely.sql`${v}`));
+					baseQuery = baseQuery.where(kysely.sql`${col("tags")}->>${key} IN (${valueList})`);
+				}
+			}
+			const countResult = await baseQuery.select(kysely.sql`COUNT(*)`.as("total")).executeTakeFirst();
+			const total = Number(countResult?.total ?? 0);
+			return {
+				data: await baseQuery.selectAll().orderBy("createdAt", "desc").limit(limit).offset(offset).execute(),
+				total,
+				limit,
+				offset
+			};
+		},
+		getRequestByRequestId: async (requestId) => {
+			return db.selectFrom("llm_requests").selectAll().where("requestId", "=", requestId).executeTakeFirst();
+		},
+		getTotalCost: async (params) => {
+			const result = await dateRangeSchema.safeParseAsync(params);
+			if (!result.success) throw new LLMOpsError(`Invalid parameters: ${result.error.message}`);
+			const { startDate, endDate, configId, variantId, environmentId, tags } = result.data;
+			let query = db.selectFrom("llm_requests").select([
+				kysely.sql`COALESCE(SUM(${col("cost")}), 0)`.as("totalCost"),
+				kysely.sql`COALESCE(SUM(${col("inputCost")}), 0)`.as("totalInputCost"),
+				kysely.sql`COALESCE(SUM(${col("outputCost")}), 0)`.as("totalOutputCost"),
+				kysely.sql`COALESCE(SUM(${col("promptTokens")}), 0)`.as("totalPromptTokens"),
+				kysely.sql`COALESCE(SUM(${col("completionTokens")}), 0)`.as("totalCompletionTokens"),
+				kysely.sql`COALESCE(SUM(${col("totalTokens")}), 0)`.as("totalTokens"),
+				kysely.sql`COALESCE(SUM(${col("cachedTokens")}), 0)`.as("totalCachedTokens"),
+				kysely.sql`COALESCE(SUM(${col("cacheSavings")}), 0)`.as("totalCacheSavings"),
+				kysely.sql`COUNT(*)`.as("requestCount")
+			]).where(kysely.sql`${col("createdAt")} >= ${startDate.toISOString()}`).where(kysely.sql`${col("createdAt")} <= ${endDate.toISOString()}`);
+			if (configId) query = query.where("configId", "=", configId);
+			if (variantId) query = query.where("variantId", "=", variantId);
+			if (environmentId) query = query.where("environmentId", "=", environmentId);
+			if (tags && Object.keys(tags).length > 0) for (const [key, values] of Object.entries(tags)) {
+				if (values.length === 0) continue;
+				if (values.length === 1) query = query.where(kysely.sql`${col("tags")}->>${key} = ${values[0]}`);
+				else {
+					const valueList = kysely.sql.join(values.map((v) => kysely.sql`${v}`));
+					query = query.where(kysely.sql`${col("tags")}->>${key} IN (${valueList})`);
+				}
+			}
+			return query.executeTakeFirst();
+		},
+		getCostByModel: async (params) => {
+			const result = await dateRangeSchema.safeParseAsync(params);
+			if (!result.success) throw new LLMOpsError(`Invalid parameters: ${result.error.message}`);
+			const { startDate, endDate } = result.data;
+			return db.selectFrom("llm_requests").select([
+				"provider",
+				"model",
+				kysely.sql`COALESCE(SUM(${col("cost")}), 0)`.as("totalCost"),
+				kysely.sql`COALESCE(SUM(${col("inputCost")}), 0)`.as("totalInputCost"),
+				kysely.sql`COALESCE(SUM(${col("outputCost")}), 0)`.as("totalOutputCost"),
+				kysely.sql`COALESCE(SUM(${col("totalTokens")}), 0)`.as("totalTokens"),
+				kysely.sql`COUNT(*)`.as("requestCount"),
+				kysely.sql`AVG(${col("latencyMs")})`.as("avgLatencyMs")
+			]).where(kysely.sql`${col("createdAt")} >= ${startDate.toISOString()}`).where(kysely.sql`${col("createdAt")} <= ${endDate.toISOString()}`).groupBy(["provider", "model"]).orderBy(kysely.sql`SUM(${col("cost")})`, "desc").execute();
+		},
+		getCostByProvider: async (params) => {
+			const result = await dateRangeSchema.safeParseAsync(params);
+			if (!result.success) throw new LLMOpsError(`Invalid parameters: ${result.error.message}`);
+			const { startDate, endDate } = result.data;
+			return db.selectFrom("llm_requests").select([
+				"provider",
+				kysely.sql`COALESCE(SUM(${col("cost")}), 0)`.as("totalCost"),
+				kysely.sql`COALESCE(SUM(${col("inputCost")}), 0)`.as("totalInputCost"),
+				kysely.sql`COALESCE(SUM(${col("outputCost")}), 0)`.as("totalOutputCost"),
+				kysely.sql`COALESCE(SUM(${col("totalTokens")}), 0)`.as("totalTokens"),
+				kysely.sql`COUNT(*)`.as("requestCount"),
+				kysely.sql`AVG(${col("latencyMs")})`.as("avgLatencyMs")
+			]).where(kysely.sql`${col("createdAt")} >= ${startDate.toISOString()}`).where(kysely.sql`${col("createdAt")} <= ${endDate.toISOString()}`).groupBy("provider").orderBy(kysely.sql`SUM(${col("cost")})`, "desc").execute();
+		},
+		getDailyCosts: async (params) => {
+			const result = await dateRangeSchema.safeParseAsync(params);
+			if (!result.success) throw new LLMOpsError(`Invalid parameters: ${result.error.message}`);
+			const { startDate, endDate } = result.data;
+			return db.selectFrom("llm_requests").select([
+				kysely.sql`DATE(${col("createdAt")})`.as("date"),
+				kysely.sql`COALESCE(SUM(${col("cost")}), 0)`.as("totalCost"),
+				kysely.sql`COALESCE(SUM(${col("inputCost")}), 0)`.as("totalInputCost"),
+				kysely.sql`COALESCE(SUM(${col("outputCost")}), 0)`.as("totalOutputCost"),
+				kysely.sql`COALESCE(SUM(${col("totalTokens")}), 0)`.as("totalTokens"),
+				kysely.sql`COUNT(*)`.as("requestCount")
+			]).where(kysely.sql`${col("createdAt")} >= ${startDate.toISOString()}`).where(kysely.sql`${col("createdAt")} <= ${endDate.toISOString()}`).groupBy(kysely.sql`DATE(${col("createdAt")})`).orderBy(kysely.sql`DATE(${col("createdAt")})`, "asc").execute();
+		},
+		getCostSummary: async (params) => {
+			const result = await costSummarySchema.safeParseAsync(params);
+			if (!result.success) throw new LLMOpsError(`Invalid parameters: ${result.error.message}`);
+			const { startDate, endDate, groupBy, configId, variantId, environmentId, tags, tagKeys } = result.data;
+			let baseQuery = db.selectFrom("llm_requests").where(kysely.sql`${col("createdAt")} >= ${startDate.toISOString()}`).where(kysely.sql`${col("createdAt")} <= ${endDate.toISOString()}`);
+			if (configId) baseQuery = baseQuery.where("configId", "=", configId);
+			if (variantId) baseQuery = baseQuery.where("variantId", "=", variantId);
+			if (environmentId) baseQuery = baseQuery.where("environmentId", "=", environmentId);
+			if (tags && Object.keys(tags).length > 0) for (const [key, values] of Object.entries(tags)) {
+				if (values.length === 0) continue;
+				if (values.length === 1) baseQuery = baseQuery.where(kysely.sql`${col("tags")}->>${key} = ${values[0]}`);
+				else {
+					const valueList = kysely.sql.join(values.map((v) => kysely.sql`${v}`));
+					baseQuery = baseQuery.where(kysely.sql`${col("tags")}->>${key} IN (${valueList})`);
+				}
+			}
+			switch (groupBy) {
+				case "day": return baseQuery.select([
+					kysely.sql`DATE(${col("createdAt")})`.as("groupKey"),
+					kysely.sql`COALESCE(SUM(${col("cost")}), 0)`.as("totalCost"),
+					kysely.sql`COUNT(*)`.as("requestCount"),
+					kysely.sql`COALESCE(SUM(${col("totalTokens")}), 0)`.as("totalTokens")
+				]).groupBy(kysely.sql`DATE(${col("createdAt")})`).orderBy(kysely.sql`DATE(${col("createdAt")})`, "asc").execute();
+				case "hour": return baseQuery.select([
+					kysely.sql`DATE_TRUNC('hour', ${col("createdAt")})`.as("groupKey"),
+					kysely.sql`COALESCE(SUM(${col("cost")}), 0)`.as("totalCost"),
+					kysely.sql`COUNT(*)`.as("requestCount"),
+					kysely.sql`COALESCE(SUM(${col("totalTokens")}), 0)`.as("totalTokens")
+				]).groupBy(kysely.sql`DATE_TRUNC('hour', ${col("createdAt")})`).orderBy(kysely.sql`DATE_TRUNC('hour', ${col("createdAt")})`, "asc").execute();
+				case "model": return baseQuery.select([
+					kysely.sql`${col("provider")} || '/' || ${col("model")}`.as("groupKey"),
+					kysely.sql`COALESCE(SUM(${col("cost")}), 0)`.as("totalCost"),
+					kysely.sql`COUNT(*)`.as("requestCount")
+				]).groupBy(["provider", "model"]).orderBy(kysely.sql`SUM(${col("cost")})`, "desc").execute();
+				case "provider": return baseQuery.select([
+					kysely.sql`${col("provider")}`.as("groupKey"),
+					kysely.sql`COALESCE(SUM(${col("cost")}), 0)`.as("totalCost"),
+					kysely.sql`COUNT(*)`.as("requestCount")
+				]).groupBy("provider").orderBy(kysely.sql`SUM(${col("cost")})`, "desc").execute();
+				case "endpoint": return baseQuery.select([
+					kysely.sql`COALESCE(${col("endpoint")}, 'unknown')`.as("groupKey"),
+					kysely.sql`COALESCE(SUM(${col("cost")}), 0)`.as("totalCost"),
+					kysely.sql`COUNT(*)`.as("requestCount")
+				]).groupBy("endpoint").orderBy(kysely.sql`SUM(${col("cost")})`, "desc").execute();
+				case "tags": {
+					const conditions = [kysely.sql`${col("createdAt")} >= ${startDate.toISOString()}`, kysely.sql`${col("createdAt")} <= ${endDate.toISOString()}`];
+					if (configId) conditions.push(kysely.sql`${col("configId")} = ${configId}`);
+					if (variantId) conditions.push(kysely.sql`${col("variantId")} = ${variantId}`);
+					if (environmentId) conditions.push(kysely.sql`${col("environmentId")} = ${environmentId}`);
+					if (tags && Object.keys(tags).length > 0) for (const [key, values] of Object.entries(tags)) {
+						if (values.length === 0) continue;
+						if (values.length === 1) conditions.push(kysely.sql`${col("tags")}->>${key} = ${values[0]}`);
+						else {
+							const valueList = kysely.sql.join(values.map((v) => kysely.sql`${v}`));
+							conditions.push(kysely.sql`${col("tags")}->>${key} IN (${valueList})`);
+						}
+					}
+					if (tagKeys && tagKeys.length > 0) {
+						const tagKeyList = kysely.sql.join(tagKeys.map((k) => kysely.sql`${k}`), kysely.sql`, `);
+						conditions.push(kysely.sql`t.key IN (${tagKeyList})`);
+					}
+					const whereClause = kysely.sql.join(conditions, kysely.sql` AND `);
+					return (await kysely.sql`
+            SELECT t.key || ':' || t.value as "groupKey",
+                   COALESCE(SUM(${col("cost")}), 0) as "totalCost",
+                   COUNT(*) as "requestCount"
+            FROM "llm_requests", jsonb_each_text(${col("tags")}) t
+            WHERE ${whereClause}
+            GROUP BY t.key, t.value
+            ORDER BY SUM(${col("cost")}) DESC
+          `.execute(db)).rows;
+				}
+				default: return baseQuery.select([
+					kysely.sql`'total'`.as("groupKey"),
+					kysely.sql`COALESCE(SUM(${col("cost")}), 0)`.as("totalCost"),
+					kysely.sql`COUNT(*)`.as("requestCount")
+				]).execute();
+			}
+		},
+		getRequestStats: async (params) => {
+			const result = await dateRangeSchema.safeParseAsync(params);
+			if (!result.success) throw new LLMOpsError(`Invalid parameters: ${result.error.message}`);
+			const { startDate, endDate, configId, variantId, environmentId, tags } = result.data;
+			let query = db.selectFrom("llm_requests").select([
+				kysely.sql`COUNT(*)`.as("totalRequests"),
+				kysely.sql`COUNT(CASE WHEN ${col("statusCode")} >= 200 AND ${col("statusCode")} < 300 THEN 1 END)`.as("successfulRequests"),
+				kysely.sql`COUNT(CASE WHEN ${col("statusCode")} >= 400 THEN 1 END)`.as("failedRequests"),
+				kysely.sql`COUNT(CASE WHEN ${col("isStreaming")} = true THEN 1 END)`.as("streamingRequests"),
+				kysely.sql`AVG(${col("latencyMs")})`.as("avgLatencyMs"),
+				kysely.sql`MAX(${col("latencyMs")})`.as("maxLatencyMs"),
+				kysely.sql`MIN(${col("latencyMs")})`.as("minLatencyMs")
+			]).where(kysely.sql`${col("createdAt")} >= ${startDate.toISOString()}`).where(kysely.sql`${col("createdAt")} <= ${endDate.toISOString()}`);
+			if (configId) query = query.where("configId", "=", configId);
+			if (variantId) query = query.where("variantId", "=", variantId);
+			if (environmentId) query = query.where("environmentId", "=", environmentId);
+			if (tags && Object.keys(tags).length > 0) for (const [key, values] of Object.entries(tags)) {
+				if (values.length === 0) continue;
+				if (values.length === 1) query = query.where(kysely.sql`${col("tags")}->>${key} = ${values[0]}`);
+				else {
+					const valueList = kysely.sql.join(values.map((v) => kysely.sql`${v}`));
+					query = query.where(kysely.sql`${col("tags")}->>${key} IN (${valueList})`);
+				}
+			}
+			return query.executeTakeFirst();
+		},
+		getDistinctTags: async () => {
+			return (await kysely.sql`
+        SELECT DISTINCT key, value
+        FROM llm_requests, jsonb_each_text(tags) AS t(key, value)
+        WHERE tags != '{}'::jsonb
+        ORDER BY key, value
+      `.execute(db)).rows;
+		}
+	};
+}
+function createTracesStore(db) {
+	return {
+		upsertTrace: async (data) => {
+			const result = await upsertTraceSchema.safeParseAsync(data);
+			if (!result.success) throw new LLMOpsError(`Invalid trace data: ${result.error.message}`);
+			const trace = result.data;
+			const now = (/* @__PURE__ */ new Date()).toISOString();
+			await kysely.sql`
+        INSERT INTO "traces" (
+          "id", "traceId", "name", "sessionId", "userId", "status",
+          "startTime", "endTime", "durationMs", "spanCount",
+          "totalInputTokens", "totalOutputTokens", "totalTokens", "totalCost",
+          "tags", "metadata", "createdAt", "updatedAt"
+        ) VALUES (
+          ${(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)}::jsonb, ${JSON.stringify(trace.metadata)}::jsonb,
+          ${now}, ${now}
+        )
+        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" = ${now}
+      `.execute(db);
+		},
+		batchInsertSpans: async (spans) => {
+			if (spans.length === 0) return { count: 0 };
+			const validatedSpans = [];
+			for (const span of spans) {
+				const result = await insertSpanSchema.safeParseAsync(span);
+				if (!result.success) {
+					require_db.logger.warn(`[batchInsertSpans] Skipping invalid span ${span.spanId}: ${result.error.message}`);
+					continue;
+				}
+				validatedSpans.push(result.data);
+			}
+			if (validatedSpans.length === 0) return { count: 0 };
+			const now = (/* @__PURE__ */ new Date()).toISOString();
+			const values = validatedSpans.map((span) => ({
+				id: (0, node_crypto.randomUUID)(),
+				traceId: span.traceId,
+				spanId: span.spanId,
+				parentSpanId: span.parentSpanId ?? null,
+				name: span.name,
+				kind: span.kind,
+				status: span.status,
+				statusMessage: span.statusMessage ?? null,
+				startTime: span.startTime.toISOString(),
+				endTime: span.endTime?.toISOString() ?? null,
+				durationMs: span.durationMs ?? null,
+				provider: span.provider ?? null,
+				model: span.model ?? null,
+				promptTokens: span.promptTokens,
+				completionTokens: span.completionTokens,
+				totalTokens: span.totalTokens,
+				cost: span.cost,
+				configId: span.configId ?? null,
+				variantId: span.variantId ?? null,
+				environmentId: span.environmentId ?? null,
+				providerConfigId: span.providerConfigId ?? null,
+				requestId: span.requestId ?? null,
+				source: span.source,
+				input: span.input != null ? JSON.stringify(span.input) : null,
+				output: span.output != null ? JSON.stringify(span.output) : null,
+				attributes: JSON.stringify(span.attributes),
+				createdAt: now,
+				updatedAt: now
+			}));
+			await db.insertInto("spans").values(values).onConflict((oc) => oc.column("spanId").doNothing()).execute();
+			return { count: values.length };
+		},
+		batchInsertSpanEvents: async (events) => {
+			if (events.length === 0) return { count: 0 };
+			const validatedEvents = [];
+			for (const event of events) {
+				const result = await insertSpanEventSchema.safeParseAsync(event);
+				if (!result.success) {
+					require_db.logger.warn(`[batchInsertSpanEvents] Skipping invalid event: ${result.error.message}`);
+					continue;
+				}
+				validatedEvents.push(result.data);
+			}
+			if (validatedEvents.length === 0) return { count: 0 };
+			const now = (/* @__PURE__ */ new Date()).toISOString();
+			const values = validatedEvents.map((event) => ({
+				id: (0, node_crypto.randomUUID)(),
+				traceId: event.traceId,
+				spanId: event.spanId,
+				name: event.name,
+				timestamp: event.timestamp.toISOString(),
+				attributes: JSON.stringify(event.attributes),
+				createdAt: now
+			}));
+			await db.insertInto("span_events").values(values).execute();
+			return { count: values.length };
+		},
+		listTraces: async (params) => {
+			const result = await listTracesSchema.safeParseAsync(params || {});
+			if (!result.success) throw new LLMOpsError(`Invalid parameters: ${result.error.message}`);
+			const { limit, offset, sessionId, userId, status, name, startDate, endDate, tags } = result.data;
+			let baseQuery = db.selectFrom("traces");
+			if (sessionId) baseQuery = baseQuery.where("sessionId", "=", sessionId);
+			if (userId) baseQuery = baseQuery.where("userId", "=", userId);
+			if (status) baseQuery = baseQuery.where("status", "=", status);
+			if (name) baseQuery = baseQuery.where(kysely.sql`${col("name")} ILIKE ${"%" + name + "%"}`);
+			if (startDate) baseQuery = baseQuery.where(kysely.sql`${col("startTime")} >= ${startDate.toISOString()}`);
+			if (endDate) baseQuery = baseQuery.where(kysely.sql`${col("startTime")} <= ${endDate.toISOString()}`);
+			if (tags && Object.keys(tags).length > 0) for (const [key, values] of Object.entries(tags)) {
+				if (values.length === 0) continue;
+				if (values.length === 1) baseQuery = baseQuery.where(kysely.sql`${col("tags")}->>${key} = ${values[0]}`);
+				else {
+					const valueList = kysely.sql.join(values.map((v) => kysely.sql`${v}`));
+					baseQuery = baseQuery.where(kysely.sql`${col("tags")}->>${key} IN (${valueList})`);
+				}
+			}
+			const countResult = await baseQuery.select(kysely.sql`COUNT(*)`.as("total")).executeTakeFirst();
+			const total = Number(countResult?.total ?? 0);
+			return {
+				data: await baseQuery.selectAll().orderBy("startTime", "desc").limit(limit).offset(offset).execute(),
+				total,
+				limit,
+				offset
+			};
+		},
+		getTraceWithSpans: async (traceId) => {
+			const trace = await db.selectFrom("traces").selectAll().where("traceId", "=", traceId).executeTakeFirst();
+			if (!trace) return void 0;
+			return {
+				trace,
+				spans: await db.selectFrom("spans").selectAll().where("traceId", "=", traceId).orderBy("startTime", "asc").execute(),
+				events: await db.selectFrom("span_events").selectAll().where("traceId", "=", traceId).orderBy("timestamp", "asc").execute()
+			};
+		},
+		getTraceStats: async (params) => {
+			const result = await traceStatsSchema.safeParseAsync(params);
+			if (!result.success) throw new LLMOpsError(`Invalid parameters: ${result.error.message}`);
+			const { startDate, endDate, sessionId, userId } = result.data;
+			let query = db.selectFrom("traces").select([
+				kysely.sql`COUNT(*)`.as("totalTraces"),
+				kysely.sql`COALESCE(AVG(${col("durationMs")}), 0)`.as("avgDurationMs"),
+				kysely.sql`COUNT(CASE WHEN ${col("status")} = 'error' THEN 1 END)`.as("errorCount"),
+				kysely.sql`COALESCE(SUM(${col("totalCost")}), 0)`.as("totalCost"),
+				kysely.sql`COALESCE(SUM(${col("totalTokens")}), 0)`.as("totalTokens"),
+				kysely.sql`COALESCE(SUM(${col("spanCount")}), 0)`.as("totalSpans")
+			]).where(kysely.sql`${col("startTime")} >= ${startDate.toISOString()}`).where(kysely.sql`${col("startTime")} <= ${endDate.toISOString()}`);
+			if (sessionId) query = query.where("sessionId", "=", sessionId);
+			if (userId) query = query.where("userId", "=", userId);
+			return query.executeTakeFirst();
+		}
+	};
+}
 /**
 * Create a PostgreSQL-backed telemetry store.
 *
 * Usage:
 * ```ts
-* import { llmops, pgStore } from '@llmops/sdk'
+* import { llmops } from '@llmops/sdk'
+* import { pgStore } from '@llmops/sdk/store/pg'
 *
 * const ops = llmops({
 *   telemetry: pgStore(process.env.DATABASE_URL),
@@ -2776,12 +2728,10 @@ function createPgStore(connectionString, options) {
 			await connection.executeQuery(kysely.CompiledQuery.raw(`SET search_path TO "${schema}"`));
 		}
 	}) });
-	const llmRequests = createLLMRequestsDataLayer(db);
-	const traces = createTracesDataLayer(db);
 	require_db.logger.debug(`pgStore: initialized with schema "${schema}"`);
 	return {
-		...llmRequests,
-		...traces,
+		...createLLMRequestsStore(db),
+		...createTracesStore(db),
 		_db: db
 	};
 }
@@ -2813,13 +2763,11 @@ exports.createDataLayer = createDataLayer;
 exports.createDatabase = require_db.createDatabase;
 exports.createDatabaseFromConnection = require_db.createDatabaseFromConnection;
 exports.createDatasetsDataLayer = createDatasetsDataLayer;
-exports.createLLMRequestsDataLayer = createLLMRequestsDataLayer;
 exports.createNeonDialect = require_neon_dialect.createNeonDialect;
 exports.createPgStore = createPgStore;
 exports.createPlaygroundDataLayer = createPlaygroundDataLayer;
 exports.createPlaygroundResultsDataLayer = createPlaygroundResultsDataLayer;
 exports.createPlaygroundRunsDataLayer = createPlaygroundRunsDataLayer;
-exports.createTracesDataLayer = createTracesDataLayer;
 exports.createWorkspaceSettingsDataLayer = createWorkspaceSettingsDataLayer;
 exports.datasetRecordsSchema = require_db.datasetRecordsSchema;
 exports.datasetVersionRecordsSchema = require_db.datasetVersionRecordsSchema;