@llmops/core 0.6.7-beta.3 → 0.6.7

package/dist/db/index.d.cts

@@ -1,2 +1,2 @@
-import { $ as SCHEMA_METADATA, A as DatasetsTable, At as playgroundsSchema, B as LLMRequest, Bt as workspaceSettingsSchema, C as Dataset, Ct as environmentSecretsSchema, D as DatasetVersionRecord, Dt as playgroundColumnSchema, E as DatasetVersion, Et as llmRequestsSchema, F as GuardrailConfig, Ft as spansSchema, G as PlaygroundResultsTable, H as Playground, I as GuardrailConfigsTable, It as targetingRulesSchema, J as PlaygroundsTable, K as PlaygroundRun, L as GuardrailResult, Lt as tracesSchema, M as EnvironmentSecret, Mt as providerGuardrailOverridesSchema, N as EnvironmentSecretsTable, Nt as schemas, O as DatasetVersionRecordsTable, Ot as playgroundResultsSchema, P as EnvironmentsTable, Pt as spanEventsSchema, Q as ProviderGuardrailOverridesTable, R as GuardrailResults, Rt as variantVersionsSchema, S as Database, St as datasetsSchema, T as DatasetRecordsTable, Tt as guardrailConfigsSchema, U as PlaygroundColumn, V as LLMRequestsTable, W as PlaygroundResult, X as ProviderConfigsTable, Y as ProviderConfig, Z as ProviderGuardrailOverride, _ as validateTableData, _t as configVariantsSchema, a as createDatabaseFromConnection, at as TableName, b as ConfigVariantsTable, bt as datasetVersionRecordsSchema, c as executeWithSchema, ct as Trace, d as getMigrations, dt as Variant, et as Selectable, f as matchType, ft as VariantVersion, g as validatePartialTableData, gt as WorkspaceSettingsTable, h as parseTableData, ht as WorkspaceSettings, i as createDatabase, it as SpansTable, j as Environment, jt as providerConfigsSchema, k as DatasetVersionsTable, kt as playgroundRunsSchema, l as MigrationOptions, lt as TracesTable, m as parsePartialTableData, mt as VariantsTable, n as DatabaseOptions, nt as SpanEvent, o as detectDatabaseType, ot as TargetingRule, p as runAutoMigrations, pt as VariantVersionsTable, q as PlaygroundRunsTable, r as DatabaseType, rt as SpanEventsTable, s as createNeonDialect, st as TargetingRulesTable, t as DatabaseConnection, tt as Span, u as MigrationResult, ut as Updateable, v as Config, vt as configsSchema, w as DatasetRecord, wt as environmentsSchema, x as ConfigsTable, xt as datasetVersionsSchema, y as ConfigVariant, yt as datasetRecordsSchema, z as Insertable, zt as variantsSchema } from "../index-BVI_reop.cjs";
+import { $ as SCHEMA_METADATA, A as DatasetsTable, At as playgroundsSchema, B as LLMRequest, Bt as workspaceSettingsSchema, C as Dataset, Ct as environmentSecretsSchema, D as DatasetVersionRecord, Dt as playgroundColumnSchema, E as DatasetVersion, Et as llmRequestsSchema, F as GuardrailConfig, Ft as spansSchema, G as PlaygroundResultsTable, H as Playground, I as GuardrailConfigsTable, It as targetingRulesSchema, J as PlaygroundsTable, K as PlaygroundRun, L as GuardrailResult, Lt as tracesSchema, M as EnvironmentSecret, Mt as providerGuardrailOverridesSchema, N as EnvironmentSecretsTable, Nt as schemas, O as DatasetVersionRecordsTable, Ot as playgroundResultsSchema, P as EnvironmentsTable, Pt as spanEventsSchema, Q as ProviderGuardrailOverridesTable, R as GuardrailResults, Rt as variantVersionsSchema, S as Database, St as datasetsSchema, T as DatasetRecordsTable, Tt as guardrailConfigsSchema, U as PlaygroundColumn, V as LLMRequestsTable, W as PlaygroundResult, X as ProviderConfigsTable, Y as ProviderConfig, Z as ProviderGuardrailOverride, _ as validateTableData, _t as configVariantsSchema, a as createDatabaseFromConnection, at as TableName, b as ConfigVariantsTable, bt as datasetVersionRecordsSchema, c as executeWithSchema, ct as Trace, d as getMigrations, dt as Variant, et as Selectable, f as matchType, ft as VariantVersion, g as validatePartialTableData, gt as WorkspaceSettingsTable, h as parseTableData, ht as WorkspaceSettings, i as createDatabase, it as SpansTable, j as Environment, jt as providerConfigsSchema, k as DatasetVersionsTable, kt as playgroundRunsSchema, l as MigrationOptions, lt as TracesTable, m as parsePartialTableData, mt as VariantsTable, n as DatabaseOptions, nt as SpanEvent, o as detectDatabaseType, ot as TargetingRule, p as runAutoMigrations, pt as VariantVersionsTable, q as PlaygroundRunsTable, r as DatabaseType, rt as SpanEventsTable, s as createNeonDialect, st as TargetingRulesTable, t as DatabaseConnection, tt as Span, u as MigrationResult, ut as Updateable, v as Config, vt as configsSchema, w as DatasetRecord, wt as environmentsSchema, x as ConfigsTable, xt as datasetVersionsSchema, y as ConfigVariant, yt as datasetRecordsSchema, z as Insertable, zt as variantsSchema } from "../index-CyVFWq76.cjs";
 export { Config, ConfigVariant, ConfigVariantsTable, ConfigsTable, Database, DatabaseConnection, DatabaseOptions, DatabaseType, Dataset, DatasetRecord, DatasetRecordsTable, DatasetVersion, DatasetVersionRecord, DatasetVersionRecordsTable, DatasetVersionsTable, DatasetsTable, Environment, EnvironmentSecret, EnvironmentSecretsTable, EnvironmentsTable, GuardrailConfig, GuardrailConfigsTable, GuardrailResult, GuardrailResults, Insertable, LLMRequest, LLMRequestsTable, MigrationOptions, MigrationResult, Playground, PlaygroundColumn, PlaygroundResult, PlaygroundResultsTable, PlaygroundRun, PlaygroundRunsTable, PlaygroundsTable, ProviderConfig, ProviderConfigsTable, ProviderGuardrailOverride, ProviderGuardrailOverridesTable, SCHEMA_METADATA, Selectable, Span, SpanEvent, SpanEventsTable, SpansTable, TableName, TargetingRule, TargetingRulesTable, Trace, TracesTable, Updateable, Variant, VariantVersion, VariantVersionsTable, VariantsTable, WorkspaceSettings, WorkspaceSettingsTable, configVariantsSchema, configsSchema, createDatabase, createDatabaseFromConnection, createNeonDialect, datasetRecordsSchema, datasetVersionRecordsSchema, datasetVersionsSchema, datasetsSchema, detectDatabaseType, environmentSecretsSchema, environmentsSchema, executeWithSchema, getMigrations, guardrailConfigsSchema, llmRequestsSchema, matchType, parsePartialTableData, parseTableData, playgroundColumnSchema, playgroundResultsSchema, playgroundRunsSchema, playgroundsSchema, providerConfigsSchema, providerGuardrailOverridesSchema, runAutoMigrations, schemas, spanEventsSchema, spansSchema, targetingRulesSchema, tracesSchema, validatePartialTableData, validateTableData, variantVersionsSchema, variantsSchema, workspaceSettingsSchema };

