rename

Author: nicohrubec

packages/core/src/index.ts

@@ -146,7 +146,7 @@ export * as metrics from './metrics/public-api';
 export type { MetricOptions } from './metrics/public-api';
 export { createConsolaReporter } from './integrations/consola';
 export { addVercelAiProcessors } from './tracing/vercel-ai';
-export { _INTERNAL_getSpanContextForToolCallId, _INTERNAL_cleanupToolCallSpan } from './tracing/vercel-ai/utils';
+export { _INTERNAL_getSpanContextForToolCallId, _INTERNAL_cleanupToolCallSpanContext } from './tracing/vercel-ai/utils';
 export { instrumentOpenAiClient } from './tracing/openai';
 export { OPENAI_INTEGRATION_NAME } from './tracing/openai/constants';
 export { instrumentAnthropicAiClient } from './tracing/anthropic-ai';

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

@@ -84,7 +84,7 @@ export function _INTERNAL_getSpanContextForToolCallId(toolCallId: string): ToolC
 /**
  * Clean up the span mapping for a tool call ID
  */
-export function _INTERNAL_cleanupToolCallSpan(toolCallId: string): void {
+export function _INTERNAL_cleanupToolCallSpanContext(toolCallId: string): void {
   toolCallSpanContextMap.delete(toolCallId);
 }
 

packages/core/test/lib/tracing/vercel-ai-tool-call-span-map.test.ts

@@ -1,7 +1,7 @@
 import { beforeEach, describe, expect, it } from 'vitest';
 import { addVercelAiProcessors } from '../../../src/tracing/vercel-ai';
 import { toolCallSpanContextMap } from '../../../src/tracing/vercel-ai/constants';
-import { _INTERNAL_cleanupToolCallSpan, _INTERNAL_getSpanContextForToolCallId } from '../../../src/tracing/vercel-ai/utils';
+import { _INTERNAL_cleanupToolCallSpanContext, _INTERNAL_getSpanContextForToolCallId } from '../../../src/tracing/vercel-ai/utils';
 import {
   AI_TOOL_CALL_ID_ATTRIBUTE,
   AI_TOOL_CALL_NAME_ATTRIBUTE,
@@ -102,7 +102,7 @@ describe('vercel-ai tool call span context map', () => {
       spanId: 'span-id-1',
     });
 
-    _INTERNAL_cleanupToolCallSpan('tool-call-1');
+    _INTERNAL_cleanupToolCallSpanContext('tool-call-1');
     expect(_INTERNAL_getSpanContextForToolCallId('tool-call-1')).toBeUndefined();
   });
 });

packages/node/src/integrations/tracing/vercelai/instrumentation.ts

@@ -1,7 +1,7 @@
 import type { InstrumentationConfig, InstrumentationModuleDefinition } from '@opentelemetry/instrumentation';
 import { InstrumentationBase, InstrumentationNodeModuleDefinition } from '@opentelemetry/instrumentation';
 import {
-  _INTERNAL_cleanupToolCallSpan,
+  _INTERNAL_cleanupToolCallSpanContextContext,
   _INTERNAL_getSpanContextForToolCallId,
   addNonEnumerableProperty,
   captureException,
@@ -99,7 +99,7 @@ function checkResultForToolErrors(result: unknown): void {
   for (const item of resultObj.content) {
     // Successful tool calls should not keep toolCallId -> span context mappings alive.
     if (isToolResultPart(item)) {
-      _INTERNAL_cleanupToolCallSpan(item.toolCallId);
+      _INTERNAL_cleanupToolCallSpanContext(item.toolCallId);
       continue;
     }
 
@@ -149,7 +149,7 @@ function checkResultForToolErrors(result: unknown): void {
 
     // Clean up the span mapping since we've processed this tool error
     // We won't get multiple { type: 'tool-error' } parts for the same toolCallId.
-    _INTERNAL_cleanupToolCallSpan(item.toolCallId);
+    _INTERNAL_cleanupToolCallSpanContext(item.toolCallId);
   }
 }