shield360 1.0.0

package/dist/instrumentation/__tests__/browser-use-wrapper.test.js

@@ -0,0 +1,284 @@
+"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 core_1 = require("@opentelemetry/core");
+const semantic_conventions_1 = require("@opentelemetry/semantic-conventions");
+const config_1 = __importDefault(require("../../config"));
+const helpers_1 = __importDefault(require("../../helpers"));
+const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+const wrapper_1 = __importDefault(require("../browser-use/wrapper"));
+const browser_use_1 = __importDefault(require("../browser-use"));
+jest.mock('../../config');
+jest.mock('../../helpers', () => ({
+    __esModule: true,
+    default: {
+        handleException: jest.fn(),
+        getChatModelCost: jest.fn().mockReturnValue(0.0025),
+    },
+    applyCustomSpanAttributes: jest.fn(),
+}));
+jest.mock('@opentelemetry/core', () => ({
+    isTracingSuppressed: jest.fn().mockReturnValue(false),
+}));
+describe('browser-use wrapper', () => {
+    let mockSpan;
+    let mockTracer;
+    beforeEach(() => {
+        mockSpan = {
+            setAttribute: jest.fn(),
+            setStatus: jest.fn(),
+            end: jest.fn(),
+        };
+        mockTracer = { startSpan: jest.fn(() => mockSpan) };
+        config_1.default.environment = 'shield360-testing';
+        config_1.default.applicationName = 'shield360-test';
+        config_1.default.captureMessageContent = true;
+        config_1.default.maxContentLength = null;
+        config_1.default.pricingInfo = {
+            chat: { 'gpt-4o-mini': { promptPrice: 0.15, completionPrice: 0.6 } },
+        };
+        core_1.isTracingSuppressed.mockReturnValue(false);
+    });
+    afterEach(() => {
+        jest.clearAllMocks();
+    });
+    function attrs() {
+        return Object.fromEntries(mockSpan.setAttribute.mock.calls);
+    }
+    it('wraps Agent.run into an invoke_agent span with browser-use metadata', async () => {
+        const mainSpan = {
+            setAttribute: jest.fn(),
+            setStatus: jest.fn(),
+            end: jest.fn(),
+        };
+        mockTracer.startSpan = jest.fn((name) => {
+            if (name === 'invoke_agent shopping-bot') {
+                return mainSpan;
+            }
+            return {
+                setAttribute: jest.fn(),
+                setStatus: jest.fn(),
+                end: jest.fn(),
+            };
+        });
+        class StubAgent {
+            constructor() {
+                this.name = 'shopping-bot';
+                this.task = 'Buy the cheapest laptop under $1000';
+                this.id = 'agent-1';
+                this.task_id = 'task-123';
+                this.session_id = 'session-456';
+                this.llm = { model: 'gpt-4o-mini' };
+                this.use_vision = true;
+                this.browser_profile = { headless: true, allowed_domains: ['example.com'] };
+            }
+            async run() {
+                return {
+                    history: [{ result: [{ is_success: true }] }],
+                    usage: { total_input_tokens: 10, total_output_tokens: 5 },
+                    final_result: () => 'done',
+                    total_duration_seconds: () => 1.5,
+                };
+            }
+        }
+        const agent = new StubAgent();
+        const wrapped = wrapper_1.default._patchAgentRun(mockTracer, '0.7.3')(StubAgent.prototype.run);
+        await wrapped.call(agent, 10);
+        expect(mockTracer.startSpan).toHaveBeenCalledWith('invoke_agent shopping-bot', {
+            kind: api_1.SpanKind.CLIENT,
+            attributes: {
+                [semantic_convention_1.default.GEN_AI_OPERATION]: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_AGENT,
+                [semantic_convention_1.default.GEN_AI_PROVIDER_NAME_OTEL]: semantic_convention_1.default.GEN_AI_SYSTEM_BROWSER_USE,
+            },
+        });
+        const a = Object.fromEntries(mainSpan.setAttribute.mock.calls);
+        expect(a[semantic_conventions_1.ATTR_TELEMETRY_SDK_NAME]).toBe('shield360');
+        expect(a[semantic_conventions_1.ATTR_SERVICE_NAME]).toBe('shield360-test');
+        expect(a[semantic_convention_1.default.ATTR_DEPLOYMENT_ENVIRONMENT]).toBe('shield360-testing');
+        expect(a[semantic_convention_1.default.GEN_AI_PROVIDER_NAME]).toBe('browser_use');
+        expect(a[semantic_convention_1.default.GEN_AI_OPERATION]).toBe('invoke_agent');
+        expect(a[semantic_convention_1.default.GEN_AI_OPERATION_TYPE]).toBe('invoke_agent');
+        expect(a[semantic_convention_1.default.GEN_AI_AGENT_TYPE]).toBe('browser');
+        expect(a[semantic_convention_1.default.GEN_AI_AGENT_ID]).toBe('agent-1');
+        expect(a[semantic_convention_1.default.GEN_AI_BROWSER_AGENT_TASK_ID]).toBe('task-123');
+        expect(a[semantic_convention_1.default.GEN_AI_AGENT_SESSION_ID]).toBe('session-456');
+        expect(a[semantic_convention_1.default.GEN_AI_AGENT_USE_VISION]).toBe(true);
+        expect(a[semantic_convention_1.default.GEN_AI_AGENT_HEADLESS]).toBe(true);
+        expect(a[semantic_convention_1.default.GEN_AI_AGENT_MAX_STEPS]).toBe(10);
+        expect(a[semantic_convention_1.default.GEN_AI_AGENT_STEP_COUNT]).toBe(1);
+        expect(a[semantic_convention_1.default.GEN_AI_AGENT_FINAL_RESULT]).toBe('done');
+        expect(a[semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS]).toBe(10);
+        expect(a[semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS]).toBe(5);
+        expect(typeof a[semantic_convention_1.default.GEN_AI_CLIENT_OPERATION_DURATION]).toBe('number');
+        expect(mainSpan.setStatus).toHaveBeenCalledWith({ code: 1 });
+        expect(mainSpan.end).toHaveBeenCalledTimes(1);
+    });
+    it('records Agent.run errors and still ends the span', async () => {
+        class StubAgent {
+            constructor() {
+                this.task = 'fail task';
+            }
+            async run() {
+                throw new Error('agent run failed');
+            }
+        }
+        const agent = new StubAgent();
+        const wrapped = wrapper_1.default._patchAgentRun(mockTracer, '0.7.3')(StubAgent.prototype.run);
+        await expect(wrapped.call(agent)).rejects.toThrow('agent run failed');
+        expect(helpers_1.default.handleException).toHaveBeenCalledWith(mockSpan, expect.objectContaining({ message: 'agent run failed' }));
+        expect(mockSpan.end).toHaveBeenCalledTimes(1);
+    });
+    it('bypasses instrumentation when tracing is suppressed', async () => {
+        core_1.isTracingSuppressed.mockReturnValue(true);
+        const original = jest.fn().mockResolvedValue({ ok: true });
+        const wrapped = wrapper_1.default._patchAgentRun(mockTracer, '0.7.3')(original);
+        await wrapped.call({ task: 'x' });
+        expect(original).toHaveBeenCalled();
+        expect(mockTracer.startSpan).not.toHaveBeenCalled();
+    });
+    it('wraps Controller.act into invoke_agent action spans (Python parity)', async () => {
+        class StubController {
+            constructor() {
+                this.registry = {
+                    get_action: () => ({ description: 'Click element by index.' }),
+                };
+            }
+            async act(action) {
+                return { is_success: true, extracted_content: `done:${Object.keys(action)[0]}` };
+            }
+        }
+        const controller = new StubController();
+        const wrapped = wrapper_1.default._patchControllerAct(mockTracer, '0.7.3')(StubController.prototype.act);
+        await wrapped.call(controller, { go_to_url: { url: 'https://example.com/page' } }, {});
+        expect(mockTracer.startSpan).toHaveBeenCalledWith('invoke_agent go_to_url', {
+            kind: api_1.SpanKind.INTERNAL,
+            attributes: {
+                [semantic_convention_1.default.GEN_AI_OPERATION]: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_AGENT,
+                [semantic_convention_1.default.GEN_AI_PROVIDER_NAME_OTEL]: semantic_convention_1.default.GEN_AI_SYSTEM_BROWSER_USE,
+            },
+        });
+        const a = attrs();
+        expect(a[semantic_convention_1.default.GEN_AI_OPERATION]).toBe('invoke_agent');
+        expect(a[semantic_convention_1.default.GEN_AI_OPERATION_TYPE]).toBe('execute_task');
+        expect(a[semantic_convention_1.default.GEN_AI_ACTION_TYPE]).toBe('go_to_url');
+        expect(a[semantic_convention_1.default.GEN_AI_ACTION_INDEX]).toBe(1);
+        expect(a['gen_ai.action.url']).toBe('https://example.com/page');
+        expect(a[semantic_convention_1.default.GEN_AI_ACTION_SUCCESS]).toBe(true);
+        expect(a[semantic_convention_1.default.GEN_AI_TOOL_CALL_ARGUMENTS]).toBe('{"url":"https://example.com/page"}');
+        expect(mockSpan.end).toHaveBeenCalledTimes(1);
+    });
+    it('records controller errors and still ends the action span', async () => {
+        class StubController {
+            constructor() {
+                this.registry = {
+                    get_action: () => ({ description: 'Navigate to URL.' }),
+                };
+            }
+            async act() {
+                throw new Error('navigation failed');
+            }
+        }
+        const controller = new StubController();
+        const wrapped = wrapper_1.default._patchControllerAct(mockTracer, '0.7.3')(StubController.prototype.act);
+        await expect(wrapped.call(controller, { go_to_url: { url: 'https://example.com' } }, {})).rejects.toThrow('navigation failed');
+        expect(helpers_1.default.handleException).toHaveBeenCalledWith(mockSpan, expect.objectContaining({ message: 'navigation failed' }));
+        expect(mockSpan.end).toHaveBeenCalledTimes(1);
+    });
+    it('wraps Agent.step into execute_task spans', async () => {
+        class StubAgent {
+            constructor() {
+                this.name = 'research-bot';
+                this.step_count = 2;
+                this.llm = { model: 'gpt-4o' };
+            }
+            async step() {
+                return { is_success: true };
+            }
+        }
+        const agent = new StubAgent();
+        const wrapped = wrapper_1.default._patchAgentStep(mockTracer, '0.7.3')(StubAgent.prototype.step);
+        await wrapped.call(agent);
+        expect(mockTracer.startSpan).toHaveBeenCalledWith('execute_task step 2', { kind: api_1.SpanKind.CLIENT });
+        expect(attrs()[semantic_convention_1.default.GEN_AI_OPERATION]).toBe('step');
+        expect(attrs()[semantic_convention_1.default.GEN_AI_OPERATION_TYPE]).toBe('execute_task');
+    });
+    it('wraps Agent.pause into browser lifecycle spans', async () => {
+        class StubAgent {
+            constructor() {
+                this.name = 'research-bot';
+            }
+            async pause() {
+                return undefined;
+            }
+        }
+        const agent = new StubAgent();
+        const wrapped = wrapper_1.default._patchAgentLifecycle(mockTracer, 'pause', '0.7.3')(StubAgent.prototype.pause);
+        await wrapped.call(agent);
+        expect(mockTracer.startSpan).toHaveBeenCalledWith('browser pause', { kind: api_1.SpanKind.CLIENT });
+        expect(attrs()[semantic_convention_1.default.GEN_AI_OPERATION]).toBe('pause');
+        expect(attrs()[semantic_convention_1.default.GEN_AI_OPERATION_TYPE]).toBe('invoke_workflow');
+    });
+    it('creates step and action spans from run history when step hooks do not fire', async () => {
+        class StubAgent {
+            constructor() {
+                this.name = 'history-bot';
+                this.task = 'complete checkout';
+            }
+            async run(_options) {
+                return {
+                    history: [
+                        {
+                            model_output: { action: [{ click: { index: 1 } }] },
+                            result: [{ is_success: true }],
+                            state: { url: 'https://shop.example/cart', title: 'Cart' },
+                        },
+                    ],
+                    usage: { total_input_tokens: 3, total_output_tokens: 2 },
+                    final_result: () => 'checked out',
+                };
+            }
+        }
+        const agent = new StubAgent();
+        const wrapped = wrapper_1.default._patchAgentRun(mockTracer, '0.7.3')(StubAgent.prototype.run);
+        await wrapped.call(agent);
+        const spanNames = mockTracer.startSpan.mock.calls.map((call) => call[0]);
+        expect(spanNames).toContain('invoke_agent history-bot');
+        expect(spanNames).toContain('execute_task step 1');
+        expect(spanNames).toContain('invoke_agent click');
+    });
+});
+describe('BrowserUseInstrumentation patch targets', () => {
+    // eslint-disable-next-line @typescript-eslint/no-var-requires
+    const { isWrapped } = require('@opentelemetry/instrumentation');
+    it('wraps Agent lifecycle methods and Controller.act', () => {
+        class Agent {
+            async run() {
+                return null;
+            }
+            async step() {
+                return null;
+            }
+            async pause() { }
+            async resume() { }
+            async stop() { }
+        }
+        class Controller {
+            async act() {
+                return null;
+            }
+        }
+        const fakeModule = { Agent, Controller };
+        const instrumentation = new browser_use_1.default();
+        instrumentation.manualPatch(fakeModule, '0.7.3');
+        expect(isWrapped(fakeModule.Agent.prototype.run)).toBe(true);
+        expect(isWrapped(fakeModule.Agent.prototype.step)).toBe(true);
+        expect(isWrapped(fakeModule.Agent.prototype.pause)).toBe(true);
+        expect(isWrapped(fakeModule.Agent.prototype.resume)).toBe(true);
+        expect(isWrapped(fakeModule.Agent.prototype.stop)).toBe(true);
+        expect(isWrapped(fakeModule.Controller.prototype.act)).toBe(true);
+    });
+});
+//# sourceMappingURL=browser-use-wrapper.test.js.map

