@traceai/cohere 0.1.0

package/dist/esnext/typeUtils.d.ts

@@ -0,0 +1,6 @@
+/**
+ * Utility function that uses the type system to check if a switch statement is exhaustive.
+ */
+export declare function assertUnreachable(_: never): never;
+export declare function isString(value: unknown): value is string;
+//# sourceMappingURL=typeUtils.d.ts.map

package/dist/esnext/typeUtils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"typeUtils.d.ts","sourceRoot":"","sources":["../../src/typeUtils.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,wBAAgB,iBAAiB,CAAC,CAAC,EAAE,KAAK,GAAG,KAAK,CAEjD;AAED,wBAAgB,QAAQ,CAAC,KAAK,EAAE,OAAO,GAAG,KAAK,IAAI,MAAM,CAExD"}

package/dist/esnext/typeUtils.js

@@ -0,0 +1,10 @@
+/**
+ * Utility function that uses the type system to check if a switch statement is exhaustive.
+ */
+export function assertUnreachable(_) {
+    throw new Error("Unreachable");
+}
+export function isString(value) {
+    return typeof value === "string";
+}
+//# sourceMappingURL=typeUtils.js.map

package/dist/esnext/typeUtils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"typeUtils.js","sourceRoot":"","sources":["../../src/typeUtils.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,MAAM,UAAU,iBAAiB,CAAC,CAAQ;IACxC,MAAM,IAAI,KAAK,CAAC,aAAa,CAAC,CAAC;AACjC,CAAC;AAED,MAAM,UAAU,QAAQ,CAAC,KAAc;IACrC,OAAO,OAAO,KAAK,KAAK,QAAQ,CAAC;AACnC,CAAC"}

package/dist/esnext/version.d.ts

@@ -0,0 +1,2 @@
+export declare const VERSION = "0.1.0";
+//# sourceMappingURL=version.d.ts.map

package/dist/esnext/version.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"version.d.ts","sourceRoot":"","sources":["../../src/version.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,OAAO,UAAU,CAAC"}

package/dist/esnext/version.js

@@ -0,0 +1,2 @@
+export const VERSION = '0.1.0';
+//# sourceMappingURL=version.js.map

package/dist/esnext/version.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"version.js","sourceRoot":"","sources":["../../src/version.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,OAAO,GAAG,OAAO,CAAC"}

package/dist/src/__tests__/instrumentation.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=instrumentation.test.d.ts.map

package/dist/src/__tests__/instrumentation.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"instrumentation.test.d.ts","sourceRoot":"","sources":["../../../src/__tests__/instrumentation.test.ts"],"names":[],"mappings":""}

package/dist/src/__tests__/instrumentation.test.js

