shield360 1.0.0

package/dist/instrumentation/__tests__/google-ai-trace-comparison.test.js

@@ -0,0 +1,229 @@
+"use strict";
+/**
+ * Cross-Language Trace Comparison Tests for Google AI Studio Integration
+ */
+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("../google-ai/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('../../config');
+jest.mock('../../helpers');
+jest.mock('../base-wrapper');
+const mockTracer = api_1.trace.getTracer('test-tracer');
+describe('Google AI Studio Cross-Language Trace Comparison', () => {
+    let mockSpan;
+    beforeEach(() => {
+        mockSpan = mockTracer.startSpan('test-span');
+        mockSpan.setAttribute = jest.fn();
+        mockSpan.addEvent = jest.fn();
+        mockSpan.end = jest.fn();
+        mockSpan.setStatus = jest.fn();
+        config_1.default.environment = 'shield360-testing';
+        config_1.default.applicationName = 'shield360-test';
+        config_1.default.captureMessageContent = true;
+        config_1.default.pricingInfo = {};
+        config_1.default.disableEvents = false;
+        helpers_1.default.getChatModelCost = jest.fn().mockReturnValue(0.001);
+        helpers_1.default.handleException = jest.fn();
+        helpers_1.default.createStreamProxy = jest.fn().mockImplementation((stream, _generator) => stream);
+        helpers_1.default.buildInputMessages = jest.fn().mockReturnValue('[{"role":"user","parts":[{"type":"text","content":"What is Gemini?"}]}]');
+        helpers_1.default.buildOutputMessages = jest.fn().mockReturnValue('[{"role":"assistant","parts":[{"type":"text","content":"Gemini is Google\'s AI model"}],"finish_reason":"STOP"}]');
+        helpers_1.default.emitInferenceEvent = jest.fn();
+        base_wrapper_1.default.recordMetrics = jest.fn();
+        base_wrapper_1.default.setBaseSpanAttributes = jest.fn().mockImplementation((span, attrs) => {
+            span.setAttribute(semantic_convention_1.default.GEN_AI_PROVIDER_NAME_OTEL, attrs.aiSystem);
+            span.setAttribute(semantic_convention_1.default.GEN_AI_ENDPOINT, attrs.genAIEndpoint);
+            span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_MODEL, attrs.model);
+            if (attrs.cost !== undefined) {
+                span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_COST, attrs.cost);
+            }
+            if (attrs.serverAddress) {
+                span.setAttribute(semantic_convention_1.default.SERVER_ADDRESS, attrs.serverAddress);
+            }
+            if (attrs.serverPort !== undefined) {
+                span.setAttribute(semantic_convention_1.default.SERVER_PORT, attrs.serverPort);
+            }
+            span.setAttribute(semantic_convention_1.default.GEN_AI_SDK_VERSION, '1.9.0');
+        });
+    });
+    afterEach(() => {
+        jest.clearAllMocks();
+    });
+    describe('Generate Content Trace Consistency', () => {
+        it('should set same attributes as Python SDK', async () => {
+            const mockArgs = [
+                {
+                    contents: [
+                        { role: 'user', parts: [{ text: 'What is Gemini?' }] },
+                    ],
+                    config: {
+                        temperature: 0.7,
+                        maxOutputTokens: 100,
+                        topP: 0.95,
+                    },
+                },
+            ];
+            const mockResponse = {
+                response: {
+                    modelVersion: 'gemini-pro',
+                    text: () => "Gemini is Google's AI model",
+                    candidates: [
+                        {
+                            content: {
+                                parts: [{ text: "Gemini is Google's AI model" }],
+                                role: 'model',
+                            },
+                            finishReason: 'STOP',
+                        },
+                    ],
+                    usageMetadata: {
+                        promptTokenCount: 5,
+                        candidatesTokenCount: 10,
+                        totalTokenCount: 15,
+                    },
+                },
+            };
+            await wrapper_1.default._generateContent({
+                args: mockArgs,
+                genAIEndpoint: 'google.generativeai.models.generate_content',
+                response: mockResponse,
+                span: mockSpan,
+                requestModel: 'gemini-pro',
+            });
+            // Provider name: gcp.gemini (matches OTel semconv well-known value for Google AI Studio)
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_PROVIDER_NAME_OTEL, 'gcp.gemini');
+            // Request model
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_MODEL, 'gemini-pro');
+            // Response model
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_RESPONSE_MODEL, 'gemini-pro');
+            // Token usage
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, 5);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS, 10);
+            // Server address + port
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.SERVER_ADDRESS, 'generativelanguage.googleapis.com');
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.SERVER_PORT, 443);
+            // Request params from config
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_TEMPERATURE, 0.7);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, 100);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_TOP_P, 0.95);
+            // is_stream
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_IS_STREAM, false);
+            // Finish reason as array
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, ['STOP']);
+            // Output type
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_OUTPUT_TYPE, 'text');
+        });
+        it('should not set sentinel values for unset params', async () => {
+            const mockArgs = [
+                {
+                    contents: 'Hello',
+                    config: {},
+                },
+            ];
+            const mockResponse = {
+                response: {
+                    text: () => 'Hi there!',
+                    candidates: [{ content: { parts: [{ text: 'Hi there!' }], role: 'model' }, finishReason: 'STOP' }],
+                    usageMetadata: { promptTokenCount: 2, candidatesTokenCount: 3, totalTokenCount: 5 },
+                },
+            };
+            jest.spyOn(helpers_1.default, 'getChatModelCost').mockReturnValue(0);
+            await wrapper_1.default._generateContent({
+                args: mockArgs,
+                genAIEndpoint: 'google.generativeai.models.generate_content',
+                response: mockResponse,
+                span: mockSpan,
+                requestModel: 'gemini-2.0-flash',
+            });
+            const setAttrCalls = mockSpan.setAttribute.mock.calls.map((c) => c[0]);
+            expect(setAttrCalls).not.toContain(semantic_convention_1.default.GEN_AI_REQUEST_TEMPERATURE);
+            expect(setAttrCalls).not.toContain(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS);
+            expect(setAttrCalls).not.toContain(semantic_convention_1.default.GEN_AI_REQUEST_TOP_P);
+            expect(setAttrCalls).not.toContain(semantic_convention_1.default.GEN_AI_REQUEST_TOP_K);
+            expect(setAttrCalls).not.toContain(semantic_convention_1.default.GEN_AI_REQUEST_STOP_SEQUENCES);
+        });
+        it('should emit inference event independently of captureMessageContent', async () => {
+            config_1.default.captureMessageContent = false;
+            const mockArgs = [{ contents: 'Hello', config: {} }];
+            const mockResponse = {
+                response: {
+                    text: () => 'Hi!',
+                    candidates: [{ content: { parts: [{ text: 'Hi!' }], role: 'model' }, finishReason: 'STOP' }],
+                    usageMetadata: { promptTokenCount: 1, candidatesTokenCount: 1, totalTokenCount: 2 },
+                },
+            };
+            jest.spyOn(helpers_1.default, 'getChatModelCost').mockReturnValue(0);
+            await wrapper_1.default._generateContent({
+                args: mockArgs,
+                genAIEndpoint: 'google.generativeai.models.generate_content',
+                response: mockResponse,
+                span: mockSpan,
+                requestModel: 'gemini-2.0-flash',
+            });
+            expect(helpers_1.default.emitInferenceEvent).toHaveBeenCalledTimes(1);
+            const eventAttrs = helpers_1.default.emitInferenceEvent.mock.calls[0][1];
+            expect(eventAttrs[semantic_convention_1.default.GEN_AI_INPUT_MESSAGES]).toBeUndefined();
+            expect(eventAttrs[semantic_convention_1.default.GEN_AI_OUTPUT_MESSAGES]).toBeUndefined();
+        });
+        it('should call recordMetrics with cost from Shield360Config.pricingInfo', async () => {
+            config_1.default.pricingInfo = { chat: { 'gemini-pro': { promptPrice: 0.1, completionPrice: 0.2 } } };
+            jest.spyOn(helpers_1.default, 'getChatModelCost').mockReturnValue(0.0025);
+            const mockArgs = [{ contents: 'test', config: {} }];
+            const mockResponse = {
+                response: {
+                    modelVersion: 'gemini-pro',
+                    text: () => 'response',
+                    candidates: [{ content: { parts: [{ text: 'response' }], role: 'model' }, finishReason: 'STOP' }],
+                    usageMetadata: { promptTokenCount: 10, candidatesTokenCount: 5, totalTokenCount: 15 },
+                },
+            };
+            await wrapper_1.default._generateContent({
+                args: mockArgs,
+                genAIEndpoint: 'google.generativeai.models.generate_content',
+                response: mockResponse,
+                span: mockSpan,
+                requestModel: 'gemini-pro',
+            });
+            expect(base_wrapper_1.default.recordMetrics).toHaveBeenCalledWith(mockSpan, expect.objectContaining({
+                model: 'gemini-pro',
+                cost: 0.0025,
+                aiSystem: 'gcp.gemini',
+            }));
+        });
+        it('should handle function calls matching Python tool attributes', async () => {
+            const mockArgs = [{ contents: 'What is the weather?', config: {} }];
+            const mockResponse = {
+                response: {
+                    text: () => '',
+                    candidates: [{
+                            content: {
+                                parts: [{
+                                        functionCall: { name: 'get_weather', args: { location: 'NYC' } },
+                                    }],
+                                role: 'model',
+                            },
+                            finishReason: 'STOP',
+                        }],
+                    usageMetadata: { promptTokenCount: 5, candidatesTokenCount: 8, totalTokenCount: 13 },
+                },
+            };
+            jest.spyOn(helpers_1.default, 'getChatModelCost').mockReturnValue(0);
+            await wrapper_1.default._generateContent({
+                args: mockArgs,
+                genAIEndpoint: 'google.generativeai.models.generate_content',
+                response: mockResponse,
+                span: mockSpan,
+                requestModel: 'gemini-2.0-flash',
+            });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_TOOL_NAME, 'get_weather');
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_TOOL_ARGS, '{"location":"NYC"}');
+        });
+    });
+});
+//# sourceMappingURL=google-ai-trace-comparison.test.js.map

