openlit 1.10.0 → 1.12.0

package/dist/instrumentation/vercel-ai/wrapper.js

@@ -8,106 +8,71 @@ const config_1 = __importDefault(require("../../config"));
 const helpers_1 = __importDefault(require("../../helpers"));
 const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
 const base_wrapper_1 = __importDefault(require("../base-wrapper"));
+function spanCreationAttrs(operationName, requestModel) {
+    return {
+        [semantic_convention_1.default.GEN_AI_OPERATION]: operationName,
+        [semantic_convention_1.default.GEN_AI_PROVIDER_NAME_OTEL]: VercelAIWrapper.aiSystem,
+        [semantic_convention_1.default.GEN_AI_REQUEST_MODEL]: requestModel,
+        [semantic_convention_1.default.SERVER_ADDRESS]: VercelAIWrapper.serverAddress,
+        [semantic_convention_1.default.SERVER_PORT]: VercelAIWrapper.serverPort,
+    };
+}
 class VercelAIWrapper extends base_wrapper_1.default {
-    static _getProviderFromModel(model) {
-        if (!model)
-            return VercelAIWrapper.aiSystem;
-        const provider = model.provider || '';
-        if (provider.startsWith('openai'))
-            return semantic_convention_1.default.GEN_AI_SYSTEM_OPENAI;
-        if (provider.startsWith('anthropic'))
-            return semantic_convention_1.default.GEN_AI_SYSTEM_ANTHROPIC;
-        if (provider.startsWith('google'))
-            return semantic_convention_1.default.GEN_AI_SYSTEM_VERTEXAI;
-        if (provider.startsWith('mistral'))
-            return semantic_convention_1.default.GEN_AI_SYSTEM_MISTRAL;
-        if (provider.startsWith('cohere'))
-            return semantic_convention_1.default.GEN_AI_SYSTEM_COHERE;
-        if (provider.startsWith('amazon') || provider.startsWith('aws'))
-            return semantic_convention_1.default.GEN_AI_SYSTEM_AWS_BEDROCK;
-        return VercelAIWrapper.aiSystem;
-    }
     static _patchGenerateText(tracer) {
-        const genAIEndpoint = 'ai.generateText';
+        const genAIEndpoint = 'vercel_ai.generateText';
         return (originalMethod) => {
             return async function (...args) {
-                const span = tracer.startSpan(genAIEndpoint, { kind: api_1.SpanKind.CLIENT });
-                return api_1.context.with(api_1.trace.setSpan(api_1.context.active(), span), async () => {
-                    let metricParams;
-                    try {
-                        const response = await originalMethod.apply(this, args);
-                        const params = args[0] || {};
-                        const model = params.model;
-                        const modelId = model?.modelId || 'unknown';
-                        const aiSystem = VercelAIWrapper._getProviderFromModel(model);
-                        const pricingInfo = await config_1.default.updatePricingJson(config_1.default.pricing_json);
-                        const cost = helpers_1.default.getChatModelCost(modelId, pricingInfo, response.usage?.promptTokens || 0, response.usage?.completionTokens || 0);
-                        VercelAIWrapper.setBaseSpanAttributes(span, { genAIEndpoint, model: modelId, cost, aiSystem });
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_OPERATION, semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT);
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_IS_STREAM, false);
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, params.maxTokens || -1);
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TEMPERATURE, params.temperature ?? 1);
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TOP_P, params.topP ?? 1);
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_MODEL, modelId);
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, response.usage?.promptTokens || 0);
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS, response.usage?.completionTokens || 0);
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_TOTAL_TOKENS, response.usage?.totalTokens || 0);
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_CLIENT_TOKEN_USAGE, response.usage?.totalTokens || 0);
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, [response.finishReason || 'stop']);
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_TYPE, semantic_convention_1.default.GEN_AI_OUTPUT_TYPE_TEXT);
-                        if (response.toolCalls?.length > 0) {
-                            const toolNames = response.toolCalls.map((t) => t.toolName || '').filter(Boolean);
-                            const toolArgs = response.toolCalls.map((t) => JSON.stringify(t.args || {}));
-                            if (toolNames.length > 0)
-                                span.setAttribute(semantic_convention_1.default.GEN_AI_TOOL_NAME, toolNames.join(', '));
-                            if (toolArgs.length > 0)
-                                span.setAttribute(semantic_convention_1.default.GEN_AI_TOOL_CALL_ARGUMENTS, toolArgs);
-                        }
-                        if (config_1.default.traceContent) {
-                            const messages = params.messages || (params.prompt ? [{ role: 'user', content: params.prompt }] : []);
-                            span.setAttribute(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, helpers_1.default.buildInputMessages(messages));
-                            span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES, helpers_1.default.buildOutputMessages(response.text || '', response.finishReason || 'stop', response.toolCalls));
-                        }
-                        metricParams = { genAIEndpoint, model: modelId, cost, aiSystem };
-                        return response;
-                    }
-                    catch (e) {
-                        helpers_1.default.handleException(span, e);
-                        throw e;
-                    }
-                    finally {
-                        span.end();
-                        if (metricParams)
-                            base_wrapper_1.default.recordMetrics(span, metricParams);
-                    }
+                const params = args[0] || {};
+                const modelId = params.model?.modelId || 'unknown';
+                const spanName = `${semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT} ${modelId}`;
+                const span = tracer.startSpan(spanName, {
+                    kind: api_1.SpanKind.CLIENT,
+                    attributes: spanCreationAttrs(semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT, modelId),
+                });
+                return api_1.context
+                    .with(api_1.trace.setSpan(api_1.context.active(), span), async () => {
+                    return originalMethod.apply(this, args);
+                })
+                    .then((response) => {
+                    return VercelAIWrapper._chatComplete({
+                        args,
+                        genAIEndpoint,
+                        response,
+                        span,
+                        outputType: semantic_convention_1.default.GEN_AI_OUTPUT_TYPE_TEXT,
+                    });
+                })
+                    .catch((e) => {
+                    helpers_1.default.handleException(span, e);
+                    base_wrapper_1.default.recordMetrics(span, {
+                        genAIEndpoint,
+                        model: modelId,
+                        aiSystem: VercelAIWrapper.aiSystem,
+                        serverAddress: VercelAIWrapper.serverAddress,
+                        serverPort: VercelAIWrapper.serverPort,
+                        errorType: e?.constructor?.name || '_OTHER',
+                    });
+                    span.end();
+                    throw e;
                 });
             };
         };
     }
     static _patchStreamText(tracer) {
-        const genAIEndpoint = 'ai.streamText';
+        const genAIEndpoint = 'vercel_ai.streamText';
         return (originalMethod) => {
             return async function (...args) {
-                const span = tracer.startSpan(genAIEndpoint, { kind: api_1.SpanKind.CLIENT });
+                const params = args[0] || {};
+                const modelId = params.model?.modelId || 'unknown';
+                const spanName = `${semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT} ${modelId}`;
+                const span = tracer.startSpan(spanName, {
+                    kind: api_1.SpanKind.CLIENT,
+                    attributes: spanCreationAttrs(semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT, modelId),
+                });
                 const startTime = Date.now();
                 const chunkTimestamps = [];
                 try {
-                    const response = await originalMethod.apply(this, args);
-                    const params = args[0] || {};
-                    const model = params.model;
-                    const modelId = model?.modelId || 'unknown';
-                    const aiSystem = VercelAIWrapper._getProviderFromModel(model);
-                    // Set request attributes immediately
-                    span.setAttribute(semantic_convention_1.default.GEN_AI_OPERATION, semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT);
-                    span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_IS_STREAM, true);
-                    span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, params.maxTokens || -1);
-                    span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TEMPERATURE, params.temperature ?? 1);
-                    span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TOP_P, params.topP ?? 1);
-                    if (config_1.default.traceContent) {
-                        const messages = params.messages || (params.prompt ? [{ role: 'user', content: params.prompt }] : []);
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, helpers_1.default.buildInputMessages(messages));
-                    }
-                    // Intercept textStream to capture per-chunk timestamps for TTFT/TBT
+                    const response = await api_1.context.with(api_1.trace.setSpan(api_1.context.active(), span), async () => originalMethod.apply(this, args));
                     try {
                         const originalTextStream = response.textStream;
                         if (originalTextStream && typeof originalTextStream.getReader === 'function') {
@@ -135,11 +100,11 @@ class VercelAIWrapper extends base_wrapper_1.default {
                         }
                     }
                     catch (_) {
-                        // Stream interception failed; TTFT/TBT won't be captured from textStream
+                        // Stream interception is best-effort; TTFT/TBT won't be captured
                     }
-                    // Observe stream completion via usage promise
                     Promise.resolve(response.usage)
                         .then(async (usage) => {
+                        let metricParams;
                         try {
                             const ttft = chunkTimestamps.length > 0 ? (chunkTimestamps[0] - startTime) / 1000 : 0;
                             let tbt = 0;
@@ -147,42 +112,65 @@ class VercelAIWrapper extends base_wrapper_1.default {
                                 const timeDiffs = chunkTimestamps.slice(1).map((t, i) => t - chunkTimestamps[i]);
                                 tbt = timeDiffs.reduce((a, b) => a + b, 0) / timeDiffs.length / 1000;
                             }
-                            const pricingInfo = await config_1.default.updatePricingJson(config_1.default.pricing_json);
-                            const cost = helpers_1.default.getChatModelCost(modelId, pricingInfo, usage?.promptTokens || 0, usage?.completionTokens || 0);
-                            VercelAIWrapper.setBaseSpanAttributes(span, { genAIEndpoint, model: modelId, cost, aiSystem });
-                            span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_MODEL, modelId);
-                            span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, usage?.promptTokens || 0);
-                            span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS, usage?.completionTokens || 0);
-                            span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_TOTAL_TOKENS, usage?.totalTokens || 0);
-                            span.setAttribute(semantic_convention_1.default.GEN_AI_CLIENT_TOKEN_USAGE, usage?.totalTokens || 0);
-                            span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_TYPE, semantic_convention_1.default.GEN_AI_OUTPUT_TYPE_TEXT);
-                            if (ttft > 0)
-                                span.setAttribute(semantic_convention_1.default.GEN_AI_SERVER_TTFT, ttft);
-                            if (tbt > 0)
-                                span.setAttribute(semantic_convention_1.default.GEN_AI_SERVER_TBT, tbt);
                             const finishReason = await Promise.resolve(response.finishReason).catch(() => 'stop');
