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

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

@@ -8,7 +8,7 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
     });
 };
 import { trace, SpanStatusCode, context } from "@opentelemetry/api";
-import { getCustomerId, getTokenStorage } from "../tracing";
+import { getCustomerIdStorage, getAgentIdStorage, getTokenStorage } from "../tracing.mjs";
 export class PaidOpenAI {
     constructor(openaiClient) {
         this.openai = openaiClient;
@@ -45,23 +45,23 @@ class ChatCompletionsWrapper {
         return __awaiter(this, void 0, void 0, function* () {
             const currentSpan = trace.getSpan(context.active());
             if (!currentSpan) {
-                console.warn("No active span found, calling OpenAI directly without tracing.");
-                return this.openai.chat.completions.create(params);
+                throw new Error("No active span found, make sure to call this inside of a callback to paid.trace().");
             }
-            const externalCustomerId = getCustomerId();
+            const externalCustomerId = getCustomerIdStorage();
+            const externalAgentId = getAgentIdStorage();
             const token = getTokenStorage();
-            const model = params.model;
-            const spanName = `trace.chat ${model}`;
-            return this.tracer.startActiveSpan(spanName, (span) => __awaiter(this, void 0, void 0, function* () {
+            if (!token || !externalCustomerId) {
+                throw new Error("No token or externalCustomerId: This wrapper should be used inside a callback to paid.trace().");
+            }
+            return this.tracer.startActiveSpan("trace.openai.chat", (span) => __awaiter(this, void 0, void 0, function* () {
                 const attributes = {
                     "gen_ai.system": "openai",
                     "gen_ai.operation.name": "chat",
                 };
-                if (externalCustomerId) {
-                    attributes["external_customer_id"] = externalCustomerId;
-                }
-                if (token) {
-                    attributes["token"] = token;
+                attributes["external_customer_id"] = externalCustomerId;
+                attributes["token"] = token;
+                if (externalAgentId) {
+                    attributes["external_agent_id"] = externalAgentId;
                 }
                 span.setAttributes(attributes);
                 try {
@@ -95,36 +95,42 @@ class ResponsesWrapper {
     }
     create(params) {
         return __awaiter(this, void 0, void 0, function* () {
-            var _a;
             const currentSpan = trace.getSpan(context.active());
             if (!currentSpan) {
-                console.warn("No active span found, calling OpenAI directly without tracing.");
-                return this.openai.responses.create(params);
+                throw new Error("No active span found, make sure to call this inside of a callback to paid.trace().");
             }
-            const externalCustomerId = getCustomerId();
+            const externalCustomerId = getCustomerIdStorage();
+            const externalAgentId = getAgentIdStorage();
             const token = getTokenStorage();
-            const model = (_a = params.model) !== null && _a !== void 0 ? _a : "unknown";
-            const spanName = `trace.responses ${model}`;
-            return this.tracer.startActiveSpan(spanName, (span) => __awaiter(this, void 0, void 0, function* () {
+            if (!token || !externalCustomerId) {
+                throw new Error("No token or externalCustomerId: This wrapper should be used inside a callback to paid.trace().");
+            }
+            return this.tracer.startActiveSpan("trace.openai.responses", (span) => __awaiter(this, void 0, void 0, function* () {
+                var _a, _b;
                 const attributes = {
                     "gen_ai.system": "openai",
                     "gen_ai.operation.name": "chat", // Equivalent to chat.completions
                 };
-                if (externalCustomerId) {
-                    attributes["external_customer_id"] = externalCustomerId;
-                }
-                if (token) {
-                    attributes["token"] = token;
+                attributes["external_customer_id"] = externalCustomerId;
+                attributes["token"] = token;
+                if (externalAgentId) {
+                    attributes["external_agent_id"] = externalAgentId;
                 }
                 span.setAttributes(attributes);
                 try {
                     const response = yield this.openai.responses.create(params);
                     if (response.usage) {
                         span.setAttributes({
-                            "gen_ai.usage.input_tokens": response.usage.prompt_tokens,
-                            "gen_ai.usage.output_tokens": response.usage.completion_tokens,
+                            "gen_ai.usage.input_tokens": response.usage.input_tokens,
+                            "gen_ai.usage.output_tokens": response.usage.output_tokens,
                             "gen_ai.response.model": response.model,
                         });
+                        if ((_a = response.usage.input_tokens_details) === null || _a === void 0 ? void 0 : _a.cached_tokens) {
+                            span.setAttribute("gen_ai.usage.cached_input_tokens", response.usage.input_tokens_details.cached_tokens);
+                        }
+                        if ((_b = response.usage.output_tokens_details) === null || _b === void 0 ? void 0 : _b.reasoning_tokens) {
+                            span.setAttribute("gen_ai.usage.reasoning_output_tokens", response.usage.output_tokens_details.reasoning_tokens);
+                        }
                     }
                     span.setStatus({ code: SpanStatusCode.OK });
                     return response;
@@ -148,26 +154,25 @@ class EmbeddingsWrapper {
     }
     create(params) {
         return __awaiter(this, void 0, void 0, function* () {
-            var _a;
             const currentSpan = trace.getSpan(context.active());
             if (!currentSpan) {
-                console.warn("No active span found, calling OpenAI directly without tracing.");
-                return this.openai.embeddings.create(params);
+                throw new Error("No active span found, make sure to call this inside of a callback to paid.trace().");
             }
-            const externalCustomerId = getCustomerId();
+            const externalCustomerId = getCustomerIdStorage();
+            const externalAgentId = getAgentIdStorage();
             const token = getTokenStorage();
-            const model = (_a = params.model) !== null && _a !== void 0 ? _a : "unknown";
-            const spanName = `trace.embeddings ${model}`;
-            return this.tracer.startActiveSpan(spanName, (span) => __awaiter(this, void 0, void 0, function* () {
+            if (!token || !externalCustomerId) {
+                throw new Error("No token or externalCustomerId: This wrapper should be used inside a callback to paid.trace().");
+            }
+            return this.tracer.startActiveSpan("trace.openai.embeddings", (span) => __awaiter(this, void 0, void 0, function* () {
                 const attributes = {
                     "gen_ai.system": "openai",
                     "gen_ai.operation.name": "embeddings",
                 };
-                if (externalCustomerId) {
-                    attributes["external_customer_id"] = externalCustomerId;
-                }
-                if (token) {
-                    attributes["token"] = token;
+                attributes["external_customer_id"] = externalCustomerId;
+                attributes["token"] = token;
+                if (externalAgentId) {
+                    attributes["external_agent_id"] = externalAgentId;
                 }
                 span.setAttributes(attributes);
                 try {
@@ -203,25 +208,26 @@ class ImagesWrapper {
             var _a;
             const currentSpan = trace.getSpan(context.active());
             if (!currentSpan) {
-                console.warn("No active span found, calling OpenAI directly without tracing.");
-                return this.openai.images.generate(params);
+                throw new Error("No active span found, make sure to call this inside of a callback to paid.trace().");
             }
-            const externalCustomerId = getCustomerId();
+            const externalCustomerId = getCustomerIdStorage();
+            const externalAgentId = getAgentIdStorage();
             const token = getTokenStorage();
             const model = (_a = params.model) !== null && _a !== void 0 ? _a : "dall-e-3";
-            const spanName = `trace.images ${model}`;
-            return this.tracer.startActiveSpan(spanName, (span) => __awaiter(this, void 0, void 0, function* () {
+            if (!token || !externalCustomerId) {
+                throw new Error("No token or externalCustomerId: This wrapper should be used inside a callback to paid.trace().");
+            }
+            return this.tracer.startActiveSpan("trace.openai.images", (span) => __awaiter(this, void 0, void 0, function* () {
                 var _a, _b, _c;
                 const attributes = {
                     "gen_ai.request.model": model,
                     "gen_ai.system": "openai",
                     "gen_ai.operation.name": "image_generation",
                 };
-                if (externalCustomerId) {
-                    attributes["external_customer_id"] = externalCustomerId;
-                }
-                if (token) {
-                    attributes["token"] = token;
+                attributes["external_customer_id"] = externalCustomerId;
+                attributes["token"] = token;
+                if (externalAgentId) {
+                    attributes["external_agent_id"] = externalAgentId;
                 }
                 span.setAttributes(attributes);
                 try {

package/dist/esm/version.d.mts

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

package/dist/esm/version.mjs

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

package/dist/esm/wrapper/BatchUsage.d.mts

@@ -1,5 +1,5 @@
 import * as core from "../core/index.mjs";
-import { Usage as UsageClient } from "../api/resources/usage/client/Client";
+import { Usage as UsageClient } from "../api/resources/usage/client/Client.mjs";
 import * as Paid from "../api/types/Signal.mjs";
 export declare class Usage extends UsageClient {
     private signals;

package/dist/esm/wrapper/BatchUsage.mjs

@@ -1,5 +1,5 @@
 import * as core from "../core/index.mjs";
-import { Usage as UsageClient } from "../api/resources/usage/client/Client";
+import { Usage as UsageClient } from "../api/resources/usage/client/Client.mjs";
 export class Usage extends UsageClient {
     constructor() {
         super(...arguments);

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@paid-ai/paid-node",
-    "version": "0.0.8-alpha0",
+    "version": "0.0.8-alpha2",
     "private": false,
     "repository": "https://github.com/AgentPaid/paid-node",
     "type": "commonjs",
@@ -36,17 +36,21 @@
         "wire:test": "yarn test:wire"
     },
     "dependencies": {
+        "@anthropic-ai/sdk": "^0.56.0",
+        "@langchain/core": "^0.3.64",
+        "@mistralai/mistralai": "^1.7.4",
         "@opentelemetry/api": "^1.9.0",
         "@opentelemetry/exporter-trace-otlp-http": "^0.202.0",
         "@opentelemetry/sdk-node": "^0.202.0",
-        "openai": "^5.5.1",
         "form-data": "^4.0.0",
         "formdata-node": "^6.0.3",
         "js-base64": "3.7.7",
         "node-fetch": "^2.7.0",
+        "openai": "^5.5.1",
         "qs": "^6.13.1",
         "readable-stream": "^4.5.2",
-        "url-join": "4.0.1"
+        "url-join": "4.0.1",
+        "winston": "^3.17.0"
     },
     "devDependencies": {
         "@jest/globals": "^29.7.0",

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

@@ -1 +0,0 @@
-export * from "./wrappers/openAiWrapper";

package/dist/cjs/tracing/index.js

@@ -1,17 +0,0 @@
-"use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __exportStar = (this && this.__exportStar) || function(m, exports) {
-    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("./wrappers/openAiWrapper"), exports);

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

@@ -1 +0,0 @@
-export * from "./wrappers/openAiWrapper";

package/dist/esm/tracing/index.mjs

@@ -1 +0,0 @@
-export * from "./wrappers/openAiWrapper";