package/dist/instrumentation/__tests__/google-ai-trace-comparison.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"google-ai-trace-comparison.test.js","sourceRoot":"","sources":["../../../src/instrumentation/__tests__/google-ai-trace-comparison.test.ts"],"names":[],"mappings":";AAAA;;GAEG;;;;;AAEH,4CAA2C;AAC3C,mEAAmD;AACnD,0DAA2C;AAC3C,4DAA4C;AAC5C,mEAA0C;AAC1C,oFAA2D;AAE3D,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;AAC1B,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;AAC3B,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;AAE7B,MAAM,UAAU,GAAG,WAAK,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;AAElD,QAAQ,CAAC,kDAAkD,EAAE,GAAG,EAAE;IAChE,IAAI,QAAa,CAAC;IAElB,UAAU,CAAC,GAAG,EAAE;QACd,QAAQ,GAAG,UAAU,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QAC7C,QAAQ,CAAC,YAAY,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAClC,QAAQ,CAAC,QAAQ,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAC9B,QAAQ,CAAC,GAAG,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QACzB,QAAQ,CAAC,SAAS,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAE9B,gBAAuB,CAAC,WAAW,GAAG,mBAAmB,CAAC;QAC1D,gBAAuB,CAAC,eAAe,GAAG,gBAAgB,CAAC;QAC3D,gBAAuB,CAAC,qBAAqB,GAAG,IAAI,CAAC;QACrD,gBAAuB,CAAC,WAAW,GAAG,EAAE,CAAC;QACzC,gBAAuB,CAAC,aAAa,GAAG,KAAK,CAAC;QAE9C,iBAAuB,CAAC,gBAAgB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QAC5E,iBAAuB,CAAC,eAAe,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QACpD,iBAAuB,CAAC,iBAAiB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,CAAC,MAAM,EAAE,UAAU,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC;QACzG,iBAAuB,CAAC,kBAAkB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,yEAAyE,CAAC,CAAC;QAClJ,iBAAuB,CAAC,mBAAmB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,kHAAkH,CAAC,CAAC;QAC5L,iBAAuB,CAAC,kBAAkB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAEvD,sBAAmB,CAAC,aAAa,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAC9C,sBAAmB,CAAC,qBAAqB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;YACxF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;YAChF,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,eAAe,EAAE,KAAK,CAAC,aAAa,CAAC,CAAC;YAC3E,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;YACxE,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;gBAC7B,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,iBAAiB,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;YACtE,CAAC;YACD,IAAI,KAAK,CAAC,aAAa,EAAE,CAAC;gBACxB,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,cAAc,EAAE,KAAK,CAAC,aAAa,CAAC,CAAC;YAC5E,CAAC;YACD,IAAI,KAAK,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;gBACnC,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,WAAW,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;YACtE,CAAC;YACD,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,OAAO,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oCAAoC,EAAE,GAAG,EAAE;QAClD,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,QAAQ,GAAG;gBACf;oBACE,QAAQ,EAAE;wBACR,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC,EAAE;qBACvD;oBACD,MAAM,EAAE;wBACN,WAAW,EAAE,GAAG;wBAChB,eAAe,EAAE,GAAG;wBACpB,IAAI,EAAE,IAAI;qBACX;iBACF;aACF,CAAC;YAEF,MAAM,YAAY,GAAG;gBACnB,QAAQ,EAAE;oBACR,YAAY,EAAE,YAAY;oBAC1B,IAAI,EAAE,GAAG,EAAE,CAAC,6BAA6B;oBACzC,UAAU,EAAE;wBACV;4BACE,OAAO,EAAE;gCACP,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,6BAA6B,EAAE,CAAC;gCAChD,IAAI,EAAE,OAAO;6BACd;4BACD,YAAY,EAAE,MAAM;yBACrB;qBACF;oBACD,aAAa,EAAE;wBACb,gBAAgB,EAAE,CAAC;wBACnB,oBAAoB,EAAE,EAAE;wBACxB,eAAe,EAAE,EAAE;qBACpB;iBACF;aACF,CAAC;YAEF,MAAM,iBAAe,CAAC,gBAAgB,CAAC;gBACrC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,6CAA6C;gBAC5D,QAAQ,EAAE,YAAY;gBACtB,IAAI,EAAE,QAAQ;gBACd,YAAY,EAAE,YAAY;aAC3B,CAAC,CAAC;YAEH,yFAAyF;YACzF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,yBAAyB,EAC5C,YAAY,CACb,CAAC;YACF,gBAAgB;YAChB,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,YAAY,CAAC,CAAC;YAC1G,iBAAiB;YACjB,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,YAAY,CAAC,CAAC;YAC3G,cAAc;YACd,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,CAAC,CAAC,CAAC;YACpG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,EAAE,CAAC,CAAC;YACtG,wBAAwB;YACxB,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,cAAc,EAAE,mCAAmC,CAAC,CAAC;YAC3H,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;YACxF,6BAA6B;YAC7B,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,GAAG,CAAC,CAAC;YACvG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,GAAG,CAAC,CAAC;YACtG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,IAAI,CAAC,CAAC;YAClG,YAAY;YACZ,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;YACvG,yBAAyB;YACzB,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;YAC/G,cAAc;YACd,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,MAAM,CAAC,CAAC;QACpG,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,QAAQ,GAAG;gBACf;oBACE,QAAQ,EAAE,OAAO;oBACjB,MAAM,EAAE,EAAE;iBACX;aACF,CAAC;YAEF,MAAM,YAAY,GAAG;gBACnB,QAAQ,EAAE;oBACR,IAAI,EAAE,GAAG,EAAE,CAAC,WAAW;oBACvB,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,YAAY,EAAE,MAAM,EAAE,CAAC;oBAClG,aAAa,EAAE,EAAE,gBAAgB,EAAE,CAAC,EAAE,oBAAoB,EAAE,CAAC,EAAE,eAAe,EAAE,CAAC,EAAE;iBACpF;aACF,CAAC;YAEF,IAAI,CAAC,KAAK,CAAC,iBAAe,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAEnE,MAAM,iBAAe,CAAC,gBAAgB,CAAC;gBACrC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,6CAA6C;gBAC5D,QAAQ,EAAE,YAAY;gBACtB,IAAI,EAAE,QAAQ;gBACd,YAAY,EAAE,kBAAkB;aACjC,CAAC,CAAC;YAEH,MAAM,YAAY,GAAG,QAAQ,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAQ,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YAC9E,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,CAAC;YAClF,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,CAAC;YACjF,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,CAAC;YAC5E,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,CAAC;YAC5E,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,6BAAkB,CAAC,6BAA6B,CAAC,CAAC;QACvF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oEAAoE,EAAE,KAAK,IAAI,EAAE;YACjF,gBAAuB,CAAC,qBAAqB,GAAG,KAAK,CAAC;YAEvD,MAAM,QAAQ,GAAG,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;YACrD,MAAM,YAAY,GAAG;gBACnB,QAAQ,EAAE;oBACR,IAAI,EAAE,GAAG,EAAE,CAAC,KAAK;oBACjB,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,YAAY,EAAE,MAAM,EAAE,CAAC;oBAC5F,aAAa,EAAE,EAAE,gBAAgB,EAAE,CAAC,EAAE,oBAAoB,EAAE,CAAC,EAAE,eAAe,EAAE,CAAC,EAAE;iBACpF;aACF,CAAC;YAEF,IAAI,CAAC,KAAK,CAAC,iBAAe,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAEnE,MAAM,iBAAe,CAAC,gBAAgB,CAAC;gBACrC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,6CAA6C;gBAC5D,QAAQ,EAAE,YAAY;gBACtB,IAAI,EAAE,QAAQ;gBACd,YAAY,EAAE,kBAAkB;aACjC,CAAC,CAAC;YAEH,MAAM,CAAC,iBAAe,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YACpE,MAAM,UAAU,GAAI,iBAAe,CAAC,kBAAgC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACtF,MAAM,CAAC,UAAU,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;YAC7E,MAAM,CAAC,UAAU,CAAC,6BAAkB,CAAC,sBAAsB,CAAC,CAAC,CAAC,aAAa,EAAE,CAAC;QAChF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;YACnF,gBAAuB,CAAC,WAAW,GAAG,EAAE,IAAI,EAAE,EAAE,YAAY,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,eAAe,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC;YAC9G,IAAI,CAAC,KAAK,CAAC,iBAAe,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;YAExE,MAAM,QAAQ,GAAG,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;YACpD,MAAM,YAAY,GAAG;gBACnB,QAAQ,EAAE;oBACR,YAAY,EAAE,YAAY;oBAC1B,IAAI,EAAE,GAAG,EAAE,CAAC,UAAU;oBACtB,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,YAAY,EAAE,MAAM,EAAE,CAAC;oBACjG,aAAa,EAAE,EAAE,gBAAgB,EAAE,EAAE,EAAE,oBAAoB,EAAE,CAAC,EAAE,eAAe,EAAE,EAAE,EAAE;iBACtF;aACF,CAAC;YAEF,MAAM,iBAAe,CAAC,gBAAgB,CAAC;gBACrC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,6CAA6C;gBAC5D,QAAQ,EAAE,YAAY;gBACtB,IAAI,EAAE,QAAQ;gBACd,YAAY,EAAE,YAAY;aAC3B,CAAC,CAAC;YAEH,MAAM,CAAC,sBAAW,CAAC,aAAa,CAAC,CAAC,oBAAoB,CACpD,QAAQ,EACR,MAAM,CAAC,gBAAgB,CAAC;gBACtB,KAAK,EAAE,YAAY;gBACnB,IAAI,EAAE,MAAM;gBACZ,QAAQ,EAAE,YAAY;aACvB,CAAC,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;YAC5E,MAAM,QAAQ,GAAG,CAAC,EAAE,QAAQ,EAAE,sBAAsB,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;YACpE,MAAM,YAAY,GAAG;gBACnB,QAAQ,EAAE;oBACR,IAAI,EAAE,GAAG,EAAE,CAAC,EAAE;oBACd,UAAU,EAAE,CAAC;4BACX,OAAO,EAAE;gCACP,KAAK,EAAE,CAAC;wCACN,YAAY,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,EAAE;qCACjE,CAAC;gCACF,IAAI,EAAE,OAAO;6BACd;4BACD,YAAY,EAAE,MAAM;yBACrB,CAAC;oBACF,aAAa,EAAE,EAAE,gBAAgB,EAAE,CAAC,EAAE,oBAAoB,EAAE,CAAC,EAAE,eAAe,EAAE,EAAE,EAAE;iBACrF;aACF,CAAC;YAEF,IAAI,CAAC,KAAK,CAAC,iBAAe,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAEnE,MAAM,iBAAe,CAAC,gBAAgB,CAAC;gBACrC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,6CAA6C;gBAC5D,QAAQ,EAAE,YAAY;gBACtB,IAAI,EAAE,QAAQ;gBACd,YAAY,EAAE,kBAAkB;aACjC,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,gBAAgB,EACnC,aAAa,CACd,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,gBAAgB,EACnC,oBAAoB,CACrB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/instrumentation/__tests__/gpu-collector.test.d.ts

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

