@llmops/core 0.5.2-beta.1 → 0.5.3
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.cjs +43 -7
- package/dist/index.d.cts +8 -1
- package/dist/index.d.mts +8 -1
- package/dist/index.mjs +43 -8
- package/package.json +2 -2
package/dist/index.cjs
CHANGED
|
@@ -2028,6 +2028,18 @@ const dateRangeSchema = require_db.zod_default.object({
|
|
|
2028
2028
|
tags: require_db.zod_default.record(require_db.zod_default.string(), require_db.zod_default.array(require_db.zod_default.string())).optional()
|
|
2029
2029
|
});
|
|
2030
2030
|
/**
|
|
2031
|
+
* Valid groupBy values for cost summary queries
|
|
2032
|
+
*/
|
|
2033
|
+
const COST_SUMMARY_GROUP_BY = [
|
|
2034
|
+
"day",
|
|
2035
|
+
"hour",
|
|
2036
|
+
"model",
|
|
2037
|
+
"provider",
|
|
2038
|
+
"config",
|
|
2039
|
+
"endpoint",
|
|
2040
|
+
"tags"
|
|
2041
|
+
];
|
|
2042
|
+
/**
|
|
2031
2043
|
* Schema for cost summary with grouping
|
|
2032
2044
|
*/
|
|
2033
2045
|
const costSummarySchema = require_db.zod_default.object({
|
|
@@ -2037,13 +2049,7 @@ const costSummarySchema = require_db.zod_default.object({
|
|
|
2037
2049
|
variantId: require_db.zod_default.string().uuid().optional(),
|
|
2038
2050
|
environmentId: require_db.zod_default.string().uuid().optional(),
|
|
2039
2051
|
tags: require_db.zod_default.record(require_db.zod_default.string(), require_db.zod_default.array(require_db.zod_default.string())).optional(),
|
|
2040
|
-
groupBy: require_db.zod_default.enum(
|
|
2041
|
-
"day",
|
|
2042
|
-
"hour",
|
|
2043
|
-
"model",
|
|
2044
|
-
"provider",
|
|
2045
|
-
"config"
|
|
2046
|
-
]).optional()
|
|
2052
|
+
groupBy: require_db.zod_default.enum(COST_SUMMARY_GROUP_BY).optional()
|
|
2047
2053
|
});
|
|
2048
2054
|
/**
|
|
2049
2055
|
* Helper to create column reference for SQL
|
|
@@ -2286,6 +2292,35 @@ const createLLMRequestsDataLayer = (db) => {
|
|
|
2286
2292
|
kysely.sql`COALESCE(SUM(${col("cost")}), 0)`.as("totalCost"),
|
|
2287
2293
|
kysely.sql`COUNT(*)`.as("requestCount")
|
|
2288
2294
|
]).groupBy("configId").orderBy(kysely.sql`SUM(${col("cost")})`, "desc").execute();
|
|
2295
|
+
case "endpoint": return baseQuery.select([
|
|
2296
|
+
kysely.sql`COALESCE(${col("endpoint")}, 'unknown')`.as("groupKey"),
|
|
2297
|
+
kysely.sql`COALESCE(SUM(${col("cost")}), 0)`.as("totalCost"),
|
|
2298
|
+
kysely.sql`COUNT(*)`.as("requestCount")
|
|
2299
|
+
]).groupBy("endpoint").orderBy(kysely.sql`SUM(${col("cost")})`, "desc").execute();
|
|
2300
|
+
case "tags": {
|
|
2301
|
+
const conditions = [kysely.sql`${col("createdAt")} >= ${startDate.toISOString()}`, kysely.sql`${col("createdAt")} <= ${endDate.toISOString()}`];
|
|
2302
|
+
if (configId) conditions.push(kysely.sql`${col("configId")} = ${configId}`);
|
|
2303
|
+
if (variantId) conditions.push(kysely.sql`${col("variantId")} = ${variantId}`);
|
|
2304
|
+
if (environmentId) conditions.push(kysely.sql`${col("environmentId")} = ${environmentId}`);
|
|
2305
|
+
if (tags && Object.keys(tags).length > 0) for (const [key, values] of Object.entries(tags)) {
|
|
2306
|
+
if (values.length === 0) continue;
|
|
2307
|
+
if (values.length === 1) conditions.push(kysely.sql`${col("tags")}->>${key} = ${values[0]}`);
|
|
2308
|
+
else {
|
|
2309
|
+
const valueList = kysely.sql.join(values.map((v) => kysely.sql`${v}`));
|
|
2310
|
+
conditions.push(kysely.sql`${col("tags")}->>${key} IN (${valueList})`);
|
|
2311
|
+
}
|
|
2312
|
+
}
|
|
2313
|
+
const whereClause = kysely.sql.join(conditions, kysely.sql` AND `);
|
|
2314
|
+
return (await kysely.sql`
|
|
2315
|
+
SELECT t.key || ':' || t.value as "groupKey",
|
|
2316
|
+
COALESCE(SUM(${col("cost")}), 0) as "totalCost",
|
|
2317
|
+
COUNT(*) as "requestCount"
|
|
2318
|
+
FROM "llm_requests", jsonb_each_text(${col("tags")}) t
|
|
2319
|
+
WHERE ${whereClause}
|
|
2320
|
+
GROUP BY t.key, t.value
|
|
2321
|
+
ORDER BY SUM(${col("cost")}) DESC
|
|
2322
|
+
`.execute(db)).rows;
|
|
2323
|
+
}
|
|
2289
2324
|
default: return baseQuery.select([
|
|
2290
2325
|
kysely.sql`'total'`.as("groupKey"),
|
|
2291
2326
|
kysely.sql`COALESCE(SUM(${col("cost")}), 0)`.as("totalCost"),
|
|
@@ -3927,6 +3962,7 @@ var ManifestRouter = class {
|
|
|
3927
3962
|
};
|
|
3928
3963
|
|
|
3929
3964
|
//#endregion
|
|
3965
|
+
exports.COST_SUMMARY_GROUP_BY = COST_SUMMARY_GROUP_BY;
|
|
3930
3966
|
exports.CacheService = CacheService;
|
|
3931
3967
|
exports.DEFAULT_PROVIDER_ENV_VARS = DEFAULT_PROVIDER_ENV_VARS;
|
|
3932
3968
|
exports.FileCacheBackend = FileCacheBackend;
|
package/dist/index.d.cts
CHANGED
|
@@ -1998,6 +1998,11 @@ declare const dateRangeSchema: z$1.ZodObject<{
|
|
|
1998
1998
|
environmentId: z$1.ZodOptional<z$1.ZodString>;
|
|
1999
1999
|
tags: z$1.ZodOptional<z$1.ZodRecord<z$1.ZodString, z$1.ZodArray<z$1.ZodString>>>;
|
|
2000
2000
|
}, z$1.core.$strip>;
|
|
2001
|
+
/**
|
|
2002
|
+
* Valid groupBy values for cost summary queries
|
|
2003
|
+
*/
|
|
2004
|
+
declare const COST_SUMMARY_GROUP_BY: readonly ["day", "hour", "model", "provider", "config", "endpoint", "tags"];
|
|
2005
|
+
type CostSummaryGroupBy = (typeof COST_SUMMARY_GROUP_BY)[number];
|
|
2001
2006
|
/**
|
|
2002
2007
|
* Schema for cost summary with grouping
|
|
2003
2008
|
*/
|
|
@@ -2012,6 +2017,8 @@ declare const costSummarySchema: z$1.ZodObject<{
|
|
|
2012
2017
|
provider: "provider";
|
|
2013
2018
|
config: "config";
|
|
2014
2019
|
model: "model";
|
|
2020
|
+
endpoint: "endpoint";
|
|
2021
|
+
tags: "tags";
|
|
2015
2022
|
day: "day";
|
|
2016
2023
|
hour: "hour";
|
|
2017
2024
|
}>>;
|
|
@@ -3916,4 +3923,4 @@ declare class ManifestRouter {
|
|
|
3916
3923
|
routeWithWeights(configIdOrSlug: string, environmentId: string, context?: RoutingContext): RoutingResult | null;
|
|
3917
3924
|
}
|
|
3918
3925
|
//#endregion
|
|
3919
|
-
export { type AnthropicProviderConfig, type AnyProviderConfig, AuthClientDatabaseConfig, AuthClientOptions, type AzureAIProviderConfig, type AzureOpenAIProviderConfig, BaseCacheConfig, type BaseProviderConfig, type BedrockProviderConfig, CacheBackend, CacheBackendType, CacheConfig, CacheEntry, CacheOptions, CacheService, CacheStats, ChatCompletionCreateParamsBase, Config, ConfigVariant, ConfigVariantsDataLayer, ConfigVariantsTable, ConfigsDataLayer, ConfigsTable, type CortexProviderConfig, CostResult, 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 };
|
|
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 };
|
package/dist/index.d.mts
CHANGED
|
@@ -1998,6 +1998,11 @@ declare const dateRangeSchema: z$1.ZodObject<{
|
|
|
1998
1998
|
environmentId: z$1.ZodOptional<z$1.ZodString>;
|
|
1999
1999
|
tags: z$1.ZodOptional<z$1.ZodRecord<z$1.ZodString, z$1.ZodArray<z$1.ZodString>>>;
|
|
2000
2000
|
}, z$1.core.$strip>;
|
|
2001
|
+
/**
|
|
2002
|
+
* Valid groupBy values for cost summary queries
|
|
2003
|
+
*/
|
|
2004
|
+
declare const COST_SUMMARY_GROUP_BY: readonly ["day", "hour", "model", "provider", "config", "endpoint", "tags"];
|
|
2005
|
+
type CostSummaryGroupBy = (typeof COST_SUMMARY_GROUP_BY)[number];
|
|
2001
2006
|
/**
|
|
2002
2007
|
* Schema for cost summary with grouping
|
|
2003
2008
|
*/
|
|
@@ -2012,6 +2017,8 @@ declare const costSummarySchema: z$1.ZodObject<{
|
|
|
2012
2017
|
provider: "provider";
|
|
2013
2018
|
config: "config";
|
|
2014
2019
|
model: "model";
|
|
2020
|
+
endpoint: "endpoint";
|
|
2021
|
+
tags: "tags";
|
|
2015
2022
|
day: "day";
|
|
2016
2023
|
hour: "hour";
|
|
2017
2024
|
}>>;
|
|
@@ -3916,4 +3923,4 @@ declare class ManifestRouter {
|
|
|
3916
3923
|
routeWithWeights(configIdOrSlug: string, environmentId: string, context?: RoutingContext): RoutingResult | null;
|
|
3917
3924
|
}
|
|
3918
3925
|
//#endregion
|
|
3919
|
-
export { type AnthropicProviderConfig, type AnyProviderConfig, AuthClientDatabaseConfig, AuthClientOptions, type AzureAIProviderConfig, type AzureOpenAIProviderConfig, BaseCacheConfig, type BaseProviderConfig, type BedrockProviderConfig, CacheBackend, CacheBackendType, CacheConfig, CacheEntry, CacheOptions, CacheService, CacheStats, ChatCompletionCreateParamsBase, type Config, type ConfigVariant, type ConfigVariantsDataLayer, type ConfigVariantsTable, type ConfigsDataLayer, type ConfigsTable, type CortexProviderConfig, CostResult, 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 };
|
|
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 };
|
package/dist/index.mjs
CHANGED
|
@@ -2024,6 +2024,18 @@ const dateRangeSchema = zod_default.object({
|
|
|
2024
2024
|
tags: zod_default.record(zod_default.string(), zod_default.array(zod_default.string())).optional()
|
|
2025
2025
|
});
|
|
2026
2026
|
/**
|
|
2027
|
+
* Valid groupBy values for cost summary queries
|
|
2028
|
+
*/
|
|
2029
|
+
const COST_SUMMARY_GROUP_BY = [
|
|
2030
|
+
"day",
|
|
2031
|
+
"hour",
|
|
2032
|
+
"model",
|
|
2033
|
+
"provider",
|
|
2034
|
+
"config",
|
|
2035
|
+
"endpoint",
|
|
2036
|
+
"tags"
|
|
2037
|
+
];
|
|
2038
|
+
/**
|
|
2027
2039
|
* Schema for cost summary with grouping
|
|
2028
2040
|
*/
|
|
2029
2041
|
const costSummarySchema = zod_default.object({
|
|
@@ -2033,13 +2045,7 @@ const costSummarySchema = zod_default.object({
|
|
|
2033
2045
|
variantId: zod_default.string().uuid().optional(),
|
|
2034
2046
|
environmentId: zod_default.string().uuid().optional(),
|
|
2035
2047
|
tags: zod_default.record(zod_default.string(), zod_default.array(zod_default.string())).optional(),
|
|
2036
|
-
groupBy: zod_default.enum(
|
|
2037
|
-
"day",
|
|
2038
|
-
"hour",
|
|
2039
|
-
"model",
|
|
2040
|
-
"provider",
|
|
2041
|
-
"config"
|
|
2042
|
-
]).optional()
|
|
2048
|
+
groupBy: zod_default.enum(COST_SUMMARY_GROUP_BY).optional()
|
|
2043
2049
|
});
|
|
2044
2050
|
/**
|
|
2045
2051
|
* Helper to create column reference for SQL
|
|
@@ -2282,6 +2288,35 @@ const createLLMRequestsDataLayer = (db) => {
|
|
|
2282
2288
|
sql`COALESCE(SUM(${col("cost")}), 0)`.as("totalCost"),
|
|
2283
2289
|
sql`COUNT(*)`.as("requestCount")
|
|
2284
2290
|
]).groupBy("configId").orderBy(sql`SUM(${col("cost")})`, "desc").execute();
|
|
2291
|
+
case "endpoint": return baseQuery.select([
|
|
2292
|
+
sql`COALESCE(${col("endpoint")}, 'unknown')`.as("groupKey"),
|
|
2293
|
+
sql`COALESCE(SUM(${col("cost")}), 0)`.as("totalCost"),
|
|
2294
|
+
sql`COUNT(*)`.as("requestCount")
|
|
2295
|
+
]).groupBy("endpoint").orderBy(sql`SUM(${col("cost")})`, "desc").execute();
|
|
2296
|
+
case "tags": {
|
|
2297
|
+
const conditions = [sql`${col("createdAt")} >= ${startDate.toISOString()}`, sql`${col("createdAt")} <= ${endDate.toISOString()}`];
|
|
2298
|
+
if (configId) conditions.push(sql`${col("configId")} = ${configId}`);
|
|
2299
|
+
if (variantId) conditions.push(sql`${col("variantId")} = ${variantId}`);
|
|
2300
|
+
if (environmentId) conditions.push(sql`${col("environmentId")} = ${environmentId}`);
|
|
2301
|
+
if (tags && Object.keys(tags).length > 0) for (const [key, values] of Object.entries(tags)) {
|
|
2302
|
+
if (values.length === 0) continue;
|
|
2303
|
+
if (values.length === 1) conditions.push(sql`${col("tags")}->>${key} = ${values[0]}`);
|
|
2304
|
+
else {
|
|
2305
|
+
const valueList = sql.join(values.map((v) => sql`${v}`));
|
|
2306
|
+
conditions.push(sql`${col("tags")}->>${key} IN (${valueList})`);
|
|
2307
|
+
}
|
|
2308
|
+
}
|
|
2309
|
+
const whereClause = sql.join(conditions, sql` AND `);
|
|
2310
|
+
return (await sql`
|
|
2311
|
+
SELECT t.key || ':' || t.value as "groupKey",
|
|
2312
|
+
COALESCE(SUM(${col("cost")}), 0) as "totalCost",
|
|
2313
|
+
COUNT(*) as "requestCount"
|
|
2314
|
+
FROM "llm_requests", jsonb_each_text(${col("tags")}) t
|
|
2315
|
+
WHERE ${whereClause}
|
|
2316
|
+
GROUP BY t.key, t.value
|
|
2317
|
+
ORDER BY SUM(${col("cost")}) DESC
|
|
2318
|
+
`.execute(db)).rows;
|
|
2319
|
+
}
|
|
2285
2320
|
default: return baseQuery.select([
|
|
2286
2321
|
sql`'total'`.as("groupKey"),
|
|
2287
2322
|
sql`COALESCE(SUM(${col("cost")}), 0)`.as("totalCost"),
|
|
@@ -3923,4 +3958,4 @@ var ManifestRouter = class {
|
|
|
3923
3958
|
};
|
|
3924
3959
|
|
|
3925
3960
|
//#endregion
|
|
3926
|
-
export { CacheService, DEFAULT_PROVIDER_ENV_VARS, FileCacheBackend, MS, ManifestBuilder, ManifestRouter, ManifestService, MemoryCacheBackend, ModelsDevPricingProvider, SCHEMA_METADATA, SupportedProviders, 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 };
|
|
3961
|
+
export { COST_SUMMARY_GROUP_BY, CacheService, DEFAULT_PROVIDER_ENV_VARS, FileCacheBackend, MS, ManifestBuilder, ManifestRouter, ManifestService, MemoryCacheBackend, ModelsDevPricingProvider, SCHEMA_METADATA, SupportedProviders, 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 };
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@llmops/core",
|
|
3
|
-
"version": "0.5.
|
|
3
|
+
"version": "0.5.3",
|
|
4
4
|
"description": "Core LLMOps functionality and utilities",
|
|
5
5
|
"type": "module",
|
|
6
6
|
"license": "Apache-2.0",
|
|
@@ -55,7 +55,7 @@
|
|
|
55
55
|
"kysely": "^0.28.8",
|
|
56
56
|
"kysely-neon": "^2.0.2",
|
|
57
57
|
"pino": "^10.1.0",
|
|
58
|
-
"@llmops/gateway": "^0.5.
|
|
58
|
+
"@llmops/gateway": "^0.5.3"
|
|
59
59
|
},
|
|
60
60
|
"devDependencies": {
|
|
61
61
|
"@types/json-logic-js": "^2.0.8",
|