@llmops/core 0.5.4 → 0.6.0

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.
package/dist/index.d.cts CHANGED
@@ -1,4 +1,4 @@
1
- import { $ as SCHEMA_METADATA, A as DatasetsTable, At as variantsSchema, B as LLMRequest, C as Dataset, Ct as playgroundRunsSchema, D as DatasetVersionRecord, Dt as schemas, E as DatasetVersion, Et as providerGuardrailOverridesSchema, F as GuardrailConfig, G as PlaygroundResultsTable, H as Playground, I as GuardrailConfigsTable, J as PlaygroundsTable, K as PlaygroundRun, L as GuardrailResult, M as EnvironmentSecret, N as EnvironmentSecretsTable, O as DatasetVersionRecordsTable, Ot as targetingRulesSchema, P as EnvironmentsTable, Q as ProviderGuardrailOverridesTable, R as GuardrailResults, S as Database, St as playgroundResultsSchema, T as DatasetRecordsTable, Tt as providerConfigsSchema, U as PlaygroundColumn, V as LLMRequestsTable, W as PlaygroundResult, X as ProviderConfigsTable, Y as ProviderConfig, Z as ProviderGuardrailOverride, _ as validateTableData, _t as environmentSecretsSchema, a as createDatabaseFromConnection, at as Variant, b as ConfigVariantsTable, bt as llmRequestsSchema, c as executeWithSchema, ct as VariantsTable, d as getMigrations, dt as configVariantsSchema, et as Selectable, f as matchType, ft as configsSchema, g as validatePartialTableData, gt as datasetsSchema, h as parseTableData, ht as datasetVersionsSchema, i as createDatabase, it as Updateable, j as Environment, jt as workspaceSettingsSchema, k as DatasetVersionsTable, kt as variantVersionsSchema, l as MigrationOptions, lt as WorkspaceSettings, m as parsePartialTableData, mt as datasetVersionRecordsSchema, n as DatabaseOptions, nt as TargetingRule, o as detectDatabaseType, ot as VariantVersion, p as runAutoMigrations, pt as datasetRecordsSchema, q as PlaygroundRunsTable, r as DatabaseType, rt as TargetingRulesTable, s as createNeonDialect, st as VariantVersionsTable, t as DatabaseConnection, tt as TableName, u as MigrationResult, ut as WorkspaceSettingsTable, v as Config, vt as environmentsSchema, w as DatasetRecord, wt as playgroundsSchema, x as ConfigsTable, xt as playgroundColumnSchema, y as ConfigVariant, yt as guardrailConfigsSchema, z as Insertable } from "./index-BtSgIKup.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-DdG7GtcE.cjs";
2
2
  import * as kysely0 from "kysely";
3
3
  import { Kysely } from "kysely";
4
4
  import z$1, { z } from "zod";
@@ -7,6 +7,19 @@ import pino from "pino";
7
7
  import { BetterAuthOptions } from "better-auth";
8
8
  import { RulesLogic } from "json-logic-js";
9
9
 
10
+ //#region src/constants/headers.d.ts
11
+ /**
12
+ * Custom HTTP headers used for trace context propagation
13
+ * between SDK clients and the LLMOps gateway.
14
+ */
15
+ declare const LLMOPS_TRACE_ID_HEADER = "x-llmops-trace-id";
16
+ declare const LLMOPS_TRACE_NAME_HEADER = "x-llmops-trace-name";
17
+ declare const LLMOPS_SPAN_NAME_HEADER = "x-llmops-span-name";
18
+ declare const LLMOPS_SESSION_ID_HEADER = "x-llmops-session-id";
19
+ declare const LLMOPS_USER_ID_HEADER = "x-llmops-user-id";
20
+ declare const LLMOPS_SPAN_ID_HEADER = "x-llmops-span-id";
21
+ declare const LLMOPS_REQUEST_ID_HEADER = "x-llmops-request-id";
22
+ //#endregion
10
23
  //#region src/providers/supported-providers.d.ts
