openlit 1.10.0 → 1.12.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 (195) hide show
  1. package/README.md +35 -1
  2. package/dist/config.d.ts +12 -4
  3. package/dist/config.js +7 -17
  4. package/dist/config.js.map +1 -1
  5. package/dist/evals/llm/anthropic.js +10 -6
  6. package/dist/evals/llm/anthropic.js.map +1 -1
  7. package/dist/evals/llm/openai.js +9 -5
  8. package/dist/evals/llm/openai.js.map +1 -1
  9. package/dist/features/__tests__/rule-engine.test.d.ts +1 -0
  10. package/dist/features/__tests__/rule-engine.test.js +146 -0
  11. package/dist/features/__tests__/rule-engine.test.js.map +1 -0
  12. package/dist/features/base.d.ts +2 -0
  13. package/dist/features/base.js +2 -0
  14. package/dist/features/base.js.map +1 -1
  15. package/dist/features/rule-engine.d.ts +6 -0
  16. package/dist/features/rule-engine.js +60 -0
  17. package/dist/features/rule-engine.js.map +1 -0
  18. package/dist/features/vault.js +1 -1
  19. package/dist/features/vault.js.map +1 -1
  20. package/dist/helpers.d.ts +93 -1
  21. package/dist/helpers.js +270 -8
  22. package/dist/helpers.js.map +1 -1
  23. package/dist/index.d.ts +6 -5
  24. package/dist/index.js +95 -50
  25. package/dist/index.js.map +1 -1
  26. package/dist/instrumentation/__tests__/anthropic-wrapper.test.js +215 -27
  27. package/dist/instrumentation/__tests__/anthropic-wrapper.test.js.map +1 -1
  28. package/dist/instrumentation/__tests__/base-wrapper.test.js +19 -23
  29. package/dist/instrumentation/__tests__/base-wrapper.test.js.map +1 -1
  30. package/dist/instrumentation/__tests__/bedrock-trace-comparison.test.d.ts +1 -0
  31. package/dist/instrumentation/__tests__/bedrock-trace-comparison.test.js +422 -0
  32. package/dist/instrumentation/__tests__/bedrock-trace-comparison.test.js.map +1 -0
  33. package/dist/instrumentation/__tests__/chroma-trace-comparison.test.js +1 -1
  34. package/dist/instrumentation/__tests__/chroma-trace-comparison.test.js.map +1 -1
  35. package/dist/instrumentation/__tests__/cohere-wrapper.test.js +150 -25
  36. package/dist/instrumentation/__tests__/cohere-wrapper.test.js.map +1 -1
  37. package/dist/instrumentation/__tests__/google-ai-trace-comparison.test.js +152 -33
  38. package/dist/instrumentation/__tests__/google-ai-trace-comparison.test.js.map +1 -1
  39. package/dist/instrumentation/__tests__/groq-trace-comparison.test.js +391 -45
  40. package/dist/instrumentation/__tests__/groq-trace-comparison.test.js.map +1 -1
  41. package/dist/instrumentation/__tests__/huggingface-trace-comparison.test.d.ts +2 -2
  42. package/dist/instrumentation/__tests__/huggingface-trace-comparison.test.js +323 -31
  43. package/dist/instrumentation/__tests__/huggingface-trace-comparison.test.js.map +1 -1
  44. package/dist/instrumentation/__tests__/langchain-wrapper.test.d.ts +1 -0
  45. package/dist/instrumentation/__tests__/langchain-wrapper.test.js +282 -0
  46. package/dist/instrumentation/__tests__/langchain-wrapper.test.js.map +1 -0
  47. package/dist/instrumentation/__tests__/milvus-trace-comparison.test.js +1 -1
  48. package/dist/instrumentation/__tests__/milvus-trace-comparison.test.js.map +1 -1
  49. package/dist/instrumentation/__tests__/mistral-trace-comparison.test.d.ts +0 -3
  50. package/dist/instrumentation/__tests__/mistral-trace-comparison.test.js +275 -68
  51. package/dist/instrumentation/__tests__/mistral-trace-comparison.test.js.map +1 -1
  52. package/dist/instrumentation/__tests__/openai-wrapper.test.js +7 -9
  53. package/dist/instrumentation/__tests__/openai-wrapper.test.js.map +1 -1
  54. package/dist/instrumentation/__tests__/qdrant-trace-comparison.test.js +1 -1
  55. package/dist/instrumentation/__tests__/qdrant-trace-comparison.test.js.map +1 -1
  56. package/dist/instrumentation/__tests__/replicate-trace-comparison.test.d.ts +2 -1
  57. package/dist/instrumentation/__tests__/replicate-trace-comparison.test.js +209 -21
  58. package/dist/instrumentation/__tests__/replicate-trace-comparison.test.js.map +1 -1
  59. package/dist/instrumentation/__tests__/together-trace-comparison.test.js +231 -51
  60. package/dist/instrumentation/__tests__/together-trace-comparison.test.js.map +1 -1
  61. package/dist/instrumentation/__tests__/vercel-ai-trace-comparison.test.d.ts +8 -0
  62. package/dist/instrumentation/__tests__/vercel-ai-trace-comparison.test.js +446 -0
  63. package/dist/instrumentation/__tests__/vercel-ai-trace-comparison.test.js.map +1 -0
  64. package/dist/instrumentation/anthropic/index.d.ts +2 -3
  65. package/dist/instrumentation/anthropic/index.js.map +1 -1
  66. package/dist/instrumentation/anthropic/wrapper.d.ts +1 -3
  67. package/dist/instrumentation/anthropic/wrapper.js +211 -91
  68. package/dist/instrumentation/anthropic/wrapper.js.map +1 -1
  69. package/dist/instrumentation/azure-ai-inference/index.d.ts +11 -0
  70. package/dist/instrumentation/azure-ai-inference/index.js +76 -0
  71. package/dist/instrumentation/azure-ai-inference/index.js.map +1 -0
  72. package/dist/instrumentation/azure-ai-inference/wrapper.d.ts +42 -0
  73. package/dist/instrumentation/azure-ai-inference/wrapper.js +515 -0
  74. package/dist/instrumentation/azure-ai-inference/wrapper.js.map +1 -0
  75. package/dist/instrumentation/base-wrapper.d.ts +2 -1
  76. package/dist/instrumentation/base-wrapper.js +35 -23
  77. package/dist/instrumentation/base-wrapper.js.map +1 -1
  78. package/dist/instrumentation/bedrock/wrapper.d.ts +21 -3
  79. package/dist/instrumentation/bedrock/wrapper.js +318 -265
  80. package/dist/instrumentation/bedrock/wrapper.js.map +1 -1
  81. package/dist/instrumentation/chroma/wrapper.js +1 -1
  82. package/dist/instrumentation/chroma/wrapper.js.map +1 -1
  83. package/dist/instrumentation/claude-agent-sdk/index.d.ts +23 -0
  84. package/dist/instrumentation/claude-agent-sdk/index.js +83 -0
  85. package/dist/instrumentation/claude-agent-sdk/index.js.map +1 -0
  86. package/dist/instrumentation/claude-agent-sdk/wrapper.d.ts +13 -0
  87. package/dist/instrumentation/claude-agent-sdk/wrapper.js +1031 -0
  88. package/dist/instrumentation/claude-agent-sdk/wrapper.js.map +1 -0
  89. package/dist/instrumentation/cohere/index.d.ts +2 -3
  90. package/dist/instrumentation/cohere/index.js.map +1 -1
  91. package/dist/instrumentation/cohere/wrapper.d.ts +1 -1
  92. package/dist/instrumentation/cohere/wrapper.js +215 -56
  93. package/dist/instrumentation/cohere/wrapper.js.map +1 -1
  94. package/dist/instrumentation/google-adk/index.d.ts +57 -0
  95. package/dist/instrumentation/google-adk/index.js +371 -0
  96. package/dist/instrumentation/google-adk/index.js.map +1 -0
  97. package/dist/instrumentation/google-adk/utils.d.ts +45 -0
  98. package/dist/instrumentation/google-adk/utils.js +663 -0
  99. package/dist/instrumentation/google-adk/utils.js.map +1 -0
  100. package/dist/instrumentation/google-adk/wrapper.d.ts +11 -0
  101. package/dist/instrumentation/google-adk/wrapper.js +391 -0
  102. package/dist/instrumentation/google-adk/wrapper.js.map +1 -0
  103. package/dist/instrumentation/google-ai/wrapper.d.ts +7 -4
  104. package/dist/instrumentation/google-ai/wrapper.js +197 -61
  105. package/dist/instrumentation/google-ai/wrapper.js.map +1 -1
  106. package/dist/instrumentation/groq/wrapper.js +137 -65
  107. package/dist/instrumentation/groq/wrapper.js.map +1 -1
  108. package/dist/instrumentation/huggingface/wrapper.js +241 -39
  109. package/dist/instrumentation/huggingface/wrapper.js.map +1 -1
  110. package/dist/instrumentation/index.d.ts +2 -2
  111. package/dist/instrumentation/index.js +64 -6
  112. package/dist/instrumentation/index.js.map +1 -1
  113. package/dist/instrumentation/langchain/index.d.ts +0 -7
  114. package/dist/instrumentation/langchain/index.js +2 -20
  115. package/dist/instrumentation/langchain/index.js.map +1 -1
  116. package/dist/instrumentation/langchain/wrapper.d.ts +35 -0
  117. package/dist/instrumentation/langchain/wrapper.js +1098 -184
  118. package/dist/instrumentation/langchain/wrapper.js.map +1 -1
  119. package/dist/instrumentation/langgraph/index.d.ts +12 -0
  120. package/dist/instrumentation/langgraph/index.js +99 -0
  121. package/dist/instrumentation/langgraph/index.js.map +1 -0
  122. package/dist/instrumentation/langgraph/wrapper.d.ts +20 -0
  123. package/dist/instrumentation/langgraph/wrapper.js +619 -0
  124. package/dist/instrumentation/langgraph/wrapper.js.map +1 -0
  125. package/dist/instrumentation/llamaindex/index.d.ts +31 -6
  126. package/dist/instrumentation/llamaindex/index.js +180 -61
  127. package/dist/instrumentation/llamaindex/index.js.map +1 -1
  128. package/dist/instrumentation/llamaindex/wrapper.d.ts +15 -3
  129. package/dist/instrumentation/llamaindex/wrapper.js +670 -179
  130. package/dist/instrumentation/llamaindex/wrapper.js.map +1 -1
  131. package/dist/instrumentation/milvus/wrapper.js +1 -1
  132. package/dist/instrumentation/milvus/wrapper.js.map +1 -1
  133. package/dist/instrumentation/mistral/wrapper.js +154 -79
  134. package/dist/instrumentation/mistral/wrapper.js.map +1 -1
  135. package/dist/instrumentation/ollama/index.js +33 -4
  136. package/dist/instrumentation/ollama/index.js.map +1 -1
  137. package/dist/instrumentation/ollama/wrapper.d.ts +28 -2
  138. package/dist/instrumentation/ollama/wrapper.js +432 -48
  139. package/dist/instrumentation/ollama/wrapper.js.map +1 -1
  140. package/dist/instrumentation/openai/index.d.ts +2 -3
  141. package/dist/instrumentation/openai/index.js.map +1 -1
  142. package/dist/instrumentation/openai/wrapper.js +293 -194
  143. package/dist/instrumentation/openai/wrapper.js.map +1 -1
  144. package/dist/instrumentation/openai-agents/index.d.ts +20 -0
  145. package/dist/instrumentation/openai-agents/index.js +174 -0
  146. package/dist/instrumentation/openai-agents/index.js.map +1 -0
  147. package/dist/instrumentation/openai-agents/processor.d.ts +35 -0
  148. package/dist/instrumentation/openai-agents/processor.js +249 -0
  149. package/dist/instrumentation/openai-agents/processor.js.map +1 -0
  150. package/dist/instrumentation/openai-agents/utils.d.ts +20 -0
  151. package/dist/instrumentation/openai-agents/utils.js +624 -0
  152. package/dist/instrumentation/openai-agents/utils.js.map +1 -0
  153. package/dist/instrumentation/pinecone/wrapper.js +2 -2
  154. package/dist/instrumentation/pinecone/wrapper.js.map +1 -1
  155. package/dist/instrumentation/qdrant/wrapper.js +1 -1
  156. package/dist/instrumentation/qdrant/wrapper.js.map +1 -1
  157. package/dist/instrumentation/replicate/wrapper.js +103 -21
  158. package/dist/instrumentation/replicate/wrapper.js.map +1 -1
  159. package/dist/instrumentation/strands/index.d.ts +21 -0
  160. package/dist/instrumentation/strands/index.js +83 -0
  161. package/dist/instrumentation/strands/index.js.map +1 -0
  162. package/dist/instrumentation/strands/processor.d.ts +45 -0
  163. package/dist/instrumentation/strands/processor.js +545 -0
  164. package/dist/instrumentation/strands/processor.js.map +1 -0
  165. package/dist/instrumentation/strands/utils.d.ts +24 -0
  166. package/dist/instrumentation/strands/utils.js +360 -0
  167. package/dist/instrumentation/strands/utils.js.map +1 -0
  168. package/dist/instrumentation/together/wrapper.js +125 -51
  169. package/dist/instrumentation/together/wrapper.js.map +1 -1
  170. package/dist/instrumentation/vercel-ai/wrapper.d.ts +28 -2
  171. package/dist/instrumentation/vercel-ai/wrapper.js +314 -164
  172. package/dist/instrumentation/vercel-ai/wrapper.js.map +1 -1
  173. package/dist/llm/anthropic.js +10 -6
  174. package/dist/llm/anthropic.js.map +1 -1
  175. package/dist/llm/openai.js +9 -5
  176. package/dist/llm/openai.js.map +1 -1
  177. package/dist/otel/__tests__/metrics.test.js +16 -27
  178. package/dist/otel/__tests__/metrics.test.js.map +1 -1
  179. package/dist/otel/events.d.ts +11 -0
  180. package/dist/otel/events.js +74 -0
  181. package/dist/otel/events.js.map +1 -0
  182. package/dist/otel/metrics.d.ts +5 -6
  183. package/dist/otel/metrics.js +66 -48
  184. package/dist/otel/metrics.js.map +1 -1
  185. package/dist/otel/tracing.d.ts +6 -2
  186. package/dist/otel/tracing.js +71 -24
  187. package/dist/otel/tracing.js.map +1 -1
  188. package/dist/otel/utils.d.ts +11 -0
  189. package/dist/otel/utils.js +34 -0
  190. package/dist/otel/utils.js.map +1 -0
  191. package/dist/semantic-convention.d.ts +44 -5
  192. package/dist/semantic-convention.js +51 -8
  193. package/dist/semantic-convention.js.map +1 -1
  194. package/dist/types.d.ts +74 -22
  195. package/package.json +41 -9
