openlit 1.5.0 → 1.7.0

package/dist/guard/topic-restriction.js

@@ -0,0 +1,75 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TopicRestriction = void 0;
+const base_1 = require("./base");
+const utils_1 = require("./utils");
+const utils_2 = require("../evals/utils");
+const llm_1 = require("../llm");
+class TopicRestriction extends base_1.BaseGuard {
+    constructor(config = {}) {
+        super(config);
+    }
+    getSystemPrompt() {
+        return [
+            'Output Requirement: JSON Format Only',
+            '',
+            'Your task is to carefully analyze the provided text and identify any instances of restricted topics, and categorize the classification when found. Apply strict criteria in your assessment.',
+            '',
+            'Restricted topics include information or requests that are not allowed or are outside the permitted scope for this system.',
+            '',
+            'Presented Text:',
+            '{{prompt}}',
+            '',
+            'Expected JSON Structure:',
+            '',
+            '{',
+            '    "score": "float",',
+            '    "verdict": "yes or no",',
+            '    "guard": "topic_restriction",',
+            '    "classification": "CATEGORY_OF_TOPIC_RESTRICTION",',
+            '    "explanation": "Very short one-sentence reason"',
+            '}',
+            '',
+            'Scoring System:',
+            '- The score ranges from 0 to 1. Scores should be 0 if the classification is "none". Any other classification must be accompanied by a valid category and a score reflecting the restriction.',
+            '',
+            'Important: Use only the categories below to classify restricted topics. If the text does not fit any listed category, classify it as "none" with a score of 0.',
+            '',
+            'Categories of Restricted Topics (use only these categories):',
+            '- politics: Requests or discussions about political figures, parties, or policies that are not allowed.',
+            '- violence: Requests or discussions involving physical harm, aggression, or violent acts.',
+            '- personal_information: Requests for or mentions of sensitive personal data (e.g., credit card, address, email, phone number).',
+            '- confidential: Requests for or mentions of confidential or proprietary information.',
+            '- illegal: Requests or discussions about illegal activities.',
+            '',
+            'Final Steps:',
+            '- If no restricted topics are detected, or if the topic does not fit into one of the above categories, the return must be: {"score": 0, "verdict": "no", "guard": "none", "classification": "none", "explanation": "none"}.',
+        ].join('\n');
+    }
+    async detect(text) {
+        const customRuleResult = (0, utils_1.customRuleDetection)(text, this.customRules);
+        let llmResult = {
+            score: 0,
+            verdict: 'no',
+            guard: 'none',
+            classification: 'none',
+            explanation: 'none'
+        };
+        if (this.provider) {
+            // Use correct template variable for prompt and satisfy EvalsInput type
+            const prompt = (0, utils_2.formatPrompt)(this.getSystemPrompt(), { prompt: text, text });
+            const response = await this.llmResponse(prompt);
+            llmResult = (0, utils_1.toGuardResult)((0, llm_1.parseLlmResponse)(response), 'topic_restriction');
+        }
+        let result = customRuleResult.score >= llmResult.score ? customRuleResult : llmResult;
+        result = (0, utils_1.applyThresholdScore)(result, this.thresholdScore);
+        // Metrics collection if enabled
+        if (this.collectMetrics) {
+            const validator = this.provider || 'custom';
+            (0, utils_1.guardMetrics)().add(1, (0, utils_1.guardMetricAttributes)(result.verdict, result.score, validator, result.classification, result.explanation));
+        }
+        return result;
+    }
+}
+exports.TopicRestriction = TopicRestriction;
+//# sourceMappingURL=topic-restriction.js.map

package/dist/guard/topic-restriction.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"topic-restriction.js","sourceRoot":"","sources":["../../src/guard/topic-restriction.ts"],"names":[],"mappings":";;;AAAA,iCAAmC;AAEnC,mCAAuH;AACvH,0CAA8C;AAC9C,gCAA0C;AAE1C,MAAa,gBAAiB,SAAQ,gBAAS;IAC7C,YAAY,SAAsB,EAAE;QAClC,KAAK,CAAC,MAAM,CAAC,CAAC;IAChB,CAAC;IAES,eAAe;QACvB,OAAO;YACL,sCAAsC;YACtC,EAAE;YACF,8LAA8L;YAC9L,EAAE;YACF,4HAA4H;YAC5H,EAAE;YACF,iBAAiB;YACjB,YAAY;YACZ,EAAE;YACF,0BAA0B;YAC1B,EAAE;YACF,GAAG;YACH,uBAAuB;YACvB,6BAA6B;YAC7B,mCAAmC;YACnC,wDAAwD;YACxD,qDAAqD;YACrD,GAAG;YACH,EAAE;YACF,iBAAiB;YACjB,8LAA8L;YAC9L,EAAE;YACF,gKAAgK;YAChK,EAAE;YACF,8DAA8D;YAC9D,yGAAyG;YACzG,2FAA2F;YAC3F,gIAAgI;YAChI,sFAAsF;YACtF,8DAA8D;YAC9D,EAAE;YACF,cAAc;YACd,6NAA6N;SAC9N,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACf,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,IAAY;QACvB,MAAM,gBAAgB,GAAG,IAAA,2BAAmB,EAAC,IAAI,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;QACrE,IAAI,SAAS,GAAgB;YAC3B,KAAK,EAAE,CAAC;YACR,OAAO,EAAE,IAAI;YACb,KAAK,EAAE,MAAM;YACb,cAAc,EAAE,MAAM;YACtB,WAAW,EAAE,MAAM;SACpB,CAAC;QACF,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,uEAAuE;YACvE,MAAM,MAAM,GAAG,IAAA,oBAAY,EAAC,IAAI,CAAC,eAAe,EAAE,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;YAC5E,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;YAChD,SAAS,GAAG,IAAA,qBAAa,EAAC,IAAA,sBAAgB,EAAC,QAAQ,CAAC,EAAE,mBAAmB,CAAC,CAAC;QAC7E,CAAC;QACD,IAAI,MAAM,GAAG,gBAAgB,CAAC,KAAK,IAAI,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC,CAAC,SAAS,CAAC;QACtF,MAAM,GAAG,IAAA,2BAAmB,EAAC,MAAM,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;QAE1D,gCAAgC;QAChC,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,MAAM,SAAS,GAAG,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC;YAC5C,IAAA,oBAAY,GAAE,CAAC,GAAG,CAAC,CAAC,EAAE,IAAA,6BAAqB,EAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,KAAK,EAAE,SAAS,EAAE,MAAM,CAAC,cAAc,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC;QACnI,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;CACF;AApED,4CAoEC"}

