@llmops/core 0.5.4 → 0.6.1-beta.1

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,20 @@ 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
+ declare const LLMOPS_INTERNAL_HEADER = "x-llmops-internal";
23
+ //#endregion
10
24
  //#region src/providers/supported-providers.d.ts
11
25
  /**
12
26
  * Supported providers - derived from @llmops/gateway providers
@@ -1969,6 +1983,10 @@ declare const insertLLMRequestSchema: z$1.ZodObject<{
1969
1983
  }>;
1970
1984
  totalLatencyMs: z$1.ZodNumber;
1971
1985
  }, z$1.core.$strip>>>;
1986
+ traceId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
1987
+ spanId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
1988
+ parentSpanId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
1989
+ sessionId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
1972
1990
  }, z$1.core.$strip>;
1973
1991
  type LLMRequestInsert = z$1.infer<typeof insertLLMRequestSchema>;
1974
1992
  /**
@@ -2069,6 +2087,10 @@ declare const createLLMRequestsDataLayer: (db: Kysely<Database>) => {
2069
2087
  action: "allowed" | "blocked" | "logged";
2070
2088
  totalLatencyMs: number;
2071
2089
  } | null;
2090
+ traceId: string | null;
2091
+ spanId: string | null;
2092
+ parentSpanId: string | null;
2093
+ sessionId: string | null;
2072
2094
  } | undefined>;
2073
2095
  /**
2074
2096
  * List LLM requests with filtering and pagination
@@ -2110,6 +2132,10 @@ declare const createLLMRequestsDataLayer: (db: Kysely<Database>) => {
2110
2132
  action: "allowed" | "blocked" | "logged";
2111
2133
  totalLatencyMs: number;
2112
2134
  } | null;
2135
+ traceId: string | null;
2136
+ spanId: string | null;
2137
+ parentSpanId: string | null;
2138
+ sessionId: string | null;
2113
2139
  }[];
2114
2140
  total: number;
2115
2141
  limit: number;
@@ -2153,6 +2179,10 @@ declare const createLLMRequestsDataLayer: (db: Kysely<Database>) => {
2153
2179
  action: "allowed" | "blocked" | "logged";
2154
2180
  totalLatencyMs: number;
2155
2181
  } | null;
2182
+ traceId: string | null;
2183
+ spanId: string | null;
2184
+ parentSpanId: string | null;
2185
+ sessionId: string | null;
2156
2186
  } | undefined>;
2157
2187
  /**
2158
2188
  * Get total cost for a date range with optional filters
@@ -3174,6 +3204,226 @@ declare const createTargetingRulesDataLayer: (db: Kysely<Database>) => {
3174
3204
  } | undefined>;
3175
3205
  };
3176
3206
  //#endregion
3207
+ //#region src/datalayer/traces.d.ts
3208
+ /**
3209
+ * Schema for upserting a trace
3210
+ */
3211
+ declare const upsertTraceSchema: z$1.ZodObject<{
3212
+ traceId: z$1.ZodString;
3213
+ name: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3214
+ sessionId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3215
+ userId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3216
+ status: z$1.ZodDefault<z$1.ZodEnum<{
3217
+ error: "error";
3218
+ unset: "unset";
3219
+ ok: "ok";
3220
+ }>>;
3221
+ startTime: z$1.ZodDate;
3222
+ endTime: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodDate>>;
3223
+ durationMs: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodNumber>>;
3224
+ spanCount: z$1.ZodDefault<z$1.ZodNumber>;
3225
+ totalInputTokens: z$1.ZodDefault<z$1.ZodNumber>;
3226
+ totalOutputTokens: z$1.ZodDefault<z$1.ZodNumber>;
3227
+ totalTokens: z$1.ZodDefault<z$1.ZodNumber>;
3228
+ totalCost: z$1.ZodDefault<z$1.ZodNumber>;
3229
+ tags: z$1.ZodDefault<z$1.ZodRecord<z$1.ZodString, z$1.ZodString>>;
3230
+ metadata: z$1.ZodDefault<z$1.ZodRecord<z$1.ZodString, z$1.ZodUnknown>>;
3231
+ }, z$1.core.$strip>;
3232
+ type TraceUpsert = z$1.infer<typeof upsertTraceSchema>;
3233
+ /**
3234
+ * Schema for inserting spans
3235
+ */
3236
+ declare const insertSpanSchema: z$1.ZodObject<{
3237
+ traceId: z$1.ZodString;
3238
+ spanId: z$1.ZodString;
3239
+ parentSpanId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3240
+ name: z$1.ZodString;
3241
+ kind: z$1.ZodDefault<z$1.ZodNumber>;
3242
+ status: z$1.ZodDefault<z$1.ZodNumber>;
3243
+ statusMessage: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3244
+ startTime: z$1.ZodDate;
3245
+ endTime: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodDate>>;
3246
+ durationMs: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodNumber>>;
3247
+ provider: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3248
+ model: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3249
+ promptTokens: z$1.ZodDefault<z$1.ZodNumber>;
3250
+ completionTokens: z$1.ZodDefault<z$1.ZodNumber>;
3251
+ totalTokens: z$1.ZodDefault<z$1.ZodNumber>;
3252
+ cost: z$1.ZodDefault<z$1.ZodNumber>;
3253
+ configId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3254
+ variantId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3255
+ environmentId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3256
+ providerConfigId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3257
+ requestId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3258
+ source: z$1.ZodDefault<z$1.ZodEnum<{
3259
+ gateway: "gateway";
3260
+ otlp: "otlp";
3261
+ }>>;
3262
+ input: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodUnknown>>;
3263
+ output: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodUnknown>>;
3264
+ attributes: z$1.ZodDefault<z$1.ZodRecord<z$1.ZodString, z$1.ZodUnknown>>;
3265
+ }, z$1.core.$strip>;
3266
+ type SpanInsert = z$1.infer<typeof insertSpanSchema>;
3267
+ /**
3268
+ * Schema for inserting span events
3269
+ */
3270
+ declare const insertSpanEventSchema: z$1.ZodObject<{
3271
+ traceId: z$1.ZodString;
3272
+ spanId: z$1.ZodString;
3273
+ name: z$1.ZodString;
3274
+ timestamp: z$1.ZodDate;
3275
+ attributes: z$1.ZodDefault<z$1.ZodRecord<z$1.ZodString, z$1.ZodUnknown>>;
3276
+ }, z$1.core.$strip>;
3277
+ type SpanEventInsert = z$1.infer<typeof insertSpanEventSchema>;
3278
+ /**
3279
+ * Schema for listing traces
3280
+ */
3281
+ declare const listTracesSchema: z$1.ZodObject<{
3282
+ limit: z$1.ZodDefault<z$1.ZodNumber>;
3283
+ offset: z$1.ZodDefault<z$1.ZodNumber>;
3284
+ sessionId: z$1.ZodOptional<z$1.ZodString>;
3285
+ userId: z$1.ZodOptional<z$1.ZodString>;
3286
+ status: z$1.ZodOptional<z$1.ZodEnum<{
3287
+ error: "error";
3288
+ unset: "unset";
3289
+ ok: "ok";
3290
+ }>>;
3291
+ name: z$1.ZodOptional<z$1.ZodString>;
3292
+ startDate: z$1.ZodOptional<z$1.ZodDate>;
3293
+ endDate: z$1.ZodOptional<z$1.ZodDate>;
3294
+ tags: z$1.ZodOptional<z$1.ZodRecord<z$1.ZodString, z$1.ZodArray<z$1.ZodString>>>;
3295
+ }, z$1.core.$strip>;
3296
+ /**
3297
+ * Schema for trace stats query
3298
+ */
3299
+ declare const traceStatsSchema: z$1.ZodObject<{
3300
+ startDate: z$1.ZodDate;
3301
+ endDate: z$1.ZodDate;
3302
+ sessionId: z$1.ZodOptional<z$1.ZodString>;
3303
+ userId: z$1.ZodOptional<z$1.ZodString>;
3304
+ }, z$1.core.$strip>;
3305
+ declare const createTracesDataLayer: (db: Kysely<Database>) => {
3306
+ /**
3307
+ * Upsert a trace — insert or update aggregates on conflict
3308
+ */
3309
+ upsertTrace: (data: TraceUpsert) => Promise<void>;
3310
+ /**
3311
+ * Batch insert spans
3312
+ */
3313
+ batchInsertSpans: (spans: SpanInsert[]) => Promise<{
3314
+ count: number;
3315
+ }>;
3316
+ /**
3317
+ * Batch insert span events
3318
+ */
3319
+ batchInsertSpanEvents: (events: SpanEventInsert[]) => Promise<{
3320
+ count: number;
3321
+ }>;
3322
+ /**
3323
+ * List traces with filtering and pagination
3324
+ */
3325
+ listTraces: (params?: z$1.infer<typeof listTracesSchema>) => Promise<{
3326
+ data: {
3327
+ name: string | null;
3328
+ id: string;
3329
+ createdAt: Date;
3330
+ updatedAt: Date;
3331
+ status: string;
3332
+ totalTokens: number;
3333
+ metadata: Record<string, unknown>;
3334
+ userId: string | null;
3335
+ tags: Record<string, string>;
3336
+ traceId: string;
3337
+ sessionId: string | null;
3338
+ startTime: Date;
3339
+ endTime: Date | null;
3340
+ durationMs: number | null;
3341
+ spanCount: number;
3342
+ totalInputTokens: number;
3343
+ totalOutputTokens: number;
3344
+ totalCost: number;
3345
+ }[];
3346
+ total: number;
3347
+ limit: number;
3348
+ offset: number;
3349
+ }>;
3350
+ /**
3351
+ * Get a single trace with all its spans and events
3352
+ */
3353
+ getTraceWithSpans: (traceId: string) => Promise<{
3354
+ trace: {
3355
+ name: string | null;
3356
+ id: string;
3357
+ createdAt: Date;
3358
+ updatedAt: Date;
3359
+ status: string;
3360
+ totalTokens: number;
3361
+ metadata: Record<string, unknown>;
3362
+ userId: string | null;
3363
+ tags: Record<string, string>;
3364
+ traceId: string;
3365
+ sessionId: string | null;
3366
+ startTime: Date;
3367
+ endTime: Date | null;
3368
+ durationMs: number | null;
3369
+ spanCount: number;
3370
+ totalInputTokens: number;
3371
+ totalOutputTokens: number;
3372
+ totalCost: number;
3373
+ };
3374
+ spans: {
3375
+ name: string;
3376
+ id: string;
3377
+ createdAt: Date;
3378
+ updatedAt: Date;
3379
+ variantId: string | null;
3380
+ provider: string | null;
3381
+ input: unknown;
3382
+ output: unknown;
3383
+ environmentId: string | null;
3384
+ configId: string | null;
3385
+ providerConfigId: string | null;
3386
+ status: number;
3387
+ promptTokens: number;
3388
+ completionTokens: number;
3389
+ totalTokens: number;
3390
+ cost: number;
3391
+ requestId: string | null;
3392
+ model: string | null;
3393
+ traceId: string;
3394
+ spanId: string;
3395
+ parentSpanId: string | null;
3396
+ startTime: Date;
3397
+ endTime: Date | null;
3398
+ durationMs: number | null;
3399
+ kind: number;
3400
+ statusMessage: string | null;
3401
+ source: string;
3402
+ attributes: Record<string, unknown>;
3403
+ }[];
3404
+ events: {
3405
+ name: string;
3406
+ id: string;
3407
+ createdAt: Date;
3408
+ traceId: string;
3409
+ spanId: string;
3410
+ attributes: Record<string, unknown>;
3411
+ timestamp: Date;
3412
+ }[];
3413
+ } | undefined>;
3414
+ /**
3415
+ * Get aggregate trace statistics for a date range
3416
+ */
3417
+ getTraceStats: (params: z$1.infer<typeof traceStatsSchema>) => Promise<{
3418
+ totalCost: number;
3419
+ totalTokens: number;
3420
+ totalTraces: number;
3421
+ avgDurationMs: number;
3422
+ errorCount: number;
3423
+ totalSpans: number;
3424
+ } | undefined>;
3425
+ };
3426
+ //#endregion
3177
3427
  //#region src/datalayer/variants.d.ts