@@ -0,0 +1,259 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const globals_1 = require("@jest/globals");
+const instrumentation_1 = require("../instrumentation");
+// Mock the dependencies
+globals_1.jest.mock("@traceai/fi-core", () => ({
+    FITracer: globals_1.jest.fn().mockImplementation(() => ({
+        startSpan: globals_1.jest.fn().mockReturnValue({
+            setStatus: globals_1.jest.fn(),
+            setAttribute: globals_1.jest.fn(),
+            recordException: globals_1.jest.fn(),
+            setAttributes: globals_1.jest.fn(),
+            addEvent: globals_1.jest.fn(),
+            end: globals_1.jest.fn(),
+            isRecording: globals_1.jest.fn().mockReturnValue(true),
+            spanContext: globals_1.jest.fn().mockReturnValue({ spanId: "test-span-id" }),
+        }),
+        startActiveSpan: globals_1.jest.fn((name, fn) => {
+            const mockSpan = {
+                setStatus: globals_1.jest.fn(),
+                recordException: globals_1.jest.fn(),
+                setAttributes: globals_1.jest.fn(),
+                addEvent: globals_1.jest.fn(),
+                end: globals_1.jest.fn(),
+                isRecording: globals_1.jest.fn().mockReturnValue(true),
+            };
+            return fn(mockSpan);
+        }),
+    })),
+    safelyJSONStringify: globals_1.jest.fn((obj) => JSON.stringify(obj)),
+    TraceConfigOptions: {},
+}));
+(0, globals_1.describe)("Cohere Instrumentation", () => {
+    let instrumentation;
+    (0, globals_1.beforeEach)(() => {
+        globals_1.jest.clearAllMocks();
+        instrumentation = new instrumentation_1.CohereInstrumentation({
+            instrumentationConfig: {
+                enabled: true,
+            },
+        });
+        // Mock the tracer property
+        const mockTracer = {
+            startSpan: globals_1.jest.fn().mockReturnValue({
+                setStatus: globals_1.jest.fn(),
+                recordException: globals_1.jest.fn(),
+                setAttributes: globals_1.jest.fn(),
+                addEvent: globals_1.jest.fn(),
+                end: globals_1.jest.fn(),
+                isRecording: globals_1.jest.fn().mockReturnValue(true),
+            }),
+            startActiveSpan: globals_1.jest.fn((name, fn) => {
+                const mockSpan = {
+                    setStatus: globals_1.jest.fn(),
+                    recordException: globals_1.jest.fn(),
+                    setAttributes: globals_1.jest.fn(),
+                    addEvent: globals_1.jest.fn(),
+                    end: globals_1.jest.fn(),
+                    isRecording: globals_1.jest.fn().mockReturnValue(true),
+                };
+                return fn(mockSpan);
+            }),
+        };
+        Object.defineProperty(instrumentation, "tracer", {
+            get: () => mockTracer,
+            configurable: true,
+        });
+    });
+    (0, globals_1.afterEach)(() => {
+        instrumentation.disable();
+    });
+    (0, globals_1.describe)("Initialization", () => {
+        (0, globals_1.it)("should create instrumentation instance", () => {
+            (0, globals_1.expect)(instrumentation).toBeInstanceOf(instrumentation_1.CohereInstrumentation);
+        });
+        (0, globals_1.it)("should initialize with default config", () => {
+            const defaultInstrumentation = new instrumentation_1.CohereInstrumentation();
+            (0, globals_1.expect)(defaultInstrumentation).toBeInstanceOf(instrumentation_1.CohereInstrumentation);
+        });
+        (0, globals_1.it)("should initialize with custom instrumentation config", () => {
+            const customInstrumentation = new instrumentation_1.CohereInstrumentation({
+                instrumentationConfig: {
+                    enabled: false,
+                },
+            });
+            (0, globals_1.expect)(customInstrumentation).toBeInstanceOf(instrumentation_1.CohereInstrumentation);
+        });
+        (0, globals_1.it)("should initialize with trace config", () => {
+            const traceConfigInstrumentation = new instrumentation_1.CohereInstrumentation({
+                traceConfig: {
+                    hideInputs: true,
+                    hideOutputs: true,
+                },
+            });
+            (0, globals_1.expect)(traceConfigInstrumentation).toBeInstanceOf(instrumentation_1.CohereInstrumentation);
+        });
+        (0, globals_1.it)("should have correct instrumentation name", () => {
+            (0, globals_1.expect)(instrumentation.instrumentationName).toBe("@traceai/cohere");
+        });
+    });
+    (0, globals_1.describe)("isPatched function", () => {
+        (0, globals_1.it)("should be a function", () => {
+            (0, globals_1.expect)(typeof instrumentation_1.isPatched).toBe("function");
+        });
+        (0, globals_1.it)("should return false initially", () => {
+            (0, globals_1.expect)(typeof (0, instrumentation_1.isPatched)()).toBe("boolean");
+        });
+    });
+    (0, globals_1.describe)("Instrumentation lifecycle", () => {
+        (0, globals_1.it)("should enable without errors", () => {
+            (0, globals_1.expect)(() => instrumentation.enable()).not.toThrow();
+        });
+        (0, globals_1.it)("should disable without errors", () => {
+            instrumentation.enable();
+            (0, globals_1.expect)(() => instrumentation.disable()).not.toThrow();
+        });
+        (0, globals_1.it)("should handle multiple enable calls", () => {
+            (0, globals_1.expect)(() => {
+                instrumentation.enable();
+                instrumentation.enable();
+            }).not.toThrow();
+        });
+        (0, globals_1.it)("should handle multiple disable calls", () => {
+            instrumentation.enable();
+            (0, globals_1.expect)(() => {
+                instrumentation.disable();
+                instrumentation.disable();
+            }).not.toThrow();
+        });
+    });
+    (0, globals_1.describe)("manuallyInstrument", () => {
+        (0, globals_1.it)("should accept a mock module with CohereClient", () => {
+            instrumentation.enable();
+            const mockModule = {
+                CohereClient: {
+                    prototype: {
+                        chat: globals_1.jest.fn(),
+                        embed: globals_1.jest.fn(),
+                        rerank: globals_1.jest.fn(),
+                    },
+                },
+            };
+            (0, globals_1.expect)(() => instrumentation.manuallyInstrument(mockModule)).not.toThrow();
+        });
+        (0, globals_1.it)("should handle module without CohereClient property", () => {
+            instrumentation.enable();
+            const mockModule = {};
+            (0, globals_1.expect)(() => instrumentation.manuallyInstrument(mockModule)).not.toThrow();
+        });
+        (0, globals_1.it)("should handle module with partial structure", () => {
+            instrumentation.enable();
+            const mockModule = {
+                CohereClient: {
+                    prototype: {},
+                },
+            };
+            (0, globals_1.expect)(() => instrumentation.manuallyInstrument(mockModule)).not.toThrow();
+        });
+    });
+});
+(0, globals_1.describe)("Cohere Instrumentation - Semantic Conventions", () => {
+    (0, globals_1.it)("should use correct LLM system identifier", () => {
+        const { LLMSystem, LLMProvider } = require("@traceai/fi-semantic-conventions");
+        (0, globals_1.expect)(LLMSystem.COHERE).toBe("cohere");
+        (0, globals_1.expect)(LLMProvider.COHERE).toBe("cohere");
+    });
+    (0, globals_1.it)("should use correct span kind for LLM and embedding operations", () => {
+        const { FISpanKind } = require("@traceai/fi-semantic-conventions");
+        (0, globals_1.expect)(FISpanKind.LLM).toBe("LLM");
+        (0, globals_1.expect)(FISpanKind.EMBEDDING).toBe("EMBEDDING");
+    });
+});
+(0, globals_1.describe)("Cohere Instrumentation - Configuration Options", () => {
+    (0, globals_1.it)("should accept empty configuration", () => {
+        const inst = new instrumentation_1.CohereInstrumentation({});
+        (0, globals_1.expect)(inst).toBeInstanceOf(instrumentation_1.CohereInstrumentation);
+    });
+    (0, globals_1.it)("should accept only instrumentationConfig", () => {
+        const inst = new instrumentation_1.CohereInstrumentation({
+            instrumentationConfig: { enabled: true },
+        });
+        (0, globals_1.expect)(inst).toBeInstanceOf(instrumentation_1.CohereInstrumentation);
+    });
+    (0, globals_1.it)("should accept only traceConfig", () => {
+        const inst = new instrumentation_1.CohereInstrumentation({
+            traceConfig: { hideInputs: false },
+        });
+        (0, globals_1.expect)(inst).toBeInstanceOf(instrumentation_1.CohereInstrumentation);
+    });
+    (0, globals_1.it)("should accept both configs", () => {
+        const inst = new instrumentation_1.CohereInstrumentation({
+            instrumentationConfig: { enabled: true },
+            traceConfig: { hideInputs: false, hideOutputs: false },
+        });
+        (0, globals_1.expect)(inst).toBeInstanceOf(instrumentation_1.CohereInstrumentation);
+    });
+});
+(0, globals_1.describe)("Cohere Instrumentation - Supported Models", () => {
+    const chatModels = [
+        "command",
+        "command-r",
+        "command-r-08-2024",
+        "command-light",
+        "command-nightly",
+    ];
+    const embedModels = [
+        "embed-english-v3.0",
+        "embed-multilingual-v3.0",
+        "embed-english-light-v3.0",
+        "embed-multilingual-light-v3.0",
+    ];
+    const rerankModels = [
+        "rerank-english-v3.0",
+        "rerank-multilingual-v3.0",
+        "rerank-english-v2.0",
+        "rerank-multilingual-v2.0",
+    ];
+    chatModels.forEach((model) => {
+        (0, globals_1.it)(`should be able to configure instrumentation for chat model: ${model}`, () => {
+            const inst = new instrumentation_1.CohereInstrumentation();
+            (0, globals_1.expect)(inst).toBeInstanceOf(instrumentation_1.CohereInstrumentation);
+        });
+    });
+    embedModels.forEach((model) => {
+        (0, globals_1.it)(`should be able to configure instrumentation for embed model: ${model}`, () => {
+            const inst = new instrumentation_1.CohereInstrumentation();
+            (0, globals_1.expect)(inst).toBeInstanceOf(instrumentation_1.CohereInstrumentation);
+        });
+    });
+    rerankModels.forEach((model) => {
+        (0, globals_1.it)(`should be able to configure instrumentation for rerank model: ${model}`, () => {
+            const inst = new instrumentation_1.CohereInstrumentation();
+            (0, globals_1.expect)(inst).toBeInstanceOf(instrumentation_1.CohereInstrumentation);
+        });
+    });
+});
+(0, globals_1.describe)("Cohere Instrumentation - Version", () => {
+    (0, globals_1.it)("should export version", () => {
+        const { VERSION } = require("../version");
+        (0, globals_1.expect)(typeof VERSION).toBe("string");
+        (0, globals_1.expect)(VERSION).toBe("0.1.0");
+    });
+});
+(0, globals_1.describe)("Cohere Instrumentation - Type Utils", () => {
+    (0, globals_1.it)("should export isString utility", () => {
+        const { isString } = require("../typeUtils");
+        (0, globals_1.expect)(typeof isString).toBe("function");
+        (0, globals_1.expect)(isString("test")).toBe(true);
+        (0, globals_1.expect)(isString(123)).toBe(false);
+        (0, globals_1.expect)(isString(null)).toBe(false);
+        (0, globals_1.expect)(isString(undefined)).toBe(false);
+    });
+    (0, globals_1.it)("should export assertUnreachable utility", () => {
+        const { assertUnreachable } = require("../typeUtils");
+        (0, globals_1.expect)(typeof assertUnreachable).toBe("function");
+        (0, globals_1.expect)(() => assertUnreachable("test")).toThrow("Unreachable");
+    });
+});
+//# sourceMappingURL=instrumentation.test.js.map