-                            span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, [finishReason || 'stop']);
-                            if (config_1.default.traceContent) {
-                                const text = await Promise.resolve(response.text).catch(() => '');
-                                span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES, helpers_1.default.buildOutputMessages(text || '', finishReason || 'stop'));
-                            }
-                            base_wrapper_1.default.recordMetrics(span, { genAIEndpoint, model: modelId, cost, aiSystem });
+                            const text = await Promise.resolve(response.text).catch(() => '');
+                            const toolCalls = await Promise.resolve(response.toolCalls).catch(() => undefined);
+                            const responseDetails = await Promise.resolve(response.response).catch(() => undefined);
+                            const result = {
+                                usage: {
+                                    promptTokens: usage?.promptTokens || 0,
+                                    completionTokens: usage?.completionTokens || 0,
+                                },
+                                finishReason: finishReason || 'stop',
+                                text: text || '',
+                                toolCalls,
+                                response: responseDetails,
+                            };
+                            metricParams = await VercelAIWrapper._chatCommonSetter({
+                                args,
+                                genAIEndpoint,
+                                result,
+                                span,
+                                isStream: true,
+                                outputType: semantic_convention_1.default.GEN_AI_OUTPUT_TYPE_TEXT,
+                                ttft,
+                                tbt,
+                            });
                         }
                         catch (e) {
                             helpers_1.default.handleException(span, e);
                         }
                         finally {
                             span.end();
+                            if (metricParams) {
+                                base_wrapper_1.default.recordMetrics(span, metricParams);
+                            }
                         }
                     })
                         .catch((e) => {
                         helpers_1.default.handleException(span, e);
+                        base_wrapper_1.default.recordMetrics(span, {
+                            genAIEndpoint,
+                            model: modelId,
+                            aiSystem: VercelAIWrapper.aiSystem,
+                            serverAddress: VercelAIWrapper.serverAddress,
+                            serverPort: VercelAIWrapper.serverPort,
+                            errorType: e?.constructor?.name || '_OTHER',
+                        });
                         span.end();
                     });
                     return response;
                 }
                 catch (e) {
                     helpers_1.default.handleException(span, e);
+                    base_wrapper_1.default.recordMetrics(span, {
+                        genAIEndpoint,
+                        model: modelId,
+                        aiSystem: VercelAIWrapper.aiSystem,
+                        serverAddress: VercelAIWrapper.serverAddress,
+                        serverPort: VercelAIWrapper.serverPort,
+                        errorType: e?.constructor?.name || '_OTHER',
+                    });
                     span.end();
                     throw e;
                 }
