@traceai/cohere 0.1.0 → 0.1.1
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +78 -369
- package/dist/esm/attributes.d.ts +11 -0
- package/dist/esm/attributes.d.ts.map +1 -0
- package/dist/esm/attributes.js +240 -0
- package/dist/esm/attributes.js.map +1 -0
- package/dist/esm/instrumentation.d.ts +11 -20
- package/dist/esm/instrumentation.d.ts.map +1 -1
- package/dist/esm/instrumentation.js +175 -344
- package/dist/esm/instrumentation.js.map +1 -1
- package/dist/esm/stream.d.ts +6 -0
- package/dist/esm/stream.d.ts.map +1 -0
- package/dist/esm/stream.js +215 -0
- package/dist/esm/stream.js.map +1 -0
- package/dist/esm/tsconfig.esm.tsbuildinfo +1 -1
- package/dist/esm/version.d.ts +1 -1
- package/dist/esm/version.js +1 -1
- package/dist/esm/version.js.map +1 -1
- package/dist/esnext/attributes.d.ts +11 -0
- package/dist/esnext/attributes.d.ts.map +1 -0
- package/dist/esnext/attributes.js +248 -0
- package/dist/esnext/attributes.js.map +1 -0
- package/dist/esnext/instrumentation.d.ts +11 -20
- package/dist/esnext/instrumentation.d.ts.map +1 -1
- package/dist/esnext/instrumentation.js +178 -331
- package/dist/esnext/instrumentation.js.map +1 -1
- package/dist/esnext/stream.d.ts +6 -0
- package/dist/esnext/stream.d.ts.map +1 -0
- package/dist/esnext/stream.js +203 -0
- package/dist/esnext/stream.js.map +1 -0
- package/dist/esnext/tsconfig.esnext.tsbuildinfo +1 -1
- package/dist/esnext/version.d.ts +1 -1
- package/dist/esnext/version.js +1 -1
- package/dist/esnext/version.js.map +1 -1
- package/dist/src/__tests__/attributes.test.d.ts +2 -0
- package/dist/src/__tests__/attributes.test.d.ts.map +1 -0
- package/dist/src/__tests__/attributes.test.js +50 -0
- package/dist/src/__tests__/attributes.test.js.map +1 -0
- package/dist/src/__tests__/auto.test.d.ts +2 -0
- package/dist/src/__tests__/auto.test.d.ts.map +1 -0
- package/dist/src/__tests__/auto.test.js +70 -0
- package/dist/src/__tests__/auto.test.js.map +1 -0
- package/dist/src/__tests__/instrumentation.test.js +1 -1
- package/dist/src/__tests__/instrumentation.test.js.map +1 -1
- package/dist/src/__tests__/lifecycle.test.d.ts +2 -0
- package/dist/src/__tests__/lifecycle.test.d.ts.map +1 -0
- package/dist/src/__tests__/lifecycle.test.js +232 -0
- package/dist/src/__tests__/lifecycle.test.js.map +1 -0
- package/dist/src/__tests__/metadata.test.d.ts +2 -0
- package/dist/src/__tests__/metadata.test.d.ts.map +1 -0
- package/dist/src/__tests__/metadata.test.js +320 -0
- package/dist/src/__tests__/metadata.test.js.map +1 -0
- package/dist/src/__tests__/sdk.test.d.ts +2 -0
- package/dist/src/__tests__/sdk.test.d.ts.map +1 -0
- package/dist/src/__tests__/sdk.test.js +883 -0
- package/dist/src/__tests__/sdk.test.js.map +1 -0
- package/dist/src/__tests__/timing.test.d.ts +2 -0
- package/dist/src/__tests__/timing.test.d.ts.map +1 -0
- package/dist/src/__tests__/timing.test.js +158 -0
- package/dist/src/__tests__/timing.test.js.map +1 -0
- package/dist/src/attributes.d.ts +11 -0
- package/dist/src/attributes.d.ts.map +1 -0
- package/dist/src/attributes.js +257 -0
- package/dist/src/attributes.js.map +1 -0
- package/dist/src/instrumentation.d.ts +11 -20
- package/dist/src/instrumentation.d.ts.map +1 -1
- package/dist/src/instrumentation.js +176 -329
- package/dist/src/instrumentation.js.map +1 -1
- package/dist/src/stream.d.ts +6 -0
- package/dist/src/stream.d.ts.map +1 -0
- package/dist/src/stream.js +215 -0
- package/dist/src/stream.js.map +1 -0
- package/dist/src/version.d.ts +1 -1
- package/dist/src/version.js +1 -1
- package/dist/src/version.js.map +1 -1
- package/dist/tsconfig.tsbuildinfo +1 -1
- package/package.json +18 -11
package/dist/esm/version.d.ts
CHANGED
|
@@ -1,2 +1,2 @@
|
|
|
1
|
-
export declare const VERSION = "0.1.
|
|
1
|
+
export declare const VERSION = "0.1.1";
|
|
2
2
|
//# sourceMappingURL=version.d.ts.map
|
package/dist/esm/version.js
CHANGED
|
@@ -1,2 +1,2 @@
|
|
|
1
|
-
export const VERSION = '0.1.
|
|
1
|
+
export const VERSION = '0.1.1'; // x-release-please-version
|
|
2
2
|
//# sourceMappingURL=version.js.map
|
package/dist/esm/version.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"version.js","sourceRoot":"","sources":["../../src/version.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,OAAO,GAAG,OAAO,CAAC"}
|
|
1
|
+
{"version":3,"file":"version.js","sourceRoot":"","sources":["../../src/version.ts"],"names":[],"mappings":"AAAA,MAAM,CAAC,MAAM,OAAO,GAAG,OAAO,CAAC,CAAC,2BAA2B"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import { Attributes } from "@opentelemetry/api";
|
|
2
|
+
import { TraceConfig } from "@traceai/fi-core";
|
|
3
|
+
export type Operation = "chat" | "chatStream" | "embed" | "rerank";
|
|
4
|
+
export type Data = Record<string, unknown>;
|
|
5
|
+
export declare const object: (value: unknown) => Data;
|
|
6
|
+
export declare const list: (value: unknown) => unknown[];
|
|
7
|
+
export declare const hideInput: (config: TraceConfig) => boolean;
|
|
8
|
+
export declare const hideOutput: (config: TraceConfig) => boolean;
|
|
9
|
+
export declare function inputAttributes(operation: Operation, request: Data, config: TraceConfig): Attributes;
|
|
10
|
+
export declare function outputAttributes(operation: Operation, response: Data, config: TraceConfig): Attributes;
|
|
11
|
+
//# sourceMappingURL=attributes.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"attributes.d.ts","sourceRoot":"","sources":["../../src/attributes.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAChD,OAAO,EAAuB,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAGpE,MAAM,MAAM,SAAS,GAAG,MAAM,GAAG,YAAY,GAAG,OAAO,GAAG,QAAQ,CAAC;AACnE,MAAM,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;AAC3C,eAAO,MAAM,MAAM,GAAI,OAAO,OAAO,KAAG,IAAwE,CAAC;AACjH,eAAO,MAAM,IAAI,GAAI,OAAO,OAAO,KAAG,OAAO,EAAuC,CAAC;AAGrF,eAAO,MAAM,SAAS,GAAI,QAAQ,WAAW,YAAoG,CAAC;AAClJ,eAAO,MAAM,UAAU,GAAI,QAAQ,WAAW,YAA6E,CAAC;AAgB5H,wBAAgB,eAAe,CAAC,SAAS,EAAE,SAAS,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,GAAG,UAAU,CA2DpG;AAyDD,wBAAgB,gBAAgB,CAAC,SAAS,EAAE,SAAS,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,EAAE,WAAW,GAAG,UAAU,CAkDtG"}
|
|
@@ -0,0 +1,248 @@
|
|
|
1
|
+
var __rest = (this && this.__rest) || function (s, e) {
|
|
2
|
+
var t = {};
|
|
3
|
+
for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0)
|
|
4
|
+
t[p] = s[p];
|
|
5
|
+
if (s != null && typeof Object.getOwnPropertySymbols === "function")
|
|
6
|
+
for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
|
|
7
|
+
if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i]))
|
|
8
|
+
t[p[i]] = s[p[i]];
|
|
9
|
+
}
|
|
10
|
+
return t;
|
|
11
|
+
};
|
|
12
|
+
import { safelyJSONStringify } from "@traceai/fi-core";
|
|
13
|
+
import { FISpanKind, LLMProvider, MimeType, SemanticConventions as S } from "@traceai/fi-semantic-conventions";
|
|
14
|
+
export const object = (value) => value !== null && typeof value === "object" ? value : {};
|
|
15
|
+
export const list = (value) => Array.isArray(value) ? value : [];
|
|
16
|
+
const json = (value) => { var _a; return (_a = safelyJSONStringify(value)) !== null && _a !== void 0 ? _a : ""; };
|
|
17
|
+
const field = (data, camel, snake) => { var _a; return (_a = data[camel]) !== null && _a !== void 0 ? _a : data[snake]; };
|
|
18
|
+
export const hideInput = (config) => config.hideInputs || config.hideInputMessages || config.hideInputText || config.hideInputImages;
|
|
19
|
+
export const hideOutput = (config) => config.hideOutputs || config.hideOutputMessages || config.hideOutputText;
|
|
20
|
+
/** Connector authentication is never telemetry content, even with input capture
|
|
21
|
+
* enabled. Project only the telemetry copy; the SDK still needs these tokens.
|
|
22
|
+
*/
|
|
23
|
+
function withoutConnectorCredentials(request) {
|
|
24
|
+
if (!Array.isArray(request.connectors))
|
|
25
|
+
return request;
|
|
26
|
+
return Object.assign(Object.assign({}, request), { connectors: request.connectors.map((connector) => {
|
|
27
|
+
const _a = object(connector), { userAccessToken: _camel, user_access_token: _snake } = _a, content = __rest(_a, ["userAccessToken", "user_access_token"]);
|
|
28
|
+
return content;
|
|
29
|
+
}) });
|
|
30
|
+
}
|
|
31
|
+
export function inputAttributes(operation, request, config) {
|
|
32
|
+
var _a, _b;
|
|
33
|
+
const embedding = operation === "embed";
|
|
34
|
+
const rerank = operation === "rerank";
|
|
35
|
+
const attrs = {
|
|
36
|
+
[S.FI_SPAN_KIND]: embedding ? FISpanKind.EMBEDDING : rerank ? FISpanKind.RERANKER : FISpanKind.LLM,
|
|
37
|
+
[S.LLM_PROVIDER]: LLMProvider.COHERE,
|
|
38
|
+
[S.GEN_AI_OPERATION_NAME]: embedding ? "embeddings" : rerank ? "rerank" : "chat",
|
|
39
|
+
};
|
|
40
|
+
if (typeof request.model === "string") {
|
|
41
|
+
attrs[embedding ? S.EMBEDDING_MODEL_NAME : S.LLM_MODEL_NAME] = request.model;
|
|
42
|
+
if (rerank)
|
|
43
|
+
attrs[S.RERANKER_MODEL_NAME] = request.model;
|
|
44
|
+
}
|
|
45
|
+
if (embedding) {
|
|
46
|
+
if (!hideInput(config)) {
|
|
47
|
+
attrs[S.INPUT_VALUE] = json((_b = (_a = request.texts) !== null && _a !== void 0 ? _a : request.inputs) !== null && _b !== void 0 ? _b : request.images);
|
|
48
|
+
attrs[S.INPUT_MIME_TYPE] = MimeType.JSON;
|
|
49
|
+
list(request.texts).forEach((text, index) => {
|
|
50
|
+
if (typeof text === "string")
|
|
51
|
+
attrs[`${S.EMBEDDING_EMBEDDINGS}.${index}.${S.EMBEDDING_TEXT}`] = text;
|
|
52
|
+
});
|
|
53
|
+
}
|
|
54
|
+
}
|
|
55
|
+
else if (rerank) {
|
|
56
|
+
attrs["cohere.rerank.documents_count"] = list(request.documents).length;
|
|
57
|
+
const topN = field(request, "topN", "top_n");
|
|
58
|
+
if (typeof topN === "number") {
|
|
59
|
+
attrs[S.RERANKER_TOP_K] = topN;
|
|
60
|
+
attrs["cohere.rerank.top_n"] = topN;
|
|
61
|
+
}
|
|
62
|
+
if (!hideInput(config)) {
|
|
63
|
+
attrs[S.INPUT_VALUE] = typeof request.query === "string" ? request.query : "";
|
|
64
|
+
attrs[S.INPUT_MIME_TYPE] = MimeType.TEXT;
|
|
65
|
+
attrs[S.RERANKER_QUERY] = attrs[S.INPUT_VALUE];
|
|
66
|
+
attrs["cohere.rerank.query"] = attrs[S.INPUT_VALUE];
|
|
67
|
+
attrs[S.RERANKER_INPUT_DOCUMENTS] = json(request.documents);
|
|
68
|
+
}
|
|
69
|
+
}
|
|
70
|
+
else {
|
|
71
|
+
// Only generation settings belong here. Chat history, tools, documents and
|
|
72
|
+
// arbitrary SDK fields must not bypass the content privacy controls.
|
|
73
|
+
const params = {};
|
|
74
|
+
for (const key of ["temperature", "maxTokens", "max_tokens", "p", "k", "seed", "frequencyPenalty", "presencePenalty"]) {
|
|
75
|
+
if (typeof request[key] === "number")
|
|
76
|
+
params[key] = request[key];
|
|
77
|
+
}
|
|
78
|
+
attrs[S.LLM_INVOCATION_PARAMETERS] = json(params);
|
|
79
|
+
if (!hideInput(config)) {
|
|
80
|
+
const messages = Array.isArray(request.messages) ? request.messages : [
|
|
81
|
+
...(request.preamble ? [{ role: "system", content: request.preamble }] : []),
|
|
82
|
+
...list(field(request, "chatHistory", "chat_history")).map((entry) => {
|
|
83
|
+
const item = object(entry);
|
|
84
|
+
return { role: item.role === "CHATBOT" ? "assistant" : String(item.role).toLowerCase(), content: item.message };
|
|
85
|
+
}),
|
|
86
|
+
{ role: "user", content: request.message },
|
|
87
|
+
];
|
|
88
|
+
attrs[S.LLM_INPUT_MESSAGES] = json(messages);
|
|
89
|
+
attrs[S.INPUT_VALUE] = typeof request.message === "string" ? request.message : json(messages);
|
|
90
|
+
attrs[S.INPUT_MIME_TYPE] = typeof request.message === "string" ? MimeType.TEXT : MimeType.JSON;
|
|
91
|
+
if (request.tools)
|
|
92
|
+
attrs[S.LLM_TOOLS] = json(request.tools);
|
|
93
|
+
}
|
|
94
|
+
}
|
|
95
|
+
if (!hideInput(config))
|
|
96
|
+
attrs[S.RAW_INPUT] = json(withoutConnectorCredentials(request));
|
|
97
|
+
return attrs;
|
|
98
|
+
}
|
|
99
|
+
/** Citation provenance is generated output, but source bodies echo inputs. Use an
|
|
100
|
+
* allowlist so document/tool payloads and future source fields cannot leak.
|
|
101
|
+
*/
|
|
102
|
+
function withoutCitationInputs(value) {
|
|
103
|
+
if (!Array.isArray(value))
|
|
104
|
+
return undefined;
|
|
105
|
+
return value.map(item => {
|
|
106
|
+
const citation = object(item);
|
|
107
|
+
const projected = {};
|
|
108
|
+
for (const key of ["start", "end", "text", "type"]) {
|
|
109
|
+
if (typeof citation[key] === "number" || typeof citation[key] === "string")
|
|
110
|
+
projected[key] = citation[key];
|
|
111
|
+
}
|
|
112
|
+
if (Array.isArray(citation.sources))
|
|
113
|
+
projected.sources = citation.sources.map(item => {
|
|
114
|
+
const source = object(item);
|
|
115
|
+
return Object.assign(Object.assign({}, (typeof source.type === "string" ? { type: source.type } : {})), (typeof source.id === "string" ? { id: source.id } : {}));
|
|
116
|
+
});
|
|
117
|
+
const documentIds = field(citation, "documentIds", "document_ids");
|
|
118
|
+
if (Array.isArray(documentIds))
|
|
119
|
+
projected.documentIds = documentIds.filter(id => typeof id === "string");
|
|
120
|
+
return projected;
|
|
121
|
+
});
|
|
122
|
+
}
|
|
123
|
+
/** Keep generated fields only when inputs are hidden. Responses can echo full
|
|
124
|
+
* chat history, embedding texts, reranked documents and citation sources.
|
|
125
|
+
* Project rather than mutate the SDK response or filter history by role: even
|
|
126
|
+
* earlier assistant messages are input to the current generation.
|
|
127
|
+
*/
|
|
128
|
+
function withoutInputEchoes(operation, response) {
|
|
129
|
+
if (operation === "embed")
|
|
130
|
+
return { embeddings: response.embeddings };
|
|
131
|
+
if (operation === "rerank")
|
|
132
|
+
return {
|
|
133
|
+
results: list(response.results).map((item) => {
|
|
134
|
+
const result = object(item);
|
|
135
|
+
return { index: result.index, relevanceScore: field(result, "relevanceScore", "relevance_score") };
|
|
136
|
+
}),
|
|
137
|
+
};
|
|
138
|
+
const message = object(response.message);
|
|
139
|
+
return {
|
|
140
|
+
text: response.text,
|
|
141
|
+
toolCalls: field(response, "toolCalls", "tool_calls"),
|
|
142
|
+
citations: withoutCitationInputs(response.citations),
|
|
143
|
+
message: {
|
|
144
|
+
role: message.role,
|
|
145
|
+
content: Array.isArray(message.content) ? message.content.map((part) => {
|
|
146
|
+
const content = object(part);
|
|
147
|
+
return { type: content.type, text: content.text, thinking: content.thinking };
|
|
148
|
+
}) : message.content,
|
|
149
|
+
toolCalls: field(message, "toolCalls", "tool_calls"),
|
|
150
|
+
toolPlan: field(message, "toolPlan", "tool_plan"),
|
|
151
|
+
citations: withoutCitationInputs(message.citations),
|
|
152
|
+
},
|
|
153
|
+
};
|
|
154
|
+
}
|
|
155
|
+
export function outputAttributes(operation, response, config) {
|
|
156
|
+
var _a, _b, _c, _d;
|
|
157
|
+
const attrs = usageAttributes(response);
|
|
158
|
+
const id = (_a = response.id) !== null && _a !== void 0 ? _a : field(response, "generationId", "generation_id");
|
|
159
|
+
if (typeof id === "string")
|
|
160
|
+
attrs[S.GEN_AI_RESPONSE_ID] = id;
|
|
161
|
+
if (hideInput(config))
|
|
162
|
+
response = withoutInputEchoes(operation, response);
|
|
163
|
+
if (operation === "embed") {
|
|
164
|
+
// raw.output would otherwise bypass hideEmbeddingVectors.
|
|
165
|
+
if (!hideOutput(config) && !config.hideEmbeddingVectors) {
|
|
166
|
+
const vectors = Array.isArray(response.embeddings) ? response.embeddings : object(response.embeddings).float;
|
|
167
|
+
list(vectors).forEach((vector, index) => {
|
|
168
|
+
if (Array.isArray(vector) && vector.every((n) => typeof n === "number")) {
|
|
169
|
+
attrs[`${S.EMBEDDING_EMBEDDINGS}.${index}.${S.EMBEDDING_VECTOR}`] = vector;
|
|
170
|
+
}
|
|
171
|
+
});
|
|
172
|
+
attrs[S.RAW_OUTPUT] = json(response);
|
|
173
|
+
}
|
|
174
|
+
}
|
|
175
|
+
else if (operation === "rerank") {
|
|
176
|
+
const results = list(response.results);
|
|
177
|
+
attrs["cohere.rerank.results_count"] = results.length;
|
|
178
|
+
results.forEach((item, index) => {
|
|
179
|
+
const result = object(item);
|
|
180
|
+
const score = field(result, "relevanceScore", "relevance_score");
|
|
181
|
+
if (typeof result.index === "number")
|
|
182
|
+
attrs[`cohere.rerank.results.${index}.index`] = result.index;
|
|
183
|
+
if (typeof score === "number")
|
|
184
|
+
attrs[`cohere.rerank.results.${index}.relevance_score`] = score;
|
|
185
|
+
const text = object(result.document).text;
|
|
186
|
+
if (!hideOutput(config) && typeof text === "string")
|
|
187
|
+
attrs[`cohere.rerank.results.${index}.document`] = text;
|
|
188
|
+
});
|
|
189
|
+
if (!hideOutput(config)) {
|
|
190
|
+
attrs[S.RERANKER_OUTPUT_DOCUMENTS] = json(results);
|
|
191
|
+
attrs[S.RAW_OUTPUT] = json(response);
|
|
192
|
+
}
|
|
193
|
+
}
|
|
194
|
+
else if (!hideOutput(config)) {
|
|
195
|
+
const message = object(response.message);
|
|
196
|
+
const text = typeof response.text === "string" ? response.text : list(message.content).map((part) => { var _a; return (_a = object(part).text) !== null && _a !== void 0 ? _a : ""; }).join("");
|
|
197
|
+
const calls = (_b = field(message, "toolCalls", "tool_calls")) !== null && _b !== void 0 ? _b : field(response, "toolCalls", "tool_calls");
|
|
198
|
+
const normalized = { role: "assistant", content: (_c = message.content) !== null && _c !== void 0 ? _c : text };
|
|
199
|
+
const citations = (_d = message.citations) !== null && _d !== void 0 ? _d : response.citations;
|
|
200
|
+
if (Array.isArray(citations))
|
|
201
|
+
normalized.citations = citations;
|
|
202
|
+
const toolPlan = field(message, "toolPlan", "tool_plan");
|
|
203
|
+
if (typeof toolPlan === "string")
|
|
204
|
+
normalized.tool_plan = toolPlan;
|
|
205
|
+
if (calls)
|
|
206
|
+
normalized.tool_calls = list(calls).map((item) => {
|
|
207
|
+
const call = object(item);
|
|
208
|
+
return call.function ? call : { function: { name: call.name, arguments: typeof call.parameters === "string" ? call.parameters : json(call.parameters) } };
|
|
209
|
+
});
|
|
210
|
+
attrs[S.OUTPUT_VALUE] = text;
|
|
211
|
+
attrs[S.OUTPUT_MIME_TYPE] = MimeType.TEXT;
|
|
212
|
+
attrs[S.LLM_OUTPUT_MESSAGES] = json([normalized]);
|
|
213
|
+
attrs[S.RAW_OUTPUT] = json(response);
|
|
214
|
+
}
|
|
215
|
+
return attrs;
|
|
216
|
+
}
|
|
217
|
+
function usageAttributes(response) {
|
|
218
|
+
var _a, _b;
|
|
219
|
+
const usage = object((_a = response.usage) !== null && _a !== void 0 ? _a : response.meta);
|
|
220
|
+
// Keep the legacy billed-token fallback only when actual usage is absent;
|
|
221
|
+
// discounted billed units must never replace or be added to actual tokens.
|
|
222
|
+
const billed = object(field(usage, "billedUnits", "billed_units"));
|
|
223
|
+
const tokens = object((_b = usage.tokens) !== null && _b !== void 0 ? _b : billed);
|
|
224
|
+
const input = field(tokens, "inputTokens", "input_tokens");
|
|
225
|
+
const output = field(tokens, "outputTokens", "output_tokens");
|
|
226
|
+
const attrs = {};
|
|
227
|
+
const numeric = (value) => typeof value === "number" && Number.isFinite(value) && value >= 0;
|
|
228
|
+
if (numeric(input))
|
|
229
|
+
attrs[S.LLM_TOKEN_COUNT_PROMPT] = input;
|
|
230
|
+
if (numeric(output))
|
|
231
|
+
attrs[S.LLM_TOKEN_COUNT_COMPLETION] = output;
|
|
232
|
+
if (numeric(input) || numeric(output))
|
|
233
|
+
attrs[S.LLM_TOKEN_COUNT_TOTAL] = (numeric(input) ? input : 0) + (numeric(output) ? output : 0);
|
|
234
|
+
const cached = field(usage, "cachedTokens", "cached_tokens");
|
|
235
|
+
if (numeric(cached))
|
|
236
|
+
attrs[S.LLM_TOKEN_COUNT_PROMPT_DETAILS_CACHE_READ] = cached;
|
|
237
|
+
for (const [camel, snake] of [
|
|
238
|
+
["inputTokens", "input_tokens"], ["outputTokens", "output_tokens"],
|
|
239
|
+
["searchUnits", "search_units"], ["classifications", "classifications"],
|
|
240
|
+
["images", "images"], ["imageTokens", "image_tokens"],
|
|
241
|
+
]) {
|
|
242
|
+
const value = field(billed, camel, snake);
|
|
243
|
+
if (numeric(value))
|
|
244
|
+
attrs[`cohere.usage.billed_units.${snake}`] = value;
|
|
245
|
+
}
|
|
246
|
+
return attrs;
|
|
247
|
+
}
|
|
248
|
+
//# sourceMappingURL=attributes.js.map
|
|
@@ -0,0 +1 @@
|
|
|
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,WAAC,OAAA,MAAA,mBAAmB,CAAC,KAAK,CAAC,mCAAI,EAAE,CAAA,EAAA,CAAC;AAClE,MAAM,KAAK,GAAG,CAAC,IAAU,EAAE,KAAa,EAAE,KAAa,EAAE,EAAE,WAAC,OAAA,MAAA,IAAI,CAAC,KAAK,CAAC,mCAAI,IAAI,CAAC,KAAK,CAAC,CAAA,EAAA,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,uCACK,OAAO,KACV,UAAU,EAAE,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE;YAC/C,MAAM,KAAqE,MAAM,CAAC,SAAS,CAAC,EAAtF,EAAE,eAAe,EAAE,MAAM,EAAE,iBAAiB,EAAE,MAAM,OAAkC,EAA7B,OAAO,cAAhE,wCAAkE,CAAoB,CAAC;YAC7F,OAAO,OAAO,CAAC;QACjB,CAAC,CAAC,IACF;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,MAAA,MAAA,OAAO,CAAC,KAAK,mCAAI,OAAO,CAAC,MAAM,mCAAI,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,uCACK,CAAC,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,GAC9D,CAAC,OAAO,MAAM,CAAC,EAAE,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAC3D;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,MAAA,QAAQ,CAAC,EAAE,mCAAI,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,WAAC,OAAA,MAAA,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,mCAAI,EAAE,CAAA,EAAA,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACvI,MAAM,KAAK,GAAG,MAAA,KAAK,CAAC,OAAO,EAAE,WAAW,EAAE,YAAY,CAAC,mCAAI,KAAK,CAAC,QAAQ,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;QACtG,MAAM,UAAU,GAAS,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,MAAA,OAAO,CAAC,OAAO,mCAAI,IAAI,EAAE,CAAC;QACjF,MAAM,SAAS,GAAG,MAAA,OAAO,CAAC,SAAS,mCAAI,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,MAAA,QAAQ,CAAC,KAAK,mCAAI,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,MAAA,KAAK,CAAC,MAAM,mCAAI,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"}
|
|
@@ -1,34 +1,25 @@
|
|
|
1
1
|
import { InstrumentationBase, InstrumentationConfig, InstrumentationNodeModuleDefinition } from "@opentelemetry/instrumentation";
|
|
2
2
|
import { TraceConfigOptions } from "@traceai/fi-core";
|
|
3
|
-
|
|
4
|
-
|
|
5
|
-
*/
|
|
3
|
+
import { Data } from "./attributes";
|
|
4
|
+
/** True when at least one SDK method is instrumented, across module copies. */
|
|
6
5
|
export declare function isPatched(): boolean;
|
|
7
|
-
/**
|
|
8
|
-
* @param instrumentationConfig The config for the instrumentation
|
|
9
|
-
* @param traceConfig The FI trace configuration for masking sensitive information
|
|
10
|
-
*/
|
|
6
|
+
/** Cohere v1/v2 instrumentation. Install before constructing SDK clients. */
|
|
11
7
|
export declare class CohereInstrumentation extends InstrumentationBase {
|
|
12
|
-
private fiTracer;
|
|
13
8
|
private _traceConfig?;
|
|
9
|
+
private patches?;
|
|
10
|
+
private manualModules?;
|
|
14
11
|
readonly instrumentationName = "@traceai/cohere";
|
|
15
|
-
constructor({ instrumentationConfig, traceConfig
|
|
12
|
+
constructor({ instrumentationConfig, traceConfig }?: {
|
|
16
13
|
instrumentationConfig?: InstrumentationConfig;
|
|
17
14
|
traceConfig?: TraceConfigOptions;
|
|
18
15
|
});
|
|
19
|
-
enable(): void;
|
|
20
16
|
protected init(): InstrumentationNodeModuleDefinition;
|
|
21
|
-
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
manuallyInstrument(module: Record<string, unknown>): void;
|
|
25
|
-
/**
|
|
26
|
-
* Patches the Cohere module
|
|
27
|
-
*/
|
|
17
|
+
enable(): void;
|
|
18
|
+
disable(): void;
|
|
19
|
+
manuallyInstrument(module: Data): void;
|
|
28
20
|
private patch;
|
|
29
|
-
/**
|
|
30
|
-
* Un-patches the Cohere module
|
|
31
|
-
*/
|
|
32
21
|
private unpatch;
|
|
22
|
+
private restorePatches;
|
|
23
|
+
private wrapMethod;
|
|
33
24
|
}
|
|
34
25
|
//# sourceMappingURL=instrumentation.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"instrumentation.d.ts","sourceRoot":"","sources":["../../src/instrumentation.ts"],"names":[],"mappings":"AAAA,OAAO,
|
|
1
|
+
{"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"}
|