Skip to content

Commit 42c4da2

Browse files
authored
ref(core): Merge embeddings operations constants (#20095)
smol cleanup, three constants when there should be one
1 parent 7ded034 commit 42c4da2

File tree

2 files changed

+2
-14
lines changed

2 files changed

+2
-14
lines changed

packages/core/src/tracing/ai/gen-ai-attributes.ts

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -222,16 +222,6 @@ export const GEN_AI_EMBEDDINGS_INPUT_ATTRIBUTE = 'gen_ai.embeddings.input';
222222
*/
223223
export const GEN_AI_EMBEDDINGS_OPERATION_ATTRIBUTE = 'gen_ai.embeddings';
224224

225-
/**
226-
* The span operation name for embedding
227-
*/
228-
export const GEN_AI_EMBED_DO_EMBED_OPERATION_ATTRIBUTE = 'gen_ai.embeddings';
229-
230-
/**
231-
* The span operation name for embedding many
232-
*/
233-
export const GEN_AI_EMBED_MANY_DO_EMBED_OPERATION_ATTRIBUTE = 'gen_ai.embeddings';
234-
235225
/**
236226
* The span operation name for reranking
237227
*/

packages/core/src/tracing/vercel-ai/utils.ts

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,7 @@
11
import type { TraceContext } from '../../types-hoist/context';
22
import type { Span, SpanAttributes, SpanJSON } from '../../types-hoist/span';
33
import {
4-
GEN_AI_EMBED_DO_EMBED_OPERATION_ATTRIBUTE,
5-
GEN_AI_EMBED_MANY_DO_EMBED_OPERATION_ATTRIBUTE,
4+
GEN_AI_EMBEDDINGS_OPERATION_ATTRIBUTE,
65
GEN_AI_EXECUTE_TOOL_OPERATION_ATTRIBUTE,
76
GEN_AI_GENERATE_CONTENT_OPERATION_ATTRIBUTE,
87
GEN_AI_INPUT_MESSAGES_ATTRIBUTE,
@@ -301,9 +300,8 @@ export function getSpanOpFromName(name: string): string | undefined {
301300
case 'ai.streamObject.doStream':
302301
return GEN_AI_GENERATE_CONTENT_OPERATION_ATTRIBUTE;
303302
case 'ai.embed.doEmbed':
304-
return GEN_AI_EMBED_DO_EMBED_OPERATION_ATTRIBUTE;
305303
case 'ai.embedMany.doEmbed':
306-
return GEN_AI_EMBED_MANY_DO_EMBED_OPERATION_ATTRIBUTE;
304+
return GEN_AI_EMBEDDINGS_OPERATION_ATTRIBUTE;
307305
case 'ai.rerank.doRerank':
308306
return GEN_AI_RERANK_DO_RERANK_OPERATION_ATTRIBUTE;
309307
case 'ai.toolCall':

0 commit comments

Comments
 (0)