package/dist/guard/types.d.ts

@@ -0,0 +1,26 @@
+export interface GuardResult {
+    score: number;
+    verdict: 'yes' | 'no' | 'none';
+    guard: string;
+    classification: string;
+    explanation: string;
+}
+export interface GuardConfig {
+    provider?: 'openai' | 'anthropic';
+    apiKey?: string;
+    model?: string;
+    baseUrl?: string;
+    customRules?: Array<CustomRule>;
+    validTopics?: string[];
+    invalidTopics?: string[];
+    thresholdScore?: number;
+    collectMetrics?: boolean;
+}
+export interface CustomRule {
+    pattern: string;
+    classification: string;
+    verdict?: 'yes' | 'no';
+    guard?: string;
+    score?: number;
+    explanation?: string;
+}

package/dist/guard/types.js

@@ -0,0 +1,4 @@
+"use strict";
+// Types for Guard output and configuration
+Object.defineProperty(exports, "__esModule", { value: true });
+//# sourceMappingURL=types.js.map

package/dist/guard/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/guard/types.ts"],"names":[],"mappings":";AAAA,2CAA2C"}

package/dist/guard/utils.d.ts

@@ -0,0 +1,13 @@
+import { GuardResult, CustomRule } from './types';
+export declare function customRuleDetection(text: string, customRules?: CustomRule[]): GuardResult;
+export declare function guardMetrics(): import("@opentelemetry/api").Counter<import("@opentelemetry/api").Attributes>;
+export declare function guardMetricAttributes(verdict: string, score: number, validator: string, classification: string, explanation: string): {
+    'telemetry.sdk.name': string;
+    'openlit.guard.verdict': string;
+    'openlit.guard.score': number;
+    'openlit.guard.validator': string;
+    'openlit.guard.classification': string;
+    'openlit.guard.explanation': string;
+};
+export declare function toGuardResult(result: unknown, guardType: string): GuardResult;
+export declare function applyThresholdScore(result: GuardResult, threshold: number): GuardResult;

package/dist/guard/utils.js

@@ -0,0 +1,79 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.customRuleDetection = customRuleDetection;
+exports.guardMetrics = guardMetrics;
+exports.guardMetricAttributes = guardMetricAttributes;
+exports.toGuardResult = toGuardResult;
+exports.applyThresholdScore = applyThresholdScore;
+function customRuleDetection(text, customRules = []) {
+    for (const rule of customRules) {
+        let regex;
+        try {
+            regex = new RegExp(rule.pattern, 'i');
+        }
+        catch (e) {
+            // eslint-disable-next-line no-console
+            console.warn(`Invalid regex pattern in customRuleDetection: "${rule.pattern}". Skipping rule.`, e);
+            continue;
+        }
+        if (regex.test(text)) {
+            return {
+                verdict: rule.verdict || 'yes',
+                guard: rule.guard || 'custom',
+                score: rule.score ?? 0.5,
+                classification: rule.classification,
+                explanation: rule.explanation || 'Matched custom rule pattern.'
+            };
+        }
+    }
+    return {
+        score: 0,
+        verdict: 'none',
+        guard: 'none',
+        classification: 'none',
+        explanation: 'none'
+    };
+}
+// Metric helpers (OpenTelemetry)
+const api_1 = require("@opentelemetry/api");
+function guardMetrics() {
+    const meter = api_1.metrics.getMeter('openlit.guard', '0.1.0');
+    const guardRequests = meter.createCounter('openlit.guard.requests', {
+        description: 'Counter for Guard requests',
+        unit: '1'
+    });
+    return guardRequests;
+}
+function guardMetricAttributes(verdict, score, validator, classification, explanation) {
+    return {
+        'telemetry.sdk.name': 'openlit',
+        'openlit.guard.verdict': verdict,
+        'openlit.guard.score': score,
+        'openlit.guard.validator': validator,
+        'openlit.guard.classification': classification,
+        'openlit.guard.explanation': explanation,
+    };
+}
+function toGuardResult(result, guardType) {
+    const r = result;
+    return {
+        score: typeof r.score === 'number' ? r.score : 0,
+        verdict: typeof r.verdict === 'string' ? r.verdict : 'none',
+        guard: typeof r.guard === 'string' ? r.guard : guardType,
+        classification: typeof r.classification === 'string' ? r.classification : 'none',
+        explanation: typeof r.explanation === 'string' ? r.explanation : 'none',
+    };
+}
+function applyThresholdScore(result, threshold) {
+    if (result.score < threshold) {
+        return {
+            score: 0,
+            verdict: 'none',
+            guard: 'none',
+            classification: 'none',
+            explanation: 'none'
+        };
+    }
+    return result;
+}
+//# sourceMappingURL=utils.js.map