11
24
  /**
12
25
  * Supported providers - derived from @llmops/gateway providers
@@ -1969,6 +1982,10 @@ declare const insertLLMRequestSchema: z$1.ZodObject<{
1969
1982
  }>;
1970
1983
  totalLatencyMs: z$1.ZodNumber;
1971
1984
  }, z$1.core.$strip>>>;
1985
+ traceId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
1986
+ spanId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
1987
+ parentSpanId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
1988
+ sessionId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
1972
1989
  }, z$1.core.$strip>;
1973
1990
  type LLMRequestInsert = z$1.infer<typeof insertLLMRequestSchema>;
1974
1991
  /**
@@ -2069,6 +2086,10 @@ declare const createLLMRequestsDataLayer: (db: Kysely<Database>) => {
2069
2086
  action: "allowed" | "blocked" | "logged";
2070
2087
  totalLatencyMs: number;
2071
2088
  } | null;
2089
+ traceId: string | null;
2090
+ spanId: string | null;
2091
+ parentSpanId: string | null;
2092
+ sessionId: string | null;
2072
2093
  } | undefined>;
2073
2094
  /**
2074
2095
  * List LLM requests with filtering and pagination
@@ -2110,6 +2131,10 @@ declare const createLLMRequestsDataLayer: (db: Kysely<Database>) => {
2110
2131
  action: "allowed" | "blocked" | "logged";
2111
2132
  totalLatencyMs: number;
2112
2133
  } | null;
2134
+ traceId: string | null;
2135
+ spanId: string | null;
2136
+ parentSpanId: string | null;
2137
+ sessionId: string | null;
2113
2138
  }[];
2114
2139
  total: number;
2115
2140
  limit: number;
@@ -2153,6 +2178,10 @@ declare const createLLMRequestsDataLayer: (db: Kysely<Database>) => {
2153
2178
  action: "allowed" | "blocked" | "logged";
2154
2179
  totalLatencyMs: number;
2155
2180
  } | null;
2181
+ traceId: string | null;
2182
+ spanId: string | null;
2183
+ parentSpanId: string | null;
2184
+ sessionId: string | null;
2156
2185
  } | undefined>;
2157
2186
  /**
2158
2187
  * Get total cost for a date range with optional filters
@@ -3174,6 +3203,226 @@ declare const createTargetingRulesDataLayer: (db: Kysely<Database>) => {
3174
3203
  } | undefined>;
3175
3204
  };
3176
3205
  //#endregion
3206
+ //#region src/datalayer/traces.d.ts
3207
+ /**
3208
+ * Schema for upserting a trace
3209
+ */
3210
+ declare const upsertTraceSchema: z$1.ZodObject<{
3211
+ traceId: z$1.ZodString;
3212
+ name: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3213
+ sessionId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3214
+ userId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3215
+ status: z$1.ZodDefault<z$1.ZodEnum<{
3216
+ error: "error";
3217
+ unset: "unset";
3218
+ ok: "ok";
3219
+ }>>;
3220
+ startTime: z$1.ZodDate;
3221
+ endTime: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodDate>>;
3222
+ durationMs: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodNumber>>;
3223
+ spanCount: z$1.ZodDefault<z$1.ZodNumber>;
3224
+ totalInputTokens: z$1.ZodDefault<z$1.ZodNumber>;
3225
+ totalOutputTokens: z$1.ZodDefault<z$1.ZodNumber>;
3226
+ totalTokens: z$1.ZodDefault<z$1.ZodNumber>;
3227
+ totalCost: z$1.ZodDefault<z$1.ZodNumber>;
3228
+ tags: z$1.ZodDefault<z$1.ZodRecord<z$1.ZodString, z$1.ZodString>>;
3229
+ metadata: z$1.ZodDefault<z$1.ZodRecord<z$1.ZodString, z$1.ZodUnknown>>;
3230
+ }, z$1.core.$strip>;
3231
+ type TraceUpsert = z$1.infer<typeof upsertTraceSchema>;
3232
+ /**
3233
+ * Schema for inserting spans
3234
+ */
3235
+ declare const insertSpanSchema: z$1.ZodObject<{
3236
+ traceId: z$1.ZodString;
3237
+ spanId: z$1.ZodString;
3238
+ parentSpanId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3239
+ name: z$1.ZodString;
3240
+ kind: z$1.ZodDefault<z$1.ZodNumber>;
3241
+ status: z$1.ZodDefault<z$1.ZodNumber>;
3242
+ statusMessage: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3243
+ startTime: z$1.ZodDate;
3244
+ endTime: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodDate>>;
3245
+ durationMs: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodNumber>>;
3246
+ provider: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3247
+ model: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3248
+ promptTokens: z$1.ZodDefault<z$1.ZodNumber>;
3249
+ completionTokens: z$1.ZodDefault<z$1.ZodNumber>;
3250
+ totalTokens: z$1.ZodDefault<z$1.ZodNumber>;
3251
+ cost: z$1.ZodDefault<z$1.ZodNumber>;
3252
+ configId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3253
+ variantId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3254
+ environmentId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3255
+ providerConfigId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3256
+ requestId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3257
+ source: z$1.ZodDefault<z$1.ZodEnum<{
3258
+ gateway: "gateway";
3259
+ otlp: "otlp";
3260
+ }>>;
3261
+ input: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodUnknown>>;
3262
+ output: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodUnknown>>;
3263
+ attributes: z$1.ZodDefault<z$1.ZodRecord<z$1.ZodString, z$1.ZodUnknown>>;
3264
+ }, z$1.core.$strip>;
3265
+ type SpanInsert = z$1.infer<typeof insertSpanSchema>;
3266
+ /**
3267
+ * Schema for inserting span events
3268
+ */
3269
+ declare const insertSpanEventSchema: z$1.ZodObject<{
3270
+ traceId: z$1.ZodString;
3271
+ spanId: z$1.ZodString;
3272
+ name: z$1.ZodString;
3273
+ timestamp: z$1.ZodDate;
3274
+ attributes: z$1.ZodDefault<z$1.ZodRecord<z$1.ZodString, z$1.ZodUnknown>>;
3275
+ }, z$1.core.$strip>;
3276
+ type SpanEventInsert = z$1.infer<typeof insertSpanEventSchema>;
3277
+ /**
3278
+ * Schema for listing traces
3279
+ */
3280
+ declare const listTracesSchema: z$1.ZodObject<{
3281
+ limit: z$1.ZodDefault<z$1.ZodNumber>;
3282
+ offset: z$1.ZodDefault<z$1.ZodNumber>;
3283
+ sessionId: z$1.ZodOptional<z$1.ZodString>;
3284
+ userId: z$1.ZodOptional<z$1.ZodString>;
3285
+ status: z$1.ZodOptional<z$1.ZodEnum<{
3286
+ error: "error";
3287
+ unset: "unset";
3288
+ ok: "ok";
3289
+ }>>;
3290
+ name: z$1.ZodOptional<z$1.ZodString>;
3291
+ startDate: z$1.ZodOptional<z$1.ZodDate>;
3292
+ endDate: z$1.ZodOptional<z$1.ZodDate>;
3293
+ tags: z$1.ZodOptional<z$1.ZodRecord<z$1.ZodString, z$1.ZodArray<z$1.ZodString>>>;
3294
+ }, z$1.core.$strip>;
3295
+ /**
3296
+ * Schema for trace stats query
3297
+ */
3298
+ declare const traceStatsSchema: z$1.ZodObject<{
3299
+ startDate: z$1.ZodDate;
3300
+ endDate: z$1.ZodDate;
3301
+ sessionId: z$1.ZodOptional<z$1.ZodString>;
3302
+ userId: z$1.ZodOptional<z$1.ZodString>;
3303
+ }, z$1.core.$strip>;
3304
+ declare const createTracesDataLayer: (db: Kysely<Database>) => {
3305
+ /**
3306
+ * Upsert a trace — insert or update aggregates on conflict
3307
+ */
3308
+ upsertTrace: (data: TraceUpsert) => Promise<void>;
3309
+ /**
3310
+ * Batch insert spans
3311
+ */
3312
+ batchInsertSpans: (spans: SpanInsert[]) => Promise<{
3313
+ count: number;
3314
+ }>;
3315
+ /**
3316
+ * Batch insert span events
3317
+ */
3318
+ batchInsertSpanEvents: (events: SpanEventInsert[]) => Promise<{
3319
+ count: number;
3320
+ }>;
3321
+ /**
3322
+ * List traces with filtering and pagination
3323
+ */
3324
+ listTraces: (params?: z$1.infer<typeof listTracesSchema>) => Promise<{
3325
+ data: {
3326
+ name: string | null;
3327
+ id: string;
3328
+ createdAt: Date;
3329
+ updatedAt: Date;
3330
+ status: string;
3331
+ totalTokens: number;
3332
+ metadata: Record<string, unknown>;
3333
+ userId: string | null;
3334
+ tags: Record<string, string>;
3335
+ traceId: string;
3336
+ sessionId: string | null;
3337
+ startTime: Date;
3338
+ endTime: Date | null;
3339
+ durationMs: number | null;
3340
+ spanCount: number;
3341
+ totalInputTokens: number;
3342
+ totalOutputTokens: number;
3343
+ totalCost: number;
3344
+ }[];
3345
+ total: number;
3346
+ limit: number;
3347
+ offset: number;
3348
+ }>;
3349
+ /**
3350
+ * Get a single trace with all its spans and events
3351
+ */
3352
+ getTraceWithSpans: (traceId: string) => Promise<{
3353
+ trace: {
3354
+ name: string | null;
3355
+ id: string;
3356
+ createdAt: Date;
3357
+ updatedAt: Date;
3358
+ status: string;
3359
+ totalTokens: number;
3360
+ metadata: Record<string, unknown>;
3361
+ userId: string | null;
3362
+ tags: Record<string, string>;
3363
+ traceId: string;
3364
+ sessionId: string | null;
3365
+ startTime: Date;
3366
+ endTime: Date | null;
3367
+ durationMs: number | null;
3368
+ spanCount: number;
3369
+ totalInputTokens: number;
3370
+ totalOutputTokens: number;
3371
+ totalCost: number;
3372
+ };
3373
+ spans: {
3374
+ name: string;
3375
+ id: string;
3376
+ createdAt: Date;
3377
+ updatedAt: Date;
3378
+ variantId: string | null;
3379
+ provider: string | null;
3380
+ input: unknown;
3381
+ output: unknown;
3382
+ environmentId: string | null;
3383
+ configId: string | null;
3384
+ providerConfigId: string | null;
3385
+ status: number;
3386
+ promptTokens: number;
3387
+ completionTokens: number;
3388
+ totalTokens: number;
3389
+ cost: number;
3390
+ requestId: string | null;
3391
+ model: string | null;
3392
+ traceId: string;
3393
+ spanId: string;
3394
+ parentSpanId: string | null;
3395
+ startTime: Date;
3396
+ endTime: Date | null;
3397
+ durationMs: number | null;
3398
+ kind: number;
3399
+ statusMessage: string | null;
3400
+ source: string;
3401
+ attributes: Record<string, unknown>;
3402
+ }[];
3403
+ events: {
3404
+ name: string;
3405
+ id: string;
3406
+ createdAt: Date;
3407
+ traceId: string;
3408
+ spanId: string;
3409
+ attributes: Record<string, unknown>;
3410
+ timestamp: Date;
3411
+ }[];
3412
+ } | undefined>;
3413
+ /**
3414
+ * Get aggregate trace statistics for a date range
3415
+ */
3416
+ getTraceStats: (params: z$1.infer<typeof traceStatsSchema>) => Promise<{
3417
+ totalCost: number;
3418
+ totalTokens: number;
3419
+ totalTraces: number;
3420
+ avgDurationMs: number;
3421
+ errorCount: number;
3422
+ totalSpans: number;
3423
+ } | undefined>;
3424
+ };
3425
+ //#endregion
3177
3426
  //#region src/datalayer/variants.d.ts
