diff --git a/packages/core/src/tracing/ai/gen-ai-attributes.ts b/packages/core/src/tracing/ai/gen-ai-attributes.ts index 288e5ad9eeb8..16cdfa17b4b3 100644 --- a/packages/core/src/tracing/ai/gen-ai-attributes.ts +++ b/packages/core/src/tracing/ai/gen-ai-attributes.ts @@ -222,16 +222,6 @@ export const GEN_AI_EMBEDDINGS_INPUT_ATTRIBUTE = 'gen_ai.embeddings.input'; */ export const GEN_AI_EMBEDDINGS_OPERATION_ATTRIBUTE = 'gen_ai.embeddings'; -/** - * The span operation name for embedding - */ -export const GEN_AI_EMBED_DO_EMBED_OPERATION_ATTRIBUTE = 'gen_ai.embeddings'; - -/** - * The span operation name for embedding many - */ -export const GEN_AI_EMBED_MANY_DO_EMBED_OPERATION_ATTRIBUTE = 'gen_ai.embeddings'; - /** * The span operation name for reranking */ diff --git a/packages/core/src/tracing/vercel-ai/utils.ts b/packages/core/src/tracing/vercel-ai/utils.ts index 2a715deab764..3e7d87a25cea 100644 --- a/packages/core/src/tracing/vercel-ai/utils.ts +++ b/packages/core/src/tracing/vercel-ai/utils.ts @@ -1,8 +1,7 @@ import type { TraceContext } from '../../types-hoist/context'; import type { Span, SpanAttributes, SpanJSON } from '../../types-hoist/span'; import { - GEN_AI_EMBED_DO_EMBED_OPERATION_ATTRIBUTE, - GEN_AI_EMBED_MANY_DO_EMBED_OPERATION_ATTRIBUTE, + GEN_AI_EMBEDDINGS_OPERATION_ATTRIBUTE, GEN_AI_EXECUTE_TOOL_OPERATION_ATTRIBUTE, GEN_AI_GENERATE_CONTENT_OPERATION_ATTRIBUTE, GEN_AI_INPUT_MESSAGES_ATTRIBUTE, @@ -297,9 +296,8 @@ export function getSpanOpFromName(name: string): string | undefined { case 'ai.streamObject.doStream': return GEN_AI_GENERATE_CONTENT_OPERATION_ATTRIBUTE; case 'ai.embed.doEmbed': - return GEN_AI_EMBED_DO_EMBED_OPERATION_ATTRIBUTE; case 'ai.embedMany.doEmbed': - return GEN_AI_EMBED_MANY_DO_EMBED_OPERATION_ATTRIBUTE; + return GEN_AI_EMBEDDINGS_OPERATION_ATTRIBUTE; case 'ai.rerank.doRerank': return GEN_AI_RERANK_DO_RERANK_OPERATION_ATTRIBUTE; case 'ai.toolCall':