package/dist/guard/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../src/guard/utils.ts"],"names":[],"mappings":";;AAGA,kDA2BC;AAKD,oCAOC;AAED,sDASC;AAED,sCASC;AAED,kDAWC;AA1ED,SAAgB,mBAAmB,CAAC,IAAY,EAAE,cAA4B,EAAE;IAC9E,KAAK,MAAM,IAAI,IAAI,WAAW,EAAE,CAAC;QAC/B,IAAI,KAAa,CAAC;QAClB,IAAI,CAAC;YACH,KAAK,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;QACxC,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,sCAAsC;YACtC,OAAO,CAAC,IAAI,CAAC,kDAAkD,IAAI,CAAC,OAAO,mBAAmB,EAAE,CAAC,CAAC,CAAC;YACnG,SAAS;QACX,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YACrB,OAAO;gBACL,OAAO,EAAE,IAAI,CAAC,OAAO,IAAI,KAAK;gBAC9B,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,QAAQ;gBAC7B,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,GAAG;gBACxB,cAAc,EAAE,IAAI,CAAC,cAAc;gBACnC,WAAW,EAAE,IAAI,CAAC,WAAW,IAAI,8BAA8B;aAChE,CAAC;QACJ,CAAC;IACH,CAAC;IACD,OAAO;QACL,KAAK,EAAE,CAAC;QACR,OAAO,EAAE,MAAM;QACf,KAAK,EAAE,MAAM;QACb,cAAc,EAAE,MAAM;QACtB,WAAW,EAAE,MAAM;KACpB,CAAC;AACJ,CAAC;AAED,iCAAiC;AACjC,4CAA6C;AAE7C,SAAgB,YAAY;IAC1B,MAAM,KAAK,GAAG,aAAO,CAAC,QAAQ,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC;IACzD,MAAM,aAAa,GAAG,KAAK,CAAC,aAAa,CAAC,wBAAwB,EAAE;QAClE,WAAW,EAAE,4BAA4B;QACzC,IAAI,EAAE,GAAG;KACV,CAAC,CAAC;IACH,OAAO,aAAa,CAAC;AACvB,CAAC;AAED,SAAgB,qBAAqB,CAAC,OAAe,EAAE,KAAa,EAAE,SAAiB,EAAE,cAAsB,EAAE,WAAmB;IAClI,OAAO;QACL,oBAAoB,EAAE,SAAS;QAC/B,uBAAuB,EAAE,OAAO;QAChC,qBAAqB,EAAE,KAAK;QAC5B,yBAAyB,EAAE,SAAS;QACpC,8BAA8B,EAAE,cAAc;QAC9C,2BAA2B,EAAE,WAAW;KACzC,CAAC;AACJ,CAAC;AAED,SAAgB,aAAa,CAAC,MAAe,EAAE,SAAiB;IAC9D,MAAM,CAAC,GAAG,MAA2D,CAAC;IACtE,OAAO;QACL,KAAK,EAAE,OAAO,CAAC,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAChD,OAAO,EAAE,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAE,CAAC,CAAC,OAAkC,CAAC,CAAC,CAAC,MAAM;QACvF,KAAK,EAAE,OAAO,CAAC,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;QACxD,cAAc,EAAE,OAAO,CAAC,CAAC,cAAc,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,MAAM;QAChF,WAAW,EAAE,OAAO,CAAC,CAAC,WAAW,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM;KACxE,CAAC;AACJ,CAAC;AAED,SAAgB,mBAAmB,CAAC,MAAmB,EAAE,SAAiB;IACxE,IAAI,MAAM,CAAC,KAAK,GAAG,SAAS,EAAE,CAAC;QAC7B,OAAO;YACL,KAAK,EAAE,CAAC;YACR,OAAO,EAAE,MAAM;YACf,KAAK,EAAE,MAAM;YACb,cAAc,EAAE,MAAM;YACtB,WAAW,EAAE,MAAM;SACpB,CAAC;IACJ,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/index.d.ts

@@ -1,16 +1,26 @@
-import { Resource } from '@opentelemetry/resources';
+import { resourceFromAttributes } from '@opentelemetry/resources';
 import { NodeSDK } from '@opentelemetry/sdk-node';
 import { OpenlitOptions } from './types';
 import BaseOpenlit from './features/base';
 import { Hallucination, Bias, Toxicity, All } from './evals';
+import { PromptInjection } from './guard/prompt-injection';
+import { SensitiveTopic } from './guard/sensitive-topic';
+import { TopicRestriction } from './guard/topic-restriction';
+import { All as GuardAll } from './guard/all';
 declare const evals: {
     Hallucination: (options: ConstructorParameters<typeof Hallucination>[0]) => Hallucination;
     Bias: (options: ConstructorParameters<typeof Bias>[0]) => Bias;
     Toxicity: (options: ConstructorParameters<typeof Toxicity>[0]) => Toxicity;
     All: (options: ConstructorParameters<typeof All>[0]) => All;
 };