3178
3427
  declare const createVariant: z$1.ZodObject<{
3179
3428
  name: z$1.ZodString;
@@ -3496,8 +3745,9 @@ type ProviderGuardrailOverridesDataLayer = ReturnType<typeof createProviderGuard
3496
3745
  type TargetingRulesDataLayer = ReturnType<typeof createTargetingRulesDataLayer>;
3497
3746
  type VariantsDataLayer = ReturnType<typeof createVariantDataLayer>;
3498
3747
  type VariantVersionsDataLayer = ReturnType<typeof createVariantVersionsDataLayer>;
3748
+ type TracesDataLayer = ReturnType<typeof createTracesDataLayer>;
3499
3749
  type WorkspaceSettingsDataLayer = ReturnType<typeof createWorkspaceSettingsDataLayer>;
3500
- type DataLayer = ConfigsDataLayer & ConfigVariantsDataLayer & DatasetsDataLayer & EnvironmentsDataLayer & EnvironmentSecretsDataLayer & GuardrailConfigsDataLayer & LLMRequestsDataLayer & PlaygroundsDataLayer & PlaygroundResultsDataLayer & PlaygroundRunsDataLayer & ProviderConfigsDataLayer & ProviderGuardrailOverridesDataLayer & TargetingRulesDataLayer & VariantsDataLayer & VariantVersionsDataLayer & WorkspaceSettingsDataLayer;
3750
+ type DataLayer = ConfigsDataLayer & ConfigVariantsDataLayer & DatasetsDataLayer & EnvironmentsDataLayer & EnvironmentSecretsDataLayer & GuardrailConfigsDataLayer & LLMRequestsDataLayer & PlaygroundsDataLayer & PlaygroundResultsDataLayer & PlaygroundRunsDataLayer & ProviderConfigsDataLayer & ProviderGuardrailOverridesDataLayer & TargetingRulesDataLayer & TracesDataLayer & VariantsDataLayer & VariantVersionsDataLayer & WorkspaceSettingsDataLayer;
3501
3751
  //#endregion
3502
3752
  //#region src/datalayer/create.d.ts
3503
3753
  /**
@@ -3923,4 +4173,4 @@ declare class ManifestRouter {
3923
4173
  routeWithWeights(configIdOrSlug: string, environmentId: string, context?: RoutingContext): RoutingResult | null;
3924
4174
  }
3925
4175
  //#endregion
3926
- export { type AnthropicProviderConfig, type AnyProviderConfig, AuthClientDatabaseConfig, AuthClientOptions, type AzureAIProviderConfig, type AzureOpenAIProviderConfig, BaseCacheConfig, type BaseProviderConfig, type BedrockProviderConfig, COST_SUMMARY_GROUP_BY, CacheBackend, CacheBackendType, CacheConfig, CacheEntry, CacheOptions, CacheService, CacheStats, ChatCompletionCreateParamsBase, Config, ConfigVariant, ConfigVariantsDataLayer, ConfigVariantsTable, ConfigsDataLayer, 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, EnvironmentSecretsDataLayer, EnvironmentSecretsTable, EnvironmentsDataLayer, EnvironmentsTable, FileCacheBackend, FileCacheConfig, type FireworksAIProviderConfig, GatewayManifest, type GoogleProviderConfig, GuardrailConfig, GuardrailConfigsDataLayer, GuardrailConfigsTable, GuardrailResult, GuardrailResults, type HuggingFaceProviderConfig, type InlineProviderConfig, type InlineProvidersConfig, Insertable, LLMOpsClient, LLMOpsConfig, type LLMOpsConfigInput, LLMRequest, LLMRequestInsert, LLMRequestsDataLayer, LLMRequestsTable, MS, ManifestBuilder, ManifestConfig, ManifestEnvironment, ManifestGuardrail, ManifestProviderGuardrailOverride, ManifestRouter, ManifestService, ManifestTargetingRule, ManifestVariantVersion, MemoryCacheBackend, MemoryCacheConfig, MigrationOptions, MigrationResult, type MistralAIProviderConfig, ModelPricing, ModelsDevPricingProvider, type OpenAIProviderConfig, type OracleProviderConfig, Playground, PlaygroundColumn, PlaygroundResult, PlaygroundResultsDataLayer, PlaygroundResultsTable, PlaygroundRun, PlaygroundRunsDataLayer, PlaygroundRunsTable, PlaygroundsDataLayer, PlaygroundsTable, Prettify, PricingProvider, ProviderConfig, type ProviderConfigMap, ProviderConfigsDataLayer, ProviderConfigsTable, ProviderGuardrailOverride, ProviderGuardrailOverridesDataLayer, ProviderGuardrailOverridesTable, type ProvidersConfig, RoutingContext, RoutingResult, SCHEMA_METADATA, type SagemakerProviderConfig, Selectable, type StabilityAIProviderConfig, SupportedProviders, TableName, TargetingRule, TargetingRulesDataLayer, TargetingRulesTable, Updateable, UsageData, type ValidatedLLMOpsConfig, Variant, VariantJsonData, VariantVersion, VariantVersionsDataLayer, VariantVersionsTable, VariantsDataLayer, VariantsTable, type VertexAIProviderConfig, type WorkersAIProviderConfig, WorkspaceSettings, WorkspaceSettingsDataLayer, WorkspaceSettingsTable, calculateCost, chatCompletionCreateParamsBaseSchema, configVariantsSchema, configsSchema, createConfigDataLayer, createConfigVariantDataLayer, createDataLayer, createDatabase, createDatabaseFromConnection, createDatasetsDataLayer, createEnvironmentDataLayer, createEnvironmentSecretDataLayer, createGuardrailConfigsDataLayer, createLLMRequestsDataLayer, createNeonDialect, createPlaygroundDataLayer, createPlaygroundResultsDataLayer, createPlaygroundRunsDataLayer, createProviderConfigsDataLayer, createProviderGuardrailOverridesDataLayer, createTargetingRulesDataLayer, createVariantDataLayer, createVariantVersionsDataLayer, createWorkspaceSettingsDataLayer, datasetRecordsSchema, datasetVersionRecordsSchema, datasetVersionsSchema, datasetsSchema, detectDatabaseType, dollarsToMicroDollars, environmentSecretsSchema, environmentsSchema, executeWithSchema, formatCost, gateway, generateId, getAuthClientOptions, getDefaultPricingProvider, getDefaultProviders, getMigrations, guardrailConfigsSchema, llmRequestsSchema, llmopsConfigSchema, logger, matchType, mergeWithDefaultProviders, microDollarsToDollars, parsePartialTableData, parseTableData, playgroundColumnSchema, playgroundResultsSchema, playgroundRunsSchema, playgroundsSchema, providerConfigsSchema, providerGuardrailOverridesSchema, runAutoMigrations, schemas, targetingRulesSchema, validateLLMOpsConfig, validatePartialTableData, validateTableData, variantJsonDataSchema, variantVersionsSchema, variantsSchema, workspaceSettingsSchema };
4176
+ export { type AnthropicProviderConfig, type AnyProviderConfig, AuthClientDatabaseConfig, AuthClientOptions, type AzureAIProviderConfig, type AzureOpenAIProviderConfig, BaseCacheConfig, type BaseProviderConfig, type BedrockProviderConfig, COST_SUMMARY_GROUP_BY, CacheBackend, CacheBackendType, CacheConfig, CacheEntry, CacheOptions, CacheService, CacheStats, ChatCompletionCreateParamsBase, Config, ConfigVariant, ConfigVariantsDataLayer, ConfigVariantsTable, ConfigsDataLayer, 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, EnvironmentSecretsDataLayer, EnvironmentSecretsTable, EnvironmentsDataLayer, EnvironmentsTable, FileCacheBackend, FileCacheConfig, type FireworksAIProviderConfig, GatewayManifest, type GoogleProviderConfig, GuardrailConfig, GuardrailConfigsDataLayer, GuardrailConfigsTable, GuardrailResult, GuardrailResults, type HuggingFaceProviderConfig, type InlineProviderConfig, type InlineProvidersConfig, Insertable, 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, LLMRequest, LLMRequestInsert, LLMRequestsDataLayer, LLMRequestsTable, MS, ManifestBuilder, ManifestConfig, ManifestEnvironment, ManifestGuardrail, ManifestProviderGuardrailOverride, ManifestRouter, ManifestService, ManifestTargetingRule, ManifestVariantVersion, MemoryCacheBackend, MemoryCacheConfig, MigrationOptions, MigrationResult, type MistralAIProviderConfig, ModelPricing, ModelsDevPricingProvider, type OpenAIProviderConfig, type OracleProviderConfig, Playground, PlaygroundColumn, PlaygroundResult, PlaygroundResultsDataLayer, PlaygroundResultsTable, PlaygroundRun, PlaygroundRunsDataLayer, PlaygroundRunsTable, PlaygroundsDataLayer, PlaygroundsTable, Prettify, PricingProvider, ProviderConfig, type ProviderConfigMap, ProviderConfigsDataLayer, ProviderConfigsTable, ProviderGuardrailOverride, ProviderGuardrailOverridesDataLayer, ProviderGuardrailOverridesTable, type ProvidersConfig, RoutingContext, RoutingResult, SCHEMA_METADATA, type SagemakerProviderConfig, Selectable, Span, SpanEvent, SpanEventInsert, SpanEventsTable, SpanInsert, SpansTable, type StabilityAIProviderConfig, SupportedProviders, TableName, TargetingRule, TargetingRulesDataLayer, TargetingRulesTable, Trace, TraceUpsert, TracesDataLayer, TracesTable, Updateable, UsageData, type ValidatedLLMOpsConfig, Variant, VariantJsonData, VariantVersion, VariantVersionsDataLayer, VariantVersionsTable, VariantsDataLayer, VariantsTable, type VertexAIProviderConfig, type WorkersAIProviderConfig, WorkspaceSettings, WorkspaceSettingsDataLayer, WorkspaceSettingsTable, calculateCost, chatCompletionCreateParamsBaseSchema, configVariantsSchema, configsSchema, createConfigDataLayer, createConfigVariantDataLayer, createDataLayer, createDatabase, createDatabaseFromConnection, createDatasetsDataLayer, createEnvironmentDataLayer, createEnvironmentSecretDataLayer, createGuardrailConfigsDataLayer, createLLMRequestsDataLayer, createNeonDialect, createPlaygroundDataLayer, createPlaygroundResultsDataLayer, createPlaygroundRunsDataLayer, createProviderConfigsDataLayer, createProviderGuardrailOverridesDataLayer, createTargetingRulesDataLayer, createTracesDataLayer, createVariantDataLayer, createVariantVersionsDataLayer, createWorkspaceSettingsDataLayer, datasetRecordsSchema, datasetVersionRecordsSchema, datasetVersionsSchema, datasetsSchema, detectDatabaseType, dollarsToMicroDollars, environmentSecretsSchema, environmentsSchema, executeWithSchema, formatCost, gateway, generateId, getAuthClientOptions, 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 };
package/dist/index.d.mts CHANGED
@@ -1,4 +1,4 @@
1
- import { $ as SCHEMA_METADATA, A as DatasetsTable, At as variantsSchema, B as LLMRequest, C as Dataset, Ct as playgroundRunsSchema, D as DatasetVersionRecord, Dt as schemas, E as DatasetVersion, Et as providerGuardrailOverridesSchema, F as GuardrailConfig, G as PlaygroundResultsTable, H as Playground, I as GuardrailConfigsTable, J as PlaygroundsTable, K as PlaygroundRun, L as GuardrailResult, M as EnvironmentSecret, N as EnvironmentSecretsTable, O as DatasetVersionRecordsTable, Ot as targetingRulesSchema, P as EnvironmentsTable, Q as ProviderGuardrailOverridesTable, R as GuardrailResults, S as Database, St as playgroundResultsSchema, T as DatasetRecordsTable, Tt as providerConfigsSchema, U as PlaygroundColumn, V as LLMRequestsTable, W as PlaygroundResult, X as ProviderConfigsTable, Y as ProviderConfig, Z as ProviderGuardrailOverride, _ as validateTableData, _t as environmentSecretsSchema, a as createDatabaseFromConnection, at as Variant, b as ConfigVariantsTable, bt as llmRequestsSchema, c as executeWithSchema, ct as VariantsTable, d as getMigrations, dt as configVariantsSchema, et as Selectable, f as matchType, ft as configsSchema, g as validatePartialTableData, gt as datasetsSchema, h as parseTableData, ht as datasetVersionsSchema, i as createDatabase, it as Updateable, j as Environment, jt as workspaceSettingsSchema, k as DatasetVersionsTable, kt as variantVersionsSchema, l as MigrationOptions, lt as WorkspaceSettings, m as parsePartialTableData, mt as datasetVersionRecordsSchema, n as DatabaseOptions, nt as TargetingRule, o as detectDatabaseType, ot as VariantVersion, p as runAutoMigrations, pt as datasetRecordsSchema, q as PlaygroundRunsTable, r as DatabaseType, rt as TargetingRulesTable, s as createNeonDialect, st as VariantVersionsTable, t as DatabaseConnection, tt as TableName, u as MigrationResult, ut as WorkspaceSettingsTable, v as Config, vt as environmentsSchema, w as DatasetRecord, wt as playgroundsSchema, x as ConfigsTable, xt as playgroundColumnSchema, y as ConfigVariant, yt as guardrailConfigsSchema, z as Insertable } from "./index-LRmy4sz9.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-BosemZ_J.mjs";
2
2
  import gateway from "@llmops/gateway";
3
3
  import * as kysely0 from "kysely";
4
4
  import { Kysely } from "kysely";
@@ -7,6 +7,19 @@ import { RulesLogic } from "json-logic-js";
7
7
  import z$1, { z } from "zod";
8
8
  import { BetterAuthOptions } from "better-auth";
9
9
 
10
+ //#region src/constants/headers.d.ts
11
+ /**
12
+ * Custom HTTP headers used for trace context propagation
13
+ * between SDK clients and the LLMOps gateway.
14
+ */
15
+ declare const LLMOPS_TRACE_ID_HEADER = "x-llmops-trace-id";
16
+ declare const LLMOPS_TRACE_NAME_HEADER = "x-llmops-trace-name";
17
+ declare const LLMOPS_SPAN_NAME_HEADER = "x-llmops-span-name";
18
+ declare const LLMOPS_SESSION_ID_HEADER = "x-llmops-session-id";
19
+ declare const LLMOPS_USER_ID_HEADER = "x-llmops-user-id";
20
+ declare const LLMOPS_SPAN_ID_HEADER = "x-llmops-span-id";
21
+ declare const LLMOPS_REQUEST_ID_HEADER = "x-llmops-request-id";
22
+ //#endregion
10
23
  //#region src/providers/supported-providers.d.ts
11
24
  /**
12
25
  * Supported providers - derived from @llmops/gateway providers
@@ -1969,6 +1982,10 @@ declare const insertLLMRequestSchema: z$1.ZodObject<{
1969
1982
  }>;
1970
1983
  totalLatencyMs: z$1.ZodNumber;
1971
1984
  }, z$1.core.$strip>>>;
1985
+ traceId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
1986
+ spanId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
1987
+ parentSpanId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
1988
+ sessionId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
1972
1989
  }, z$1.core.$strip>;
1973
1990
  type LLMRequestInsert = z$1.infer<typeof insertLLMRequestSchema>;
1974
1991
  /**
@@ -2069,6 +2086,10 @@ declare const createLLMRequestsDataLayer: (db: Kysely<Database>) => {
2069
2086
  action: "allowed" | "blocked" | "logged";
2070
2087
  totalLatencyMs: number;
2071
2088
  } | null;
2089
+ traceId: string | null;
2090
+ spanId: string | null;
2091
+ parentSpanId: string | null;
2092
+ sessionId: string | null;
2072
2093
  } | undefined>;
2073
2094
  /**
2074
2095
  * List LLM requests with filtering and pagination
@@ -2110,6 +2131,10 @@ declare const createLLMRequestsDataLayer: (db: Kysely<Database>) => {
2110
2131
  action: "allowed" | "blocked" | "logged";
2111
2132
  totalLatencyMs: number;
2112
2133
  } | null;
2134
+ traceId: string | null;
2135
+ spanId: string | null;
2136
+ parentSpanId: string | null;
2137
+ sessionId: string | null;
2113
2138
  }[];
2114
2139
  total: number;
2115
2140
  limit: number;
@@ -2153,6 +2178,10 @@ declare const createLLMRequestsDataLayer: (db: Kysely<Database>) => {
2153
2178
  action: "allowed" | "blocked" | "logged";
2154
2179
  totalLatencyMs: number;
2155
2180
  } | null;
2181
+ traceId: string | null;
2182
+ spanId: string | null;
2183
+ parentSpanId: string | null;
2184
+ sessionId: string | null;
2156
2185
  } | undefined>;
2157
2186
  /**
2158
2187
  * Get total cost for a date range with optional filters
@@ -3174,6 +3203,226 @@ declare const createTargetingRulesDataLayer: (db: Kysely<Database>) => {
3174
3203
  } | undefined>;
3175
3204
  };
3176
3205
  //#endregion
3206
+ //#region src/datalayer/traces.d.ts
3207
+ /**
3208
+ * Schema for upserting a trace
3209
+ */
3210
+ declare const upsertTraceSchema: z$1.ZodObject<{
3211
+ traceId: z$1.ZodString;
3212
+ name: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3213
+ sessionId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3214
+ userId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3215
+ status: z$1.ZodDefault<z$1.ZodEnum<{
3216
+ error: "error";
3217
+ unset: "unset";
3218
+ ok: "ok";
3219
+ }>>;
3220
+ startTime: z$1.ZodDate;
3221
+ endTime: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodDate>>;
3222
+ durationMs: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodNumber>>;
3223
+ spanCount: z$1.ZodDefault<z$1.ZodNumber>;
3224
+ totalInputTokens: z$1.ZodDefault<z$1.ZodNumber>;
3225
+ totalOutputTokens: z$1.ZodDefault<z$1.ZodNumber>;
3226
+ totalTokens: z$1.ZodDefault<z$1.ZodNumber>;
3227
+ totalCost: z$1.ZodDefault<z$1.ZodNumber>;
3228
+ tags: z$1.ZodDefault<z$1.ZodRecord<z$1.ZodString, z$1.ZodString>>;
3229
+ metadata: z$1.ZodDefault<z$1.ZodRecord<z$1.ZodString, z$1.ZodUnknown>>;
3230
+ }, z$1.core.$strip>;
3231
+ type TraceUpsert = z$1.infer<typeof upsertTraceSchema>;
3232
+ /**
3233
+ * Schema for inserting spans
3234
+ */
3235
+ declare const insertSpanSchema: z$1.ZodObject<{
3236
+ traceId: z$1.ZodString;
3237
+ spanId: z$1.ZodString;
3238
+ parentSpanId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3239
+ name: z$1.ZodString;
3240
+ kind: z$1.ZodDefault<z$1.ZodNumber>;
3241
+ status: z$1.ZodDefault<z$1.ZodNumber>;
3242
+ statusMessage: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3243
+ startTime: z$1.ZodDate;
3244
+ endTime: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodDate>>;
3245
+ durationMs: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodNumber>>;
3246
+ provider: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3247
+ model: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3248
+ promptTokens: z$1.ZodDefault<z$1.ZodNumber>;
3249
+ completionTokens: z$1.ZodDefault<z$1.ZodNumber>;
3250
+ totalTokens: z$1.ZodDefault<z$1.ZodNumber>;
3251
+ cost: z$1.ZodDefault<z$1.ZodNumber>;
3252
+ configId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3253
+ variantId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3254
+ environmentId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3255
+ providerConfigId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3256
+ requestId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3257
+ source: z$1.ZodDefault<z$1.ZodEnum<{
3258
+ gateway: "gateway";
3259
+ otlp: "otlp";
3260
+ }>>;
3261
+ input: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodUnknown>>;
3262
+ output: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodUnknown>>;
3263
+ attributes: z$1.ZodDefault<z$1.ZodRecord<z$1.ZodString, z$1.ZodUnknown>>;
3264
+ }, z$1.core.$strip>;
3265
+ type SpanInsert = z$1.infer<typeof insertSpanSchema>;
3266
+ /**
3267
+ * Schema for inserting span events
3268
+ */
3269
+ declare const insertSpanEventSchema: z$1.ZodObject<{
3270
+ traceId: z$1.ZodString;
3271
+ spanId: z$1.ZodString;
3272
+ name: z$1.ZodString;
3273
+ timestamp: z$1.ZodDate;
3274
+ attributes: z$1.ZodDefault<z$1.ZodRecord<z$1.ZodString, z$1.ZodUnknown>>;
3275
+ }, z$1.core.$strip>;
3276
+ type SpanEventInsert = z$1.infer<typeof insertSpanEventSchema>;
3277
+ /**
3278
+ * Schema for listing traces
3279
+ */
3280
+ declare const listTracesSchema: z$1.ZodObject<{
3281
+ limit: z$1.ZodDefault<z$1.ZodNumber>;
3282
+ offset: z$1.ZodDefault<z$1.ZodNumber>;
3283
+ sessionId: z$1.ZodOptional<z$1.ZodString>;
3284
+ userId: z$1.ZodOptional<z$1.ZodString>;
3285
+ status: z$1.ZodOptional<z$1.ZodEnum<{
3286
+ error: "error";
3287
+ unset: "unset";
3288
+ ok: "ok";
3289
+ }>>;
3290
+ name: z$1.ZodOptional<z$1.ZodString>;
3291
+ startDate: z$1.ZodOptional<z$1.ZodDate>;
3292
+ endDate: z$1.ZodOptional<z$1.ZodDate>;
3293
+ tags: z$1.ZodOptional<z$1.ZodRecord<z$1.ZodString, z$1.ZodArray<z$1.ZodString>>>;
3294
+ }, z$1.core.$strip>;
3295
+ /**
3296
+ * Schema for trace stats query
3297
+ */
3298
+ declare const traceStatsSchema: z$1.ZodObject<{
3299
+ startDate: z$1.ZodDate;
3300
+ endDate: z$1.ZodDate;
3301
+ sessionId: z$1.ZodOptional<z$1.ZodString>;
3302
+ userId: z$1.ZodOptional<z$1.ZodString>;
3303
+ }, z$1.core.$strip>;
3304
+ declare const createTracesDataLayer: (db: Kysely<Database>) => {
3305
+ /**
3306
+ * Upsert a trace — insert or update aggregates on conflict
3307
+ */
3308
+ upsertTrace: (data: TraceUpsert) => Promise<void>;
3309
+ /**
3310
+ * Batch insert spans
3311
+ */
3312
+ batchInsertSpans: (spans: SpanInsert[]) => Promise<{
3313
+ count: number;
3314
+ }>;
3315
+ /**
3316
+ * Batch insert span events
3317
+ */
3318
+ batchInsertSpanEvents: (events: SpanEventInsert[]) => Promise<{
3319
+ count: number;
3320
+ }>;
3321
+ /**
3322
+ * List traces with filtering and pagination
3323
+ */
3324
+ listTraces: (params?: z$1.infer<typeof listTracesSchema>) => Promise<{
3325
+ data: {
3326
+ name: string | null;
3327
+ id: string;
3328
+ createdAt: Date;
3329
+ updatedAt: Date;
3330
+ status: string;
3331
+ totalTokens: number;
3332
+ metadata: Record<string, unknown>;
3333
+ userId: string | null;
3334
+ tags: Record<string, string>;
3335
+ traceId: string;
3336
+ sessionId: string | null;
3337
+ startTime: Date;
3338
+ endTime: Date | null;
3339
+ durationMs: number | null;
3340
+ spanCount: number;
3341
+ totalInputTokens: number;
3342
+ totalOutputTokens: number;
3343
+ totalCost: number;
3344
+ }[];
3345
+ total: number;
3346
+ limit: number;
3347
+ offset: number;
3348
+ }>;
3349
+ /**
3350
+ * Get a single trace with all its spans and events
3351
+ */
3352
+ getTraceWithSpans: (traceId: string) => Promise<{
3353
+ trace: {
3354
+ name: string | null;
3355
+ id: string;
3356
+ createdAt: Date;
3357
+ updatedAt: Date;
3358
+ status: string;
3359
+ totalTokens: number;
3360
+ metadata: Record<string, unknown>;
3361
+ userId: string | null;
3362
+ tags: Record<string, string>;
3363
+ traceId: string;
3364
+ sessionId: string | null;
3365
+ startTime: Date;
3366
+ endTime: Date | null;
3367
+ durationMs: number | null;
3368
+ spanCount: number;
3369
+ totalInputTokens: number;
3370
+ totalOutputTokens: number;
3371
+ totalCost: number;
3372
+ };
3373
+ spans: {
3374
+ name: string;
3375
+ id: string;
3376
+ createdAt: Date;
3377
+ updatedAt: Date;
3378
+ variantId: string | null;
3379
+ provider: string | null;
3380
+ input: unknown;
3381
+ output: unknown;
3382
+ environmentId: string | null;
3383
+ configId: string | null;
3384
+ providerConfigId: string | null;
3385
+ status: number;
3386
+ promptTokens: number;
3387
+ completionTokens: number;
3388
+ totalTokens: number;
3389
+ cost: number;
3390
+ requestId: string | null;
3391
+ model: string | null;
3392
+ traceId: string;
3393
+ spanId: string;
3394
+ parentSpanId: string | null;
3395
+ startTime: Date;
3396
+ endTime: Date | null;
3397
+ durationMs: number | null;
3398
+ kind: number;
3399
+ statusMessage: string | null;
3400
+ source: string;
3401
+ attributes: Record<string, unknown>;
3402
+ }[];
3403
+ events: {
3404
+ name: string;
3405
+ id: string;
3406
+ createdAt: Date;
3407
+ traceId: string;
3408
+ spanId: string;
3409
+ attributes: Record<string, unknown>;
3410
+ timestamp: Date;
3411
+ }[];
3412
+ } | undefined>;
3413
+ /**
3414
+ * Get aggregate trace statistics for a date range
3415
+ */
3416
+ getTraceStats: (params: z$1.infer<typeof traceStatsSchema>) => Promise<{
3417
+ totalCost: number;
3418
+ totalTokens: number;
3419
+ totalTraces: number;
3420
+ avgDurationMs: number;
3421
+ errorCount: number;
3422
+ totalSpans: number;
3423
+ } | undefined>;
3424
+ };
3425
+ //#endregion
3177
3426
  //#region src/datalayer/variants.d.ts
3178
3427
  declare const createVariant: z$1.ZodObject<{
3179
3428
  name: z$1.ZodString;
@@ -3496,8 +3745,9 @@ type ProviderGuardrailOverridesDataLayer = ReturnType<typeof createProviderGuard
3496
3745
  type TargetingRulesDataLayer = ReturnType<typeof createTargetingRulesDataLayer>;
3497
3746
  type VariantsDataLayer = ReturnType<typeof createVariantDataLayer>;
3498
3747
  type VariantVersionsDataLayer = ReturnType<typeof createVariantVersionsDataLayer>;
3748
+ type TracesDataLayer = ReturnType<typeof createTracesDataLayer>;
3499
3749
  type WorkspaceSettingsDataLayer = ReturnType<typeof createWorkspaceSettingsDataLayer>;
3500
- type DataLayer = ConfigsDataLayer & ConfigVariantsDataLayer & DatasetsDataLayer & EnvironmentsDataLayer & EnvironmentSecretsDataLayer & GuardrailConfigsDataLayer & LLMRequestsDataLayer & PlaygroundsDataLayer & PlaygroundResultsDataLayer & PlaygroundRunsDataLayer & ProviderConfigsDataLayer & ProviderGuardrailOverridesDataLayer & TargetingRulesDataLayer & VariantsDataLayer & VariantVersionsDataLayer & WorkspaceSettingsDataLayer;
3750
+ type DataLayer = ConfigsDataLayer & ConfigVariantsDataLayer & DatasetsDataLayer & EnvironmentsDataLayer & EnvironmentSecretsDataLayer & GuardrailConfigsDataLayer & LLMRequestsDataLayer & PlaygroundsDataLayer & PlaygroundResultsDataLayer & PlaygroundRunsDataLayer & ProviderConfigsDataLayer & ProviderGuardrailOverridesDataLayer & TargetingRulesDataLayer & TracesDataLayer & VariantsDataLayer & VariantVersionsDataLayer & WorkspaceSettingsDataLayer;
3501
3751
  //#endregion
3502
3752
  //#region src/datalayer/create.d.ts
3503
3753
  /**
@@ -3923,4 +4173,4 @@ declare class ManifestRouter {
3923
4173
  routeWithWeights(configIdOrSlug: string, environmentId: string, context?: RoutingContext): RoutingResult | null;
3924
4174
  }
3925
4175
  //#endregion
3926
- export { type AnthropicProviderConfig, type AnyProviderConfig, AuthClientDatabaseConfig, AuthClientOptions, type AzureAIProviderConfig, type AzureOpenAIProviderConfig, BaseCacheConfig, type BaseProviderConfig, type BedrockProviderConfig, COST_SUMMARY_GROUP_BY, CacheBackend, CacheBackendType, CacheConfig, CacheEntry, CacheOptions, CacheService, CacheStats, ChatCompletionCreateParamsBase, type Config, type ConfigVariant, type ConfigVariantsDataLayer, type ConfigVariantsTable, type ConfigsDataLayer, type ConfigsTable, type CortexProviderConfig, CostResult, type CostSummaryGroupBy, DEFAULT_PROVIDER_ENV_VARS, type DataLayer, type Database, DatabaseConnection, DatabaseOptions, DatabaseType, type Dataset, type DatasetRecord, DatasetRecordsTable, type DatasetVersion, type DatasetVersionRecord, DatasetVersionRecordsTable, DatasetVersionsTable, type DatasetsDataLayer, DatasetsTable, type Environment, type EnvironmentSecret, type EnvironmentSecretsDataLayer, type EnvironmentSecretsTable, type EnvironmentsDataLayer, type EnvironmentsTable, FileCacheBackend, FileCacheConfig, type FireworksAIProviderConfig, type GatewayManifest, type GoogleProviderConfig, type GuardrailConfig, type GuardrailConfigsDataLayer, GuardrailConfigsTable, type GuardrailResult, type GuardrailResults, type HuggingFaceProviderConfig, type InlineProviderConfig, type InlineProvidersConfig, Insertable, LLMOpsClient, LLMOpsConfig, type LLMOpsConfigInput, type LLMRequest, type LLMRequestInsert, type LLMRequestsDataLayer, LLMRequestsTable, MS, ManifestBuilder, type ManifestConfig, type ManifestEnvironment, type ManifestGuardrail, type ManifestProviderGuardrailOverride, ManifestRouter, ManifestService, type ManifestTargetingRule, type ManifestVariantVersion, MemoryCacheBackend, MemoryCacheConfig, MigrationOptions, MigrationResult, type MistralAIProviderConfig, ModelPricing, ModelsDevPricingProvider, type OpenAIProviderConfig, type OracleProviderConfig, type Playground, type PlaygroundColumn, type PlaygroundResult, type PlaygroundResultsDataLayer, PlaygroundResultsTable, type PlaygroundRun, type PlaygroundRunsDataLayer, PlaygroundRunsTable, type PlaygroundsDataLayer, PlaygroundsTable, Prettify, PricingProvider, type ProviderConfig, type ProviderConfigMap, type ProviderConfigsDataLayer, ProviderConfigsTable, type ProviderGuardrailOverride, type ProviderGuardrailOverridesDataLayer, ProviderGuardrailOverridesTable, type ProvidersConfig, type RoutingContext, type RoutingResult, SCHEMA_METADATA, type SagemakerProviderConfig, Selectable, type StabilityAIProviderConfig, SupportedProviders, type TableName, type TargetingRule, type TargetingRulesDataLayer, type TargetingRulesTable, Updateable, UsageData, type ValidatedLLMOpsConfig, type Variant, VariantJsonData, type VariantVersion, type VariantVersionsDataLayer, VariantVersionsTable, type VariantsDataLayer, type VariantsTable, type VertexAIProviderConfig, type WorkersAIProviderConfig, type WorkspaceSettings, type WorkspaceSettingsDataLayer, WorkspaceSettingsTable, calculateCost, chatCompletionCreateParamsBaseSchema, configVariantsSchema, configsSchema, createConfigDataLayer, createConfigVariantDataLayer, createDataLayer, createDatabase, createDatabaseFromConnection, createDatasetsDataLayer, createEnvironmentDataLayer, createEnvironmentSecretDataLayer, createGuardrailConfigsDataLayer, createLLMRequestsDataLayer, createNeonDialect, createPlaygroundDataLayer, createPlaygroundResultsDataLayer, createPlaygroundRunsDataLayer, createProviderConfigsDataLayer, createProviderGuardrailOverridesDataLayer, createTargetingRulesDataLayer, createVariantDataLayer, createVariantVersionsDataLayer, createWorkspaceSettingsDataLayer, datasetRecordsSchema, datasetVersionRecordsSchema, datasetVersionsSchema, datasetsSchema, detectDatabaseType, dollarsToMicroDollars, environmentSecretsSchema, environmentsSchema, executeWithSchema, formatCost, gateway, generateId, getAuthClientOptions, getDefaultPricingProvider, getDefaultProviders, getMigrations, guardrailConfigsSchema, llmRequestsSchema, llmopsConfigSchema, logger, matchType, mergeWithDefaultProviders, microDollarsToDollars, parsePartialTableData, parseTableData, playgroundColumnSchema, playgroundResultsSchema, playgroundRunsSchema, playgroundsSchema, providerConfigsSchema, providerGuardrailOverridesSchema, runAutoMigrations, schemas, targetingRulesSchema, validateLLMOpsConfig, validatePartialTableData, validateTableData, variantJsonDataSchema, variantVersionsSchema, variantsSchema, workspaceSettingsSchema };
4176
+ export { type AnthropicProviderConfig, type AnyProviderConfig, AuthClientDatabaseConfig, AuthClientOptions, type AzureAIProviderConfig, type AzureOpenAIProviderConfig, BaseCacheConfig, type BaseProviderConfig, type BedrockProviderConfig, COST_SUMMARY_GROUP_BY, CacheBackend, CacheBackendType, CacheConfig, CacheEntry, CacheOptions, CacheService, CacheStats, ChatCompletionCreateParamsBase, type Config, type ConfigVariant, type ConfigVariantsDataLayer, type ConfigVariantsTable, type ConfigsDataLayer, type ConfigsTable, type CortexProviderConfig, CostResult, type CostSummaryGroupBy, DEFAULT_PROVIDER_ENV_VARS, type DataLayer, type Database, DatabaseConnection, DatabaseOptions, DatabaseType, type Dataset, type DatasetRecord, DatasetRecordsTable, type DatasetVersion, type DatasetVersionRecord, DatasetVersionRecordsTable, DatasetVersionsTable, type DatasetsDataLayer, DatasetsTable, type Environment, type EnvironmentSecret, type EnvironmentSecretsDataLayer, type EnvironmentSecretsTable, type EnvironmentsDataLayer, type EnvironmentsTable, FileCacheBackend, FileCacheConfig, type FireworksAIProviderConfig, type GatewayManifest, type GoogleProviderConfig, type GuardrailConfig, type GuardrailConfigsDataLayer, GuardrailConfigsTable, type GuardrailResult, type GuardrailResults, type HuggingFaceProviderConfig, type InlineProviderConfig, type InlineProvidersConfig, Insertable, 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, type LLMRequest, type LLMRequestInsert, type LLMRequestsDataLayer, LLMRequestsTable, MS, ManifestBuilder, type ManifestConfig, type ManifestEnvironment, type ManifestGuardrail, type ManifestProviderGuardrailOverride, ManifestRouter, ManifestService, type ManifestTargetingRule, type ManifestVariantVersion, MemoryCacheBackend, MemoryCacheConfig, MigrationOptions, MigrationResult, type MistralAIProviderConfig, ModelPricing, ModelsDevPricingProvider, type OpenAIProviderConfig, type OracleProviderConfig, type Playground, type PlaygroundColumn, type PlaygroundResult, type PlaygroundResultsDataLayer, PlaygroundResultsTable, type PlaygroundRun, type PlaygroundRunsDataLayer, PlaygroundRunsTable, type PlaygroundsDataLayer, PlaygroundsTable, Prettify, PricingProvider, type ProviderConfig, type ProviderConfigMap, type ProviderConfigsDataLayer, ProviderConfigsTable, type ProviderGuardrailOverride, type ProviderGuardrailOverridesDataLayer, ProviderGuardrailOverridesTable, type ProvidersConfig, type RoutingContext, type RoutingResult, SCHEMA_METADATA, type SagemakerProviderConfig, Selectable, Span, SpanEvent, type SpanEventInsert, SpanEventsTable, type SpanInsert, SpansTable, type StabilityAIProviderConfig, SupportedProviders, type TableName, type TargetingRule, type TargetingRulesDataLayer, type TargetingRulesTable, Trace, type TraceUpsert, type TracesDataLayer, TracesTable, Updateable, UsageData, type ValidatedLLMOpsConfig, type Variant, VariantJsonData, type VariantVersion, type VariantVersionsDataLayer, VariantVersionsTable, type VariantsDataLayer, type VariantsTable, type VertexAIProviderConfig, type WorkersAIProviderConfig, type WorkspaceSettings, type WorkspaceSettingsDataLayer, WorkspaceSettingsTable, calculateCost, chatCompletionCreateParamsBaseSchema, configVariantsSchema, configsSchema, createConfigDataLayer, createConfigVariantDataLayer, createDataLayer, createDatabase, createDatabaseFromConnection, createDatasetsDataLayer, createEnvironmentDataLayer, createEnvironmentSecretDataLayer, createGuardrailConfigsDataLayer, createLLMRequestsDataLayer, createNeonDialect, createPlaygroundDataLayer, createPlaygroundResultsDataLayer, createPlaygroundRunsDataLayer, createProviderConfigsDataLayer, createProviderGuardrailOverridesDataLayer, createTargetingRulesDataLayer, createTracesDataLayer, createVariantDataLayer, createVariantVersionsDataLayer, createWorkspaceSettingsDataLayer, datasetRecordsSchema, datasetVersionRecordsSchema, datasetVersionsSchema, datasetsSchema, detectDatabaseType, dollarsToMicroDollars, environmentSecretsSchema, environmentsSchema, executeWithSchema, formatCost, gateway, generateId, getAuthClientOptions, 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 };