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

package/dist/index.d.cts

@@ -1,5 +1,3 @@
-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-DjIHdwhi.cjs";
-import { Kysely } from "kysely";
 import z$1, { z } from "zod";
 import gateway from "@llmops/gateway";
 import pino from "pino";
@@ -716,10 +714,10 @@ declare const chatCompletionCreateParamsBaseSchema: z.ZodObject<{
   seed: z.ZodNullable<z.ZodOptional<z.ZodNumber>>;
   service_tier: z.ZodNullable<z.ZodOptional<z.ZodEnum<{
     default: "default";
-    priority: "priority";
     auto: "auto";
     flex: "flex";
     scale: "scale";
+    priority: "priority";
   }>>>;
   stop: z.ZodNullable<z.ZodOptional<z.ZodUnion<readonly [z.ZodString, z.ZodArray<z.ZodString>]>>>;
   store: z.ZodNullable<z.ZodOptional<z.ZodBoolean>>;
@@ -905,10 +903,10 @@ declare const variantJsonDataSchema: z.ZodObject<{
   seed: z.ZodNullable<z.ZodOptional<z.ZodNumber>>;
   service_tier: z.ZodNullable<z.ZodOptional<z.ZodEnum<{
     default: "default";
-    priority: "priority";
     auto: "auto";
     flex: "flex";
     scale: "scale";
+    priority: "priority";
   }>>>;
   stop: z.ZodNullable<z.ZodOptional<z.ZodUnion<readonly [z.ZodString, z.ZodArray<z.ZodString>]>>>;
   store: z.ZodNullable<z.ZodOptional<z.ZodBoolean>>;
@@ -1187,725 +1185,6 @@ declare const logger: pino.Logger<never, boolean>;
 //#region src/utils/id.d.ts
 declare const generateId: (size?: number) => string;
 //#endregion
-//#region src/datalayer/datasets.d.ts
-declare const createDataset: z$1.ZodObject<{
-  name: z$1.ZodString;
-  description: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
-}, z$1.core.$strip>;
-declare const updateDataset: z$1.ZodObject<{
-  datasetId: z$1.ZodString;
-  name: z$1.ZodOptional<z$1.ZodString>;
-  description: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
-}, z$1.core.$strip>;
-declare const getDatasetById: z$1.ZodObject<{
-  datasetId: z$1.ZodString;
-}, z$1.core.$strip>;
-declare const deleteDataset: z$1.ZodObject<{
-  datasetId: z$1.ZodString;
-}, z$1.core.$strip>;
-declare const listDatasets: z$1.ZodObject<{
-  limit: z$1.ZodOptional<z$1.ZodNumber>;
-  offset: z$1.ZodOptional<z$1.ZodNumber>;
-}, z$1.core.$strip>;
-declare const createRecord: z$1.ZodObject<{
-  datasetId: z$1.ZodString;
-  input: z$1.ZodRecord<z$1.ZodString, z$1.ZodUnknown>;
-  expected: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodRecord<z$1.ZodString, z$1.ZodUnknown>>>;
-  metadata: z$1.ZodOptional<z$1.ZodRecord<z$1.ZodString, z$1.ZodUnknown>>;
-}, z$1.core.$strip>;
-declare const updateRecord: z$1.ZodObject<{
-  recordId: z$1.ZodString;
-  input: z$1.ZodOptional<z$1.ZodRecord<z$1.ZodString, z$1.ZodUnknown>>;
-  expected: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodRecord<z$1.ZodString, z$1.ZodUnknown>>>;
-  metadata: z$1.ZodOptional<z$1.ZodRecord<z$1.ZodString, z$1.ZodUnknown>>;
-}, z$1.core.$strip>;
-declare const deleteRecord: z$1.ZodObject<{
-  recordId: z$1.ZodString;
-}, z$1.core.$strip>;
-declare const listRecords: z$1.ZodObject<{
-  datasetId: z$1.ZodString;
-  limit: z$1.ZodOptional<z$1.ZodNumber>;
-  offset: z$1.ZodOptional<z$1.ZodNumber>;
-}, z$1.core.$strip>;
-declare const createVersion: z$1.ZodObject<{
-  datasetId: z$1.ZodString;
-  name: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
-  description: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
-}, z$1.core.$strip>;
-declare const getVersionById: z$1.ZodObject<{
-  versionId: z$1.ZodString;
-}, z$1.core.$strip>;
-declare const listVersions: z$1.ZodObject<{
-  datasetId: z$1.ZodString;
-  limit: z$1.ZodOptional<z$1.ZodNumber>;
-  offset: z$1.ZodOptional<z$1.ZodNumber>;
-}, z$1.core.$strip>;
-declare const getVersionRecords: z$1.ZodObject<{
-  versionId: z$1.ZodString;
-  limit: z$1.ZodOptional<z$1.ZodNumber>;
-  offset: z$1.ZodOptional<z$1.ZodNumber>;
-}, z$1.core.$strip>;
-declare const createDatasetsDataLayer: (db: Kysely<Database>) => {
-  createDataset: (params: z$1.infer<typeof createDataset>) => Promise<{
-    name: string;
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    description: string | null;
-    recordCount: number;
-    latestVersionNumber: number;
-  } | undefined>;
-  updateDataset: (params: z$1.infer<typeof updateDataset>) => Promise<{
-    name: string;
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    description: string | null;
-    recordCount: number;
-    latestVersionNumber: number;
-  } | undefined>;
-  getDatasetById: (params: z$1.infer<typeof getDatasetById>) => Promise<{
-    name: string;
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    description: string | null;
-    recordCount: number;
-    latestVersionNumber: number;
-  } | undefined>;
-  deleteDataset: (params: z$1.infer<typeof deleteDataset>) => Promise<{
-    name: string;
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    description: string | null;
-    recordCount: number;
-    latestVersionNumber: number;
-  } | undefined>;
-  listDatasets: (params?: z$1.infer<typeof listDatasets>) => Promise<{
-    name: string;
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    description: string | null;
-    recordCount: number;
-    latestVersionNumber: number;
-  }[]>;
-  countDatasets: () => Promise<number>;
-  createRecord: (params: z$1.infer<typeof createRecord>) => Promise<{
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    input: Record<string, unknown>;
-    datasetId: string;
-    expected: Record<string, unknown> | null;
-    metadata: Record<string, unknown>;
-  } | undefined>;
-  updateRecord: (params: z$1.infer<typeof updateRecord>) => Promise<{
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    input: Record<string, unknown>;
-    datasetId: string;
-    expected: Record<string, unknown> | null;
-    metadata: Record<string, unknown>;
-  } | undefined>;
-  deleteRecord: (params: z$1.infer<typeof deleteRecord>) => Promise<{
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    input: Record<string, unknown>;
-    datasetId: string;
-    expected: Record<string, unknown> | null;
-    metadata: Record<string, unknown>;
-  } | undefined>;
-  listRecords: (params: z$1.infer<typeof listRecords>) => Promise<{
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    input: Record<string, unknown>;
-    datasetId: string;
-    expected: Record<string, unknown> | null;
-    metadata: Record<string, unknown>;
-  }[]>;
-  getRecordById: (recordId: string) => Promise<{
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    input: Record<string, unknown>;
-    datasetId: string;
-    expected: Record<string, unknown> | null;
-    metadata: Record<string, unknown>;
-  } | undefined>;
-  createVersion: (params: z$1.infer<typeof createVersion>) => Promise<{
-    name: string | null;
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    datasetId: string;
-    description: string | null;
-    recordCount: number;
-    versionNumber: number;
-    snapshotHash: string;
-  } | undefined>;
-  getVersionById: (params: z$1.infer<typeof getVersionById>) => Promise<{
-    name: string | null;
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    datasetId: string;
-    description: string | null;
-    recordCount: number;
-    versionNumber: number;
-    snapshotHash: string;
-  } | undefined>;
-  listVersions: (params: z$1.infer<typeof listVersions>) => Promise<{
-    name: string | null;
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    datasetId: string;
-    description: string | null;
-    recordCount: number;
-    versionNumber: number;
-    snapshotHash: string;
-  }[]>;
-  getVersionRecords: (params: z$1.infer<typeof getVersionRecords>) => Promise<{
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    input: Record<string, unknown>;
-    datasetId: string;
-    expected: Record<string, unknown> | null;
-    metadata: Record<string, unknown>;
-    position: number;
-  }[]>;
-};
-//#endregion
-//#region src/datalayer/playgrounds.d.ts
-declare const createNewPlayground: z$1.ZodObject<{
-  name: z$1.ZodString;
-  datasetId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
-  columns: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodArray<z$1.ZodObject<{
-    id: z$1.ZodString;
-    name: z$1.ZodString;
-    position: z$1.ZodNumber;
-    providerConfigId: z$1.ZodOptional<z$1.ZodUnion<readonly [z$1.ZodString, z$1.ZodNull]>>;
-    providerSlug: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
-    modelName: z$1.ZodString;
-    messages: z$1.ZodArray<z$1.ZodObject<{
-      role: z$1.ZodEnum<{
-        system: "system";
-        user: "user";
-        assistant: "assistant";
-      }>;
-      content: z$1.ZodString;
-    }, z$1.core.$strip>>;
-    temperature: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodNumber>>;
-    maxTokens: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodNumber>>;
-    topP: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodNumber>>;
-    frequencyPenalty: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodNumber>>;
-    presencePenalty: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodNumber>>;
-  }, z$1.core.$strip>>>>;
-}, z$1.core.$strip>;
-declare const updatePlayground: z$1.ZodObject<{
-  playgroundId: z$1.ZodUUID;
-  name: z$1.ZodOptional<z$1.ZodString>;
-  datasetId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
-  columns: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodArray<z$1.ZodObject<{
-    id: z$1.ZodString;
-    name: z$1.ZodString;
-    position: z$1.ZodNumber;
-    providerConfigId: z$1.ZodOptional<z$1.ZodUnion<readonly [z$1.ZodString, z$1.ZodNull]>>;
-    providerSlug: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
-    modelName: z$1.ZodString;
-    messages: z$1.ZodArray<z$1.ZodObject<{
-      role: z$1.ZodEnum<{
-        system: "system";
-        user: "user";
-        assistant: "assistant";
-      }>;
-      content: z$1.ZodString;
-    }, z$1.core.$strip>>;
-    temperature: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodNumber>>;
-    maxTokens: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodNumber>>;
-    topP: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodNumber>>;
-    frequencyPenalty: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodNumber>>;
-    presencePenalty: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodNumber>>;
-  }, z$1.core.$strip>>>>;
-}, z$1.core.$strip>;
-declare const getPlaygroundById: z$1.ZodObject<{
-  playgroundId: z$1.ZodUUID;
-}, z$1.core.$strip>;
-declare const deletePlayground: z$1.ZodObject<{
-  playgroundId: z$1.ZodUUID;
-}, z$1.core.$strip>;
-declare const listPlaygrounds: z$1.ZodObject<{
-  limit: z$1.ZodOptional<z$1.ZodNumber>;
-  offset: z$1.ZodOptional<z$1.ZodNumber>;
-}, z$1.core.$strip>;
-declare const createPlaygroundDataLayer: (db: Kysely<Database>) => {
-  createNewPlayground: (params: z$1.infer<typeof createNewPlayground>) => Promise<{
-    name: string;
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    datasetId: string | null;
-    columns: {
-      id: string;
-      name: string;
-      position: number;
-      modelName: string;
-      messages: {
-        role: "system" | "user" | "assistant";
-        content: string;
-      }[];
-      providerConfigId?: string | null | undefined;
-      providerSlug?: string | null | undefined;
-      temperature?: number | null | undefined;
-      maxTokens?: number | null | undefined;
-      topP?: number | null | undefined;
-      frequencyPenalty?: number | null | undefined;
-      presencePenalty?: number | null | undefined;
-    }[] | null;
-  } | undefined>;
-  updatePlayground: (params: z$1.infer<typeof updatePlayground>) => Promise<{
-    name: string;
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    datasetId: string | null;
-    columns: {
-      id: string;
-      name: string;
-      position: number;
-      modelName: string;
-      messages: {
-        role: "system" | "user" | "assistant";
-        content: string;
-      }[];
-      providerConfigId?: string | null | undefined;
-      providerSlug?: string | null | undefined;
-      temperature?: number | null | undefined;
-      maxTokens?: number | null | undefined;
-      topP?: number | null | undefined;
-      frequencyPenalty?: number | null | undefined;
-      presencePenalty?: number | null | undefined;
-    }[] | null;
-  } | undefined>;
-  getPlaygroundById: (params: z$1.infer<typeof getPlaygroundById>) => Promise<{
-    name: string;
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    datasetId: string | null;
-    columns: {
-      id: string;
-      name: string;
-      position: number;
-      modelName: string;
-      messages: {
-        role: "system" | "user" | "assistant";
-        content: string;
-      }[];
-      providerConfigId?: string | null | undefined;
-      providerSlug?: string | null | undefined;
-      temperature?: number | null | undefined;
-      maxTokens?: number | null | undefined;
-      topP?: number | null | undefined;
-      frequencyPenalty?: number | null | undefined;
-      presencePenalty?: number | null | undefined;
-    }[] | null;
-  } | undefined>;
-  deletePlayground: (params: z$1.infer<typeof deletePlayground>) => Promise<{
-    name: string;
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    datasetId: string | null;
-    columns: {
-      id: string;
-      name: string;
-      position: number;
-      modelName: string;
-      messages: {
-        role: "system" | "user" | "assistant";
-        content: string;
-      }[];
-      providerConfigId?: string | null | undefined;
-      providerSlug?: string | null | undefined;
-      temperature?: number | null | undefined;
-      maxTokens?: number | null | undefined;
-      topP?: number | null | undefined;
-      frequencyPenalty?: number | null | undefined;
-      presencePenalty?: number | null | undefined;
-    }[] | null;
-  } | undefined>;
-  listPlaygrounds: (params?: z$1.infer<typeof listPlaygrounds>) => Promise<{
-    name: string;
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    datasetId: string | null;
-    columns: {
-      id: string;
-      name: string;
-      position: number;
-      modelName: string;
-      messages: {
-        role: "system" | "user" | "assistant";
-        content: string;
-      }[];
-      providerConfigId?: string | null | undefined;
-      providerSlug?: string | null | undefined;
-      temperature?: number | null | undefined;
-      maxTokens?: number | null | undefined;
-      topP?: number | null | undefined;
-      frequencyPenalty?: number | null | undefined;
-      presencePenalty?: number | null | undefined;
-    }[] | null;
-  }[]>;
-  countPlaygrounds: () => Promise<number>;
-};
-//#endregion
-//#region src/datalayer/playgroundResults.d.ts
-declare const createPlaygroundResult: z$1.ZodObject<{
-  runId: z$1.ZodString;
-  columnId: z$1.ZodString;
-  datasetRecordId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
-  inputVariables: z$1.ZodDefault<z$1.ZodRecord<z$1.ZodString, z$1.ZodUnknown>>;
-  status: z$1.ZodDefault<z$1.ZodEnum<{
-    pending: "pending";
-    running: "running";
-    completed: "completed";
-    failed: "failed";
-  }>>;
-}, z$1.core.$strip>;
-declare const createPlaygroundResultsBatch: z$1.ZodObject<{
-  results: z$1.ZodArray<z$1.ZodObject<{
-    runId: z$1.ZodString;
-    columnId: z$1.ZodString;
-    datasetRecordId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
-    inputVariables: z$1.ZodDefault<z$1.ZodRecord<z$1.ZodString, z$1.ZodUnknown>>;
-    status: z$1.ZodDefault<z$1.ZodEnum<{
-      pending: "pending";
-      running: "running";
-      completed: "completed";
-      failed: "failed";
-    }>>;
-  }, z$1.core.$strip>>;
-}, z$1.core.$strip>;
-declare const updatePlaygroundResult: z$1.ZodObject<{
-  resultId: z$1.ZodString;
-  outputContent: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
-  status: z$1.ZodOptional<z$1.ZodEnum<{
-    pending: "pending";
-    running: "running";
-    completed: "completed";
-    failed: "failed";
-  }>>;
-  error: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
-  latencyMs: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodNumber>>;
-  promptTokens: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodNumber>>;
-  completionTokens: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodNumber>>;
-  totalTokens: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodNumber>>;
-  cost: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodNumber>>;
-}, z$1.core.$strip>;
-declare const getPlaygroundResultById: z$1.ZodObject<{
-  resultId: z$1.ZodString;
-}, z$1.core.$strip>;
-declare const listPlaygroundResults: z$1.ZodObject<{
-  runId: z$1.ZodString;
-  columnId: z$1.ZodOptional<z$1.ZodString>;
-  limit: z$1.ZodOptional<z$1.ZodNumber>;
-  offset: z$1.ZodOptional<z$1.ZodNumber>;
-}, z$1.core.$strip>;
-declare const deletePlaygroundResultsByRunId: z$1.ZodObject<{
-  runId: z$1.ZodString;
-}, z$1.core.$strip>;
-declare const createPlaygroundResultsDataLayer: (db: Kysely<Database>) => {
-  createPlaygroundResult: (params: z$1.infer<typeof createPlaygroundResult>) => Promise<{
-    error: string | null;
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    status: string;
-    runId: string;
-    columnId: string;
-    datasetRecordId: string | null;
-    inputVariables: Record<string, unknown>;
-    outputContent: string | null;
-    latencyMs: number | null;
-    promptTokens: number | null;
-    completionTokens: number | null;
-    totalTokens: number | null;
-    cost: number | null;
-  } | undefined>;
-  createPlaygroundResultsBatch: (params: z$1.infer<typeof createPlaygroundResultsBatch>) => Promise<{
-    error: string | null;
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    status: string;
-    runId: string;
-    columnId: string;
-    datasetRecordId: string | null;
-    inputVariables: Record<string, unknown>;
-    outputContent: string | null;
-    latencyMs: number | null;
-    promptTokens: number | null;
-    completionTokens: number | null;
-    totalTokens: number | null;
-    cost: number | null;
-  }[]>;
-  updatePlaygroundResult: (params: z$1.infer<typeof updatePlaygroundResult>) => Promise<{
-    error: string | null;
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    status: string;
-    runId: string;
-    columnId: string;
-    datasetRecordId: string | null;
-    inputVariables: Record<string, unknown>;
-    outputContent: string | null;
-    latencyMs: number | null;
-    promptTokens: number | null;
-    completionTokens: number | null;
-    totalTokens: number | null;
-    cost: number | null;
-  } | undefined>;
-  getPlaygroundResultById: (params: z$1.infer<typeof getPlaygroundResultById>) => Promise<{
-    error: string | null;
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    status: string;
-    runId: string;
-    columnId: string;
-    datasetRecordId: string | null;
-    inputVariables: Record<string, unknown>;
-    outputContent: string | null;
-    latencyMs: number | null;
-    promptTokens: number | null;
-    completionTokens: number | null;
-    totalTokens: number | null;
-    cost: number | null;
-  } | undefined>;
-  listPlaygroundResults: (params: z$1.infer<typeof listPlaygroundResults>) => Promise<{
-    error: string | null;
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    status: string;
-    runId: string;
-    columnId: string;
-    datasetRecordId: string | null;
-    inputVariables: Record<string, unknown>;
-    outputContent: string | null;
-    latencyMs: number | null;
-    promptTokens: number | null;
-    completionTokens: number | null;
-    totalTokens: number | null;
-    cost: number | null;
-  }[]>;
-  deletePlaygroundResultsByRunId: (params: z$1.infer<typeof deletePlaygroundResultsByRunId>) => Promise<{
-    error: string | null;
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    status: string;
-    runId: string;
-    columnId: string;
-    datasetRecordId: string | null;
-    inputVariables: Record<string, unknown>;
-    outputContent: string | null;
-    latencyMs: number | null;
-    promptTokens: number | null;
-    completionTokens: number | null;
-    totalTokens: number | null;
-    cost: number | null;
-  }[]>;
-  countPlaygroundResults: (runId: string) => Promise<number>;
-  countCompletedPlaygroundResults: (runId: string) => Promise<number>;
-};
-//#endregion
-//#region src/datalayer/playgroundRuns.d.ts
-declare const createPlaygroundRun: z$1.ZodObject<{
-  playgroundId: z$1.ZodString;
-  datasetId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
-  datasetVersionId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
-  status: z$1.ZodDefault<z$1.ZodEnum<{
-    pending: "pending";
-    running: "running";
-    completed: "completed";
-    failed: "failed";
-    cancelled: "cancelled";
-  }>>;
-  totalRecords: z$1.ZodDefault<z$1.ZodNumber>;
-}, z$1.core.$strip>;
-declare const updatePlaygroundRun: z$1.ZodObject<{
-  runId: z$1.ZodString;
-  status: z$1.ZodOptional<z$1.ZodEnum<{
-    pending: "pending";
-    running: "running";
-    completed: "completed";
-    failed: "failed";
-    cancelled: "cancelled";
-  }>>;
-  startedAt: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodDate>>;
-  completedAt: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodDate>>;
-  completedRecords: z$1.ZodOptional<z$1.ZodNumber>;
-}, z$1.core.$strip>;
-declare const getPlaygroundRunById: z$1.ZodObject<{
-  runId: z$1.ZodString;
-}, z$1.core.$strip>;
-declare const listPlaygroundRuns: z$1.ZodObject<{
-  playgroundId: z$1.ZodString;
-  limit: z$1.ZodOptional<z$1.ZodNumber>;
-  offset: z$1.ZodOptional<z$1.ZodNumber>;
-}, z$1.core.$strip>;
-declare const deletePlaygroundRun: z$1.ZodObject<{
-  runId: z$1.ZodString;
-}, z$1.core.$strip>;
-declare const createPlaygroundRunsDataLayer: (db: Kysely<Database>) => {
-  createPlaygroundRun: (params: z$1.infer<typeof createPlaygroundRun>) => Promise<{
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    datasetId: string | null;
-    playgroundId: string;
-    datasetVersionId: string | null;
-    status: string;
-    startedAt: Date | null;
-    completedAt: Date | null;
-    totalRecords: number;
-    completedRecords: number;
-  } | undefined>;
-  updatePlaygroundRun: (params: z$1.infer<typeof updatePlaygroundRun>) => Promise<{
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    datasetId: string | null;
-    playgroundId: string;
-    datasetVersionId: string | null;
-    status: string;
-    startedAt: Date | null;
-    completedAt: Date | null;
-    totalRecords: number;
-    completedRecords: number;
-  } | undefined>;
-  getPlaygroundRunById: (params: z$1.infer<typeof getPlaygroundRunById>) => Promise<{
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    datasetId: string | null;
-    playgroundId: string;
-    datasetVersionId: string | null;
-    status: string;
-    startedAt: Date | null;
-    completedAt: Date | null;
-    totalRecords: number;
-    completedRecords: number;
-  } | undefined>;
-  listPlaygroundRuns: (params: z$1.infer<typeof listPlaygroundRuns>) => Promise<{
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    datasetId: string | null;
-    playgroundId: string;
-    datasetVersionId: string | null;
-    status: string;
-    startedAt: Date | null;
-    completedAt: Date | null;
-    totalRecords: number;
-    completedRecords: number;
-  }[]>;
-  deletePlaygroundRun: (params: z$1.infer<typeof deletePlaygroundRun>) => Promise<{
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    datasetId: string | null;
-    playgroundId: string;
-    datasetVersionId: string | null;
-    status: string;
-    startedAt: Date | null;
-    completedAt: Date | null;
-    totalRecords: number;
-    completedRecords: number;
-  } | undefined>;
-  countPlaygroundRuns: (playgroundId: string) => Promise<number>;
-};
-//#endregion
-//#region src/datalayer/workspaceSettings.d.ts
-declare const updateWorkspaceSettings: z$1.ZodObject<{
-  name: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
-  setupComplete: z$1.ZodOptional<z$1.ZodBoolean>;
-  superAdminId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
-}, z$1.core.$strip>;
-declare const createWorkspaceSettingsDataLayer: (db: Kysely<Database>) => {
-  /**
-   * Get workspace settings (creates default if not exists)
-   */
-  getWorkspaceSettings: () => Promise<{
-    name: string | null;
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    setupComplete: boolean;
-    superAdminId: string | null;
-  } | undefined>;
-  /**
-   * Update workspace settings
-   */
-  updateWorkspaceSettings: (params: z$1.infer<typeof updateWorkspaceSettings>) => Promise<{
-    name: string | null;
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    setupComplete: boolean;
-    superAdminId: string | null;
-  } | undefined>;
-  /**
-   * Get the super admin user ID
-   */
-  getSuperAdminId: () => Promise<string | null>;
-  /**
-   * Set the super admin user ID (only if not already set)
-   */
-  setSuperAdminId: (userId: string) => Promise<boolean>;
-  /**
-   * Check if initial setup has been completed
-   */
-  isSetupComplete: () => Promise<boolean>;
-  /**
-   * Mark initial setup as complete
-   */
-  markSetupComplete: () => Promise<{
-    name: string | null;
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    setupComplete: boolean;
-    superAdminId: string | null;
-  } | undefined>;
-};
-//#endregion
-//#region src/datalayer/interface.d.ts
-type DatasetsDataLayer = ReturnType<typeof createDatasetsDataLayer>;
-type PlaygroundsDataLayer = ReturnType<typeof createPlaygroundDataLayer>;
-type PlaygroundResultsDataLayer = ReturnType<typeof createPlaygroundResultsDataLayer>;
-type PlaygroundRunsDataLayer = ReturnType<typeof createPlaygroundRunsDataLayer>;
-type WorkspaceSettingsDataLayer = ReturnType<typeof createWorkspaceSettingsDataLayer>;
-type DataLayer = DatasetsDataLayer & PlaygroundsDataLayer & PlaygroundResultsDataLayer & PlaygroundRunsDataLayer & WorkspaceSettingsDataLayer;
-//#endregion
-//#region src/datalayer/create.d.ts
-/**
- * Create all datalayers from a Kysely database instance.
- * Returns a flat object with all datalayer methods spread together.
- */
-declare function createDataLayer(db: Kysely<Database>): DataLayer;
-//#endregion
 //#region src/pricing/types.d.ts
 /**
  * Pricing types for cost tracking
@@ -2159,46 +1438,8 @@ declare const insertLLMRequestSchema: z$1.ZodObject<{
   sessionId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
 }, z$1.core.$strip>;
 type LLMRequestInsert = z$1.infer<typeof insertLLMRequestSchema>;
-declare const listRequestsSchema: z$1.ZodObject<{
-  limit: z$1.ZodDefault<z$1.ZodNumber>;
-  offset: z$1.ZodDefault<z$1.ZodNumber>;
-  configId: z$1.ZodOptional<z$1.ZodString>;
-  variantId: z$1.ZodOptional<z$1.ZodString>;
-  environmentId: z$1.ZodOptional<z$1.ZodString>;
-  providerConfigId: z$1.ZodOptional<z$1.ZodString>;
-  provider: z$1.ZodOptional<z$1.ZodString>;
-  model: z$1.ZodOptional<z$1.ZodString>;
-  startDate: z$1.ZodOptional<z$1.ZodDate>;
-  endDate: z$1.ZodOptional<z$1.ZodDate>;
-  tags: z$1.ZodOptional<z$1.ZodRecord<z$1.ZodString, z$1.ZodArray<z$1.ZodString>>>;
-}, z$1.core.$strip>;
-declare const dateRangeSchema: z$1.ZodObject<{
-  startDate: z$1.ZodDate;
-  endDate: z$1.ZodDate;
-  configId: z$1.ZodOptional<z$1.ZodString>;
-  variantId: z$1.ZodOptional<z$1.ZodString>;
-  environmentId: z$1.ZodOptional<z$1.ZodString>;
-  tags: z$1.ZodOptional<z$1.ZodRecord<z$1.ZodString, z$1.ZodArray<z$1.ZodString>>>;
-}, z$1.core.$strip>;
 declare const COST_SUMMARY_GROUP_BY: readonly ["day", "hour", "model", "provider", "endpoint", "tags"];
 type CostSummaryGroupBy = (typeof COST_SUMMARY_GROUP_BY)[number];
-declare const costSummarySchema: z$1.ZodObject<{
-  startDate: z$1.ZodDate;
-  endDate: z$1.ZodDate;
-  configId: z$1.ZodOptional<z$1.ZodString>;
-  variantId: z$1.ZodOptional<z$1.ZodString>;
-  environmentId: z$1.ZodOptional<z$1.ZodString>;
-  tags: z$1.ZodOptional<z$1.ZodRecord<z$1.ZodString, z$1.ZodArray<z$1.ZodString>>>;
-  groupBy: z$1.ZodOptional<z$1.ZodEnum<{
-    provider: "provider";
-    model: "model";
-    endpoint: "endpoint";
-    tags: "tags";
-    day: "day";
-    hour: "hour";
-  }>>;
-  tagKeys: z$1.ZodOptional<z$1.ZodArray<z$1.ZodString>>;
-}, z$1.core.$strip>;
 declare const upsertTraceSchema: z$1.ZodObject<{
   traceId: z$1.ZodString;
   name: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
@@ -2261,220 +1502,71 @@ declare const insertSpanEventSchema: z$1.ZodObject<{
   attributes: z$1.ZodDefault<z$1.ZodRecord<z$1.ZodString, z$1.ZodUnknown>>;
 }, z$1.core.$strip>;
 type SpanEventInsert = z$1.infer<typeof insertSpanEventSchema>;
-declare const listTracesSchema: z$1.ZodObject<{
-  limit: z$1.ZodDefault<z$1.ZodNumber>;
-  offset: z$1.ZodDefault<z$1.ZodNumber>;
-  sessionId: z$1.ZodOptional<z$1.ZodString>;
-  userId: z$1.ZodOptional<z$1.ZodString>;
-  status: z$1.ZodOptional<z$1.ZodEnum<{
-    error: "error";
-    unset: "unset";
-    ok: "ok";
-  }>>;
-  name: z$1.ZodOptional<z$1.ZodString>;
-  startDate: z$1.ZodOptional<z$1.ZodDate>;
-  endDate: z$1.ZodOptional<z$1.ZodDate>;
-  tags: z$1.ZodOptional<z$1.ZodRecord<z$1.ZodString, z$1.ZodArray<z$1.ZodString>>>;
-}, z$1.core.$strip>;
-declare const traceStatsSchema: z$1.ZodObject<{
-  startDate: z$1.ZodDate;
-  endDate: z$1.ZodDate;
-  sessionId: z$1.ZodOptional<z$1.ZodString>;
-  userId: z$1.ZodOptional<z$1.ZodString>;
-}, z$1.core.$strip>;
-declare function createLLMRequestsStore(db: Kysely<Database>): {
+declare function createLLMRequestsStore(pool: any): {
   batchInsertRequests: (requests: LLMRequestInsert[]) => Promise<{
     count: number;
   }>;
-  insertRequest: (request: LLMRequestInsert) => Promise<{
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    variantId: string | null;
-    provider: string;
-    environmentId: string | null;
-    configId: string | null;
-    providerConfigId: string | null;
-    latencyMs: number;
-    promptTokens: number;
-    completionTokens: number;
-    totalTokens: number;
-    cost: number;
-    requestId: string;
-    model: string;
-    cachedTokens: number;
-    cacheCreationTokens: number;
-    cacheSavings: number;
-    inputCost: number;
-    outputCost: number;
-    endpoint: string;
-    statusCode: number;
-    isStreaming: boolean;
-    userId: string | null;
-    tags: Record<string, string>;
-    guardrailResults: {
-      results: {
-        checkId: string;
-        functionId: string;
-        hookType: "beforeRequestHook" | "afterRequestHook";
-        verdict: boolean;
-        latencyMs: number;
-      }[];
-      action: "allowed" | "blocked" | "logged";
-      totalLatencyMs: number;
-    } | null;
-    traceId: string | null;
-    spanId: string | null;
-    parentSpanId: string | null;
-    sessionId: string | null;
-  } | undefined>;
-  listRequests: (params?: z$1.infer<typeof listRequestsSchema>) => Promise<{
-    data: {
-      id: string;
-      createdAt: Date;
-      updatedAt: Date;
-      variantId: string | null;
-      provider: string;
-      environmentId: string | null;
-      configId: string | null;
-      providerConfigId: string | null;
-      latencyMs: number;
-      promptTokens: number;
-      completionTokens: number;
-      totalTokens: number;
-      cost: number;
-      requestId: string;
-      model: string;
-      cachedTokens: number;
-      cacheCreationTokens: number;
-      cacheSavings: number;
-      inputCost: number;
-      outputCost: number;
-      endpoint: string;
-      statusCode: number;
-      isStreaming: boolean;
-      userId: string | null;
-      tags: Record<string, string>;
-      guardrailResults: {
-        results: {
-          checkId: string;
-          functionId: string;
-          hookType: "beforeRequestHook" | "afterRequestHook";
-          verdict: boolean;
-          latencyMs: number;
-        }[];
-        action: "allowed" | "blocked" | "logged";
-        totalLatencyMs: number;
-      } | null;
-      traceId: string | null;
-      spanId: string | null;
-      parentSpanId: string | null;
-      sessionId: string | null;
-    }[];
-    total: number;
+  insertRequest: (request: LLMRequestInsert) => Promise<any>;
+  listRequests: (params?: {
+    limit?: number;
+    offset?: number;
+    configId?: string;
+    variantId?: string;
+    environmentId?: string;
+    providerConfigId?: string;
+    provider?: string;
+    model?: string;
+    startDate?: Date;
+    endDate?: Date;
+    tags?: Record<string, string[]>;
+  }) => Promise<{
+    data: any;
+    total: any;
     limit: number;
     offset: number;
   }>;
-  getRequestByRequestId: (requestId: string) => Promise<{
-    id: string;
-    createdAt: Date;
-    updatedAt: Date;
-    variantId: string | null;
-    provider: string;
-    environmentId: string | null;
-    configId: string | null;
-    providerConfigId: string | null;
-    latencyMs: number;
-    promptTokens: number;
-    completionTokens: number;
-    totalTokens: number;
-    cost: number;
-    requestId: string;
-    model: string;
-    cachedTokens: number;
-    cacheCreationTokens: number;
-    cacheSavings: number;
-    inputCost: number;
-    outputCost: number;
-    endpoint: string;
-    statusCode: number;
-    isStreaming: boolean;
-    userId: string | null;
-    tags: Record<string, string>;
-    guardrailResults: {
-      results: {
-        checkId: string;
-        functionId: string;
-        hookType: "beforeRequestHook" | "afterRequestHook";
-        verdict: boolean;
-        latencyMs: number;
-      }[];
-      action: "allowed" | "blocked" | "logged";
-      totalLatencyMs: number;
-    } | null;
-    traceId: string | null;
-    spanId: string | null;
-    parentSpanId: string | null;
-    sessionId: string | null;
-  } | undefined>;
-  getTotalCost: (params: z$1.infer<typeof dateRangeSchema>) => Promise<{
-    totalCost: number;
-    totalInputCost: number;
-    totalOutputCost: number;
-    totalPromptTokens: number;
-    totalCompletionTokens: number;
-    totalTokens: number;
-    totalCachedTokens: number;
-    totalCacheSavings: number;
-    requestCount: number;
-  } | undefined>;
-  getCostByModel: (params: z$1.infer<typeof dateRangeSchema>) => Promise<{
-    provider: string;
-    model: string;
-    totalCost: number;
-    totalInputCost: number;
-    totalOutputCost: number;
-    totalTokens: number;
-    requestCount: number;
-    avgLatencyMs: number;
-  }[]>;
-  getCostByProvider: (params: z$1.infer<typeof dateRangeSchema>) => Promise<{
-    provider: string;
-    totalCost: number;
-    totalInputCost: number;
-    totalOutputCost: number;
-    totalTokens: number;
-    requestCount: number;
-    avgLatencyMs: number;
-  }[]>;
-  getDailyCosts: (params: z$1.infer<typeof dateRangeSchema>) => Promise<{
-    totalCost: number;
-    totalInputCost: number;
-    totalOutputCost: number;
-    totalTokens: number;
-    requestCount: number;
-    date: string;
-  }[]>;
-  getCostSummary: (params: z$1.infer<typeof costSummarySchema>) => Promise<{
-    totalCost: number;
-    requestCount: number;
-    groupKey: string;
-  }[]>;
-  getRequestStats: (params: z$1.infer<typeof dateRangeSchema>) => Promise<{
-    avgLatencyMs: number;
-    totalRequests: number;
-    successfulRequests: number;
-    failedRequests: number;
-    streamingRequests: number;
-    maxLatencyMs: number;
-    minLatencyMs: number;
-  } | undefined>;
-  getDistinctTags: () => Promise<{
-    key: string;
-    value: string;
-  }[]>;
+  getRequestByRequestId: (requestId: string) => Promise<any>;
+  getTotalCost: (params: {
+    startDate: Date;
+    endDate: Date;
+    configId?: string;
+    variantId?: string;
+    environmentId?: string;
+    tags?: Record<string, string[]>;
+  }) => Promise<any>;
+  getCostByModel: (params: {
+    startDate: Date;
+    endDate: Date;
+  }) => Promise<any>;
+  getCostByProvider: (params: {
+    startDate: Date;
+    endDate: Date;
+  }) => Promise<any>;
+  getDailyCosts: (params: {
+    startDate: Date;
+    endDate: Date;
+  }) => Promise<any>;
+  getCostSummary: (params: {
+    startDate: Date;
+    endDate: Date;
+    configId?: string;
+    variantId?: string;
+    environmentId?: string;
+    groupBy?: CostSummaryGroupBy;
+    tags?: Record<string, string[]>;
+    tagKeys?: string[];
+  }) => Promise<any>;
+  getRequestStats: (params: {
+    startDate: Date;
+    endDate: Date;
+    configId?: string;
+    variantId?: string;
+    environmentId?: string;
+    tags?: Record<string, string[]>;
+  }) => Promise<any>;
+  getDistinctTags: () => Promise<any>;
 };
-declare function createTracesStore(db: Kysely<Database>): {
+declare function createTracesStore(pool: any): {
   upsertTrace: (data: TraceUpsert) => Promise<void>;
   batchInsertSpans: (spans: SpanInsert[]) => Promise<{
     count: number;
@@ -2482,103 +1574,37 @@ declare function createTracesStore(db: Kysely<Database>): {
   batchInsertSpanEvents: (events: SpanEventInsert[]) => Promise<{
     count: number;
   }>;
-  listTraces: (params?: z$1.infer<typeof listTracesSchema>) => Promise<{
-    data: {
-      name: string | null;
-      id: string;
-      createdAt: Date;
-      updatedAt: Date;
-      status: string;
-      totalTokens: number;
-      metadata: Record<string, unknown>;
-      userId: string | null;
-      tags: Record<string, string>;
-      traceId: string;
-      sessionId: string | null;
-      startTime: Date;
-      endTime: Date | null;
-      durationMs: number | null;
-      spanCount: number;
-      totalInputTokens: number;
-      totalOutputTokens: number;
-      totalCost: number;
-    }[];
-    total: number;
+  listTraces: (params?: {
+    limit?: number;
+    offset?: number;
+    sessionId?: string;
+    userId?: string;
+    status?: string;
+    name?: string;
+    startDate?: Date;
+    endDate?: Date;
+    tags?: Record<string, string[]>;
+  }) => Promise<{
+    data: any;
+    total: any;
     limit: number;
     offset: number;
   }>;
   getTraceWithSpans: (traceId: string) => Promise<{
-    trace: {
-      name: string | null;
-      id: string;
-      createdAt: Date;
-      updatedAt: Date;
-      status: string;
-      totalTokens: number;
-      metadata: Record<string, unknown>;
-      userId: string | null;
-      tags: Record<string, string>;
-      traceId: string;
-      sessionId: string | null;
-      startTime: Date;
-      endTime: Date | null;
-      durationMs: number | null;
-      spanCount: number;
-      totalInputTokens: number;
-      totalOutputTokens: number;
-      totalCost: number;
-    };
-    spans: {
-      name: string;
-      id: string;
-      createdAt: Date;
-      updatedAt: Date;
-      variantId: string | null;
-      provider: string | null;
-      input: unknown;
-      output: unknown;
-      environmentId: string | null;
-      configId: string | null;
-      providerConfigId: string | null;
-      status: number;
-      promptTokens: number;
-      completionTokens: number;
-      totalTokens: number;
-      cost: number;
-      requestId: string | null;
-      model: string | null;
-      traceId: string;
-      spanId: string;
-      parentSpanId: string | null;
-      startTime: Date;
-      endTime: Date | null;
-      durationMs: number | null;
-      kind: number;
-      statusMessage: string | null;
-      source: string;
-      attributes: Record<string, unknown>;
-    }[];
-    events: {
-      name: string;
-      id: string;
-      createdAt: Date;
-      traceId: string;
-      spanId: string;
-      attributes: Record<string, unknown>;
-      timestamp: Date;
-    }[];
-  } | undefined>;
-  getTraceStats: (params: z$1.infer<typeof traceStatsSchema>) => Promise<{
-    totalCost: number;
-    totalTokens: number;
-    totalTraces: number;
-    avgDurationMs: number;
-    errorCount: number;
-    totalSpans: number;
+    trace: any;
+    spans: any;
+    events: any;
   } | undefined>;
+  getTraceStats: (params: {
+    startDate: Date;
+    endDate: Date;
+    sessionId?: string;
+    userId?: string;
+  }) => Promise<any>;
 };
 type PgStore = ReturnType<typeof createLLMRequestsStore> & ReturnType<typeof createTracesStore> & {
-  _db: Kysely<Database>;
+  _pool: unknown;
+  _schema: string;
 };
 /**
  * Create a PostgreSQL-backed telemetry store.
@@ -2602,6 +1628,6 @@ declare function createPgStore(connectionString: string, options?: {
  * TelemetryStore provides read + write access to telemetry data.
  * Implemented by pgStore (and future sqliteStore, etc.)
  */
-type TelemetryStore = Omit<PgStore, '_db'>;
+type TelemetryStore = Omit<PgStore, '_pool' | '_schema'>;
 //#endregion
-export { type AnthropicProviderConfig, type AnyProviderConfig, type AzureAIProviderConfig, type AzureOpenAIProviderConfig, BaseCacheConfig, type BaseProviderConfig, type BedrockProviderConfig, COST_SUMMARY_GROUP_BY, CacheBackend, CacheBackendType, CacheConfig, CacheEntry, CacheOptions, CacheService, CacheStats, ChatCompletionCreateParamsBase, Config, ConfigVariant, ConfigVariantsTable, ConfigsTable, type CortexProviderConfig, CostResult, CostSummaryGroupBy, DEFAULT_PROVIDER_ENV_VARS, DataLayer, Database, DatabaseConnection, DatabaseOptions, DatabaseType, Dataset, DatasetRecord, DatasetRecordsTable, DatasetVersion, DatasetVersionRecord, DatasetVersionRecordsTable, DatasetVersionsTable, DatasetsDataLayer, DatasetsTable, Environment, EnvironmentSecret, EnvironmentSecretsTable, EnvironmentsTable, FileCacheBackend, FileCacheConfig, type FireworksAIProviderConfig, type GoogleProviderConfig, GuardrailConfig, GuardrailConfigsTable, GuardrailResult, GuardrailResults, type HuggingFaceProviderConfig, type InlineProviderConfig, type InlineProvidersConfig, Insertable, LLMOPS_INTERNAL_HEADER, LLMOPS_REQUEST_ID_HEADER, LLMOPS_SESSION_ID_HEADER, LLMOPS_SPAN_ID_HEADER, LLMOPS_SPAN_NAME_HEADER, LLMOPS_TRACE_ID_HEADER, LLMOPS_TRACE_NAME_HEADER, LLMOPS_USER_ID_HEADER, LLMOpsClient, LLMOpsConfig, type LLMOpsConfigInput, LLMOpsPricingProvider, LLMRequest, LLMRequestInsert, LLMRequestsTable, MS, MemoryCacheBackend, MemoryCacheConfig, MigrationOptions, MigrationResult, type MistralAIProviderConfig, ModelPricing, type OpenAIProviderConfig, type OracleProviderConfig, PgStore, Playground, PlaygroundColumn, PlaygroundResult, PlaygroundResultsDataLayer, PlaygroundResultsTable, PlaygroundRun, PlaygroundRunsDataLayer, PlaygroundRunsTable, PlaygroundsDataLayer, PlaygroundsTable, Prettify, PricingProvider, ProviderConfig, type ProviderConfigMap, ProviderConfigsTable, ProviderGuardrailOverride, ProviderGuardrailOverridesTable, type ProvidersConfig, SCHEMA_METADATA, type SagemakerProviderConfig, Selectable, Span, SpanEvent, SpanEventInsert, SpanEventsTable, SpanInsert, SpansTable, type StabilityAIProviderConfig, SupportedProviders, TableName, TargetingRule, TargetingRulesTable, TelemetryStore, Trace, TraceUpsert, TracesTable, Updateable, UsageData, type ValidatedLLMOpsConfig, Variant, VariantJsonData, VariantVersion, VariantVersionsTable, VariantsTable, type VertexAIProviderConfig, type WorkersAIProviderConfig, WorkspaceSettings, WorkspaceSettingsDataLayer, WorkspaceSettingsTable, calculateCacheAwareCost, calculateCost, chatCompletionCreateParamsBaseSchema, configVariantsSchema, configsSchema, createDataLayer, createDatabase, createDatabaseFromConnection, createDatasetsDataLayer, createNeonDialect, createPgStore, createPlaygroundDataLayer, createPlaygroundResultsDataLayer, createPlaygroundRunsDataLayer, createWorkspaceSettingsDataLayer, datasetRecordsSchema, datasetVersionRecordsSchema, datasetVersionsSchema, datasetsSchema, detectDatabaseType, dollarsToMicroDollars, environmentSecretsSchema, environmentsSchema, executeWithSchema, formatCost, gateway, generateId, getDefaultPricingProvider, getDefaultProviders, getMigrations, guardrailConfigsSchema, llmRequestsSchema, llmopsConfigSchema, logger, matchType, mergeWithDefaultProviders, microDollarsToDollars, parsePartialTableData, parseTableData, playgroundColumnSchema, playgroundResultsSchema, playgroundRunsSchema, playgroundsSchema, providerConfigsSchema, providerGuardrailOverridesSchema, runAutoMigrations, schemas, spanEventsSchema, spansSchema, targetingRulesSchema, tracesSchema, validateLLMOpsConfig, validatePartialTableData, validateTableData, variantJsonDataSchema, variantVersionsSchema, variantsSchema, workspaceSettingsSchema };
+export { type AnthropicProviderConfig, type AnyProviderConfig, type AzureAIProviderConfig, type AzureOpenAIProviderConfig, BaseCacheConfig, type BaseProviderConfig, type BedrockProviderConfig, COST_SUMMARY_GROUP_BY, CacheBackend, CacheBackendType, CacheConfig, CacheEntry, CacheOptions, CacheService, CacheStats, ChatCompletionCreateParamsBase, type CortexProviderConfig, CostResult, CostSummaryGroupBy, DEFAULT_PROVIDER_ENV_VARS, FileCacheBackend, FileCacheConfig, type FireworksAIProviderConfig, type GoogleProviderConfig, type HuggingFaceProviderConfig, type InlineProviderConfig, type InlineProvidersConfig, LLMOPS_INTERNAL_HEADER, LLMOPS_REQUEST_ID_HEADER, LLMOPS_SESSION_ID_HEADER, LLMOPS_SPAN_ID_HEADER, LLMOPS_SPAN_NAME_HEADER, LLMOPS_TRACE_ID_HEADER, LLMOPS_TRACE_NAME_HEADER, LLMOPS_USER_ID_HEADER, LLMOpsClient, LLMOpsConfig, type LLMOpsConfigInput, LLMOpsPricingProvider, LLMRequestInsert, MS, MemoryCacheBackend, MemoryCacheConfig, type MistralAIProviderConfig, ModelPricing, type OpenAIProviderConfig, type OracleProviderConfig, PgStore, Prettify, PricingProvider, type ProviderConfigMap, type ProvidersConfig, type SagemakerProviderConfig, SpanEventInsert, SpanInsert, type StabilityAIProviderConfig, SupportedProviders, TelemetryStore, TraceUpsert, UsageData, type ValidatedLLMOpsConfig, VariantJsonData, type VertexAIProviderConfig, type WorkersAIProviderConfig, calculateCacheAwareCost, calculateCost, chatCompletionCreateParamsBaseSchema, createPgStore, dollarsToMicroDollars, formatCost, gateway, generateId, getDefaultPricingProvider, getDefaultProviders, llmopsConfigSchema, logger, mergeWithDefaultProviders, microDollarsToDollars, validateLLMOpsConfig, variantJsonDataSchema };