package/dist/src/__tests__/instrumentation.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"instrumentation.test.js","sourceRoot":"","sources":["../../../src/__tests__/instrumentation.test.ts"],"names":[],"mappings":";;AAAA,2CAAkF;AAClF,wDAAsE;AAEtE,wBAAwB;AACxB,cAAI,CAAC,IAAI,CAAC,kBAAkB,EAAE,GAAG,EAAE,CAAC,CAAC;IACnC,QAAQ,EAAE,cAAI,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC,CAAC;QAC5C,SAAS,EAAE,cAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;YACnC,SAAS,EAAE,cAAI,CAAC,EAAE,EAAE;YACpB,YAAY,EAAE,cAAI,CAAC,EAAE,EAAE;YACvB,eAAe,EAAE,cAAI,CAAC,EAAE,EAAE;YAC1B,aAAa,EAAE,cAAI,CAAC,EAAE,EAAE;YACxB,QAAQ,EAAE,cAAI,CAAC,EAAE,EAAE;YACnB,GAAG,EAAE,cAAI,CAAC,EAAE,EAAE;YACd,WAAW,EAAE,cAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC;YAC5C,WAAW,EAAE,cAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,EAAE,MAAM,EAAE,cAAc,EAAE,CAAC;SACnE,CAAC;QACF,eAAe,EAAE,cAAI,CAAC,EAAE,CAAC,CAAC,IAAY,EAAE,EAAO,EAAE,EAAE;YACjD,MAAM,QAAQ,GAAG;gBACf,SAAS,EAAE,cAAI,CAAC,EAAE,EAAE;gBACpB,eAAe,EAAE,cAAI,CAAC,EAAE,EAAE;gBAC1B,aAAa,EAAE,cAAI,CAAC,EAAE,EAAE;gBACxB,QAAQ,EAAE,cAAI,CAAC,EAAE,EAAE;gBACnB,GAAG,EAAE,cAAI,CAAC,EAAE,EAAE;gBACd,WAAW,EAAE,cAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC;aAC7C,CAAC;YACF,OAAO,EAAE,CAAC,QAAQ,CAAC,CAAC;QACtB,CAAC,CAAC;KACH,CAAC,CAAC;IACH,mBAAmB,EAAE,cAAI,CAAC,EAAE,CAAC,CAAC,GAAQ,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;IAC/D,kBAAkB,EAAE,EAAE;CACvB,CAAC,CAAC,CAAC;AAEJ,IAAA,kBAAQ,EAAC,wBAAwB,EAAE,GAAG,EAAE;IACtC,IAAI,eAAsC,CAAC;IAE3C,IAAA,oBAAU,EAAC,GAAG,EAAE;QACd,cAAI,CAAC,aAAa,EAAE,CAAC;QAErB,eAAe,GAAG,IAAI,uCAAqB,CAAC;YAC1C,qBAAqB,EAAE;gBACrB,OAAO,EAAE,IAAI;aACd;SACF,CAAC,CAAC;QAEH,2BAA2B;QAC3B,MAAM,UAAU,GAAG;YACjB,SAAS,EAAE,cAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC;gBACnC,SAAS,EAAE,cAAI,CAAC,EAAE,EAAE;gBACpB,eAAe,EAAE,cAAI,CAAC,EAAE,EAAE;gBAC1B,aAAa,EAAE,cAAI,CAAC,EAAE,EAAE;gBACxB,QAAQ,EAAE,cAAI,CAAC,EAAE,EAAE;gBACnB,GAAG,EAAE,cAAI,CAAC,EAAE,EAAE;gBACd,WAAW,EAAE,cAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC;aAC7C,CAAC;YACF,eAAe,EAAE,cAAI,CAAC,EAAE,CAAC,CAAC,IAAY,EAAE,EAAO,EAAE,EAAE;gBACjD,MAAM,QAAQ,GAAG;oBACf,SAAS,EAAE,cAAI,CAAC,EAAE,EAAE;oBACpB,eAAe,EAAE,cAAI,CAAC,EAAE,EAAE;oBAC1B,aAAa,EAAE,cAAI,CAAC,EAAE,EAAE;oBACxB,QAAQ,EAAE,cAAI,CAAC,EAAE,EAAE;oBACnB,GAAG,EAAE,cAAI,CAAC,EAAE,EAAE;oBACd,WAAW,EAAE,cAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,IAAI,CAAC;iBAC7C,CAAC;gBACF,OAAO,EAAE,CAAC,QAAQ,CAAC,CAAC;YACtB,CAAC,CAAC;SACH,CAAC;QAEF,MAAM,CAAC,cAAc,CAAC,eAAe,EAAE,QAAQ,EAAE;YAC/C,GAAG,EAAE,GAAG,EAAE,CAAC,UAAU;YACrB,YAAY,EAAE,IAAI;SACnB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAA,mBAAS,EAAC,GAAG,EAAE;QACb,eAAe,CAAC,OAAO,EAAE,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,IAAA,kBAAQ,EAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,IAAA,YAAE,EAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,IAAA,gBAAM,EAAC,eAAe,CAAC,CAAC,cAAc,CAAC,uCAAqB,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,IAAA,YAAE,EAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,sBAAsB,GAAG,IAAI,uCAAqB,EAAE,CAAC;YAC3D,IAAA,gBAAM,EAAC,sBAAsB,CAAC,CAAC,cAAc,CAAC,uCAAqB,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC;QAEH,IAAA,YAAE,EAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,qBAAqB,GAAG,IAAI,uCAAqB,CAAC;gBACtD,qBAAqB,EAAE;oBACrB,OAAO,EAAE,KAAK;iBACf;aACF,CAAC,CAAC;YACH,IAAA,gBAAM,EAAC,qBAAqB,CAAC,CAAC,cAAc,CAAC,uCAAqB,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;QAEH,IAAA,YAAE,EAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,0BAA0B,GAAG,IAAI,uCAAqB,CAAC;gBAC3D,WAAW,EAAE;oBACX,UAAU,EAAE,IAAI;oBAChB,WAAW,EAAE,IAAI;iBAClB;aACF,CAAC,CAAC;YACH,IAAA,gBAAM,EAAC,0BAA0B,CAAC,CAAC,cAAc,CAAC,uCAAqB,CAAC,CAAC;QAC3E,CAAC,CAAC,CAAC;QAEH,IAAA,YAAE,EAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,IAAA,gBAAM,EAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAA,kBAAQ,EAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,IAAA,YAAE,EAAC,sBAAsB,EAAE,GAAG,EAAE;YAC9B,IAAA,gBAAM,EAAC,OAAO,2BAAS,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;QAEH,IAAA,YAAE,EAAC,+BAA+B,EAAE,GAAG,EAAE;YACvC,IAAA,gBAAM,EAAC,OAAO,IAAA,2BAAS,GAAE,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAC7C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAA,kBAAQ,EAAC,2BAA2B,EAAE,GAAG,EAAE;QACzC,IAAA,YAAE,EAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,IAAA,gBAAM,EAAC,GAAG,EAAE,CAAC,eAAe,CAAC,MAAM,EAAE,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,IAAA,YAAE,EAAC,+BAA+B,EAAE,GAAG,EAAE;YACvC,eAAe,CAAC,MAAM,EAAE,CAAC;YACzB,IAAA,gBAAM,EAAC,GAAG,EAAE,CAAC,eAAe,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QACxD,CAAC,CAAC,CAAC;QAEH,IAAA,YAAE,EAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,IAAA,gBAAM,EAAC,GAAG,EAAE;gBACV,eAAe,CAAC,MAAM,EAAE,CAAC;gBACzB,eAAe,CAAC,MAAM,EAAE,CAAC;YAC3B,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC,CAAC,CAAC;QAEH,IAAA,YAAE,EAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,eAAe,CAAC,MAAM,EAAE,CAAC;YACzB,IAAA,gBAAM,EAAC,GAAG,EAAE;gBACV,eAAe,CAAC,OAAO,EAAE,CAAC;gBAC1B,eAAe,CAAC,OAAO,EAAE,CAAC;YAC5B,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QACnB,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,IAAA,kBAAQ,EAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,IAAA,YAAE,EAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,eAAe,CAAC,MAAM,EAAE,CAAC;YAEzB,MAAM,UAAU,GAAG;gBACjB,YAAY,EAAE;oBACZ,SAAS,EAAE;wBACT,IAAI,EAAE,cAAI,CAAC,EAAE,EAAE;wBACf,KAAK,EAAE,cAAI,CAAC,EAAE,EAAE;wBAChB,MAAM,EAAE,cAAI,CAAC,EAAE,EAAE;qBAClB;iBACF;aACF,CAAC;YAEF,IAAA,gBAAM,EAAC,GAAG,EAAE,CAAC,eAAe,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QAC7E,CAAC,CAAC,CAAC;QAEH,IAAA,YAAE,EAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,eAAe,CAAC,MAAM,EAAE,CAAC;YAEzB,MAAM,UAAU,GAAG,EAAE,CAAC;YAEtB,IAAA,gBAAM,EAAC,GAAG,EAAE,CAAC,eAAe,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QAC7E,CAAC,CAAC,CAAC;QAEH,IAAA,YAAE,EAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,eAAe,CAAC,MAAM,EAAE,CAAC;YAEzB,MAAM,UAAU,GAAG;gBACjB,YAAY,EAAE;oBACZ,SAAS,EAAE,EAAE;iBACd;aACF,CAAC;YAEF,IAAA,gBAAM,EAAC,GAAG,EAAE,CAAC,eAAe,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QAC7E,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAA,kBAAQ,EAAC,+CAA+C,EAAE,GAAG,EAAE;IAC7D,IAAA,YAAE,EAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,kCAAkC,CAAC,CAAC;QAC/E,IAAA,gBAAM,EAAC,SAAS,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACxC,IAAA,gBAAM,EAAC,WAAW,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC5C,CAAC,CAAC,CAAC;IAEH,IAAA,YAAE,EAAC,+DAA+D,EAAE,GAAG,EAAE;QACvE,MAAM,EAAE,UAAU,EAAE,GAAG,OAAO,CAAC,kCAAkC,CAAC,CAAC;QACnE,IAAA,gBAAM,EAAC,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,IAAA,gBAAM,EAAC,UAAU,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAA,kBAAQ,EAAC,gDAAgD,EAAE,GAAG,EAAE;IAC9D,IAAA,YAAE,EAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,IAAI,GAAG,IAAI,uCAAqB,CAAC,EAAE,CAAC,CAAC;QAC3C,IAAA,gBAAM,EAAC,IAAI,CAAC,CAAC,cAAc,CAAC,uCAAqB,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,IAAA,YAAE,EAAC,0CAA0C,EAAE,GAAG,EAAE;QAClD,MAAM,IAAI,GAAG,IAAI,uCAAqB,CAAC;YACrC,qBAAqB,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;SACzC,CAAC,CAAC;QACH,IAAA,gBAAM,EAAC,IAAI,CAAC,CAAC,cAAc,CAAC,uCAAqB,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,IAAA,YAAE,EAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,IAAI,GAAG,IAAI,uCAAqB,CAAC;YACrC,WAAW,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE;SACnC,CAAC,CAAC;QACH,IAAA,gBAAM,EAAC,IAAI,CAAC,CAAC,cAAc,CAAC,uCAAqB,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,IAAA,YAAE,EAAC,4BAA4B,EAAE,GAAG,EAAE;QACpC,MAAM,IAAI,GAAG,IAAI,uCAAqB,CAAC;YACrC,qBAAqB,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;YACxC,WAAW,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,WAAW,EAAE,KAAK,EAAE;SACvD,CAAC,CAAC;QACH,IAAA,gBAAM,EAAC,IAAI,CAAC,CAAC,cAAc,CAAC,uCAAqB,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAA,kBAAQ,EAAC,2CAA2C,EAAE,GAAG,EAAE;IACzD,MAAM,UAAU,GAAG;QACjB,SAAS;QACT,WAAW;QACX,mBAAmB;QACnB,eAAe;QACf,iBAAiB;KAClB,CAAC;IAEF,MAAM,WAAW,GAAG;QAClB,oBAAoB;QACpB,yBAAyB;QACzB,0BAA0B;QAC1B,+BAA+B;KAChC,CAAC;IAEF,MAAM,YAAY,GAAG;QACnB,qBAAqB;QACrB,0BAA0B;QAC1B,qBAAqB;QACrB,0BAA0B;KAC3B,CAAC;IAEF,UAAU,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;QAC3B,IAAA,YAAE,EAAC,+DAA+D,KAAK,EAAE,EAAE,GAAG,EAAE;YAC9E,MAAM,IAAI,GAAG,IAAI,uCAAqB,EAAE,CAAC;YACzC,IAAA,gBAAM,EAAC,IAAI,CAAC,CAAC,cAAc,CAAC,uCAAqB,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,WAAW,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;QAC5B,IAAA,YAAE,EAAC,gEAAgE,KAAK,EAAE,EAAE,GAAG,EAAE;YAC/E,MAAM,IAAI,GAAG,IAAI,uCAAqB,EAAE,CAAC;YACzC,IAAA,gBAAM,EAAC,IAAI,CAAC,CAAC,cAAc,CAAC,uCAAqB,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,YAAY,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;QAC7B,IAAA,YAAE,EAAC,iEAAiE,KAAK,EAAE,EAAE,GAAG,EAAE;YAChF,MAAM,IAAI,GAAG,IAAI,uCAAqB,EAAE,CAAC;YACzC,IAAA,gBAAM,EAAC,IAAI,CAAC,CAAC,cAAc,CAAC,uCAAqB,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAA,kBAAQ,EAAC,kCAAkC,EAAE,GAAG,EAAE;IAChD,IAAA,YAAE,EAAC,uBAAuB,EAAE,GAAG,EAAE;QAC/B,MAAM,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;QAC1C,IAAA,gBAAM,EAAC,OAAO,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACtC,IAAA,gBAAM,EAAC,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAChC,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,IAAA,kBAAQ,EAAC,qCAAqC,EAAE,GAAG,EAAE;IACnD,IAAA,YAAE,EAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;QAC7C,IAAA,gBAAM,EAAC,OAAO,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACzC,IAAA,gBAAM,EAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACpC,IAAA,gBAAM,EAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAClC,IAAA,gBAAM,EAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,IAAA,gBAAM,EAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,IAAA,YAAE,EAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,EAAE,iBAAiB,EAAE,GAAG,OAAO,CAAC,cAAc,CAAC,CAAC;QACtD,IAAA,gBAAM,EAAC,OAAO,iBAAiB,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QAClD,IAAA,gBAAM,EAAC,GAAG,EAAE,CAAC,iBAAiB,CAAC,MAAe,CAAC,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;IAC1E,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/src/index.d.ts