@@ -190,83 +178,103 @@ class VercelAIWrapper extends base_wrapper_1.default {
         };
     }
     static _patchGenerateObject(tracer) {
-        const genAIEndpoint = 'ai.generateObject';
+        const genAIEndpoint = 'vercel_ai.generateObject';
         return (originalMethod) => {
             return async function (...args) {
-                const span = tracer.startSpan(genAIEndpoint, { kind: api_1.SpanKind.CLIENT });
-                return api_1.context.with(api_1.trace.setSpan(api_1.context.active(), span), async () => {
-                    let metricParams;
-                    try {
-                        const response = await originalMethod.apply(this, args);
-                        const params = args[0] || {};
-                        const model = params.model;
-                        const modelId = model?.modelId || 'unknown';
-                        const aiSystem = VercelAIWrapper._getProviderFromModel(model);
-                        const pricingInfo = await config_1.default.updatePricingJson(config_1.default.pricing_json);
-                        const cost = helpers_1.default.getChatModelCost(modelId, pricingInfo, response.usage?.promptTokens || 0, response.usage?.completionTokens || 0);
-                        VercelAIWrapper.setBaseSpanAttributes(span, { genAIEndpoint, model: modelId, cost, aiSystem });
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_OPERATION, semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT);
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_IS_STREAM, false);
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, params.maxTokens || -1);
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TEMPERATURE, params.temperature ?? 1);
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_MODEL, modelId);
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, response.usage?.promptTokens || 0);
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS, response.usage?.completionTokens || 0);
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_TOTAL_TOKENS, response.usage?.totalTokens || 0);
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_CLIENT_TOKEN_USAGE, response.usage?.totalTokens || 0);
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, [response.finishReason || 'stop']);
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_TYPE, semantic_convention_1.default.GEN_AI_OUTPUT_TYPE_JSON);
-                        if (config_1.default.traceContent) {
-                            const messages = params.messages || (params.prompt ? [{ role: 'user', content: params.prompt }] : []);
-                            span.setAttribute(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, helpers_1.default.buildInputMessages(messages));
-                            span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES, helpers_1.default.buildOutputMessages(JSON.stringify(response.object || {}), response.finishReason || 'stop'));
-                        }
-                        metricParams = { genAIEndpoint, model: modelId, cost, aiSystem };
-                        return response;
-                    }
-                    catch (e) {
-                        helpers_1.default.handleException(span, e);
-                        throw e;
-                    }
-                    finally {
-                        span.end();
-                        if (metricParams)
-                            base_wrapper_1.default.recordMetrics(span, metricParams);
-                    }
+                const params = args[0] || {};
+                const modelId = params.model?.modelId || 'unknown';
+                const spanName = `${semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT} ${modelId}`;
+                const span = tracer.startSpan(spanName, {
+                    kind: api_1.SpanKind.CLIENT,
+                    attributes: spanCreationAttrs(semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT, modelId),
+                });
+                return api_1.context
+                    .with(api_1.trace.setSpan(api_1.context.active(), span), async () => {
+                    return originalMethod.apply(this, args);
+                })
+                    .then((response) => {
+                    const result = {
+                        ...response,
+                        text: JSON.stringify(response.object || {}),
+                    };
+                    return VercelAIWrapper._chatComplete({
+                        args,
+                        genAIEndpoint,
+                        response,
+                        span,
+                        outputType: semantic_convention_1.default.GEN_AI_OUTPUT_TYPE_JSON,
+                        resultOverride: result,
+                    });
+                })
+                    .catch((e) => {
+                    helpers_1.default.handleException(span, e);
+                    base_wrapper_1.default.recordMetrics(span, {
+                        genAIEndpoint,
+                        model: modelId,
+                        aiSystem: VercelAIWrapper.aiSystem,
+                        serverAddress: VercelAIWrapper.serverAddress,
+                        serverPort: VercelAIWrapper.serverPort,
+                        errorType: e?.constructor?.name || '_OTHER',
+                    });
+                    span.end();
+                    throw e;
                 });
             };
         };
     }
     static _patchEmbed(tracer) {
-        const genAIEndpoint = 'ai.embed';
+        const genAIEndpoint = 'vercel_ai.embed';
         return (originalMethod) => {
             return async function (...args) {
-                const span = tracer.startSpan(genAIEndpoint, { kind: api_1.SpanKind.CLIENT });
+                const params = args[0] || {};
+                const modelId = params.model?.modelId || 'unknown';
+                const spanName = `${semantic_convention_1.default.GEN_AI_OPERATION_TYPE_EMBEDDING} ${modelId}`;
+                const span = tracer.startSpan(spanName, {
+                    kind: api_1.SpanKind.CLIENT,
+                    attributes: spanCreationAttrs(semantic_convention_1.default.GEN_AI_OPERATION_TYPE_EMBEDDING, modelId),
+                });
                 return api_1.context.with(api_1.trace.setSpan(api_1.context.active(), span), async () => {
+                    const captureContent = config_1.default.captureMessageContent;
                     let metricParams;
                     try {
                         const response = await originalMethod.apply(this, args);
-                        const params = args[0] || {};
-                        const model = params.model;
-                        const modelId = model?.modelId || 'unknown';
-                        const aiSystem = VercelAIWrapper._getProviderFromModel(model);
-                        const pricingInfo = await config_1.default.updatePricingJson(config_1.default.pricing_json);
-                        const cost = helpers_1.default.getEmbedModelCost(modelId, pricingInfo, response.usage?.tokens || 0);
-                        VercelAIWrapper.setBaseSpanAttributes(span, { genAIEndpoint, model: modelId, cost, aiSystem });
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_OPERATION, semantic_convention_1.default.GEN_AI_OPERATION_TYPE_EMBEDDING);
+                        const pricingInfo = config_1.default.pricingInfo || {};
+                        const inputTokens = response.usage?.tokens || 0;
+                        const cost = helpers_1.default.getEmbedModelCost(modelId, pricingInfo, inputTokens);
+                        VercelAIWrapper.setBaseSpanAttributes(span, {
+                            genAIEndpoint,
+                            model: modelId,
+                            cost,
+                            aiSystem: VercelAIWrapper.aiSystem,
+                            serverAddress: VercelAIWrapper.serverAddress,
+                            serverPort: VercelAIWrapper.serverPort,
+                        });
                         span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_IS_STREAM, false);
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_MODEL, modelId);
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, response.usage?.tokens || 0);
-                        span.setAttribute(semantic_convention_1.default.GEN_AI_CLIENT_TOKEN_USAGE, response.usage?.tokens || 0);
-                        if (config_1.default.traceContent && params.value !== undefined) {
+                        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, inputTokens);
+                        if (captureContent && params.value !== undefined) {
                             const inputStr = typeof params.value === 'string' ? params.value : JSON.stringify(params.value);
                             span.setAttribute(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, inputStr);
                         }
-                        metricParams = { genAIEndpoint, model: modelId, cost, aiSystem };
+                        metricParams = {
+                            genAIEndpoint,
+                            model: modelId,
+                            cost,
+                            aiSystem: VercelAIWrapper.aiSystem,
+                            serverAddress: VercelAIWrapper.serverAddress,
+                            serverPort: VercelAIWrapper.serverPort,
+                        };
                         return response;
                     }
                     catch (e) {
                         helpers_1.default.handleException(span, e);
+                        metricParams = {
+                            genAIEndpoint,
+                            model: modelId,
+                            aiSystem: VercelAIWrapper.aiSystem,
+                            serverAddress: VercelAIWrapper.serverAddress,
+                            serverPort: VercelAIWrapper.serverPort,
+                            errorType: e?.constructor?.name || '_OTHER',
+                        };
                         throw e;
                     }
                     finally {
@@ -278,7 +286,149 @@ class VercelAIWrapper extends base_wrapper_1.default {
             };
         };
     }
