openlit 1.9.1 → 1.10.0

package/README.md

@@ -24,7 +24,7 @@ This project proudly follows and maintains the [Semantic Conventions](https://gi
 
 ## ⚡ Features
 
-- 🔎 **Auto Instrumentation**: Works with 8 LLM providers with just one line of code.
+- 🔎 **Auto Instrumentation**: Works with 12+ LLM providers, vector databases, and AI frameworks with just one line of code.
 - 🔭 **OpenTelemetry-Native Observability SDKs**: Vendor-neutral SDKs that can send traces and metrics to your existing observability tool like Prometheus and Jaeger.
 - 💲 **Cost Tracking for Custom and Fine-Tuned Models**: Pass custom pricing files for accurate budgeting of custom and fine-tuned models.
 - 🚀 **Support for OpenLIT Features**: Includes support for prompt management and secrets management features available in OpenLIT.
@@ -42,8 +42,24 @@ This project proudly follows and maintains the [Semantic Conventions](https://gi
 | [✅ Google AI Studio](https://docs.openlit.io/latest/integrations/google-ai-studio)         |
 | [✅ Together AI](https://docs.openlit.io/latest/integrations/together)                      |
 | [✅ Ollama](https://docs.openlit.io/latest/integrations/ollama)                             |
+| [✅ AWS Bedrock](https://docs.openlit.io/latest/integrations/bedrock)                        |
+| [✅ Hugging Face](https://docs.openlit.io/latest/integrations/huggingface)                  |
+| [✅ Replicate](https://docs.openlit.io/latest/integrations/replicate)                      |
 | [✅ Azure OpenAI](https://docs.openlit.io/latest/integrations/azure-openai) *(via OpenAI SDK)* |
 
+| Vector Databases                                                                            |
+| ------------------------------------------------------------------------------------------- |
+| [✅ Chroma](https://docs.openlit.io/latest/integrations/chromadb)                           |
+| [✅ Pinecone](https://docs.openlit.io/latest/integrations/pinecone)                         |
+| [✅ Qdrant](https://docs.openlit.io/latest/integrations/qdrant)                             |
+| [✅ Milvus](https://docs.openlit.io/latest/integrations/milvus)                             |
+
+| Frameworks                                                                                  |
+| ------------------------------------------------------------------------------------------- |
+| [✅ LangChain](https://docs.openlit.io/latest/integrations/langchain)                       |
+| [✅ LlamaIndex](https://docs.openlit.io/latest/integrations/llama-index)                    |
+| [✅ Vercel AI SDK](https://docs.openlit.io/latest/integrations/vercel-ai)                   |
+
 ## Supported Destinations
 
 - [✅ OpenTelemetry Collector](https://docs.openlit.io/latest/connections/otelcol)

package/dist/constant.d.ts

@@ -1,7 +1,6 @@
 export declare const SDK_NAME = "openlit";
-export declare const SDK_VERSION = "1.9.1";
+export declare const SDK_VERSION: string;
 export declare const DEFAULT_ENVIRONMENT = "default";
 export declare const DEFAULT_APPLICATION_NAME = "default";
 export declare const INSTRUMENTATION_PREFIX = "@openlit";
-export declare const TELEMETRY_SDK_NAME = "opentelemetry";
 export declare const OPENLIT_URL = "http://127.0.0.1:3000";

package/dist/constant.js

@@ -1,11 +1,14 @@
 "use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.OPENLIT_URL = exports.TELEMETRY_SDK_NAME = exports.INSTRUMENTATION_PREFIX = exports.DEFAULT_APPLICATION_NAME = exports.DEFAULT_ENVIRONMENT = exports.SDK_VERSION = exports.SDK_NAME = void 0;
+exports.OPENLIT_URL = exports.INSTRUMENTATION_PREFIX = exports.DEFAULT_APPLICATION_NAME = exports.DEFAULT_ENVIRONMENT = exports.SDK_VERSION = exports.SDK_NAME = void 0;
+const package_json_1 = __importDefault(require("../package.json"));
 exports.SDK_NAME = 'openlit';
-exports.SDK_VERSION = '1.9.1';
+exports.SDK_VERSION = package_json_1.default.version;
 exports.DEFAULT_ENVIRONMENT = 'default';
 exports.DEFAULT_APPLICATION_NAME = 'default';
 exports.INSTRUMENTATION_PREFIX = '@openlit';
-exports.TELEMETRY_SDK_NAME = 'opentelemetry';
 exports.OPENLIT_URL = 'http://127.0.0.1:3000';
 //# sourceMappingURL=constant.js.map

package/dist/constant.js.map

@@ -1 +1 @@
-{"version":3,"file":"constant.js","sourceRoot":"","sources":["../src/constant.ts"],"names":[],"mappings":";;;AAAa,QAAA,QAAQ,GAAG,SAAS,CAAC;AACrB,QAAA,WAAW,GAAG,OAAO,CAAC;AACtB,QAAA,mBAAmB,GAAG,SAAS,CAAC;AAChC,QAAA,wBAAwB,GAAG,SAAS,CAAC;AACrC,QAAA,sBAAsB,GAAG,UAAU,CAAC;AACpC,QAAA,kBAAkB,GAAG,eAAe,CAAC;AACrC,QAAA,WAAW,GAAG,uBAAuB,CAAC"}
+{"version":3,"file":"constant.js","sourceRoot":"","sources":["../src/constant.ts"],"names":[],"mappings":";;;;;;AAAA,mEAAkC;AAErB,QAAA,QAAQ,GAAG,SAAS,CAAC;AACrB,QAAA,WAAW,GAAG,sBAAG,CAAC,OAAO,CAAC;AAC1B,QAAA,mBAAmB,GAAG,SAAS,CAAC;AAChC,QAAA,wBAAwB,GAAG,SAAS,CAAC;AACrC,QAAA,sBAAsB,GAAG,UAAU,CAAC;AACpC,QAAA,WAAW,GAAG,uBAAuB,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/openlit/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 {};

package/dist/instrumentation/__tests__/chroma-trace-comparison.test.js

@@ -0,0 +1,209 @@
+"use strict";
+/**
+ * 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/openlit/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
+ */
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const wrapper_1 = __importDefault(require("../chroma/wrapper"));
+const config_1 = __importDefault(require("../../config"));
+const helpers_1 = __importDefault(require("../../helpers"));
+const semantic_convention_1 = __importDefault(require("../../semantic-convention"));
+jest.mock('../../config');
+jest.mock('../../helpers');
+describe('ChromaDB Cross-Language Trace Comparison', () => {
+    let mockSpan;
+    let mockTracer;
+    beforeEach(() => {
+        mockSpan = {
+            setAttribute: jest.fn(),
+            addEvent: jest.fn(),
+            end: jest.fn(),
+            setStatus: jest.fn(),
+        };
+        mockTracer = {
+            startSpan: jest.fn().mockReturnValue(mockSpan),
+        };
+        config_1.default.environment = 'openlit-testing';
+        config_1.default.applicationName = 'openlit-test';
+        config_1.default.traceContent = true;
+        helpers_1.default.handleException = jest.fn();
+    });
+    afterEach(() => {
+        jest.clearAllMocks();
+    });
+    // ── Helper: invoke wrapper ────────────────────────────────────────────────
+    async function invokeWrapped(method, dbOperation, collectionName, params, returnValue = {}) {
+        const patchFn = wrapper_1.default._patchCollectionMethod(mockTracer, dbOperation);
+        const originalMethod = jest.fn().mockResolvedValue(returnValue);
+        const fakeCollectionInstance = { name: collectionName };
+        const wrapped = patchFn(originalMethod);
+        await wrapped.call(fakeCollectionInstance, params);
+    }
+    // ── Common DB attributes ──────────────────────────────────────────────────
+    it('should set db.system.name = "chroma" matching Python DB_SYSTEM_CHROMA', async () => {
+        await invokeWrapped('add', semantic_convention_1.default.DB_OPERATION_INSERT, 'test_col', {
+            ids: ['id1'],
+            embeddings: [[0.1, 0.2]],
+        });
+        expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_SYSTEM_NAME, 'chroma');
+    });
+    it('should set server.address = localhost and server.port = 8000 (Python defaults)', async () => {
+        await invokeWrapped('add', semantic_convention_1.default.DB_OPERATION_INSERT, 'test_col', {
+            ids: ['id1'],
+        });
+        expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.SERVER_ADDRESS, 'localhost');
+        expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.SERVER_PORT, 8000);
+    });
+    it('should set db.collection.name from collection instance', async () => {
+        await invokeWrapped('add', semantic_convention_1.default.DB_OPERATION_INSERT, 'my_collection', {
+            ids: ['id1'],
+        });
+        expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_COLLECTION_NAME, 'my_collection');
+    });
+    // ── add → INSERT ──────────────────────────────────────────────────────────
+    describe('add() → INSERT (matches Python DB_OPERATION_INSERT)', () => {
+        it('should set db.operation.name = "INSERT"', async () => {
+            await invokeWrapped('add', semantic_convention_1.default.DB_OPERATION_INSERT, 'col', {
+                ids: ['id1', 'id2'],
+                embeddings: [[0.1], [0.2]],
+                documents: ['doc1', 'doc2'],
+            });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_OPERATION_NAME, 'INSERT');
+        });
+        it('should set db.vector.count = number of ids', async () => {
+            await invokeWrapped('add', semantic_convention_1.default.DB_OPERATION_INSERT, 'col', {
+                ids: ['id1', 'id2', 'id3'],
+            });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_VECTOR_COUNT, 3);
+        });
+        it('should set db.documents_count when documents provided', async () => {
+            await invokeWrapped('add', semantic_convention_1.default.DB_OPERATION_INSERT, 'col', {
+                ids: ['id1'],
+                documents: ['doc1'],
+            });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_DOCUMENTS_COUNT, 1);
+        });
+    });
+    // ── query → QUERY ─────────────────────────────────────────────────────────
+    describe('query() → QUERY (vector similarity search)', () => {
+        it('should set db.operation.name = "QUERY"', async () => {
+            await invokeWrapped('query', semantic_convention_1.default.DB_OPERATION_QUERY, 'col', {
+                queryEmbeddings: [[0.1, 0.2]],
+                nResults: 5,
+            }, { ids: [['id1', 'id2']] });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_OPERATION_NAME, 'QUERY');
+        });
+        it('should set db.vector.query.top_k from nResults', async () => {
+            await invokeWrapped('query', semantic_convention_1.default.DB_OPERATION_QUERY, 'col', {
+                queryEmbeddings: [[0.1]],
+                nResults: 10,
+            }, { ids: [['id1']] });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_VECTOR_QUERY_TOP_K, 10);
+        });
+        it('should set db.filter when where clause provided (matching Python DB_FILTER)', async () => {
+            const where = { source: 'test' };
+            await invokeWrapped('query', semantic_convention_1.default.DB_OPERATION_QUERY, 'col', {
+                queryEmbeddings: [[0.1]],
+                nResults: 5,
+                where,
+            });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_FILTER, JSON.stringify(where));
+        });
+        it('should set db.n_results to number of matched results', async () => {
+            await invokeWrapped('query', semantic_convention_1.default.DB_OPERATION_QUERY, 'col', {
+                queryEmbeddings: [[0.1]],
+                nResults: 5,
+            }, { ids: [['r1', 'r2', 'r3']] });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_N_RESULTS, 3);
+        });
+    });
+    // ── get → GET ────────────────────────────────────────────────────────────
+    describe('get() → GET (retrieve by IDs)', () => {
+        it('should set db.operation.name = "GET"', async () => {
+            await invokeWrapped('get', semantic_convention_1.default.DB_OPERATION_GET, 'col', {
+                ids: ['id1', 'id2'],
+            }, { ids: ['id1', 'id2'] });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_OPERATION_NAME, 'GET');
+        });
+        it('should set db.ids_count from ids array length', async () => {
+            await invokeWrapped('get', semantic_convention_1.default.DB_OPERATION_GET, 'col', {
+                ids: ['id1', 'id2'],
+            });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_ID_COUNT, 2);
+        });
+    });
+    // ── delete → DELETE ───────────────────────────────────────────────────────
+    describe('delete() → DELETE', () => {
+        it('should set db.operation.name = "DELETE"', async () => {
+            await invokeWrapped('delete', semantic_convention_1.default.DB_OPERATION_DELETE, 'col', {
+                ids: ['id1'],
+            });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_OPERATION_NAME, 'DELETE');
+        });
+        it('should set db.filter when where clause provided', async () => {
+            const where = { category: 'old' };
+            await invokeWrapped('delete', semantic_convention_1.default.DB_OPERATION_DELETE, 'col', {
+                where,
+            });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_FILTER, JSON.stringify(where));
+        });
+    });
+    // ── peek → PEEK ───────────────────────────────────────────────────────────
+    describe('peek() → PEEK', () => {
+        it('should set db.operation.name = "PEEK"', async () => {
+            await invokeWrapped('peek', semantic_convention_1.default.DB_OPERATION_PEEK, 'col', 5, {
+                ids: ['id1', 'id2', 'id3', 'id4', 'id5'],
+            });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_OPERATION_NAME, 'PEEK');
+        });
+    });
+    // ── upsert → UPSERT ───────────────────────────────────────────────────────
+    describe('upsert() → UPSERT', () => {
+        it('should set db.operation.name = "UPSERT" and db.vector.count', async () => {
+            await invokeWrapped('upsert', semantic_convention_1.default.DB_OPERATION_UPSERT, 'col', {
+                ids: ['id1', 'id2'],
+                embeddings: [[0.1], [0.2]],
+            });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_OPERATION_NAME, 'UPSERT');
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_VECTOR_COUNT, 2);
+        });
+    });
+    // ── update → UPDATE ───────────────────────────────────────────────────────
+    describe('update() → UPDATE', () => {
+        it('should set db.operation.name = "UPDATE"', async () => {
+            await invokeWrapped('update', semantic_convention_1.default.DB_OPERATION_UPDATE, 'col', {
+                ids: ['id1'],
+                documents: ['new content'],
+            });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.DB_OPERATION_NAME, 'UPDATE');
+        });
+    });
+    // ── Error handling ────────────────────────────────────────────────────────
+    it('should call handleException on error', async () => {
+        const patchFn = wrapper_1.default._patchCollectionMethod(mockTracer, semantic_convention_1.default.DB_OPERATION_QUERY);
+        const error = new Error('Collection not found');
+        const originalMethod = jest.fn().mockRejectedValue(error);
+        const fakeInstance = { name: 'missing_col' };
+        const wrapped = patchFn(originalMethod);
+        await expect(wrapped.call(fakeInstance, {})).rejects.toThrow('Collection not found');
+        expect(helpers_1.default.handleException).toHaveBeenCalledWith(mockSpan, error);
+        expect(mockSpan.end).toHaveBeenCalled();
+    });
+});
+//# sourceMappingURL=chroma-trace-comparison.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"chroma-trace-comparison.test.js","sourceRoot":"","sources":["../../../src/instrumentation/__tests__/chroma-trace-comparison.test.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;;;GAgBG;;;;;AAEH,gEAA8C;AAC9C,0DAAyC;AACzC,4DAA0C;AAC1C,oFAA2D;AAE3D,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;AAC1B,IAAI,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;AAE3B,QAAQ,CAAC,0CAA0C,EAAE,GAAG,EAAE;IACxD,IAAI,QAAa,CAAC;IAClB,IAAI,UAAe,CAAC;IAEpB,UAAU,CAAC,GAAG,EAAE;QACd,QAAQ,GAAG;YACT,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;YACvB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;YACnB,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;YACd,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;SACrB,CAAC;QAEF,UAAU,GAAG;YACX,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,QAAQ,CAAC;SAC/C,CAAC;QAED,gBAAqB,CAAC,WAAW,GAAG,iBAAiB,CAAC;QACtD,gBAAqB,CAAC,eAAe,GAAG,cAAc,CAAC;QACvD,gBAAqB,CAAC,YAAY,GAAG,IAAI,CAAC;QAC1C,iBAAqB,CAAC,eAAe,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,6EAA6E;IAE7E,KAAK,UAAU,aAAa,CAC1B,MAAc,EACd,WAAmB,EACnB,cAAsB,EACtB,MAAW,EACX,cAAmB,EAAE;QAErB,MAAM,OAAO,GAAG,iBAAa,CAAC,sBAAsB,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QAC9E,MAAM,cAAc,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;QAChE,MAAM,sBAAsB,GAAG,EAAE,IAAI,EAAE,cAAc,EAAE,CAAC;QACxD,MAAM,OAAO,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;QACxC,MAAM,OAAO,CAAC,IAAI,CAAC,sBAAsB,EAAE,MAAM,CAAC,CAAC;IACrD,CAAC;IAED,6EAA6E;IAE7E,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,aAAa,CAAC,KAAK,EAAE,6BAAkB,CAAC,mBAAmB,EAAE,UAAU,EAAE;YAC7E,GAAG,EAAE,CAAC,KAAK,CAAC;YACZ,UAAU,EAAE,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;SACzB,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,cAAc,EACjC,QAAQ,CACT,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gFAAgF,EAAE,KAAK,IAAI,EAAE;QAC9F,MAAM,aAAa,CAAC,KAAK,EAAE,6BAAkB,CAAC,mBAAmB,EAAE,UAAU,EAAE;YAC7E,GAAG,EAAE,CAAC,KAAK,CAAC;SACb,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,cAAc,EAAE,WAAW,CAAC,CAAC;QACnG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;IAC3F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;QACtE,MAAM,aAAa,CAAC,KAAK,EAAE,6BAAkB,CAAC,mBAAmB,EAAE,eAAe,EAAE;YAClF,GAAG,EAAE,CAAC,KAAK,CAAC;SACb,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,kBAAkB,EACrC,eAAe,CAChB,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,6EAA6E;IAE7E,QAAQ,CAAC,qDAAqD,EAAE,GAAG,EAAE;QACnE,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,aAAa,CAAC,KAAK,EAAE,6BAAkB,CAAC,mBAAmB,EAAE,KAAK,EAAE;gBACxE,GAAG,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC;gBACnB,UAAU,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC;gBAC1B,SAAS,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC;aAC5B,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,iBAAiB,EACpC,QAAQ,CACT,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4CAA4C,EAAE,KAAK,IAAI,EAAE;YAC1D,MAAM,aAAa,CAAC,KAAK,EAAE,6BAAkB,CAAC,mBAAmB,EAAE,KAAK,EAAE;gBACxE,GAAG,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC;aAC3B,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC;QAC5F,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,aAAa,CAAC,KAAK,EAAE,6BAAkB,CAAC,mBAAmB,EAAE,KAAK,EAAE;gBACxE,GAAG,EAAE,CAAC,KAAK,CAAC;gBACZ,SAAS,EAAE,CAAC,MAAM,CAAC;aACpB,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,CAAC,CAAC,CAAC;QAC/F,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,6EAA6E;IAE7E,QAAQ,CAAC,4CAA4C,EAAE,GAAG,EAAE;QAC1D,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,aAAa,CAAC,OAAO,EAAE,6BAAkB,CAAC,kBAAkB,EAAE,KAAK,EAAE;gBACzE,eAAe,EAAE,CAAC,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;gBAC7B,QAAQ,EAAE,CAAC;aACZ,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;YAE9B,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,iBAAiB,EACpC,OAAO,CACR,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,aAAa,CAAC,OAAO,EAAE,6BAAkB,CAAC,kBAAkB,EAAE,KAAK,EAAE;gBACzE,eAAe,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC;gBACxB,QAAQ,EAAE,EAAE;aACb,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC;YAEvB,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,qBAAqB,EAAE,EAAE,CAAC,CAAC;QACnG,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6EAA6E,EAAE,KAAK,IAAI,EAAE;YAC3F,MAAM,KAAK,GAAG,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;YACjC,MAAM,aAAa,CAAC,OAAO,EAAE,6BAAkB,CAAC,kBAAkB,EAAE,KAAK,EAAE;gBACzE,eAAe,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC;gBACxB,QAAQ,EAAE,CAAC;gBACX,KAAK;aACN,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,SAAS,EAC5B,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CACtB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,aAAa,CAAC,OAAO,EAAE,6BAAkB,CAAC,kBAAkB,EAAE,KAAK,EAAE;gBACzE,eAAe,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC;gBACxB,QAAQ,EAAE,CAAC;aACZ,EAAE,EAAE,GAAG,EAAE,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC;YAElC,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;QACzF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,4EAA4E;IAE5E,QAAQ,CAAC,+BAA+B,EAAE,GAAG,EAAE;QAC7C,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,aAAa,CAAC,KAAK,EAAE,6BAAkB,CAAC,gBAAgB,EAAE,KAAK,EAAE;gBACrE,GAAG,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC;aACpB,EAAE,EAAE,GAAG,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,CAAC,CAAC;YAE5B,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,iBAAiB,EACpC,KAAK,CACN,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,aAAa,CAAC,KAAK,EAAE,6BAAkB,CAAC,gBAAgB,EAAE,KAAK,EAAE;gBACrE,GAAG,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC;aACpB,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC;QACxF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,6EAA6E;IAE7E,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,aAAa,CAAC,QAAQ,EAAE,6BAAkB,CAAC,mBAAmB,EAAE,KAAK,EAAE;gBAC3E,GAAG,EAAE,CAAC,KAAK,CAAC;aACb,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,iBAAiB,EACpC,QAAQ,CACT,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;YAC/D,MAAM,KAAK,GAAG,EAAE,QAAQ,EAAE,KAAK,EAAE,CAAC;YAClC,MAAM,aAAa,CAAC,QAAQ,EAAE,6BAAkB,CAAC,mBAAmB,EAAE,KAAK,EAAE;gBAC3E,KAAK;aACN,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,SAAS,EAC5B,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CACtB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,6EAA6E;IAE7E,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,aAAa,CAAC,MAAM,EAAE,6BAAkB,CAAC,iBAAiB,EAAE,KAAK,EAAE,CAAC,EAAE;gBAC1E,GAAG,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC;aACzC,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,iBAAiB,EACpC,MAAM,CACP,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,6EAA6E;IAE7E,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;YAC3E,MAAM,aAAa,CAAC,QAAQ,EAAE,6BAAkB,CAAC,mBAAmB,EAAE,KAAK,EAAE;gBAC3E,GAAG,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC;gBACnB,UAAU,EAAE,CAAC,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC;aAC3B,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,iBAAiB,EACpC,QAAQ,CACT,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,eAAe,EAAE,CAAC,CAAC,CAAC;QAC5F,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,6EAA6E;IAE7E,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,aAAa,CAAC,QAAQ,EAAE,6BAAkB,CAAC,mBAAmB,EAAE,KAAK,EAAE;gBAC3E,GAAG,EAAE,CAAC,KAAK,CAAC;gBACZ,SAAS,EAAE,CAAC,aAAa,CAAC;aAC3B,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,iBAAiB,EACpC,QAAQ,CACT,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,6EAA6E;IAE7E,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,MAAM,OAAO,GAAG,iBAAa,CAAC,sBAAsB,CAAC,UAAU,EAAE,6BAAkB,CAAC,kBAAkB,CAAC,CAAC;QACxG,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC;QAChD,MAAM,cAAc,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAC1D,MAAM,YAAY,GAAG,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;QAC7C,MAAM,OAAO,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;QAExC,MAAM,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAC;QACrF,MAAM,CAAC,iBAAa,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAC5E,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,gBAAgB,EAAE,CAAC;IAC1C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,7 @@
+/**
+ * Cross-Language Trace Comparison Tests for HuggingFace Inference Integration
+ *
+ * Verifies that the TypeScript SDK generates traces consistent with the Python SDK
+ * for HuggingFace Inference API chatCompletion and textGeneration.
+ */
+export {};