@@ -0,0 +1,3 @@
+export * from './instrumentation';
+export * from './version';
+//# sourceMappingURL=index.d.ts.map

package/dist/src/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,mBAAmB,CAAC;AAClC,cAAc,WAAW,CAAC"}

package/dist/src/index.js

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

package/dist/src/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,oDAAkC;AAClC,4CAA0B"}

package/dist/src/instrumentation.d.ts

@@ -0,0 +1,34 @@
+import { InstrumentationBase, InstrumentationConfig, InstrumentationNodeModuleDefinition } from "@opentelemetry/instrumentation";
+import { TraceConfigOptions } from "@traceai/fi-core";
+/**
+ * Function to check if instrumentation is enabled / disabled
+ */
+export declare function isPatched(): boolean;
+/**
+ * @param instrumentationConfig The config for the instrumentation
+ * @param traceConfig The FI trace configuration for masking sensitive information
+ */
+export declare class CohereInstrumentation extends InstrumentationBase {
+    private fiTracer;
+    private _traceConfig?;
+    readonly instrumentationName = "@traceai/cohere";
+    constructor({ instrumentationConfig, traceConfig, }?: {
+        instrumentationConfig?: InstrumentationConfig;
+        traceConfig?: TraceConfigOptions;
+    });
+    enable(): void;
+    protected init(): InstrumentationNodeModuleDefinition;
+    /**
+     * Manually instruments the Cohere module
+     */
+    manuallyInstrument(module: Record<string, unknown>): void;
+    /**
+     * Patches the Cohere module
+     */
+    private patch;
+    /**
+     * Un-patches the Cohere module
+     */
+    private unpatch;
+}
+//# sourceMappingURL=instrumentation.d.ts.map