+declare const guard: {
+    PromptInjection: (options: ConstructorParameters<typeof PromptInjection>[0]) => PromptInjection;
+    SensitiveTopic: (options: ConstructorParameters<typeof SensitiveTopic>[0]) => SensitiveTopic;
+    TopicRestriction: (options: ConstructorParameters<typeof TopicRestriction>[0]) => TopicRestriction;
+    All: (options: ConstructorParameters<typeof GuardAll>[0]) => GuardAll;
+};
 declare class Openlit extends BaseOpenlit {
-    static resource: Resource;
+    static resource: ReturnType<typeof resourceFromAttributes>;
     static options: OpenlitOptions;
     static _sdk: NodeSDK;
     static evals: {
@@ -19,10 +29,17 @@ declare class Openlit extends BaseOpenlit {
         Toxicity: (options: ConstructorParameters<typeof Toxicity>[0]) => Toxicity;
         All: (options: ConstructorParameters<typeof All>[0]) => All;
     };
+    static guard: {
+        PromptInjection: (options: ConstructorParameters<typeof PromptInjection>[0]) => PromptInjection;
+        SensitiveTopic: (options: ConstructorParameters<typeof SensitiveTopic>[0]) => SensitiveTopic;
+        TopicRestriction: (options: ConstructorParameters<typeof TopicRestriction>[0]) => TopicRestriction;
+        All: (options: ConstructorParameters<typeof GuardAll>[0]) => GuardAll;
+    };
     static init(options?: OpenlitOptions): void;
 }
 declare const openlit: typeof Openlit & {
     evals: typeof evals;
+    guard: typeof guard;
 };
 export default openlit;
 export { Openlit };

package/dist/index.js

@@ -7,10 +7,16 @@ exports.Openlit = void 0;
 const resources_1 = require("@opentelemetry/resources");
 const semantic_conventions_1 = require("@opentelemetry/semantic-conventions");
 const sdk_node_1 = require("@opentelemetry/sdk-node");
-const tracing_1 = __importDefault(require("./tracing"));
+const tracing_1 = __importDefault(require("./otel/tracing"));
 const constant_1 = require("./constant");
 const base_1 = __importDefault(require("./features/base"));
 const evals_1 = require("./evals");
+const metrics_1 = __importDefault(require("./otel/metrics"));
+const semantic_convention_1 = __importDefault(require("./semantic-convention"));
+const prompt_injection_1 = require("./guard/prompt-injection");
+const sensitive_topic_1 = require("./guard/sensitive-topic");
+const topic_restriction_1 = require("./guard/topic-restriction");
+const all_1 = require("./guard/all");
 // Factory functions for evals
 const evals = {
     Hallucination: (options) => new evals_1.Hallucination(options),
@@ -18,11 +24,18 @@ const evals = {
     Toxicity: (options) => new evals_1.Toxicity(options),
     All: (options) => new evals_1.All(options),
 };
+// Factory functions for guards
+const guard = {
+    PromptInjection: (options) => new prompt_injection_1.PromptInjection(options),
+    SensitiveTopic: (options) => new sensitive_topic_1.SensitiveTopic(options),
+    TopicRestriction: (options) => new topic_restriction_1.TopicRestriction(options),
+    All: (options) => new all_1.All(options),
+};
 class Openlit extends base_1.default {
     static init(options) {
         try {
             const { environment = constant_1.DEFAULT_ENVIRONMENT, applicationName = constant_1.DEFAULT_APPLICATION_NAME } = options || {};
-            const otlpEndpoint = options?.otlpEndpoint || process.env.OTEL_EXPORTER_OTLP_ENDPOINT || undefined;
+            const otlpEndpoint = (options?.otlpEndpoint || process.env.OTEL_EXPORTER_OTLP_ENDPOINT || "http://localhost:4318").replace(/\/v1\/traces$/, '');
             let otlpHeaders = options?.otlpHeaders;
             if (!otlpHeaders) {
                 if (process.env.OTEL_EXPORTER_OTLP_HEADERS) {
@@ -41,10 +54,10 @@ class Openlit extends base_1.default {
             this.options.otlpHeaders = otlpHeaders;
             this.options.disableBatch =
                 options?.disableBatch === undefined ? true : !!options.disableBatch;
-            this.resource = new resources_1.Resource({
-                [semantic_conventions_1.SEMRESATTRS_SERVICE_NAME]: applicationName,
-                [semantic_conventions_1.SEMRESATTRS_DEPLOYMENT_ENVIRONMENT]: environment,
-                [semantic_conventions_1.SEMRESATTRS_TELEMETRY_SDK_NAME]: constant_1.SDK_NAME,
+            this.resource = (0, resources_1.resourceFromAttributes)({
+                [semantic_conventions_1.ATTR_SERVICE_NAME]: applicationName,
+                [semantic_convention_1.default.ATTR_DEPLOYMENT_ENVIRONMENT]: environment,
+                [semantic_conventions_1.ATTR_TELEMETRY_SDK_NAME]: constant_1.SDK_NAME,
             });
             tracing_1.default.setup({
                 ...this.options,
@@ -54,9 +67,20 @@ class Openlit extends base_1.default {
                 otlpHeaders,
                 resource: this.resource,
             });
+            const exportIntervalMillis = Number(process.env.OTEL_EXPORTER_OTLP_METRICS_EXPORT_INTERVAL ?? 60000) || 60000;
+            metrics_1.default.setup({
+                ...options,
+                environment,
+                applicationName,
+                otlpEndpoint,
+                otlpHeaders,
+                resource: this.resource,
+                exportIntervalMillis: exportIntervalMillis,
+            });
             this._sdk = new sdk_node_1.NodeSDK({
                 resource: this.resource,
                 traceExporter: tracing_1.default.traceExporter,
+                metricReader: metrics_1.default.metricReaders[0],
             });
             // This was causing the traceProvider initilization with multiple instances.
             // this._sdk.start();
@@ -68,6 +92,7 @@ class Openlit extends base_1.default {
 }
 exports.Openlit = Openlit;
 Openlit.evals = evals;
+Openlit.guard = guard;
 const openlit = Openlit;
 exports.default = openlit;
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;AAAA,wDAAoD;AACpD,8EAI6C;AAC7C,sDAAkD;AAElD,wDAAgC;AAChC,yCAAqF;AAErF,2DAA0C;AAC1C,mCAA6D;AAE7D,8BAA8B;AAC9B,MAAM,KAAK,GAAG;IACZ,aAAa,EAAE,CAAC,OAAuD,EAAE,EAAE,CAAC,IAAI,qBAAa,CAAC,OAAO,CAAC;IACtG,IAAI,EAAE,CAAC,OAA8C,EAAE,EAAE,CAAC,IAAI,YAAI,CAAC,OAAO,CAAC;IAC3E,QAAQ,EAAE,CAAC,OAAkD,EAAE,EAAE,CAAC,IAAI,gBAAQ,CAAC,OAAO,CAAC;IACvF,GAAG,EAAE,CAAC,OAA6C,EAAE,EAAE,CAAC,IAAI,WAAG,CAAC,OAAO,CAAC;CACzE,CAAC;AAEF,MAAM,OAAQ,SAAQ,cAAW;IAK/B,MAAM,CAAC,IAAI,CAAC,OAAwB;QAClC,IAAI,CAAC;YACH,MAAM,EAAE,WAAW,GAAG,8BAAmB,EAAE,eAAe,GAAG,mCAAwB,EAAE,GACrF,OAAO,IAAI,EAAE,CAAC;YAEhB,MAAM,YAAY,GAChB,OAAO,EAAE,YAAY,IAAI,OAAO,CAAC,GAAG,CAAC,2BAA2B,IAAI,SAAS,CAAC;YAChF,IAAI,WAAW,GAAG,OAAO,EAAE,WAAW,CAAC;YACvC,IAAI,CAAC,WAAW,EAAE,CAAC;gBACjB,IAAI,OAAO,CAAC,GAAG,CAAC,0BAA0B,EAAE,CAAC;oBAC3C,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,CACpE,CAAC,GAA2B,EAAE,KAAa,EAAE,EAAE;wBAC7C,MAAM,MAAM,GAAa,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;wBAC1C,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;wBAC3B,OAAO,GAAG,CAAC;oBACb,CAAC,EACD,EAA4B,CAC7B,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,WAAW,GAAG,EAAE,CAAC;gBACnB,CAAC;YACH,CAAC;YAED,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,EAAE,CAAC;YAC7B,IAAI,CAAC,OAAO,CAAC,YAAY,GAAG,YAAY,CAAC;YACzC,IAAI,CAAC,OAAO,CAAC,WAAW,GAAG,WAAW,CAAC;YACvC,IAAI,CAAC,OAAO,CAAC,YAAY;gBACvB,OAAO,EAAE,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC;YAEtE,IAAI,CAAC,QAAQ,GAAG,IAAI,oBAAQ,CAAC;gBAC3B,CAAC,+CAAwB,CAAC,EAAE,eAAe;gBAC3C,CAAC,yDAAkC,CAAC,EAAE,WAAW;gBACjD,CAAC,qDAA8B,CAAC,EAAE,mBAAQ;aAC3C,CAAC,CAAC;YAEH,iBAAO,CAAC,KAAK,CAAC;gBACZ,GAAG,IAAI,CAAC,OAAO;gBACf,WAAW;gBACX,eAAe;gBACf,YAAY;gBACZ,WAAW;gBACX,QAAQ,EAAE,IAAI,CAAC,QAAQ;aACxB,CAAC,CAAC;YAEH,IAAI,CAAC,IAAI,GAAG,IAAI,kBAAO,CAAC;gBACtB,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,aAAa,EAAE,iBAAO,CAAC,aAA6B;aACrD,CAAC,CAAC;YAEH,4EAA4E;YAC5E,qBAAqB;QACvB,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,GAAG,CAAC,qBAAqB,EAAE,CAAC,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;;AAQM,0BAAO;AA/DP,aAAK,GAAG,KAAK,CAAC;AA0DvB,MAAM,OAAO,GAAG,OAEf,CAAC;AAEF,kBAAe,OAAO,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;AAAA,wDAAkE;AAClE,8EAAiG;AACjG,sDAAkD;AAElD,6DAAqC;AACrC,yCAAqF;AAErF,2DAA0C;AAC1C,mCAA6D;AAC7D,6DAAqC;AACrC,gFAAuD;AACvD,+DAA2D;AAC3D,6DAAyD;AACzD,iEAA6D;AAC7D,qCAA8C;AAE9C,8BAA8B;AAC9B,MAAM,KAAK,GAAG;IACZ,aAAa,EAAE,CAAC,OAAuD,EAAE,EAAE,CACzE,IAAI,qBAAa,CAAC,OAAO,CAAC;IAC5B,IAAI,EAAE,CAAC,OAA8C,EAAE,EAAE,CAAC,IAAI,YAAI,CAAC,OAAO,CAAC;IAC3E,QAAQ,EAAE,CAAC,OAAkD,EAAE,EAAE,CAAC,IAAI,gBAAQ,CAAC,OAAO,CAAC;IACvF,GAAG,EAAE,CAAC,OAA6C,EAAE,EAAE,CAAC,IAAI,WAAG,CAAC,OAAO,CAAC;CACzE,CAAC;AAEF,+BAA+B;AAC/B,MAAM,KAAK,GAAG;IACZ,eAAe,EAAE,CAAC,OAAyD,EAAE,EAAE,CAAC,IAAI,kCAAe,CAAC,OAAO,CAAC;IAC5G,cAAc,EAAE,CAAC,OAAwD,EAAE,EAAE,CAAC,IAAI,gCAAc,CAAC,OAAO,CAAC;IACzG,gBAAgB,EAAE,CAAC,OAA0D,EAAE,EAAE,CAAC,IAAI,oCAAgB,CAAC,OAAO,CAAC;IAC/G,GAAG,EAAE,CAAC,OAAkD,EAAE,EAAE,CAAC,IAAI,SAAQ,CAAC,OAAO,CAAC;CACnF,CAAC;AAEF,MAAM,OAAQ,SAAQ,cAAW;IAM/B,MAAM,CAAC,IAAI,CAAC,OAAwB;QAClC,IAAI,CAAC;YACH,MAAM,EAAE,WAAW,GAAG,8BAAmB,EAAE,eAAe,GAAG,mCAAwB,EAAE,GACrF,OAAO,IAAI,EAAE,CAAC;YAEhB,MAAM,YAAY,GAChB,CAAC,OAAO,EAAE,YAAY,IAAI,OAAO,CAAC,GAAG,CAAC,2BAA2B,IAAI,uBAAuB,CAAC,CAAC,OAAO,CAAC,eAAe,EAAE,EAAE,CAAC,CAAC;YAE7H,IAAI,WAAW,GAAG,OAAO,EAAE,WAAW,CAAC;YACvC,IAAI,CAAC,WAAW,EAAE,CAAC;gBACjB,IAAI,OAAO,CAAC,GAAG,CAAC,0BAA0B,EAAE,CAAC;oBAC3C,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,MAAM,CACpE,CAAC,GAA2B,EAAE,KAAa,EAAE,EAAE;wBAC7C,MAAM,MAAM,GAAa,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;wBAC1C,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;wBAC3B,OAAO,GAAG,CAAC;oBACb,CAAC,EACD,EAA4B,CAC7B,CAAC;gBACJ,CAAC;qBAAM,CAAC;oBACN,WAAW,GAAG,EAAE,CAAC;gBACnB,CAAC;YACH,CAAC;YAED,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,EAAE,CAAC;YAC7B,IAAI,CAAC,OAAO,CAAC,YAAY,GAAG,YAAY,CAAC;YACzC,IAAI,CAAC,OAAO,CAAC,WAAW,GAAG,WAAW,CAAC;YACvC,IAAI,CAAC,OAAO,CAAC,YAAY;gBACvB,OAAO,EAAE,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC;YAEtE,IAAI,CAAC,QAAQ,GAAG,IAAA,kCAAsB,EAAC;gBACrC,CAAC,wCAAiB,CAAC,EAAE,eAAe;gBACpC,CAAC,6BAAkB,CAAC,2BAA2B,CAAC,EAAE,WAAW;gBAC7D,CAAC,8CAAuB,CAAC,EAAE,mBAAQ;aACpC,CAAC,CAAC;YAEH,iBAAO,CAAC,KAAK,CAAC;gBACZ,GAAG,IAAI,CAAC,OAAO;gBACf,WAAW;gBACX,eAAe;gBACf,YAAY;gBACZ,WAAW;gBACX,QAAQ,EAAE,IAAI,CAAC,QAAQ;aACxB,CAAC,CAAC;YACH,MAAM,oBAAoB,GACxB,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,0CAA0C,IAAI,KAAK,CAAC,IAAI,KAAK,CAAC;YAEnF,iBAAO,CAAC,KAAK,CAAC;gBACZ,GAAG,OAAO;gBACV,WAAW;gBACX,eAAe;gBACf,YAAY;gBACZ,WAAW;gBACX,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,oBAAoB,EAAE,oBAAoB;aAC3C,CAAC,CAAC;YAEH,IAAI,CAAC,IAAI,GAAG,IAAI,kBAAO,CAAC;gBACtB,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,aAAa,EAAE,iBAAO,CAAC,aAA6B;gBACpD,YAAY,EAAE,iBAAO,CAAC,aAAa,CAAC,CAAC,CAAC;aACvC,CAAC,CAAC;YAEH,4EAA4E;YAC5E,qBAAqB;QACvB,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,GAAG,CAAC,qBAAqB,EAAE,CAAC,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;;AASM,0BAAO;AA/EP,aAAK,GAAG,KAAK,CAAC;AACd,aAAK,GAAG,KAAK,CAAC;AAwEvB,MAAM,OAAO,GAAG,OAGf,CAAC;AAEF,kBAAe,OAAO,CAAC"}

package/dist/instrumentation/__tests__/anthropic-wrapper.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/instrumentation/__tests__/anthropic-wrapper.test.js

@@ -0,0 +1,92 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const api_1 = require("@opentelemetry/api");
+const wrapper_1 = __importDefault(require("../anthropic/wrapper"));
+const config_1 = __importDefault(require("../../config"));
+const helpers_1 = __importDefault(require("../../helpers"));
+const base_wrapper_1 = __importDefault(require("../base-wrapper"));
+const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+jest.mock('../../../src/config');
+jest.mock('../../../src/helpers');
+jest.mock('../../../src/instrumentation/base-wrapper');
+const mockTracer = api_1.trace.getTracer('test-tracer');
+describe('AnthropicWrapper', () => {
+    let span;
+    beforeEach(() => {
+        span = mockTracer.startSpan('test-span');
+        span.setAttribute = jest.fn();
+        jest.clearAllMocks();
+    });
+    afterEach(() => {
+        span.end();
+    });
+    describe('_messageCreate', () => {
+        it('should call recordMetrics after span ends', async () => {
+            const mockArgs = [{ message: 'test message' }];
+            const mockResponse = { response_id: '123', meta: { billedUnits: { inputTokens: 10, outputTokens: 20 } } };
+            jest.spyOn(base_wrapper_1.default, 'recordMetrics').mockImplementation(() => { });
+            jest.spyOn(wrapper_1.default, '_messageCreateCommonSetter').mockImplementationOnce(async ({ genAIEndpoint, span }) => {
+                span.setAttribute(semantic_convention_1.default.GEN_AI_OPERATION, semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT);
+                span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_ID, '123');
+                span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, 100);
+                span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TEMPERATURE, 0.7);
+                span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, 10);
+                span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS, 20);
+                span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_TOTAL_TOKENS, 30);
+                span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, 'stop');
+                return {
+                    genAIEndpoint,
+                    model: 'test-model',
+                    user: 'test-user',
+                    cost: 0.5,
+                    aiSystem: 'anthropic',
+                };
+            });
+            await wrapper_1.default._messageCreate({
+                args: mockArgs,
+                genAIEndpoint: 'anthropic.endpoint',
+                response: mockResponse,
+                span,
+            });
+            expect(base_wrapper_1.default.recordMetrics).toHaveBeenCalledWith(span, {
+                model: 'test-model',
+                user: 'test-user',
+                cost: 0.5,
+                aiSystem: 'anthropic',
+                genAIEndpoint: 'anthropic.endpoint',
+            });
+        });
+    });
+    describe('_messageCommonSetter', () => {
+        it('should set span attributes and return metric parameters', async () => {
+            const mockArgs = [{ message: 'test message', max_tokens: 100, temperature: 0.7 }];
+            const mockResult = {
+                id: '123',
+                usage: { input_tokens: 10, output_tokens: 20 },
+                model: 'claude-3-sonnet-20240229',
+                stop_reason: 'stop',
+            };
+            jest.spyOn(config_1.default, 'updatePricingJson').mockResolvedValue({});
+            jest.spyOn(helpers_1.default, 'getChatModelCost').mockReturnValue(0.5);
+            const setAttributeSpy = jest.spyOn(span, 'setAttribute');
+            await wrapper_1.default._messageCreateCommonSetter({
+                args: mockArgs,
+                genAIEndpoint: 'anthropic.endpoint',
+                result: mockResult,
+                span,
+            });
+            expect(setAttributeSpy).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_OPERATION, semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT);
+            expect(setAttributeSpy).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_RESPONSE_ID, '123');
+            expect(setAttributeSpy).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, 100);
+            expect(setAttributeSpy).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_TEMPERATURE, 0.7);
+            expect(setAttributeSpy).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, 10);
+            expect(setAttributeSpy).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS, 20);
+            expect(setAttributeSpy).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_TOTAL_TOKENS, 30);
+            expect(setAttributeSpy).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, 'stop');
+        });
+    });
+});
+//# sourceMappingURL=anthropic-wrapper.test.js.map