package/dist/instrumentation/__tests__/gpu-collector.test.js

@@ -0,0 +1,247 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const gpu_1 = require("../gpu");
+const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+// Columns match NVIDIA_SMI_FIELDS order:
+// index, uuid, name, util.gpu, util.encoder, util.decoder, temp, fan,
+// mem.total, mem.used, mem.free, power.draw, enforced.power.limit, power.limit
+const NVIDIA_CSV = '0, GPU-11111111-2222-3333-4444-555555555555, NVIDIA A100-SXM4-40GB, 87, 15, 5, 65, 30, 40960, 30000, 10960, 250.50, 400.00, 400.00\n' +
+    '1, GPU-aaaaaaaa-bbbb-cccc-dddd-eeeeeeeeeeee, NVIDIA A100-SXM4-40GB, 12, 0, 0, 40, 25, 40960, 1024, 39936, 60.25, 0, 400.00\n';
+const ROCM_JSON = JSON.stringify({
+    card0: {
+        'GPU ID': '0x740f',
+        'Unique ID': '0x123456789abcdef',
+        'Card series': 'AMD Instinct MI210',
+        'GPU use (%)': '42',
+        'Temperature (Sensor edge) (C)': '55.0',
+        'Fan speed (%)': '35',
+        'VRAM Total Memory (B)': String(64 * 1024 * 1024 * 1024),
+        'VRAM Total Used Memory (B)': String(16 * 1024 * 1024 * 1024),
+        'Average Graphics Package Power (W)': '180.0',
+        'Max Graphics Package Power (W)': '300.0',
+    },
+    system: { 'Driver version': '6.2' },
+});
+function nvidiaRunner() {
+    return async (cmd, args) => {
+        if (cmd !== 'nvidia-smi')
+            throw new Error(`${cmd}: command not found`);
+        if (args[0] === '-L')
+            return 'GPU 0: NVIDIA A100\nGPU 1: NVIDIA A100\n';
+        return NVIDIA_CSV;
+    };
+}
+function amdRunner() {
+    return async (cmd) => {
+        if (cmd !== 'rocm-smi')
+            throw new Error(`${cmd}: command not found`);
+        return ROCM_JSON;
+    };
+}
+const noGpuRunner = async (cmd) => {
+    throw new Error(`${cmd}: command not found`);
+};
+function makeFakeMeter() {
+    const gauges = [];
+    const observations = [];
+    let callback = null;
+    const meter = {
+        createObservableGauge: (name) => {
+            const token = { name };
+            gauges.push({ name, token });
+            return token;
+        },
+        addBatchObservableCallback: (cb) => {
+            callback = cb;
+        },
+    };
+    const runCallback = async () => {
+        if (!callback)
+            throw new Error('no callback registered');
+        await callback({
+            observe: (gauge, value, attributes) => observations.push({ gauge, value, attributes }),
+        });
+    };
+    return { meter: meter, gauges, observations, runCallback };
+}
+describe('GpuMetricsCollector', () => {
+    it('detects nvidia GPUs', async () => {
+        expect(await new gpu_1.GpuMetricsCollector(nvidiaRunner()).detectGpuType()).toBe('nvidia');
+    });
+    it('detects amd GPUs when nvidia-smi is missing', async () => {
+        expect(await new gpu_1.GpuMetricsCollector(amdRunner()).detectGpuType()).toBe('amd');
+    });
+    it('returns null when no supported GPU tooling exists', async () => {
+        expect(await new gpu_1.GpuMetricsCollector(noGpuRunner).detectGpuType()).toBeNull();
+    });
+    it('setup registers all 11 gauges with Python-matching metric names', async () => {
+        const { meter, gauges } = makeFakeMeter();
+        const collector = new gpu_1.GpuMetricsCollector(nvidiaRunner());
+        const gpuType = await collector.setup({
+            meter,
+            environment: 'staging',
+            applicationName: 'my-app',
+        });
+        expect(gpuType).toBe('nvidia');
+        expect(gauges.map((g) => g.name)).toEqual([
+            'gpu.utilization',
+            'gpu.enc.utilization',
+            'gpu.dec.utilization',
+            'gpu.temperature',
+            'gpu.fan_speed',
+            'gpu.memory.available',
+            'gpu.memory.total',
+            'gpu.memory.used',
+            'gpu.memory.free',
+            'gpu.power.draw',
+            'gpu.power.limit',
+        ]);
+    });
+    it('setup returns null and registers nothing without a GPU', async () => {
+        const { meter, gauges } = makeFakeMeter();
+        const gpuType = await new gpu_1.GpuMetricsCollector(noGpuRunner).setup({
+            meter,
+            environment: 'staging',
+            applicationName: 'my-app',
+        });
+        expect(gpuType).toBeNull();
+        expect(gauges).toHaveLength(0);
+    });
+    it('observes nvidia stats per GPU with Python-matching attributes', async () => {
+        const { meter, observations, runCallback } = makeFakeMeter();
+        const collector = new gpu_1.GpuMetricsCollector(nvidiaRunner());
+        await collector.setup({ meter, environment: 'staging', applicationName: 'my-app' });
+        await runCallback();
+        // 2 GPUs x 11 metrics
+        expect(observations).toHaveLength(22);
+        const gpu0 = observations.filter((o) => o.attributes[semantic_convention_1.default.GPU_INDEX] === '0');
+        expect(gpu0).toHaveLength(11);
+        expect(gpu0[0].attributes).toEqual({
+            'telemetry.sdk.name': 'shield360',
+            'service.name': 'my-app',
+            [semantic_convention_1.default.ATTR_DEPLOYMENT_ENVIRONMENT]: 'staging',
+            [semantic_convention_1.default.GPU_INDEX]: '0',
+            [semantic_convention_1.default.GPU_UUID]: 'GPU-11111111-2222-3333-4444-555555555555',
+            [semantic_convention_1.default.GPU_NAME]: 'NVIDIA A100-SXM4-40GB',
+        });
+        const valueOf = (metric, index) => observations.find((o) => o.gauge.name === metric &&
+            o.attributes[semantic_convention_1.default.GPU_INDEX] === index)?.value;
+        expect(valueOf('gpu.utilization', '0')).toBe(87);
+        expect(valueOf('gpu.enc.utilization', '0')).toBe(15);
+        expect(valueOf('gpu.dec.utilization', '0')).toBe(5);
+        expect(valueOf('gpu.temperature', '0')).toBe(65);
+        expect(valueOf('gpu.fan_speed', '0')).toBe(30);
+        expect(valueOf('gpu.memory.total', '0')).toBe(40960);
+        expect(valueOf('gpu.memory.used', '0')).toBe(30000);
+        expect(valueOf('gpu.memory.free', '0')).toBe(10960);
+        expect(valueOf('gpu.memory.available', '0')).toBe(10960);
+        expect(valueOf('gpu.power.draw', '0')).toBe(250.5);
+        expect(valueOf('gpu.power.limit', '0')).toBe(400);
+        expect(valueOf('gpu.utilization', '1')).toBe(12);
+        // GPU 1: enforced.power.limit is 0 / N/A → fall back to power.limit
+        expect(valueOf('gpu.power.limit', '1')).toBe(400);
+        expect(valueOf('gpu.enc.utilization', '1')).toBe(0);
+        expect(valueOf('gpu.dec.utilization', '1')).toBe(0);
+    });
+    it('observes amd stats from rocm-smi JSON', async () => {
+        const { meter, observations, runCallback } = makeFakeMeter();
+        const collector = new gpu_1.GpuMetricsCollector(amdRunner());
+        await collector.setup({ meter, environment: 'default', applicationName: 'default' });
+        await runCallback();
+        expect(observations).toHaveLength(11);
+        const attrs = observations[0].attributes;
+        expect(attrs[semantic_convention_1.default.GPU_INDEX]).toBe('0');
+        expect(attrs[semantic_convention_1.default.GPU_NAME]).toBe('AMD Instinct MI210');
+        expect(attrs[semantic_convention_1.default.GPU_UUID]).toBe('0x123456789abcdef');
+        const valueOf = (metric) => observations.find((o) => o.gauge.name === metric)?.value;
+        expect(valueOf('gpu.utilization')).toBe(42);
+        expect(valueOf('gpu.enc.utilization')).toBe(0);
+        expect(valueOf('gpu.dec.utilization')).toBe(0);
+        expect(valueOf('gpu.temperature')).toBe(55);
+        expect(valueOf('gpu.fan_speed')).toBe(35);
+        expect(valueOf('gpu.memory.total')).toBe(64 * 1024);
+        expect(valueOf('gpu.memory.used')).toBe(16 * 1024);
+        expect(valueOf('gpu.memory.free')).toBe(48 * 1024);
+        // Python AMD memory_available == total (not free)
+        expect(valueOf('gpu.memory.available')).toBe(64 * 1024);
+        expect(valueOf('gpu.power.draw')).toBe(180);
+        expect(valueOf('gpu.power.limit')).toBe(300);
+    });
+    it('swallows collection errors without throwing from the callback', async () => {
+        let detect = true;
+        const flakyRunner = async (cmd, args) => {
+            if (cmd !== 'nvidia-smi')
+                throw new Error('not found');
+            if (args[0] === '-L' && detect)
+                return 'GPU 0: NVIDIA A100';
+            throw new Error('nvidia-smi crashed');
+        };
+        const { meter, observations, runCallback } = makeFakeMeter();
+        const collector = new gpu_1.GpuMetricsCollector(flakyRunner);
+        await collector.setup({ meter, environment: 'default', applicationName: 'default' });
+        detect = false;
+        await expect(runCallback()).resolves.toBeUndefined();
+        expect(observations).toHaveLength(0);
+    });
+    it('swallows invalid AMD JSON without throwing from the callback', async () => {
+        const badJsonRunner = async (cmd, args) => {
+            if (cmd === 'nvidia-smi')
+                throw new Error('not found');
+            if (cmd !== 'rocm-smi')
+                throw new Error('not found');
+            if (args.includes('--showid') && args.length === 2)
+                return '{}';
+            return 'not-json{';
+        };
+        const { meter, observations, runCallback } = makeFakeMeter();
+        const collector = new gpu_1.GpuMetricsCollector(badJsonRunner);
+        await collector.setup({ meter, environment: 'default', applicationName: 'default' });
+        await expect(runCallback()).resolves.toBeUndefined();
+        expect(observations).toHaveLength(0);
+    });
+    it('parses malformed numeric fields as 0', async () => {
+        const badCsvRunner = async (cmd, args) => {
+            if (cmd !== 'nvidia-smi')
+                throw new Error('not found');
+            if (args[0] === '-L')
+                return 'GPU 0: NVIDIA A100';
+            return '0, GPU-uuid, NVIDIA A100, [N/A], [N/A], [N/A], 65, [N/A], 40960, 30000, 10960, [N/A], [N/A], 400.00\n';
+        };
+        const collector = new gpu_1.GpuMetricsCollector(badCsvRunner);
+        const { meter, runCallback, observations } = makeFakeMeter();
+        await collector.setup({ meter, environment: 'default', applicationName: 'default' });
+        await runCallback();
+        const valueOf = (metric) => observations.find((o) => o.gauge.name === metric)?.value;
+        expect(valueOf('gpu.utilization')).toBe(0);
+        expect(valueOf('gpu.enc.utilization')).toBe(0);
+        expect(valueOf('gpu.dec.utilization')).toBe(0);
+        expect(valueOf('gpu.power.draw')).toBe(0);
+        // enforced N/A → fall back to power.limit
+        expect(valueOf('gpu.power.limit')).toBe(400);
+    });
+});
+describe('GpuStats typing', () => {
+    it('exposes all metric fields', () => {
+        const stats = {
+            index: '0',
+            uuid: 'u',
+            name: 'n',
+            utilization: 0,
+            utilization_enc: 0,
+            utilization_dec: 0,
+            temperature: 0,
+            fan_speed: 0,
+            memory_available: 0,
+            memory_total: 0,
+            memory_used: 0,
+            memory_free: 0,
+            power_draw: 0,
+            power_limit: 0,
+        };
+        expect(Object.keys(stats)).toHaveLength(14);
+    });
+});
+//# sourceMappingURL=gpu-collector.test.js.map