package/dist/db/index.d.mts

@@ -1,2 +1,2 @@
-import { $ as SCHEMA_METADATA, A as DatasetsTable, At as playgroundsSchema, B as LLMRequest, Bt as workspaceSettingsSchema, C as Dataset, Ct as environmentSecretsSchema, D as DatasetVersionRecord, Dt as playgroundColumnSchema, E as DatasetVersion, Et as llmRequestsSchema, F as GuardrailConfig, Ft as spansSchema, G as PlaygroundResultsTable, H as Playground, I as GuardrailConfigsTable, It as targetingRulesSchema, J as PlaygroundsTable, K as PlaygroundRun, L as GuardrailResult, Lt as tracesSchema, M as EnvironmentSecret, Mt as providerGuardrailOverridesSchema, N as EnvironmentSecretsTable, Nt as schemas, O as DatasetVersionRecordsTable, Ot as playgroundResultsSchema, P as EnvironmentsTable, Pt as spanEventsSchema, Q as ProviderGuardrailOverridesTable, R as GuardrailResults, Rt as variantVersionsSchema, S as Database, St as datasetsSchema, T as DatasetRecordsTable, Tt as guardrailConfigsSchema, U as PlaygroundColumn, V as LLMRequestsTable, W as PlaygroundResult, X as ProviderConfigsTable, Y as ProviderConfig, Z as ProviderGuardrailOverride, _ as validateTableData, _t as configVariantsSchema, a as createDatabaseFromConnection, at as TableName, b as ConfigVariantsTable, bt as datasetVersionRecordsSchema, c as executeWithSchema, ct as Trace, d as getMigrations, dt as Variant, et as Selectable, f as matchType, ft as VariantVersion, g as validatePartialTableData, gt as WorkspaceSettingsTable, h as parseTableData, ht as WorkspaceSettings, i as createDatabase, it as SpansTable, j as Environment, jt as providerConfigsSchema, k as DatasetVersionsTable, kt as playgroundRunsSchema, l as MigrationOptions, lt as TracesTable, m as parsePartialTableData, mt as VariantsTable, n as DatabaseOptions, nt as SpanEvent, o as detectDatabaseType, ot as TargetingRule, p as runAutoMigrations, pt as VariantVersionsTable, q as PlaygroundRunsTable, r as DatabaseType, rt as SpanEventsTable, s as createNeonDialect, st as TargetingRulesTable, t as DatabaseConnection, tt as Span, u as MigrationResult, ut as Updateable, v as Config, vt as configsSchema, w as DatasetRecord, wt as environmentsSchema, x as ConfigsTable, xt as datasetVersionsSchema, y as ConfigVariant, yt as datasetRecordsSchema, z as Insertable, zt as variantsSchema } from "../index-2POEhF2_.mjs";
+import { $ as SCHEMA_METADATA, A as DatasetsTable, At as playgroundsSchema, B as LLMRequest, Bt as workspaceSettingsSchema, C as Dataset, Ct as environmentSecretsSchema, D as DatasetVersionRecord, Dt as playgroundColumnSchema, E as DatasetVersion, Et as llmRequestsSchema, F as GuardrailConfig, Ft as spansSchema, G as PlaygroundResultsTable, H as Playground, I as GuardrailConfigsTable, It as targetingRulesSchema, J as PlaygroundsTable, K as PlaygroundRun, L as GuardrailResult, Lt as tracesSchema, M as EnvironmentSecret, Mt as providerGuardrailOverridesSchema, N as EnvironmentSecretsTable, Nt as schemas, O as DatasetVersionRecordsTable, Ot as playgroundResultsSchema, P as EnvironmentsTable, Pt as spanEventsSchema, Q as ProviderGuardrailOverridesTable, R as GuardrailResults, Rt as variantVersionsSchema, S as Database, St as datasetsSchema, T as DatasetRecordsTable, Tt as guardrailConfigsSchema, U as PlaygroundColumn, V as LLMRequestsTable, W as PlaygroundResult, X as ProviderConfigsTable, Y as ProviderConfig, Z as ProviderGuardrailOverride, _ as validateTableData, _t as configVariantsSchema, a as createDatabaseFromConnection, at as TableName, b as ConfigVariantsTable, bt as datasetVersionRecordsSchema, c as executeWithSchema, ct as Trace, d as getMigrations, dt as Variant, et as Selectable, f as matchType, ft as VariantVersion, g as validatePartialTableData, gt as WorkspaceSettingsTable, h as parseTableData, ht as WorkspaceSettings, i as createDatabase, it as SpansTable, j as Environment, jt as providerConfigsSchema, k as DatasetVersionsTable, kt as playgroundRunsSchema, l as MigrationOptions, lt as TracesTable, m as parsePartialTableData, mt as VariantsTable, n as DatabaseOptions, nt as SpanEvent, o as detectDatabaseType, ot as TargetingRule, p as runAutoMigrations, pt as VariantVersionsTable, q as PlaygroundRunsTable, r as DatabaseType, rt as SpanEventsTable, s as createNeonDialect, st as TargetingRulesTable, t as DatabaseConnection, tt as Span, u as MigrationResult, ut as Updateable, v as Config, vt as configsSchema, w as DatasetRecord, wt as environmentsSchema, x as ConfigsTable, xt as datasetVersionsSchema, y as ConfigVariant, yt as datasetRecordsSchema, z as Insertable, zt as variantsSchema } from "../index-D3onb7gK.mjs";
 export { Config, ConfigVariant, ConfigVariantsTable, ConfigsTable, Database, DatabaseConnection, DatabaseOptions, DatabaseType, Dataset, DatasetRecord, DatasetRecordsTable, DatasetVersion, DatasetVersionRecord, DatasetVersionRecordsTable, DatasetVersionsTable, DatasetsTable, Environment, EnvironmentSecret, EnvironmentSecretsTable, EnvironmentsTable, GuardrailConfig, GuardrailConfigsTable, GuardrailResult, GuardrailResults, Insertable, LLMRequest, LLMRequestsTable, MigrationOptions, MigrationResult, Playground, PlaygroundColumn, PlaygroundResult, PlaygroundResultsTable, PlaygroundRun, PlaygroundRunsTable, PlaygroundsTable, ProviderConfig, ProviderConfigsTable, ProviderGuardrailOverride, ProviderGuardrailOverridesTable, SCHEMA_METADATA, Selectable, Span, SpanEvent, SpanEventsTable, SpansTable, TableName, TargetingRule, TargetingRulesTable, Trace, TracesTable, Updateable, Variant, VariantVersion, VariantVersionsTable, VariantsTable, WorkspaceSettings, WorkspaceSettingsTable, configVariantsSchema, configsSchema, createDatabase, createDatabaseFromConnection, createNeonDialect, datasetRecordsSchema, datasetVersionRecordsSchema, datasetVersionsSchema, datasetsSchema, detectDatabaseType, environmentSecretsSchema, environmentsSchema, executeWithSchema, getMigrations, guardrailConfigsSchema, llmRequestsSchema, matchType, parsePartialTableData, parseTableData, playgroundColumnSchema, playgroundResultsSchema, playgroundRunsSchema, playgroundsSchema, providerConfigsSchema, providerGuardrailOverridesSchema, runAutoMigrations, schemas, spanEventsSchema, spansSchema, targetingRulesSchema, tracesSchema, validatePartialTableData, validateTableData, variantVersionsSchema, variantsSchema, workspaceSettingsSchema };