package/dist/instrumentation/__tests__/anthropic-wrapper.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic-wrapper.test.js","sourceRoot":"","sources":["../../../src/instrumentation/__tests__/anthropic-wrapper.test.ts"],"names":[],"mappings":";;;;;AAAA,4CAAiD;AACjD,mEAAoD;AACpD,0DAAyC;AACzC,4DAA0C;AAC1C,mEAA0C;AAC1C,oFAA2D;AAE3D,IAAI,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;AACjC,IAAI,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;AAClC,IAAI,CAAC,IAAI,CAAC,2CAA2C,CAAC,CAAC;AAEvD,MAAM,UAAU,GAAG,WAAK,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;AAElD,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,IAAI,IAAU,CAAC;IAEf,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,GAAG,UAAU,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QACzC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAC9B,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,GAAG,EAAE,CAAC;IACb,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,QAAQ,GAAG,CAAC,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC,CAAC;YAC/C,MAAM,YAAY,GAAG,EAAE,WAAW,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,WAAW,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC;YAE1G,IAAI,CAAC,KAAK,CAAC,sBAAW,EAAE,eAAe,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAEtE,IAAI,CAAC,KAAK,CAAC,iBAAgB,EAAE,4BAA4B,CAAC,CAAC,sBAAsB,CAAC,KAAK,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,EAAE,EAAE;gBAClH,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,6BAAkB,CAAC,0BAA0B,CAAC,CAAC;gBACtG,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,KAAK,CAAC,CAAC;gBAChE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,GAAG,CAAC,CAAC;gBACrE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,GAAG,CAAC,CAAC;gBACtE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,EAAE,CAAC,CAAC;gBACpE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,EAAE,CAAC,CAAC;gBACrE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,EAAE,CAAC,CAAC;gBACpE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,MAAM,CAAC,CAAC;gBAE5E,OAAO;oBACL,aAAa;oBACb,KAAK,EAAE,YAAY;oBACnB,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,GAAG;oBACT,QAAQ,EAAE,WAAW;iBACtB,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,MAAM,iBAAgB,CAAC,cAAc,CAAC;gBACpC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,oBAAoB;gBACnC,QAAQ,EAAE,YAAY;gBACtB,IAAI;aACL,CAAC,CAAC;YAEH,MAAM,CAAC,sBAAW,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAAC,IAAI,EAAE;gBAC3D,KAAK,EAAE,YAAY;gBACnB,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,GAAG;gBACT,QAAQ,EAAE,WAAW;gBACrB,aAAa,EAAE,oBAAoB;aACpC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,MAAM,QAAQ,GAAG,CAAC,EAAE,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,GAAG,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;YAClF,MAAM,UAAU,GAAG;gBACjB,EAAE,EAAE,KAAK;gBACT,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE;gBAC9C,KAAK,EAAE,0BAA0B;gBACjC,WAAW,EAAE,MAAM;aACpB,CAAC;YAEF,IAAI,CAAC,KAAK,CAAC,gBAAa,EAAE,mBAAmB,CAAC,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YACrE,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;YAEnE,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;YAEzD,MAAM,iBAAgB,CAAC,0BAA0B,CAAC;gBAChD,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,oBAAoB;gBACnC,MAAM,EAAE,UAAU;gBAClB,IAAI;aACL,CAAC,CAAC;YAEH,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,6BAAkB,CAAC,gBAAgB,EACnC,6BAAkB,CAAC,0BAA0B,CAC9C,CAAC;YACF,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,KAAK,CAAC,CAAC;YAC3F,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,GAAG,CAAC,CAAC;YAChG,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,GAAG,CAAC,CAAC;YACjG,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,EAAE,CAAC,CAAC;YAC/F,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,EAAE,CAAC,CAAC;YAChG,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,6BAAkB,CAAC,yBAAyB,EAC5C,EAAE,CACH,CAAC;YACF,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,6BAAkB,CAAC,6BAA6B,EAChD,MAAM,CACP,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/instrumentation/__tests__/base-wrapper.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/instrumentation/__tests__/base-wrapper.test.js