package/dist/instrumentation/__tests__/huggingface-trace-comparison.test.js

@@ -0,0 +1,237 @@
+"use strict";
+/**
+ * Cross-Language Trace Comparison Tests for HuggingFace Inference Integration
+ *
+ * Verifies that the TypeScript SDK generates traces consistent with the Python SDK
+ * for HuggingFace Inference API chatCompletion and textGeneration.
+ */
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const wrapper_1 = __importDefault(require("../huggingface/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');
+describe('HuggingFace Cross-Language Trace Comparison', () => {
+    let mockSpan;
+    let mockTracer;
+    beforeEach(() => {
+        mockSpan = {
+            setAttribute: jest.fn(),
+            addEvent: jest.fn(),
+            end: jest.fn(),
+            setStatus: jest.fn(),
+        };
+        mockTracer = {
+            startSpan: jest.fn().mockReturnValue(mockSpan),
+        };
+        config_1.default.environment = 'openlit-testing';
+        config_1.default.applicationName = 'openlit-test';
+        config_1.default.traceContent = true;
+        config_1.default.pricing_json = {};
+        config_1.default.updatePricingJson = jest.fn().mockResolvedValue({});
+        helpers_1.default.getChatModelCost = jest.fn().mockReturnValue(0.0005);
+        helpers_1.default.generalTokens = jest.fn().mockReturnValue(8);
+        helpers_1.default.buildInputMessages = jest.fn().mockReturnValue('[]');
+        helpers_1.default.buildOutputMessages = jest.fn().mockReturnValue('[]');
+        helpers_1.default.handleException = jest.fn();
+        helpers_1.default.createStreamProxy = jest.fn().mockImplementation((stream, _gen) => stream);
+        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, 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();
+    });
+    // ── Chat Completion ───────────────────────────────────────────────────────
+    describe('Chat Completion', () => {
+        const mockArgs = [
+            {
+                model: 'meta-llama/Meta-Llama-3-8B-Instruct',
+                messages: [{ role: 'user', content: 'What is LLM Observability?' }],
+                max_tokens: 100,
+                temperature: 0.7,
+                stream: false,
+            },
+        ];
+        const mockResponse = {
+            id: 'hf-chat-123',
+            created: Date.now(),
+            model: 'meta-llama/Meta-Llama-3-8B-Instruct',
+            choices: [
+                {
+                    index: 0,
+                    finish_reason: 'stop',
+                    message: { role: 'assistant', content: 'LLM Observability is...' },
+                },
+            ],
+            usage: { prompt_tokens: 12, completion_tokens: 25, total_tokens: 37 },
+        };
+        it('should set gen_ai.system = "huggingface" matching Python GEN_AI_SYSTEM_HUGGING_FACE', async () => {
+            await wrapper_1.default._chatCompletion({
+                args: mockArgs,
+                genAIEndpoint: 'huggingface.chat.completions',
+                response: mockResponse,
+                span: mockSpan,
+            });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_PROVIDER_NAME, 'huggingface');
+        });
+        it('should set gen_ai.operation.name = "chat" matching Python', async () => {
+            await wrapper_1.default._chatCompletion({
+                args: mockArgs,
+                genAIEndpoint: 'huggingface.chat.completions',
+                response: mockResponse,
+                span: mockSpan,
+            });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_OPERATION, semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT);
+        });
+        it('should set token usage attributes matching Python SDK', async () => {
+            await wrapper_1.default._chatCompletion({
+                args: mockArgs,
+                genAIEndpoint: 'huggingface.chat.completions',
+                response: mockResponse,
+                span: mockSpan,
+            });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, 12);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS, 25);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_TOTAL_TOKENS, 37);
+        });
+        it('should set server.address and server.port matching Python', async () => {
+            await wrapper_1.default._chatCompletion({
+                args: mockArgs,
+                genAIEndpoint: 'huggingface.chat.completions',
+                response: mockResponse,
+                span: mockSpan,
+            });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.SERVER_ADDRESS, 'api-inference.huggingface.co');
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.SERVER_PORT, 443);
+        });
+        it('should set request params: temperature, max_tokens, is_stream', async () => {
+            await wrapper_1.default._chatCompletion({
+                args: mockArgs,
+                genAIEndpoint: 'huggingface.chat.completions',
+                response: mockResponse,
+                span: mockSpan,
+            });
+            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_IS_STREAM, false);
+        });
+        it('should set finish_reason and output_type', async () => {
+            await wrapper_1.default._chatCompletion({
+                args: mockArgs,
+                genAIEndpoint: 'huggingface.chat.completions',
+                response: mockResponse,
+                span: mockSpan,
+            });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, ['stop']);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_OUTPUT_TYPE, semantic_convention_1.default.GEN_AI_OUTPUT_TYPE_TEXT);
+        });
+        it('should call recordMetrics after span ends', async () => {
+            await wrapper_1.default._chatCompletion({
+                args: mockArgs,
+                genAIEndpoint: 'huggingface.chat.completions',
+                response: mockResponse,
+                span: mockSpan,
+            });
+            expect(mockSpan.end).toHaveBeenCalled();
+            expect(base_wrapper_1.default.recordMetrics).toHaveBeenCalled();
+        });
+    });
+    // ── Streaming ─────────────────────────────────────────────────────────────
+    describe('Streaming Chat Completion', () => {
+        it('should set is_stream=true and accumulate content across chunks', async () => {
+            const mockArgs = [
+                {
+                    model: 'meta-llama/Meta-Llama-3-8B-Instruct',
+                    messages: [{ role: 'user', content: 'Hello' }],
+                    stream: true,
+                },
+            ];
+            async function* mockStream() {
+                yield {
+                    id: 'hf-stream-1',
+                    created: Date.now(),
+                    model: 'meta-llama/Meta-Llama-3-8B-Instruct',
+                    choices: [{ delta: { content: 'Hello' } }],
+                };
+                yield {
+                    id: 'hf-stream-1',
+                    created: Date.now(),
+                    model: 'meta-llama/Meta-Llama-3-8B-Instruct',
+                    choices: [{ delta: { content: ' world' }, finish_reason: 'stop' }],
+                };
+            }
+            const generator = wrapper_1.default._chatCompletionGenerator({
+                args: mockArgs,
+                genAIEndpoint: 'huggingface.chat.completions',
+                response: mockStream(),
+                span: mockSpan,
+            });
+            for await (const _ of generator) { /* consume */ }
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_IS_STREAM, true);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_PROVIDER_NAME, 'huggingface');
+            expect(mockSpan.end).toHaveBeenCalled();
+        });
+    });
+    // ── Text Generation ───────────────────────────────────────────────────────
+    describe('Text Generation', () => {
+        it('should set text_completion operation and correct attributes', async () => {
+            const mockArgs = [
+                {
+                    model: 'gpt2',
+                    inputs: 'The meaning of life is',
+                    parameters: { max_new_tokens: 50, temperature: 0.9 },
+                },
+            ];
+            const mockResponse = { generated_text: 'The meaning of life is 42.' };
+            await wrapper_1.default._textGeneration({
+                args: mockArgs,
+                genAIEndpoint: 'huggingface.text.generation',
+                response: mockResponse,
+                span: mockSpan,
+            });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_OPERATION, semantic_convention_1.default.GEN_AI_OPERATION_TYPE_TEXT_COMPLETION);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_PROVIDER_NAME, 'huggingface');
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, 50);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_TEMPERATURE, 0.9);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_IS_STREAM, false);
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_OUTPUT_TYPE, semantic_convention_1.default.GEN_AI_OUTPUT_TYPE_TEXT);
+            expect(mockSpan.end).toHaveBeenCalled();
+        });
+        it('should capture prompt and completion content when traceContent=true', async () => {
+            const mockArgs = [
+                {
+                    model: 'gpt2',
+                    inputs: 'The meaning of life is',
+                    parameters: {},
+                },
+            ];
+            const mockResponse = { generated_text: '42.' };
+            await wrapper_1.default._textGeneration({
+                args: mockArgs,
+                genAIEndpoint: 'huggingface.text.generation',
+                response: mockResponse,
+                span: mockSpan,
+            });
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_CONTENT_PROMPT_EVENT, 'The meaning of life is');
+            expect(mockSpan.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_CONTENT_COMPLETION_EVENT, '42.');
+        });
+    });
+});
+//# sourceMappingURL=huggingface-trace-comparison.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"huggingface-trace-comparison.test.js","sourceRoot":"","sources":["../../../src/instrumentation/__tests__/huggingface-trace-comparison.test.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;;;AAEH,qEAAwD;AACxD,0DAAyC;AACzC,4DAA0C;AAC1C,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,QAAQ,CAAC,6CAA6C,EAAE,GAAG,EAAE;IAC3D,IAAI,QAAa,CAAC;IAClB,IAAI,UAAe,CAAC;IAEpB,UAAU,CAAC,GAAG,EAAE;QACd,QAAQ,GAAG;YACT,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;YACvB,QAAQ,EAAE,IAAI,CAAC,EAAE,EAAE;YACnB,GAAG,EAAE,IAAI,CAAC,EAAE,EAAE;YACd,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;SACrB,CAAC;QAEF,UAAU,GAAG;YACX,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,QAAQ,CAAC;SAC/C,CAAC;QAED,gBAAqB,CAAC,WAAW,GAAG,iBAAiB,CAAC;QACtD,gBAAqB,CAAC,eAAe,GAAG,cAAc,CAAC;QACvD,gBAAqB,CAAC,YAAY,GAAG,IAAI,CAAC;QAC1C,gBAAqB,CAAC,YAAY,GAAG,EAAE,CAAC;QACxC,gBAAqB,CAAC,iBAAiB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;QAE1E,iBAAqB,CAAC,gBAAgB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC;QAC3E,iBAAqB,CAAC,aAAa,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QACnE,iBAAqB,CAAC,kBAAkB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAC3E,iBAAqB,CAAC,mBAAmB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC;QAC5E,iBAAqB,CAAC,eAAe,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAClD,iBAAqB,CAAC,iBAAiB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,CAAC,MAAM,EAAE,IAAI,EAAE,EAAE,CAAC,MAAM,CAAC,CAAC;QAEjG,sBAAmB,CAAC,aAAa,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAC9C,sBAAmB,CAAC,qBAAqB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,CAAC,IAAS,EAAE,KAAU,EAAE,EAAE;YAClG,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,oBAAoB,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;YAC3E,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;gBAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,iBAAiB,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;YAClG,IAAI,KAAK,CAAC,aAAa;gBAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,cAAc,EAAE,KAAK,CAAC,aAAa,CAAC,CAAC;YACnG,IAAI,KAAK,CAAC,UAAU,KAAK,SAAS;gBAAE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,WAAW,EAAE,KAAK,CAAC,UAAU,CAAC,CAAC;YACxG,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,6EAA6E;IAE7E,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,MAAM,QAAQ,GAAG;YACf;gBACE,KAAK,EAAE,qCAAqC;gBAC5C,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,4BAA4B,EAAE,CAAC;gBACnE,UAAU,EAAE,GAAG;gBACf,WAAW,EAAE,GAAG;gBAChB,MAAM,EAAE,KAAK;aACd;SACF,CAAC;QAEF,MAAM,YAAY,GAAG;YACnB,EAAE,EAAE,aAAa;YACjB,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE;YACnB,KAAK,EAAE,qCAAqC;YAC5C,OAAO,EAAE;gBACP;oBACE,KAAK,EAAE,CAAC;oBACR,aAAa,EAAE,MAAM;oBACrB,OAAO,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,yBAAyB,EAAE;iBACnE;aACF;YACD,KAAK,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,iBAAiB,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE;SACtE,CAAC;QAEF,EAAE,CAAC,qFAAqF,EAAE,KAAK,IAAI,EAAE;YACnG,MAAM,iBAAkB,CAAC,eAAe,CAAC;gBACvC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,8BAA8B;gBAC7C,QAAQ,EAAE,YAAY;gBACtB,IAAI,EAAE,QAAQ;aACf,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,oBAAoB,EACvC,aAAa,CACd,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,iBAAkB,CAAC,eAAe,CAAC;gBACvC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,8BAA8B;gBAC7C,QAAQ,EAAE,YAAY;gBACtB,IAAI,EAAE,QAAQ;aACf,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,gBAAgB,EACnC,6BAAkB,CAAC,0BAA0B,CAC9C,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,MAAM,iBAAkB,CAAC,eAAe,CAAC;gBACvC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,8BAA8B;gBAC7C,QAAQ,EAAE,YAAY;gBACtB,IAAI,EAAE,QAAQ;aACf,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,EAAE,CAAC,CAAC;YACrG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,EAAE,CAAC,CAAC;YACtG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,EAAE,CAAC,CAAC;QACvG,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,MAAM,iBAAkB,CAAC,eAAe,CAAC;gBACvC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,8BAA8B;gBAC7C,QAAQ,EAAE,YAAY;gBACtB,IAAI,EAAE,QAAQ;aACf,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,cAAc,EACjC,8BAA8B,CAC/B,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC;QAC1F,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,KAAK,IAAI,EAAE;YAC7E,MAAM,iBAAkB,CAAC,eAAe,CAAC;gBACvC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,8BAA8B;gBAC7C,QAAQ,EAAE,YAAY;gBACtB,IAAI,EAAE,QAAQ;aACf,CAAC,CAAC;YAEH,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,wBAAwB,EAAE,KAAK,CAAC,CAAC;QACzG,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,iBAAkB,CAAC,eAAe,CAAC;gBACvC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,8BAA8B;gBAC7C,QAAQ,EAAE,YAAY;gBACtB,IAAI,EAAE,QAAQ;aACf,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,6BAA6B,EAChD,CAAC,MAAM,CAAC,CACT,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,kBAAkB,EACrC,6BAAkB,CAAC,uBAAuB,CAC3C,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,iBAAkB,CAAC,eAAe,CAAC;gBACvC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,8BAA8B;gBAC7C,QAAQ,EAAE,YAAY;gBACtB,IAAI,EAAE,QAAQ;aACf,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,gBAAgB,EAAE,CAAC;YACxC,MAAM,CAAC,sBAAW,CAAC,aAAa,CAAC,CAAC,gBAAgB,EAAE,CAAC;QACvD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,6EAA6E;IAE7E,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;QACzC,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;YAC9E,MAAM,QAAQ,GAAG;gBACf;oBACE,KAAK,EAAE,qCAAqC;oBAC5C,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;oBAC9C,MAAM,EAAE,IAAI;iBACb;aACF,CAAC;YAEF,KAAK,SAAS,CAAC,CAAC,UAAU;gBACxB,MAAM;oBACJ,EAAE,EAAE,aAAa;oBACjB,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE;oBACnB,KAAK,EAAE,qCAAqC;oBAC5C,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC;iBAC3C,CAAC;gBACF,MAAM;oBACJ,EAAE,EAAE,aAAa;oBACjB,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE;oBACnB,KAAK,EAAE,qCAAqC;oBAC5C,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,CAAC;iBACnE,CAAC;YACJ,CAAC;YAED,MAAM,SAAS,GAAG,iBAAkB,CAAC,wBAAwB,CAAC;gBAC5D,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,8BAA8B;gBAC7C,QAAQ,EAAE,UAAU,EAAE;gBACtB,IAAI,EAAE,QAAQ;aACf,CAAC,CAAC;YAEH,IAAI,KAAK,EAAE,MAAM,CAAC,IAAI,SAAS,EAAE,CAAC,CAAC,aAAa,CAAC,CAAC;YAElD,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,IAAI,CAAC,CAAC;YACtG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,oBAAoB,EACvC,aAAa,CACd,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC1C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,6EAA6E;IAE7E,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,6DAA6D,EAAE,KAAK,IAAI,EAAE;YAC3E,MAAM,QAAQ,GAAG;gBACf;oBACE,KAAK,EAAE,MAAM;oBACb,MAAM,EAAE,wBAAwB;oBAChC,UAAU,EAAE,EAAE,cAAc,EAAE,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE;iBACrD;aACF,CAAC;YAEF,MAAM,YAAY,GAAG,EAAE,cAAc,EAAE,4BAA4B,EAAE,CAAC;YAEtE,MAAM,iBAAkB,CAAC,eAAe,CAAC;gBACvC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,6BAA6B;gBAC5C,QAAQ,EAAE,YAAY;gBACtB,IAAI,EAAE,QAAQ;aACf,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,gBAAgB,EACnC,6BAAkB,CAAC,qCAAqC,CACzD,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,oBAAoB,EACvC,aAAa,CACd,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,EAAE,CAAC,CAAC;YACrG,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,wBAAwB,EAAE,KAAK,CAAC,CAAC;YACvG,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,kBAAkB,EACrC,6BAAkB,CAAC,uBAAuB,CAC3C,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;YACnF,MAAM,QAAQ,GAAG;gBACf;oBACE,KAAK,EAAE,MAAM;oBACb,MAAM,EAAE,wBAAwB;oBAChC,UAAU,EAAE,EAAE;iBACf;aACF,CAAC;YACF,MAAM,YAAY,GAAG,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC;YAE/C,MAAM,iBAAkB,CAAC,eAAe,CAAC;gBACvC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,6BAA6B;gBAC5C,QAAQ,EAAE,YAAY;gBACtB,IAAI,EAAE,QAAQ;aACf,CAAC,CAAC;YAEH,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,2BAA2B,EAC9C,wBAAwB,CACzB,CAAC;YACF,MAAM,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAChD,6BAAkB,CAAC,+BAA+B,EAClD,KAAK,CACN,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

@@ -0,0 +1,17 @@
+/**
+ * Cross-Language Trace Comparison Tests for Milvus Integration
+ *
+ * Verifies TypeScript SDK trace attributes match Python SDK.
+ *
+ * Python SDK reference: sdk/python/src/openlit/instrumentation/milvus/milvus.py
+ *
+ * Key alignment:
+ *   - db.system.name = 'milvus'
+ *   - db.operation.name: SEARCH, INSERT, UPSERT, DELETE, QUERY
+ *   - server.address, server.port (default localhost:19530)
+ *   - db.vector.query.top_k for search
+ *   - db.filter for expr/filter params
+ *   - db.vector.count for insert/upsert
+ *   - db.query.summary
+ */
+export {};