@llmops/core 0.6.7-beta.3 → 0.6.8

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -1,4 +1,4 @@
1
- const require_db = require('./db-i0OOYxJm.cjs');
1
+ const require_db = require('./db-DsEVRoua.cjs');
2
2
  let kysely = require("kysely");
3
3
 
4
4
  //#region src/db/bun-sqlite-dialect.ts
package/dist/db/index.cjs CHANGED
@@ -1,5 +1,5 @@
1
- const require_db = require('../db-i0OOYxJm.cjs');
2
- const require_neon_dialect = require('../neon-dialect-oh8u9vRy.cjs');
1
+ const require_db = require('../db-DsEVRoua.cjs');
2
+ const require_neon_dialect = require('../neon-dialect-Dn80Acj9.cjs');
3
3
 
4
4
  exports.SCHEMA_METADATA = require_db.SCHEMA_METADATA;
5
5
  exports.configVariantsSchema = require_db.configVariantsSchema;
@@ -1,2 +1,2 @@
1
- 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";
1
+ 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";
2
2
  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 };
@@ -1,2 +1,2 @@
1
- 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";
1
+ 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";
2
2
  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.mjs CHANGED
@@ -1,4 +1,4 @@
1
- import { A as schemas, C as llmRequestsSchema, D as playgroundsSchema, E as playgroundRunsSchema, F as variantVersionsSchema, I as variantsSchema, L as workspaceSettingsSchema, M as spansSchema, N as targetingRulesSchema, O as providerConfigsSchema, P as tracesSchema, S as guardrailConfigsSchema, T as playgroundResultsSchema, _ as datasetVersionRecordsSchema, a as matchType, b as environmentSecretsSchema, d as validatePartialTableData, f as validateTableData, g as datasetRecordsSchema, h as configsSchema, i as getMigrations, j as spanEventsSchema, k as providerGuardrailOverridesSchema, l as parsePartialTableData, m as configVariantsSchema, n as createDatabaseFromConnection, o as runAutoMigrations, p as SCHEMA_METADATA, r as detectDatabaseType, t as createDatabase, u as parseTableData, v as datasetVersionsSchema, w as playgroundColumnSchema, x as environmentsSchema, y as datasetsSchema } from "../db-Cd1KR24Y.mjs";
1
+ import { A as schemas, C as llmRequestsSchema, D as playgroundsSchema, E as playgroundRunsSchema, F as variantVersionsSchema, I as variantsSchema, L as workspaceSettingsSchema, M as spansSchema, N as targetingRulesSchema, O as providerConfigsSchema, P as tracesSchema, S as guardrailConfigsSchema, T as playgroundResultsSchema, _ as datasetVersionRecordsSchema, a as matchType, b as environmentSecretsSchema, d as validatePartialTableData, f as validateTableData, g as datasetRecordsSchema, h as configsSchema, i as getMigrations, j as spanEventsSchema, k as providerGuardrailOverridesSchema, l as parsePartialTableData, m as configVariantsSchema, n as createDatabaseFromConnection, o as runAutoMigrations, p as SCHEMA_METADATA, r as detectDatabaseType, t as createDatabase, u as parseTableData, v as datasetVersionsSchema, w as playgroundColumnSchema, x as environmentsSchema, y as datasetsSchema } from "../db-H8SfNI38.mjs";
2
2
  import { n as executeWithSchema, t as createNeonDialect } from "../neon-dialect-DySGBYUi.mjs";
3
3
 
4
4
  export { SCHEMA_METADATA, 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 };
@@ -39,7 +39,6 @@ var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__ge
39
39
 
40
40
  //#endregion
41
41
  let kysely = require("kysely");
42
- let better_auth_db = require("better-auth/db");
43
42
  let pino = require("pino");
44
43
  pino = __toESM(pino);
45
44
 
@@ -14545,10 +14544,29 @@ async function getMigrations(db, dbType, options) {
14545
14544
  } catch (error$47) {
14546
14545
  logger.warn({ error: error$47 }, "Failed to set search_path for Better Auth migrations");
14547
14546
  }