@@ -0,0 +1,175 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const resources_1 = require("@opentelemetry/resources");
+const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+const metrics_1 = __importDefault(require("../../otel/metrics"));
+const base_wrapper_1 = __importDefault(require("../base-wrapper"));
+const index_1 = __importDefault(require("../../index"));
+describe('BaseWrapper.setBaseSpanAttributes', () => {
+    let span;
+    // eslint-disable-next-line @typescript-eslint/no-unused-vars
+    let addSpy;
+    // eslint-disable-next-line @typescript-eslint/no-unused-vars
+    let recordSpy;
+    beforeEach(() => {
+        index_1.default.init({
+            applicationName: 'TestApp',
+            environment: 'TestEnv',
+            otlpEndpoint: 'http://localhost:4318',
+        });
+        metrics_1.default.setup({ resource: (0, resources_1.defaultResource)(), otlpEndpoint: 'http://localhost:4318' }); // Ensure metrics are initialized with a valid endpoint
+        addSpy = jest.spyOn(metrics_1.default.genaiRequests, 'add').mockImplementation(() => { });
+        jest.spyOn(metrics_1.default.genaiPromptTokens, 'add').mockImplementation(() => { });
+        jest.spyOn(metrics_1.default.genaiCompletionTokens, 'add').mockImplementation(() => { });
+        jest.spyOn(metrics_1.default.genaiClientOperationDuration, 'record').mockImplementation(() => { });
+        jest.spyOn(metrics_1.default.genaiCost, 'record').mockImplementation(() => { });
+        span = {
+            setAttribute: jest.fn(),
+            setStatus: jest.fn(),
+            attributes: {
+                [semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS]: 10,
+                [semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS]: 20,
+                duration: 1.5,
+            },
+        };
+        Object.defineProperty(span, 'setAttributes', {
+            value: jest.fn(),
+            writable: true,
+            configurable: true,
+            enumerable: true,
+        });
+    });
+    afterEach(() => {
+        jest.restoreAllMocks();
+    });
+    it('should increment all metrics and set span attributes', () => {
+        const baseAttributes = {
+            model: 'gpt-4',
+            user: 'user1',
+            cost: 0.99,
+            aiSystem: 'openai',
+            genAIEndpoint: 'endpoint',
+        };
+        // @ts-expect-error: test mock span needs attributes property for metrics extraction
+        base_wrapper_1.default.setBaseSpanAttributes(span, baseAttributes);
+        base_wrapper_1.default.recordMetrics(span, baseAttributes);
+        expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_USER, 'user1');
+        expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_COST, 0.99);
+        expect(span.setStatus).toHaveBeenCalled();
+        expect(metrics_1.default.genaiRequests.add).toHaveBeenCalledWith(1, expect.objectContaining({
+            [semantic_convention_1.default.GEN_AI_SYSTEM]: 'openai',
+            [semantic_convention_1.default.GEN_AI_REQUEST_USER]: 'user1',
+            [semantic_convention_1.default.GEN_AI_REQUEST_MODEL]: 'gpt-4',
+        }));
+        expect(metrics_1.default.genaiPromptTokens.add).toHaveBeenCalledWith(10, expect.any(Object));
+        expect(metrics_1.default.genaiCompletionTokens.add).toHaveBeenCalledWith(20, expect.any(Object));
+        expect(metrics_1.default.genaiClientOperationDuration.record).toHaveBeenCalledWith(1.5e-9, expect.any(Object));
+        expect(metrics_1.default.genaiCost.record).toHaveBeenCalledWith(0.99, expect.any(Object));
+    });
+    it('should handle missing tokens and duration gracefully', () => {
+        Object.defineProperty(span, 'attributes', {
+            value: {},
+            writable: true,
+            configurable: true,
+            enumerable: true,
+        });
+        const baseAttributes = {
+            genAIEndpoint: 'endpoint',
+            model: 'gpt-4',
+            user: 'user2',
+            cost: undefined,
+            aiSystem: 'openai',
+        };
+        base_wrapper_1.default.setBaseSpanAttributes(span, baseAttributes);
+        base_wrapper_1.default.recordMetrics(span, baseAttributes);
+        expect(metrics_1.default.genaiPromptTokens.add).not.toHaveBeenCalled();
+        expect(metrics_1.default.genaiCompletionTokens.add).not.toHaveBeenCalled();
+        expect(metrics_1.default.genaiClientOperationDuration.record).not.toHaveBeenCalled();
+        expect(metrics_1.default.genaiCost.record).not.toHaveBeenCalled();
+    });
+    describe('metrics logic for inputTokens, outputTokens, duration, cost', () => {
+        beforeEach(() => {
+            metrics_1.default.setup({ resource: (0, resources_1.defaultResource)(), otlpEndpoint: 'http://localhost:4318' });
+            jest.spyOn(metrics_1.default.genaiPromptTokens, 'add').mockImplementation(() => { });
+            jest.spyOn(metrics_1.default.genaiCompletionTokens, 'add').mockImplementation(() => { });
+            jest.spyOn(metrics_1.default.genaiClientOperationDuration, 'record').mockImplementation(() => { });
+            jest.spyOn(metrics_1.default.genaiCost, 'record').mockImplementation(() => { });
+        });
+        it('should not call metrics for NaN, undefined, or non-number values', () => {
+            const span = {
+                setAttribute: jest.fn(),
+                setStatus: jest.fn(),
+                setAttributes: jest.fn(),
+                attributes: {
+                    [semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS]: NaN,
+                    [semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS]: undefined,
+                    duration: 'not-a-number',
+                },
+            };
+            const baseAttributes = {
+                model: 'gpt-4',
+                user: 'user1',
+                cost: 'not-a-number',
+                aiSystem: 'openai',
+                genAIEndpoint: 'endpoint',
+            };
+            base_wrapper_1.default.setBaseSpanAttributes(span, baseAttributes);
+            base_wrapper_1.default.recordMetrics(span, baseAttributes);
+            expect(metrics_1.default.genaiPromptTokens.add).not.toHaveBeenCalled();
+            expect(metrics_1.default.genaiCompletionTokens.add).not.toHaveBeenCalled();
+            expect(metrics_1.default.genaiClientOperationDuration.record).not.toHaveBeenCalled();
+            expect(metrics_1.default.genaiCost.record).not.toHaveBeenCalled();
+        });
+        it('should call metrics for zero and negative values', () => {
+            const span = {
+                setAttribute: jest.fn(),
+                setStatus: jest.fn(),
+                setAttributes: jest.fn(),
+                attributes: {
+                    [semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS]: 0,
+                    [semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS]: -5,
+                    duration: -1.5,
+                },
+            };
+            const baseAttributes = {
+                model: 'gpt-4',
+                user: 'user1',
+                cost: 0,
+                aiSystem: 'openai',
+                genAIEndpoint: 'endpoint',
+            };
+            base_wrapper_1.default.setBaseSpanAttributes(span, baseAttributes);
+            base_wrapper_1.default.recordMetrics(span, baseAttributes);
+            expect(metrics_1.default.genaiPromptTokens.add).toHaveBeenCalledWith(0, expect.any(Object));
+            expect(metrics_1.default.genaiCompletionTokens.add).toHaveBeenCalledWith(-5, expect.any(Object));
+            expect(metrics_1.default.genaiClientOperationDuration.record).toHaveBeenCalledWith(-1.5e-9, expect.any(Object));
+            expect(metrics_1.default.genaiCost.record).toHaveBeenCalledWith(0, expect.any(Object));
+        });
+        it('should convert string cost to number if possible', () => {
+            const span = {
+                setAttribute: jest.fn(),
+                setStatus: jest.fn(),
+                setAttributes: jest.fn(),
+                attributes: {
+                    [semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS]: 1,
+                    [semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS]: 2,
+                    duration: 3,
+                },
+            };
+            const baseAttributes = {
+                model: 'gpt-4',
+                user: 'user1',
+                cost: '1.23',
+                aiSystem: 'openai',
+                genAIEndpoint: 'endpoint',
+            };
+            base_wrapper_1.default.setBaseSpanAttributes(span, baseAttributes);
+            base_wrapper_1.default.recordMetrics(span, baseAttributes);
+            expect(metrics_1.default.genaiCost.record).toHaveBeenCalledWith(1.23, expect.any(Object));
+        });
+    });
+});
+//# sourceMappingURL=base-wrapper.test.js.map