3178
3428
  declare const createVariant: z$1.ZodObject<{
3179
3429
  name: z$1.ZodString;
@@ -3496,8 +3746,9 @@ type ProviderGuardrailOverridesDataLayer = ReturnType<typeof createProviderGuard
3496
3746
  type TargetingRulesDataLayer = ReturnType<typeof createTargetingRulesDataLayer>;
3497
3747
  type VariantsDataLayer = ReturnType<typeof createVariantDataLayer>;
3498
3748
  type VariantVersionsDataLayer = ReturnType<typeof createVariantVersionsDataLayer>;
3749
+ type TracesDataLayer = ReturnType<typeof createTracesDataLayer>;
3499
3750
  type WorkspaceSettingsDataLayer = ReturnType<typeof createWorkspaceSettingsDataLayer>;
3500
- type DataLayer = ConfigsDataLayer & ConfigVariantsDataLayer & DatasetsDataLayer & EnvironmentsDataLayer & EnvironmentSecretsDataLayer & GuardrailConfigsDataLayer & LLMRequestsDataLayer & PlaygroundsDataLayer & PlaygroundResultsDataLayer & PlaygroundRunsDataLayer & ProviderConfigsDataLayer & ProviderGuardrailOverridesDataLayer & TargetingRulesDataLayer & VariantsDataLayer & VariantVersionsDataLayer & WorkspaceSettingsDataLayer;
3751
+ type DataLayer = ConfigsDataLayer & ConfigVariantsDataLayer & DatasetsDataLayer & EnvironmentsDataLayer & EnvironmentSecretsDataLayer & GuardrailConfigsDataLayer & LLMRequestsDataLayer & PlaygroundsDataLayer & PlaygroundResultsDataLayer & PlaygroundRunsDataLayer & ProviderConfigsDataLayer & ProviderGuardrailOverridesDataLayer & TargetingRulesDataLayer & TracesDataLayer & VariantsDataLayer & VariantVersionsDataLayer & WorkspaceSettingsDataLayer;
3501
3752
  //#endregion
3502
3753
  //#region src/datalayer/create.d.ts
3503
3754
  /**
@@ -3923,4 +4174,4 @@ declare class ManifestRouter {
3923
4174
  routeWithWeights(configIdOrSlug: string, environmentId: string, context?: RoutingContext): RoutingResult | null;
3924
4175
  }
3925
4176
  //#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 };
4177
+ 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_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, 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,20 @@ 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
+ declare const LLMOPS_INTERNAL_HEADER = "x-llmops-internal";
23
+ //#endregion
10
24
  //#region src/providers/supported-providers.d.ts
11
25
  /**
12
26
  * Supported providers - derived from @llmops/gateway providers
@@ -1969,6 +1983,10 @@ declare const insertLLMRequestSchema: z$1.ZodObject<{
1969
1983
  }>;
1970
1984
  totalLatencyMs: z$1.ZodNumber;
1971
1985
  }, z$1.core.$strip>>>;
1986
+ traceId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
1987
+ spanId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
1988
+ parentSpanId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
1989
+ sessionId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
1972
1990
  }, z$1.core.$strip>;
1973
1991
  type LLMRequestInsert = z$1.infer<typeof insertLLMRequestSchema>;
1974
1992
  /**
@@ -2069,6 +2087,10 @@ declare const createLLMRequestsDataLayer: (db: Kysely<Database>) => {
2069
2087
  action: "allowed" | "blocked" | "logged";
2070
2088
  totalLatencyMs: number;
2071
2089
  } | null;
2090
+ traceId: string | null;
2091
+ spanId: string | null;
2092
+ parentSpanId: string | null;
2093
+ sessionId: string | null;
2072
2094
  } | undefined>;
2073
2095
  /**
2074
2096
  * List LLM requests with filtering and pagination
@@ -2110,6 +2132,10 @@ declare const createLLMRequestsDataLayer: (db: Kysely<Database>) => {
2110
2132
  action: "allowed" | "blocked" | "logged";
2111
2133
  totalLatencyMs: number;
2112
2134
  } | null;
2135
+ traceId: string | null;
2136
+ spanId: string | null;
2137
+ parentSpanId: string | null;
2138
+ sessionId: string | null;
2113
2139
  }[];
2114
2140
  total: number;
2115
2141
  limit: number;
@@ -2153,6 +2179,10 @@ declare const createLLMRequestsDataLayer: (db: Kysely<Database>) => {
2153
2179
  action: "allowed" | "blocked" | "logged";
2154
2180
  totalLatencyMs: number;
2155
2181
  } | null;
2182
+ traceId: string | null;
2183
+ spanId: string | null;
2184
+ parentSpanId: string | null;
2185
+ sessionId: string | null;
2156
2186
  } | undefined>;
2157
2187
  /**
2158
2188
  * Get total cost for a date range with optional filters
@@ -3174,6 +3204,226 @@ declare const createTargetingRulesDataLayer: (db: Kysely<Database>) => {
3174
3204
  } | undefined>;
3175
3205
  };
3176
3206
  //#endregion
3207
+ //#region src/datalayer/traces.d.ts
3208
+ /**
3209
+ * Schema for upserting a trace
3210
+ */
3211
+ declare const upsertTraceSchema: z$1.ZodObject<{
3212
+ traceId: z$1.ZodString;
3213
+ name: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3214
+ sessionId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3215
+ userId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3216
+ status: z$1.ZodDefault<z$1.ZodEnum<{
3217
+ error: "error";
3218
+ unset: "unset";
3219
+ ok: "ok";
3220
+ }>>;
3221
+ startTime: z$1.ZodDate;
3222
+ endTime: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodDate>>;
3223
+ durationMs: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodNumber>>;
3224
+ spanCount: z$1.ZodDefault<z$1.ZodNumber>;
3225
+ totalInputTokens: z$1.ZodDefault<z$1.ZodNumber>;
3226
+ totalOutputTokens: z$1.ZodDefault<z$1.ZodNumber>;
3227
+ totalTokens: z$1.ZodDefault<z$1.ZodNumber>;
3228
+ totalCost: z$1.ZodDefault<z$1.ZodNumber>;
3229
+ tags: z$1.ZodDefault<z$1.ZodRecord<z$1.ZodString, z$1.ZodString>>;
3230
+ metadata: z$1.ZodDefault<z$1.ZodRecord<z$1.ZodString, z$1.ZodUnknown>>;
3231
+ }, z$1.core.$strip>;
3232
+ type TraceUpsert = z$1.infer<typeof upsertTraceSchema>;
3233
+ /**
3234
+ * Schema for inserting spans
3235
+ */
3236
+ declare const insertSpanSchema: z$1.ZodObject<{
3237
+ traceId: z$1.ZodString;
3238
+ spanId: z$1.ZodString;
3239
+ parentSpanId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3240
+ name: z$1.ZodString;
3241
+ kind: z$1.ZodDefault<z$1.ZodNumber>;
3242
+ status: z$1.ZodDefault<z$1.ZodNumber>;
3243
+ statusMessage: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3244
+ startTime: z$1.ZodDate;
3245
+ endTime: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodDate>>;
3246
+ durationMs: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodNumber>>;
3247
+ provider: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3248
+ model: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3249
+ promptTokens: z$1.ZodDefault<z$1.ZodNumber>;
3250
+ completionTokens: z$1.ZodDefault<z$1.ZodNumber>;
3251
+ totalTokens: z$1.ZodDefault<z$1.ZodNumber>;
3252
+ cost: z$1.ZodDefault<z$1.ZodNumber>;
3253
+ configId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3254
+ variantId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3255
+ environmentId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3256
+ providerConfigId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3257
+ requestId: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodString>>;
3258
+ source: z$1.ZodDefault<z$1.ZodEnum<{
3259
+ gateway: "gateway";
3260
+ otlp: "otlp";
3261
+ }>>;
3262
+ input: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodUnknown>>;
3263
+ output: z$1.ZodOptional<z$1.ZodNullable<z$1.ZodUnknown>>;
3264
+ attributes: z$1.ZodDefault<z$1.ZodRecord<z$1.ZodString, z$1.ZodUnknown>>;
3265
+ }, z$1.core.$strip>;
3266
+ type SpanInsert = z$1.infer<typeof insertSpanSchema>;
3267
+ /**
3268
+ * Schema for inserting span events
3269
+ */
3270
+ declare const insertSpanEventSchema: z$1.ZodObject<{
3271
+ traceId: z$1.ZodString;
3272
+ spanId: z$1.ZodString;
3273
+ name: z$1.ZodString;
3274
+ timestamp: z$1.ZodDate;
3275
+ attributes: z$1.ZodDefault<z$1.ZodRecord<z$1.ZodString, z$1.ZodUnknown>>;
3276
+ }, z$1.core.$strip>;
3277
+ type SpanEventInsert = z$1.infer<typeof insertSpanEventSchema>;
3278
+ /**
3279
+ * Schema for listing traces
3280
+ */
3281
+ declare const listTracesSchema: z$1.ZodObject<{
3282
+ limit: z$1.ZodDefault<z$1.ZodNumber>;
3283
+ offset: z$1.ZodDefault<z$1.ZodNumber>;
3284
+ sessionId: z$1.ZodOptional<z$1.ZodString>;
3285
+ userId: z$1.ZodOptional<z$1.ZodString>;
3286
+ status: z$1.ZodOptional<z$1.ZodEnum<{
3287
+ error: "error";
3288
+ unset: "unset";
3289
+ ok: "ok";
3290
+ }>>;
3291
+ name: z$1.ZodOptional<z$1.ZodString>;
3292
+ startDate: z$1.ZodOptional<z$1.ZodDate>;
3293
+ endDate: z$1.ZodOptional<z$1.ZodDate>;
3294
+ tags: z$1.ZodOptional<z$1.ZodRecord<z$1.ZodString, z$1.ZodArray<z$1.ZodString>>>;
3295
+ }, z$1.core.$strip>;
3296
+ /**
3297
+ * Schema for trace stats query
3298
+ */
3299
+ declare const traceStatsSchema: z$1.ZodObject<{
3300
+ startDate: z$1.ZodDate;
3301
+ endDate: z$1.ZodDate;
3302
+ sessionId: z$1.ZodOptional<z$1.ZodString>;
3303
+ userId: z$1.ZodOptional<z$1.ZodString>;
3304
+ }, z$1.core.$strip>;
3305
+ declare const createTracesDataLayer: (db: Kysely<Database>) => {
3306
+ /**
3307
+ * Upsert a trace — insert or update aggregates on conflict
3308
+ */
3309
+ upsertTrace: (data: TraceUpsert) => Promise<void>;
3310
+ /**
3311
+ * Batch insert spans
3312
+ */
3313
+ batchInsertSpans: (spans: SpanInsert[]) => Promise<{
3314
+ count: number;
3315
+ }>;
3316
+ /**
3317
+ * Batch insert span events
3318
+ */
3319
+ batchInsertSpanEvents: (events: SpanEventInsert[]) => Promise<{
3320
+ count: number;
3321
+ }>;
3322
+ /**
3323
+ * List traces with filtering and pagination
3324
+ */
3325
+ listTraces: (params?: z$1.infer<typeof listTracesSchema>) => Promise<{
3326
+ data: {
3327
+ name: string | null;
3328
+ id: string;
3329
+ createdAt: Date;
3330
+ updatedAt: Date;
3331
+ status: string;
3332
+ totalTokens: number;
3333
+ metadata: Record<string, unknown>;
3334
+ userId: string | null;
3335
+ tags: Record<string, string>;
3336
+ traceId: string;
3337
+ sessionId: string | null;
3338
+ startTime: Date;
3339
+ endTime: Date | null;
3340
+ durationMs: number | null;
3341
+ spanCount: number;
3342
+ totalInputTokens: number;
3343
+ totalOutputTokens: number;
3344
+ totalCost: number;
3345
+ }[];
3346
+ total: number;
3347
+ limit: number;
3348
+ offset: number;
3349
+ }>;
3350
+ /**
3351
+ * Get a single trace with all its spans and events
3352
+ */
3353
+ getTraceWithSpans: (traceId: string) => Promise<{
3354
+ trace: {
3355
+ name: string | null;
3356
+ id: string;
3357
+ createdAt: Date;
3358
+ updatedAt: Date;
3359
+ status: string;
3360
+ totalTokens: number;
3361
+ metadata: Record<string, unknown>;
3362
+ userId: string | null;
3363
+ tags: Record<string, string>;
3364
+ traceId: string;
3365
+ sessionId: string | null;
3366
+ startTime: Date;
3367
+ endTime: Date | null;
3368
+ durationMs: number | null;
3369
+ spanCount: number;
3370
+ totalInputTokens: number;
3371
+ totalOutputTokens: number;
3372
+ totalCost: number;
3373
+ };
3374
+ spans: {
3375
+ name: string;
3376
+ id: string;
3377
+ createdAt: Date;
3378
+ updatedAt: Date;
3379
+ variantId: string | null;
3380
+ provider: string | null;
3381
+ input: unknown;
3382
+ output: unknown;
3383
+ environmentId: string | null;
3384
+ configId: string | null;
3385
+ providerConfigId: string | null;
3386
+ status: number;
3387
+ promptTokens: number;
3388
+ completionTokens: number;
3389
+ totalTokens: number;
3390
+ cost: number;
3391
+ requestId: string | null;
3392
+ model: string | null;
3393
+ traceId: string;
3394
+ spanId: string;
3395
+ parentSpanId: string | null;
3396
+ startTime: Date;
3397
+ endTime: Date | null;
3398
+ durationMs: number | null;
3399
+ kind: number;
3400
+ statusMessage: string | null;
3401
+ source: string;
3402
+ attributes: Record<string, unknown>;
3403
+ }[];
3404
+ events: {
3405
+ name: string;
3406
+ id: string;
3407
+ createdAt: Date;
3408
+ traceId: string;
3409
+ spanId: string;
3410
+ attributes: Record<string, unknown>;
3411
+ timestamp: Date;
3412
+ }[];
3413
+ } | undefined>;
3414
+ /**
3415
+ * Get aggregate trace statistics for a date range
3416
+ */
3417
+ getTraceStats: (params: z$1.infer<typeof traceStatsSchema>) => Promise<{
3418
+ totalCost: number;
3419
+ totalTokens: number;
3420
+ totalTraces: number;
3421
+ avgDurationMs: number;
3422
+ errorCount: number;
3423
+ totalSpans: number;
3424
+ } | undefined>;
3425
+ };
3426
+ //#endregion
3177
3427
  //#region src/datalayer/variants.d.ts
3178
3428
  declare const createVariant: z$1.ZodObject<{
3179
3429
  name: z$1.ZodString;
@@ -3496,8 +3746,9 @@ type ProviderGuardrailOverridesDataLayer = ReturnType<typeof createProviderGuard
3496
3746
  type TargetingRulesDataLayer = ReturnType<typeof createTargetingRulesDataLayer>;
3497
3747
  type VariantsDataLayer = ReturnType<typeof createVariantDataLayer>;
3498
3748
  type VariantVersionsDataLayer = ReturnType<typeof createVariantVersionsDataLayer>;
3749
+ type TracesDataLayer = ReturnType<typeof createTracesDataLayer>;
3499
3750
  type WorkspaceSettingsDataLayer = ReturnType<typeof createWorkspaceSettingsDataLayer>;
3500
- type DataLayer = ConfigsDataLayer & ConfigVariantsDataLayer & DatasetsDataLayer & EnvironmentsDataLayer & EnvironmentSecretsDataLayer & GuardrailConfigsDataLayer & LLMRequestsDataLayer & PlaygroundsDataLayer & PlaygroundResultsDataLayer & PlaygroundRunsDataLayer & ProviderConfigsDataLayer & ProviderGuardrailOverridesDataLayer & TargetingRulesDataLayer & VariantsDataLayer & VariantVersionsDataLayer & WorkspaceSettingsDataLayer;
3751
+ type DataLayer = ConfigsDataLayer & ConfigVariantsDataLayer & DatasetsDataLayer & EnvironmentsDataLayer & EnvironmentSecretsDataLayer & GuardrailConfigsDataLayer & LLMRequestsDataLayer & PlaygroundsDataLayer & PlaygroundResultsDataLayer & PlaygroundRunsDataLayer & ProviderConfigsDataLayer & ProviderGuardrailOverridesDataLayer & TargetingRulesDataLayer & TracesDataLayer & VariantsDataLayer & VariantVersionsDataLayer & WorkspaceSettingsDataLayer;
3501
3752
  //#endregion
3502
3753
  //#region src/datalayer/create.d.ts
3503
3754
  /**
@@ -3923,4 +4174,4 @@ declare class ManifestRouter {
3923
4174
  routeWithWeights(configIdOrSlug: string, environmentId: string, context?: RoutingContext): RoutingResult | null;
3924
4175
  }
3925
4176
  //#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 };
4177
+ 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_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, 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 };