package/dist/instrumentation/__tests__/browser-use-wrapper.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"browser-use-wrapper.test.js","sourceRoot":"","sources":["../../../src/instrumentation/__tests__/browser-use-wrapper.test.ts"],"names":[],"mappings":";;;;;AAAA,4CAA8C;AAC9C,8CAA0D;AAC1D,8EAAiG;AACjG,0DAA2C;AAC3C,4DAA4C;AAC5C,oFAA2D;AAC3D,qEAAuD;AACvD,iEAAuD;AAEvD,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;AAC1B,IAAI,CAAC,IAAI,CAAC,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC;IAChC,UAAU,EAAE,IAAI;IAChB,OAAO,EAAE;QACP,eAAe,EAAE,IAAI,CAAC,EAAE,EAAE;QAC1B,gBAAgB,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,MAAM,CAAC;KACpD;IACD,yBAAyB,EAAE,IAAI,CAAC,EAAE,EAAE;CACrC,CAAC,CAAC,CAAC;AACJ,IAAI,CAAC,IAAI,CAAC,qBAAqB,EAAE,GAAG,EAAE,CAAC,CAAC;IACtC,mBAAmB,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC;CACtD,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,IAAI,QAAa,CAAC;IAClB,IAAI,UAAe,CAAC;IAEpB,UAAU,CAAC,GAAG,EAAE;QACd,QAAQ,GAAG;YACT,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;YACvB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;YACpB,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;SACf,CAAC;QACF,UAAU,GAAG,EAAE,SAAS,EAAE,IAAI,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC;QAEnD,gBAAuB,CAAC,WAAW,GAAG,mBAAmB,CAAC;QAC1D,gBAAuB,CAAC,eAAe,GAAG,gBAAgB,CAAC;QAC3D,gBAAuB,CAAC,qBAAqB,GAAG,IAAI,CAAC;QACrD,gBAAuB,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAChD,gBAAuB,CAAC,WAAW,GAAG;YACrC,IAAI,EAAE,EAAE,aAAa,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,eAAe,EAAE,GAAG,EAAE,EAAE;SACrE,CAAC;QACD,0BAAiC,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,SAAS,KAAK;QACZ,OAAO,MAAM,CAAC,WAAW,CAAE,QAAQ,CAAC,YAA0B,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC7E,CAAC;IAED,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACnF,MAAM,QAAQ,GAAG;YACf,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;YACvB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;YACpB,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;SACf,CAAC;QACF,UAAU,CAAC,SAAS,GAAG,IAAI,CAAC,EAAE,CAAC,CAAC,IAAY,EAAE,EAAE;YAC9C,IAAI,IAAI,KAAK,2BAA2B,EAAE,CAAC;gBACzC,OAAO,QAAQ,CAAC;YAClB,CAAC;YACD,OAAO;gBACL,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;gBACvB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;gBACpB,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;aACf,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,MAAM,SAAS;YAAf;gBACE,SAAI,GAAG,cAAc,CAAC;gBACtB,SAAI,GAAG,qCAAqC,CAAC;gBAC7C,OAAE,GAAG,SAAS,CAAC;gBACf,YAAO,GAAG,UAAU,CAAC;gBACrB,eAAU,GAAG,aAAa,CAAC;gBAC3B,QAAG,GAAG,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC;gBAC/B,eAAU,GAAG,IAAI,CAAC;gBAClB,oBAAe,GAAG,EAAE,QAAQ,EAAE,IAAI,EAAE,eAAe,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC;YAUzE,CAAC;YARC,KAAK,CAAC,GAAG;gBACP,OAAO;oBACL,OAAO,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;oBAC7C,KAAK,EAAE,EAAE,kBAAkB,EAAE,EAAE,EAAE,mBAAmB,EAAE,CAAC,EAAE;oBACzD,YAAY,EAAE,GAAG,EAAE,CAAC,MAAM;oBAC1B,sBAAsB,EAAE,GAAG,EAAE,CAAC,GAAG;iBAClC,CAAC;YACJ,CAAC;SACF;QAED,MAAM,KAAK,GAAG,IAAI,SAAS,EAAE,CAAC;QAC9B,MAAM,OAAO,GAAG,iBAAiB,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QAC/F,MAAM,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAE9B,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,2BAA2B,EAAE;YAC7E,IAAI,EAAE,cAAQ,CAAC,MAAM;YACrB,UAAU,EAAE;gBACV,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,6BAAkB,CAAC,2BAA2B;gBACrF,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAC5C,6BAAkB,CAAC,yBAAyB;aAC/C;SACF,CAAC,CAAC;QAEH,MAAM,CAAC,GAAG,MAAM,CAAC,WAAW,CAAE,QAAQ,CAAC,YAA0B,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC9E,MAAM,CAAC,CAAC,CAAC,8CAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACrD,MAAM,CAAC,CAAC,CAAC,wCAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACpD,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,2BAA2B,CAAC,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QACpF,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QACvE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACpE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACzE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAChE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC9D,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,4BAA4B,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC5E,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,uBAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAC1E,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,uBAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACjE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/D,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,sBAAsB,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC9D,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,uBAAuB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC9D,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACrE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjE,MAAM,CAAC,OAAO,CAAC,CAAC,6BAAkB,CAAC,gCAAgC,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACrF,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;QAC7D,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,MAAM,SAAS;YAAf;gBACE,SAAI,GAAG,WAAW,CAAC;YAKrB,CAAC;YAHC,KAAK,CAAC,GAAG;gBACP,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;YACtC,CAAC;SACF;QAED,MAAM,KAAK,GAAG,IAAI,SAAS,EAAE,CAAC;QAC9B,MAAM,OAAO,GAAG,iBAAiB,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QAE/F,MAAM,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;QACtE,MAAM,CAAC,iBAAe,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1D,QAAQ,EACR,MAAM,CAAC,gBAAgB,CAAC,EAAE,OAAO,EAAE,kBAAkB,EAAE,CAAC,CACzD,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QAClE,0BAAiC,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QACzD,MAAM,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC;QAC3D,MAAM,OAAO,GAAG,iBAAiB,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,QAAQ,CAAC,CAAC;QAEhF,MAAM,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC;QAClC,MAAM,CAAC,QAAQ,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACpC,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;QACnF,MAAM,cAAc;YAApB;gBACE,aAAQ,GAAG;oBACT,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,WAAW,EAAE,yBAAyB,EAAE,CAAC;iBAC/D,CAAC;YAKJ,CAAC;YAHC,KAAK,CAAC,GAAG,CAAC,MAA+B;gBACvC,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,iBAAiB,EAAE,QAAQ,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;YACnF,CAAC;SACF;QAED,MAAM,UAAU,GAAG,IAAI,cAAc,EAAE,CAAC;QACxC,MAAM,OAAO,GAAG,iBAAiB,CAAC,mBAAmB,CAAC,UAAU,EAAE,OAAO,CAAC,CACxE,cAAc,CAAC,SAAS,CAAC,GAAG,CAC7B,CAAC;QACF,MAAM,OAAO,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,EAAE,GAAG,EAAE,0BAA0B,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;QAEvF,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,wBAAwB,EAAE;YAC1E,IAAI,EAAE,cAAQ,CAAC,QAAQ;YACvB,UAAU,EAAE;gBACV,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,6BAAkB,CAAC,2BAA2B;gBACrF,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAC5C,6BAAkB,CAAC,yBAAyB;aAC/C;SACF,CAAC,CAAC;QAEH,MAAM,CAAC,GAAG,KAAK,EAAE,CAAC;QAClB,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACpE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;QACzE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACnE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1D,MAAM,CAAC,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;QAChE,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/D,MAAM,CAAC,CAAC,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,CAAC,CAAC,IAAI,CAC3D,oCAAoC,CACrC,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;QACxE,MAAM,cAAc;YAApB;gBACE,aAAQ,GAAG;oBACT,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,WAAW,EAAE,kBAAkB,EAAE,CAAC;iBACxD,CAAC;YAKJ,CAAC;YAHC,KAAK,CAAC,GAAG;gBACP,MAAM,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC;YACvC,CAAC;SACF;QAED,MAAM,UAAU,GAAG,IAAI,cAAc,EAAE,CAAC;QACxC,MAAM,OAAO,GAAG,iBAAiB,CAAC,mBAAmB,CAAC,UAAU,EAAE,OAAO,CAAC,CACxE,cAAc,CAAC,SAAS,CAAC,GAAG,CAC7B,CAAC;QAEF,MAAM,MAAM,CACV,OAAO,CAAC,IAAI,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,EAAE,GAAG,EAAE,qBAAqB,EAAE,EAAE,EAAE,EAAE,CAAC,CAC5E,CAAC,OAAO,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAC;QAEvC,MAAM,CAAC,iBAAe,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1D,QAAQ,EACR,MAAM,CAAC,gBAAgB,CAAC,EAAE,OAAO,EAAE,mBAAmB,EAAE,CAAC,CAC1D,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,SAAS;YAAf;gBACE,SAAI,GAAG,cAAc,CAAC;gBACtB,eAAU,GAAG,CAAC,CAAC;gBACf,QAAG,GAAG,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;YAK5B,CAAC;YAHC,KAAK,CAAC,IAAI;gBACR,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;YAC9B,CAAC;SACF;QAED,MAAM,KAAK,GAAG,IAAI,SAAS,EAAE,CAAC;QAC9B,MAAM,OAAO,GAAG,iBAAiB,CAAC,eAAe,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QACjG,MAAM,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAE1B,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,qBAAqB,EAAE,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QACpG,MAAM,CAAC,KAAK,EAAE,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAClE,MAAM,CAAC,KAAK,EAAE,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IACjF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;QAC9D,MAAM,SAAS;YAAf;gBACE,SAAI,GAAG,cAAc,CAAC;YAKxB,CAAC;YAHC,KAAK,CAAC,KAAK;gBACT,OAAO,SAAS,CAAC;YACnB,CAAC;SACF;QAED,MAAM,KAAK,GAAG,IAAI,SAAS,EAAE,CAAC;QAC9B,MAAM,OAAO,GAAG,iBAAiB,CAAC,oBAAoB,CAAC,UAAU,EAAE,OAAO,EAAE,OAAO,CAAC,CAClF,SAAS,CAAC,SAAS,CAAC,KAAK,CAC1B,CAAC;QACF,MAAM,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAE1B,MAAM,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,eAAe,EAAE,EAAE,IAAI,EAAE,cAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;QAC9F,MAAM,CAAC,KAAK,EAAE,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACnE,MAAM,CAAC,KAAK,EAAE,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;IACpF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4EAA4E,EAAE,KAAK,IAAI,EAAE;QAC1F,MAAM,SAAS;YAAf;gBACE,SAAI,GAAG,aAAa,CAAC;gBACrB,SAAI,GAAG,mBAAmB,CAAC;YAe7B,CAAC;YAbC,KAAK,CAAC,GAAG,CAAC,QAAkC;gBAC1C,OAAO;oBACL,OAAO,EAAE;wBACP;4BACE,YAAY,EAAE,EAAE,MAAM,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC,EAAE;4BACnD,MAAM,EAAE,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;4BAC9B,KAAK,EAAE,EAAE,GAAG,EAAE,2BAA2B,EAAE,KAAK,EAAE,MAAM,EAAE;yBAC3D;qBACF;oBACD,KAAK,EAAE,EAAE,kBAAkB,EAAE,CAAC,EAAE,mBAAmB,EAAE,CAAC,EAAE;oBACxD,YAAY,EAAE,GAAG,EAAE,CAAC,aAAa;iBAClC,CAAC;YACJ,CAAC;SACF;QAED,MAAM,KAAK,GAAG,IAAI,SAAS,EAAE,CAAC;QAC9B,MAAM,OAAO,GAAG,iBAAiB,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,SAAS,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;QAC/F,MAAM,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAE1B,MAAM,SAAS,GAAI,UAAU,CAAC,SAAuB,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAW,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QAC/F,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,0BAA0B,CAAC,CAAC;QACxD,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,qBAAqB,CAAC,CAAC;QACnD,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,oBAAoB,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,yCAAyC,EAAE,GAAG,EAAE;IACvD,8DAA8D;IAC9D,MAAM,EAAE,SAAS,EAAE,GAAG,OAAO,CAAC,gCAAgC,CAAC,CAAC;IAEhE,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;QAC1D,MAAM,KAAK;YACT,KAAK,CAAC,GAAG;gBACP,OAAO,IAAI,CAAC;YACd,CAAC;YACD,KAAK,CAAC,IAAI;gBACR,OAAO,IAAI,CAAC;YACd,CAAC;YACD,KAAK,CAAC,KAAK,KAAI,CAAC;YAChB,KAAK,CAAC,MAAM,KAAI,CAAC;YACjB,KAAK,CAAC,IAAI,KAAI,CAAC;SAChB;QAED,MAAM,UAAU;YACd,KAAK,CAAC,GAAG;gBACP,OAAO,IAAI,CAAC;YACd,CAAC;SACF;QAED,MAAM,UAAU,GAAG,EAAE,KAAK,EAAE,UAAU,EAAS,CAAC;QAChD,MAAM,eAAe,GAAG,IAAI,qBAAyB,EAAE,CAAC;QACxD,eAAe,CAAC,WAAW,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QAEjD,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7D,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9D,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/D,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChE,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9D,MAAM,CAAC,SAAS,CAAC,UAAU,CAAC,UAAU,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/instrumentation/__tests__/cache-pricing.test.d.ts