14548
- const { toBeAdded: authChangesToBeAdded, toBeCreated: authChangesToBeCreated, runMigrations: runAuthMigrations } = await (0, better_auth_db.getMigrations)(getAuthClientOptions({ database: {
14547
+ let getAuthMigrations;
14548
+ try {
14549
+ ({getMigrations: getAuthMigrations} = await import("better-auth/db"));
14550
+ } catch (error$47) {
14551
+ logger.warn({ error: error$47 }, "Failed to import better-auth/db; skipping auth migrations");
14552
+ return {
14553
+ toBeCreated,
14554
+ toBeAdded,
14555
+ runMigrations: async () => {
14556
+ for (const migration of migrations) await migration.execute();
14557
+ },
14558
+ compileMigrations: async () => {
14559
+ return migrations.map((m) => m.compile().sql).join(";\n\n") + ";";
14560
+ },
14561
+ migrations,
14562
+ needsMigration: toBeCreated.length > 0 || toBeAdded.length > 0
14563
+ };
14564
+ }
14565
+ const authOptions = getAuthClientOptions({ database: {
14549
14566
  db,
14550
14567
  type: dbType === "neon" ? "postgres" : dbType
14551
- } }));
14568
+ } });
14569
+ const { toBeAdded: authChangesToBeAdded, toBeCreated: authChangesToBeCreated, runMigrations: runAuthMigrations } = await getAuthMigrations(authOptions);
14552
14570
  async function runMigrations() {
14553
14571
  for (const migration of migrations) await migration.execute();
14554
14572
  await runAuthMigrations();
@@ -14695,12 +14713,12 @@ async function createDatabaseFromConnection(rawConnection, options) {
14695
14713
  case "sqlite":
14696
14714
  if ("aggregate" in rawConnection && !("createSession" in rawConnection)) dialect = new kysely.SqliteDialect({ database: rawConnection });
14697
14715
  else if ("fileControl" in rawConnection) {
14698
- const { BunSqliteDialect } = await Promise.resolve().then(() => require("./bun-sqlite-dialect-BUnuGpx1.cjs"));
14716
+ const { BunSqliteDialect } = await Promise.resolve().then(() => require("./bun-sqlite-dialect-CweOAhRE.cjs"));
14699
14717
  dialect = new BunSqliteDialect({ database: rawConnection });
14700
14718
  } else if ("createSession" in rawConnection && typeof window === "undefined") try {
14701
14719
  const { DatabaseSync } = await import("node:sqlite");
14702
14720
  if (rawConnection instanceof DatabaseSync) {
14703
- const { NodeSqliteDialect } = await Promise.resolve().then(() => require("./node-sqlite-dialect-CbwETvHG.cjs"));
14721
+ const { NodeSqliteDialect } = await Promise.resolve().then(() => require("./node-sqlite-dialect-DyFzt-h5.cjs"));
14704
14722
  dialect = new NodeSqliteDialect({ database: rawConnection });
14705
14723
  }
14706
14724
  } catch {}
@@ -14718,7 +14736,7 @@ async function createDatabaseFromConnection(rawConnection, options) {
14718
14736
  break;
14719
14737
  case "neon": {
14720
14738
  if (typeof rawConnection === "function") {
14721
- const { createNeonDialect: createNeonDialect$1 } = await Promise.resolve().then(() => require("./neon-dialect-DavGzunb.cjs"));
14739
+ const { createNeonDialect: createNeonDialect$1 } = await Promise.resolve().then(() => require("./neon-dialect-Bk6Cncio.cjs"));
14722
14740
  dialect = createNeonDialect$1(rawConnection);
14723
14741
  break;
14724
14742
  }
@@ -14726,7 +14744,7 @@ async function createDatabaseFromConnection(rawConnection, options) {
14726
14744
  if (!connectionString) throw new Error("Neon connection string is required. Pass it directly as the database option or set one of: NEON_CONNECTION_STRING, NEON_PG_URL, DATABASE_URL, POSTGRES_URL");
14727
14745
  const { neon } = await import("@neondatabase/serverless");
14728
14746
  const sql$1 = neon(connectionString);
14729
- const { createNeonDialect } = await Promise.resolve().then(() => require("./neon-dialect-DavGzunb.cjs"));
14747
+ const { createNeonDialect } = await Promise.resolve().then(() => require("./neon-dialect-Bk6Cncio.cjs"));
14730
14748
  dialect = createNeonDialect(sql$1);
14731
14749
  break;
14732
14750
  }
@@ -1,5 +1,4 @@
1
1
  import { CompiledQuery, Kysely, MssqlDialect, MysqlDialect, PostgresDialect, SqliteDialect, sql } from "kysely";
2
- import { getMigrations } from "better-auth/db";
3
2
  import pino from "pino";
4
3
 
5
4
  //#region rolldown:runtime
@@ -14369,7 +14368,7 @@ async function ensurePostgresSchemaExists(db, schema) {
14369
14368
  logger.warn({ error: error$47 }, "Failed to ensure PostgreSQL schema exists");
14370
14369
  }
14371
14370
  }
14372
- async function getMigrations$1(db, dbType, options) {
14371
+ async function getMigrations(db, dbType, options) {
14373
14372
  let currentSchema = "public";
14374
14373
  if (dbType === "postgres" || dbType === "neon") {
14375
14374
  if (options?.schema) {
@@ -14521,10 +14520,29 @@ async function getMigrations$1(db, dbType, options) {
14521
14520
  } catch (error$47) {
14522
14521
  logger.warn({ error: error$47 }, "Failed to set search_path for Better Auth migrations");
14523
14522
  }
14524
- const { toBeAdded: authChangesToBeAdded, toBeCreated: authChangesToBeCreated, runMigrations: runAuthMigrations } = await getMigrations(getAuthClientOptions({ database: {
14523
+ let getAuthMigrations;
14524
+ try {
14525
+ ({getMigrations: getAuthMigrations} = await import("better-auth/db"));
14526
+ } catch (error$47) {
14527
+ logger.warn({ error: error$47 }, "Failed to import better-auth/db; skipping auth migrations");
14528
+ return {
14529
+ toBeCreated,
14530
+ toBeAdded,
14531
+ runMigrations: async () => {
14532
+ for (const migration of migrations) await migration.execute();
14533
+ },
14534
+ compileMigrations: async () => {
14535
+ return migrations.map((m) => m.compile().sql).join(";\n\n") + ";";
14536
+ },
14537
+ migrations,
14538
+ needsMigration: toBeCreated.length > 0 || toBeAdded.length > 0
14539
+ };
14540
+ }
14541
+ const authOptions = getAuthClientOptions({ database: {
14525
14542
  db,
14526
14543
  type: dbType === "neon" ? "postgres" : dbType
14527
- } }));
14544
+ } });
14545
+ const { toBeAdded: authChangesToBeAdded, toBeCreated: authChangesToBeCreated, runMigrations: runAuthMigrations } = await getAuthMigrations(authOptions);
14528
14546
  async function runMigrations() {
14529
14547
  for (const migration of migrations) await migration.execute();
14530
14548
  await runAuthMigrations();
@@ -14593,7 +14611,7 @@ async function runAutoMigrations(db, dbType, options) {
14593
14611
  } catch (error$47) {
14594
14612
  logger.debug(`Constraint fixup skipped or failed: ${error$47}`);
14595
14613
  }
14596
- const { toBeCreated, toBeAdded, runMigrations, needsMigration } = await getMigrations$1(db, dbType, options);
14614
+ const { toBeCreated, toBeAdded, runMigrations, needsMigration } = await getMigrations(db, dbType, options);
14597
14615
  if (!needsMigration) {
14598
14616
  logger.debug("Auto-migration: No migrations needed");
14599
14617
  return {
@@ -14714,4 +14732,4 @@ async function createDatabaseFromConnection(rawConnection, options) {
14714
14732
  }
14715
14733
 
14716
14734
  //#endregion
14717
- export { schemas as A, any as B, llmRequestsSchema as C, playgroundsSchema as D, playgroundRunsSchema as E, variantVersionsSchema as F, object as G, boolean$1 as H, variantsSchema as I, union as J, record as K, workspaceSettingsSchema as L, spansSchema as M, targetingRulesSchema as N, providerConfigsSchema as O, tracesSchema as P, zod_default as R, guardrailConfigsSchema as S, playgroundResultsSchema as T, literal as U, array as V, number$1 as W, unknown as Y, datasetVersionRecordsSchema as _, matchType as a, environmentSecretsSchema as b, logger as c, validatePartialTableData as d, validateTableData as f, datasetRecordsSchema as g, configsSchema as h, getMigrations$1 as i, spanEventsSchema as j, providerGuardrailOverridesSchema as k, parsePartialTableData as l, configVariantsSchema as m, createDatabaseFromConnection as n, runAutoMigrations as o, SCHEMA_METADATA as p, string$1 as q, detectDatabaseType as r, getAuthClientOptions as s, createDatabase as t, parseTableData as u, datasetVersionsSchema as v, playgroundColumnSchema as w, environmentsSchema as x, datasetsSchema as y, _enum as z };
14735
+ export { schemas as A, any as B, llmRequestsSchema as C, playgroundsSchema as D, playgroundRunsSchema as E, variantVersionsSchema as F, object as G, boolean$1 as H, variantsSchema as I, union as J, record as K, workspaceSettingsSchema as L, spansSchema as M, targetingRulesSchema as N, providerConfigsSchema as O, tracesSchema as P, zod_default as R, guardrailConfigsSchema as S, playgroundResultsSchema as T, literal as U, array as V, number$1 as W, unknown as Y, datasetVersionRecordsSchema as _, matchType as a, environmentSecretsSchema as b, logger as c, validatePartialTableData as d, validateTableData as f, datasetRecordsSchema as g, configsSchema as h, getMigrations as i, spanEventsSchema as j, providerGuardrailOverridesSchema as k, parsePartialTableData as l, configVariantsSchema as m, createDatabaseFromConnection as n, runAutoMigrations as o, SCHEMA_METADATA as p, string$1 as q, detectDatabaseType as r, getAuthClientOptions as s, createDatabase as t, parseTableData as u, datasetVersionsSchema as v, playgroundColumnSchema as w, environmentsSchema as x, datasetsSchema as y, _enum as z };
@@ -2588,22 +2588,6 @@ declare function validateTableData<T extends TableName>(table: T, data: unknown)
2588
2588
  id: string;
2589
2589
  createdAt: Date;
2590
2590
  updatedAt: Date;
2591
- }> | zod0.ZodSafeParseSuccess<{
2592
- providerConfigId: string;
2593
- guardrailConfigId: string;
2594
- enabled: boolean;
2595
- id: string;
2596
- createdAt: Date;
2597
- updatedAt: Date;
2598
- parameters?: Record<string, unknown> | null | undefined;
2599
- }> | zod0.ZodSafeParseError<{
2600
- providerConfigId: string;
2601
- guardrailConfigId: string;
2602
- enabled: boolean;
2603
- id: string;
2604
- createdAt: Date;
2605
- updatedAt: Date;
2606
- parameters?: Record<string, unknown> | null | undefined;
2607
2591
  }> | zod0.ZodSafeParseSuccess<{
2608
2592
  datasetId: string;
2609
2593
  versionNumber: number;
@@ -2654,6 +2638,22 @@ declare function validateTableData<T extends TableName>(table: T, data: unknown)
2654
2638
  id: string;
2655
2639
  createdAt: Date;
2656
2640
  updatedAt: Date;
2641
+ }> | zod0.ZodSafeParseSuccess<{
2642
+ providerConfigId: string;
2643
+ guardrailConfigId: string;
2644
+ enabled: boolean;
2645
+ id: string;
2646
+ createdAt: Date;
2647
+ updatedAt: Date;
2648
+ parameters?: Record<string, unknown> | null | undefined;
2649
+ }> | zod0.ZodSafeParseError<{
2650
+ providerConfigId: string;
2651
+ guardrailConfigId: string;
2652
+ enabled: boolean;
2653
+ id: string;
2654
+ createdAt: Date;
2655
+ updatedAt: Date;
2656
+ parameters?: Record<string, unknown> | null | undefined;
2657
2657
  }> | zod0.ZodSafeParseSuccess<{
2658
2658
  requestId: string;
2659
2659
  provider: string;
@@ -2960,22 +2960,6 @@ declare function validatePartialTableData<T extends TableName>(table: T, data: u
2960
2960
  id?: string | undefined;
2961
2961
  createdAt?: Date | undefined;
2962
2962
  updatedAt?: Date | undefined;
2963
- }> | zod0.ZodSafeParseSuccess<{
2964
- providerConfigId?: string | undefined;
2965
- guardrailConfigId?: string | undefined;
2966
- enabled?: boolean | undefined;
2967
- parameters?: Record<string, unknown> | null | undefined;
2968
- id?: string | undefined;
2969
- createdAt?: Date | undefined;
2970
- updatedAt?: Date | undefined;
2971
- }> | zod0.ZodSafeParseError<{
2972
- providerConfigId?: string | undefined;
2973
- guardrailConfigId?: string | undefined;
2974
- enabled?: boolean | undefined;
2975
- parameters?: Record<string, unknown> | null | undefined;
2976
- id?: string | undefined;
2977
- createdAt?: Date | undefined;
2978
- updatedAt?: Date | undefined;
2979
2963
  }> | zod0.ZodSafeParseSuccess<{
2980
2964
  datasetId?: string | undefined;
2981
2965
  versionNumber?: number | undefined;
@@ -3026,6 +3010,22 @@ declare function validatePartialTableData<T extends TableName>(table: T, data: u
3026
3010
  id?: string | undefined;
3027
3011
  createdAt?: Date | undefined;
3028
3012
  updatedAt?: Date | undefined;
3013
+ }> | zod0.ZodSafeParseSuccess<{
3014
+ providerConfigId?: string | undefined;
3015
+ guardrailConfigId?: string | undefined;
3016
+ enabled?: boolean | undefined;
3017
+ parameters?: Record<string, unknown> | null | undefined;
3018
+ id?: string | undefined;
3019
+ createdAt?: Date | undefined;
3020
+ updatedAt?: Date | undefined;
3021
+ }> | zod0.ZodSafeParseError<{
3022
+ providerConfigId?: string | undefined;
3023
+ guardrailConfigId?: string | undefined;
3024
+ enabled?: boolean | undefined;
3025
+ parameters?: Record<string, unknown> | null | undefined;
3026
+ id?: string | undefined;
3027
+ createdAt?: Date | undefined;
3028
+ updatedAt?: Date | undefined;
3029
3029
  }> | zod0.ZodSafeParseSuccess<{
3030
3030
  requestId?: string | undefined;
3031
3031
  configId?: string | null | undefined;
@@ -3255,14 +3255,6 @@ declare function parseTableData<T extends TableName>(table: T, data: unknown): {
3255
3255
  id: string;
3256
3256
  createdAt: Date;
3257
3257
  updatedAt: Date;
3258
- } | {
3259
- providerConfigId: string;
3260
- guardrailConfigId: string;
3261
- enabled: boolean;
3262
- id: string;
3263
- createdAt: Date;
3264
- updatedAt: Date;
3265
- parameters?: Record<string, unknown> | null | undefined;
3266
3258
  } | {
3267
3259
  datasetId: string;
3268
3260
  versionNumber: number;
@@ -3288,6 +3280,14 @@ declare function parseTableData<T extends TableName>(table: T, data: unknown): {
3288
3280
  id: string;
3289
3281
  createdAt: Date;
3290
3282
  updatedAt: Date;
3283
+ } | {
3284
+ providerConfigId: string;
3285
+ guardrailConfigId: string;
3286
+ enabled: boolean;
3287
+ id: string;
3288
+ createdAt: Date;
3289
+ updatedAt: Date;
3290
+ parameters?: Record<string, unknown> | null | undefined;
3291
3291
  } | {
3292
3292
  requestId: string;
3293
3293
  provider: string;
@@ -3443,14 +3443,6 @@ declare function parsePartialTableData<T extends TableName>(table: T, data: unkn
3443
3443
  id?: string | undefined;
3444
3444
  createdAt?: Date | undefined;
3445
3445
  updatedAt?: Date | undefined;
3446
- } | {
3447
- providerConfigId?: string | undefined;
3448
- guardrailConfigId?: string | undefined;
3449
- enabled?: boolean | undefined;
3450
- parameters?: Record<string, unknown> | null | undefined;
3451
- id?: string | undefined;
3452
- createdAt?: Date | undefined;
3453
- updatedAt?: Date | undefined;
3454
3446
  } | {
3455
3447
  datasetId?: string | undefined;
3456
3448
  versionNumber?: number | undefined;
@@ -3476,6 +3468,14 @@ declare function parsePartialTableData<T extends TableName>(table: T, data: unkn
3476
3468
  id?: string | undefined;
3477
3469
  createdAt?: Date | undefined;
3478
3470
  updatedAt?: Date | undefined;
3471
+ } | {
3472
+ providerConfigId?: string | undefined;
3473
+ guardrailConfigId?: string | undefined;
3474
+ enabled?: boolean | undefined;
3475
+ parameters?: Record<string, unknown> | null | undefined;
3476
+ id?: string | undefined;
3477
+ createdAt?: Date | undefined;
3478
+ updatedAt?: Date | undefined;
3479
3479
  } | {
3480
3480
  requestId?: string | undefined;
3481
3481
  configId?: string | null | undefined;
@@ -2588,22 +2588,6 @@ declare function validateTableData<T extends TableName>(table: T, data: unknown)
2588
2588
  id: string;
2589
2589
  createdAt: Date;
2590
2590
  updatedAt: Date;
2591
- }> | zod0.ZodSafeParseSuccess<{
2592
- providerConfigId: string;
2593
- guardrailConfigId: string;
2594
- enabled: boolean;
2595
- id: string;
2596
- createdAt: Date;
2597
- updatedAt: Date;
2598
- parameters?: Record<string, unknown> | null | undefined;
2599
- }> | zod0.ZodSafeParseError<{
2600
- providerConfigId: string;
2601
- guardrailConfigId: string;
2602
- enabled: boolean;
2603
- id: string;
2604
- createdAt: Date;
2605
- updatedAt: Date;
2606
- parameters?: Record<string, unknown> | null | undefined;
2607
2591
  }> | zod0.ZodSafeParseSuccess<{
2608
2592
  datasetId: string;
2609
2593
  versionNumber: number;
@@ -2654,6 +2638,22 @@ declare function validateTableData<T extends TableName>(table: T, data: unknown)
2654
2638
  id: string;
2655
2639
  createdAt: Date;
2656
2640
  updatedAt: Date;
2641
+ }> | zod0.ZodSafeParseSuccess<{
2642
+ providerConfigId: string;
2643
+ guardrailConfigId: string;
2644
+ enabled: boolean;
2645
+ id: string;
2646
+ createdAt: Date;
2647
+ updatedAt: Date;
2648
+ parameters?: Record<string, unknown> | null | undefined;
2649
+ }> | zod0.ZodSafeParseError<{
2650
+ providerConfigId: string;
2651
+ guardrailConfigId: string;
2652
+ enabled: boolean;
2653
+ id: string;
2654
+ createdAt: Date;
2655
+ updatedAt: Date;
2656
+ parameters?: Record<string, unknown> | null | undefined;
2657
2657
  }> | zod0.ZodSafeParseSuccess<{
2658
2658
  requestId: string;
2659
2659
  provider: string;
@@ -2960,22 +2960,6 @@ declare function validatePartialTableData<T extends TableName>(table: T, data: u
2960
2960
  id?: string | undefined;
2961
2961
  createdAt?: Date | undefined;
2962
2962
  updatedAt?: Date | undefined;
2963
- }> | zod0.ZodSafeParseSuccess<{
2964
- providerConfigId?: string | undefined;
2965
- guardrailConfigId?: string | undefined;
2966
- enabled?: boolean | undefined;
2967
- parameters?: Record<string, unknown> | null | undefined;
2968
- id?: string | undefined;
2969
- createdAt?: Date | undefined;
2970
- updatedAt?: Date | undefined;
2971
- }> | zod0.ZodSafeParseError<{
2972
- providerConfigId?: string | undefined;
2973
- guardrailConfigId?: string | undefined;
2974
- enabled?: boolean | undefined;
2975
- parameters?: Record<string, unknown> | null | undefined;
2976
- id?: string | undefined;
2977
- createdAt?: Date | undefined;
2978
- updatedAt?: Date | undefined;
2979
2963
  }> | zod0.ZodSafeParseSuccess<{
2980
2964
  datasetId?: string | undefined;
2981
2965
  versionNumber?: number | undefined;
@@ -3026,6 +3010,22 @@ declare function validatePartialTableData<T extends TableName>(table: T, data: u
3026
3010
  id?: string | undefined;
3027
3011
  createdAt?: Date | undefined;
3028
3012
  updatedAt?: Date | undefined;
3013
+ }> | zod0.ZodSafeParseSuccess<{
3014
+ providerConfigId?: string | undefined;
3015
+ guardrailConfigId?: string | undefined;
3016
+ enabled?: boolean | undefined;
3017
+ parameters?: Record<string, unknown> | null | undefined;
3018
+ id?: string | undefined;
3019
+ createdAt?: Date | undefined;
3020
+ updatedAt?: Date | undefined;
3021
+ }> | zod0.ZodSafeParseError<{
3022
+ providerConfigId?: string | undefined;
3023
+ guardrailConfigId?: string | undefined;
3024
+ enabled?: boolean | undefined;
3025
+ parameters?: Record<string, unknown> | null | undefined;
3026
+ id?: string | undefined;
3027
+ createdAt?: Date | undefined;
3028
+ updatedAt?: Date | undefined;
3029
3029
  }> | zod0.ZodSafeParseSuccess<{
3030
3030
  requestId?: string | undefined;
3031
3031
  configId?: string | null | undefined;
@@ -3255,14 +3255,6 @@ declare function parseTableData<T extends TableName>(table: T, data: unknown): {
3255
3255
  id: string;
3256
3256
  createdAt: Date;
3257
3257
  updatedAt: Date;
3258
- } | {
3259
- providerConfigId: string;
3260
- guardrailConfigId: string;
3261
- enabled: boolean;
3262
- id: string;
3263
- createdAt: Date;
3264
- updatedAt: Date;
3265
- parameters?: Record<string, unknown> | null | undefined;
3266
3258
  } | {
3267
3259
  datasetId: string;
3268
3260
  versionNumber: number;
@@ -3288,6 +3280,14 @@ declare function parseTableData<T extends TableName>(table: T, data: unknown): {
3288
3280
  id: string;
3289
3281
  createdAt: Date;
3290
3282
  updatedAt: Date;
3283
+ } | {
3284
+ providerConfigId: string;
3285
+ guardrailConfigId: string;
3286
+ enabled: boolean;
3287
+ id: string;
3288
+ createdAt: Date;
3289
+ updatedAt: Date;
3290
+ parameters?: Record<string, unknown> | null | undefined;
3291
3291
  } | {
3292
3292
  requestId: string;
3293
3293
  provider: string;
@@ -3443,14 +3443,6 @@ declare function parsePartialTableData<T extends TableName>(table: T, data: unkn
3443
3443
  id?: string | undefined;
3444
3444
  createdAt?: Date | undefined;
3445
3445
  updatedAt?: Date | undefined;
3446
- } | {
3447
- providerConfigId?: string | undefined;
3448
- guardrailConfigId?: string | undefined;
3449
- enabled?: boolean | undefined;
3450
- parameters?: Record<string, unknown> | null | undefined;
3451
- id?: string | undefined;
3452
- createdAt?: Date | undefined;
3453
- updatedAt?: Date | undefined;
3454
3446
  } | {
3455
3447
  datasetId?: string | undefined;
3456
3448
  versionNumber?: number | undefined;
@@ -3476,6 +3468,14 @@ declare function parsePartialTableData<T extends TableName>(table: T, data: unkn
3476
3468
  id?: string | undefined;
3477
3469
  createdAt?: Date | undefined;
3478
3470
  updatedAt?: Date | undefined;
3471
+ } | {
3472
+ providerConfigId?: string | undefined;
3473
+ guardrailConfigId?: string | undefined;
3474
+ enabled?: boolean | undefined;
3475
+ parameters?: Record<string, unknown> | null | undefined;
3476
+ id?: string | undefined;
3477
+ createdAt?: Date | undefined;
3478
+ updatedAt?: Date | undefined;
3479
3479
  } | {
3480
3480
  requestId?: string | undefined;
3481
3481
  configId?: string | null | undefined;
package/dist/index.cjs CHANGED
@@ -1,10 +1,8 @@
1
- const require_db = require('./db-i0OOYxJm.cjs');
2
- const require_neon_dialect = require('./neon-dialect-oh8u9vRy.cjs');
1
+ const require_db = require('./db-DsEVRoua.cjs');
2
+ const require_neon_dialect = require('./neon-dialect-Dn80Acj9.cjs');
3
3
  let __llmops_gateway = require("@llmops/gateway");
4
4
  __llmops_gateway = require_db.__toESM(__llmops_gateway);
5
5
  let kysely = require("kysely");
6
- let node_fs_promises = require("node:fs/promises");
7
- node_fs_promises = require_db.__toESM(node_fs_promises);
8
6
  let node_path = require("node:path");
9
7
  node_path = require_db.__toESM(node_path);
10
8
  let __better_auth_utils_random = require("@better-auth/utils/random");
@@ -643,7 +641,7 @@ const llmopsConfigBaseSchema = require_db.object({
643
641
  const llmopsConfigSchema = llmopsConfigBaseSchema.transform((config) => ({
644
642
  ...config,
645
643
  providers: mergeWithDefaultProviders(config.providers)
646
- })).refine((config) => config.database !== void 0 || config.providers && config.providers.length > 0, "Either database or providers must be configured. Set a database connection, configure providers explicitly, or set API key environment variables (e.g., OPENAI_API_KEY).");
644
+ }));
647
645
  function validateLLMOpsConfig(config) {
648
646
  const result = llmopsConfigSchema.safeParse(config ?? {});
649
647
  if (!result.success) {
@@ -864,14 +862,16 @@ var FileCacheBackend = class {
864
862
  if (!this.loaded) await this.loadPromise;
865
863
  }
866
864
  async ensureDataDir() {
865
+ const { mkdir } = await import("node:fs/promises");
867
866
  const dir = node_path.dirname(this.cacheFile);
868
867
  try {
869
- await node_fs_promises.mkdir(dir, { recursive: true });
868
+ await mkdir(dir, { recursive: true });
870
869
  } catch {}
871
870
  }
872
871
  async loadCache() {
873
872
  try {
874
- const content = await node_fs_promises.readFile(this.cacheFile, "utf-8");
873
+ const { readFile } = await import("node:fs/promises");
874
+ const content = await readFile(this.cacheFile, "utf-8");
875
875
  this.data = JSON.parse(content);
876
876
  this.updateStats();
877
877
  this.loaded = true;
@@ -883,7 +883,8 @@ var FileCacheBackend = class {
883
883
  async saveCache() {
884
884
  try {
885
885
  await this.ensureDataDir();
886
- await node_fs_promises.writeFile(this.cacheFile, JSON.stringify(this.data, null, 2));
886
+ const { writeFile } = await import("node:fs/promises");
887
+ await writeFile(this.cacheFile, JSON.stringify(this.data, null, 2));
887
888
  } catch {}
888
889
  }
889
890
  scheduleSave() {
@@ -2070,7 +2071,8 @@ const costSummarySchema = require_db.zod_default.object({
2070
2071
  variantId: require_db.zod_default.string().uuid().optional(),
2071
2072
  environmentId: require_db.zod_default.string().uuid().optional(),
2072
2073
  tags: require_db.zod_default.record(require_db.zod_default.string(), require_db.zod_default.array(require_db.zod_default.string())).optional(),
2073
- groupBy: require_db.zod_default.enum(COST_SUMMARY_GROUP_BY).optional()
2074
+ groupBy: require_db.zod_default.enum(COST_SUMMARY_GROUP_BY).optional(),
2075
+ tagKeys: require_db.zod_default.array(require_db.zod_default.string()).optional()
2074
2076
  });
2075
2077
  /**
2076
2078
  * Helper to create column reference for SQL
@@ -2286,7 +2288,7 @@ const createLLMRequestsDataLayer = (db) => {
2286
2288
  getCostSummary: async (params) => {
2287
2289
  const result = await costSummarySchema.safeParseAsync(params);
2288
2290
  if (!result.success) throw new LLMOpsError(`Invalid parameters: ${result.error.message}`);
2289
- const { startDate, endDate, groupBy, configId, variantId, environmentId, tags } = result.data;
2291
+ const { startDate, endDate, groupBy, configId, variantId, environmentId, tags, tagKeys } = result.data;
2290
2292
  let baseQuery = db.selectFrom("llm_requests").where(kysely.sql`${col$1("createdAt")} >= ${startDate.toISOString()}`).where(kysely.sql`${col$1("createdAt")} <= ${endDate.toISOString()}`);
2291
2293
  if (configId) baseQuery = baseQuery.where("configId", "=", configId);
2292
2294
  if (variantId) baseQuery = baseQuery.where("variantId", "=", variantId);
@@ -2345,6 +2347,10 @@ const createLLMRequestsDataLayer = (db) => {
2345
2347
  conditions.push(kysely.sql`${col$1("tags")}->>${key} IN (${valueList})`);
2346
2348
  }
2347
2349
  }
2350
+ if (tagKeys && tagKeys.length > 0) {
2351
+ const tagKeyList = kysely.sql.join(tagKeys.map((k) => kysely.sql`${k}`), kysely.sql`, `);
2352
+ conditions.push(kysely.sql`t.key IN (${tagKeyList})`);
2353
+ }
2348
2354
  const whereClause = kysely.sql.join(conditions, kysely.sql` AND `);
2349
2355
  return (await kysely.sql`
2350
2356
  SELECT t.key || ':' || t.value as "groupKey",