@anyway-sh/node-server-sdk 0.22.9 → 0.22.11

package/dist/index.js

@@ -90,7 +90,7 @@ class PromptNotFoundError extends TraceloopError {
     }
 }
 
-var version = "0.22.9";
+var version = "0.22.11";
 
 /**
  * Base class for handling annotation operations with the Traceloop API.
@@ -3262,6 +3262,10 @@ function parseKeyPairsIntoRecord(keyPairs) {
 }
 
 const ALL_INSTRUMENTATION_LIBRARIES = "all";
+let _pricingCalculator = null;
+const setPricingCalculator = (calculator) => {
+    _pricingCalculator = calculator;
+};
 const spanAgentNames = new Map();
 const SPAN_AGENT_NAME_TTL = 5 * 60 * 1000;
 const AI_TELEMETRY_METADATA_AGENT = "ai.telemetry.metadata.agent";
@@ -3411,13 +3415,9 @@ const ensureSpanCompatibility = (span) => {
 const onSpanEnd = (originalOnEnd, instrumentationLibraries) => {
     return (span) => {
         var _a, _b, _c;
-        const libName = ((_a = span.instrumentationScope) === null || _a === void 0 ? void 0 : _a.name) ||
-            ((_b = span.instrumentationLibrary) === null || _b === void 0 ? void 0 : _b.name) ||
-            "unknown";
-        console.log(`[Anyway Debug] onSpanEnd: span="${span.name}" lib="${libName}"`);
         if (instrumentationLibraries &&
-            !instrumentationLibraries.includes(libName)) {
-            console.log(`[Anyway Debug] Dropping span="${span.name}" - lib="${libName}" not in allowed list`);
+            !instrumentationLibraries.includes(((_a = span.instrumentationScope) === null || _a === void 0 ? void 0 : _a.name) ||
+                ((_b = span.instrumentationLibrary) === null || _b === void 0 ? void 0 : _b.name))) {
             return;
         }
         transformAiSdkSpanAttributes(span);
@@ -3443,8 +3443,10 @@ const onSpanEnd = (originalOnEnd, instrumentationLibraries) => {
         if (Math.random() < 0.01) {
             cleanupExpiredSpanAgentNames();
         }
+        if (_pricingCalculator) {
+            _pricingCalculator.addCostAttributes(span);
+        }
         const compatibleSpan = ensureSpanCompatibility(span);
-        console.log(`[Anyway Debug] Span queued for export: "${span.name}" lib="${libName}"`);
         originalOnEnd(compatibleSpan);
     };
 };
@@ -3730,19 +3732,6 @@ const startTracing = (options) => {
             url: `${baseUrl}/v1/traces`,
             headers,
         }));
-    // [Anyway Debug] Wrap exporter to log outgoing calls
-    const origExport = traceExporter.export.bind(traceExporter);
-    traceExporter.export = (spans, resultCallback) => {
-        var _a, _b;
-        console.log(`[Anyway Debug] Exporting ${spans.length} span(s) to ${baseUrl}/v1/traces`);
-        for (const s of spans) {
-            console.log(`[Anyway Debug]   - ${s.name} (${((_a = s.instrumentationLibrary) === null || _a === void 0 ? void 0 : _a.name) || ((_b = s.instrumentationScope) === null || _b === void 0 ? void 0 : _b.name) || "unknown"})`);
-        }
-        return origExport(spans, (result) => {
-            console.log(`[Anyway Debug] Export result: code=${result.code}${result.error ? ` error=${result.error.message}` : ""}`);
-            resultCallback(result);
-        });
-    };
     _spanProcessor = createSpanProcessor({
         apiKey: options.apiKey,
         baseUrl: options.baseUrl,
@@ -3880,6 +3869,674 @@ const initializeRegistry = (options) => {
     });
 };
 
+const DATE_SUFFIX_PATTERN = /-\d{4}-\d{2}-\d{2}$/;
+const DATE_COMPACT_SUFFIX_PATTERN = /-\d{8}$/;
+class PricingCalculator {
+    constructor(pricingData) {
+        var _a;
+        this.chatModels = (_a = pricingData.chat) !== null && _a !== void 0 ? _a : {};
+    }
+    findPricing(modelName) {
+        if (!modelName)
+            return null;
+        const models = this.chatModels;
+        // 1. Exact match
+        if (models[modelName]) {
+            return this.normalize(models[modelName]);
+        }
+        // 2. Strip date suffix and retry
+        let stripped = modelName.replace(DATE_SUFFIX_PATTERN, "");
+        stripped = stripped.replace(DATE_COMPACT_SUFFIX_PATTERN, "");
+        if (stripped !== modelName && models[stripped]) {
+            return this.normalize(models[stripped]);
+        }
+        // 3. Prefix match (longest wins)
+        let bestMatch = null;
+        let bestLen = 0;
+        for (const baseModel of Object.keys(models)) {
+            if (modelName.startsWith(baseModel) && baseModel.length > bestLen) {
+                bestMatch = baseModel;
+                bestLen = baseModel.length;
+            }
+        }
+        if (bestMatch) {
+            return this.normalize(models[bestMatch]);
+        }
+        return null;
+    }
+    addCostAttributes(span) {
+        const attrs = span.attributes;
+        if (!attrs)
+            return;
+        const model = attrs[incubating.ATTR_GEN_AI_RESPONSE_MODEL] ||
+            attrs[incubating.ATTR_GEN_AI_REQUEST_MODEL];
+        if (!model)
+            return;
+        const inputTokens = attrs[incubating.ATTR_GEN_AI_USAGE_INPUT_TOKENS];
+        const outputTokens = attrs[incubating.ATTR_GEN_AI_USAGE_OUTPUT_TOKENS];
+        if (inputTokens == null && outputTokens == null)
+            return;
+        const pricing = this.findPricing(model);
+        if (!pricing)
+            return;
+        const inputCost = (inputTokens !== null && inputTokens !== void 0 ? inputTokens : 0) * pricing.inputCostPerToken;
+        const outputCost = (outputTokens !== null && outputTokens !== void 0 ? outputTokens : 0) * pricing.outputCostPerToken;
+        span.attributes["gen_ai.usage.input_cost"] = inputCost;
+        span.attributes["gen_ai.usage.output_cost"] = outputCost;
+        span.attributes["gen_ai.usage.cost"] = inputCost + outputCost;
+    }
+    normalize(pricing) {
+        var _a, _b;
+        return {
+            inputCostPerToken: ((_a = pricing.promptPrice) !== null && _a !== void 0 ? _a : 0) / 1000,
+            outputCostPerToken: ((_b = pricing.completionPrice) !== null && _b !== void 0 ? _b : 0) / 1000,
+        };
+    }
+}
+
+var embeddings = {
+	"text-embedding-ada-002": 0.0001,
+	"text-embedding-3-small": 0.00002,
+	"text-embedding-3-large": 0.00013,
+	ada: 0.0001,
+	"ada-v2": 0.0001,
+	"text-ada-001": 0.0001,
+	"azure_text-embedding-ada-002": 0.0001,
+	"azure_text-embedding-3-small": 0.00002,
+	"azure_text-embedding-3-large": 0.00013,
+	azure_ada: 0.0001,
+	"azure_ada-v2": 0.0001,
+	"azure_text-ada-001": 0.0001,
+	"embed-english-v3.0": 0.0001,
+	"embed-multilingual-v3.0": 0.0001,
+	"embed-english-light-v3.0": 0.0001,
+	"embed-multilingual-light-v3.0": 0.0001,
+	"embed-english-v2.0": 0.0001,
+	"embed-english-light-v2.0": 0.0001,
+	"embed-multilingual-v2.0": 0.0001,
+	"mistral-embed": 0.0001,
+	"amazon.titan-embed-text-v1": 0.0001,
+	"amazon.titan-embed-text-v2": 0.00002,
+	"textembedding-gecko": 0.0001,
+	"textembedding-gecko@001": 0.0001,
+	"textembedding-gecko@002": 0.0001,
+	"textembedding-gecko@003": 0.0001
+};
+var images = {
+	"dall-e-3": {
+		standard: {
+			"1024x1024": 0.04,
+			"1024x1792": 0.02,
+			"1792x1024": 0.08
+		},
+		hd: {
+			"1024x1024": 0.08,
+			"1024x1792": 0.12,
+			"1792x1024": 0.12
+		}
+	},
+	"azure_dall-e-3": {
+		standard: {
+			"1024x1024": 0.04,
+			"1024x1792": 0.02,
+			"1792x1024": 0.08
+		},
+		hd: {
+			"1024x1024": 0.08,
+			"1024x1792": 0.12,
+			"1792x1024": 0.12
+		}
+	},
+	"dall-e-2": {
+		standard: {
+			"1024x1024": 0.02,
+			"512x512": 0.018,
+			"256x256": 0.016
+		}
+	},
+	"black-forest-labs/FLUX.1-dev": {
+		standard: {
+			"1000000": 0.025
+		}
+	},
+	"black-forest-labs/FLUX.1-canny": {
+		standard: {
+			"1000000": 0.025
+		}
+	},
+	"black-forest-labs/FLUX.1-depth": {
+		standard: {
+			"1000000": 0.025
+		}
+	},
+	"black-forest-labs/FLUX.1-redux": {
+		standard: {
+			"1000000": 0.025
+		}
+	},
+	"black-forest-labs/FLUX.1-schnell": {
+		standard: {
+			"1000000": 0.0027
+		}
+	},
+	"black-forest-labs/FLUX.1-pro": {
+		standard: {
+			"1000000": 0.04
+		}
+	},
+	"black-forest-labs/FLUX.1.1-pro": {
+		standard: {
+			"1000000": 0.05
+		}
+	},
+	"stabilityai/stable-diffusion-xl-base-1.0": {
+		standard: {
+			"512X512": 0.001,
+			"1024x1024": 0.01
+		}
+	},
+	"amazon.titan-image-generator-v1": {
+		standard: {
+			"512x512": 0.008,
+			"1024x1024": 0.01
+		},
+		premium: {
+			"512x512": 0.01,
+			"1024x1024": 0.012
+		}
+	}
+};
+var audio = {
+	"tts-1": 0.015,
+	"tts-1-hd": 0.03,
+	eleven_multilingual_v2: 0.24,
+	eleven_multilingual_v1: 0.24,
+	eleven_monolingual_v1: 0.24,
+	eleven_english_v1: 0.24,
+	eleven_turbo_v2: 0.24,
+	eleven_english_sts_v2: 0.24,
+	eleven_multilingual_sts_v2: 0.24,
+	best: 0.00010277777,
+	nano: 0.00003333333
+};
+var chat = {
+	"gpt-4.1-nano-2025-04-14": {
+		promptPrice: 0.0001,
+		completionPrice: 0.0004
+	},
+	"gpt-4.1-nano": {
+		promptPrice: 0.0001,
+		completionPrice: 0.0004
+	},
+	"gpt-4.1-mini-2025-04-14": {
+		promptPrice: 0.0004,
+		completionPrice: 0.0016
+	},
+	"gpt-4.1-mini": {
+		promptPrice: 0.0004,
+		completionPrice: 0.0016
+	},
+	"gpt-4.1-2025-04-14": {
+		promptPrice: 0.002,
+		completionPrice: 0.008
+	},
+	"gpt-4.1": {
+		promptPrice: 0.002,
+		completionPrice: 0.008
+	},
+	"gpt-4o": {
+		promptPrice: 0.0005,
+		completionPrice: 0.0015
+	},
+	"gpt-4o-2024-08-06": {
+		promptPrice: 0.0025,
+		completionPrice: 0.01
+	},
+	"gpt-4o-2024-05-13": {
+		promptPrice: 0.005,
+		completionPrice: 0.015
+	},
+	"gpt-4o-mini": {
+		promptPrice: 0.00015,
+		completionPrice: 0.0006
+	},
+	"gpt-4o-mini-2024-07-18": {
+		promptPrice: 0.00015,
+		completionPrice: 0.0006
+	},
+	"o1-mini": {
+		promptPrice: 0.003,
+		completionPrice: 0.012
+	},
+	"o1-mini-2024-09-12": {
+		promptPrice: 0.003,
+		completionPrice: 0.012
+	},
+	"o1-preview": {
+		promptPrice: 0.015,
+		completionPrice: 0.06
+	},
+	"o1-preview-2024-09-12": {
+		promptPrice: 0.015,
+		completionPrice: 0.06
+	},
+	"gpt-3.5-turbo": {
+		promptPrice: 0.0005,
+		completionPrice: 0.0015
+	},
+	"gpt-3.5-turbo-0125": {
+		promptPrice: 0.0005,
+		completionPrice: 0.0015
+	},
+	"azure_gpt-35-turbo": {
+		promptPrice: 0.0005,
+		completionPrice: 0.0015
+	},
+	"azure_gpt-35-turbo-16k": {
+		promptPrice: 0.0005,
+		completionPrice: 0.0015
+	},
+	"azure_gpt-35-turbo-instruct": {
+		promptPrice: 0.0015,
+		completionPrice: 0.002
+	},
+	"gpt-4": {
+		promptPrice: 0.03,
+		completionPrice: 0.06
+	},
+	"gpt-4-turbo": {
+		promptPrice: 0.01,
+		completionPrice: 0.03
+	},
+	"gpt-4-32k": {
+		promptPrice: 0.06,
+		completionPrice: 0.12
+	},
+	"gpt-4-1106-preview": {
+		promptPrice: 0.01,
+		completionPrice: 0.03
+	},
+	"gpt-4-0125-preview": {
+		promptPrice: 0.01,
+		completionPrice: 0.03
+	},
+	"gpt-4-preview": {
+		promptPrice: 0.01,
+		completionPrice: 0.03
+	},
+	"gpt-4-1106-vision-preview": {
+		promptPrice: 0.01,
+		completionPrice: 0.03
+	},
+	"gpt-4-vision-preview": {
+		promptPrice: 0.01,
+		completionPrice: 0.03
+	},
+	"azure_gpt-4": {
+		promptPrice: 0.03,
+		completionPrice: 0.06
+	},
+	"azure_gpt-4-32k": {
+		promptPrice: 0.06,
+		completionPrice: 0.12
+	},
+	"claude-3-opus-20240229": {
+		promptPrice: 0.015,
+		completionPrice: 0.075
+	},
+	"claude-3-sonnet-20240229": {
+		promptPrice: 0.003,
+		completionPrice: 0.015
+	},
+	"claude-3-haiku-20240307": {
+		promptPrice: 0.00025,
+		completionPrice: 0.00125
+	},
+	command: {
+		promptPrice: 0.001,
+		completionPrice: 0.002
+	},
+	"command-nightly": {
+		promptPrice: 0.001,
+		completionPrice: 0.002
+	},
+	"command-light": {
+		promptPrice: 0.0003,
+		completionPrice: 0.0006
+	},
+	"command-light-nightly": {
+		promptPrice: 0.0003,
+		completionPrice: 0.0006
+	},
+	"open-mistral-7b": {
+		promptPrice: 0.00025,
+		completionPrice: 0.00025
+	},
+	"open-mixtral-8x7b": {
+		promptPrice: 0.0007,
+		completionPrice: 0.0007
+	},
+	"mistral-small-latest": {
+		promptPrice: 0.002,
+		completionPrice: 0.006
+	},
+	"mistral-medium-latest": {
+		promptPrice: 0.0027,
+		completionPrice: 0.0081
+	},
+	"mistral-large-latest": {
+		promptPrice: 0.008,
+		completionPrice: 0.024
+	},
+	"amazon.titan-text-express-v1": {
+		promptPrice: 0.0008,
+		completionPrice: 0.0016
+	},
+	"amazon.titan-text-lite-v1": {
+		promptPrice: 0.0003,
+		completionPrice: 0.0004
+	},
+	"mistral.mistral-7b-instruct-v0:2": {
+		promptPrice: 0.00015,
+		completionPrice: 0.0002
+	},
+	"mistral.mixtral-8x7b-instruct-v0:1": {
+		promptPrice: 0.00045,
+		completionPrice: 0.0007
+	},
+	"mistral.mistral-large-2402-v1:0": {
+		promptPrice: 0.008,
+		completionPrice: 0.024
+	},
+	"anthropic.claude-3-haiku-20240307-v1:0": {
+		promptPrice: 0.00025,
+		completionPrice: 0.00125
+	},
+	"anthropic.claude-v2": {
+		promptPrice: 0.008,
+		completionPrice: 0.024
+	},
+	"anthropic.claude-3-sonnet-20240229-v1:0": {
+		promptPrice: 0.003,
+		completionPrice: 0.015
+	},
+	"anthropic.claude-3-opus-20240229-v1:0": {
+		promptPrice: 0.015,
+		completionPrice: 0.075
+	},
+	"meta.llama3-8b-instruct-v1:0": {
+		promptPrice: 0.0004,
+		completionPrice: 0.0006
+	},
+	"meta.llama3-70b-instruct-v1:0": {
+		promptPrice: 0.00265,
+		completionPrice: 0.0035
+	},
+	"meta.llama2-13b-chat-v1": {
+		promptPrice: 0.00075,
+		completionPrice: 0.001
+	},
+	"meta.llama2-70b-chat-v1": {
+		promptPrice: 0.00195,
+		completionPrice: 0.00256
+	},
+	"cohere.command-text-v14": {
+		promptPrice: 0.0015,
+		completionPrice: 0.002
+	},
+	"cohere.command-light-text-v14": {
+		promptPrice: 0.0003,
+		completionPrice: 0.0006
+	},
+	"ai21.j2-mid-v1": {
+		promptPrice: 0.0125,
+		completionPrice: 0.0125
+	},
+	"ai21.j2-ultra-v1": {
+		promptPrice: 0.0188,
+		completionPrice: 0.0188
+	},
+	"gemini-1.0-pro": {
+		promptPrice: 0.0005,
+		completionPrice: 0.0015
+	},
+	"gemini-1.5-flash": {
+		promptPrice: 0.000075,
+		completionPrice: 0.0003
+	},
+	"gemini-1.5-pro": {
+		promptPrice: 0.00125,
+		completionPrice: 0.005
+	},
+	"gemini-1.0-pro-002": {
+		promptPrice: 0.0005,
+		completionPrice: 0.0015
+	},
+	"gemini-1.0-pro-001": {
+		promptPrice: 0.0005,
+		completionPrice: 0.0015
+	},
+	"gemini-1.5-pro-preview-0409": {
+		promptPrice: 0.005,
+		completionPrice: 0.015
+	},
+	"gemini-1.5-pro-preview-0514": {
+		promptPrice: 0.005,
+		completionPrice: 0.015
+	},
+	"gemini-1.5-flash-preview-0514": {
+		promptPrice: 0.0005,
+		completionPrice: 0.0015
+	},
+	"gemini-2.0-flash": {
+		promptPrice: 0.0001,
+		completionPrice: 0.0004
+	},
+	"gemini-2.0-flash-lite": {
+		promptPrice: 0.000075,
+		completionPrice: 0.0003
+	},
+	"gemini-2.5-flash": {
+		promptPrice: 0.0003,
+		completionPrice: 0.0025
+	},
+	"gemini-2.5-flash-preview": {
+		promptPrice: 0.0003,
+		completionPrice: 0.0025
+	},
+	"gemini-2.5-flash-lite": {
+		promptPrice: 0.0001,
+		completionPrice: 0.0004
+	},
+	"gemini-2.5-flash-lite-preview": {
+		promptPrice: 0.0001,
+		completionPrice: 0.0004
+	},
+	"gemini-2.5-pro": {
+		promptPrice: 0.00125,
+		completionPrice: 0.01
+	},
+	"text-bison": {
+		promptPrice: 0.001,
+		completionPrice: 0.002
+	},
+	"text-bison@002": {
+		promptPrice: 0.001,
+		completionPrice: 0.002
+	},
+	"text-bison-32k": {
+		promptPrice: 0.001,
+		completionPrice: 0.002
+	},
+	"text-bison-32k@002": {
+		promptPrice: 0.001,
+		completionPrice: 0.002
+	},
+	"text-unicorn": {
+		promptPrice: 0.01,
+		completionPrice: 0.03
+	},
+	"text-unicorn@001": {
+		promptPrice: 0.01,
+		completionPrice: 0.03
+	},
+	"chat-bison": {
+		promptPrice: 0.001,
+		completionPrice: 0.002
+	},
+	"chat-bison@002": {
+		promptPrice: 0.001,
+		completionPrice: 0.002
+	},
+	"chat-bison-32k": {
+		promptPrice: 0.001,
+		completionPrice: 0.002
+	},
+	"chat-bison-32k@002": {
+		promptPrice: 0.001,
+		completionPrice: 0.002
+	},
+	"llama3-8b-8192": {
+		promptPrice: 0.00005,
+		completionPrice: 0.0001
+	},
+	"llama3-70b-8192": {
+		promptPrice: 0.00059,
+		completionPrice: 0.00079
+	},
+	"mixtral-8x7b-32768": {
+		promptPrice: 0.00024,
+		completionPrice: 0.00024
+	},
+	"gemma-7b-it": {
+		promptPrice: 0.0001,
+		completionPrice: 0.0001
+	},
+	"reka-core": {
+		promptPrice: 0.002,
+		completionPrice: 0.002
+	},
+	"reka-core-20240415": {
+		promptPrice: 0.002,
+		completionPrice: 0.002
+	},
+	"reka-core-20240501": {
+		promptPrice: 0.002,
+		completionPrice: 0.002
+	},
+	"reka-flash": {
+		promptPrice: 0.0002,
+		completionPrice: 0.0008
+	},
+	"reka-flash-20240226": {
+		promptPrice: 0.0002,
+		completionPrice: 0.0008
+	},
+	"reka-edge": {
+		promptPrice: 0.0001,
+		completionPrice: 0.0001
+	},
+	"reka-edge-20240208": {
+		promptPrice: 0.0001,
+		completionPrice: 0.0001
+	},
+	"reka-spark": {
+		promptPrice: 0.00005,
+		completionPrice: 0.00005
+	},
+	"grok-beta": {
+		promptPrice: 0.0005,
+		completionPrice: 0.0015
+	},
+	"grok-vision-beta": {
+		promptPrice: 0.0005,
+		completionPrice: 0.0015
+	},
+	"jamba-1.5-mini": {
+		promptPrice: 0.0002,
+		completionPrice: 0.0004
+	},
+	"jamba-1.5-large": {
+		promptPrice: 0.002,
+		completionPrice: 0.008
+	},
+	"meta-llama/Llama-3.3-70B-Instruct-Turbo": {
+		promptPrice: 0.00088,
+		completionPrice: 0.00088
+	},
+	"meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo-vLLM-json": {
+		promptPrice: 0.00088,
+		completionPrice: 0.00088
+	},
+	"meta-llama/Meta-Llama-3.1-405B-Instruct-Lite-Pro-lora": {
+		promptPrice: 0.00088,
+		completionPrice: 0.00088
+	},
+	"meta-llama/Meta-Llama-3.1-405B-Instruct-Turbo": {
+		promptPrice: 0.00088,
+		completionPrice: 0.00088
+	},
+	"Qwen/QwQ-32B-Preview": {
+		promptPrice: 0.0012,
+		completionPrice: 0.0012
+	},
+	"codellama/CodeLlama-34b-Instruct-hf": {
+		promptPrice: 0.0008,
+		completionPrice: 0.0008
+	},
+	"databricks/dbrx-instruct": {
+		promptPrice: 0.0012,
+		completionPrice: 0.0012
+	},
+	"deepseek-ai/deepseek-llm-67b-chat": {
+		promptPrice: 0.0009,
+		completionPrice: 0.0009
+	},
+	"google/gemma-2b-it": {
+		promptPrice: 0.0001,
+		completionPrice: 0.0001
+	},
+	"google/gemma-2-27b-it": {
+		promptPrice: 0.0008,
+		completionPrice: 0.0008
+	},
+	"google/gemma-2-9b-it": {
+		promptPrice: 0.0003,
+		completionPrice: 0.0003
+	},
+	"Gryphe/MythoMax-L2-13b-Lite": {
+		promptPrice: 0.0003,
+		completionPrice: 0.0003
+	},
+	"nvidia/Llama-3.1-Nemotron-70B-Instruct-HF": {
+		promptPrice: 0.0088,
+		completionPrice: 0.0088
+	},
+	"deepseek-chat": {
+		promptPrice: 0.00014,
+		completionPrice: 0.00028
+	},
+	"deepseek-reasoner": {
+		promptPrice: 0.00055,
+		completionPrice: 0.00219
+	}
+};
+var defaultPricing = {
+	embeddings: embeddings,
+	images: images,
+	audio: audio,
+	chat: chat
+};
+
+function loadPricing(pricingJsonPath) {
+    if (pricingJsonPath) {
+        // eslint-disable-next-line @typescript-eslint/no-var-requires
+        const fs = require("fs");
+        return JSON.parse(fs.readFileSync(pricingJsonPath, "utf-8"));
+    }
+    return defaultPricing;
+}
+
 let _configuration;
 let _client;
 /**
@@ -3942,15 +4599,21 @@ const initialize = (options = {}) => {
     if (!options.silenceInitializationMessage) {
         console.log(`Traceloop exporting traces to ${_configuration.exporter ? "a custom exporter" : _configuration.baseUrl}`);
     }
-    console.log(`[Anyway Debug] Trace endpoint: ${_configuration.baseUrl}/v1/traces`);
-    console.log(`[Anyway Debug] API key present: ${!!_configuration.apiKey}`);
-    console.log(`[Anyway Debug] Using custom exporter: ${!!_configuration.exporter}`);
     if (options.tracingEnabled === undefined || options.tracingEnabled) {
         if (options.logLevel) {
             api.diag.setLogger(new api.DiagConsoleLogger(), logLevelToOtelLogLevel(options.logLevel));
         }
         startTracing(_configuration);
     }
+    if (options.pricingEnabled !== false) {
+        try {
+            const pricingData = loadPricing(options.pricingJsonPath);
+            setPricingCalculator(new PricingCalculator(pricingData));
+        }
+        catch (e) {
+            console.warn(`[Traceloop] Failed to initialize pricing: ${e.message}`);
+        }
+    }
     initializeRegistry(_configuration);
     if (options.apiKey) {
         _client = new TraceloopClient({
@@ -4462,6 +5125,7 @@ exports.isAttachment = isAttachment;
 exports.isAttachmentReference = isAttachmentReference;
 exports.isExternalAttachment = isExternalAttachment;
 exports.reportCustomMetric = reportCustomMetric;
+exports.setPricingCalculator = setPricingCalculator;
 exports.task = task;
 exports.tool = tool;
 exports.traceloopInstrumentationLibraries = traceloopInstrumentationLibraries;