package/dist/src/instrumentation.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"instrumentation.d.ts","sourceRoot":"","sources":["../../src/instrumentation.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,mBAAmB,EACnB,qBAAqB,EACrB,mCAAmC,EAEpC,MAAM,gCAAgC,CAAC;AAmBxC,OAAO,EAGL,kBAAkB,EACnB,MAAM,kBAAkB,CAAC;AAU1B;;GAEG;AACH,wBAAgB,SAAS,YAExB;AAyGD;;;GAGG;AACH,qBAAa,qBAAsB,SAAQ,mBAAmB;IAC5D,OAAO,CAAC,QAAQ,CAAY;IAC5B,OAAO,CAAC,YAAY,CAAC,CAAqB;IAC1C,SAAgB,mBAAmB,qBAAqB;gBAE5C,EACV,qBAAqB,EACrB,WAAW,GACZ,GAAE;QACD,qBAAqB,CAAC,EAAE,qBAAqB,CAAC;QAC9C,WAAW,CAAC,EAAE,kBAAkB,CAAC;KAC7B;IASU,MAAM,IAAI,IAAI;IAK9B,SAAS,CAAC,IAAI;IAUd;;OAEG;IACH,kBAAkB,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAIlD;;OAEG;IACH,OAAO,CAAC,KAAK;IAoPb;;OAEG;IACH,OAAO,CAAC,OAAO;CA2BhB"}