@loongsuite/otel-util-genai 0.1.0-beta.0

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.
Files changed (103) hide show
  1. package/LICENSE +201 -0
  2. package/README.md +262 -0
  3. package/dist/environment-variables.d.ts +8 -0
  4. package/dist/environment-variables.d.ts.map +1 -0
  5. package/dist/environment-variables.js +21 -0
  6. package/dist/environment-variables.js.map +1 -0
  7. package/dist/event-log/converter.d.ts +19 -0
  8. package/dist/event-log/converter.d.ts.map +1 -0
  9. package/dist/event-log/converter.js +216 -0
  10. package/dist/event-log/converter.js.map +1 -0
  11. package/dist/event-log/field-mapping.d.ts +53 -0
  12. package/dist/event-log/field-mapping.d.ts.map +1 -0
  13. package/dist/event-log/field-mapping.js +400 -0
  14. package/dist/event-log/field-mapping.js.map +1 -0
  15. package/dist/event-log/grouping.d.ts +67 -0
  16. package/dist/event-log/grouping.d.ts.map +1 -0
  17. package/dist/event-log/grouping.js +241 -0
  18. package/dist/event-log/grouping.js.map +1 -0
  19. package/dist/event-log/index.d.ts +5 -0
  20. package/dist/event-log/index.d.ts.map +1 -0
  21. package/dist/event-log/index.js +18 -0
  22. package/dist/event-log/index.js.map +1 -0
  23. package/dist/event-log/parent-context.d.ts +14 -0
  24. package/dist/event-log/parent-context.d.ts.map +1 -0
  25. package/dist/event-log/parent-context.js +45 -0
  26. package/dist/event-log/parent-context.js.map +1 -0
  27. package/dist/event-log/readable-spans.d.ts +36 -0
  28. package/dist/event-log/readable-spans.d.ts.map +1 -0
  29. package/dist/event-log/readable-spans.js +73 -0
  30. package/dist/event-log/readable-spans.js.map +1 -0
  31. package/dist/event-log/types.d.ts +78 -0
  32. package/dist/event-log/types.d.ts.map +1 -0
  33. package/dist/event-log/types.js +35 -0
  34. package/dist/event-log/types.js.map +1 -0
  35. package/dist/extended-environment-variables.d.ts +12 -0
  36. package/dist/extended-environment-variables.d.ts.map +1 -0
  37. package/dist/extended-environment-variables.js +25 -0
  38. package/dist/extended-environment-variables.js.map +1 -0
  39. package/dist/extended-handler.d.ts +50 -0
  40. package/dist/extended-handler.d.ts.map +1 -0
  41. package/dist/extended-handler.js +334 -0
  42. package/dist/extended-handler.js.map +1 -0
  43. package/dist/extended-metrics.d.ts +13 -0
  44. package/dist/extended-metrics.d.ts.map +1 -0
  45. package/dist/extended-metrics.js +27 -0
  46. package/dist/extended-metrics.js.map +1 -0
  47. package/dist/extended-span-utils.d.ts +14 -0
  48. package/dist/extended-span-utils.d.ts.map +1 -0
  49. package/dist/extended-span-utils.js +427 -0
  50. package/dist/extended-span-utils.js.map +1 -0
  51. package/dist/extended-types.d.ts +150 -0
  52. package/dist/extended-types.d.ts.map +1 -0
  53. package/dist/extended-types.js +46 -0
  54. package/dist/extended-types.js.map +1 -0
  55. package/dist/handler.d.ts +23 -0
  56. package/dist/handler.d.ts.map +1 -0
  57. package/dist/handler.js +127 -0
  58. package/dist/handler.js.map +1 -0
  59. package/dist/index.d.ts +22 -0
  60. package/dist/index.d.ts.map +1 -0
  61. package/dist/index.js +43 -0
  62. package/dist/index.js.map +1 -0
  63. package/dist/instruments.d.ts +4 -0
  64. package/dist/instruments.d.ts.map +1 -0
  65. package/dist/instruments.js +37 -0
  66. package/dist/instruments.js.map +1 -0
  67. package/dist/memory/memory-types.d.ts +26 -0
  68. package/dist/memory/memory-types.d.ts.map +1 -0
  69. package/dist/memory/memory-types.js +17 -0
  70. package/dist/memory/memory-types.js.map +1 -0
  71. package/dist/memory/memory-utils.d.ts +7 -0
  72. package/dist/memory/memory-utils.d.ts.map +1 -0
  73. package/dist/memory/memory-utils.js +121 -0
  74. package/dist/memory/memory-utils.js.map +1 -0
  75. package/dist/metrics.d.ts +11 -0
  76. package/dist/metrics.d.ts.map +1 -0
  77. package/dist/metrics.js +77 -0
  78. package/dist/metrics.js.map +1 -0
  79. package/dist/semconv/gen-ai-extended-attributes.d.ts +101 -0
  80. package/dist/semconv/gen-ai-extended-attributes.d.ts.map +1 -0
  81. package/dist/semconv/gen-ai-extended-attributes.js +120 -0
  82. package/dist/semconv/gen-ai-extended-attributes.js.map +1 -0
  83. package/dist/semconv/gen-ai-memory-attributes.d.ts +28 -0
  84. package/dist/semconv/gen-ai-memory-attributes.d.ts.map +1 -0
  85. package/dist/semconv/gen-ai-memory-attributes.js +42 -0
  86. package/dist/semconv/gen-ai-memory-attributes.js.map +1 -0
  87. package/dist/span-utils.d.ts +18 -0
  88. package/dist/span-utils.d.ts.map +1 -0
  89. package/dist/span-utils.js +236 -0
  90. package/dist/span-utils.js.map +1 -0
  91. package/dist/types.d.ts +117 -0
  92. package/dist/types.d.ts.map +1 -0
  93. package/dist/types.js +33 -0
  94. package/dist/types.js.map +1 -0
  95. package/dist/utils.d.ts +8 -0
  96. package/dist/utils.d.ts.map +1 -0
  97. package/dist/utils.js +88 -0
  98. package/dist/utils.js.map +1 -0
  99. package/dist/version.d.ts +2 -0
  100. package/dist/version.d.ts.map +1 -0
  101. package/dist/version.js +2 -0
  102. package/dist/version.js.map +1 -0
  103. package/package.json +72 -0
