@paid-ai/paid-node 0.0.8-alpha2 → 0.0.8-alpha4

package/dist/cjs/Client.d.ts

@@ -44,5 +44,4 @@ export declare class PaidClient {
     initializeTracing(collectorEndpoint?: string): Promise<void>;
     trace<T extends (...args: any[]) => any>(externalCustomerId: string, fn: T, externalAgentId?: string, ...args: Parameters<T>): Promise<ReturnType<T>>;
     signal(eventName: string, data?: Record<string, any>): void;
-    getSpanProcessorAndInitialize(collectorEndpoint?: string): Promise<import("@opentelemetry/sdk-trace-base").SpanProcessor>;
 }

package/dist/cjs/Client.js

@@ -86,9 +86,9 @@ class PaidClient {
         var _a;
         return ((_a = this._usage) !== null && _a !== void 0 ? _a : (this._usage = new BatchUsage_js_1.Usage(this._options)));
     }
-    // Use this method if you want paid to initialize tracing automatically.
-    initializeTracing() {
-        return __awaiter(this, arguments, void 0, function* (collectorEndpoint = "https://collector.agentpaid.io:4318/v1/traces") {
+    // Need to call this method before using tracing or creating wrappers.
+    initializeTracing(collectorEndpoint) {
+        return __awaiter(this, void 0, void 0, function* () {
             const tokenSupplier = this._options.token;
             const token = typeof tokenSupplier === "function" ? yield tokenSupplier() : tokenSupplier;
             const resolvedToken = yield Promise.resolve(token);
@@ -106,14 +106,5 @@ class PaidClient {
     signal(eventName, data) {
         return (0, signal_js_1._signal)(eventName, data);
     }
-    // Use this method if you're setting up OTEL SDK in your code and want to add Paid's span processor.
-    getSpanProcessorAndInitialize() {
-        return __awaiter(this, arguments, void 0, function* (collectorEndpoint = "https://collector.agentpaid.io:4318/v1/traces") {
-            const tokenSupplier = this._options.token;
-            const token = typeof tokenSupplier === "function" ? yield tokenSupplier() : tokenSupplier;
-            const resolvedToken = yield Promise.resolve(token);
-            return (0, tracing_js_1._getSpanProcessorAndInitialize)(resolvedToken, collectorEndpoint);
-        });
-    }
 }
 exports.PaidClient = PaidClient;

package/dist/cjs/tracing/signal.js

@@ -3,22 +3,18 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports._signal = _signal;
 const api_1 = require("@opentelemetry/api");
 const tracing_js_1 = require("./tracing.js");
+const tracing_js_2 = require("./tracing.js");
 function _signal(eventName, data) {
     if (!eventName) {
         throw new Error("Event name is required for signal.");
     }
-    // Check if there's an active span (from _trace())
-    const currentSpan = api_1.trace.getActiveSpan();
-    if (!currentSpan) {
-        throw new Error("Cannot send signal: you should call signal() within trace()");
-    }
     const externalCustomerId = (0, tracing_js_1.getCustomerIdStorage)();
     const externalAgentId = (0, tracing_js_1.getAgentIdStorage)();
     const token = (0, tracing_js_1.getTokenStorage)();
     if (!externalCustomerId || !externalAgentId || !token) {
         throw new Error(`Missing some of: external_customer_id: ${externalCustomerId}, external_agent_id: ${externalAgentId}, or token. Make sure to call signal() within trace()`);
     }
-    const tracer = api_1.trace.getTracer("paid.node");
+    const tracer = tracing_js_2.paidTracer;
     tracer.startActiveSpan("trace.signal", (span) => {
         try {
             const attributes = {

package/dist/cjs/tracing/tracing.d.ts

@@ -1,9 +1,8 @@
-import { SpanProcessor } from "@opentelemetry/sdk-trace-base";
 import winston from "winston";
 export declare const logger: winston.Logger;
+export declare let paidTracer: import("@opentelemetry/api").Tracer;
 export declare const getCustomerIdStorage: () => string | null | undefined;
 export declare const getAgentIdStorage: () => string | null | undefined;
 export declare const getTokenStorage: () => string | null | undefined;
-export declare function _initializeTracing(apiKey: string, collectorEndpoint: string): void;
-export declare function _getSpanProcessorAndInitialize(apiKey: string, collectorEndpoint: string): SpanProcessor;
+export declare function _initializeTracing(apiKey: string, collectorEndpoint?: string): void;
 export declare function _trace<T extends (...args: any[]) => any>(externalCustomerId: string, fn: T, externalAgentId: string | undefined, ...args: Parameters<T>): Promise<ReturnType<T>>;

package/dist/cjs/tracing/tracing.js

@@ -12,11 +12,10 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getTokenStorage = exports.getAgentIdStorage = exports.getCustomerIdStorage = exports.logger = void 0;
+exports.getTokenStorage = exports.getAgentIdStorage = exports.getCustomerIdStorage = exports.paidTracer = exports.logger = void 0;
 exports._initializeTracing = _initializeTracing;
-exports._getSpanProcessorAndInitialize = _getSpanProcessorAndInitialize;
 exports._trace = _trace;
-const sdk_node_1 = require("@opentelemetry/sdk-node");
+const sdk_trace_node_1 = require("@opentelemetry/sdk-trace-node");
 const exporter_trace_otlp_http_1 = require("@opentelemetry/exporter-trace-otlp-http");
 const sdk_trace_base_1 = require("@opentelemetry/sdk-trace-base");
 const api_1 = require("@opentelemetry/api");
@@ -31,6 +30,12 @@ const logLevel = process.env.PAID_LOG_LEVEL;
 if (logLevel) {
     exports.logger.level = logLevel;
 }
+const COLLECTOR_ENDPOINT = process.env.PAID_COLLECTOR_ENDPOINT || "https://collector.agentpaid.io:4318/v1/traces";
+// set up default tracing provider
+let paidExporter = new exporter_trace_otlp_http_1.OTLPTraceExporter({ url: COLLECTOR_ENDPOINT });
+let spanProcessor = new sdk_trace_base_1.BatchSpanProcessor(paidExporter);
+let paidTracerProvider = new sdk_trace_node_1.NodeTracerProvider({ spanProcessors: [spanProcessor] });
+exports.paidTracer = paidTracerProvider.getTracer("paid.node");
 // storage for passing info to child spans
 const customerIdStorage = new async_hooks_1.AsyncLocalStorage();
 const agentIdStorage = new async_hooks_1.AsyncLocalStorage();
@@ -63,19 +68,6 @@ function setupGracefulShutdown(shuttable) {
         });
     });
 }
-function isOTELInitialized() {
-    const provider = api_1.trace.getTracerProvider();
-    if (!provider) {
-        return false;
-    }
-    if (provider.constructor.name !== "ProxyTracerProvider" && provider.constructor.name !== "NodeTracerProvider") {
-        return true;
-    }
-    if (provider._delegate) {
-        return provider._delegate.constructor.name !== "NoopTracerProvider";
-    }
-    return false;
-}
 function _initializeTracing(apiKey, collectorEndpoint) {
     if (getToken()) {
         throw new Error("Tracing SDK is already initialized.");
@@ -83,40 +75,23 @@ function _initializeTracing(apiKey, collectorEndpoint) {
     if (!apiKey) {
         throw new Error("Cannot initialize tracing SDK - first initialize the PaidClient with an API key");
     }
-    try {
-        new URL(collectorEndpoint);
-    }
-    catch (_a) {
-        throw new Error(`Collector endpoint [${collectorEndpoint}] must be a valid URL`);
-    }
-    if (isOTELInitialized()) {
-        throw new Error("OTEL SDK is already initialized with a different provider.\n" +
-            "If you're already using OTEL in your code - " +
-            "please use function Paid.getSpanProcessorAndInitialize() " +
-            "and add the span processor to your OTEL SDK initialization");
+    if (collectorEndpoint) {
+        paidExporter = new exporter_trace_otlp_http_1.OTLPTraceExporter({ url: collectorEndpoint });
+        spanProcessor = new sdk_trace_base_1.BatchSpanProcessor(paidExporter);
+        paidTracerProvider = new sdk_trace_node_1.NodeTracerProvider({ spanProcessors: [spanProcessor] });
+        exports.paidTracer = paidTracerProvider.getTracer("paid.node");
     }
-    const sdk = new sdk_node_1.NodeSDK({
-        traceExporter: new exporter_trace_otlp_http_1.OTLPTraceExporter({
-            url: collectorEndpoint,
-        }),
-    });
-    sdk.start();
-    setupGracefulShutdown(sdk);
-    setToken(apiKey);
-    exports.logger.info(`Paid tracing SDK initialized with collector endpoint: ${collectorEndpoint}`);
-}
-function _getSpanProcessorAndInitialize(apiKey, collectorEndpoint) {
-    setToken(apiKey);
-    const spanProcessor = new sdk_trace_base_1.BatchSpanProcessor(new exporter_trace_otlp_http_1.OTLPTraceExporter({ url: collectorEndpoint }));
     setupGracefulShutdown(spanProcessor);
-    return spanProcessor;
+    setToken(apiKey);
+    const collectorAddress = collectorEndpoint || COLLECTOR_ENDPOINT;
+    exports.logger.info(`Paid tracing SDK initialized with collector endpoint: ${collectorAddress}`);
 }
 function _trace(externalCustomerId, fn, externalAgentId, ...args) {
     return __awaiter(this, void 0, void 0, function* () {
-        const tracer = api_1.trace.getTracer("paid.node");
+        const tracer = exports.paidTracer;
         const token = getToken();
         if (!token || !externalCustomerId) {
-            throw new Error(`Token [${token}] or external customer ID [${externalCustomerId}] is missing`);
+            throw new Error(`Paid tracing is not initialized. Make sure to call initializeTracing() first.`);
         }
         return tracer.startActiveSpan("paid.node", (span) => __awaiter(this, void 0, void 0, function* () {
             span.setAttribute("external_customer_id", externalCustomerId);

package/dist/cjs/tracing/wrappers/anthropicWrapper.js

@@ -15,7 +15,7 @@ const tracing_js_1 = require("../tracing.js");
 class PaidAnthropic {
     constructor(anthropicClient) {
         this.anthropic = anthropicClient;
-        this.tracer = api_1.trace.getTracer("paid.node");
+        this.tracer = tracing_js_1.paidTracer;
     }
     get messages() {
         return new MessagesWrapper(this.anthropic, this.tracer);
@@ -29,10 +29,6 @@ class MessagesWrapper {
     }
     create(params) {
         return __awaiter(this, void 0, void 0, function* () {
-            const currentSpan = api_1.trace.getSpan(api_1.context.active());
-            if (!currentSpan) {
-                throw new Error("No active span found, make sure to call this inside of a callback to paid.trace().");
-            }
             const externalCustomerId = (0, tracing_js_1.getCustomerIdStorage)();
             const externalAgentId = (0, tracing_js_1.getAgentIdStorage)();
             const token = (0, tracing_js_1.getTokenStorage)();

package/dist/cjs/tracing/wrappers/langchainCallback.js

@@ -18,7 +18,7 @@ class PaidLangChainCallback extends base_1.BaseCallbackHandler {
         super();
         this.name = "PaidLangChainCallback";
         this.spans = new Map();
-        this.tracer = api_1.trace.getTracer("paid.node");
+        this.tracer = tracing_js_1.paidTracer;
     }
     extractProvider(serialized) {
         if (!(serialized === null || serialized === void 0 ? void 0 : serialized.id))
@@ -40,10 +40,6 @@ class PaidLangChainCallback extends base_1.BaseCallbackHandler {
     }
     handleLLMStart(llm, prompts, runId, parentRunId, extraParams, tags, metadata) {
         return __awaiter(this, void 0, void 0, function* () {
-            const currentSpan = api_1.trace.getSpan(api_1.context.active());
-            if (!currentSpan) {
-                throw new Error("No active span found, make sure to call this inside of a callback to paid.trace().");
-            }
             const externalCustomerId = (0, tracing_js_1.getCustomerIdStorage)();
             const externalAgentId = (0, tracing_js_1.getAgentIdStorage)();
             const token = (0, tracing_js_1.getTokenStorage)();

package/dist/cjs/tracing/wrappers/mistralWrapper.js

@@ -15,7 +15,7 @@ const tracing_js_1 = require("../tracing.js");
 class PaidMistral {
     constructor(mistralClient) {
         this.mistral = mistralClient;
-        this.tracer = api_1.trace.getTracer("paid.node");
+        this.tracer = tracing_js_1.paidTracer;
     }
     get ocr() {
         return new OCRWrapper(this.mistral, this.tracer);
@@ -29,10 +29,6 @@ class OCRWrapper {
     }
     process(request, options) {
         return __awaiter(this, void 0, void 0, function* () {
-            const currentSpan = api_1.trace.getSpan(api_1.context.active());
-            if (!currentSpan) {
-                throw new Error("No active span found, make sure to call this inside of a callback to paid.trace().");
-            }
             const externalCustomerId = (0, tracing_js_1.getCustomerIdStorage)();
             const externalAgentId = (0, tracing_js_1.getAgentIdStorage)();
             const token = (0, tracing_js_1.getTokenStorage)();

package/dist/cjs/tracing/wrappers/openAiWrapper.js

@@ -15,7 +15,7 @@ const tracing_js_1 = require("../tracing.js");
 class PaidOpenAI {
     constructor(openaiClient) {
         this.openai = openaiClient;
-        this.tracer = api_1.trace.getTracer("paid.node");
+        this.tracer = tracing_js_1.paidTracer;
     }
     get chat() {
         return new ChatWrapper(this.openai, this.tracer);
@@ -47,10 +47,6 @@ class ChatCompletionsWrapper {
     }
     create(params) {
         return __awaiter(this, void 0, void 0, function* () {
-            const currentSpan = api_1.trace.getSpan(api_1.context.active());
-            if (!currentSpan) {
-                throw new Error("No active span found, make sure to call this inside of a callback to paid.trace().");
-            }
             const externalCustomerId = (0, tracing_js_1.getCustomerIdStorage)();
             const externalAgentId = (0, tracing_js_1.getAgentIdStorage)();
             const token = (0, tracing_js_1.getTokenStorage)();
@@ -99,10 +95,6 @@ class ResponsesWrapper {
     }
     create(params) {
         return __awaiter(this, void 0, void 0, function* () {
-            const currentSpan = api_1.trace.getSpan(api_1.context.active());
-            if (!currentSpan) {
-                throw new Error("No active span found, make sure to call this inside of a callback to paid.trace().");
-            }
             const externalCustomerId = (0, tracing_js_1.getCustomerIdStorage)();
             const externalAgentId = (0, tracing_js_1.getAgentIdStorage)();
             const token = (0, tracing_js_1.getTokenStorage)();
@@ -158,10 +150,6 @@ class EmbeddingsWrapper {
     }
     create(params) {
         return __awaiter(this, void 0, void 0, function* () {
-            const currentSpan = api_1.trace.getSpan(api_1.context.active());
-            if (!currentSpan) {
-                throw new Error("No active span found, make sure to call this inside of a callback to paid.trace().");
-            }
             const externalCustomerId = (0, tracing_js_1.getCustomerIdStorage)();
             const externalAgentId = (0, tracing_js_1.getAgentIdStorage)();
             const token = (0, tracing_js_1.getTokenStorage)();
@@ -210,10 +198,6 @@ class ImagesWrapper {
     generate(params) {
         return __awaiter(this, void 0, void 0, function* () {
             var _a;
-            const currentSpan = api_1.trace.getSpan(api_1.context.active());
-            if (!currentSpan) {
-                throw new Error("No active span found, make sure to call this inside of a callback to paid.trace().");
-            }
             const externalCustomerId = (0, tracing_js_1.getCustomerIdStorage)();
             const externalAgentId = (0, tracing_js_1.getAgentIdStorage)();
             const token = (0, tracing_js_1.getTokenStorage)();

package/dist/cjs/tracing/wrappers/vercelAIWrapper.d.ts

@@ -0,0 +1,22 @@
+import { generateText as originalGenerateText, streamText as originalStreamText, generateObject as originalGenerateObject, streamObject as originalStreamObject, embed as originalEmbed, embedMany as originalEmbedMany } from "ai";
+type GenerateTextParams = Parameters<typeof originalGenerateText>[0];
+type StreamTextParams = Parameters<typeof originalStreamText>[0];
+type GenerateObjectParams = Parameters<typeof originalGenerateObject>[0];
+type StreamObjectParams = Parameters<typeof originalStreamObject>[0];
+type EmbedParams = Parameters<typeof originalEmbed>[0];
+type EmbedManyParams = Parameters<typeof originalEmbedMany>[0];
+export declare function generateText(params: GenerateTextParams): Promise<ReturnType<typeof originalGenerateText>>;
+export declare function streamText(params: StreamTextParams): Promise<ReturnType<typeof originalStreamText>>;
+export declare function generateObject(params: GenerateObjectParams): Promise<ReturnType<typeof originalGenerateObject>>;
+export declare function streamObject(params: StreamObjectParams): Promise<ReturnType<typeof originalStreamObject>>;
+export declare function embed(params: EmbedParams): Promise<ReturnType<typeof originalEmbed>>;
+export declare function embedMany(params: EmbedManyParams): Promise<ReturnType<typeof originalEmbedMany>>;
+declare const _default: {
+    generateText: typeof generateText;
+    streamText: typeof streamText;
+    generateObject: typeof generateObject;
+    streamObject: typeof streamObject;
+    embed: typeof embed;
+    embedMany: typeof embedMany;
+};
+export default _default;

package/dist/cjs/tracing/wrappers/vercelAIWrapper.js

@@ -0,0 +1,327 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.generateText = generateText;
+exports.streamText = streamText;
+exports.generateObject = generateObject;
+exports.streamObject = streamObject;
+exports.embed = embed;
+exports.embedMany = embedMany;
+const api_1 = require("@opentelemetry/api");
+const tracing_js_1 = require("../tracing.js");
+const ai_1 = require("ai");
+function getModelInfo(model) {
+    if (model === null || model === void 0 ? void 0 : model.modelId) {
+        const modelId = model.modelId;
+        if (modelId.startsWith('gpt-') || modelId.startsWith('text-embedding-') || modelId.startsWith('dall-e-')) {
+            return { system: 'openai', modelName: modelId };
+        }
+        if (modelId.startsWith('claude-')) {
+            return { system: 'anthropic', modelName: modelId };
+        }
+        if (modelId.startsWith('mistral-') || modelId.startsWith('codestral-')) {
+            return { system: 'mistral', modelName: modelId };
+        }
+        if (modelId.includes('gemini')) {
+            return { system: 'google', modelName: modelId };
+        }
+    }
+    if (model === null || model === void 0 ? void 0 : model.provider) {
+        return { system: model.provider, modelName: model.modelId };
+    }
+    return { system: 'unknown' };
+}
+function extractUsageMetrics(usage) {
+    const usageAttrs = {};
+    const inputTokens = usage.promptTokens || usage.prompt_tokens || usage.inputTokens;
+    const outputTokens = usage.completionTokens || usage.completion_tokens || usage.outputTokens;
+    const cachedTokens = usage.cachedPromptTokens || usage.cached_prompt_tokens || usage.cachedInputTokens;
+    if (inputTokens !== undefined) {
+        usageAttrs["gen_ai.usage.input_tokens"] = inputTokens;
+    }
+    if (outputTokens !== undefined) {
+        usageAttrs["gen_ai.usage.output_tokens"] = outputTokens;
+    }
+    if (cachedTokens !== undefined) {
+        usageAttrs["gen_ai.usage.cached_input_tokens"] = cachedTokens;
+    }
+    if (usage.tokens !== undefined && inputTokens === undefined) {
+        usageAttrs["gen_ai.usage.input_tokens"] = usage.tokens;
+    }
+    return usageAttrs;
+}
+function validateContext() {
+    const externalCustomerId = (0, tracing_js_1.getCustomerIdStorage)();
+    const token = (0, tracing_js_1.getTokenStorage)();
+    if (!token || !externalCustomerId) {
+        throw new Error("No token or externalCustomerId: This wrapper should be used inside a callback to paid.trace().");
+    }
+    return {
+        externalCustomerId,
+        externalAgentId: (0, tracing_js_1.getAgentIdStorage)(),
+        token,
+    };
+}
+function generateText(params) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const context = validateContext();
+        const { system: aiSystem, modelName } = getModelInfo(params.model);
+        return tracing_js_1.paidTracer.startActiveSpan("trace.ai-sdk.generateText", (span) => __awaiter(this, void 0, void 0, function* () {
+            var _a;
+            const attributes = {
+                "gen_ai.system": aiSystem,
+                "gen_ai.operation.name": "chat",
+                "external_customer_id": context.externalCustomerId,
+                "token": context.token,
+            };
+            if (context.externalAgentId) {
+                attributes["external_agent_id"] = context.externalAgentId;
+            }
+            if (modelName) {
+                attributes["gen_ai.request.model"] = modelName;
+            }
+            span.setAttributes(attributes);
+            try {
+                const result = yield (0, ai_1.generateText)(params);
+                if (result.usage) {
+                    const usageAttrs = extractUsageMetrics(result.usage);
+                    span.setAttributes(usageAttrs);
+                }
+                if ((_a = result.response) === null || _a === void 0 ? void 0 : _a.modelId) {
+                    span.setAttribute("gen_ai.response.model", result.response.modelId);
+                }
+                span.setStatus({ code: api_1.SpanStatusCode.OK });
+                return result;
+            }
+            catch (error) {
+                span.setStatus({ code: api_1.SpanStatusCode.ERROR, message: error.message });
+                span.recordException(error);
+                throw error;
+            }
+            finally {
+                span.end();
+            }
+        }));
+    });
+}
+function streamText(params) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const context = validateContext();
+        const { system: aiSystem, modelName } = getModelInfo(params.model);
+        return tracing_js_1.paidTracer.startActiveSpan("trace.ai-sdk.streamText", (span) => __awaiter(this, void 0, void 0, function* () {
+            const attributes = {
+                "gen_ai.system": aiSystem,
+                "gen_ai.operation.name": "chat",
+                "external_customer_id": context.externalCustomerId,
+                "token": context.token,
+            };
+            if (context.externalAgentId) {
+                attributes["external_agent_id"] = context.externalAgentId;
+            }
+            if (modelName) {
+                attributes["gen_ai.request.model"] = modelName;
+            }
+            span.setAttributes(attributes);
+            try {
+                const originalOnFinish = params.onFinish;
+                const wrappedParams = Object.assign(Object.assign({}, params), { onFinish: (result) => {
+                        if (result.usage) {
+                            const usageAttrs = extractUsageMetrics(result.usage);
+                            span.setAttributes(usageAttrs);
+                        }
+                        if (originalOnFinish) {
+                            originalOnFinish(result);
+                        }
+                        span.setStatus({ code: api_1.SpanStatusCode.OK });
+                        span.end();
+                    } });
+                const result = (0, ai_1.streamText)(wrappedParams);
+                return result;
+            }
+            catch (error) {
+                span.setStatus({ code: api_1.SpanStatusCode.ERROR, message: error.message });
+                span.recordException(error);
+                throw error;
+            }
+            finally {
+                span.end();
+            }
+        }));
+    });
+}
+function generateObject(params) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const context = validateContext();
+        const { system: aiSystem, modelName } = getModelInfo(params.model);
+        return tracing_js_1.paidTracer.startActiveSpan("trace.ai-sdk.generateObject", (span) => __awaiter(this, void 0, void 0, function* () {
+            var _a;
+            const attributes = {
+                "gen_ai.system": aiSystem,
+                "gen_ai.operation.name": "chat",
+                "external_customer_id": context.externalCustomerId,
+                "token": context.token,
+            };
+            if (context.externalAgentId) {
+                attributes["external_agent_id"] = context.externalAgentId;
+            }
+            if (modelName) {
+                attributes["gen_ai.request.model"] = modelName;
+            }
+            span.setAttributes(attributes);
+            try {
+                const result = yield (0, ai_1.generateObject)(params);
+                if (result.usage) {
+                    const usageAttrs = extractUsageMetrics(result.usage);
+                    span.setAttributes(usageAttrs);
+                }
+                if ((_a = result.response) === null || _a === void 0 ? void 0 : _a.modelId) {
+                    span.setAttribute("gen_ai.response.model", result.response.modelId);
+                }
+                span.setStatus({ code: api_1.SpanStatusCode.OK });
+                return result;
+            }
+            catch (error) {
+                span.setStatus({ code: api_1.SpanStatusCode.ERROR, message: error.message });
+                span.recordException(error);
+                throw error;
+            }
+            finally {
+                span.end();
+            }
+        }));
+    });
+}
+function streamObject(params) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const context = validateContext();
+        const { system: aiSystem, modelName } = getModelInfo(params.model);
+        return tracing_js_1.paidTracer.startActiveSpan("trace.ai-sdk.streamObject", (span) => __awaiter(this, void 0, void 0, function* () {
+            const attributes = {
+                "gen_ai.system": aiSystem,
+                "gen_ai.operation.name": "chat",
+                "external_customer_id": context.externalCustomerId,
+                "token": context.token,
+            };
+            if (context.externalAgentId) {
+                attributes["external_agent_id"] = context.externalAgentId;
+            }
+            if (modelName) {
+                attributes["gen_ai.request.model"] = modelName;
+            }
+            span.setAttributes(attributes);
+            try {
+                const originalOnFinish = params.onFinish;
+                const wrappedParams = Object.assign(Object.assign({}, params), { onFinish: (result) => {
+                        if (result.usage) {
+                            const usageAttrs = extractUsageMetrics(result.usage);
+                            span.setAttributes(usageAttrs);
+                        }
+                        if (originalOnFinish) {
+                            originalOnFinish(result);
+                        }
+                        span.setStatus({ code: api_1.SpanStatusCode.OK });
+                        span.end();
+                    } });
+                const result = (0, ai_1.streamObject)(wrappedParams);
+                return result;
+            }
+            catch (error) {
+                span.setStatus({ code: api_1.SpanStatusCode.ERROR, message: error.message });
+                span.recordException(error);
+                throw error;
+            }
+        }));
+    });
+}
+function embed(params) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const context = validateContext();
+        const { system: aiSystem, modelName } = getModelInfo(params.model);
+        return tracing_js_1.paidTracer.startActiveSpan("trace.ai-sdk.embed", (span) => __awaiter(this, void 0, void 0, function* () {
+            const attributes = {
+                "gen_ai.system": aiSystem,
+                "gen_ai.operation.name": "embeddings",
+                "external_customer_id": context.externalCustomerId,
+                "token": context.token,
+            };
+            if (context.externalAgentId) {
+                attributes["external_agent_id"] = context.externalAgentId;
+            }
+            if (modelName) {
+                attributes["gen_ai.request.model"] = modelName;
+            }
+            span.setAttributes(attributes);
+            try {
+                const result = yield (0, ai_1.embed)(params);
+                if (result.usage) {
+                    const usageAttrs = extractUsageMetrics(result.usage);
+                    span.setAttributes(usageAttrs);
+                }
+                span.setStatus({ code: api_1.SpanStatusCode.OK });
+                return result;
+            }
+            catch (error) {
+                span.setStatus({ code: api_1.SpanStatusCode.ERROR, message: error.message });
+                span.recordException(error);
+                throw error;
+            }
+            finally {
+                span.end();
+            }
+        }));
+    });
+}
+function embedMany(params) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const context = validateContext();
+        const { system: aiSystem, modelName } = getModelInfo(params.model);
+        return tracing_js_1.paidTracer.startActiveSpan("trace.ai-sdk.embedMany", (span) => __awaiter(this, void 0, void 0, function* () {
+            const attributes = {
+                "gen_ai.system": aiSystem,
+                "gen_ai.operation.name": "embeddings",
+                "external_customer_id": context.externalCustomerId,
+                "token": context.token,
+            };
+            if (context.externalAgentId) {
+                attributes["external_agent_id"] = context.externalAgentId;
+            }
+            if (modelName) {
+                attributes["gen_ai.request.model"] = modelName;
+            }
+            span.setAttributes(attributes);
+            try {
+                const result = yield (0, ai_1.embedMany)(params);
+                if (result.usage) {
+                    const usageAttrs = extractUsageMetrics(result.usage);
+                    span.setAttributes(usageAttrs);
+                }
+                span.setStatus({ code: api_1.SpanStatusCode.OK });
+                return result;
+            }
+            catch (error) {
+                span.setStatus({ code: api_1.SpanStatusCode.ERROR, message: error.message });
+                span.recordException(error);
+                throw error;
+            }
+            finally {
+                span.end();
+            }
+        }));
+    });
+}
+exports.default = {
+    generateText,
+    streamText,
+    generateObject,
+    streamObject,
+    embed,
+    embedMany,
+};

package/dist/cjs/vercel/index.d.ts

@@ -0,0 +1 @@
+export { generateText as paidGenerateText, streamText as paidStreamText, generateObject as paidGenerateObject, streamObject as paidStreamObject, embed as paidEmbed, embedMany as paidEmbedMany, } from "../tracing/wrappers/vercelAIWrapper.js";

package/dist/cjs/vercel/index.js

@@ -0,0 +1,10 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.paidEmbedMany = exports.paidEmbed = exports.paidStreamObject = exports.paidGenerateObject = exports.paidStreamText = exports.paidGenerateText = void 0;
+var vercelAIWrapper_js_1 = require("../tracing/wrappers/vercelAIWrapper.js");
+Object.defineProperty(exports, "paidGenerateText", { enumerable: true, get: function () { return vercelAIWrapper_js_1.generateText; } });
+Object.defineProperty(exports, "paidStreamText", { enumerable: true, get: function () { return vercelAIWrapper_js_1.streamText; } });
+Object.defineProperty(exports, "paidGenerateObject", { enumerable: true, get: function () { return vercelAIWrapper_js_1.generateObject; } });
+Object.defineProperty(exports, "paidStreamObject", { enumerable: true, get: function () { return vercelAIWrapper_js_1.streamObject; } });
+Object.defineProperty(exports, "paidEmbed", { enumerable: true, get: function () { return vercelAIWrapper_js_1.embed; } });
+Object.defineProperty(exports, "paidEmbedMany", { enumerable: true, get: function () { return vercelAIWrapper_js_1.embedMany; } });

package/dist/cjs/version.d.ts

@@ -1 +1 @@
-export declare const SDK_VERSION = "0.0.8-alpha2";
+export declare const SDK_VERSION = "0.0.8-alpha4";

package/dist/cjs/version.js

@@ -1,4 +1,4 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.SDK_VERSION = void 0;
-exports.SDK_VERSION = "0.0.8-alpha2";
+exports.SDK_VERSION = "0.0.8-alpha4";