package/dist/{index-2POEhF2_.d.mts → index-CyVFWq76.d.cts}

@@ -1,7 +1,7 @@
 import { ColumnType, Dialect, Generated, Kysely, MssqlDialect, MysqlDialect, PostgresDialect, SqliteDialect } from "kysely";
-import { NeonQueryFunction } from "@neondatabase/serverless";
 import * as zod0 from "zod";
 import { z } from "zod";
+import { NeonQueryFunction } from "@neondatabase/serverless";
 
 //#region src/db/schema.d.ts
 declare const configsSchema: z.ZodObject<{
@@ -2588,22 +2588,6 @@ declare function validateTableData<T extends TableName>(table: T, data: unknown)
   id: string;
   createdAt: Date;
   updatedAt: Date;
-}> | zod0.ZodSafeParseSuccess<{
-  providerConfigId: string;
-  guardrailConfigId: string;
-  enabled: boolean;
-  id: string;
-  createdAt: Date;
-  updatedAt: Date;
-  parameters?: Record<string, unknown> | null | undefined;
-}> | zod0.ZodSafeParseError<{
-  providerConfigId: string;
-  guardrailConfigId: string;
-  enabled: boolean;
-  id: string;
-  createdAt: Date;
-  updatedAt: Date;
-  parameters?: Record<string, unknown> | null | undefined;
 }> | zod0.ZodSafeParseSuccess<{
   datasetId: string;
   versionNumber: number;
@@ -2654,6 +2638,22 @@ declare function validateTableData<T extends TableName>(table: T, data: unknown)
   id: string;
   createdAt: Date;
   updatedAt: Date;
+}> | zod0.ZodSafeParseSuccess<{
+  providerConfigId: string;
+  guardrailConfigId: string;
+  enabled: boolean;
+  id: string;
+  createdAt: Date;
+  updatedAt: Date;
+  parameters?: Record<string, unknown> | null | undefined;
+}> | zod0.ZodSafeParseError<{
+  providerConfigId: string;
+  guardrailConfigId: string;
+  enabled: boolean;
+  id: string;
+  createdAt: Date;
+  updatedAt: Date;
+  parameters?: Record<string, unknown> | null | undefined;
 }> | zod0.ZodSafeParseSuccess<{
   requestId: string;
   provider: string;
@@ -2960,22 +2960,6 @@ declare function validatePartialTableData<T extends TableName>(table: T, data: u
   id?: string | undefined;
   createdAt?: Date | undefined;
   updatedAt?: Date | undefined;
-}> | zod0.ZodSafeParseSuccess<{
-  providerConfigId?: string | undefined;
-  guardrailConfigId?: string | undefined;
-  enabled?: boolean | undefined;
-  parameters?: Record<string, unknown> | null | undefined;
-  id?: string | undefined;
-  createdAt?: Date | undefined;
-  updatedAt?: Date | undefined;
-}> | zod0.ZodSafeParseError<{
-  providerConfigId?: string | undefined;
-  guardrailConfigId?: string | undefined;
-  enabled?: boolean | undefined;
-  parameters?: Record<string, unknown> | null | undefined;
-  id?: string | undefined;
-  createdAt?: Date | undefined;
-  updatedAt?: Date | undefined;
 }> | zod0.ZodSafeParseSuccess<{
   datasetId?: string | undefined;
   versionNumber?: number | undefined;
@@ -3026,6 +3010,22 @@ declare function validatePartialTableData<T extends TableName>(table: T, data: u
   id?: string | undefined;
   createdAt?: Date | undefined;
   updatedAt?: Date | undefined;
+}> | zod0.ZodSafeParseSuccess<{
+  providerConfigId?: string | undefined;
+  guardrailConfigId?: string | undefined;
+  enabled?: boolean | undefined;
+  parameters?: Record<string, unknown> | null | undefined;
+  id?: string | undefined;
+  createdAt?: Date | undefined;
+  updatedAt?: Date | undefined;
+}> | zod0.ZodSafeParseError<{
+  providerConfigId?: string | undefined;
+  guardrailConfigId?: string | undefined;
+  enabled?: boolean | undefined;
+  parameters?: Record<string, unknown> | null | undefined;
+  id?: string | undefined;
+  createdAt?: Date | undefined;
+  updatedAt?: Date | undefined;
 }> | zod0.ZodSafeParseSuccess<{
   requestId?: string | undefined;
   configId?: string | null | undefined;
@@ -3255,14 +3255,6 @@ declare function parseTableData<T extends TableName>(table: T, data: unknown): {
   id: string;
   createdAt: Date;
   updatedAt: Date;
-} | {
-  providerConfigId: string;
-  guardrailConfigId: string;
-  enabled: boolean;
-  id: string;
-  createdAt: Date;
-  updatedAt: Date;
-  parameters?: Record<string, unknown> | null | undefined;
 } | {
   datasetId: string;
   versionNumber: number;
@@ -3288,6 +3280,14 @@ declare function parseTableData<T extends TableName>(table: T, data: unknown): {
   id: string;
   createdAt: Date;
   updatedAt: Date;
+} | {
+  providerConfigId: string;
+  guardrailConfigId: string;
+  enabled: boolean;
+  id: string;
+  createdAt: Date;
+  updatedAt: Date;
+  parameters?: Record<string, unknown> | null | undefined;
 } | {
   requestId: string;
   provider: string;
@@ -3443,14 +3443,6 @@ declare function parsePartialTableData<T extends TableName>(table: T, data: unkn
   id?: string | undefined;
   createdAt?: Date | undefined;
   updatedAt?: Date | undefined;
-} | {
-  providerConfigId?: string | undefined;
-  guardrailConfigId?: string | undefined;
-  enabled?: boolean | undefined;
-  parameters?: Record<string, unknown> | null | undefined;
-  id?: string | undefined;
-  createdAt?: Date | undefined;
-  updatedAt?: Date | undefined;
 } | {
   datasetId?: string | undefined;
   versionNumber?: number | undefined;
@@ -3476,6 +3468,14 @@ declare function parsePartialTableData<T extends TableName>(table: T, data: unkn
   id?: string | undefined;
   createdAt?: Date | undefined;
   updatedAt?: Date | undefined;
+} | {
+  providerConfigId?: string | undefined;
+  guardrailConfigId?: string | undefined;
+  enabled?: boolean | undefined;
+  parameters?: Record<string, unknown> | null | undefined;
+  id?: string | undefined;
+  createdAt?: Date | undefined;
+  updatedAt?: Date | undefined;
 } | {
   requestId?: string | undefined;
   configId?: string | null | undefined;

package/dist/{index-BVI_reop.d.cts → index-D3onb7gK.d.mts}

@@ -1,7 +1,7 @@
 import { ColumnType, Dialect, Generated, Kysely, MssqlDialect, MysqlDialect, PostgresDialect, SqliteDialect } from "kysely";
+import { NeonQueryFunction } from "@neondatabase/serverless";
 import * as zod0 from "zod";
 import { z } from "zod";
-import { NeonQueryFunction } from "@neondatabase/serverless";
 
 //#region src/db/schema.d.ts
 declare const configsSchema: z.ZodObject<{
@@ -2588,22 +2588,6 @@ declare function validateTableData<T extends TableName>(table: T, data: unknown)
   id: string;
   createdAt: Date;
   updatedAt: Date;
-}> | zod0.ZodSafeParseSuccess<{
-  providerConfigId: string;
-  guardrailConfigId: string;
-  enabled: boolean;
-  id: string;
-  createdAt: Date;
-  updatedAt: Date;
-  parameters?: Record<string, unknown> | null | undefined;
-}> | zod0.ZodSafeParseError<{
-  providerConfigId: string;
-  guardrailConfigId: string;
-  enabled: boolean;
-  id: string;
-  createdAt: Date;
-  updatedAt: Date;
-  parameters?: Record<string, unknown> | null | undefined;
 }> | zod0.ZodSafeParseSuccess<{
   datasetId: string;
   versionNumber: number;
@@ -2654,6 +2638,22 @@ declare function validateTableData<T extends TableName>(table: T, data: unknown)
   id: string;
   createdAt: Date;
   updatedAt: Date;
+}> | zod0.ZodSafeParseSuccess<{
+  providerConfigId: string;
+  guardrailConfigId: string;
+  enabled: boolean;
+  id: string;
+  createdAt: Date;
+  updatedAt: Date;
+  parameters?: Record<string, unknown> | null | undefined;
+}> | zod0.ZodSafeParseError<{
+  providerConfigId: string;
+  guardrailConfigId: string;
+  enabled: boolean;
+  id: string;
+  createdAt: Date;
+  updatedAt: Date;
+  parameters?: Record<string, unknown> | null | undefined;
 }> | zod0.ZodSafeParseSuccess<{
   requestId: string;
   provider: string;
@@ -2960,22 +2960,6 @@ declare function validatePartialTableData<T extends TableName>(table: T, data: u
   id?: string | undefined;
   createdAt?: Date | undefined;
   updatedAt?: Date | undefined;
-}> | zod0.ZodSafeParseSuccess<{
-  providerConfigId?: string | undefined;
-  guardrailConfigId?: string | undefined;
-  enabled?: boolean | undefined;
-  parameters?: Record<string, unknown> | null | undefined;
-  id?: string | undefined;
-  createdAt?: Date | undefined;
-  updatedAt?: Date | undefined;
-}> | zod0.ZodSafeParseError<{
-  providerConfigId?: string | undefined;
-  guardrailConfigId?: string | undefined;
-  enabled?: boolean | undefined;
-  parameters?: Record<string, unknown> | null | undefined;
-  id?: string | undefined;
-  createdAt?: Date | undefined;
-  updatedAt?: Date | undefined;
 }> | zod0.ZodSafeParseSuccess<{
   datasetId?: string | undefined;
   versionNumber?: number | undefined;
@@ -3026,6 +3010,22 @@ declare function validatePartialTableData<T extends TableName>(table: T, data: u
   id?: string | undefined;
   createdAt?: Date | undefined;
   updatedAt?: Date | undefined;
+}> | zod0.ZodSafeParseSuccess<{
+  providerConfigId?: string | undefined;
+  guardrailConfigId?: string | undefined;
+  enabled?: boolean | undefined;
+  parameters?: Record<string, unknown> | null | undefined;
+  id?: string | undefined;
+  createdAt?: Date | undefined;
+  updatedAt?: Date | undefined;
+}> | zod0.ZodSafeParseError<{
+  providerConfigId?: string | undefined;
+  guardrailConfigId?: string | undefined;
+  enabled?: boolean | undefined;
+  parameters?: Record<string, unknown> | null | undefined;
+  id?: string | undefined;
+  createdAt?: Date | undefined;
+  updatedAt?: Date | undefined;
 }> | zod0.ZodSafeParseSuccess<{
   requestId?: string | undefined;
   configId?: string | null | undefined;
@@ -3255,14 +3255,6 @@ declare function parseTableData<T extends TableName>(table: T, data: unknown): {
   id: string;
   createdAt: Date;
   updatedAt: Date;
-} | {
-  providerConfigId: string;
-  guardrailConfigId: string;
-  enabled: boolean;
-  id: string;
-  createdAt: Date;
-  updatedAt: Date;
-  parameters?: Record<string, unknown> | null | undefined;
 } | {
   datasetId: string;
   versionNumber: number;
@@ -3288,6 +3280,14 @@ declare function parseTableData<T extends TableName>(table: T, data: unknown): {
   id: string;
   createdAt: Date;
   updatedAt: Date;
+} | {
+  providerConfigId: string;
+  guardrailConfigId: string;
+  enabled: boolean;
+  id: string;
+  createdAt: Date;
+  updatedAt: Date;
+  parameters?: Record<string, unknown> | null | undefined;
 } | {
   requestId: string;
   provider: string;
@@ -3443,14 +3443,6 @@ declare function parsePartialTableData<T extends TableName>(table: T, data: unkn
   id?: string | undefined;
   createdAt?: Date | undefined;
   updatedAt?: Date | undefined;
-} | {
-  providerConfigId?: string | undefined;
-  guardrailConfigId?: string | undefined;
-  enabled?: boolean | undefined;
-  parameters?: Record<string, unknown> | null | undefined;
-  id?: string | undefined;
-  createdAt?: Date | undefined;
-  updatedAt?: Date | undefined;
 } | {
   datasetId?: string | undefined;
   versionNumber?: number | undefined;
@@ -3476,6 +3468,14 @@ declare function parsePartialTableData<T extends TableName>(table: T, data: unkn
   id?: string | undefined;
   createdAt?: Date | undefined;
   updatedAt?: Date | undefined;
+} | {
+  providerConfigId?: string | undefined;
+  guardrailConfigId?: string | undefined;
+  enabled?: boolean | undefined;
+  parameters?: Record<string, unknown> | null | undefined;
+  id?: string | undefined;
+  createdAt?: Date | undefined;
+  updatedAt?: Date | undefined;
 } | {
   requestId?: string | undefined;
   configId?: string | null | undefined;

package/dist/index.cjs

@@ -2070,7 +2070,8 @@ const costSummarySchema = require_db.zod_default.object({
 	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()
+	groupBy: require_db.zod_default.enum(COST_SUMMARY_GROUP_BY).optional(),
+	tagKeys: require_db.zod_default.array(require_db.zod_default.string()).optional()
 });
 /**
 * Helper to create column reference for SQL
@@ -2286,7 +2287,7 @@ const createLLMRequestsDataLayer = (db) => {
 		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 } = result.data;
+			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);
@@ -2345,6 +2346,10 @@ const createLLMRequestsDataLayer = (db) => {
 							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",