@@ -0,0 +1,427 @@
1
+ // Copyright The OpenTelemetry Authors
2
+ //
3
+ // Licensed under the Apache License, Version 2.0 (the "License");
4
+ // you may not use this file except in compliance with the License.
5
+ // You may obtain a copy of the License at
6
+ //
7
+ // http://www.apache.org/licenses/LICENSE-2.0
8
+ //
9
+ // Unless required by applicable law or agreed to in writing, software
10
+ // distributed under the License is distributed on an "AS IS" BASIS,
11
+ // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ // See the License for the specific language governing permissions and
13
+ // limitations under the License.
14
+ import { ContentCapturingMode } from "./types.js";
15
+ import { GEN_AI_OPERATION_NAME, GEN_AI_SPAN_KIND, GEN_AI_REQUEST_MODEL, GEN_AI_PROVIDER_NAME, GEN_AI_REQUEST_TEMPERATURE, GEN_AI_REQUEST_TOP_P, GEN_AI_REQUEST_TOP_K, GEN_AI_REQUEST_FREQUENCY_PENALTY, GEN_AI_REQUEST_PRESENCE_PENALTY, GEN_AI_REQUEST_MAX_TOKENS, GEN_AI_REQUEST_SEED, GEN_AI_REQUEST_STOP_SEQUENCES, GEN_AI_REQUEST_ENCODING_FORMATS, GEN_AI_REQUEST_CHOICE_COUNT, GEN_AI_RESPONSE_FINISH_REASONS, GEN_AI_RESPONSE_MODEL, GEN_AI_RESPONSE_ID, GEN_AI_RESPONSE_TIME_TO_FIRST_TOKEN, GEN_AI_USAGE_INPUT_TOKENS, GEN_AI_USAGE_OUTPUT_TOKENS, GEN_AI_USAGE_TOTAL_TOKENS, GEN_AI_USAGE_CACHE_CREATION_INPUT_TOKENS, GEN_AI_USAGE_CACHE_READ_INPUT_TOKENS, GEN_AI_OUTPUT_TYPE, GEN_AI_CONVERSATION_ID, GEN_AI_AGENT_ID, GEN_AI_AGENT_NAME, GEN_AI_AGENT_DESCRIPTION, GEN_AI_DATA_SOURCE_ID, GEN_AI_TOOL_CALL_ID, GEN_AI_TOOL_NAME, GEN_AI_TOOL_DESCRIPTION, GEN_AI_TOOL_TYPE, GEN_AI_TOOL_CALL_ARGUMENTS, GEN_AI_TOOL_CALL_RESULT, GEN_AI_EMBEDDINGS_DIMENSION_COUNT, GEN_AI_RETRIEVAL_QUERY_TEXT, GEN_AI_RETRIEVAL_DOCUMENTS, GEN_AI_RERANK_DOCUMENTS_COUNT, GEN_AI_RERANK_SCORING_PROMPT, GEN_AI_RERANK_RETURN_DOCUMENTS, GEN_AI_RERANK_MAX_CHUNKS_PER_DOC, GEN_AI_RERANK_DEVICE, GEN_AI_RERANK_BATCH_SIZE, GEN_AI_RERANK_MAX_LENGTH, GEN_AI_RERANK_NORMALIZE, GEN_AI_RERANK_INPUT_DOCUMENTS, GEN_AI_RERANK_OUTPUT_DOCUMENTS, GEN_AI_SESSION_ID, GEN_AI_USER_ID, GEN_AI_REACT_FINISH_REASON, GEN_AI_REACT_ROUND, SERVER_ADDRESS, SERVER_PORT, ERROR_TYPE, GenAiSpanKindValues, GenAiOperationNameValues, GenAiExtendedOperationNameValues, } from "./semconv/gen-ai-extended-attributes.js";
16
+ import { getLlmMessagesAttributesForSpan, getToolDefinitionsForSpan, } from "./span-utils.js";
17
+ import { isExperimentalMode, getContentCapturingMode, shouldEmitEvent, genAiJsonDumps, shouldCaptureContentInSpan, } from "./utils.js";
18
+ // ==================== Embedding ====================
19
+ export function applyEmbeddingFinishAttributes(span, invocation) {
20
+ span.updateName(`${GenAiOperationNameValues.EMBEDDINGS} ${invocation.requestModel}`.trim());
21
+ const attrs = {
22
+ [GEN_AI_OPERATION_NAME]: GenAiOperationNameValues.EMBEDDINGS,
23
+ [GEN_AI_SPAN_KIND]: GenAiSpanKindValues.EMBEDDING,
24
+ };
25
+ if (invocation.requestModel) {
26
+ attrs[GEN_AI_REQUEST_MODEL] = invocation.requestModel;
27
+ }
28
+ if (invocation.provider != null) {
29
+ attrs[GEN_AI_PROVIDER_NAME] = invocation.provider;
30
+ }
31
+ if (invocation.serverPort != null) {
32
+ attrs[SERVER_PORT] = invocation.serverPort;
33
+ }
34
+ if (invocation.dimensionCount != null) {
35
+ attrs[GEN_AI_EMBEDDINGS_DIMENSION_COUNT] = invocation.dimensionCount;
36
+ }
37
+ if (invocation.encodingFormats != null) {
38
+ attrs[GEN_AI_REQUEST_ENCODING_FORMATS] = invocation.encodingFormats;
39
+ }
40
+ if (invocation.inputTokens != null) {
41
+ attrs[GEN_AI_USAGE_INPUT_TOKENS] = invocation.inputTokens;
42
+ attrs[GEN_AI_USAGE_TOTAL_TOKENS] = invocation.inputTokens;
43
+ }
44
+ if (invocation.serverAddress != null) {
45
+ attrs[SERVER_ADDRESS] = invocation.serverAddress;
46
+ }
47
+ if (invocation.attributes) {
48
+ Object.assign(attrs, invocation.attributes);
49
+ }
50
+ span.setAttributes(attrs);
51
+ }
52
+ // ==================== Create Agent ====================
53
+ export function applyCreateAgentFinishAttributes(span, invocation) {
54
+ span.updateName(`${GenAiOperationNameValues.CREATE_AGENT} ${invocation.agentName ?? ""}`.trim());
55
+ const attrs = {
56
+ [GEN_AI_OPERATION_NAME]: GenAiOperationNameValues.CREATE_AGENT,
57
+ [GEN_AI_SPAN_KIND]: GenAiSpanKindValues.AGENT,
58
+ };
59
+ if (invocation.provider) {
60
+ attrs[GEN_AI_PROVIDER_NAME] = invocation.provider;
61
+ }
62
+ if (invocation.agentDescription != null) {
63
+ attrs[GEN_AI_AGENT_DESCRIPTION] = invocation.agentDescription;
64
+ }
65
+ if (invocation.agentId != null) {
66
+ attrs[GEN_AI_AGENT_ID] = invocation.agentId;
67
+ }
68
+ if (invocation.agentName != null) {
69
+ attrs[GEN_AI_AGENT_NAME] = invocation.agentName;
70
+ }
71
+ if (invocation.requestModel != null) {
72
+ attrs[GEN_AI_REQUEST_MODEL] = invocation.requestModel;
73
+ }
74
+ if (invocation.serverPort != null) {
75
+ attrs[SERVER_PORT] = invocation.serverPort;
76
+ }
77
+ if (invocation.serverAddress != null) {
78
+ attrs[SERVER_ADDRESS] = invocation.serverAddress;
79
+ }
80
+ if (invocation.attributes) {
81
+ Object.assign(attrs, invocation.attributes);
82
+ }
83
+ span.setAttributes(attrs);
84
+ }
85
+ // ==================== Execute Tool ====================
86
+ function getToolCallDataAttributes(toolCallArguments, toolCallResult) {
87
+ if (!isExperimentalMode() || !shouldCaptureContentInSpan())
88
+ return {};
89
+ const attrs = {};
90
+ if (toolCallArguments != null) {
91
+ attrs[GEN_AI_TOOL_CALL_ARGUMENTS] =
92
+ typeof toolCallArguments === "string"
93
+ ? toolCallArguments
94
+ : genAiJsonDumps(toolCallArguments);
95
+ }
96
+ if (toolCallResult != null) {
97
+ attrs[GEN_AI_TOOL_CALL_RESULT] =
98
+ typeof toolCallResult === "string"
99
+ ? toolCallResult
100
+ : genAiJsonDumps(toolCallResult);
101
+ }
102
+ return attrs;
103
+ }
104
+ export function applyExecuteToolFinishAttributes(span, invocation) {
105
+ span.updateName(`${GenAiOperationNameValues.EXECUTE_TOOL} ${invocation.toolName}`.trim());
106
+ const attrs = {
107
+ [GEN_AI_OPERATION_NAME]: GenAiOperationNameValues.EXECUTE_TOOL,
108
+ [GEN_AI_SPAN_KIND]: GenAiSpanKindValues.TOOL,
109
+ };
110
+ if (invocation.toolCallId != null) {
111
+ attrs[GEN_AI_TOOL_CALL_ID] = invocation.toolCallId;
112
+ }
113
+ if (invocation.toolDescription != null) {
114
+ attrs[GEN_AI_TOOL_DESCRIPTION] = invocation.toolDescription;
115
+ }
116
+ if (invocation.toolName) {
117
+ attrs[GEN_AI_TOOL_NAME] = invocation.toolName;
118
+ }
119
+ if (invocation.toolType != null) {
120
+ attrs[GEN_AI_TOOL_TYPE] = invocation.toolType;
121
+ }
122
+ Object.assign(attrs, getToolCallDataAttributes(invocation.toolCallArguments, invocation.toolCallResult));
123
+ if (invocation.attributes) {
124
+ Object.assign(attrs, invocation.attributes);
125
+ }
126
+ span.setAttributes(attrs);
127
+ }
128
+ // ==================== Invoke Agent ====================
129
+ function getInvokeAgentSpanName(invocation) {
130
+ if (invocation.agentName) {
131
+ return `${GenAiOperationNameValues.INVOKE_AGENT} ${invocation.agentName}`.trim();
132
+ }
133
+ return GenAiOperationNameValues.INVOKE_AGENT;
134
+ }
135
+ export function applyInvokeAgentFinishAttributes(span, invocation) {
136
+ span.updateName(getInvokeAgentSpanName(invocation));
137
+ const attrs = {
138
+ [GEN_AI_OPERATION_NAME]: GenAiOperationNameValues.INVOKE_AGENT,
139
+ [GEN_AI_SPAN_KIND]: GenAiSpanKindValues.AGENT,
140
+ };
141
+ if (invocation.provider) {
142
+ attrs[GEN_AI_PROVIDER_NAME] = invocation.provider;
143
+ }
144
+ if (invocation.agentDescription != null) {
145
+ attrs[GEN_AI_AGENT_DESCRIPTION] = invocation.agentDescription;
146
+ }
147
+ if (invocation.agentId != null) {
148
+ attrs[GEN_AI_AGENT_ID] = invocation.agentId;
149
+ }
150
+ if (invocation.agentName != null) {
151
+ attrs[GEN_AI_AGENT_NAME] = invocation.agentName;
152
+ }
153
+ if (invocation.conversationId != null) {
154
+ attrs[GEN_AI_CONVERSATION_ID] = invocation.conversationId;
155
+ }
156
+ if (invocation.dataSourceId != null) {
157
+ attrs[GEN_AI_DATA_SOURCE_ID] = invocation.dataSourceId;
158
+ }
159
+ if (invocation.requestModel != null) {
160
+ attrs[GEN_AI_REQUEST_MODEL] = invocation.requestModel;
161
+ }
162
+ // Request attributes
163
+ if (invocation.temperature != null) {
164
+ attrs[GEN_AI_REQUEST_TEMPERATURE] = invocation.temperature;
165
+ }
166
+ if (invocation.topP != null) {
167
+ attrs[GEN_AI_REQUEST_TOP_P] = invocation.topP;
168
+ }
169
+ if (invocation.frequencyPenalty != null) {
170
+ attrs[GEN_AI_REQUEST_FREQUENCY_PENALTY] = invocation.frequencyPenalty;
171
+ }
172
+ if (invocation.presencePenalty != null) {
173
+ attrs[GEN_AI_REQUEST_PRESENCE_PENALTY] = invocation.presencePenalty;
174
+ }
175
+ if (invocation.maxTokens != null) {
176
+ attrs[GEN_AI_REQUEST_MAX_TOKENS] = invocation.maxTokens;
177
+ }
178
+ if (invocation.seed != null) {
179
+ attrs[GEN_AI_REQUEST_SEED] = invocation.seed;
180
+ }
181
+ if (invocation.stopSequences != null) {
182
+ attrs[GEN_AI_REQUEST_STOP_SEQUENCES] = invocation.stopSequences;
183
+ }
184
+ // Response attributes
185
+ if (invocation.finishReasons != null) {
186
+ attrs[GEN_AI_RESPONSE_FINISH_REASONS] = invocation.finishReasons;
187
+ }
188
+ if (invocation.responseId != null) {
189
+ attrs[GEN_AI_RESPONSE_ID] = invocation.responseId;
190
+ }
191
+ if (invocation.responseModelName != null) {
192
+ attrs[GEN_AI_RESPONSE_MODEL] = invocation.responseModelName;
193
+ }
194
+ if (invocation.inputTokens != null) {
195
+ attrs[GEN_AI_USAGE_INPUT_TOKENS] = invocation.inputTokens;
196
+ }
197
+ if (invocation.outputTokens != null) {
198
+ attrs[GEN_AI_USAGE_OUTPUT_TOKENS] = invocation.outputTokens;
199
+ }
200
+ if (invocation.usageCacheCreationInputTokens != null) {
201
+ attrs[GEN_AI_USAGE_CACHE_CREATION_INPUT_TOKENS] =
202
+ invocation.usageCacheCreationInputTokens;
203
+ }
204
+ if (invocation.usageCacheReadInputTokens != null) {
205
+ attrs[GEN_AI_USAGE_CACHE_READ_INPUT_TOKENS] =
206
+ invocation.usageCacheReadInputTokens;
207
+ }
208
+ let totalTokens = 0;
209
+ if (invocation.inputTokens != null)
210
+ totalTokens += invocation.inputTokens;
211
+ if (invocation.outputTokens != null)
212
+ totalTokens += invocation.outputTokens;
213
+ if (totalTokens > 0) {
214
+ attrs[GEN_AI_USAGE_TOTAL_TOKENS] = totalTokens;
215
+ }
216
+ if (invocation.monotonicFirstTokenS != null &&
217
+ invocation.monotonicStartS != null &&
218
+ invocation.monotonicFirstTokenS >= invocation.monotonicStartS) {
219
+ attrs[GEN_AI_RESPONSE_TIME_TO_FIRST_TOKEN] = Math.round((invocation.monotonicFirstTokenS - invocation.monotonicStartS) * 1e9);
220
+ }
221
+ // Additional span-specific attributes
222
+ if (invocation.outputType != null) {
223
+ attrs[GEN_AI_OUTPUT_TYPE] = invocation.outputType;
224
+ }
225
+ if (invocation.choiceCount != null && invocation.choiceCount !== 1) {
226
+ attrs[GEN_AI_REQUEST_CHOICE_COUNT] = invocation.choiceCount;
227
+ }
228
+ if (invocation.serverPort != null) {
229
+ attrs[SERVER_PORT] = invocation.serverPort;
230
+ }
231
+ if (invocation.serverAddress != null) {
232
+ attrs[SERVER_ADDRESS] = invocation.serverAddress;
233
+ }
234
+ Object.assign(attrs, getLlmMessagesAttributesForSpan(invocation.inputMessages ?? [], invocation.outputMessages ?? [], invocation.systemInstruction));
235
+ Object.assign(attrs, getToolDefinitionsForSpan(invocation.toolDefinitions));
236
+ if (invocation.attributes) {
237
+ Object.assign(attrs, invocation.attributes);
238
+ }
239
+ span.setAttributes(attrs);
240
+ }
241
+ export function maybeEmitInvokeAgentEvent(logger, span, invocation, error) {
242
+ if (!isExperimentalMode() || !shouldEmitEvent() || !logger)
243
+ return;
244
+ const attrs = {
245
+ [GEN_AI_OPERATION_NAME]: GenAiOperationNameValues.INVOKE_AGENT,
246
+ [GEN_AI_SPAN_KIND]: GenAiSpanKindValues.AGENT,
247
+ };
248
+ if (invocation.provider) {
249
+ attrs[GEN_AI_PROVIDER_NAME] = invocation.provider;
250
+ }
251
+ if (invocation.agentName != null) {
252
+ attrs[GEN_AI_AGENT_NAME] = invocation.agentName;
253
+ }
254
+ if (error) {
255
+ attrs[ERROR_TYPE] = error.type;
256
+ }
257
+ logger.emit({
258
+ name: "gen_ai.client.agent.invoke.operation.details",
259
+ attributes: attrs,
260
+ });
261
+ }
262
+ // ==================== Retrieval ====================
263
+ function getRetrievalDocumentsAttributes(documents) {
264
+ if (!isExperimentalMode() || !documents?.length)
265
+ return {};
266
+ const mode = getContentCapturingMode();
267
+ const shouldRecordFull = mode === ContentCapturingMode.SPAN_ONLY ||
268
+ mode === ContentCapturingMode.SPAN_AND_EVENT;
269
+ const dicts = documents.map((doc) => {
270
+ if (shouldRecordFull) {
271
+ return { ...doc };
272
+ }
273
+ return { id: doc.id, score: doc.score };
274
+ });
275
+ return { [GEN_AI_RETRIEVAL_DOCUMENTS]: genAiJsonDumps(dicts) };
276
+ }
277
+ export function applyRetrievalFinishAttributes(span, invocation) {
278
+ const opName = GenAiExtendedOperationNameValues.RETRIEVAL;
279
+ const spanName = invocation.dataSourceId
280
+ ? `${opName} ${invocation.dataSourceId}`.trim()
281
+ : opName;
282
+ span.updateName(spanName);
283
+ const attrs = {
284
+ [GEN_AI_OPERATION_NAME]: opName,
285
+ [GEN_AI_SPAN_KIND]: GenAiSpanKindValues.RETRIEVER,
286
+ };
287
+ if (invocation.dataSourceId != null) {
288
+ attrs[GEN_AI_DATA_SOURCE_ID] = invocation.dataSourceId;
289
+ }
290
+ if (invocation.provider != null) {
291
+ attrs[GEN_AI_PROVIDER_NAME] = invocation.provider;
292
+ }
293
+ if (invocation.requestModel != null) {
294
+ attrs[GEN_AI_REQUEST_MODEL] = invocation.requestModel;
295
+ }
296
+ if (invocation.topK != null) {
297
+ attrs[GEN_AI_REQUEST_TOP_K] = invocation.topK;
298
+ }
299
+ if (invocation.query != null &&
300
+ isExperimentalMode() &&
301
+ shouldCaptureContentInSpan()) {
302
+ attrs[GEN_AI_RETRIEVAL_QUERY_TEXT] = invocation.query;
303
+ }
304
+ if (invocation.serverAddress != null) {
305
+ attrs[SERVER_ADDRESS] = invocation.serverAddress;
306
+ }
307
+ if (invocation.serverPort != null) {
308
+ attrs[SERVER_PORT] = invocation.serverPort;
309
+ }
310
+ Object.assign(attrs, getRetrievalDocumentsAttributes(invocation.documents));
311
+ if (invocation.attributes) {
312
+ Object.assign(attrs, invocation.attributes);
313
+ }
314
+ span.setAttributes(attrs);
315
+ }
316
+ // ==================== Rerank ====================
317
+ function getRerankDocumentsAttributes(inputDocuments, outputDocuments) {
318
+ if (!isExperimentalMode() || !shouldCaptureContentInSpan())
319
+ return {};
320
+ const attrs = {};
321
+ if (inputDocuments != null) {
322
+ attrs[GEN_AI_RERANK_INPUT_DOCUMENTS] =
323
+ typeof inputDocuments === "string"
324
+ ? inputDocuments
325
+ : genAiJsonDumps(inputDocuments);
326
+ }
327
+ if (outputDocuments != null) {
328
+ attrs[GEN_AI_RERANK_OUTPUT_DOCUMENTS] =
329
+ typeof outputDocuments === "string"
330
+ ? outputDocuments
331
+ : genAiJsonDumps(outputDocuments);
332
+ }
333
+ return attrs;
334
+ }
335
+ export function applyRerankFinishAttributes(span, invocation) {
336
+ span.updateName(`${GenAiExtendedOperationNameValues.RERANK_DOCUMENTS} ${invocation.requestModel ?? ""}`.trim());
337
+ const attrs = {
338
+ [GEN_AI_OPERATION_NAME]: GenAiExtendedOperationNameValues.RERANK_DOCUMENTS,
339
+ [GEN_AI_SPAN_KIND]: GenAiSpanKindValues.RERANKER,
340
+ };
341
+ if (invocation.provider) {
342
+ attrs[GEN_AI_PROVIDER_NAME] = invocation.provider;
343
+ }
344
+ if (invocation.requestModel != null) {
345
+ attrs[GEN_AI_REQUEST_MODEL] = invocation.requestModel;
346
+ }
347
+ if (invocation.topK != null) {
348
+ attrs[GEN_AI_REQUEST_TOP_K] = invocation.topK;
349
+ }
350
+ if (invocation.documentsCount != null) {
351
+ attrs[GEN_AI_RERANK_DOCUMENTS_COUNT] = invocation.documentsCount;
352
+ }
353
+ if (invocation.temperature != null) {
354
+ attrs[GEN_AI_REQUEST_TEMPERATURE] = invocation.temperature;
355
+ }
356
+ if (invocation.maxTokens != null) {
357
+ attrs[GEN_AI_REQUEST_MAX_TOKENS] = invocation.maxTokens;
358
+ }
359
+ if (invocation.scoringPrompt != null) {
360
+ attrs[GEN_AI_RERANK_SCORING_PROMPT] = invocation.scoringPrompt;
361
+ }
362
+ if (invocation.returnDocuments != null) {
363
+ attrs[GEN_AI_RERANK_RETURN_DOCUMENTS] = invocation.returnDocuments;
364
+ }
365
+ if (invocation.maxChunksPerDoc != null) {
366
+ attrs[GEN_AI_RERANK_MAX_CHUNKS_PER_DOC] = invocation.maxChunksPerDoc;
367
+ }
368
+ if (invocation.device != null) {
369
+ attrs[GEN_AI_RERANK_DEVICE] = invocation.device;
370
+ }
371
+ if (invocation.batchSize != null) {
372
+ attrs[GEN_AI_RERANK_BATCH_SIZE] = invocation.batchSize;
373
+ }
374
+ if (invocation.maxLength != null) {
375
+ attrs[GEN_AI_RERANK_MAX_LENGTH] = invocation.maxLength;
376
+ }
377
+ if (invocation.normalize != null) {
378
+ attrs[GEN_AI_RERANK_NORMALIZE] = invocation.normalize;
379
+ }
380
+ Object.assign(attrs, getRerankDocumentsAttributes(invocation.inputDocuments, invocation.outputDocuments));
381
+ if (invocation.attributes) {
382
+ Object.assign(attrs, invocation.attributes);
383
+ }
384
+ span.setAttributes(attrs);
385
+ }
386
+ // ==================== Entry ====================
387
+ export function applyEntryFinishAttributes(span, invocation) {
388
+ span.updateName("enter_ai_application_system");
389
+ const attrs = {
390
+ [GEN_AI_OPERATION_NAME]: GenAiExtendedOperationNameValues.ENTER,
391
+ [GEN_AI_SPAN_KIND]: GenAiSpanKindValues.ENTRY,
392
+ };
393
+ if (invocation.sessionId != null) {
394
+ attrs[GEN_AI_SESSION_ID] = invocation.sessionId;
395
+ }
396
+ if (invocation.userId != null) {
397
+ attrs[GEN_AI_USER_ID] = invocation.userId;
398
+ }
399
+ if (invocation.responseTimeToFirstToken != null) {
400
+ attrs[GEN_AI_RESPONSE_TIME_TO_FIRST_TOKEN] =
401
+ invocation.responseTimeToFirstToken;
402
+ }
403
+ Object.assign(attrs, getLlmMessagesAttributesForSpan(invocation.inputMessages ?? [], invocation.outputMessages ?? []));
404
+ if (invocation.attributes) {
405
+ Object.assign(attrs, invocation.attributes);
406
+ }
407
+ span.setAttributes(attrs);
408
+ }
409
+ // ==================== ReAct Step ====================
410
+ export function applyReactStepFinishAttributes(span, invocation) {
411
+ span.updateName("react step");
412
+ const attrs = {
413
+ [GEN_AI_OPERATION_NAME]: GenAiExtendedOperationNameValues.REACT,
414
+ [GEN_AI_SPAN_KIND]: GenAiSpanKindValues.STEP,
415
+ };
416
+ if (invocation.finishReason != null) {
417
+ attrs[GEN_AI_REACT_FINISH_REASON] = invocation.finishReason;
418
+ }
419
+ if (invocation.round != null) {
420
+ attrs[GEN_AI_REACT_ROUND] = invocation.round;
421
+ }
422
+ if (invocation.attributes) {
423
+ Object.assign(attrs, invocation.attributes);
424
+ }
425
+ span.setAttributes(attrs);
426
+ }
427
+ //# sourceMappingURL=extended-span-utils.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"extended-span-utils.js","sourceRoot":"","sources":["../src/extended-span-utils.ts"],"names":[],"mappings":"AAAA,sCAAsC;AACtC,EAAE;AACF,kEAAkE;AAClE,mEAAmE;AACnE,0CAA0C;AAC1C,EAAE;AACF,iDAAiD;AACjD,EAAE;AACF,sEAAsE;AACtE,oEAAoE;AACpE,2EAA2E;AAC3E,sEAAsE;AACtE,iCAAiC;AAIjC,OAAO,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAC;AAYlD,OAAO,EACL,qBAAqB,EACrB,gBAAgB,EAChB,oBAAoB,EACpB,oBAAoB,EACpB,0BAA0B,EAC1B,oBAAoB,EACpB,oBAAoB,EACpB,gCAAgC,EAChC,+BAA+B,EAC/B,yBAAyB,EACzB,mBAAmB,EACnB,6BAA6B,EAC7B,+BAA+B,EAC/B,2BAA2B,EAC3B,8BAA8B,EAC9B,qBAAqB,EACrB,kBAAkB,EAClB,mCAAmC,EACnC,yBAAyB,EACzB,0BAA0B,EAC1B,yBAAyB,EACzB,wCAAwC,EACxC,oCAAoC,EACpC,kBAAkB,EAClB,sBAAsB,EACtB,eAAe,EACf,iBAAiB,EACjB,wBAAwB,EACxB,qBAAqB,EACrB,mBAAmB,EACnB,gBAAgB,EAChB,uBAAuB,EACvB,gBAAgB,EAChB,0BAA0B,EAC1B,uBAAuB,EACvB,iCAAiC,EACjC,2BAA2B,EAC3B,0BAA0B,EAC1B,6BAA6B,EAC7B,4BAA4B,EAC5B,8BAA8B,EAC9B,gCAAgC,EAChC,oBAAoB,EACpB,wBAAwB,EACxB,wBAAwB,EACxB,uBAAuB,EACvB,6BAA6B,EAC7B,8BAA8B,EAC9B,iBAAiB,EACjB,cAAc,EACd,0BAA0B,EAC1B,kBAAkB,EAClB,cAAc,EACd,WAAW,EACX,UAAU,EACV,mBAAmB,EACnB,wBAAwB,EACxB,gCAAgC,GACjC,MAAM,yCAAyC,CAAC;AACjD,OAAO,EACL,+BAA+B,EAC/B,yBAAyB,GAE1B,MAAM,iBAAiB,CAAC;AACzB,OAAO,EACL,kBAAkB,EAClB,uBAAuB,EACvB,eAAe,EACf,cAAc,EACd,0BAA0B,GAC3B,MAAM,YAAY,CAAC;AAEpB,sDAAsD;AAEtD,MAAM,UAAU,8BAA8B,CAC5C,IAAU,EACV,UAA+B;IAE/B,IAAI,CAAC,UAAU,CACb,GAAG,wBAAwB,CAAC,UAAU,IAAI,UAAU,CAAC,YAAY,EAAE,CAAC,IAAI,EAAE,CAC3E,CAAC;IAEF,MAAM,KAAK,GAA4B;QACrC,CAAC,qBAAqB,CAAC,EAAE,wBAAwB,CAAC,UAAU;QAC5D,CAAC,gBAAgB,CAAC,EAAE,mBAAmB,CAAC,SAAS;KAClD,CAAC;IAEF,IAAI,UAAU,CAAC,YAAY,EAAE,CAAC;QAC5B,KAAK,CAAC,oBAAoB,CAAC,GAAG,UAAU,CAAC,YAAY,CAAC;IACxD,CAAC;IACD,IAAI,UAAU,CAAC,QAAQ,IAAI,IAAI,EAAE,CAAC;QAChC,KAAK,CAAC,oBAAoB,CAAC,GAAG,UAAU,CAAC,QAAQ,CAAC;IACpD,CAAC;IACD,IAAI,UAAU,CAAC,UAAU,IAAI,IAAI,EAAE,CAAC;QAClC,KAAK,CAAC,WAAW,CAAC,GAAG,UAAU,CAAC,UAAU,CAAC;IAC7C,CAAC;IACD,IAAI,UAAU,CAAC,cAAc,IAAI,IAAI,EAAE,CAAC;QACtC,KAAK,CAAC,iCAAiC,CAAC,GAAG,UAAU,CAAC,cAAc,CAAC;IACvE,CAAC;IACD,IAAI,UAAU,CAAC,eAAe,IAAI,IAAI,EAAE,CAAC;QACvC,KAAK,CAAC,+BAA+B,CAAC,GAAG,UAAU,CAAC,eAAe,CAAC;IACtE,CAAC;IACD,IAAI,UAAU,CAAC,WAAW,IAAI,IAAI,EAAE,CAAC;QACnC,KAAK,CAAC,yBAAyB,CAAC,GAAG,UAAU,CAAC,WAAW,CAAC;QAC1D,KAAK,CAAC,yBAAyB,CAAC,GAAG,UAAU,CAAC,WAAW,CAAC;IAC5D,CAAC;IACD,IAAI,UAAU,CAAC,aAAa,IAAI,IAAI,EAAE,CAAC;QACrC,KAAK,CAAC,cAAc,CAAC,GAAG,UAAU,CAAC,aAAa,CAAC;IACnD,CAAC;IACD,IAAI,UAAU,CAAC,UAAU,EAAE,CAAC;QAC1B,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,UAAU,CAAC,UAAU,CAAC,CAAC;IAC9C,CAAC;IAED,IAAI,CAAC,aAAa,CAAC,KAAkD,CAAC,CAAC;AACzE,CAAC;AAED,yDAAyD;AAEzD,MAAM,UAAU,gCAAgC,CAC9C,IAAU,EACV,UAAiC;IAEjC,IAAI,CAAC,UAAU,CACb,GAAG,wBAAwB,CAAC,YAAY,IAAI,UAAU,CAAC,SAAS,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,CAChF,CAAC;IAEF,MAAM,KAAK,GAA4B;QACrC,CAAC,qBAAqB,CAAC,EAAE,wBAAwB,CAAC,YAAY;QAC9D,CAAC,gBAAgB,CAAC,EAAE,mBAAmB,CAAC,KAAK;KAC9C,CAAC;IAEF,IAAI,UAAU,CAAC,QAAQ,EAAE,CAAC;QACxB,KAAK,CAAC,oBAAoB,CAAC,GAAG,UAAU,CAAC,QAAQ,CAAC;IACpD,CAAC;IACD,IAAI,UAAU,CAAC,gBAAgB,IAAI,IAAI,EAAE,CAAC;QACxC,KAAK,CAAC,wBAAwB,CAAC,GAAG,UAAU,CAAC,gBAAgB,CAAC;IAChE,CAAC;IACD,IAAI,UAAU,CAAC,OAAO,IAAI,IAAI,EAAE,CAAC;QAC/B,KAAK,CAAC,eAAe,CAAC,GAAG,UAAU,CAAC,OAAO,CAAC;IAC9C,CAAC;IACD,IAAI,UAAU,CAAC,SAAS,IAAI,IAAI,EAAE,CAAC;QACjC,KAAK,CAAC,iBAAiB,CAAC,GAAG,UAAU,CAAC,SAAS,CAAC;IAClD,CAAC;IACD,IAAI,UAAU,CAAC,YAAY,IAAI,IAAI,EAAE,CAAC;QACpC,KAAK,CAAC,oBAAoB,CAAC,GAAG,UAAU,CAAC,YAAY,CAAC;IACxD,CAAC;IACD,IAAI,UAAU,CAAC,UAAU,IAAI,IAAI,EAAE,CAAC;QAClC,KAAK,CAAC,WAAW,CAAC,GAAG,UAAU,CAAC,UAAU,CAAC;IAC7C,CAAC;IACD,IAAI,UAAU,CAAC,aAAa,IAAI,IAAI,EAAE,CAAC;QACrC,KAAK,CAAC,cAAc,CAAC,GAAG,UAAU,CAAC,aAAa,CAAC;IACnD,CAAC;IACD,IAAI,UAAU,CAAC,UAAU,EAAE,CAAC;QAC1B,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,UAAU,CAAC,UAAU,CAAC,CAAC;IAC9C,CAAC;IAED,IAAI,CAAC,aAAa,CAAC,KAAkD,CAAC,CAAC;AACzE,CAAC;AAED,yDAAyD;AAEzD,SAAS,yBAAyB,CAChC,iBAA0B,EAC1B,cAAuB;IAEvB,IAAI,CAAC,kBAAkB,EAAE,IAAI,CAAC,0BAA0B,EAAE;QAAE,OAAO,EAAE,CAAC;IAEtE,MAAM,KAAK,GAA4B,EAAE,CAAC;IAC1C,IAAI,iBAAiB,IAAI,IAAI,EAAE,CAAC;QAC9B,KAAK,CAAC,0BAA0B,CAAC;YAC/B,OAAO,iBAAiB,KAAK,QAAQ;gBACnC,CAAC,CAAC,iBAAiB;gBACnB,CAAC,CAAC,cAAc,CAAC,iBAAiB,CAAC,CAAC;IAC1C,CAAC;IACD,IAAI,cAAc,IAAI,IAAI,EAAE,CAAC;QAC3B,KAAK,CAAC,uBAAuB,CAAC;YAC5B,OAAO,cAAc,KAAK,QAAQ;gBAChC,CAAC,CAAC,cAAc;gBAChB,CAAC,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC;IACvC,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,gCAAgC,CAC9C,IAAU,EACV,UAAiC;IAEjC,IAAI,CAAC,UAAU,CACb,GAAG,wBAAwB,CAAC,YAAY,IAAI,UAAU,CAAC,QAAQ,EAAE,CAAC,IAAI,EAAE,CACzE,CAAC;IAEF,MAAM,KAAK,GAA4B;QACrC,CAAC,qBAAqB,CAAC,EAAE,wBAAwB,CAAC,YAAY;QAC9D,CAAC,gBAAgB,CAAC,EAAE,mBAAmB,CAAC,IAAI;KAC7C,CAAC;IAEF,IAAI,UAAU,CAAC,UAAU,IAAI,IAAI,EAAE,CAAC;QAClC,KAAK,CAAC,mBAAmB,CAAC,GAAG,UAAU,CAAC,UAAU,CAAC;IACrD,CAAC;IACD,IAAI,UAAU,CAAC,eAAe,IAAI,IAAI,EAAE,CAAC;QACvC,KAAK,CAAC,uBAAuB,CAAC,GAAG,UAAU,CAAC,eAAe,CAAC;IAC9D,CAAC;IACD,IAAI,UAAU,CAAC,QAAQ,EAAE,CAAC;QACxB,KAAK,CAAC,gBAAgB,CAAC,GAAG,UAAU,CAAC,QAAQ,CAAC;IAChD,CAAC;IACD,IAAI,UAAU,CAAC,QAAQ,IAAI,IAAI,EAAE,CAAC;QAChC,KAAK,CAAC,gBAAgB,CAAC,GAAG,UAAU,CAAC,QAAQ,CAAC;IAChD,CAAC;IAED,MAAM,CAAC,MAAM,CACX,KAAK,EACL,yBAAyB,CACvB,UAAU,CAAC,iBAAiB,EAC5B,UAAU,CAAC,cAAc,CAC1B,CACF,CAAC;IACF,IAAI,UAAU,CAAC,UAAU,EAAE,CAAC;QAC1B,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,UAAU,CAAC,UAAU,CAAC,CAAC;IAC9C,CAAC;IAED,IAAI,CAAC,aAAa,CAAC,KAAkD,CAAC,CAAC;AACzE,CAAC;AAED,yDAAyD;AAEzD,SAAS,sBAAsB,CAAC,UAAiC;IAC/D,IAAI,UAAU,CAAC,SAAS,EAAE,CAAC;QACzB,OAAO,GAAG,wBAAwB,CAAC,YAAY,IAAI,UAAU,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,CAAC;IACnF,CAAC;IACD,OAAO,wBAAwB,CAAC,YAAY,CAAC;AAC/C,CAAC;AAED,MAAM,UAAU,gCAAgC,CAC9C,IAAU,EACV,UAAiC;IAEjC,IAAI,CAAC,UAAU,CAAC,sBAAsB,CAAC,UAAU,CAAC,CAAC,CAAC;IAEpD,MAAM,KAAK,GAA4B;QACrC,CAAC,qBAAqB,CAAC,EAAE,wBAAwB,CAAC,YAAY;QAC9D,CAAC,gBAAgB,CAAC,EAAE,mBAAmB,CAAC,KAAK;KAC9C,CAAC;IAEF,IAAI,UAAU,CAAC,QAAQ,EAAE,CAAC;QACxB,KAAK,CAAC,oBAAoB,CAAC,GAAG,UAAU,CAAC,QAAQ,CAAC;IACpD,CAAC;IACD,IAAI,UAAU,CAAC,gBAAgB,IAAI,IAAI,EAAE,CAAC;QACxC,KAAK,CAAC,wBAAwB,CAAC,GAAG,UAAU,CAAC,gBAAgB,CAAC;IAChE,CAAC;IACD,IAAI,UAAU,CAAC,OAAO,IAAI,IAAI,EAAE,CAAC;QAC/B,KAAK,CAAC,eAAe,CAAC,GAAG,UAAU,CAAC,OAAO,CAAC;IAC9C,CAAC;IACD,IAAI,UAAU,CAAC,SAAS,IAAI,IAAI,EAAE,CAAC;QACjC,KAAK,CAAC,iBAAiB,CAAC,GAAG,UAAU,CAAC,SAAS,CAAC;IAClD,CAAC;IACD,IAAI,UAAU,CAAC,cAAc,IAAI,IAAI,EAAE,CAAC;QACtC,KAAK,CAAC,sBAAsB,CAAC,GAAG,UAAU,CAAC,cAAc,CAAC;IAC5D,CAAC;IACD,IAAI,UAAU,CAAC,YAAY,IAAI,IAAI,EAAE,CAAC;QACpC,KAAK,CAAC,qBAAqB,CAAC,GAAG,UAAU,CAAC,YAAY,CAAC;IACzD,CAAC;IACD,IAAI,UAAU,CAAC,YAAY,IAAI,IAAI,EAAE,CAAC;QACpC,KAAK,CAAC,oBAAoB,CAAC,GAAG,UAAU,CAAC,YAAY,CAAC;IACxD,CAAC;IAED,qBAAqB;IACrB,IAAI,UAAU,CAAC,WAAW,IAAI,IAAI,EAAE,CAAC;QACnC,KAAK,CAAC,0BAA0B,CAAC,GAAG,UAAU,CAAC,WAAW,CAAC;IAC7D,CAAC;IACD,IAAI,UAAU,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC;QAC5B,KAAK,CAAC,oBAAoB,CAAC,GAAG,UAAU,CAAC,IAAI,CAAC;IAChD,CAAC;IACD,IAAI,UAAU,CAAC,gBAAgB,IAAI,IAAI,EAAE,CAAC;QACxC,KAAK,CAAC,gCAAgC,CAAC,GAAG,UAAU,CAAC,gBAAgB,CAAC;IACxE,CAAC;IACD,IAAI,UAAU,CAAC,eAAe,IAAI,IAAI,EAAE,CAAC;QACvC,KAAK,CAAC,+BAA+B,CAAC,GAAG,UAAU,CAAC,eAAe,CAAC;IACtE,CAAC;IACD,IAAI,UAAU,CAAC,SAAS,IAAI,IAAI,EAAE,CAAC;QACjC,KAAK,CAAC,yBAAyB,CAAC,GAAG,UAAU,CAAC,SAAS,CAAC;IAC1D,CAAC;IACD,IAAI,UAAU,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC;QAC5B,KAAK,CAAC,mBAAmB,CAAC,GAAG,UAAU,CAAC,IAAI,CAAC;IAC/C,CAAC;IACD,IAAI,UAAU,CAAC,aAAa,IAAI,IAAI,EAAE,CAAC;QACrC,KAAK,CAAC,6BAA6B,CAAC,GAAG,UAAU,CAAC,aAAa,CAAC;IAClE,CAAC;IAED,sBAAsB;IACtB,IAAI,UAAU,CAAC,aAAa,IAAI,IAAI,EAAE,CAAC;QACrC,KAAK,CAAC,8BAA8B,CAAC,GAAG,UAAU,CAAC,aAAa,CAAC;IACnE,CAAC;IACD,IAAI,UAAU,CAAC,UAAU,IAAI,IAAI,EAAE,CAAC;QAClC,KAAK,CAAC,kBAAkB,CAAC,GAAG,UAAU,CAAC,UAAU,CAAC;IACpD,CAAC;IACD,IAAI,UAAU,CAAC,iBAAiB,IAAI,IAAI,EAAE,CAAC;QACzC,KAAK,CAAC,qBAAqB,CAAC,GAAG,UAAU,CAAC,iBAAiB,CAAC;IAC9D,CAAC;IACD,IAAI,UAAU,CAAC,WAAW,IAAI,IAAI,EAAE,CAAC;QACnC,KAAK,CAAC,yBAAyB,CAAC,GAAG,UAAU,CAAC,WAAW,CAAC;IAC5D,CAAC;IACD,IAAI,UAAU,CAAC,YAAY,IAAI,IAAI,EAAE,CAAC;QACpC,KAAK,CAAC,0BAA0B,CAAC,GAAG,UAAU,CAAC,YAAY,CAAC;IAC9D,CAAC;IACD,IAAI,UAAU,CAAC,6BAA6B,IAAI,IAAI,EAAE,CAAC;QACrD,KAAK,CAAC,wCAAwC,CAAC;YAC7C,UAAU,CAAC,6BAA6B,CAAC;IAC7C,CAAC;IACD,IAAI,UAAU,CAAC,yBAAyB,IAAI,IAAI,EAAE,CAAC;QACjD,KAAK,CAAC,oCAAoC,CAAC;YACzC,UAAU,CAAC,yBAAyB,CAAC;IACzC,CAAC;IAED,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,IAAI,UAAU,CAAC,WAAW,IAAI,IAAI;QAAE,WAAW,IAAI,UAAU,CAAC,WAAW,CAAC;IAC1E,IAAI,UAAU,CAAC,YAAY,IAAI,IAAI;QAAE,WAAW,IAAI,UAAU,CAAC,YAAY,CAAC;IAC5E,IAAI,WAAW,GAAG,CAAC,EAAE,CAAC;QACpB,KAAK,CAAC,yBAAyB,CAAC,GAAG,WAAW,CAAC;IACjD,CAAC;IAED,IACE,UAAU,CAAC,oBAAoB,IAAI,IAAI;QACvC,UAAU,CAAC,eAAe,IAAI,IAAI;QAClC,UAAU,CAAC,oBAAoB,IAAI,UAAU,CAAC,eAAe,EAC7D,CAAC;QACD,KAAK,CAAC,mCAAmC,CAAC,GAAG,IAAI,CAAC,KAAK,CACrD,CAAC,UAAU,CAAC,oBAAoB,GAAG,UAAU,CAAC,eAAe,CAAC,GAAG,GAAG,CACrE,CAAC;IACJ,CAAC;IAED,sCAAsC;IACtC,IAAI,UAAU,CAAC,UAAU,IAAI,IAAI,EAAE,CAAC;QAClC,KAAK,CAAC,kBAAkB,CAAC,GAAG,UAAU,CAAC,UAAU,CAAC;IACpD,CAAC;IACD,IAAI,UAAU,CAAC,WAAW,IAAI,IAAI,IAAI,UAAU,CAAC,WAAW,KAAK,CAAC,EAAE,CAAC;QACnE,KAAK,CAAC,2BAA2B,CAAC,GAAG,UAAU,CAAC,WAAW,CAAC;IAC9D,CAAC;IACD,IAAI,UAAU,CAAC,UAAU,IAAI,IAAI,EAAE,CAAC;QAClC,KAAK,CAAC,WAAW,CAAC,GAAG,UAAU,CAAC,UAAU,CAAC;IAC7C,CAAC;IACD,IAAI,UAAU,CAAC,aAAa,IAAI,IAAI,EAAE,CAAC;QACrC,KAAK,CAAC,cAAc,CAAC,GAAG,UAAU,CAAC,aAAa,CAAC;IACnD,CAAC;IAED,MAAM,CAAC,MAAM,CACX,KAAK,EACL,+BAA+B,CAC7B,UAAU,CAAC,aAAa,IAAI,EAAE,EAC9B,UAAU,CAAC,cAAc,IAAI,EAAE,EAC/B,UAAU,CAAC,iBAAiB,CAC7B,CACF,CAAC;IACF,MAAM,CAAC,MAAM,CACX,KAAK,EACL,yBAAyB,CAAC,UAAU,CAAC,eAAe,CAAC,CACtD,CAAC;IACF,IAAI,UAAU,CAAC,UAAU,EAAE,CAAC;QAC1B,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,UAAU,CAAC,UAAU,CAAC,CAAC;IAC9C,CAAC;IAED,IAAI,CAAC,aAAa,CAAC,KAAkD,CAAC,CAAC;AACzE,CAAC;AAED,MAAM,UAAU,yBAAyB,CACvC,MAAsC,EACtC,IAAU,EACV,UAAiC,EACjC,KAAyB;IAEzB,IAAI,CAAC,kBAAkB,EAAE,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,MAAM;QAAE,OAAO;IAEnE,MAAM,KAAK,GAA4B;QACrC,CAAC,qBAAqB,CAAC,EAAE,wBAAwB,CAAC,YAAY;QAC9D,CAAC,gBAAgB,CAAC,EAAE,mBAAmB,CAAC,KAAK;KAC9C,CAAC;IACF,IAAI,UAAU,CAAC,QAAQ,EAAE,CAAC;QACxB,KAAK,CAAC,oBAAoB,CAAC,GAAG,UAAU,CAAC,QAAQ,CAAC;IACpD,CAAC;IACD,IAAI,UAAU,CAAC,SAAS,IAAI,IAAI,EAAE,CAAC;QACjC,KAAK,CAAC,iBAAiB,CAAC,GAAG,UAAU,CAAC,SAAS,CAAC;IAClD,CAAC;IAED,IAAI,KAAK,EAAE,CAAC;QACV,KAAK,CAAC,UAAU,CAAC,GAAG,KAAK,CAAC,IAAI,CAAC;IACjC,CAAC;IAED,MAAM,CAAC,IAAI,CAAC;QACV,IAAI,EAAE,8CAA8C;QACpD,UAAU,EAAE,KAAK;KAClB,CAAC,CAAC;AACL,CAAC;AAED,sDAAsD;AAEtD,SAAS,+BAA+B,CACtC,SAA0C;IAE1C,IAAI,CAAC,kBAAkB,EAAE,IAAI,CAAC,SAAS,EAAE,MAAM;QAAE,OAAO,EAAE,CAAC;IAE3D,MAAM,IAAI,GAAG,uBAAuB,EAAE,CAAC;IACvC,MAAM,gBAAgB,GACpB,IAAI,KAAK,oBAAoB,CAAC,SAAS;QACvC,IAAI,KAAK,oBAAoB,CAAC,cAAc,CAAC;IAE/C,MAAM,KAAK,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;QAClC,IAAI,gBAAgB,EAAE,CAAC;YACrB,OAAO,EAAE,GAAG,GAAG,EAAE,CAAC;QACpB,CAAC;QACD,OAAO,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,KAAK,EAAE,GAAG,CAAC,KAAK,EAAE,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,OAAO,EAAE,CAAC,0BAA0B,CAAC,EAAE,cAAc,CAAC,KAAK,CAAC,EAAE,CAAC;AACjE,CAAC;AAED,MAAM,UAAU,8BAA8B,CAC5C,IAAU,EACV,UAA+B;IAE/B,MAAM,MAAM,GAAG,gCAAgC,CAAC,SAAS,CAAC;IAC1D,MAAM,QAAQ,GAAG,UAAU,CAAC,YAAY;QACtC,CAAC,CAAC,GAAG,MAAM,IAAI,UAAU,CAAC,YAAY,EAAE,CAAC,IAAI,EAAE;QAC/C,CAAC,CAAC,MAAM,CAAC;IACX,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;IAE1B,MAAM,KAAK,GAA4B;QACrC,CAAC,qBAAqB,CAAC,EAAE,MAAM;QAC/B,CAAC,gBAAgB,CAAC,EAAE,mBAAmB,CAAC,SAAS;KAClD,CAAC;IAEF,IAAI,UAAU,CAAC,YAAY,IAAI,IAAI,EAAE,CAAC;QACpC,KAAK,CAAC,qBAAqB,CAAC,GAAG,UAAU,CAAC,YAAY,CAAC;IACzD,CAAC;IACD,IAAI,UAAU,CAAC,QAAQ,IAAI,IAAI,EAAE,CAAC;QAChC,KAAK,CAAC,oBAAoB,CAAC,GAAG,UAAU,CAAC,QAAQ,CAAC;IACpD,CAAC;IACD,IAAI,UAAU,CAAC,YAAY,IAAI,IAAI,EAAE,CAAC;QACpC,KAAK,CAAC,oBAAoB,CAAC,GAAG,UAAU,CAAC,YAAY,CAAC;IACxD,CAAC;IACD,IAAI,UAAU,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC;QAC5B,KAAK,CAAC,oBAAoB,CAAC,GAAG,UAAU,CAAC,IAAI,CAAC;IAChD,CAAC;IACD,IACE,UAAU,CAAC,KAAK,IAAI,IAAI;QACxB,kBAAkB,EAAE;QACpB,0BAA0B,EAAE,EAC5B,CAAC;QACD,KAAK,CAAC,2BAA2B,CAAC,GAAG,UAAU,CAAC,KAAK,CAAC;IACxD,CAAC;IACD,IAAI,UAAU,CAAC,aAAa,IAAI,IAAI,EAAE,CAAC;QACrC,KAAK,CAAC,cAAc,CAAC,GAAG,UAAU,CAAC,aAAa,CAAC;IACnD,CAAC;IACD,IAAI,UAAU,CAAC,UAAU,IAAI,IAAI,EAAE,CAAC;QAClC,KAAK,CAAC,WAAW,CAAC,GAAG,UAAU,CAAC,UAAU,CAAC;IAC7C,CAAC;IAED,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,+BAA+B,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC;IAC5E,IAAI,UAAU,CAAC,UAAU,EAAE,CAAC;QAC1B,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,UAAU,CAAC,UAAU,CAAC,CAAC;IAC9C,CAAC;IAED,IAAI,CAAC,aAAa,CAAC,KAAkD,CAAC,CAAC;AACzE,CAAC;AAED,mDAAmD;AAEnD,SAAS,4BAA4B,CACnC,cAAuB,EACvB,eAAwB;IAExB,IAAI,CAAC,kBAAkB,EAAE,IAAI,CAAC,0BAA0B,EAAE;QAAE,OAAO,EAAE,CAAC;IAEtE,MAAM,KAAK,GAA4B,EAAE,CAAC;IAC1C,IAAI,cAAc,IAAI,IAAI,EAAE,CAAC;QAC3B,KAAK,CAAC,6BAA6B,CAAC;YAClC,OAAO,cAAc,KAAK,QAAQ;gBAChC,CAAC,CAAC,cAAc;gBAChB,CAAC,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC;IACvC,CAAC;IACD,IAAI,eAAe,IAAI,IAAI,EAAE,CAAC;QAC5B,KAAK,CAAC,8BAA8B,CAAC;YACnC,OAAO,eAAe,KAAK,QAAQ;gBACjC,CAAC,CAAC,eAAe;gBACjB,CAAC,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC;IACxC,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,MAAM,UAAU,2BAA2B,CACzC,IAAU,EACV,UAA4B;IAE5B,IAAI,CAAC,UAAU,CACb,GAAG,gCAAgC,CAAC,gBAAgB,IAAI,UAAU,CAAC,YAAY,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,CAC/F,CAAC;IAEF,MAAM,KAAK,GAA4B;QACrC,CAAC,qBAAqB,CAAC,EAAE,gCAAgC,CAAC,gBAAgB;QAC1E,CAAC,gBAAgB,CAAC,EAAE,mBAAmB,CAAC,QAAQ;KACjD,CAAC;IAEF,IAAI,UAAU,CAAC,QAAQ,EAAE,CAAC;QACxB,KAAK,CAAC,oBAAoB,CAAC,GAAG,UAAU,CAAC,QAAQ,CAAC;IACpD,CAAC;IACD,IAAI,UAAU,CAAC,YAAY,IAAI,IAAI,EAAE,CAAC;QACpC,KAAK,CAAC,oBAAoB,CAAC,GAAG,UAAU,CAAC,YAAY,CAAC;IACxD,CAAC;IACD,IAAI,UAAU,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC;QAC5B,KAAK,CAAC,oBAAoB,CAAC,GAAG,UAAU,CAAC,IAAI,CAAC;IAChD,CAAC;IACD,IAAI,UAAU,CAAC,cAAc,IAAI,IAAI,EAAE,CAAC;QACtC,KAAK,CAAC,6BAA6B,CAAC,GAAG,UAAU,CAAC,cAAc,CAAC;IACnE,CAAC;IACD,IAAI,UAAU,CAAC,WAAW,IAAI,IAAI,EAAE,CAAC;QACnC,KAAK,CAAC,0BAA0B,CAAC,GAAG,UAAU,CAAC,WAAW,CAAC;IAC7D,CAAC;IACD,IAAI,UAAU,CAAC,SAAS,IAAI,IAAI,EAAE,CAAC;QACjC,KAAK,CAAC,yBAAyB,CAAC,GAAG,UAAU,CAAC,SAAS,CAAC;IAC1D,CAAC;IACD,IAAI,UAAU,CAAC,aAAa,IAAI,IAAI,EAAE,CAAC;QACrC,KAAK,CAAC,4BAA4B,CAAC,GAAG,UAAU,CAAC,aAAa,CAAC;IACjE,CAAC;IACD,IAAI,UAAU,CAAC,eAAe,IAAI,IAAI,EAAE,CAAC;QACvC,KAAK,CAAC,8BAA8B,CAAC,GAAG,UAAU,CAAC,eAAe,CAAC;IACrE,CAAC;IACD,IAAI,UAAU,CAAC,eAAe,IAAI,IAAI,EAAE,CAAC;QACvC,KAAK,CAAC,gCAAgC,CAAC,GAAG,UAAU,CAAC,eAAe,CAAC;IACvE,CAAC;IACD,IAAI,UAAU,CAAC,MAAM,IAAI,IAAI,EAAE,CAAC;QAC9B,KAAK,CAAC,oBAAoB,CAAC,GAAG,UAAU,CAAC,MAAM,CAAC;IAClD,CAAC;IACD,IAAI,UAAU,CAAC,SAAS,IAAI,IAAI,EAAE,CAAC;QACjC,KAAK,CAAC,wBAAwB,CAAC,GAAG,UAAU,CAAC,SAAS,CAAC;IACzD,CAAC;IACD,IAAI,UAAU,CAAC,SAAS,IAAI,IAAI,EAAE,CAAC;QACjC,KAAK,CAAC,wBAAwB,CAAC,GAAG,UAAU,CAAC,SAAS,CAAC;IACzD,CAAC;IACD,IAAI,UAAU,CAAC,SAAS,IAAI,IAAI,EAAE,CAAC;QACjC,KAAK,CAAC,uBAAuB,CAAC,GAAG,UAAU,CAAC,SAAS,CAAC;IACxD,CAAC;IAED,MAAM,CAAC,MAAM,CACX,KAAK,EACL,4BAA4B,CAC1B,UAAU,CAAC,cAAc,EACzB,UAAU,CAAC,eAAe,CAC3B,CACF,CAAC;IACF,IAAI,UAAU,CAAC,UAAU,EAAE,CAAC;QAC1B,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,UAAU,CAAC,UAAU,CAAC,CAAC;IAC9C,CAAC;IAED,IAAI,CAAC,aAAa,CAAC,KAAkD,CAAC,CAAC;AACzE,CAAC;AAED,kDAAkD;AAElD,MAAM,UAAU,0BAA0B,CACxC,IAAU,EACV,UAA2B;IAE3B,IAAI,CAAC,UAAU,CAAC,6BAA6B,CAAC,CAAC;IAE/C,MAAM,KAAK,GAA4B;QACrC,CAAC,qBAAqB,CAAC,EAAE,gCAAgC,CAAC,KAAK;QAC/D,CAAC,gBAAgB,CAAC,EAAE,mBAAmB,CAAC,KAAK;KAC9C,CAAC;IAEF,IAAI,UAAU,CAAC,SAAS,IAAI,IAAI,EAAE,CAAC;QACjC,KAAK,CAAC,iBAAiB,CAAC,GAAG,UAAU,CAAC,SAAS,CAAC;IAClD,CAAC;IACD,IAAI,UAAU,CAAC,MAAM,IAAI,IAAI,EAAE,CAAC;QAC9B,KAAK,CAAC,cAAc,CAAC,GAAG,UAAU,CAAC,MAAM,CAAC;IAC5C,CAAC;IACD,IAAI,UAAU,CAAC,wBAAwB,IAAI,IAAI,EAAE,CAAC;QAChD,KAAK,CAAC,mCAAmC,CAAC;YACxC,UAAU,CAAC,wBAAwB,CAAC;IACxC,CAAC;IAED,MAAM,CAAC,MAAM,CACX,KAAK,EACL,+BAA+B,CAC7B,UAAU,CAAC,aAAa,IAAI,EAAE,EAC9B,UAAU,CAAC,cAAc,IAAI,EAAE,CAChC,CACF,CAAC;IACF,IAAI,UAAU,CAAC,UAAU,EAAE,CAAC;QAC1B,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,UAAU,CAAC,UAAU,CAAC,CAAC;IAC9C,CAAC;IAED,IAAI,CAAC,aAAa,CAAC,KAAkD,CAAC,CAAC;AACzE,CAAC;AAED,uDAAuD;AAEvD,MAAM,UAAU,8BAA8B,CAC5C,IAAU,EACV,UAA+B;IAE/B,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;IAE9B,MAAM,KAAK,GAA4B;QACrC,CAAC,qBAAqB,CAAC,EAAE,gCAAgC,CAAC,KAAK;QAC/D,CAAC,gBAAgB,CAAC,EAAE,mBAAmB,CAAC,IAAI;KAC7C,CAAC;IAEF,IAAI,UAAU,CAAC,YAAY,IAAI,IAAI,EAAE,CAAC;QACpC,KAAK,CAAC,0BAA0B,CAAC,GAAG,UAAU,CAAC,YAAY,CAAC;IAC9D,CAAC;IACD,IAAI,UAAU,CAAC,KAAK,IAAI,IAAI,EAAE,CAAC;QAC7B,KAAK,CAAC,kBAAkB,CAAC,GAAG,UAAU,CAAC,KAAK,CAAC;IAC/C,CAAC;IACD,IAAI,UAAU,CAAC,UAAU,EAAE,CAAC;QAC1B,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,UAAU,CAAC,UAAU,CAAC,CAAC;IAC9C,CAAC;IAED,IAAI,CAAC,aAAa,CAAC,KAAkD,CAAC,CAAC;AACzE,CAAC"}
@@ -0,0 +1,150 @@
1
+ import type { Span, Context } from "@opentelemetry/api";
2
+ import type { InputMessage, MessagePart, OutputMessage, ToolDefinition } from "./types.js";
3
+ export interface RetrievalDocument {
4
+ id: string | null;
5
+ score: number | null;
6
+ content?: string | null;
7
+ metadata?: Record<string, unknown> | null;
8
+ }
9
+ export interface EmbeddingInvocation {
10
+ requestModel: string;
11
+ contextToken?: Context | null;
12
+ span?: Span | null;
13
+ provider?: string | null;
14
+ responseModelName?: string | null;
15
+ responseId?: string | null;
16
+ inputTokens?: number | null;
17
+ outputTokens?: number | null;
18
+ attributes?: Record<string, unknown>;
19
+ dimensionCount?: number | null;
20
+ encodingFormats?: string[] | null;
21
+ serverAddress?: string | null;
22
+ serverPort?: number | null;
23
+ monotonicStartS?: number | null;
24
+ }
25
+ export declare function createEmbeddingInvocation(requestModel: string, init?: Partial<EmbeddingInvocation>): EmbeddingInvocation;
26
+ export interface ExecuteToolInvocation {
27
+ toolName: string;
28
+ contextToken?: Context | null;
29
+ span?: Span | null;
30
+ provider?: string | null;
31
+ attributes?: Record<string, unknown>;
32
+ toolCallId?: string | null;
33
+ toolDescription?: string | null;
34
+ toolType?: string | null;
35
+ toolCallArguments?: unknown;
36
+ toolCallResult?: unknown;
37
+ monotonicStartS?: number | null;
38
+ }
39
+ export declare function createExecuteToolInvocation(toolName: string, init?: Partial<ExecuteToolInvocation>): ExecuteToolInvocation;
40
+ export interface CreateAgentInvocation {
41
+ provider: string;
42
+ contextToken?: Context | null;
43
+ span?: Span | null;
44
+ agentName?: string | null;
45
+ attributes?: Record<string, unknown>;
46
+ agentId?: string | null;
47
+ agentDescription?: string | null;
48
+ requestModel?: string | null;
49
+ serverAddress?: string | null;
50
+ serverPort?: number | null;
51
+ monotonicStartS?: number | null;
52
+ }
53
+ export declare function createCreateAgentInvocation(provider: string, init?: Partial<CreateAgentInvocation>): CreateAgentInvocation;
54
+ export interface InvokeAgentInvocation {
55
+ provider: string;
56
+ contextToken?: Context | null;
57
+ span?: Span | null;
58
+ agentName?: string | null;
59
+ inputMessages?: InputMessage[];
60
+ outputMessages?: OutputMessage[];
61
+ toolDefinitions?: ToolDefinition[];
62
+ systemInstruction?: MessagePart[];
63
+ attributes?: Record<string, unknown>;
64
+ agentId?: string | null;
65
+ agentDescription?: string | null;
66
+ conversationId?: string | null;
67
+ dataSourceId?: string | null;
68
+ requestModel?: string | null;
69
+ responseModelName?: string | null;
70
+ responseId?: string | null;
71
+ finishReasons?: string[] | null;
72
+ inputTokens?: number | null;
73
+ outputTokens?: number | null;
74
+ usageCacheCreationInputTokens?: number | null;
75
+ usageCacheReadInputTokens?: number | null;
76
+ outputType?: string | null;
77
+ choiceCount?: number | null;
78
+ seed?: number | null;
79
+ frequencyPenalty?: number | null;
80
+ maxTokens?: number | null;
81
+ presencePenalty?: number | null;
82
+ stopSequences?: string[] | null;
83
+ temperature?: number | null;
84
+ topP?: number | null;
85
+ serverAddress?: string | null;
86
+ serverPort?: number | null;
87
+ monotonicStartS?: number | null;
88
+ monotonicEndS?: number | null;
89
+ monotonicFirstTokenS?: number | null;
90
+ }
91
+ export declare function createInvokeAgentInvocation(provider: string, init?: Partial<InvokeAgentInvocation>): InvokeAgentInvocation;
92
+ export interface RetrievalInvocation {
93
+ contextToken?: Context | null;
94
+ span?: Span | null;
95
+ attributes?: Record<string, unknown>;
96
+ query?: string | null;
97
+ documents?: RetrievalDocument[];
98
+ dataSourceId?: string | null;
99
+ provider?: string | null;
100
+ requestModel?: string | null;
101
+ topK?: number | null;
102
+ serverAddress?: string | null;
103
+ serverPort?: number | null;
104
+ monotonicStartS?: number | null;
105
+ }
106
+ export declare function createRetrievalInvocation(init?: Partial<RetrievalInvocation>): RetrievalInvocation;
107
+ export interface RerankInvocation {
108
+ provider: string;
109
+ contextToken?: Context | null;
110
+ span?: Span | null;
111
+ requestModel?: string | null;
112
+ attributes?: Record<string, unknown>;
113
+ topK?: number | null;
114
+ documentsCount?: number | null;
115
+ temperature?: number | null;
116
+ maxTokens?: number | null;
117
+ scoringPrompt?: string | null;
118
+ returnDocuments?: boolean | null;
119
+ maxChunksPerDoc?: number | null;
120
+ device?: string | null;
121
+ batchSize?: number | null;
122
+ maxLength?: number | null;
123
+ normalize?: boolean | null;
124
+ inputDocuments?: unknown;
125
+ outputDocuments?: unknown;
126
+ monotonicStartS?: number | null;
127
+ }
128
+ export declare function createRerankInvocation(provider: string, init?: Partial<RerankInvocation>): RerankInvocation;
129
+ export interface EntryInvocation {
130
+ contextToken?: Context | null;
131
+ span?: Span | null;
132
+ attributes?: Record<string, unknown>;
133
+ sessionId?: string | null;
134
+ userId?: string | null;
135
+ inputMessages?: InputMessage[];
136
+ outputMessages?: OutputMessage[];
137
+ responseTimeToFirstToken?: number | null;
138
+ monotonicStartS?: number | null;
139
+ }
140
+ export declare function createEntryInvocation(init?: Partial<EntryInvocation>): EntryInvocation;
141
+ export interface ReactStepInvocation {
142
+ contextToken?: Context | null;
143
+ span?: Span | null;
144
+ attributes?: Record<string, unknown>;
145
+ finishReason?: string | null;
146
+ round?: number | null;
147
+ monotonicStartS?: number | null;
148
+ }
149
+ export declare function createReactStepInvocation(init?: Partial<ReactStepInvocation>): ReactStepInvocation;
150
+ //# sourceMappingURL=extended-types.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"extended-types.d.ts","sourceRoot":"","sources":["../src/extended-types.ts"],"names":[],"mappings":"AAcA,OAAO,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AACxD,OAAO,KAAK,EACV,YAAY,EACZ,WAAW,EACX,aAAa,EACb,cAAc,EACf,MAAM,YAAY,CAAC;AAEpB,MAAM,WAAW,iBAAiB;IAChC,EAAE,EAAE,MAAM,GAAG,IAAI,CAAC;IAClB,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IACrB,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;CAC3C;AAED,MAAM,WAAW,mBAAmB;IAClC,YAAY,EAAE,MAAM,CAAC;IACrB,YAAY,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;IAC9B,IAAI,CAAC,EAAE,IAAI,GAAG,IAAI,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,iBAAiB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAClC,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACrC,cAAc,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,eAAe,CAAC,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;IAClC,aAAa,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC9B,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACjC;AAED,wBAAgB,yBAAyB,CACvC,YAAY,EAAE,MAAM,EACpB,IAAI,CAAC,EAAE,OAAO,CAAC,mBAAmB,CAAC,GAClC,mBAAmB,CAErB;AAED,MAAM,WAAW,qBAAqB;IACpC,QAAQ,EAAE,MAAM,CAAC;IACjB,YAAY,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;IAC9B,IAAI,CAAC,EAAE,IAAI,GAAG,IAAI,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACrC,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,iBAAiB,CAAC,EAAE,OAAO,CAAC;IAC5B,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACjC;AAED,wBAAgB,2BAA2B,CACzC,QAAQ,EAAE,MAAM,EAChB,IAAI,CAAC,EAAE,OAAO,CAAC,qBAAqB,CAAC,GACpC,qBAAqB,CAEvB;AAED,MAAM,WAAW,qBAAqB;IACpC,QAAQ,EAAE,MAAM,CAAC;IACjB,YAAY,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;IAC9B,IAAI,CAAC,EAAE,IAAI,GAAG,IAAI,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACrC,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,gBAAgB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACjC,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,aAAa,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC9B,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACjC;AAED,wBAAgB,2BAA2B,CACzC,QAAQ,EAAE,MAAM,EAChB,IAAI,CAAC,EAAE,OAAO,CAAC,qBAAqB,CAAC,GACpC,qBAAqB,CAEvB;AAED,MAAM,WAAW,qBAAqB;IACpC,QAAQ,EAAE,MAAM,CAAC;IACjB,YAAY,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;IAC9B,IAAI,CAAC,EAAE,IAAI,GAAG,IAAI,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,aAAa,CAAC,EAAE,YAAY,EAAE,CAAC;IAC/B,cAAc,CAAC,EAAE,aAAa,EAAE,CAAC;IACjC,eAAe,CAAC,EAAE,cAAc,EAAE,CAAC;IACnC,iBAAiB,CAAC,EAAE,WAAW,EAAE,CAAC;IAClC,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACrC,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,gBAAgB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACjC,cAAc,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,iBAAiB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAClC,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,aAAa,CAAC,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;IAChC,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,6BAA6B,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC9C,yBAAyB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1C,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,IAAI,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACrB,gBAAgB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACjC,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,aAAa,CAAC,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC;IAChC,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,IAAI,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACrB,aAAa,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC9B,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,aAAa,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC9B,oBAAoB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACtC;AAED,wBAAgB,2BAA2B,CACzC,QAAQ,EAAE,MAAM,EAChB,IAAI,CAAC,EAAE,OAAO,CAAC,qBAAqB,CAAC,GACpC,qBAAqB,CAUvB;AAED,MAAM,WAAW,mBAAmB;IAClC,YAAY,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;IAC9B,IAAI,CAAC,EAAE,IAAI,GAAG,IAAI,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACrC,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,SAAS,CAAC,EAAE,iBAAiB,EAAE,CAAC;IAChC,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,IAAI,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACrB,aAAa,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC9B,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACjC;AAED,wBAAgB,yBAAyB,CACvC,IAAI,CAAC,EAAE,OAAO,CAAC,mBAAmB,CAAC,GAClC,mBAAmB,CAErB;AAED,MAAM,WAAW,gBAAgB;IAC/B,QAAQ,EAAE,MAAM,CAAC;IACjB,YAAY,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;IAC9B,IAAI,CAAC,EAAE,IAAI,GAAG,IAAI,CAAC;IACnB,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACrC,IAAI,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACrB,cAAc,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,aAAa,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC9B,eAAe,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;IACjC,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAChC,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,SAAS,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;IAC3B,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,eAAe,CAAC,EAAE,OAAO,CAAC;IAC1B,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACjC;AAED,wBAAgB,sBAAsB,CACpC,QAAQ,EAAE,MAAM,EAChB,IAAI,CAAC,EAAE,OAAO,CAAC,gBAAgB,CAAC,GAC/B,gBAAgB,CAElB;AAED,MAAM,WAAW,eAAe;IAC9B,YAAY,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;IAC9B,IAAI,CAAC,EAAE,IAAI,GAAG,IAAI,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACrC,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,MAAM,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACvB,aAAa,CAAC,EAAE,YAAY,EAAE,CAAC;IAC/B,cAAc,CAAC,EAAE,aAAa,EAAE,CAAC;IACjC,wBAAwB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACzC,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACjC;AAED,wBAAgB,qBAAqB,CACnC,IAAI,CAAC,EAAE,OAAO,CAAC,eAAe,CAAC,GAC9B,eAAe,CAEjB;AAED,MAAM,WAAW,mBAAmB;IAClC,YAAY,CAAC,EAAE,OAAO,GAAG,IAAI,CAAC;IAC9B,IAAI,CAAC,EAAE,IAAI,GAAG,IAAI,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACrC,YAAY,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC7B,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,eAAe,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CACjC;AAED,wBAAgB,yBAAyB,CACvC,IAAI,CAAC,EAAE,OAAO,CAAC,mBAAmB,CAAC,GAClC,mBAAmB,CAErB"}
@@ -0,0 +1,46 @@
1
+ // Copyright The OpenTelemetry Authors
2
+ //
3
+ // Licensed under the Apache License, Version 2.0 (the "License");
4
+ // you may not use this file except in compliance with the License.
5
+ // You may obtain a copy of the License at
6
+ //
7
+ // http://www.apache.org/licenses/LICENSE-2.0
8
+ //
9
+ // Unless required by applicable law or agreed to in writing, software
10
+ // distributed under the License is distributed on an "AS IS" BASIS,
11
+ // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ // See the License for the specific language governing permissions and
13
+ // limitations under the License.
14
+ export function createEmbeddingInvocation(requestModel, init) {
15
+ return { requestModel, attributes: {}, ...init };
16
+ }
17
+ export function createExecuteToolInvocation(toolName, init) {
18
+ return { toolName, attributes: {}, ...init };
19
+ }
20
+ export function createCreateAgentInvocation(provider, init) {
21
+ return { provider, attributes: {}, ...init };
22
+ }
23
+ export function createInvokeAgentInvocation(provider, init) {
24
+ return {
25
+ provider,
26
+ inputMessages: [],
27
+ outputMessages: [],
28
+ toolDefinitions: [],
29
+ systemInstruction: [],
30
+ attributes: {},
31
+ ...init,
32
+ };
33
+ }
34
+ export function createRetrievalInvocation(init) {
35
+ return { documents: [], attributes: {}, ...init };
36
+ }
37
+ export function createRerankInvocation(provider, init) {
38
+ return { provider, attributes: {}, ...init };
39
+ }
40
+ export function createEntryInvocation(init) {
41
+ return { inputMessages: [], outputMessages: [], attributes: {}, ...init };
42
+ }
43
+ export function createReactStepInvocation(init) {
44
+ return { attributes: {}, ...init };
45
+ }
46
+ //# sourceMappingURL=extended-types.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"extended-types.js","sourceRoot":"","sources":["../src/extended-types.ts"],"names":[],"mappings":"AAAA,sCAAsC;AACtC,EAAE;AACF,kEAAkE;AAClE,mEAAmE;AACnE,0CAA0C;AAC1C,EAAE;AACF,iDAAiD;AACjD,EAAE;AACF,sEAAsE;AACtE,oEAAoE;AACpE,2EAA2E;AAC3E,sEAAsE;AACtE,iCAAiC;AAkCjC,MAAM,UAAU,yBAAyB,CACvC,YAAoB,EACpB,IAAmC;IAEnC,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,EAAE,EAAE,GAAG,IAAI,EAAE,CAAC;AACnD,CAAC;AAgBD,MAAM,UAAU,2BAA2B,CACzC,QAAgB,EAChB,IAAqC;IAErC,OAAO,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE,GAAG,IAAI,EAAE,CAAC;AAC/C,CAAC;AAgBD,MAAM,UAAU,2BAA2B,CACzC,QAAgB,EAChB,IAAqC;IAErC,OAAO,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE,GAAG,IAAI,EAAE,CAAC;AAC/C,CAAC;AAwCD,MAAM,UAAU,2BAA2B,CACzC,QAAgB,EAChB,IAAqC;IAErC,OAAO;QACL,QAAQ;QACR,aAAa,EAAE,EAAE;QACjB,cAAc,EAAE,EAAE;QAClB,eAAe,EAAE,EAAE;QACnB,iBAAiB,EAAE,EAAE;QACrB,UAAU,EAAE,EAAE;QACd,GAAG,IAAI;KACR,CAAC;AACJ,CAAC;AAiBD,MAAM,UAAU,yBAAyB,CACvC,IAAmC;IAEnC,OAAO,EAAE,SAAS,EAAE,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE,GAAG,IAAI,EAAE,CAAC;AACpD,CAAC;AAwBD,MAAM,UAAU,sBAAsB,CACpC,QAAgB,EAChB,IAAgC;IAEhC,OAAO,EAAE,QAAQ,EAAE,UAAU,EAAE,EAAE,EAAE,GAAG,IAAI,EAAE,CAAC;AAC/C,CAAC;AAcD,MAAM,UAAU,qBAAqB,CACnC,IAA+B;IAE/B,OAAO,EAAE,aAAa,EAAE,EAAE,EAAE,cAAc,EAAE,EAAE,EAAE,UAAU,EAAE,EAAE,EAAE,GAAG,IAAI,EAAE,CAAC;AAC5E,CAAC;AAWD,MAAM,UAAU,yBAAyB,CACvC,IAAmC;IAEnC,OAAO,EAAE,UAAU,EAAE,EAAE,EAAE,GAAG,IAAI,EAAE,CAAC;AACrC,CAAC"}