@traceai/cohere 0.1.0 → 0.1.1

package/dist/esm/attributes.js

@@ -0,0 +1,240 @@
+import { safelyJSONStringify } from "@traceai/fi-core";
+import { FISpanKind, LLMProvider, MimeType, SemanticConventions as S } from "@traceai/fi-semantic-conventions";
+export const object = (value) => value !== null && typeof value === "object" ? value : {};
+export const list = (value) => Array.isArray(value) ? value : [];
+const json = (value) => safelyJSONStringify(value) ?? "";
+const field = (data, camel, snake) => data[camel] ?? data[snake];
+export const hideInput = (config) => config.hideInputs || config.hideInputMessages || config.hideInputText || config.hideInputImages;
+export const hideOutput = (config) => config.hideOutputs || config.hideOutputMessages || config.hideOutputText;
+/** Connector authentication is never telemetry content, even with input capture
+ * enabled. Project only the telemetry copy; the SDK still needs these tokens.
+ */
+function withoutConnectorCredentials(request) {
+    if (!Array.isArray(request.connectors))
+        return request;
+    return {
+        ...request,
+        connectors: request.connectors.map((connector) => {
+            const { userAccessToken: _camel, user_access_token: _snake, ...content } = object(connector);
+            return content;
+        }),
+    };
+}
+export function inputAttributes(operation, request, config) {
+    const embedding = operation === "embed";
+    const rerank = operation === "rerank";
+    const attrs = {
+        [S.FI_SPAN_KIND]: embedding ? FISpanKind.EMBEDDING : rerank ? FISpanKind.RERANKER : FISpanKind.LLM,
+        [S.LLM_PROVIDER]: LLMProvider.COHERE,
+        [S.GEN_AI_OPERATION_NAME]: embedding ? "embeddings" : rerank ? "rerank" : "chat",
+    };
+    if (typeof request.model === "string") {
+        attrs[embedding ? S.EMBEDDING_MODEL_NAME : S.LLM_MODEL_NAME] = request.model;
+        if (rerank)
+            attrs[S.RERANKER_MODEL_NAME] = request.model;
+    }
+    if (embedding) {
+        if (!hideInput(config)) {
+            attrs[S.INPUT_VALUE] = json(request.texts ?? request.inputs ?? request.images);
+            attrs[S.INPUT_MIME_TYPE] = MimeType.JSON;
+            list(request.texts).forEach((text, index) => {
+                if (typeof text === "string")
+                    attrs[`${S.EMBEDDING_EMBEDDINGS}.${index}.${S.EMBEDDING_TEXT}`] = text;
+            });
+        }
+    }
+    else if (rerank) {
+        attrs["cohere.rerank.documents_count"] = list(request.documents).length;
+        const topN = field(request, "topN", "top_n");
+        if (typeof topN === "number") {
+            attrs[S.RERANKER_TOP_K] = topN;
+            attrs["cohere.rerank.top_n"] = topN;
+        }
+        if (!hideInput(config)) {
+            attrs[S.INPUT_VALUE] = typeof request.query === "string" ? request.query : "";
+            attrs[S.INPUT_MIME_TYPE] = MimeType.TEXT;
+            attrs[S.RERANKER_QUERY] = attrs[S.INPUT_VALUE];
+            attrs["cohere.rerank.query"] = attrs[S.INPUT_VALUE];
+            attrs[S.RERANKER_INPUT_DOCUMENTS] = json(request.documents);
+        }
+    }
+    else {
+        // Only generation settings belong here. Chat history, tools, documents and
+        // arbitrary SDK fields must not bypass the content privacy controls.
+        const params = {};
+        for (const key of ["temperature", "maxTokens", "max_tokens", "p", "k", "seed", "frequencyPenalty", "presencePenalty"]) {
+            if (typeof request[key] === "number")
+                params[key] = request[key];
+        }
+        attrs[S.LLM_INVOCATION_PARAMETERS] = json(params);
+        if (!hideInput(config)) {
+            const messages = Array.isArray(request.messages) ? request.messages : [
+                ...(request.preamble ? [{ role: "system", content: request.preamble }] : []),
+                ...list(field(request, "chatHistory", "chat_history")).map((entry) => {
+                    const item = object(entry);
+                    return { role: item.role === "CHATBOT" ? "assistant" : String(item.role).toLowerCase(), content: item.message };
+                }),
+                { role: "user", content: request.message },
+            ];
+            attrs[S.LLM_INPUT_MESSAGES] = json(messages);
+            attrs[S.INPUT_VALUE] = typeof request.message === "string" ? request.message : json(messages);
+            attrs[S.INPUT_MIME_TYPE] = typeof request.message === "string" ? MimeType.TEXT : MimeType.JSON;
+            if (request.tools)
+                attrs[S.LLM_TOOLS] = json(request.tools);
+        }
+    }
+    if (!hideInput(config))
+        attrs[S.RAW_INPUT] = json(withoutConnectorCredentials(request));
+    return attrs;
+}
+/** Citation provenance is generated output, but source bodies echo inputs. Use an
+ * allowlist so document/tool payloads and future source fields cannot leak.
+ */
+function withoutCitationInputs(value) {
+    if (!Array.isArray(value))
+        return undefined;
+    return value.map(item => {
+        const citation = object(item);
+        const projected = {};
+        for (const key of ["start", "end", "text", "type"]) {
+            if (typeof citation[key] === "number" || typeof citation[key] === "string")
+                projected[key] = citation[key];
+        }
+        if (Array.isArray(citation.sources))
+            projected.sources = citation.sources.map(item => {
+                const source = object(item);
+                return {
+                    ...(typeof source.type === "string" ? { type: source.type } : {}),
+                    ...(typeof source.id === "string" ? { id: source.id } : {}),
+                };
+            });
+        const documentIds = field(citation, "documentIds", "document_ids");
+        if (Array.isArray(documentIds))
+            projected.documentIds = documentIds.filter(id => typeof id === "string");
+        return projected;
+    });
+}
+/** Keep generated fields only when inputs are hidden. Responses can echo full
+ * chat history, embedding texts, reranked documents and citation sources.
+ * Project rather than mutate the SDK response or filter history by role: even
+ * earlier assistant messages are input to the current generation.
+ */
+function withoutInputEchoes(operation, response) {
+    if (operation === "embed")
+        return { embeddings: response.embeddings };
+    if (operation === "rerank")
+        return {
+            results: list(response.results).map((item) => {
+                const result = object(item);
+                return { index: result.index, relevanceScore: field(result, "relevanceScore", "relevance_score") };
+            }),
+        };
+    const message = object(response.message);
+    return {
+        text: response.text,
+        toolCalls: field(response, "toolCalls", "tool_calls"),
+        citations: withoutCitationInputs(response.citations),
+        message: {
+            role: message.role,
+            content: Array.isArray(message.content) ? message.content.map((part) => {
+                const content = object(part);
+                return { type: content.type, text: content.text, thinking: content.thinking };
+            }) : message.content,
+            toolCalls: field(message, "toolCalls", "tool_calls"),
+            toolPlan: field(message, "toolPlan", "tool_plan"),
+            citations: withoutCitationInputs(message.citations),
+        },
+    };
+}
+export function outputAttributes(operation, response, config) {
+    const attrs = usageAttributes(response);
+    const id = response.id ?? field(response, "generationId", "generation_id");
+    if (typeof id === "string")
+        attrs[S.GEN_AI_RESPONSE_ID] = id;
+    if (hideInput(config))
+        response = withoutInputEchoes(operation, response);
+    if (operation === "embed") {
+        // raw.output would otherwise bypass hideEmbeddingVectors.
+        if (!hideOutput(config) && !config.hideEmbeddingVectors) {
+            const vectors = Array.isArray(response.embeddings) ? response.embeddings : object(response.embeddings).float;
+            list(vectors).forEach((vector, index) => {
+                if (Array.isArray(vector) && vector.every((n) => typeof n === "number")) {
+                    attrs[`${S.EMBEDDING_EMBEDDINGS}.${index}.${S.EMBEDDING_VECTOR}`] = vector;
+                }
+            });
+            attrs[S.RAW_OUTPUT] = json(response);
+        }
+    }
+    else if (operation === "rerank") {
+        const results = list(response.results);
+        attrs["cohere.rerank.results_count"] = results.length;
+        results.forEach((item, index) => {
+            const result = object(item);
+            const score = field(result, "relevanceScore", "relevance_score");
+            if (typeof result.index === "number")
+                attrs[`cohere.rerank.results.${index}.index`] = result.index;
+            if (typeof score === "number")
+                attrs[`cohere.rerank.results.${index}.relevance_score`] = score;
+            const text = object(result.document).text;
+            if (!hideOutput(config) && typeof text === "string")
+                attrs[`cohere.rerank.results.${index}.document`] = text;
+        });
+        if (!hideOutput(config)) {
+            attrs[S.RERANKER_OUTPUT_DOCUMENTS] = json(results);
+            attrs[S.RAW_OUTPUT] = json(response);
+        }
+    }
+    else if (!hideOutput(config)) {
+        const message = object(response.message);
+        const text = typeof response.text === "string" ? response.text : list(message.content).map((part) => object(part).text ?? "").join("");
+        const calls = field(message, "toolCalls", "tool_calls") ?? field(response, "toolCalls", "tool_calls");
+        const normalized = { role: "assistant", content: message.content ?? text };
+        const citations = message.citations ?? response.citations;
+        if (Array.isArray(citations))
+            normalized.citations = citations;
+        const toolPlan = field(message, "toolPlan", "tool_plan");
+        if (typeof toolPlan === "string")
+            normalized.tool_plan = toolPlan;
+        if (calls)
+            normalized.tool_calls = list(calls).map((item) => {
+                const call = object(item);
+                return call.function ? call : { function: { name: call.name, arguments: typeof call.parameters === "string" ? call.parameters : json(call.parameters) } };
+            });
+        attrs[S.OUTPUT_VALUE] = text;
+        attrs[S.OUTPUT_MIME_TYPE] = MimeType.TEXT;
+        attrs[S.LLM_OUTPUT_MESSAGES] = json([normalized]);
+        attrs[S.RAW_OUTPUT] = json(response);
+    }
+    return attrs;
+}
+function usageAttributes(response) {
+    const usage = object(response.usage ?? response.meta);
+    // Keep the legacy billed-token fallback only when actual usage is absent;
+    // discounted billed units must never replace or be added to actual tokens.
+    const billed = object(field(usage, "billedUnits", "billed_units"));
+    const tokens = object(usage.tokens ?? billed);
+    const input = field(tokens, "inputTokens", "input_tokens");
+    const output = field(tokens, "outputTokens", "output_tokens");
+    const attrs = {};
+    const numeric = (value) => typeof value === "number" && Number.isFinite(value) && value >= 0;
+    if (numeric(input))
+        attrs[S.LLM_TOKEN_COUNT_PROMPT] = input;
+    if (numeric(output))
+        attrs[S.LLM_TOKEN_COUNT_COMPLETION] = output;
+    if (numeric(input) || numeric(output))
+        attrs[S.LLM_TOKEN_COUNT_TOTAL] = (numeric(input) ? input : 0) + (numeric(output) ? output : 0);
+    const cached = field(usage, "cachedTokens", "cached_tokens");
+    if (numeric(cached))
+        attrs[S.LLM_TOKEN_COUNT_PROMPT_DETAILS_CACHE_READ] = cached;
+    for (const [camel, snake] of [
+        ["inputTokens", "input_tokens"], ["outputTokens", "output_tokens"],
+        ["searchUnits", "search_units"], ["classifications", "classifications"],
+        ["images", "images"], ["imageTokens", "image_tokens"],
+    ]) {
+        const value = field(billed, camel, snake);
+        if (numeric(value))
+            attrs[`cohere.usage.billed_units.${snake}`] = value;
+    }
+    return attrs;
+}
+//# sourceMappingURL=attributes.js.map