@@ -25,71 +25,259 @@ describe('AnthropicWrapper', () => {
25
25
  });
26
26
  describe('_messageCreate', () => {
27
27
  it('should call recordMetrics after span ends', async () => {
28
- const mockArgs = [{ message: 'test message' }];
29
- const mockResponse = { response_id: '123', meta: { billedUnits: { inputTokens: 10, outputTokens: 20 } } };
28
+ const mockArgs = [{ model: 'claude-3-5-sonnet-latest', messages: [{ role: 'user', content: 'test' }] }];
29
+ const mockResponse = {
30
+ id: 'msg_123',
31
+ model: 'claude-3-5-sonnet-latest',
32
+ content: [{ type: 'text', text: 'Hello' }],
33
+ stop_reason: 'end_turn',
34
+ usage: { input_tokens: 10, output_tokens: 20 },
35
+ };
30
36
  jest.spyOn(base_wrapper_1.default, 'recordMetrics').mockImplementation(() => { });
31
37
  jest.spyOn(wrapper_1.default, '_messageCreateCommonSetter').mockImplementationOnce(async ({ genAIEndpoint, span }) => {
32
- span.setAttribute(semantic_convention_1.default.GEN_AI_OPERATION, semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT);
33
- span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_ID, '123');
38
+ span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_ID, 'msg_123');
34
39
  span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, 100);
35
40
  span.setAttribute(semantic_convention_1.default.GEN_AI_REQUEST_TEMPERATURE, 0.7);
36
41
  span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, 10);
37
42
  span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS, 20);
38
- span.setAttribute(semantic_convention_1.default.GEN_AI_USAGE_TOTAL_TOKENS, 30);
39
- span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, 'stop');
43
+ span.setAttribute(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, ['stop']);
40
44
  return {
41
45
  genAIEndpoint,
42
- model: 'test-model',
43
- user: 'test-user',
46
+ model: 'claude-3-5-sonnet-latest',
47
+ user: undefined,
44
48
  cost: 0.5,
45
49
  aiSystem: 'anthropic',
46
- serverAddress: 'api.anthropic.com',
47
- serverPort: 443,
48
50
  };
49
51
  });
50
52
  await wrapper_1.default._messageCreate({
51
53
  args: mockArgs,
52
- genAIEndpoint: 'anthropic.endpoint',
54
+ genAIEndpoint: 'anthropic.resources.messages',
53
55
  response: mockResponse,
54
56
  span,
55
57
  });
56
58
  expect(base_wrapper_1.default.recordMetrics).toHaveBeenCalledWith(span, {
57
- model: 'test-model',
58
- user: 'test-user',
59
+ model: 'claude-3-5-sonnet-latest',
60
+ user: undefined,
59
61
  cost: 0.5,
60
62
  aiSystem: 'anthropic',
61
- genAIEndpoint: 'anthropic.endpoint',
62
- serverAddress: 'api.anthropic.com',
63
- serverPort: 443,
63
+ genAIEndpoint: 'anthropic.resources.messages',
64
64
  });
65
65
  });
66
+ it('should re-throw errors from commonSetter', async () => {
67
+ const mockArgs = [{ model: 'claude-3-5-sonnet-latest', messages: [] }];
68
+ const mockResponse = {};
69
+ jest.spyOn(base_wrapper_1.default, 'recordMetrics').mockImplementation(() => { });
70
+ jest.spyOn(wrapper_1.default, '_messageCreateCommonSetter').mockRejectedValueOnce(new Error('test error'));
71
+ await expect(wrapper_1.default._messageCreate({
72
+ args: mockArgs,
73
+ genAIEndpoint: 'anthropic.resources.messages',
74
+ response: mockResponse,
75
+ span,
76
+ })).rejects.toThrow('test error');
77
+ });
66
78
  });