+    static async _chatComplete({ args, genAIEndpoint, response, span, outputType, resultOverride, }) {
+        let metricParams;
+        try {
+            metricParams = await VercelAIWrapper._chatCommonSetter({
+                args,
+                genAIEndpoint,
+                result: resultOverride || response,
+                span,
+                isStream: false,
+                outputType,
+            });
+            return response;
+        }
+        catch (e) {
+            helpers_1.default.handleException(span, e);
+            throw e;
+        }
+        finally {
+            span.end();
+            if (metricParams) {
+                base_wrapper_1.default.recordMetrics(span, metricParams);
+            }
+        }
+    }
+    static async _chatCommonSetter({ args, genAIEndpoint, result, span, isStream, outputType, ttft = 0, tbt = 0, }) {
+        const captureContent = config_1.default.captureMessageContent;
+        const params = args[0] || {};
+        const modelId = params.model?.modelId || 'unknown';
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TEMPERATURE, params.temperature ?? 1);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TOP_P, params.topP ?? 1);
+        if (params.maxTokens != null) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, params.maxTokens);
+        }
+        span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_IS_STREAM, isStream);
+        if (params.seed != null) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_SEED, Number(params.seed));
+        }
+        if (params.frequencyPenalty) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_FREQUENCY_PENALTY, params.frequencyPenalty);
+        }
+        if (params.presencePenalty) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_PRESENCE_PENALTY, params.presencePenalty);
+        }
+        if (params.stopSequences) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_STOP_SEQUENCES, Array.isArray(params.stopSequences) ? params.stopSequences : [params.stopSequences]);
+        }
+        if (params.topK != null) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TOP_K, params.topK);
+        }
+        const messages = params.messages || (params.prompt ? [{ role: 'user', content: params.prompt }] : []);
+        if (captureContent) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_INPUT_MESSAGES, helpers_1.default.buildInputMessages(messages, params.system));
+        }
+        const responseId = result.response?.id;
+        const responseModel = result.response?.modelId || modelId;
+        const inputTokens = result.usage?.promptTokens || 0;
+        const outputTokens = result.usage?.completionTokens || 0;
+        const pricingInfo = config_1.default.pricingInfo || {};
+        const cost = helpers_1.default.getChatModelCost(modelId, pricingInfo, inputTokens, outputTokens);
+        VercelAIWrapper.setBaseSpanAttributes(span, {
+            genAIEndpoint,
+            model: modelId,
+            cost,
+            aiSystem: VercelAIWrapper.aiSystem,
+            serverAddress: VercelAIWrapper.serverAddress,
+            serverPort: VercelAIWrapper.serverPort,
+        });
+        span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_MODEL, responseModel);
+        if (responseId) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_ID, responseId);
+        }
+        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, inputTokens);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS, outputTokens);
+        const finishReason = result.finishReason || 'stop';
+        span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, [finishReason]);
+        span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_TYPE, outputType);
+        if (ttft > 0) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_SERVER_TTFT, ttft);
+        }
+        if (tbt > 0) {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_SERVER_TBT, tbt);
+        }
+        if (result.toolCalls?.length > 0) {
+            const toolNames = result.toolCalls.map((t) => t.toolName || '').filter(Boolean);
+            const toolIds = result.toolCalls.map((t) => t.toolCallId || '').filter(Boolean);
+            const toolArgs = result.toolCalls.map((t) => JSON.stringify(t.args || {})).filter(Boolean);
+            if (toolNames.length > 0) {
+                span.setAttribute(semantic_convention_1.default.GEN_AI_TOOL_NAME, toolNames.join(', '));
+            }
+            if (toolIds.length > 0) {
+                span.setAttribute(semantic_convention_1.default.GEN_AI_TOOL_CALL_ID, toolIds.join(', '));
+            }
+            if (toolArgs.length > 0) {
+                span.setAttribute(semantic_convention_1.default.GEN_AI_TOOL_ARGS, toolArgs.join(', '));
+            }
+        }
+        const normalizedToolCalls = result.toolCalls?.map((t) => ({
+            id: t.toolCallId || '',
+            name: t.toolName || '',
+            arguments: t.args || {},
+        }));
+        let inputMessagesJson;
+        let outputMessagesJson;
+        if (captureContent) {
+            outputMessagesJson = helpers_1.default.buildOutputMessages(result.text || '', finishReason, normalizedToolCalls);
+            span.setAttribute(semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES, outputMessagesJson);
+            inputMessagesJson = helpers_1.default.buildInputMessages(messages, params.system);
+        }
+        if (!config_1.default.disableEvents) {
+            const eventAttrs = {
+                [semantic_convention_1.default.GEN_AI_OPERATION]: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT,
+                [semantic_convention_1.default.GEN_AI_REQUEST_MODEL]: modelId,
+                [semantic_convention_1.default.GEN_AI_RESPONSE_MODEL]: responseModel,
+                [semantic_convention_1.default.SERVER_ADDRESS]: VercelAIWrapper.serverAddress,
+                [semantic_convention_1.default.SERVER_PORT]: VercelAIWrapper.serverPort,
+                [semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON]: [finishReason],
+                [semantic_convention_1.default.GEN_AI_OUTPUT_TYPE]: outputType,
+                [semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS]: inputTokens,
+                [semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS]: outputTokens,
+            };
+            if (responseId) {
+                eventAttrs[semantic_convention_1.default.GEN_AI_RESPONSE_ID] = responseId;
+            }
+            if (captureContent) {
+                if (inputMessagesJson)
+                    eventAttrs[semantic_convention_1.default.GEN_AI_INPUT_MESSAGES] = inputMessagesJson;
+                if (outputMessagesJson)
+                    eventAttrs[semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES] = outputMessagesJson;
+            }
+            helpers_1.default.emitInferenceEvent(span, eventAttrs);
+        }
+        return {
+            genAIEndpoint,
+            model: modelId,
+            cost,
+            aiSystem: VercelAIWrapper.aiSystem,
+            serverAddress: VercelAIWrapper.serverAddress,
+            serverPort: VercelAIWrapper.serverPort,
+        };
+    }
 }
-VercelAIWrapper.aiSystem = 'vercel_ai';
+VercelAIWrapper.aiSystem = semantic_convention_1.default.GEN_AI_SYSTEM_VERCEL_AI;
+VercelAIWrapper.serverAddress = 'vercel.ai';
+VercelAIWrapper.serverPort = 443;
 exports.default = VercelAIWrapper;
 //# sourceMappingURL=wrapper.js.map