package/dist/esm/attributes.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"attributes.js","sourceRoot":"","sources":["../../src/attributes.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAe,MAAM,kBAAkB,CAAC;AACpE,OAAO,EAAE,UAAU,EAAE,WAAW,EAAE,QAAQ,EAAE,mBAAmB,IAAI,CAAC,EAAE,MAAM,kCAAkC,CAAC;AAI/G,MAAM,CAAC,MAAM,MAAM,GAAG,CAAC,KAAc,EAAQ,EAAE,CAAC,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAa,CAAC,CAAC,CAAC,EAAE,CAAC;AACjH,MAAM,CAAC,MAAM,IAAI,GAAG,CAAC,KAAc,EAAa,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;AACrF,MAAM,IAAI,GAAG,CAAC,KAAc,EAAE,EAAE,CAAC,mBAAmB,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;AAClE,MAAM,KAAK,GAAG,CAAC,IAAU,EAAE,KAAa,EAAE,KAAa,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,IAAI,CAAC,KAAK,CAAC,CAAC;AACvF,MAAM,CAAC,MAAM,SAAS,GAAG,CAAC,MAAmB,EAAE,EAAE,CAAC,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,iBAAiB,IAAI,MAAM,CAAC,aAAa,IAAI,MAAM,CAAC,eAAe,CAAC;AAClJ,MAAM,CAAC,MAAM,UAAU,GAAG,CAAC,MAAmB,EAAE,EAAE,CAAC,MAAM,CAAC,WAAW,IAAI,MAAM,CAAC,kBAAkB,IAAI,MAAM,CAAC,cAAc,CAAC;AAE5H;;GAEG;AACH,SAAS,2BAA2B,CAAC,OAAa;IAChD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC;QAAE,OAAO,OAAO,CAAC;IACvD,OAAO;QACL,GAAG,OAAO;QACV,UAAU,EAAE,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE;YAC/C,MAAM,EAAE,eAAe,EAAE,MAAM,EAAE,iBAAiB,EAAE,MAAM,EAAE,GAAG,OAAO,EAAE,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC;YAC7F,OAAO,OAAO,CAAC;QACjB,CAAC,CAAC;KACH,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,SAAoB,EAAE,OAAa,EAAE,MAAmB;IACtF,MAAM,SAAS,GAAG,SAAS,KAAK,OAAO,CAAC;IACxC,MAAM,MAAM,GAAG,SAAS,KAAK,QAAQ,CAAC;IACtC,MAAM,KAAK,GAAe;QACxB,CAAC,CAAC,CAAC,YAAY,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,GAAG;QAClG,CAAC,CAAC,CAAC,YAAY,CAAC,EAAE,WAAW,CAAC,MAAM;QACpC,CAAC,CAAC,CAAC,qBAAqB,CAAC,EAAE,SAAS,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM;KACjF,CAAC;IACF,IAAI,OAAO,OAAO,CAAC,KAAK,KAAK,QAAQ,EAAE,CAAC;QACtC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC;QAC7E,IAAI,MAAM;YAAE,KAAK,CAAC,CAAC,CAAC,mBAAmB,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC;IAC3D,CAAC;IACD,IAAI,SAAS,EAAE,CAAC;QACd,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;YACvB,KAAK,CAAC,CAAC,CAAC,WAAW,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,CAAC;YAC/E,KAAK,CAAC,CAAC,CAAC,eAAe,CAAC,GAAG,QAAQ,CAAC,IAAI,CAAC;YACzC,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;gBAC1C,IAAI,OAAO,IAAI,KAAK,QAAQ;oBAAE,KAAK,CAAC,GAAG,CAAC,CAAC,oBAAoB,IAAI,KAAK,IAAI,CAAC,CAAC,cAAc,EAAE,CAAC,GAAG,IAAI,CAAC;YACvG,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;SAAM,IAAI,MAAM,EAAE,CAAC;QAClB,KAAK,CAAC,+BAA+B,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,MAAM,CAAC;QACxE,MAAM,IAAI,GAAG,KAAK,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;QAC7C,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC7B,KAAK,CAAC,CAAC,CAAC,cAAc,CAAC,GAAG,IAAI,CAAC;YAC/B,KAAK,CAAC,qBAAqB,CAAC,GAAG,IAAI,CAAC;QACtC,CAAC;QACD,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;YACvB,KAAK,CAAC,CAAC,CAAC,WAAW,CAAC,GAAG,OAAO,OAAO,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;YAC9E,KAAK,CAAC,CAAC,CAAC,eAAe,CAAC,GAAG,QAAQ,CAAC,IAAI,CAAC;YACzC,KAAK,CAAC,CAAC,CAAC,cAAc,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;YAC/C,KAAK,CAAC,qBAAqB,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC;YACpD,KAAK,CAAC,CAAC,CAAC,wBAAwB,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC;SAAM,CAAC;QACN,2EAA2E;QAC3E,qEAAqE;QACrE,MAAM,MAAM,GAAS,EAAE,CAAC;QACxB,KAAK,MAAM,GAAG,IAAI,CAAC,aAAa,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,EAAE,GAAG,EAAE,MAAM,EAAE,kBAAkB,EAAE,iBAAiB,CAAC,EAAE,CAAC;YACtH,IAAI,OAAO,OAAO,CAAC,GAAG,CAAC,KAAK,QAAQ;gBAAE,MAAM,CAAC,GAAG,CAAC,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC;QACnE,CAAC;QACD,KAAK,CAAC,CAAC,CAAC,yBAAyB,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC;QAClD,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;YACvB,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC;gBACpE,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC5E,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,aAAa,EAAE,cAAc,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;oBACnE,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;oBAC3B,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC;gBAClH,CAAC,CAAC;gBACF,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE;aAC3C,CAAC;YACF,KAAK,CAAC,CAAC,CAAC,kBAAkB,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC7C,KAAK,CAAC,CAAC,CAAC,WAAW,CAAC,GAAG,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC9F,KAAK,CAAC,CAAC,CAAC,eAAe,CAAC,GAAG,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC;YAC/F,IAAI,OAAO,CAAC,KAAK;gBAAE,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAC9D,CAAC;IACH,CAAC;IACD,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;QAAE,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC,2BAA2B,CAAC,OAAO,CAAC,CAAC,CAAC;IACxF,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;GAEG;AACH,SAAS,qBAAqB,CAAC,KAAc;IAC3C,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IAC5C,OAAO,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;QACtB,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;QAC9B,MAAM,SAAS,GAAS,EAAE,CAAC;QAC3B,KAAK,MAAM,GAAG,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC;YACnD,IAAI,OAAO,QAAQ,CAAC,GAAG,CAAC,KAAK,QAAQ,IAAI,OAAO,QAAQ,CAAC,GAAG,CAAC,KAAK,QAAQ;gBAAE,SAAS,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC;QAC7G,CAAC;QACD,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC;YAAE,SAAS,CAAC,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;gBACnF,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;gBAC5B,OAAO;oBACL,GAAG,CAAC,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;oBACjE,GAAG,CAAC,OAAO,MAAM,CAAC,EAAE,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBAC5D,CAAC;YACJ,CAAC,CAAC,CAAC;QACH,MAAM,WAAW,GAAG,KAAK,CAAC,QAAQ,EAAE,aAAa,EAAE,cAAc,CAAC,CAAC;QACnE,IAAI,KAAK,CAAC,OAAO,CAAC,WAAW,CAAC;YAAE,SAAS,CAAC,WAAW,GAAG,WAAW,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,KAAK,QAAQ,CAAC,CAAC;QACzG,OAAO,SAAS,CAAC;IACnB,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;;GAIG;AACH,SAAS,kBAAkB,CAAC,SAAoB,EAAE,QAAc;IAC9D,IAAI,SAAS,KAAK,OAAO;QAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,CAAC,UAAU,EAAE,CAAC;IACtE,IAAI,SAAS,KAAK,QAAQ;QAAE,OAAO;YACjC,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;gBAC3C,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;gBAC5B,OAAO,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,cAAc,EAAE,KAAK,CAAC,MAAM,EAAE,gBAAgB,EAAE,iBAAiB,CAAC,EAAE,CAAC;YACrG,CAAC,CAAC;SACH,CAAC;IACF,MAAM,OAAO,GAAG,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IACzC,OAAO;QACL,IAAI,EAAE,QAAQ,CAAC,IAAI;QACnB,SAAS,EAAE,KAAK,CAAC,QAAQ,EAAE,WAAW,EAAE,YAAY,CAAC;QACrD,SAAS,EAAE,qBAAqB,CAAC,QAAQ,CAAC,SAAS,CAAC;QACpD,OAAO,EAAE;YACP,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;gBACrE,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;gBAC7B,OAAO,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,IAAI,EAAE,OAAO,CAAC,IAAI,EAAE,QAAQ,EAAE,OAAO,CAAC,QAAQ,EAAE,CAAC;YAChF,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO;YACpB,SAAS,EAAE,KAAK,CAAC,OAAO,EAAE,WAAW,EAAE,YAAY,CAAC;YACpD,QAAQ,EAAE,KAAK,CAAC,OAAO,EAAE,UAAU,EAAE,WAAW,CAAC;YACjD,SAAS,EAAE,qBAAqB,CAAC,OAAO,CAAC,SAAS,CAAC;SACpD;KACF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,SAAoB,EAAE,QAAc,EAAE,MAAmB;IACxF,MAAM,KAAK,GAAe,eAAe,CAAC,QAAQ,CAAC,CAAC;IACpD,MAAM,EAAE,GAAG,QAAQ,CAAC,EAAE,IAAI,KAAK,CAAC,QAAQ,EAAE,cAAc,EAAE,eAAe,CAAC,CAAC;IAC3E,IAAI,OAAO,EAAE,KAAK,QAAQ;QAAE,KAAK,CAAC,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,CAAC;IAC7D,IAAI,SAAS,CAAC,MAAM,CAAC;QAAE,QAAQ,GAAG,kBAAkB,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAC1E,IAAI,SAAS,KAAK,OAAO,EAAE,CAAC;QAC1B,0DAA0D;QAC1D,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC;YACxD,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC;YAC7G,IAAI,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE;gBACtC,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,EAAE,CAAC;oBACxE,KAAK,CAAC,GAAG,CAAC,CAAC,oBAAoB,IAAI,KAAK,IAAI,CAAC,CAAC,gBAAgB,EAAE,CAAC,GAAG,MAAM,CAAC;gBAC7E,CAAC;YACH,CAAC,CAAC,CAAC;YACH,KAAK,CAAC,CAAC,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvC,CAAC;IACH,CAAC;SAAM,IAAI,SAAS,KAAK,QAAQ,EAAE,CAAC;QAClC,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QACvC,KAAK,CAAC,6BAA6B,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;QACtD,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE;YAC9B,MAAM,MAAM,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;YAC5B,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,EAAE,gBAAgB,EAAE,iBAAiB,CAAC,CAAC;YACjE,IAAI,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ;gBAAE,KAAK,CAAC,yBAAyB,KAAK,QAAQ,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC;YACnG,IAAI,OAAO,KAAK,KAAK,QAAQ;gBAAE,KAAK,CAAC,yBAAyB,KAAK,kBAAkB,CAAC,GAAG,KAAK,CAAC;YAC/F,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC;YAC1C,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,OAAO,IAAI,KAAK,QAAQ;gBAAE,KAAK,CAAC,yBAAyB,KAAK,WAAW,CAAC,GAAG,IAAI,CAAC;QAC/G,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YACxB,KAAK,CAAC,CAAC,CAAC,yBAAyB,CAAC,GAAG,IAAI,CAAC,OAAO,CAAC,CAAC;YACnD,KAAK,CAAC,CAAC,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvC,CAAC;IACH,CAAC;SAAM,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;QAC/B,MAAM,OAAO,GAAG,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QACzC,MAAM,IAAI,GAAG,OAAO,QAAQ,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACvI,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,EAAE,WAAW,EAAE,YAAY,CAAC,IAAI,KAAK,CAAC,QAAQ,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;QACtG,MAAM,UAAU,GAAS,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,IAAI,EAAE,CAAC;QACjF,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,QAAQ,CAAC,SAAS,CAAC;QAC1D,IAAI,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC;YAAE,UAAU,CAAC,SAAS,GAAG,SAAS,CAAC;QAC/D,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,EAAE,UAAU,EAAE,WAAW,CAAC,CAAC;QACzD,IAAI,OAAO,QAAQ,KAAK,QAAQ;YAAE,UAAU,CAAC,SAAS,GAAG,QAAQ,CAAC;QAClE,IAAI,KAAK;YAAE,UAAU,CAAC,UAAU,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;gBAC1D,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;gBAC1B,OAAO,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,SAAS,EAAE,OAAO,IAAI,CAAC,UAAU,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,EAAE,EAAE,CAAC;YAC5J,CAAC,CAAC,CAAC;QACH,KAAK,CAAC,CAAC,CAAC,YAAY,CAAC,GAAG,IAAI,CAAC;QAC7B,KAAK,CAAC,CAAC,CAAC,gBAAgB,CAAC,GAAG,QAAQ,CAAC,IAAI,CAAC;QAC1C,KAAK,CAAC,CAAC,CAAC,mBAAmB,CAAC,GAAG,IAAI,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;QAClD,KAAK,CAAC,CAAC,CAAC,UAAU,CAAC,GAAG,IAAI,CAAC,QAAQ,CAAC,CAAC;IACvC,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,eAAe,CAAC,QAAc;IACrC,MAAM,KAAK,GAAG,MAAM,CAAC,QAAQ,CAAC,KAAK,IAAI,QAAQ,CAAC,IAAI,CAAC,CAAC;IACtD,0EAA0E;IAC1E,2EAA2E;IAC3E,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,EAAE,aAAa,EAAE,cAAc,CAAC,CAAC,CAAC;IACnE,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM,IAAI,MAAM,CAAC,CAAC;IAC9C,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,EAAE,aAAa,EAAE,cAAc,CAAC,CAAC;IAC3D,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,EAAE,cAAc,EAAE,eAAe,CAAC,CAAC;IAC9D,MAAM,KAAK,GAAe,EAAE,CAAC;IAC7B,MAAM,OAAO,GAAG,CAAC,KAAc,EAAmB,EAAE,CAAC,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;IACvH,IAAI,OAAO,CAAC,KAAK,CAAC;QAAE,KAAK,CAAC,CAAC,CAAC,sBAAsB,CAAC,GAAG,KAAK,CAAC;IAC5D,IAAI,OAAO,CAAC,MAAM,CAAC;QAAE,KAAK,CAAC,CAAC,CAAC,0BAA0B,CAAC,GAAG,MAAM,CAAC;IAClE,IAAI,OAAO,CAAC,KAAK,CAAC,IAAI,OAAO,CAAC,MAAM,CAAC;QAAE,KAAK,CAAC,CAAC,CAAC,qBAAqB,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACtI,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,EAAE,cAAc,EAAE,eAAe,CAAC,CAAC;IAC7D,IAAI,OAAO,CAAC,MAAM,CAAC;QAAE,KAAK,CAAC,CAAC,CAAC,yCAAyC,CAAC,GAAG,MAAM,CAAC;IACjF,KAAK,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI;QAC3B,CAAC,aAAa,EAAE,cAAc,CAAC,EAAE,CAAC,cAAc,EAAE,eAAe,CAAC;QAClE,CAAC,aAAa,EAAE,cAAc,CAAC,EAAE,CAAC,iBAAiB,EAAE,iBAAiB,CAAC;QACvE,CAAC,QAAQ,EAAE,QAAQ,CAAC,EAAE,CAAC,aAAa,EAAE,cAAc,CAAC;KACtD,EAAE,CAAC;QACF,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC;QAC1C,IAAI,OAAO,CAAC,KAAK,CAAC;YAAE,KAAK,CAAC,6BAA6B,KAAK,EAAE,CAAC,GAAG,KAAK,CAAC;IAC1E,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC"}