67
- describe('_messageCommonSetter', () => {
79
+ describe('_messageCreateCommonSetter', () => {
68
80
  it('should set span attributes and return metric parameters', async () => {
69
- const mockArgs = [{ message: 'test message', max_tokens: 100, temperature: 0.7 }];
81
+ const mockArgs = [{
82
+ model: 'claude-3-5-sonnet-latest',
83
+ messages: [{ role: 'user', content: 'test message' }],
84
+ max_tokens: 100,
85
+ temperature: 0.7,
86
+ }];
70
87
  const mockResult = {
71
- id: '123',
88
+ id: 'msg_123',
72
89
  usage: { input_tokens: 10, output_tokens: 20 },
73
- model: 'claude-3-sonnet-20240229',
74
- stop_reason: 'stop',
90
+ model: 'claude-3-5-sonnet-20241022',
91
+ stop_reason: 'end_turn',
92
+ content: [{ type: 'text', text: 'Hello' }],
75
93
  };
76
- jest.spyOn(config_1.default, 'updatePricingJson').mockResolvedValue({});
94
+ config_1.default.pricingInfo = {};
95
+ config_1.default.disableEvents = true;
77
96
  jest.spyOn(helpers_1.default, 'getChatModelCost').mockReturnValue(0.5);
78
97
  const setAttributeSpy = jest.spyOn(span, 'setAttribute');
79
- await wrapper_1.default._messageCreateCommonSetter({
98
+ const metricParams = await wrapper_1.default._messageCreateCommonSetter({
80
99
  args: mockArgs,
81
- genAIEndpoint: 'anthropic.endpoint',
100
+ genAIEndpoint: 'anthropic.resources.messages',
82
101
  result: mockResult,
83
102
  span,
84
103
  });
85
- expect(setAttributeSpy).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_OPERATION, semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT);
86
- expect(setAttributeSpy).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_RESPONSE_ID, '123');
104
+ expect(setAttributeSpy).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_RESPONSE_ID, 'msg_123');
87
105
  expect(setAttributeSpy).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, 100);
88
106
  expect(setAttributeSpy).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_TEMPERATURE, 0.7);
89
107
  expect(setAttributeSpy).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS, 10);
90
108
  expect(setAttributeSpy).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS, 20);
91
- expect(setAttributeSpy).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_TOTAL_TOKENS, 30);
92
109
  expect(setAttributeSpy).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, ['stop']);
110
+ expect(setAttributeSpy).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_RESPONSE_MODEL, 'claude-3-5-sonnet-20241022');
111
+ expect(setAttributeSpy).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_OUTPUT_TYPE, 'text');
112
+ expect(setAttributeSpy).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_IS_STREAM, false);
113
+ expect(setAttributeSpy).not.toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_TOTAL_TOKENS, expect.anything());
114
+ expect(metricParams).toEqual({
115
+ genAIEndpoint: 'anthropic.resources.messages',
116
+ model: 'claude-3-5-sonnet-latest',
117
+ user: undefined,
118
+ cost: 0.5,
119
+ aiSystem: 'anthropic',
120
+ });
121
+ });
122
+ it('should map Anthropic finish reasons to OTel standard', async () => {
123
+ const makeResult = (stop_reason) => ({
124
+ id: 'msg_1',
125
+ usage: { input_tokens: 5, output_tokens: 5 },
126
+ model: 'claude-3-5-sonnet-latest',
127
+ stop_reason,
128
+ content: [{ type: 'text', text: 'hi' }],
129
+ });
130
+ config_1.default.pricingInfo = {};
131
+ config_1.default.disableEvents = true;
132
+ jest.spyOn(helpers_1.default, 'getChatModelCost').mockReturnValue(0);
133
+ const testCases = [
134
+ { anthropic: 'end_turn', otel: 'stop' },
135
+ { anthropic: 'max_tokens', otel: 'length' },
136
+ { anthropic: 'stop_sequence', otel: 'stop' },
137
+ { anthropic: 'tool_use', otel: 'tool_call' },
138
+ ];
139
+ for (const { anthropic, otel } of testCases) {
140
+ jest.clearAllMocks();
141
+ span.setAttribute = jest.fn();
142
+ await wrapper_1.default._messageCreateCommonSetter({
143
+ args: [{ model: 'claude-3-5-sonnet-latest', messages: [] }],
144
+ genAIEndpoint: 'anthropic.resources.messages',
145
+ result: makeResult(anthropic),
146
+ span,
147
+ });
148
+ expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, [otel]);
149
+ }
150
+ });
151
+ it('should not set max_tokens when not provided', async () => {
152
+ const mockArgs = [{ model: 'claude-3-5-sonnet-latest', messages: [], temperature: 1 }];
153
+ const mockResult = {
154
+ id: 'msg_1',
155
+ usage: { input_tokens: 5, output_tokens: 5 },
156
+ model: 'claude-3-5-sonnet-latest',
157
+ stop_reason: 'end_turn',
158
+ content: [{ type: 'text', text: 'hi' }],
159
+ };
160
+ config_1.default.pricingInfo = {};
161
+ config_1.default.disableEvents = true;
162
+ jest.spyOn(helpers_1.default, 'getChatModelCost').mockReturnValue(0);
163
+ const setAttributeSpy = jest.spyOn(span, 'setAttribute');
164
+ await wrapper_1.default._messageCreateCommonSetter({
165
+ args: mockArgs,
166
+ genAIEndpoint: 'anthropic.resources.messages',
167
+ result: mockResult,
168
+ span,
169
+ });
170
+ expect(setAttributeSpy).not.toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_MAX_TOKENS, expect.anything());
171
+ });
172
+ it('should not set seed when not provided', async () => {
173
+ const mockArgs = [{ model: 'claude-3-5-sonnet-latest', messages: [] }];
174
+ const mockResult = {
175
+ id: 'msg_1',
176
+ usage: { input_tokens: 5, output_tokens: 5 },
177
+ model: 'claude-3-5-sonnet-latest',
178
+ stop_reason: 'end_turn',
179
+ content: [{ type: 'text', text: 'hi' }],
180
+ };
181
+ config_1.default.pricingInfo = {};
182
+ config_1.default.disableEvents = true;
183
+ jest.spyOn(helpers_1.default, 'getChatModelCost').mockReturnValue(0);
184
+ const setAttributeSpy = jest.spyOn(span, 'setAttribute');
185
+ await wrapper_1.default._messageCreateCommonSetter({
186
+ args: mockArgs,
187
+ genAIEndpoint: 'anthropic.resources.messages',
188
+ result: mockResult,
189
+ span,
190
+ });
191
+ expect(setAttributeSpy).not.toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_SEED, expect.anything());
192
+ });
193
+ it('should emit inference event when events are enabled', async () => {
194
+ const mockArgs = [{
195
+ model: 'claude-3-5-sonnet-latest',
196
+ messages: [{ role: 'user', content: 'hello' }],
197
+ }];
198
+ const mockResult = {
199
+ id: 'msg_123',
200
+ usage: { input_tokens: 10, output_tokens: 20 },
201
+ model: 'claude-3-5-sonnet-latest',
202
+ stop_reason: 'end_turn',
203
+ content: [{ type: 'text', text: 'Hi there' }],
204
+ };
205
+ config_1.default.pricingInfo = {};
206
+ config_1.default.disableEvents = false;
207
+ jest.spyOn(helpers_1.default, 'getChatModelCost').mockReturnValue(0.5);
208
+ const emitSpy = jest.spyOn(helpers_1.default, 'emitInferenceEvent').mockImplementation(() => { });
209
+ await wrapper_1.default._messageCreateCommonSetter({
210
+ args: mockArgs,
211
+ genAIEndpoint: 'anthropic.resources.messages',
212
+ result: mockResult,
213
+ span,
214
+ });
215
+ expect(emitSpy).toHaveBeenCalledWith(span, expect.objectContaining({
216
+ [semantic_convention_1.default.GEN_AI_OPERATION]: semantic_convention_1.default.GEN_AI_OPERATION_TYPE_CHAT,
217
+ [semantic_convention_1.default.GEN_AI_REQUEST_MODEL]: 'claude-3-5-sonnet-latest',
218
+ [semantic_convention_1.default.GEN_AI_RESPONSE_MODEL]: 'claude-3-5-sonnet-latest',
219
+ [semantic_convention_1.default.SERVER_ADDRESS]: 'api.anthropic.com',
220
+ [semantic_convention_1.default.SERVER_PORT]: 443,
221
+ [semantic_convention_1.default.GEN_AI_RESPONSE_ID]: 'msg_123',
222
+ [semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON]: ['stop'],
223
+ [semantic_convention_1.default.GEN_AI_OUTPUT_TYPE]: 'text',
224
+ [semantic_convention_1.default.GEN_AI_USAGE_INPUT_TOKENS]: 10,
225
+ [semantic_convention_1.default.GEN_AI_USAGE_OUTPUT_TOKENS]: 20,
226
+ }));
227
+ });
228
+ it('should set tool call attributes when tool_use blocks are present', async () => {
229
+ const mockArgs = [{ model: 'claude-3-5-sonnet-latest', messages: [] }];
230
+ const mockResult = {
231
+ id: 'msg_1',
232
+ usage: { input_tokens: 10, output_tokens: 20 },
233
+ model: 'claude-3-5-sonnet-latest',
234
+ stop_reason: 'tool_use',
235
+ content: [
236
+ { type: 'text', text: '' },
237
+ { type: 'tool_use', id: 'toolu_123', name: 'get_weather', input: { location: 'Paris' } },
238
+ ],
239
+ };
240
+ config_1.default.pricingInfo = {};
241
+ config_1.default.disableEvents = true;
242
+ jest.spyOn(helpers_1.default, 'getChatModelCost').mockReturnValue(0);
243
+ const setAttributeSpy = jest.spyOn(span, 'setAttribute');
244
+ await wrapper_1.default._messageCreateCommonSetter({
245
+ args: mockArgs,
246
+ genAIEndpoint: 'anthropic.resources.messages',
247
+ result: mockResult,
248
+ span,
249
+ });
250
+ expect(setAttributeSpy).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_OUTPUT_TYPE, 'json');
251
+ expect(setAttributeSpy).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_TOOL_NAME, 'get_weather');
252
+ expect(setAttributeSpy).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_TOOL_CALL_ID, 'toolu_123');
253
+ expect(setAttributeSpy).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_RESPONSE_FINISH_REASON, ['tool_call']);
254
+ });
255
+ it('should set cache token attributes when present', async () => {
256
+ const mockArgs = [{ model: 'claude-3-5-sonnet-latest', messages: [] }];
257
+ const mockResult = {
258
+ id: 'msg_1',
259
+ usage: {
260
+ input_tokens: 10,
261
+ output_tokens: 20,
262
+ cache_creation_input_tokens: 5,
263
+ cache_read_input_tokens: 3,
264
+ },
265
+ model: 'claude-3-5-sonnet-latest',
266
+ stop_reason: 'end_turn',
267
+ content: [{ type: 'text', text: 'cached response' }],
268
+ };
269
+ config_1.default.pricingInfo = {};
270
+ config_1.default.disableEvents = true;
271
+ jest.spyOn(helpers_1.default, 'getChatModelCost').mockReturnValue(0);
272
+ const setAttributeSpy = jest.spyOn(span, 'setAttribute');
273
+ await wrapper_1.default._messageCreateCommonSetter({
274
+ args: mockArgs,
275
+ genAIEndpoint: 'anthropic.resources.messages',
276
+ result: mockResult,
277
+ span,
278
+ });
279
+ expect(setAttributeSpy).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_CACHE_CREATION_INPUT_TOKENS, 5);
280
+ expect(setAttributeSpy).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_CACHE_READ_INPUT_TOKENS, 3);
93
281
  });