@@ -0,0 +1,15 @@
+/**
+ * Unit tests for cache-aware cost calculation in `Shield360Helper.getChatModelCost`.
+ *
+ * Mirrors the Python `test_cache_pricing.py` coverage and validates the two
+ * provider token-accounting conventions:
+ *   - Anthropic native API -> promptTokens is exclusive of cache tokens.
+ *   - OpenAI / LangChain -> promptTokens is inclusive of cache tokens.
+ *
+ * The same cached tokens must never be billed twice, and models without cache
+ * pricing must produce exactly the legacy cost regardless of convention.
+ *
+ * Also covers instrumentor call-site patterns (claude-agent-sdk inclusive sum,
+ * bedrock exclusive, openai cache_write_tokens, strands/cursor heuristic).
+ */
+export {};

package/dist/instrumentation/__tests__/cache-pricing.test.js

@@ -0,0 +1,155 @@
+"use strict";
+/**
+ * Unit tests for cache-aware cost calculation in `Shield360Helper.getChatModelCost`.
+ *
+ * Mirrors the Python `test_cache_pricing.py` coverage and validates the two
+ * provider token-accounting conventions:
+ *   - Anthropic native API -> promptTokens is exclusive of cache tokens.
+ *   - OpenAI / LangChain -> promptTokens is inclusive of cache tokens.
+ *
+ * The same cached tokens must never be billed twice, and models without cache
+ * pricing must produce exactly the legacy cost regardless of convention.
+ *
+ * Also covers instrumentor call-site patterns (claude-agent-sdk inclusive sum,
+ * bedrock exclusive, openai cache_write_tokens, strands/cursor heuristic).
+ */
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const helpers_1 = __importDefault(require("../../helpers"));
+const PRICING = {
+    chat: {
+        'claude-3-5-sonnet-20241022': {
+            promptPrice: 0.003,
+            completionPrice: 0.015,
+            cacheReadPrice: 0.0003,
+            cacheCreationPrice: 0.00375,
+        },
+        'model-no-cache-price': {
+            promptPrice: 0.003,
+            completionPrice: 0.015,
+        },
+        'gpt-cache-write': {
+            promptPrice: 0.0025,
+            completionPrice: 0.01,
+            cacheReadPrice: 0.00125,
+            cacheCreationPrice: 0.003125,
+        },
+    },
+};
+const MODEL = 'claude-3-5-sonnet-20241022';
+// 200 uncached + 5000 cache-read + 1000 cache-creation + 300 output
+const RAW_INPUT = 200;
+const CACHE_READ = 5000;
+const CACHE_CREATION = 1000;
+const OUTPUT = 300;
+const INCLUSIVE_INPUT = RAW_INPUT + CACHE_READ + CACHE_CREATION;
+function groundTruth() {
+    return ((RAW_INPUT / 1000) * 0.003 +
+        (OUTPUT / 1000) * 0.015 +
+        (CACHE_READ / 1000) * 0.0003 +
+        (CACHE_CREATION / 1000) * 0.00375);
+}
+describe('Shield360Helper.getChatModelCost — cache pricing', () => {
+    describe('no cache tokens', () => {
+        it('matches the legacy formula', () => {
+            const cost = helpers_1.default.getChatModelCost(MODEL, PRICING, 1000, 500);
+            expect(cost).toBeCloseTo((1000 / 1000) * 0.003 + (500 / 1000) * 0.015, 10);
+        });
+        it('returns 0 for an unknown model', () => {
+            expect(helpers_1.default.getChatModelCost('nope', PRICING, 10, 10)).toBe(0);
+        });
+        it('returns 0 when pricing info is missing', () => {
+            expect(helpers_1.default.getChatModelCost(MODEL, {}, 10, 10)).toBe(0);
+        });
+    });
+    describe('Anthropic convention (promptTokens exclusive of cache)', () => {
+        it('adds cache cost on top of the prompt base', () => {
+            const cost = helpers_1.default.getChatModelCost(MODEL, PRICING, 800, 500, 100, 100);
+            const expected = (800 / 1000) * 0.003 +
+                (500 / 1000) * 0.015 +
+                (100 / 1000) * 0.0003 +
+                (100 / 1000) * 0.00375;
+            expect(cost).toBeCloseTo(expected, 10);
+        });
+        it('leaves the prompt base untouched when no cache price exists', () => {
+            const cost = helpers_1.default.getChatModelCost('model-no-cache-price', PRICING, 800, 500, 100, 100);
+            expect(cost).toBeCloseTo((800 / 1000) * 0.003 + (500 / 1000) * 0.015, 10);
+        });
+    });
+    describe('OpenAI / LangChain convention (promptTokens inclusive of cache)', () => {
+        it('does not bill cache tokens twice', () => {
+            // 1000 total input = 800 uncached + 100 cache read + 100 cache creation
+            const cost = helpers_1.default.getChatModelCost(MODEL, PRICING, 1000, 500, 100, 100, true);
+            const expected = (800 / 1000) * 0.003 +
+                (500 / 1000) * 0.015 +
+                (100 / 1000) * 0.0003 +
+                (100 / 1000) * 0.00375;
+            expect(cost).toBeCloseTo(expected, 10);
+        });
+        it('matches the exclusive call describing the same request', () => {
+            const inclusive = helpers_1.default.getChatModelCost(MODEL, PRICING, 1000, 500, 100, 100, true);
+            const exclusive = helpers_1.default.getChatModelCost(MODEL, PRICING, 800, 500, 100, 100);
+            expect(inclusive).toBeCloseTo(exclusive, 10);
+        });
+        it('matches the legacy total when no cache price exists (no regression)', () => {
+            const cost = helpers_1.default.getChatModelCost('model-no-cache-price', PRICING, 1000, 500, 100, 100, true);
+            expect(cost).toBeCloseTo((1000 / 1000) * 0.003 + (500 / 1000) * 0.015, 10);
+        });
+        it('handles OpenAI-style cache read only (no cache creation)', () => {
+            // OpenAI: prompt_tokens includes cached_tokens; no cache-creation charge.
+            const pricing = {
+                chat: { 'gpt-cache': { promptPrice: 0.0025, completionPrice: 0.01, cacheReadPrice: 0.00125 } },
+            };
+            const cost = helpers_1.default.getChatModelCost('gpt-cache', pricing, 1000, 200, 400, 0, true);
+            const expected = (600 / 1000) * 0.0025 + (200 / 1000) * 0.01 + (400 / 1000) * 0.00125;
+            expect(cost).toBeCloseTo(expected, 10);
+        });
+        it('clamps the prompt base at 0 for malformed token data', () => {
+            const cost = helpers_1.default.getChatModelCost(MODEL, PRICING, 100, 0, 200, 200, true);
+            const expected = (200 / 1000) * 0.0003 + (200 / 1000) * 0.00375;
+            expect(cost).toBeCloseTo(expected, 10);
+        });
+    });
+    describe('instrumentor call-site patterns', () => {
+        it('claude-agent-sdk: inclusive sum + cache args re-prices correctly', () => {
+            // extractUsage builds input = raw + cache_read + cache_creation
+            const cost = helpers_1.default.getChatModelCost(MODEL, PRICING, INCLUSIVE_INPUT, OUTPUT, CACHE_READ, CACHE_CREATION, true);
+            expect(cost).toBeCloseTo(groundTruth(), 10);
+        });
+        it('bedrock: exclusive cacheRead/cacheWrite does not undercharge', () => {
+            const cost = helpers_1.default.getChatModelCost(MODEL, PRICING, RAW_INPUT, OUTPUT, CACHE_READ, CACHE_CREATION, false);
+            expect(cost).toBeCloseTo(groundTruth(), 10);
+        });
+        it('openai: maps cache_write_tokens into creation (inclusive)', () => {
+            // prompt_tokens inclusive; cache_write_tokens billed at cacheCreationPrice
+            const cost = helpers_1.default.getChatModelCost('gpt-cache-write', PRICING, 1000, 200, 400, 100, true);
+            const expected = (500 / 1000) * 0.0025 +
+                (200 / 1000) * 0.01 +
+                (400 / 1000) * 0.00125 +
+                (100 / 1000) * 0.003125;
+            expect(cost).toBeCloseTo(expected, 10);
+        });
+        it('strands/cursor heuristic: input >= cache sum → inclusive', () => {
+            const cacheTotal = CACHE_READ + CACHE_CREATION;
+            const inclusive = INCLUSIVE_INPUT >= cacheTotal;
+            expect(inclusive).toBe(true);
+            const cost = helpers_1.default.getChatModelCost(MODEL, PRICING, INCLUSIVE_INPUT, OUTPUT, CACHE_READ, CACHE_CREATION, inclusive);
+            expect(cost).toBeCloseTo(groundTruth(), 10);
+        });
+        it('strands/cursor heuristic: input < cache sum → exclusive', () => {
+            const cacheTotal = CACHE_READ + CACHE_CREATION;
+            const inclusive = RAW_INPUT >= cacheTotal;
+            expect(inclusive).toBe(false);
+            const cost = helpers_1.default.getChatModelCost(MODEL, PRICING, RAW_INPUT, OUTPUT, CACHE_READ, CACHE_CREATION, inclusive);
+            expect(cost).toBeCloseTo(groundTruth(), 10);
+        });
+        it('billing inclusive total at prompt price overcharges (documents pre-fix)', () => {
+            const overcharged = helpers_1.default.getChatModelCost(MODEL, PRICING, INCLUSIVE_INPUT, OUTPUT);
+            expect(overcharged).toBeCloseTo((INCLUSIVE_INPUT / 1000) * 0.003 + (OUTPUT / 1000) * 0.015, 10);
+            expect(overcharged).toBeGreaterThan(groundTruth());
+        });
+    });
+});
+//# sourceMappingURL=cache-pricing.test.js.map

