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

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/version.d.ts

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

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-alpha3";

package/dist/esm/Client.d.mts

@@ -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/esm/Client.mjs

@@ -17,7 +17,7 @@ import { Agents } from "./api/resources/agents/client/Client.mjs";
 import { Contacts } from "./api/resources/contacts/client/Client.mjs";
 import { Orders } from "./api/resources/orders/client/Client.mjs";
 import { Usage } from "./wrapper/BatchUsage.mjs";
-import { _trace, _initializeTracing, _getSpanProcessorAndInitialize } from "./tracing/tracing.mjs";
+import { _trace, _initializeTracing } from "./tracing/tracing.mjs";
 import { _signal } from "./tracing/signal.mjs";
 export class PaidClient {
     constructor(_options) {
@@ -50,9 +50,9 @@ export class PaidClient {
         var _a;
         return ((_a = this._usage) !== null && _a !== void 0 ? _a : (this._usage = new 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);
@@ -70,13 +70,4 @@ export class PaidClient {
     signal(eventName, data) {
         return _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 _getSpanProcessorAndInitialize(resolvedToken, collectorEndpoint);
-        });
-    }
 }

package/dist/esm/tracing/signal.mjs

@@ -1,21 +1,17 @@
-import { trace, SpanStatusCode } from "@opentelemetry/api";
+import { SpanStatusCode } from "@opentelemetry/api";
 import { getCustomerIdStorage, getAgentIdStorage, getTokenStorage } from "./tracing.mjs";
+import { paidTracer } from "./tracing.mjs";
 export 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 = trace.getActiveSpan();
-    if (!currentSpan) {
-        throw new Error("Cannot send signal: you should call signal() within trace()");
-    }
     const externalCustomerId = getCustomerIdStorage();
     const externalAgentId = getAgentIdStorage();
     const token = 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 = trace.getTracer("paid.node");
+    const tracer = paidTracer;
     tracer.startActiveSpan("trace.signal", (span) => {
         try {
             const attributes = {

package/dist/esm/tracing/tracing.d.mts

@@ -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/esm/tracing/tracing.mjs

@@ -7,10 +7,10 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         step((generator = generator.apply(thisArg, _arguments || [])).next());
     });
 };
-import { NodeSDK } from "@opentelemetry/sdk-node";
+import { NodeTracerProvider } from "@opentelemetry/sdk-trace-node";
 import { OTLPTraceExporter } from "@opentelemetry/exporter-trace-otlp-http";
 import { BatchSpanProcessor } from "@opentelemetry/sdk-trace-base";
-import { trace, SpanStatusCode } from "@opentelemetry/api";
+import { SpanStatusCode } from "@opentelemetry/api";
 import { AsyncLocalStorage } from "async_hooks";
 import winston from "winston";
 export const logger = winston.createLogger({
@@ -22,6 +22,12 @@ const logLevel = process.env.PAID_LOG_LEVEL;
 if (logLevel) {
     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 OTLPTraceExporter({ url: COLLECTOR_ENDPOINT });
+let spanProcessor = new BatchSpanProcessor(paidExporter);
+let paidTracerProvider = new NodeTracerProvider({ spanProcessors: [spanProcessor] });
+export let paidTracer = paidTracerProvider.getTracer("paid.node");
 // storage for passing info to child spans
 const customerIdStorage = new AsyncLocalStorage();
 const agentIdStorage = new AsyncLocalStorage();
@@ -51,19 +57,6 @@ function setupGracefulShutdown(shuttable) {
         });
     });
 }
-function isOTELInitialized() {
-    const provider = 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;
-}
 export function _initializeTracing(apiKey, collectorEndpoint) {
     if (getToken()) {
         throw new Error("Tracing SDK is already initialized.");
@@ -71,40 +64,23 @@ export 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 OTLPTraceExporter({ url: collectorEndpoint });
+        spanProcessor = new BatchSpanProcessor(paidExporter);
+        paidTracerProvider = new NodeTracerProvider({ spanProcessors: [spanProcessor] });
+        paidTracer = paidTracerProvider.getTracer("paid.node");
     }
-    const sdk = new NodeSDK({
-        traceExporter: new OTLPTraceExporter({
-            url: collectorEndpoint,
-        }),
-    });
-    sdk.start();
-    setupGracefulShutdown(sdk);
-    setToken(apiKey);
-    logger.info(`Paid tracing SDK initialized with collector endpoint: ${collectorEndpoint}`);
-}
-export function _getSpanProcessorAndInitialize(apiKey, collectorEndpoint) {
-    setToken(apiKey);
-    const spanProcessor = new BatchSpanProcessor(new OTLPTraceExporter({ url: collectorEndpoint }));
     setupGracefulShutdown(spanProcessor);
-    return spanProcessor;
+    setToken(apiKey);
+    const collectorAddress = collectorEndpoint || COLLECTOR_ENDPOINT;
+    logger.info(`Paid tracing SDK initialized with collector endpoint: ${collectorAddress}`);
 }
 export function _trace(externalCustomerId, fn, externalAgentId, ...args) {
     return __awaiter(this, void 0, void 0, function* () {
-        const tracer = trace.getTracer("paid.node");
+        const tracer = 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/esm/tracing/wrappers/anthropicWrapper.mjs

@@ -7,12 +7,12 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         step((generator = generator.apply(thisArg, _arguments || [])).next());
     });
 };
-import { trace, SpanStatusCode, context } from "@opentelemetry/api";
-import { getCustomerIdStorage, getAgentIdStorage, getTokenStorage } from "../tracing.mjs";
+import { SpanStatusCode } from "@opentelemetry/api";
+import { getCustomerIdStorage, getAgentIdStorage, getTokenStorage, paidTracer } from "../tracing.mjs";
 export class PaidAnthropic {
     constructor(anthropicClient) {
         this.anthropic = anthropicClient;
-        this.tracer = trace.getTracer("paid.node");
+        this.tracer = paidTracer;
     }
     get messages() {
         return new MessagesWrapper(this.anthropic, this.tracer);
@@ -25,10 +25,6 @@ class MessagesWrapper {
     }
     create(params) {
         return __awaiter(this, void 0, void 0, function* () {
-            const currentSpan = trace.getSpan(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 = getCustomerIdStorage();
             const externalAgentId = getAgentIdStorage();
             const token = getTokenStorage();

package/dist/esm/tracing/wrappers/langchainCallback.mjs

@@ -8,14 +8,14 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
     });
 };
 import { BaseCallbackHandler } from "@langchain/core/callbacks/base";
-import { trace, SpanStatusCode, context } from "@opentelemetry/api";
-import { getCustomerIdStorage, getAgentIdStorage, getTokenStorage } from "../tracing.mjs";
+import { SpanStatusCode } from "@opentelemetry/api";
+import { getCustomerIdStorage, getAgentIdStorage, getTokenStorage, paidTracer } from "../tracing.mjs";
 export class PaidLangChainCallback extends BaseCallbackHandler {
     constructor() {
         super();
         this.name = "PaidLangChainCallback";
         this.spans = new Map();
-        this.tracer = trace.getTracer("paid.node");
+        this.tracer = paidTracer;
     }
     extractProvider(serialized) {
         if (!(serialized === null || serialized === void 0 ? void 0 : serialized.id))
@@ -37,10 +37,6 @@ export class PaidLangChainCallback extends BaseCallbackHandler {
     }
     handleLLMStart(llm, prompts, runId, parentRunId, extraParams, tags, metadata) {
         return __awaiter(this, void 0, void 0, function* () {
-            const currentSpan = trace.getSpan(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 = getCustomerIdStorage();
             const externalAgentId = getAgentIdStorage();
             const token = getTokenStorage();

package/dist/esm/tracing/wrappers/mistralWrapper.mjs

@@ -7,12 +7,12 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         step((generator = generator.apply(thisArg, _arguments || [])).next());
     });
 };
-import { trace, SpanStatusCode, context } from "@opentelemetry/api";
-import { getCustomerIdStorage, getAgentIdStorage, getTokenStorage } from "../tracing.mjs";
+import { SpanStatusCode } from "@opentelemetry/api";
+import { getCustomerIdStorage, getAgentIdStorage, getTokenStorage, paidTracer } from "../tracing.mjs";
 export class PaidMistral {
     constructor(mistralClient) {
         this.mistral = mistralClient;
-        this.tracer = trace.getTracer("paid.node");
+        this.tracer = paidTracer;
     }
     get ocr() {
         return new OCRWrapper(this.mistral, this.tracer);
@@ -25,10 +25,6 @@ class OCRWrapper {
     }
     process(request, options) {
         return __awaiter(this, void 0, void 0, function* () {
-            const currentSpan = trace.getSpan(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 = getCustomerIdStorage();
             const externalAgentId = getAgentIdStorage();
             const token = getTokenStorage();

package/dist/esm/tracing/wrappers/openAiWrapper.mjs

@@ -7,12 +7,12 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         step((generator = generator.apply(thisArg, _arguments || [])).next());
     });
 };
-import { trace, SpanStatusCode, context } from "@opentelemetry/api";
-import { getCustomerIdStorage, getAgentIdStorage, getTokenStorage } from "../tracing.mjs";
+import { SpanStatusCode } from "@opentelemetry/api";
+import { getCustomerIdStorage, getAgentIdStorage, getTokenStorage, paidTracer } from "../tracing.mjs";
 export class PaidOpenAI {
     constructor(openaiClient) {
         this.openai = openaiClient;
-        this.tracer = trace.getTracer("paid.node");
+        this.tracer = paidTracer;
     }
     get chat() {
         return new ChatWrapper(this.openai, this.tracer);
@@ -43,10 +43,6 @@ class ChatCompletionsWrapper {
     }
     create(params) {
         return __awaiter(this, void 0, void 0, function* () {
-            const currentSpan = trace.getSpan(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 = getCustomerIdStorage();
             const externalAgentId = getAgentIdStorage();
             const token = getTokenStorage();
@@ -95,10 +91,6 @@ class ResponsesWrapper {
     }
     create(params) {
         return __awaiter(this, void 0, void 0, function* () {
-            const currentSpan = trace.getSpan(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 = getCustomerIdStorage();
             const externalAgentId = getAgentIdStorage();
             const token = getTokenStorage();
@@ -154,10 +146,6 @@ class EmbeddingsWrapper {
     }
     create(params) {
         return __awaiter(this, void 0, void 0, function* () {
-            const currentSpan = trace.getSpan(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 = getCustomerIdStorage();
             const externalAgentId = getAgentIdStorage();
             const token = getTokenStorage();
@@ -206,10 +194,6 @@ class ImagesWrapper {
     generate(params) {
         return __awaiter(this, void 0, void 0, function* () {
             var _a;
-            const currentSpan = trace.getSpan(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 = getCustomerIdStorage();
             const externalAgentId = getAgentIdStorage();
             const token = getTokenStorage();

package/dist/esm/version.d.mts

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

package/dist/esm/version.mjs

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

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@paid-ai/paid-node",
-    "version": "0.0.8-alpha2",
+    "version": "0.0.8-alpha3",
     "private": false,
     "repository": "https://github.com/AgentPaid/paid-node",
     "type": "commonjs",