package/dist/instrumentation/__tests__/gpu-collector.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gpu-collector.test.js","sourceRoot":"","sources":["../../../src/instrumentation/__tests__/gpu-collector.test.ts"],"names":[],"mappings":";;;;;AAAA,gCAAsE;AACtE,oFAA2D;AAE3D,yCAAyC;AACzC,sEAAsE;AACtE,+EAA+E;AAC/E,MAAM,UAAU,GACd,sIAAsI;IACtI,8HAA8H,CAAC;AAEjI,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;IAC/B,KAAK,EAAE;QACL,QAAQ,EAAE,QAAQ;QAClB,WAAW,EAAE,mBAAmB;QAChC,aAAa,EAAE,oBAAoB;QACnC,aAAa,EAAE,IAAI;QACnB,+BAA+B,EAAE,MAAM;QACvC,eAAe,EAAE,IAAI;QACrB,uBAAuB,EAAE,MAAM,CAAC,EAAE,GAAG,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC;QACxD,4BAA4B,EAAE,MAAM,CAAC,EAAE,GAAG,IAAI,GAAG,IAAI,GAAG,IAAI,CAAC;QAC7D,oCAAoC,EAAE,OAAO;QAC7C,gCAAgC,EAAE,OAAO;KAC1C;IACD,MAAM,EAAE,EAAE,gBAAgB,EAAE,KAAK,EAAE;CACpC,CAAC,CAAC;AAEH,SAAS,YAAY;IACnB,OAAO,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;QACzB,IAAI,GAAG,KAAK,YAAY;YAAE,MAAM,IAAI,KAAK,CAAC,GAAG,GAAG,qBAAqB,CAAC,CAAC;QACvE,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI;YAAE,OAAO,0CAA0C,CAAC;QACxE,OAAO,UAAU,CAAC;IACpB,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,SAAS;IAChB,OAAO,KAAK,EAAE,GAAG,EAAE,EAAE;QACnB,IAAI,GAAG,KAAK,UAAU;YAAE,MAAM,IAAI,KAAK,CAAC,GAAG,GAAG,qBAAqB,CAAC,CAAC;QACrE,OAAO,SAAS,CAAC;IACnB,CAAC,CAAC;AACJ,CAAC;AAED,MAAM,WAAW,GAAkB,KAAK,EAAE,GAAG,EAAE,EAAE;IAC/C,MAAM,IAAI,KAAK,CAAC,GAAG,GAAG,qBAAqB,CAAC,CAAC;AAC/C,CAAC,CAAC;AAQF,SAAS,aAAa;IACpB,MAAM,MAAM,GAA2C,EAAE,CAAC;IAC1D,MAAM,YAAY,GAAsB,EAAE,CAAC;IAC3C,IAAI,QAAQ,GAID,IAAI,CAAC;IAEhB,MAAM,KAAK,GAAG;QACZ,qBAAqB,EAAE,CAAC,IAAY,EAAE,EAAE;YACtC,MAAM,KAAK,GAAG,EAAE,IAAI,EAAE,CAAC;YACvB,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;YAC7B,OAAO,KAAK,CAAC;QACf,CAAC;QACD,0BAA0B,EAAE,CAAC,EAAmB,EAAE,EAAE;YAClD,QAAQ,GAAG,EAAE,CAAC;QAChB,CAAC;KACF,CAAC;IAEF,MAAM,WAAW,GAAG,KAAK,IAAI,EAAE;QAC7B,IAAI,CAAC,QAAQ;YAAE,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;QACzD,MAAM,QAAQ,CAAC;YACb,OAAO,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC;SACvF,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,OAAO,EAAE,KAAK,EAAE,KAAY,EAAE,MAAM,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC;AACpE,CAAC;AAED,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,EAAE,CAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;QACnC,MAAM,CAAC,MAAM,IAAI,yBAAmB,CAAC,YAAY,EAAE,CAAC,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACvF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;QAC3D,MAAM,CAAC,MAAM,IAAI,yBAAmB,CAAC,SAAS,EAAE,CAAC,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACjF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QACjE,MAAM,CAAC,MAAM,IAAI,yBAAmB,CAAC,WAAW,CAAC,CAAC,aAAa,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC;IAChF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;QAC/E,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,aAAa,EAAE,CAAC;QAC1C,MAAM,SAAS,GAAG,IAAI,yBAAmB,CAAC,YAAY,EAAE,CAAC,CAAC;QAC1D,MAAM,OAAO,GAAG,MAAM,SAAS,CAAC,KAAK,CAAC;YACpC,KAAK;YACL,WAAW,EAAE,SAAS;YACtB,eAAe,EAAE,QAAQ;SAC1B,CAAC,CAAC;QAEH,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC/B,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC;YACxC,iBAAiB;YACjB,qBAAqB;YACrB,qBAAqB;YACrB,iBAAiB;YACjB,eAAe;YACf,sBAAsB;YACtB,kBAAkB;YAClB,iBAAiB;YACjB,iBAAiB;YACjB,gBAAgB;YAChB,iBAAiB;SAClB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,aAAa,EAAE,CAAC;QAC1C,MAAM,OAAO,GAAG,MAAM,IAAI,yBAAmB,CAAC,WAAW,CAAC,CAAC,KAAK,CAAC;YAC/D,KAAK;YACL,WAAW,EAAE,SAAS;YACtB,eAAe,EAAE,QAAQ;SAC1B,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC3B,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,MAAM,EAAE,KAAK,EAAE,YAAY,EAAE,WAAW,EAAE,GAAG,aAAa,EAAE,CAAC;QAC7D,MAAM,SAAS,GAAG,IAAI,yBAAmB,CAAC,YAAY,EAAE,CAAC,CAAC;QAC1D,MAAM,SAAS,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,SAAS,EAAE,eAAe,EAAE,QAAQ,EAAE,CAAC,CAAC;QACpF,MAAM,WAAW,EAAE,CAAC;QAEpB,sBAAsB;QACtB,MAAM,CAAC,YAAY,CAAC,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;QAEtC,MAAM,IAAI,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,6BAAkB,CAAC,SAAS,CAAC,KAAK,GAAG,CAAC,CAAC;QAC5F,MAAM,CAAC,IAAI,CAAC,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;QAC9B,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC;YACjC,oBAAoB,EAAE,WAAW;YACjC,cAAc,EAAE,QAAQ;YACxB,CAAC,6BAAkB,CAAC,2BAA2B,CAAC,EAAE,SAAS;YAC3D,CAAC,6BAAkB,CAAC,SAAS,CAAC,EAAE,GAAG;YACnC,CAAC,6BAAkB,CAAC,QAAQ,CAAC,EAAE,0CAA0C;YACzE,CAAC,6BAAkB,CAAC,QAAQ,CAAC,EAAE,uBAAuB;SACvD,CAAC,CAAC;QAEH,MAAM,OAAO,GAAG,CAAC,MAAc,EAAE,KAAa,EAAE,EAAE,CAChD,YAAY,CAAC,IAAI,CACf,CAAC,CAAC,EAAE,EAAE,CACH,CAAC,CAAC,KAA0B,CAAC,IAAI,KAAK,MAAM;YAC7C,CAAC,CAAC,UAAU,CAAC,6BAAkB,CAAC,SAAS,CAAC,KAAK,KAAK,CACvD,EAAE,KAAK,CAAC;QAEX,MAAM,CAAC,OAAO,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjD,MAAM,CAAC,OAAO,CAAC,qBAAqB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACrD,MAAM,CAAC,OAAO,CAAC,qBAAqB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpD,MAAM,CAAC,OAAO,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjD,MAAM,CAAC,OAAO,CAAC,eAAe,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC/C,MAAM,CAAC,OAAO,CAAC,kBAAkB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACrD,MAAM,CAAC,OAAO,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpD,MAAM,CAAC,OAAO,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpD,MAAM,CAAC,OAAO,CAAC,sBAAsB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzD,MAAM,CAAC,OAAO,CAAC,gBAAgB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnD,MAAM,CAAC,OAAO,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAClD,MAAM,CAAC,OAAO,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjD,oEAAoE;QACpE,MAAM,CAAC,OAAO,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAClD,MAAM,CAAC,OAAO,CAAC,qBAAqB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpD,MAAM,CAAC,OAAO,CAAC,qBAAqB,EAAE,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;QACrD,MAAM,EAAE,KAAK,EAAE,YAAY,EAAE,WAAW,EAAE,GAAG,aAAa,EAAE,CAAC;QAC7D,MAAM,SAAS,GAAG,IAAI,yBAAmB,CAAC,SAAS,EAAE,CAAC,CAAC;QACvD,MAAM,SAAS,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,SAAS,EAAE,eAAe,EAAE,SAAS,EAAE,CAAC,CAAC;QACrF,MAAM,WAAW,EAAE,CAAC;QAEpB,MAAM,CAAC,YAAY,CAAC,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;QACtC,MAAM,KAAK,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,6BAAkB,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACtD,MAAM,CAAC,KAAK,CAAC,6BAAkB,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QACtE,MAAM,CAAC,KAAK,CAAC,6BAAkB,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAErE,MAAM,OAAO,GAAG,CAAC,MAAc,EAAE,EAAE,CACjC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAE,CAAC,CAAC,KAA0B,CAAC,IAAI,KAAK,MAAM,CAAC,EAAE,KAAK,CAAC;QACjF,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC5C,MAAM,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC5C,MAAM,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC1C,MAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QACpD,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QACnD,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QACnD,kDAAkD;QAClD,MAAM,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC,CAAC,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;QACxD,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC5C,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;QAC7E,IAAI,MAAM,GAAG,IAAI,CAAC;QAClB,MAAM,WAAW,GAAkB,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;YACrD,IAAI,GAAG,KAAK,YAAY;gBAAE,MAAM,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC;YACvD,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI,IAAI,MAAM;gBAAE,OAAO,oBAAoB,CAAC;YAC5D,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;QACxC,CAAC,CAAC;QACF,MAAM,EAAE,KAAK,EAAE,YAAY,EAAE,WAAW,EAAE,GAAG,aAAa,EAAE,CAAC;QAC7D,MAAM,SAAS,GAAG,IAAI,yBAAmB,CAAC,WAAW,CAAC,CAAC;QACvD,MAAM,SAAS,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,SAAS,EAAE,eAAe,EAAE,SAAS,EAAE,CAAC,CAAC;QACrF,MAAM,GAAG,KAAK,CAAC;QACf,MAAM,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;QACrD,MAAM,CAAC,YAAY,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC5E,MAAM,aAAa,GAAkB,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;YACvD,IAAI,GAAG,KAAK,YAAY;gBAAE,MAAM,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC;YACvD,IAAI,GAAG,KAAK,UAAU;gBAAE,MAAM,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC;YACrD,IAAI,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,IAAI,CAAC;YAChE,OAAO,WAAW,CAAC;QACrB,CAAC,CAAC;QACF,MAAM,EAAE,KAAK,EAAE,YAAY,EAAE,WAAW,EAAE,GAAG,aAAa,EAAE,CAAC;QAC7D,MAAM,SAAS,GAAG,IAAI,yBAAmB,CAAC,aAAa,CAAC,CAAC;QACzD,MAAM,SAAS,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,SAAS,EAAE,eAAe,EAAE,SAAS,EAAE,CAAC,CAAC;QACrF,MAAM,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,QAAQ,CAAC,aAAa,EAAE,CAAC;QACrD,MAAM,CAAC,YAAY,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IACvC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,YAAY,GAAkB,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;YACtD,IAAI,GAAG,KAAK,YAAY;gBAAE,MAAM,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC;YACvD,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI;gBAAE,OAAO,oBAAoB,CAAC;YAClD,OAAO,uGAAuG,CAAC;QACjH,CAAC,CAAC;QACF,MAAM,SAAS,GAAG,IAAI,yBAAmB,CAAC,YAAY,CAAC,CAAC;QACxD,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,aAAa,EAAE,CAAC;QAC7D,MAAM,SAAS,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,SAAS,EAAE,eAAe,EAAE,SAAS,EAAE,CAAC,CAAC;QACrF,MAAM,WAAW,EAAE,CAAC;QAEpB,MAAM,OAAO,GAAG,CAAC,MAAc,EAAE,EAAE,CACjC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAE,CAAC,CAAC,KAA0B,CAAC,IAAI,KAAK,MAAM,CAAC,EAAE,KAAK,CAAC;QAEjF,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC3C,MAAM,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC/C,MAAM,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC1C,0CAA0C;QAC1C,MAAM,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACnC,MAAM,KAAK,GAAa;YACtB,KAAK,EAAE,GAAG;YACV,IAAI,EAAE,GAAG;YACT,IAAI,EAAE,GAAG;YACT,WAAW,EAAE,CAAC;YACd,eAAe,EAAE,CAAC;YAClB,eAAe,EAAE,CAAC;YAClB,WAAW,EAAE,CAAC;YACd,SAAS,EAAE,CAAC;YACZ,gBAAgB,EAAE,CAAC;YACnB,YAAY,EAAE,CAAC;YACf,WAAW,EAAE,CAAC;YACd,WAAW,EAAE,CAAC;YACd,UAAU,EAAE,CAAC;YACb,WAAW,EAAE,CAAC;SACf,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,YAAY,CAAC,EAAE,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,11 @@
+/**
+ * Cross-Language Trace Comparison Tests for the DigitalOcean Gradient Integration
+ *
+ * These verify that the TypeScript Gradient instrumentation emits the same span
+ * attributes / events as the Python SDK reference
+ * (sdk/python/src/shield360/instrumentation/gradient). Gradient is OpenAI-shaped:
+ * responses carry a `model` field, requests support seed / frequency_penalty /
+ * presence_penalty, and streaming usage arrives on the final chunk as `usage`.
+ * The provider name is `digitalocean` and chat is served from inference.do-ai.run.
+ */
+export {};