package/dist/instrumentation/__tests__/cache-pricing.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cache-pricing.test.js","sourceRoot":"","sources":["../../../src/instrumentation/__tests__/cache-pricing.test.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;GAaG;;;;;AAEH,4DAA4C;AAE5C,MAAM,OAAO,GAAG;IACd,IAAI,EAAE;QACJ,4BAA4B,EAAE;YAC5B,WAAW,EAAE,KAAK;YAClB,eAAe,EAAE,KAAK;YACtB,cAAc,EAAE,MAAM;YACtB,kBAAkB,EAAE,OAAO;SAC5B;QACD,sBAAsB,EAAE;YACtB,WAAW,EAAE,KAAK;YAClB,eAAe,EAAE,KAAK;SACvB;QACD,iBAAiB,EAAE;YACjB,WAAW,EAAE,MAAM;YACnB,eAAe,EAAE,IAAI;YACrB,cAAc,EAAE,OAAO;YACvB,kBAAkB,EAAE,QAAQ;SAC7B;KACF;CACF,CAAC;AAEF,MAAM,KAAK,GAAG,4BAA4B,CAAC;AAE3C,oEAAoE;AACpE,MAAM,SAAS,GAAG,GAAG,CAAC;AACtB,MAAM,UAAU,GAAG,IAAI,CAAC;AACxB,MAAM,cAAc,GAAG,IAAI,CAAC;AAC5B,MAAM,MAAM,GAAG,GAAG,CAAC;AACnB,MAAM,eAAe,GAAG,SAAS,GAAG,UAAU,GAAG,cAAc,CAAC;AAEhE,SAAS,WAAW;IAClB,OAAO,CACL,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,KAAK;QAC1B,CAAC,MAAM,GAAG,IAAI,CAAC,GAAG,KAAK;QACvB,CAAC,UAAU,GAAG,IAAI,CAAC,GAAG,MAAM;QAC5B,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,OAAO,CAClC,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,kDAAkD,EAAE,GAAG,EAAE;IAChE,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,IAAI,GAAG,iBAAe,CAAC,gBAAgB,CAAC,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,CAAC,CAAC;YACzE,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,CAAC,CAAC,IAAI,GAAG,IAAI,CAAC,GAAG,KAAK,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,KAAK,EAAE,EAAE,CAAC,CAAC;QAC7E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;YACxC,MAAM,CAAC,iBAAe,CAAC,gBAAgB,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,CAAC,iBAAe,CAAC,gBAAgB,CAAC,KAAK,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,wDAAwD,EAAE,GAAG,EAAE;QACtE,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,IAAI,GAAG,iBAAe,CAAC,gBAAgB,CAAC,KAAK,EAAE,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YAClF,MAAM,QAAQ,GACZ,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,KAAK;gBACpB,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,KAAK;gBACpB,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,MAAM;gBACrB,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,OAAO,CAAC;YACzB,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;YACrE,MAAM,IAAI,GAAG,iBAAe,CAAC,gBAAgB,CAC3C,sBAAsB,EACtB,OAAO,EACP,GAAG,EACH,GAAG,EACH,GAAG,EACH,GAAG,CACJ,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,CAAC,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,KAAK,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,KAAK,EAAE,EAAE,CAAC,CAAC;QAC5E,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iEAAiE,EAAE,GAAG,EAAE;QAC/E,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;YAC1C,wEAAwE;YACxE,MAAM,IAAI,GAAG,iBAAe,CAAC,gBAAgB,CAAC,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC;YACzF,MAAM,QAAQ,GACZ,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,KAAK;gBACpB,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,KAAK;gBACpB,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,MAAM;gBACrB,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,OAAO,CAAC;YACzB,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,MAAM,SAAS,GAAG,iBAAe,CAAC,gBAAgB,CAAC,KAAK,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC;YAC9F,MAAM,SAAS,GAAG,iBAAe,CAAC,gBAAgB,CAAC,KAAK,EAAE,OAAO,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC;YACvF,MAAM,CAAC,SAAS,CAAC,CAAC,WAAW,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,GAAG,EAAE;YAC7E,MAAM,IAAI,GAAG,iBAAe,CAAC,gBAAgB,CAC3C,sBAAsB,EACtB,OAAO,EACP,IAAI,EACJ,GAAG,EACH,GAAG,EACH,GAAG,EACH,IAAI,CACL,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,CAAC,CAAC,IAAI,GAAG,IAAI,CAAC,GAAG,KAAK,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,KAAK,EAAE,EAAE,CAAC,CAAC;QAC7E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;YAClE,0EAA0E;YAC1E,MAAM,OAAO,GAAG;gBACd,IAAI,EAAE,EAAE,WAAW,EAAE,EAAE,WAAW,EAAE,MAAM,EAAE,eAAe,EAAE,IAAI,EAAE,cAAc,EAAE,OAAO,EAAE,EAAE;aAC/F,CAAC;YACF,MAAM,IAAI,GAAG,iBAAe,CAAC,gBAAgB,CAAC,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC;YAC7F,MAAM,QAAQ,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,MAAM,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,IAAI,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,OAAO,CAAC;YACtF,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,IAAI,GAAG,iBAAe,CAAC,gBAAgB,CAAC,KAAK,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC;YACtF,MAAM,QAAQ,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,MAAM,GAAG,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,OAAO,CAAC;YAChE,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iCAAiC,EAAE,GAAG,EAAE;QAC/C,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;YAC1E,gEAAgE;YAChE,MAAM,IAAI,GAAG,iBAAe,CAAC,gBAAgB,CAC3C,KAAK,EACL,OAAO,EACP,eAAe,EACf,MAAM,EACN,UAAU,EACV,cAAc,EACd,IAAI,CACL,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,CAAC,WAAW,EAAE,EAAE,EAAE,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,GAAG,EAAE;YACtE,MAAM,IAAI,GAAG,iBAAe,CAAC,gBAAgB,CAC3C,KAAK,EACL,OAAO,EACP,SAAS,EACT,MAAM,EACN,UAAU,EACV,cAAc,EACd,KAAK,CACN,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,CAAC,WAAW,EAAE,EAAE,EAAE,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;YACnE,2EAA2E;YAC3E,MAAM,IAAI,GAAG,iBAAe,CAAC,gBAAgB,CAC3C,iBAAiB,EACjB,OAAO,EACP,IAAI,EACJ,GAAG,EACH,GAAG,EACH,GAAG,EACH,IAAI,CACL,CAAC;YACF,MAAM,QAAQ,GACZ,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,MAAM;gBACrB,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,IAAI;gBACnB,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,OAAO;gBACtB,CAAC,GAAG,GAAG,IAAI,CAAC,GAAG,QAAQ,CAAC;YAC1B,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;YAClE,MAAM,UAAU,GAAG,UAAU,GAAG,cAAc,CAAC;YAC/C,MAAM,SAAS,GAAG,eAAe,IAAI,UAAU,CAAC;YAChD,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC7B,MAAM,IAAI,GAAG,iBAAe,CAAC,gBAAgB,CAC3C,KAAK,EACL,OAAO,EACP,eAAe,EACf,MAAM,EACN,UAAU,EACV,cAAc,EACd,SAAS,CACV,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,CAAC,WAAW,EAAE,EAAE,EAAE,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yDAAyD,EAAE,GAAG,EAAE;YACjE,MAAM,UAAU,GAAG,UAAU,GAAG,cAAc,CAAC;YAC/C,MAAM,SAAS,GAAG,SAAS,IAAI,UAAU,CAAC;YAC1C,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC9B,MAAM,IAAI,GAAG,iBAAe,CAAC,gBAAgB,CAC3C,KAAK,EACL,OAAO,EACP,SAAS,EACT,MAAM,EACN,UAAU,EACV,cAAc,EACd,SAAS,CACV,CAAC;YACF,MAAM,CAAC,IAAI,CAAC,CAAC,WAAW,CAAC,WAAW,EAAE,EAAE,EAAE,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yEAAyE,EAAE,GAAG,EAAE;YACjF,MAAM,WAAW,GAAG,iBAAe,CAAC,gBAAgB,CAClD,KAAK,EACL,OAAO,EACP,eAAe,EACf,MAAM,CACP,CAAC;YACF,MAAM,CAAC,WAAW,CAAC,CAAC,WAAW,CAC7B,CAAC,eAAe,GAAG,IAAI,CAAC,GAAG,KAAK,GAAG,CAAC,MAAM,GAAG,IAAI,CAAC,GAAG,KAAK,EAC1D,EAAE,CACH,CAAC;YACF,MAAM,CAAC,WAAW,CAAC,CAAC,eAAe,CAAC,WAAW,EAAE,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/instrumentation/__tests__/chroma-trace-comparison.test.d.ts

@@ -0,0 +1,18 @@
+/**
+ * Cross-Language Trace Comparison Tests for ChromaDB Integration
+ *
+ * Verifies that TypeScript SDK generates traces consistent with the Python SDK
+ * for ChromaDB Collection operations.
+ *
+ * Python SDK reference: sdk/python/src/shield360/instrumentation/chroma/utils.py
+ *
+ * Key alignment:
+ *   - db.system.name = 'chroma'
+ *   - db.operation.name: INSERT (add), QUERY (query), GET (get), DELETE, PEEK, UPDATE, UPSERT
+ *   - db.collection.name
+ *   - server.address, server.port (default localhost:8000)
+ *   - db.vector.query.top_k for query
+ *   - db.filter for where-clause filtering
+ *   - db.query.summary
+ */
+export {};