94
282
  });
95
283
  });
@@ -1 +1 @@
1
- {"version":3,"file":"anthropic-wrapper.test.js","sourceRoot":"","sources":["../../../src/instrumentation/__tests__/anthropic-wrapper.test.ts"],"names":[],"mappings":";;;;;AAAA,4CAAiD;AACjD,mEAAoD;AACpD,0DAAyC;AACzC,4DAA0C;AAC1C,mEAA0C;AAC1C,oFAA2D;AAE3D,IAAI,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;AACjC,IAAI,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;AAClC,IAAI,CAAC,IAAI,CAAC,2CAA2C,CAAC,CAAC;AAEvD,MAAM,UAAU,GAAG,WAAK,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;AAElD,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,IAAI,IAAU,CAAC;IAEf,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,GAAG,UAAU,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QACzC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAC9B,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,GAAG,EAAE,CAAC;IACb,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,QAAQ,GAAG,CAAC,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC,CAAC;YAC/C,MAAM,YAAY,GAAG,EAAE,WAAW,EAAE,KAAK,EAAE,IAAI,EAAE,EAAE,WAAW,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC;YAE1G,IAAI,CAAC,KAAK,CAAC,sBAAW,EAAE,eAAe,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAEtE,IAAI,CAAC,KAAK,CAAC,iBAAgB,EAAE,4BAA4B,CAAC,CAAC,sBAAsB,CAAC,KAAK,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,EAAE,EAAE;gBAClH,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,gBAAgB,EAAE,6BAAkB,CAAC,0BAA0B,CAAC,CAAC;gBACtG,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,KAAK,CAAC,CAAC;gBAChE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,GAAG,CAAC,CAAC;gBACrE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,GAAG,CAAC,CAAC;gBACtE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,EAAE,CAAC,CAAC;gBACpE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,EAAE,CAAC,CAAC;gBACrE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,EAAE,CAAC,CAAC;gBACpE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,MAAM,CAAC,CAAC;gBAE5E,OAAO;oBACL,aAAa;oBACb,KAAK,EAAE,YAAY;oBACnB,IAAI,EAAE,WAAW;oBACjB,IAAI,EAAE,GAAG;oBACT,QAAQ,EAAE,WAAW;oBACrB,aAAa,EAAE,mBAAmB;oBAClC,UAAU,EAAE,GAAG;iBAChB,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,MAAM,iBAAgB,CAAC,cAAc,CAAC;gBACpC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,oBAAoB;gBACnC,QAAQ,EAAE,YAAY;gBACtB,IAAI;aACL,CAAC,CAAC;YAEH,MAAM,CAAC,sBAAW,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAAC,IAAI,EAAE;gBAC3D,KAAK,EAAE,YAAY;gBACnB,IAAI,EAAE,WAAW;gBACjB,IAAI,EAAE,GAAG;gBACT,QAAQ,EAAE,WAAW;gBACrB,aAAa,EAAE,oBAAoB;gBACnC,aAAa,EAAE,mBAAmB;gBAClC,UAAU,EAAE,GAAG;aAChB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,MAAM,QAAQ,GAAG,CAAC,EAAE,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,GAAG,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;YAClF,MAAM,UAAU,GAAG;gBACjB,EAAE,EAAE,KAAK;gBACT,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE;gBAC9C,KAAK,EAAE,0BAA0B;gBACjC,WAAW,EAAE,MAAM;aACpB,CAAC;YAEF,IAAI,CAAC,KAAK,CAAC,gBAAa,EAAE,mBAAmB,CAAC,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YACrE,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;YAEnE,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;YAEzD,MAAM,iBAAgB,CAAC,0BAA0B,CAAC;gBAChD,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,oBAAoB;gBACnC,MAAM,EAAE,UAAU;gBAClB,IAAI;aACL,CAAC,CAAC;YAEH,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,6BAAkB,CAAC,gBAAgB,EACnC,6BAAkB,CAAC,0BAA0B,CAC9C,CAAC;YACF,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,KAAK,CAAC,CAAC;YAC3F,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,GAAG,CAAC,CAAC;YAChG,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,GAAG,CAAC,CAAC;YACjG,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,EAAE,CAAC,CAAC;YAC/F,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,EAAE,CAAC,CAAC;YAChG,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,6BAAkB,CAAC,yBAAyB,EAC5C,EAAE,CACH,CAAC;YACF,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,6BAAkB,CAAC,6BAA6B,EAChD,CAAC,MAAM,CAAC,CACT,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
1
+ {"version":3,"file":"anthropic-wrapper.test.js","sourceRoot":"","sources":["../../../src/instrumentation/__tests__/anthropic-wrapper.test.ts"],"names":[],"mappings":";;;;;AAAA,4CAAiD;AACjD,mEAAoD;AACpD,0DAAyC;AACzC,4DAA0C;AAC1C,mEAA0C;AAC1C,oFAA2D;AAE3D,IAAI,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;AACjC,IAAI,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;AAClC,IAAI,CAAC,IAAI,CAAC,2CAA2C,CAAC,CAAC;AAEvD,MAAM,UAAU,GAAG,WAAK,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;AAElD,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,IAAI,IAAU,CAAC;IAEf,UAAU,CAAC,GAAG,EAAE;QACd,IAAI,GAAG,UAAU,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QACzC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;QAC9B,IAAI,CAAC,aAAa,EAAE,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,GAAG,EAAE,CAAC;IACb,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,QAAQ,GAAG,CAAC,EAAE,KAAK,EAAE,0BAA0B,EAAE,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC;YACxG,MAAM,YAAY,GAAG;gBACnB,EAAE,EAAE,SAAS;gBACb,KAAK,EAAE,0BAA0B;gBACjC,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;gBAC1C,WAAW,EAAE,UAAU;gBACvB,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE;aAC/C,CAAC;YAEF,IAAI,CAAC,KAAK,CAAC,sBAAW,EAAE,eAAe,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAEtE,IAAI,CAAC,KAAK,CAAC,iBAAgB,EAAE,4BAA4B,CAAC,CAAC,sBAAsB,CAAC,KAAK,EAAE,EAAE,aAAa,EAAE,IAAI,EAAE,EAAE,EAAE;gBAClH,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAC;gBACpE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,GAAG,CAAC,CAAC;gBACrE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,GAAG,CAAC,CAAC;gBACtE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,EAAE,CAAC,CAAC;gBACpE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,EAAE,CAAC,CAAC;gBACrE,IAAI,CAAC,YAAY,CAAC,6BAAkB,CAAC,6BAA6B,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;gBAE9E,OAAO;oBACL,aAAa;oBACb,KAAK,EAAE,0BAA0B;oBACjC,IAAI,EAAE,SAAS;oBACf,IAAI,EAAE,GAAG;oBACT,QAAQ,EAAE,WAAW;iBACtB,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,MAAM,iBAAgB,CAAC,cAAc,CAAC;gBACpC,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,8BAA8B;gBAC7C,QAAQ,EAAE,YAAY;gBACtB,IAAI;aACL,CAAC,CAAC;YAEH,MAAM,CAAC,sBAAW,CAAC,aAAa,CAAC,CAAC,oBAAoB,CAAC,IAAI,EAAE;gBAC3D,KAAK,EAAE,0BAA0B;gBACjC,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,GAAG;gBACT,QAAQ,EAAE,WAAW;gBACrB,aAAa,EAAE,8BAA8B;aAC9C,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,QAAQ,GAAG,CAAC,EAAE,KAAK,EAAE,0BAA0B,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;YACvE,MAAM,YAAY,GAAG,EAAE,CAAC;YAExB,IAAI,CAAC,KAAK,CAAC,sBAAW,EAAE,eAAe,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YACtE,IAAI,CAAC,KAAK,CAAC,iBAAgB,EAAE,4BAA4B,CAAC,CAAC,qBAAqB,CAAC,IAAI,KAAK,CAAC,YAAY,CAAC,CAAC,CAAC;YAE1G,MAAM,MAAM,CACV,iBAAgB,CAAC,cAAc,CAAC;gBAC9B,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,8BAA8B;gBAC7C,QAAQ,EAAE,YAAY;gBACtB,IAAI;aACL,CAAC,CACH,CAAC,OAAO,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;QAC1C,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;YACvE,MAAM,QAAQ,GAAG,CAAC;oBAChB,KAAK,EAAE,0BAA0B;oBACjC,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC;oBACrD,UAAU,EAAE,GAAG;oBACf,WAAW,EAAE,GAAG;iBACjB,CAAC,CAAC;YACH,MAAM,UAAU,GAAG;gBACjB,EAAE,EAAE,SAAS;gBACb,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE;gBAC9C,KAAK,EAAE,4BAA4B;gBACnC,WAAW,EAAE,UAAU;gBACvB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC3C,CAAC;YAED,gBAAqB,CAAC,WAAW,GAAG,EAAE,CAAC;YACvC,gBAAqB,CAAC,aAAa,GAAG,IAAI,CAAC;YAC5C,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;YAEnE,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;YAEzD,MAAM,YAAY,GAAG,MAAM,iBAAgB,CAAC,0BAA0B,CAAC;gBACrE,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,8BAA8B;gBAC7C,MAAM,EAAE,UAAU;gBAClB,IAAI;aACL,CAAC,CAAC;YAEH,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,kBAAkB,EAAE,SAAS,CAAC,CAAC;YAC/F,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,GAAG,CAAC,CAAC;YAChG,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,GAAG,CAAC,CAAC;YACjG,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,yBAAyB,EAAE,EAAE,CAAC,CAAC;YAC/F,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,0BAA0B,EAAE,EAAE,CAAC,CAAC;YAChG,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,6BAAkB,CAAC,6BAA6B,EAChD,CAAC,MAAM,CAAC,CACT,CAAC;YACF,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,6BAAkB,CAAC,qBAAqB,EACxC,4BAA4B,CAC7B,CAAC;YACF,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,6BAAkB,CAAC,kBAAkB,EACrC,MAAM,CACP,CAAC;YACF,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,wBAAwB,EAAE,KAAK,CAAC,CAAC;YAEjG,MAAM,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,oBAAoB,CAC9C,6BAAkB,CAAC,yBAAyB,EAC5C,MAAM,CAAC,QAAQ,EAAE,CAClB,CAAC;YAEF,MAAM,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC;gBAC3B,aAAa,EAAE,8BAA8B;gBAC7C,KAAK,EAAE,0BAA0B;gBACjC,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,GAAG;gBACT,QAAQ,EAAE,WAAW;aACtB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,UAAU,GAAG,CAAC,WAAmB,EAAE,EAAE,CAAC,CAAC;gBAC3C,EAAE,EAAE,OAAO;gBACX,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE;gBAC5C,KAAK,EAAE,0BAA0B;gBACjC,WAAW;gBACX,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;aACxC,CAAC,CAAC;YAEF,gBAAqB,CAAC,WAAW,GAAG,EAAE,CAAC;YACvC,gBAAqB,CAAC,aAAa,GAAG,IAAI,CAAC;YAC5C,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAEjE,MAAM,SAAS,GAAG;gBAChB,EAAE,SAAS,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,EAAE;gBACvC,EAAE,SAAS,EAAE,YAAY,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAC3C,EAAE,SAAS,EAAE,eAAe,EAAE,IAAI,EAAE,MAAM,EAAE;gBAC5C,EAAE,SAAS,EAAE,UAAU,EAAE,IAAI,EAAE,WAAW,EAAE;aAC7C,CAAC;YAEF,KAAK,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,IAAI,SAAS,EAAE,CAAC;gBAC5C,IAAI,CAAC,aAAa,EAAE,CAAC;gBACrB,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC;gBAE9B,MAAM,iBAAgB,CAAC,0BAA0B,CAAC;oBAChD,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,0BAA0B,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;oBAC3D,aAAa,EAAE,8BAA8B;oBAC7C,MAAM,EAAE,UAAU,CAAC,SAAS,CAAC;oBAC7B,IAAI;iBACL,CAAC,CAAC;gBAEH,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAC5C,6BAAkB,CAAC,6BAA6B,EAChD,CAAC,IAAI,CAAC,CACP,CAAC;YACJ,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,QAAQ,GAAG,CAAC,EAAE,KAAK,EAAE,0BAA0B,EAAE,QAAQ,EAAE,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC,CAAC;YACvF,MAAM,UAAU,GAAG;gBACjB,EAAE,EAAE,OAAO;gBACX,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE;gBAC5C,KAAK,EAAE,0BAA0B;gBACjC,WAAW,EAAE,UAAU;gBACvB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;aACxC,CAAC;YAED,gBAAqB,CAAC,WAAW,GAAG,EAAE,CAAC;YACvC,gBAAqB,CAAC,aAAa,GAAG,IAAI,CAAC;YAC5C,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YACjE,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;YAEzD,MAAM,iBAAgB,CAAC,0BAA0B,CAAC;gBAChD,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,8BAA8B;gBAC7C,MAAM,EAAE,UAAU;gBAClB,IAAI;aACL,CAAC,CAAC;YAEH,MAAM,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,oBAAoB,CAC9C,6BAAkB,CAAC,yBAAyB,EAC5C,MAAM,CAAC,QAAQ,EAAE,CAClB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,QAAQ,GAAG,CAAC,EAAE,KAAK,EAAE,0BAA0B,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;YACvE,MAAM,UAAU,GAAG;gBACjB,EAAE,EAAE,OAAO;gBACX,KAAK,EAAE,EAAE,YAAY,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE;gBAC5C,KAAK,EAAE,0BAA0B;gBACjC,WAAW,EAAE,UAAU;gBACvB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;aACxC,CAAC;YAED,gBAAqB,CAAC,WAAW,GAAG,EAAE,CAAC;YACvC,gBAAqB,CAAC,aAAa,GAAG,IAAI,CAAC;YAC5C,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YACjE,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;YAEzD,MAAM,iBAAgB,CAAC,0BAA0B,CAAC;gBAChD,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,8BAA8B;gBAC7C,MAAM,EAAE,UAAU;gBAClB,IAAI;aACL,CAAC,CAAC;YAEH,MAAM,CAAC,eAAe,CAAC,CAAC,GAAG,CAAC,oBAAoB,CAC9C,6BAAkB,CAAC,mBAAmB,EACtC,MAAM,CAAC,QAAQ,EAAE,CAClB,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,QAAQ,GAAG,CAAC;oBAChB,KAAK,EAAE,0BAA0B;oBACjC,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC;iBAC/C,CAAC,CAAC;YACH,MAAM,UAAU,GAAG;gBACjB,EAAE,EAAE,SAAS;gBACb,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE;gBAC9C,KAAK,EAAE,0BAA0B;gBACjC,WAAW,EAAE,UAAU;gBACvB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;aAC9C,CAAC;YAED,gBAAqB,CAAC,WAAW,GAAG,EAAE,CAAC;YACvC,gBAAqB,CAAC,aAAa,GAAG,KAAK,CAAC;YAC7C,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;YACnE,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,oBAAoB,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAE7F,MAAM,iBAAgB,CAAC,0BAA0B,CAAC;gBAChD,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,8BAA8B;gBAC7C,MAAM,EAAE,UAAU;gBAClB,IAAI;aACL,CAAC,CAAC;YAEH,MAAM,CAAC,OAAO,CAAC,CAAC,oBAAoB,CAAC,IAAI,EAAE,MAAM,CAAC,gBAAgB,CAAC;gBACjE,CAAC,6BAAkB,CAAC,gBAAgB,CAAC,EAAE,6BAAkB,CAAC,0BAA0B;gBACpF,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,EAAE,0BAA0B;gBACrE,CAAC,6BAAkB,CAAC,qBAAqB,CAAC,EAAE,0BAA0B;gBACtE,CAAC,6BAAkB,CAAC,cAAc,CAAC,EAAE,mBAAmB;gBACxD,CAAC,6BAAkB,CAAC,WAAW,CAAC,EAAE,GAAG;gBACrC,CAAC,6BAAkB,CAAC,kBAAkB,CAAC,EAAE,SAAS;gBAClD,CAAC,6BAAkB,CAAC,6BAA6B,CAAC,EAAE,CAAC,MAAM,CAAC;gBAC5D,CAAC,6BAAkB,CAAC,kBAAkB,CAAC,EAAE,MAAM;gBAC/C,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,EAAE;gBAClD,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,EAAE,EAAE;aACpD,CAAC,CAAC,CAAC;QACN,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;YAChF,MAAM,QAAQ,GAAG,CAAC,EAAE,KAAK,EAAE,0BAA0B,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;YACvE,MAAM,UAAU,GAAG;gBACjB,EAAE,EAAE,OAAO;gBACX,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE;gBAC9C,KAAK,EAAE,0BAA0B;gBACjC,WAAW,EAAE,UAAU;gBACvB,OAAO,EAAE;oBACP,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,EAAE;oBAC1B,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE;iBACzF;aACF,CAAC;YAED,gBAAqB,CAAC,WAAW,GAAG,EAAE,CAAC;YACvC,gBAAqB,CAAC,aAAa,GAAG,IAAI,CAAC;YAC5C,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YACjE,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;YAEzD,MAAM,iBAAgB,CAAC,0BAA0B,CAAC;gBAChD,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,8BAA8B;gBAC7C,MAAM,EAAE,UAAU;gBAClB,IAAI;aACL,CAAC,CAAC;YAEH,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,6BAAkB,CAAC,kBAAkB,EACrC,MAAM,CACP,CAAC;YACF,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,6BAAkB,CAAC,gBAAgB,EACnC,aAAa,CACd,CAAC;YACF,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,6BAAkB,CAAC,mBAAmB,EACtC,WAAW,CACZ,CAAC;YACF,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,6BAAkB,CAAC,6BAA6B,EAChD,CAAC,WAAW,CAAC,CACd,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,QAAQ,GAAG,CAAC,EAAE,KAAK,EAAE,0BAA0B,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;YACvE,MAAM,UAAU,GAAG;gBACjB,EAAE,EAAE,OAAO;gBACX,KAAK,EAAE;oBACL,YAAY,EAAE,EAAE;oBAChB,aAAa,EAAE,EAAE;oBACjB,2BAA2B,EAAE,CAAC;oBAC9B,uBAAuB,EAAE,CAAC;iBAC3B;gBACD,KAAK,EAAE,0BAA0B;gBACjC,WAAW,EAAE,UAAU;gBACvB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC;aACrD,CAAC;YAED,gBAAqB,CAAC,WAAW,GAAG,EAAE,CAAC;YACvC,gBAAqB,CAAC,aAAa,GAAG,IAAI,CAAC;YAC5C,IAAI,CAAC,KAAK,CAAC,iBAAa,EAAE,kBAAkB,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YACjE,MAAM,eAAe,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;YAEzD,MAAM,iBAAgB,CAAC,0BAA0B,CAAC;gBAChD,IAAI,EAAE,QAAQ;gBACd,aAAa,EAAE,8BAA8B;gBAC7C,MAAM,EAAE,UAAU;gBAClB,IAAI;aACL,CAAC,CAAC;YAEH,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,6BAAkB,CAAC,wCAAwC,EAC3D,CAAC,CACF,CAAC;YACF,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC1C,6BAAkB,CAAC,oCAAoC,EACvD,CAAC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
@@ -10,20 +10,15 @@ const base_wrapper_1 = __importDefault(require("../base-wrapper"));
10
10
  const index_1 = __importDefault(require("../../index"));
11
11
  describe('BaseWrapper.setBaseSpanAttributes', () => {
12
12
  let span;
13
- // eslint-disable-next-line @typescript-eslint/no-unused-vars
14
- let addSpy;
15
- // eslint-disable-next-line @typescript-eslint/no-unused-vars
16
- let recordSpy;
17
13
  beforeEach(() => {
18
14
  index_1.default.init({
19
15
  applicationName: 'TestApp',
20
16
  environment: 'TestEnv',
21
17
  otlpEndpoint: 'http://localhost:4318',
22
18
  });
23
- metrics_1.default.setup({ resource: (0, resources_1.defaultResource)(), otlpEndpoint: 'http://localhost:4318' }); // Ensure metrics are initialized with a valid endpoint
24
- addSpy = jest.spyOn(metrics_1.default.genaiRequests, 'add').mockImplementation(() => { });
25
- jest.spyOn(metrics_1.default.genaiPromptTokens, 'add').mockImplementation(() => { });
26
- jest.spyOn(metrics_1.default.genaiCompletionTokens, 'add').mockImplementation(() => { });
19
+ metrics_1.default.resetForTesting();
20
+ metrics_1.default.setup({ resource: (0, resources_1.defaultResource)(), otlpEndpoint: 'http://localhost:4318', environment: 'TestEnv', applicationName: 'TestApp', disableBatch: false, captureMessageContent: true, disableMetrics: false, disableEvents: false });
21
+ jest.spyOn(metrics_1.default.genaiClientUsageTokens, 'record').mockImplementation(() => { });
27
22
  jest.spyOn(metrics_1.default.genaiClientOperationDuration, 'record').mockImplementation(() => { });
28
23
  jest.spyOn(metrics_1.default.genaiCost, 'record').mockImplementation(() => { });
29
24
  span = {
@@ -59,13 +54,12 @@ describe('BaseWrapper.setBaseSpanAttributes', () => {
59
54
  expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_REQUEST_USER, 'user1');
60
55
  expect(span.setAttribute).toHaveBeenCalledWith(semantic_convention_1.default.GEN_AI_USAGE_COST, 0.99);
61
56
  expect(span.setStatus).toHaveBeenCalled();
62
- expect(metrics_1.default.genaiRequests.add).toHaveBeenCalledWith(1, expect.objectContaining({
63
- [semantic_convention_1.default.GEN_AI_PROVIDER_NAME]: 'openai',
64
- [semantic_convention_1.default.GEN_AI_REQUEST_USER]: 'user1',
65
- [semantic_convention_1.default.GEN_AI_REQUEST_MODEL]: 'gpt-4',
57
+ expect(metrics_1.default.genaiClientUsageTokens.record).toHaveBeenCalledWith(10, expect.objectContaining({
58
+ [semantic_convention_1.default.GEN_AI_TOKEN_TYPE]: semantic_convention_1.default.GEN_AI_TOKEN_TYPE_INPUT,
59
+ }));
60
+ expect(metrics_1.default.genaiClientUsageTokens.record).toHaveBeenCalledWith(20, expect.objectContaining({
61
+ [semantic_convention_1.default.GEN_AI_TOKEN_TYPE]: semantic_convention_1.default.GEN_AI_TOKEN_TYPE_OUTPUT,
66
62
  }));
67
- expect(metrics_1.default.genaiPromptTokens.add).toHaveBeenCalledWith(10, expect.any(Object));
68
- expect(metrics_1.default.genaiCompletionTokens.add).toHaveBeenCalledWith(20, expect.any(Object));
69
63
  expect(metrics_1.default.genaiClientOperationDuration.record).toHaveBeenCalledWith(1.5e-9, expect.any(Object));
70
64
  expect(metrics_1.default.genaiCost.record).toHaveBeenCalledWith(0.99, expect.any(Object));
71
65
  });
@@ -85,16 +79,15 @@ describe('BaseWrapper.setBaseSpanAttributes', () => {
85
79
  };
86
80
  base_wrapper_1.default.setBaseSpanAttributes(span, baseAttributes);
87
81
  base_wrapper_1.default.recordMetrics(span, baseAttributes);
88
- expect(metrics_1.default.genaiPromptTokens.add).not.toHaveBeenCalled();
89
- expect(metrics_1.default.genaiCompletionTokens.add).not.toHaveBeenCalled();
82
+ expect(metrics_1.default.genaiClientUsageTokens.record).not.toHaveBeenCalled();
90
83
  expect(metrics_1.default.genaiClientOperationDuration.record).not.toHaveBeenCalled();
91
84
  expect(metrics_1.default.genaiCost.record).not.toHaveBeenCalled();
92
85
  });
93
86
  describe('metrics logic for inputTokens, outputTokens, duration, cost', () => {
94
87
  beforeEach(() => {
95
- metrics_1.default.setup({ resource: (0, resources_1.defaultResource)(), otlpEndpoint: 'http://localhost:4318' });
96
- jest.spyOn(metrics_1.default.genaiPromptTokens, 'add').mockImplementation(() => { });
97
- jest.spyOn(metrics_1.default.genaiCompletionTokens, 'add').mockImplementation(() => { });
88
+ metrics_1.default.resetForTesting();
89
+ metrics_1.default.setup({ resource: (0, resources_1.defaultResource)(), otlpEndpoint: 'http://localhost:4318', environment: 'TestEnv', applicationName: 'TestApp', disableBatch: false, captureMessageContent: true, disableMetrics: false, disableEvents: false });
90
+ jest.spyOn(metrics_1.default.genaiClientUsageTokens, 'record').mockImplementation(() => { });
98
91
  jest.spyOn(metrics_1.default.genaiClientOperationDuration, 'record').mockImplementation(() => { });
99
92
  jest.spyOn(metrics_1.default.genaiCost, 'record').mockImplementation(() => { });
100
93
  });
@@ -118,8 +111,7 @@ describe('BaseWrapper.setBaseSpanAttributes', () => {
118
111
  };
119
112
  base_wrapper_1.default.setBaseSpanAttributes(span, baseAttributes);
120
113
  base_wrapper_1.default.recordMetrics(span, baseAttributes);
121
- expect(metrics_1.default.genaiPromptTokens.add).not.toHaveBeenCalled();
122
- expect(metrics_1.default.genaiCompletionTokens.add).not.toHaveBeenCalled();
114
+ expect(metrics_1.default.genaiClientUsageTokens.record).not.toHaveBeenCalled();
123
115
  expect(metrics_1.default.genaiClientOperationDuration.record).not.toHaveBeenCalled();
124
116
  expect(metrics_1.default.genaiCost.record).not.toHaveBeenCalled();
125
117
  });
@@ -143,8 +135,12 @@ describe('BaseWrapper.setBaseSpanAttributes', () => {
143
135
  };
144
136
  base_wrapper_1.default.setBaseSpanAttributes(span, baseAttributes);
145
137
  base_wrapper_1.default.recordMetrics(span, baseAttributes);
146
- expect(metrics_1.default.genaiPromptTokens.add).toHaveBeenCalledWith(0, expect.any(Object));
147
- expect(metrics_1.default.genaiCompletionTokens.add).toHaveBeenCalledWith(-5, expect.any(Object));
138
+ expect(metrics_1.default.genaiClientUsageTokens.record).toHaveBeenCalledWith(0, expect.objectContaining({
139
+ [semantic_convention_1.default.GEN_AI_TOKEN_TYPE]: semantic_convention_1.default.GEN_AI_TOKEN_TYPE_INPUT,
140
+ }));
141
+ expect(metrics_1.default.genaiClientUsageTokens.record).toHaveBeenCalledWith(-5, expect.objectContaining({
142
+ [semantic_convention_1.default.GEN_AI_TOKEN_TYPE]: semantic_convention_1.default.GEN_AI_TOKEN_TYPE_OUTPUT,
143
+ }));
148
144
  expect(metrics_1.default.genaiClientOperationDuration.record).toHaveBeenCalledWith(-1.5e-9, expect.any(Object));
149
145
  expect(metrics_1.default.genaiCost.record).toHaveBeenCalledWith(0, expect.any(Object));
150
146
  });
@@ -1 +1 @@
1
- {"version":3,"file":"base-wrapper.test.js","sourceRoot":"","sources":["../../../src/instrumentation/__tests__/base-wrapper.test.ts"],"names":[],"mappings":";;;;;AACA,wDAA2D;AAC3D,oFAA2D;AAC3D,iEAAyC;AACzC,mEAA0C;AAC1C,wDAAkC;AAElC,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;IAIjD,IAAI,IAAc,CAAC;IACnB,6DAA6D;IAC7D,IAAI,MAAwB,CAAC;IAC7B,6DAA6D;IAC7D,IAAI,SAA2B,CAAC;IAGhC,UAAU,CAAC,GAAG,EAAE;QACd,eAAO,CAAC,IAAI,CAAC;YACX,eAAe,EAAE,SAAS;YAC1B,WAAW,EAAE,SAAS;YACtB,YAAY,EAAE,uBAAuB;SACtC,CAAC,CAAC;QACH,iBAAO,CAAC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAA,2BAAe,GAAE,EAAE,YAAY,EAAE,uBAAuB,EAAE,CAAC,CAAC,CAAC,uDAAuD;QAC9I,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,iBAAO,CAAC,aAAc,EAAE,KAAK,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAChF,IAAI,CAAC,KAAK,CAAC,iBAAO,CAAC,iBAAkB,EAAE,KAAK,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAC3E,IAAI,CAAC,KAAK,CAAC,iBAAO,CAAC,qBAAsB,EAAE,KAAK,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QAC/E,IAAI,CAAC,KAAK,CAAC,iBAAO,CAAC,4BAA6B,EAAE,QAAQ,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QACzF,IAAI,CAAC,KAAK,CAAC,iBAAO,CAAC,SAAU,EAAE,QAAQ,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QACtE,IAAI,GAAG;YACL,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;YACvB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;YACpB,UAAU,EAAE;gBACV,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,EAAE;gBAClD,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,EAAE,EAAE;gBACnD,QAAQ,EAAE,GAAG;aACd;SACF,CAAC;QACF,MAAM,CAAC,cAAc,CAAC,IAAI,EAAE,eAAe,EAAE;YAC3C,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;YAChB,QAAQ,EAAE,IAAI;YACd,YAAY,EAAE,IAAI;YAClB,UAAU,EAAE,IAAI;SACjB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,eAAe,EAAE,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,cAAc,GAAG;YACrB,KAAK,EAAE,OAAO;YACd,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,IAAI;YACV,QAAQ,EAAE,QAAQ;YAClB,aAAa,EAAE,UAAU;SAC1B,CAAC;QACF,oFAAoF;QACpF,sBAAW,CAAC,qBAAqB,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;QACxD,sBAAW,CAAC,aAAa,CAAC,IAAuB,EAAE,cAAc,CAAC,CAAC;QAEnE,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,mBAAmB,EAAE,OAAO,CAAC,CAAC;QAChG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;QAC3F,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC1C,MAAM,CAAC,iBAAO,CAAC,aAAc,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAAC,CAAC,EAAE,MAAM,CAAC,gBAAgB,CAAC;YACjF,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,EAAE,QAAQ;YACnD,CAAC,6BAAkB,CAAC,mBAAmB,CAAC,EAAE,OAAO;YACjD,CAAC,6BAAkB,CAAC,oBAAoB,CAAC,EAAE,OAAO;SACnD,CAAC,CAAC,CAAC;QACJ,MAAM,CAAC,iBAAO,CAAC,iBAAkB,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAAC,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;QACpF,MAAM,CAAC,iBAAO,CAAC,qBAAsB,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAAC,EAAE,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;QACxF,MAAM,CAAC,iBAAO,CAAC,4BAA6B,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;QACtG,MAAM,CAAC,iBAAO,CAAC,SAAU,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;IACnF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,CAAC,cAAc,CAAC,IAAI,EAAE,YAAY,EAAE;YACxC,KAAK,EAAE,EAAE;YACT,QAAQ,EAAE,IAAI;YACd,YAAY,EAAE,IAAI;YAClB,UAAU,EAAE,IAAI;SACjB,CAAC,CAAC;QACH,MAAM,cAAc,GAAG;YACrB,aAAa,EAAE,UAAU;YACzB,KAAK,EAAE,OAAO;YACd,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,SAAS;YACf,QAAQ,EAAE,QAAQ;SACnB,CAAC;QACF,sBAAW,CAAC,qBAAqB,CAAC,IAAuB,EAAE,cAAc,CAAC,CAAC;QAC3E,sBAAW,CAAC,aAAa,CAAC,IAAuB,EAAE,cAAc,CAAC,CAAC;QAEnE,MAAM,CAAC,iBAAO,CAAC,iBAAkB,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC9D,MAAM,CAAC,iBAAO,CAAC,qBAAsB,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAClE,MAAM,CAAC,iBAAO,CAAC,4BAA6B,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC5E,MAAM,CAAC,iBAAO,CAAC,SAAU,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,6DAA6D,EAAE,GAAG,EAAE;QAC3E,UAAU,CAAC,GAAG,EAAE;YACd,iBAAO,CAAC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAA,2BAAe,GAAE,EAAE,YAAY,EAAE,uBAAuB,EAAE,CAAC,CAAC;YACtF,IAAI,CAAC,KAAK,CAAC,iBAAO,CAAC,iBAAkB,EAAE,KAAK,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAC3E,IAAI,CAAC,KAAK,CAAC,iBAAO,CAAC,qBAAsB,EAAE,KAAK,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YAC/E,IAAI,CAAC,KAAK,CAAC,iBAAO,CAAC,4BAA6B,EAAE,QAAQ,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YACzF,IAAI,CAAC,KAAK,CAAC,iBAAO,CAAC,SAAU,EAAE,QAAQ,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;YAC1E,MAAM,IAAI,GAAa;gBACrB,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;gBACvB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;gBACpB,aAAa,EAAE,IAAI,CAAC,EAAE,EAAE;gBACxB,UAAU,EAAE;oBACV,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,GAAG;oBACnD,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,EAAE,SAAS;oBAC1D,QAAQ,EAAE,cAAc;iBACzB;aACF,CAAC;YACF,MAAM,cAAc,GAAG;gBACrB,KAAK,EAAE,OAAO;gBACd,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,cAAc;gBACpB,QAAQ,EAAE,QAAQ;gBAClB,aAAa,EAAE,UAAU;aAC1B,CAAC;YACF,sBAAW,CAAC,qBAAqB,CAAC,IAAuB,EAAE,cAAc,CAAC,CAAC;YAC3E,sBAAW,CAAC,aAAa,CAAC,IAAuB,EAAE,cAAc,CAAC,CAAC;YAEnE,MAAM,CAAC,iBAAO,CAAC,iBAAkB,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAC9D,MAAM,CAAC,iBAAO,CAAC,qBAAsB,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAClE,MAAM,CAAC,iBAAO,CAAC,4BAA6B,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAC5E,MAAM,CAAC,iBAAO,CAAC,SAAU,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,IAAI,GAAa;gBACrB,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;gBACvB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;gBACpB,aAAa,EAAE,IAAI,CAAC,EAAE,EAAE;gBACxB,UAAU,EAAE;oBACV,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,CAAC;oBACjD,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,EAAE,CAAC,CAAC;oBACnD,QAAQ,EAAE,CAAC,GAAG;iBACf;aACF,CAAC;YACF,MAAM,cAAc,GAAG;gBACrB,KAAK,EAAE,OAAO;gBACd,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,CAAC;gBACP,QAAQ,EAAE,QAAQ;gBAClB,aAAa,EAAE,UAAU;aAC1B,CAAC;YACF,sBAAW,CAAC,qBAAqB,CAAC,IAAuB,EAAE,cAAc,CAAC,CAAC;YAC3E,sBAAW,CAAC,aAAa,CAAC,IAAuB,EAAE,cAAc,CAAC,CAAC;YAEnE,MAAM,CAAC,iBAAO,CAAC,iBAAkB,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAAC,CAAC,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;YACnF,MAAM,CAAC,iBAAO,CAAC,qBAAsB,CAAC,GAAG,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;YACxF,MAAM,CAAC,iBAAO,CAAC,4BAA6B,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;YACvG,MAAM,CAAC,iBAAO,CAAC,SAAU,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,CAAC,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;QAChF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,IAAI,GAAa;gBACrB,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;gBACvB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;gBACpB,aAAa,EAAE,IAAI,CAAC,EAAE,EAAE;gBACxB,UAAU,EAAE;oBACV,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,CAAC;oBACjD,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,EAAE,CAAC;oBAClD,QAAQ,EAAE,CAAC;iBACZ;aACF,CAAC;YACF,MAAM,cAAc,GAAG;gBACrB,KAAK,EAAE,OAAO;gBACd,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,MAAM;gBACZ,QAAQ,EAAE,QAAQ;gBAClB,aAAa,EAAE,UAAU;aAC1B,CAAC;YACF,sBAAW,CAAC,qBAAqB,CAAC,IAAuB,EAAE,cAAc,CAAC,CAAC;YAC3E,sBAAW,CAAC,aAAa,CAAC,IAAuB,EAAE,cAAc,CAAC,CAAC;YAEnE,MAAM,CAAC,iBAAO,CAAC,SAAU,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;QACnF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
1
+ {"version":3,"file":"base-wrapper.test.js","sourceRoot":"","sources":["../../../src/instrumentation/__tests__/base-wrapper.test.ts"],"names":[],"mappings":";;;;;AACA,wDAA2D;AAC3D,oFAA2D;AAC3D,iEAAyC;AACzC,mEAA0C;AAC1C,wDAAkC;AAElC,QAAQ,CAAC,mCAAmC,EAAE,GAAG,EAAE;IAIjD,IAAI,IAAc,CAAC;IACnB,UAAU,CAAC,GAAG,EAAE;QACd,eAAO,CAAC,IAAI,CAAC;YACX,eAAe,EAAE,SAAS;YAC1B,WAAW,EAAE,SAAS;YACtB,YAAY,EAAE,uBAAuB;SACtC,CAAC,CAAC;QACH,iBAAO,CAAC,eAAe,EAAE,CAAC;QAC1B,iBAAO,CAAC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAA,2BAAe,GAAE,EAAE,YAAY,EAAE,uBAAuB,EAAE,WAAW,EAAE,SAAS,EAAE,eAAe,EAAE,SAAS,EAAE,YAAY,EAAE,KAAK,EAAE,qBAAqB,EAAE,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE,aAAa,EAAE,KAAK,EAAS,CAAC,CAAC;QAChP,IAAI,CAAC,KAAK,CAAC,iBAAO,CAAC,sBAAuB,EAAE,QAAQ,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QACnF,IAAI,CAAC,KAAK,CAAC,iBAAO,CAAC,4BAA6B,EAAE,QAAQ,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QACzF,IAAI,CAAC,KAAK,CAAC,iBAAO,CAAC,SAAU,EAAE,QAAQ,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QACtE,IAAI,GAAG;YACL,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;YACvB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;YACpB,UAAU,EAAE;gBACV,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,EAAE;gBAClD,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,EAAE,EAAE;gBACnD,QAAQ,EAAE,GAAG;aACd;SACF,CAAC;QACF,MAAM,CAAC,cAAc,CAAC,IAAI,EAAE,eAAe,EAAE;YAC3C,KAAK,EAAE,IAAI,CAAC,EAAE,EAAE;YAChB,QAAQ,EAAE,IAAI;YACd,YAAY,EAAE,IAAI;YAClB,UAAU,EAAE,IAAI;SACjB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,eAAe,EAAE,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,cAAc,GAAG;YACrB,KAAK,EAAE,OAAO;YACd,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,IAAI;YACV,QAAQ,EAAE,QAAQ;YAClB,aAAa,EAAE,UAAU;SAC1B,CAAC;QACF,oFAAoF;QACpF,sBAAW,CAAC,qBAAqB,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;QACxD,sBAAW,CAAC,aAAa,CAAC,IAAuB,EAAE,cAAc,CAAC,CAAC;QAEnE,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,mBAAmB,EAAE,OAAO,CAAC,CAAC;QAChG,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC,6BAAkB,CAAC,iBAAiB,EAAE,IAAI,CAAC,CAAC;QAC3F,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,gBAAgB,EAAE,CAAC;QAC1C,MAAM,CAAC,iBAAO,CAAC,sBAAuB,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,EAAE,EAAE,MAAM,CAAC,gBAAgB,CAAC;YAC9F,CAAC,6BAAkB,CAAC,iBAAiB,CAAC,EAAE,6BAAkB,CAAC,uBAAuB;SACnF,CAAC,CAAC,CAAC;QACJ,MAAM,CAAC,iBAAO,CAAC,sBAAuB,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,EAAE,EAAE,MAAM,CAAC,gBAAgB,CAAC;YAC9F,CAAC,6BAAkB,CAAC,iBAAiB,CAAC,EAAE,6BAAkB,CAAC,wBAAwB;SACpF,CAAC,CAAC,CAAC;QACJ,MAAM,CAAC,iBAAO,CAAC,4BAA6B,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;QACtG,MAAM,CAAC,iBAAO,CAAC,SAAU,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;IACnF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,CAAC,cAAc,CAAC,IAAI,EAAE,YAAY,EAAE;YACxC,KAAK,EAAE,EAAE;YACT,QAAQ,EAAE,IAAI;YACd,YAAY,EAAE,IAAI;YAClB,UAAU,EAAE,IAAI;SACjB,CAAC,CAAC;QACH,MAAM,cAAc,GAAG;YACrB,aAAa,EAAE,UAAU;YACzB,KAAK,EAAE,OAAO;YACd,IAAI,EAAE,OAAO;YACb,IAAI,EAAE,SAAS;YACf,QAAQ,EAAE,QAAQ;SACnB,CAAC;QACF,sBAAW,CAAC,qBAAqB,CAAC,IAAuB,EAAE,cAAc,CAAC,CAAC;QAC3E,sBAAW,CAAC,aAAa,CAAC,IAAuB,EAAE,cAAc,CAAC,CAAC;QAEnE,MAAM,CAAC,iBAAO,CAAC,sBAAuB,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACtE,MAAM,CAAC,iBAAO,CAAC,4BAA6B,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC5E,MAAM,CAAC,iBAAO,CAAC,SAAU,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,6DAA6D,EAAE,GAAG,EAAE;QAC3E,UAAU,CAAC,GAAG,EAAE;YACd,iBAAO,CAAC,eAAe,EAAE,CAAC;YAC1B,iBAAO,CAAC,KAAK,CAAC,EAAE,QAAQ,EAAE,IAAA,2BAAe,GAAE,EAAE,YAAY,EAAE,uBAAuB,EAAE,WAAW,EAAE,SAAS,EAAE,eAAe,EAAE,SAAS,EAAE,YAAY,EAAE,KAAK,EAAE,qBAAqB,EAAE,IAAI,EAAE,cAAc,EAAE,KAAK,EAAE,aAAa,EAAE,KAAK,EAAS,CAAC,CAAC;YAChP,IAAI,CAAC,KAAK,CAAC,iBAAO,CAAC,sBAAuB,EAAE,QAAQ,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YACnF,IAAI,CAAC,KAAK,CAAC,iBAAO,CAAC,4BAA6B,EAAE,QAAQ,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;YACzF,IAAI,CAAC,KAAK,CAAC,iBAAO,CAAC,SAAU,EAAE,QAAQ,CAAC,CAAC,kBAAkB,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QACxE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;YAC1E,MAAM,IAAI,GAAa;gBACrB,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;gBACvB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;gBACpB,aAAa,EAAE,IAAI,CAAC,EAAE,EAAE;gBACxB,UAAU,EAAE;oBACV,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,GAAG;oBACnD,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,EAAE,SAAS;oBAC1D,QAAQ,EAAE,cAAc;iBACzB;aACF,CAAC;YACF,MAAM,cAAc,GAAG;gBACrB,KAAK,EAAE,OAAO;gBACd,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,cAAc;gBACpB,QAAQ,EAAE,QAAQ;gBAClB,aAAa,EAAE,UAAU;aAC1B,CAAC;YACF,sBAAW,CAAC,qBAAqB,CAAC,IAAuB,EAAE,cAAc,CAAC,CAAC;YAC3E,sBAAW,CAAC,aAAa,CAAC,IAAuB,EAAE,cAAc,CAAC,CAAC;YAEnE,MAAM,CAAC,iBAAO,CAAC,sBAAuB,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YACtE,MAAM,CAAC,iBAAO,CAAC,4BAA6B,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YAC5E,MAAM,CAAC,iBAAO,CAAC,SAAU,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,IAAI,GAAa;gBACrB,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;gBACvB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;gBACpB,aAAa,EAAE,IAAI,CAAC,EAAE,EAAE;gBACxB,UAAU,EAAE;oBACV,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,CAAC;oBACjD,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,EAAE,CAAC,CAAC;oBACnD,QAAQ,EAAE,CAAC,GAAG;iBACf;aACF,CAAC;YACF,MAAM,cAAc,GAAG;gBACrB,KAAK,EAAE,OAAO;gBACd,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,CAAC;gBACP,QAAQ,EAAE,QAAQ;gBAClB,aAAa,EAAE,UAAU;aAC1B,CAAC;YACF,sBAAW,CAAC,qBAAqB,CAAC,IAAuB,EAAE,cAAc,CAAC,CAAC;YAC3E,sBAAW,CAAC,aAAa,CAAC,IAAuB,EAAE,cAAc,CAAC,CAAC;YAEnE,MAAM,CAAC,iBAAO,CAAC,sBAAuB,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,CAAC,EAAE,MAAM,CAAC,gBAAgB,CAAC;gBAC7F,CAAC,6BAAkB,CAAC,iBAAiB,CAAC,EAAE,6BAAkB,CAAC,uBAAuB;aACnF,CAAC,CAAC,CAAC;YACJ,MAAM,CAAC,iBAAO,CAAC,sBAAuB,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,CAAC,CAAC,EAAE,MAAM,CAAC,gBAAgB,CAAC;gBAC9F,CAAC,6BAAkB,CAAC,iBAAiB,CAAC,EAAE,6BAAkB,CAAC,wBAAwB;aACpF,CAAC,CAAC,CAAC;YACJ,MAAM,CAAC,iBAAO,CAAC,4BAA6B,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;YACvG,MAAM,CAAC,iBAAO,CAAC,SAAU,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,CAAC,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;QAChF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,IAAI,GAAa;gBACrB,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;gBACvB,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE;gBACpB,aAAa,EAAE,IAAI,CAAC,EAAE,EAAE;gBACxB,UAAU,EAAE;oBACV,CAAC,6BAAkB,CAAC,yBAAyB,CAAC,EAAE,CAAC;oBACjD,CAAC,6BAAkB,CAAC,0BAA0B,CAAC,EAAE,CAAC;oBAClD,QAAQ,EAAE,CAAC;iBACZ;aACF,CAAC;YACF,MAAM,cAAc,GAAG;gBACrB,KAAK,EAAE,OAAO;gBACd,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,MAAM;gBACZ,QAAQ,EAAE,QAAQ;gBAClB,aAAa,EAAE,UAAU;aAC1B,CAAC;YACF,sBAAW,CAAC,qBAAqB,CAAC,IAAuB,EAAE,cAAc,CAAC,CAAC;YAC3E,sBAAW,CAAC,aAAa,CAAC,IAAuB,EAAE,cAAc,CAAC,CAAC;YAEnE,MAAM,CAAC,iBAAO,CAAC,SAAU,CAAC,MAAM,CAAC,CAAC,oBAAoB,CAAC,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;QACnF,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}