package/dist/esm/instrumentation.d.ts

@@ -1,34 +1,25 @@
 import { InstrumentationBase, InstrumentationConfig, InstrumentationNodeModuleDefinition } from "@opentelemetry/instrumentation";
 import { TraceConfigOptions } from "@traceai/fi-core";
-/**
- * Function to check if instrumentation is enabled / disabled
- */
+import { Data } from "./attributes.js";
+/** True when at least one SDK method is instrumented, across module copies. */
 export declare function isPatched(): boolean;
-/**
- * @param instrumentationConfig The config for the instrumentation
- * @param traceConfig The FI trace configuration for masking sensitive information
- */
+/** Cohere v1/v2 instrumentation. Install before constructing SDK clients. */
 export declare class CohereInstrumentation extends InstrumentationBase {
-    private fiTracer;
     private _traceConfig?;
+    private patches?;
+    private manualModules?;
     readonly instrumentationName = "@traceai/cohere";
-    constructor({ instrumentationConfig, traceConfig, }?: {
+    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
-     */
+    enable(): void;
+    disable(): void;
+    manuallyInstrument(module: Data): void;
     private patch;
-    /**
-     * Un-patches the Cohere module
-     */
     private unpatch;
+    private restorePatches;
+    private wrapMethod;
 }
 //# sourceMappingURL=instrumentation.d.ts.map

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

@@ -1 +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"}
+{"version":3,"file":"instrumentation.d.ts","sourceRoot":"","sources":["../../src/instrumentation.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,qBAAqB,EAAE,mCAAmC,EAAE,MAAM,gCAAgC,CAAC;AAIjI,OAAO,EAAiC,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AAErF,OAAO,EAAE,IAAI,EAAoE,MAAM,cAAc,CAAC;AAUtG,+EAA+E;AAC/E,wBAAgB,SAAS,IAAI,OAAO,CAA2B;AAE/D,6EAA6E;AAC7E,qBAAa,qBAAsB,SAAQ,mBAAmB;IAC5D,OAAO,CAAC,YAAY,CAAC,CAAqB;IAE1C,OAAO,CAAC,OAAO,CAAC,CAAU;IAC1B,OAAO,CAAC,aAAa,CAAC,CAAY;IAClC,SAAgB,mBAAmB,qBAAqB;gBAE5C,EAAE,qBAAqB,EAAE,WAAW,EAAE,GAAE;QAClD,qBAAqB,CAAC,EAAE,qBAAqB,CAAC;QAC9C,WAAW,CAAC,EAAE,kBAAkB,CAAC;KAC7B;IAKN,SAAS,CAAC,IAAI;IAIE,MAAM,IAAI,IAAI;IAKd,OAAO,IAAI,IAAI;IAK/B,kBAAkB,CAAC,MAAM,EAAE,IAAI,GAAG,IAAI;IAKtC,OAAO,CAAC,KAAK;IA+Bb,OAAO,CAAC,OAAO;IAEf,OAAO,CAAC,cAAc;IAYtB,OAAO,CAAC,UAAU;CA8EnB"}