@paid-ai/paid-node 0.5.0-alpha2 → 0.5.0

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

@@ -8,28 +8,29 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
     });
 };
 import { SpanStatusCode } from "@opentelemetry/api";
-import { getCustomerIdStorage, getAgentIdStorage, getTokenStorage, paidTracer } from "../tracing.mjs";
 import { generateText as originalGenerateText, streamText as originalStreamText, generateObject as originalGenerateObject, streamObject as originalStreamObject, embed as originalEmbed, embedMany as originalEmbedMany, } from "ai";
+import { getPaidTracer, getToken } from "tracing/tracing.js";
+import { getTracingContext } from "tracing/tracingContext.js";
 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("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("claude-")) {
+            return { system: "anthropic", modelName: modelId };
         }
-        if (modelId.startsWith('mistral-') || modelId.startsWith('codestral-')) {
-            return { system: 'mistral', modelName: modelId };
+        if (modelId.startsWith("mistral-") || modelId.startsWith("codestral-")) {
+            return { system: "mistral", modelName: modelId };
         }
-        if (modelId.includes('gemini')) {
-            return { system: 'google', 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' };
+    return { system: "unknown" };
 }
 function extractUsageMetrics(usage) {
     const usageAttrs = {};
@@ -51,14 +52,14 @@ function extractUsageMetrics(usage) {
     return usageAttrs;
 }
 function validateContext() {
-    const externalCustomerId = getCustomerIdStorage();
-    const token = getTokenStorage();
+    const token = getToken();
+    const { externalCustomerId, externalProductId } = getTracingContext();
     if (!token || !externalCustomerId) {
         throw new Error("No token or externalCustomerId: This wrapper should be used inside a callback to paid.trace().");
     }
     return {
         externalCustomerId,
-        externalAgentId: getAgentIdStorage(),
+        externalProductId,
         token,
     };
 }
@@ -66,16 +67,20 @@ export function generateText(params) {
     return __awaiter(this, void 0, void 0, function* () {
         const context = validateContext();
         const { system: aiSystem, modelName } = getModelInfo(params.model);
-        return paidTracer.startActiveSpan("trace.ai-sdk.generateText", (span) => __awaiter(this, void 0, void 0, function* () {
+        const tracer = getPaidTracer();
+        if (!tracer) {
+            throw new Error("Paid tracer is not initialized, Make sure to call 'initializeTracing()' first");
+        }
+        return tracer.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,
+                external_customer_id: context.externalCustomerId,
+                token: context.token,
             };
-            if (context.externalAgentId) {
-                attributes["external_agent_id"] = context.externalAgentId;
+            if (context.externalProductId) {
+                attributes["external_agent_id"] = context.externalProductId;
             }
             if (modelName) {
                 attributes["gen_ai.request.model"] = modelName;
@@ -108,15 +113,19 @@ export function streamText(params) {
     return __awaiter(this, void 0, void 0, function* () {
         const context = validateContext();
         const { system: aiSystem, modelName } = getModelInfo(params.model);
-        return paidTracer.startActiveSpan("trace.ai-sdk.streamText", (span) => __awaiter(this, void 0, void 0, function* () {
+        const tracer = getPaidTracer();
+        if (!tracer) {
+            throw new Error("Paid tracer is not initialized, Make sure to call 'initializeTracing()' first");
+        }
+        return tracer.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,
+                external_customer_id: context.externalCustomerId,
+                token: context.token,
             };
-            if (context.externalAgentId) {
-                attributes["external_agent_id"] = context.externalAgentId;
+            if (context.externalProductId) {
+                attributes["external_agent_id"] = context.externalProductId;
             }
             if (modelName) {
                 attributes["gen_ai.request.model"] = modelName;
@@ -153,16 +162,20 @@ export function generateObject(params) {
     return __awaiter(this, void 0, void 0, function* () {
         const context = validateContext();
         const { system: aiSystem, modelName } = getModelInfo(params.model);
-        return paidTracer.startActiveSpan("trace.ai-sdk.generateObject", (span) => __awaiter(this, void 0, void 0, function* () {
+        const tracer = getPaidTracer();
+        if (!tracer) {
+            throw new Error("Paid tracer is not initialized, Make sure to call 'initializeTracing()' first");
+        }
+        return tracer.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,
+                external_customer_id: context.externalCustomerId,
+                token: context.token,
             };
-            if (context.externalAgentId) {
-                attributes["external_agent_id"] = context.externalAgentId;
+            if (context.externalProductId) {
+                attributes["external_agent_id"] = context.externalProductId;
             }
             if (modelName) {
                 attributes["gen_ai.request.model"] = modelName;
@@ -195,15 +208,19 @@ export function streamObject(params) {
     return __awaiter(this, void 0, void 0, function* () {
         const context = validateContext();
         const { system: aiSystem, modelName } = getModelInfo(params.model);
-        return paidTracer.startActiveSpan("trace.ai-sdk.streamObject", (span) => __awaiter(this, void 0, void 0, function* () {
+        const tracer = getPaidTracer();
+        if (!tracer) {
+            throw new Error("Paid tracer is not initialized, Make sure to call 'initializeTracing()' first");
+        }
+        return tracer.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,
+                external_customer_id: context.externalCustomerId,
+                token: context.token,
             };
-            if (context.externalAgentId) {
-                attributes["external_agent_id"] = context.externalAgentId;
+            if (context.externalProductId) {
+                attributes["external_agent_id"] = context.externalProductId;
             }
             if (modelName) {
                 attributes["gen_ai.request.model"] = modelName;
@@ -237,15 +254,19 @@ export function embed(params) {
     return __awaiter(this, void 0, void 0, function* () {
         const context = validateContext();
         const { system: aiSystem, modelName } = getModelInfo(params.model);
-        return paidTracer.startActiveSpan("trace.ai-sdk.embed", (span) => __awaiter(this, void 0, void 0, function* () {
+        const tracer = getPaidTracer();
+        if (!tracer) {
+            throw new Error("Paid tracer is not initialized, Make sure to call 'initializeTracing()' first");
+        }
+        return tracer.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,
+                external_customer_id: context.externalCustomerId,
+                token: context.token,
             };
-            if (context.externalAgentId) {
-                attributes["external_agent_id"] = context.externalAgentId;
+            if (context.externalProductId) {
+                attributes["external_agent_id"] = context.externalProductId;
             }
             if (modelName) {
                 attributes["gen_ai.request.model"] = modelName;
@@ -275,15 +296,19 @@ export function embedMany(params) {
     return __awaiter(this, void 0, void 0, function* () {
         const context = validateContext();
         const { system: aiSystem, modelName } = getModelInfo(params.model);
-        return paidTracer.startActiveSpan("trace.ai-sdk.embedMany", (span) => __awaiter(this, void 0, void 0, function* () {
+        const tracer = getPaidTracer();
+        if (!tracer) {
+            throw new Error("Paid tracer is not initialized, Make sure to call 'initializeTracing()' first");
+        }
+        return tracer.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,
+                external_customer_id: context.externalCustomerId,
+                token: context.token,
             };
-            if (context.externalAgentId) {
-                attributes["external_agent_id"] = context.externalAgentId;
+            if (context.externalProductId) {
+                attributes["external_agent_id"] = context.externalProductId;
             }
             if (modelName) {
                 attributes["gen_ai.request.model"] = modelName;

package/dist/esm/version.d.mts

@@ -1 +1 @@
-export declare const SDK_VERSION = "0.5.0-alpha2";
+export declare const SDK_VERSION = "0.9.0";

package/dist/esm/version.mjs

@@ -1 +1 @@
-export const SDK_VERSION = "0.5.0-alpha2";
+export const SDK_VERSION = "0.9.0";

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@paid-ai/paid-node",
-    "version": "0.5.0-alpha2",
+    "version": "0.5.0",
     "private": false,
     "repository": "https://github.com/paid-ai/paid-node",
     "type": "commonjs",
@@ -20,6 +20,7 @@
             },
             "default": "./dist/cjs/index.js"
         },
+        "./package.json": "./package.json",
         "./openai": {
             "types": "./dist/cjs/openai-wrapper/index.d.ts",
             "import": {
@@ -104,7 +105,30 @@
             },
             "default": "./dist/cjs/integrations/adapters/nextjs/index.js"
         },
-        "./package.json": "./package.json"
+        "./integrations/express": {
+            "types": "./dist/cjs/integrations/adapters/expressjs/index.d.ts",
+            "import": {
+                "types": "./dist/esm/integrations/adapters/expressjs/index.d.mts",
+                "default": "./dist/esm/integrations/adapters/expressjs/index.mjs"
+            },
+            "require": {
+                "types": "./dist/cjs/integrations/adapters/expressjs/index.d.ts",
+                "default": "./dist/cjs/integrations/adapters/expressjs/index.js"
+            },
+            "default": "./dist/cjs/integrations/adapters/expressjs/index.js"
+        },
+        "./integrations/nestjs": {
+            "types": "./dist/cjs/integrations/adapters/nestjs/index.d.ts",
+            "import": {
+                "types": "./dist/esm/integrations/adapters/nestjs/index.d.mts",
+                "default": "./dist/esm/integrations/adapters/nestjs/index.mjs"
+            },
+            "require": {
+                "types": "./dist/cjs/integrations/adapters/nestjs/index.d.ts",
+                "default": "./dist/cjs/integrations/adapters/nestjs/index.js"
+            },
+            "default": "./dist/cjs/integrations/adapters/nestjs/index.js"
+        }
     },
     "files": [
         "dist",
@@ -128,8 +152,15 @@
         "readable-stream": "^4.5.2",
         "js-base64": "3.7.7",
         "@opentelemetry/api": ">=1.9.0",
-        "@opentelemetry/exporter-trace-otlp-http": ">=0.202.0",
-        "@opentelemetry/sdk-node": ">=0.202.0",
+        "@opentelemetry/exporter-trace-otlp-http": ">=0.208.0",
+        "@opentelemetry/exporter-trace-otlp-proto": ">=0.208.0",
+        "@opentelemetry/sdk-node": ">=0.208.0",
+        "@opentelemetry/sdk-trace-node": "^2.2.0",
+        "@opentelemetry/semantic-conventions": ">=1.38.0",
+        "@arizeai/openinference-instrumentation-anthropic": ">=0.1.2",
+        "@arizeai/openinference-instrumentation-openai": ">=4.0.0",
+        "@arizeai/openinference-semantic-conventions": ">=2.1.2",
+        "@traceloop/instrumentation-bedrock": ">=0.22.0",
         "winston": ">=3.0.0"
     },
     "peerDependencies": {
@@ -139,6 +170,7 @@
         "@mistralai/mistralai": ">=1.0.0",
         "ai": ">=5.0.0",
         "openai": ">=4.85.4",
+        "@types/express": ">=4.17.0",
         "next": ">=13.0.0"
     },
     "peerDependenciesMeta": {
@@ -160,6 +192,9 @@
         "openai": {
             "optional": true
         },
+        "@types/express": {
+            "optional": true
+        },
         "next": {
             "optional": true
         }
@@ -186,6 +221,8 @@
         "@mistralai/mistralai": ">=1.0.0",
         "ai": ">=5.0.0",
         "openai": ">=4.85.4",
+        "@aws-sdk/client-bedrock-runtime": "^3.941.0",
+        "@types/express": "^4.17.21",
         "next": ">=13.0.0"
     },
     "browser": {