@@ -21,11 +21,6 @@ import {
2121 GEN_AI_USAGE_INPUT_TOKENS_ATTRIBUTE ,
2222 GEN_AI_USAGE_OUTPUT_TOKENS_ATTRIBUTE ,
2323 GEN_AI_USAGE_TOTAL_TOKENS_ATTRIBUTE ,
24- OPENAI_RESPONSE_ID_ATTRIBUTE ,
25- OPENAI_RESPONSE_MODEL_ATTRIBUTE ,
26- OPENAI_RESPONSE_TIMESTAMP_ATTRIBUTE ,
27- OPENAI_USAGE_COMPLETION_TOKENS_ATTRIBUTE ,
28- OPENAI_USAGE_PROMPT_TOKENS_ATTRIBUTE ,
2924} from '../../../../../packages/core/src/tracing/ai/gen-ai-attributes' ;
3025import { cleanupChildProcesses , createEsmAndCjsTests } from '../../../utils/runner' ;
3126
@@ -52,11 +47,6 @@ describe('OpenAI integration', () => {
5247 [ GEN_AI_USAGE_INPUT_TOKENS_ATTRIBUTE ] : 10 ,
5348 [ GEN_AI_USAGE_OUTPUT_TOKENS_ATTRIBUTE ] : 15 ,
5449 [ GEN_AI_USAGE_TOTAL_TOKENS_ATTRIBUTE ] : 25 ,
55- [ OPENAI_RESPONSE_ID_ATTRIBUTE ] : 'chatcmpl-mock123' ,
56- [ OPENAI_RESPONSE_MODEL_ATTRIBUTE ] : 'gpt-3.5-turbo' ,
57- [ OPENAI_RESPONSE_TIMESTAMP_ATTRIBUTE ] : '2023-03-01T06:31:28.000Z' ,
58- [ OPENAI_USAGE_COMPLETION_TOKENS_ATTRIBUTE ] : 15 ,
59- [ OPENAI_USAGE_PROMPT_TOKENS_ATTRIBUTE ] : 10 ,
6050 } ,
6151 description : 'chat gpt-3.5-turbo' ,
6252 op : 'gen_ai.chat' ,
@@ -77,11 +67,6 @@ describe('OpenAI integration', () => {
7767 [ GEN_AI_USAGE_INPUT_TOKENS_ATTRIBUTE ] : 5 ,
7868 [ GEN_AI_USAGE_OUTPUT_TOKENS_ATTRIBUTE ] : 8 ,
7969 [ GEN_AI_USAGE_TOTAL_TOKENS_ATTRIBUTE ] : 13 ,
80- [ OPENAI_RESPONSE_ID_ATTRIBUTE ] : 'resp_mock456' ,
81- [ OPENAI_RESPONSE_MODEL_ATTRIBUTE ] : 'gpt-3.5-turbo' ,
82- [ OPENAI_RESPONSE_TIMESTAMP_ATTRIBUTE ] : '2023-03-01T06:31:30.000Z' ,
83- [ OPENAI_USAGE_COMPLETION_TOKENS_ATTRIBUTE ] : 8 ,
84- [ OPENAI_USAGE_PROMPT_TOKENS_ATTRIBUTE ] : 5 ,
8570 } ,
8671 description : 'chat gpt-3.5-turbo' ,
8772 op : 'gen_ai.chat' ,
@@ -118,12 +103,7 @@ describe('OpenAI integration', () => {
118103 [ GEN_AI_USAGE_INPUT_TOKENS_ATTRIBUTE ] : 12 ,
119104 [ GEN_AI_USAGE_OUTPUT_TOKENS_ATTRIBUTE ] : 18 ,
120105 [ GEN_AI_USAGE_TOTAL_TOKENS_ATTRIBUTE ] : 30 ,
121- [ OPENAI_RESPONSE_ID_ATTRIBUTE ] : 'chatcmpl-stream-123' ,
122- [ OPENAI_RESPONSE_MODEL_ATTRIBUTE ] : 'gpt-4' ,
123106 [ GEN_AI_RESPONSE_STREAMING_ATTRIBUTE ] : true ,
124- [ OPENAI_RESPONSE_TIMESTAMP_ATTRIBUTE ] : '2023-03-01T06:31:40.000Z' ,
125- [ OPENAI_USAGE_COMPLETION_TOKENS_ATTRIBUTE ] : 18 ,
126- [ OPENAI_USAGE_PROMPT_TOKENS_ATTRIBUTE ] : 12 ,
127107 } ,
128108 description : 'chat gpt-4' ,
129109 op : 'gen_ai.chat' ,
@@ -145,12 +125,7 @@ describe('OpenAI integration', () => {
145125 [ GEN_AI_USAGE_INPUT_TOKENS_ATTRIBUTE ] : 6 ,
146126 [ GEN_AI_USAGE_OUTPUT_TOKENS_ATTRIBUTE ] : 10 ,
147127 [ GEN_AI_USAGE_TOTAL_TOKENS_ATTRIBUTE ] : 16 ,
148- [ OPENAI_RESPONSE_ID_ATTRIBUTE ] : 'resp_stream_456' ,
149- [ OPENAI_RESPONSE_MODEL_ATTRIBUTE ] : 'gpt-4' ,
150128 [ GEN_AI_RESPONSE_STREAMING_ATTRIBUTE ] : true ,
151- [ OPENAI_RESPONSE_TIMESTAMP_ATTRIBUTE ] : '2023-03-01T06:31:50.000Z' ,
152- [ OPENAI_USAGE_COMPLETION_TOKENS_ATTRIBUTE ] : 10 ,
153- [ OPENAI_USAGE_PROMPT_TOKENS_ATTRIBUTE ] : 6 ,
154129 } ,
155130 description : 'chat gpt-4' ,
156131 op : 'gen_ai.chat' ,
@@ -199,11 +174,6 @@ describe('OpenAI integration', () => {
199174 [ GEN_AI_USAGE_INPUT_TOKENS_ATTRIBUTE ] : 10 ,
200175 [ GEN_AI_USAGE_OUTPUT_TOKENS_ATTRIBUTE ] : 15 ,
201176 [ GEN_AI_USAGE_TOTAL_TOKENS_ATTRIBUTE ] : 25 ,
202- [ OPENAI_RESPONSE_ID_ATTRIBUTE ] : 'chatcmpl-mock123' ,
203- [ OPENAI_RESPONSE_MODEL_ATTRIBUTE ] : 'gpt-3.5-turbo' ,
204- [ OPENAI_RESPONSE_TIMESTAMP_ATTRIBUTE ] : '2023-03-01T06:31:28.000Z' ,
205- [ OPENAI_USAGE_COMPLETION_TOKENS_ATTRIBUTE ] : 15 ,
206- [ OPENAI_USAGE_PROMPT_TOKENS_ATTRIBUTE ] : 10 ,
207177 } ,
208178 description : 'chat gpt-3.5-turbo' ,
209179 op : 'gen_ai.chat' ,
@@ -227,11 +197,6 @@ describe('OpenAI integration', () => {
227197 [ GEN_AI_USAGE_INPUT_TOKENS_ATTRIBUTE ] : 5 ,
228198 [ GEN_AI_USAGE_OUTPUT_TOKENS_ATTRIBUTE ] : 8 ,
229199 [ GEN_AI_USAGE_TOTAL_TOKENS_ATTRIBUTE ] : 13 ,
230- [ OPENAI_RESPONSE_ID_ATTRIBUTE ] : 'resp_mock456' ,
231- [ OPENAI_RESPONSE_MODEL_ATTRIBUTE ] : 'gpt-3.5-turbo' ,
232- [ OPENAI_RESPONSE_TIMESTAMP_ATTRIBUTE ] : '2023-03-01T06:31:30.000Z' ,
233- [ OPENAI_USAGE_COMPLETION_TOKENS_ATTRIBUTE ] : 8 ,
234- [ OPENAI_USAGE_PROMPT_TOKENS_ATTRIBUTE ] : 5 ,
235200 } ,
236201 description : 'chat gpt-3.5-turbo' ,
237202 op : 'gen_ai.chat' ,
@@ -276,12 +241,7 @@ describe('OpenAI integration', () => {
276241 [ GEN_AI_USAGE_INPUT_TOKENS_ATTRIBUTE ] : 12 ,
277242 [ GEN_AI_USAGE_OUTPUT_TOKENS_ATTRIBUTE ] : 18 ,
278243 [ GEN_AI_USAGE_TOTAL_TOKENS_ATTRIBUTE ] : 30 ,
279- [ OPENAI_RESPONSE_ID_ATTRIBUTE ] : 'chatcmpl-stream-123' ,
280- [ OPENAI_RESPONSE_MODEL_ATTRIBUTE ] : 'gpt-4' ,
281244 [ GEN_AI_RESPONSE_STREAMING_ATTRIBUTE ] : true ,
282- [ OPENAI_RESPONSE_TIMESTAMP_ATTRIBUTE ] : '2023-03-01T06:31:40.000Z' ,
283- [ OPENAI_USAGE_COMPLETION_TOKENS_ATTRIBUTE ] : 18 ,
284- [ OPENAI_USAGE_PROMPT_TOKENS_ATTRIBUTE ] : 12 ,
285245 } ) ,
286246 description : 'chat gpt-4' ,
287247 op : 'gen_ai.chat' ,
@@ -307,12 +267,7 @@ describe('OpenAI integration', () => {
307267 [ GEN_AI_USAGE_INPUT_TOKENS_ATTRIBUTE ] : 6 ,
308268 [ GEN_AI_USAGE_OUTPUT_TOKENS_ATTRIBUTE ] : 10 ,
309269 [ GEN_AI_USAGE_TOTAL_TOKENS_ATTRIBUTE ] : 16 ,
310- [ OPENAI_RESPONSE_ID_ATTRIBUTE ] : 'resp_stream_456' ,
311- [ OPENAI_RESPONSE_MODEL_ATTRIBUTE ] : 'gpt-4' ,
312270 [ GEN_AI_RESPONSE_STREAMING_ATTRIBUTE ] : true ,
313- [ OPENAI_RESPONSE_TIMESTAMP_ATTRIBUTE ] : '2023-03-01T06:31:50.000Z' ,
314- [ OPENAI_USAGE_COMPLETION_TOKENS_ATTRIBUTE ] : 10 ,
315- [ OPENAI_USAGE_PROMPT_TOKENS_ATTRIBUTE ] : 6 ,
316271 } ) ,
317272 description : 'chat gpt-4' ,
318273 op : 'gen_ai.chat' ,
@@ -406,8 +361,6 @@ describe('OpenAI integration', () => {
406361 [ GEN_AI_RESPONSE_MODEL_ATTRIBUTE ] : 'text-embedding-3-small' ,
407362 [ GEN_AI_USAGE_INPUT_TOKENS_ATTRIBUTE ] : 10 ,
408363 [ GEN_AI_USAGE_TOTAL_TOKENS_ATTRIBUTE ] : 10 ,
409- [ OPENAI_RESPONSE_MODEL_ATTRIBUTE ] : 'text-embedding-3-small' ,
410- [ OPENAI_USAGE_PROMPT_TOKENS_ATTRIBUTE ] : 10 ,
411364 } ,
412365 description : 'embeddings text-embedding-3-small' ,
413366 op : 'gen_ai.embeddings' ,
@@ -448,8 +401,6 @@ describe('OpenAI integration', () => {
448401 [ GEN_AI_RESPONSE_MODEL_ATTRIBUTE ] : 'text-embedding-3-small' ,
449402 [ GEN_AI_USAGE_INPUT_TOKENS_ATTRIBUTE ] : 10 ,
450403 [ GEN_AI_USAGE_TOTAL_TOKENS_ATTRIBUTE ] : 10 ,
451- [ OPENAI_RESPONSE_MODEL_ATTRIBUTE ] : 'text-embedding-3-small' ,
452- [ OPENAI_USAGE_PROMPT_TOKENS_ATTRIBUTE ] : 10 ,
453404 } ,
454405 description : 'embeddings text-embedding-3-small' ,
455406 op : 'gen_ai.embeddings' ,
@@ -483,8 +434,6 @@ describe('OpenAI integration', () => {
483434 [ GEN_AI_RESPONSE_MODEL_ATTRIBUTE ] : 'text-embedding-3-small' ,
484435 [ GEN_AI_USAGE_INPUT_TOKENS_ATTRIBUTE ] : 10 ,
485436 [ GEN_AI_USAGE_TOTAL_TOKENS_ATTRIBUTE ] : 10 ,
486- [ OPENAI_RESPONSE_MODEL_ATTRIBUTE ] : 'text-embedding-3-small' ,
487- [ OPENAI_USAGE_PROMPT_TOKENS_ATTRIBUTE ] : 10 ,
488437 } ,
489438 description : 'embeddings text-embedding-3-small' ,
490439 op : 'gen_ai.embeddings' ,
@@ -542,11 +491,6 @@ describe('OpenAI integration', () => {
542491 [ GEN_AI_USAGE_INPUT_TOKENS_ATTRIBUTE ] : 10 ,
543492 [ GEN_AI_USAGE_OUTPUT_TOKENS_ATTRIBUTE ] : 15 ,
544493 [ GEN_AI_USAGE_TOTAL_TOKENS_ATTRIBUTE ] : 25 ,
545- [ OPENAI_RESPONSE_ID_ATTRIBUTE ] : 'chatcmpl-mock123' ,
546- [ OPENAI_RESPONSE_MODEL_ATTRIBUTE ] : 'gpt-3.5-turbo' ,
547- [ OPENAI_RESPONSE_TIMESTAMP_ATTRIBUTE ] : '2023-03-01T06:31:28.000Z' ,
548- [ OPENAI_USAGE_COMPLETION_TOKENS_ATTRIBUTE ] : 15 ,
549- [ OPENAI_USAGE_PROMPT_TOKENS_ATTRIBUTE ] : 10 ,
550494 } ,
551495 op : 'gen_ai.chat' ,
552496 origin : 'auto.ai.openai' ,
@@ -589,11 +533,6 @@ describe('OpenAI integration', () => {
589533 [ GEN_AI_USAGE_INPUT_TOKENS_ATTRIBUTE ] : 10 ,
590534 [ GEN_AI_USAGE_OUTPUT_TOKENS_ATTRIBUTE ] : 15 ,
591535 [ GEN_AI_USAGE_TOTAL_TOKENS_ATTRIBUTE ] : 25 ,
592- [ OPENAI_RESPONSE_ID_ATTRIBUTE ] : 'chatcmpl-mock123' ,
593- [ OPENAI_RESPONSE_MODEL_ATTRIBUTE ] : 'gpt-3.5-turbo' ,
594- [ OPENAI_RESPONSE_TIMESTAMP_ATTRIBUTE ] : '2023-03-01T06:31:28.000Z' ,
595- [ OPENAI_USAGE_COMPLETION_TOKENS_ATTRIBUTE ] : 15 ,
596- [ OPENAI_USAGE_PROMPT_TOKENS_ATTRIBUTE ] : 10 ,
597536 } ,
598537 op : 'gen_ai.chat' ,
599538 origin : 'auto.ai.openai' ,
0 commit comments