.

Author: nicohrubec

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

@@ -94,7 +94,7 @@ function mapVercelAiOperationName(operationName: string): string {
  * Post-process spans emitted by the Vercel AI SDK.
  * This is supposed to be used in `client.on('spanStart', ...)
  */
-function onVercelAiSpanStart(span: Span, enableTruncation: boolean): void {
+function onVercelAiSpanStart(span: Span, client: Client): void {
   const { data: attributes, description: name } = spanToJSON(span);
 
   if (!name) {
@@ -114,6 +114,9 @@ function onVercelAiSpanStart(span: Span, enableTruncation: boolean): void {
     return;
   }
 
+  const integration = client.getIntegrationByName('VercelAI') as { options?: { enableTruncation?: boolean } } | undefined;
+  const enableTruncation = integration?.options?.enableTruncation ?? true;
+
   processGenerateSpan(span, name, attributes, enableTruncation);
 }
 
@@ -444,9 +447,8 @@ function processGenerateSpan(span: Span, name: string, attributes: SpanAttribute
 /**
  * Add event processors to the given client to process Vercel AI spans.
  */
-export function addVercelAiProcessors(client: Client, options?: { enableTruncation?: boolean }): void {
-  const enableTruncation = options?.enableTruncation ?? true;
-  client.on('spanStart', span => onVercelAiSpanStart(span, enableTruncation));
+export function addVercelAiProcessors(client: Client): void {
+  client.on('spanStart', span => onVercelAiSpanStart(span, client));
   // Note: We cannot do this on `spanEnd`, because the span cannot be mutated anymore at this point
   client.addEventProcessor(Object.assign(vercelAiEventProcessor, { id: 'VercelAiEventProcessor' }));
 }

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

@@ -30,11 +30,10 @@ const _vercelAIIntegration = ((options: VercelAiOptions = {}) => {
       // Note that this can only be detected if the 'Modules' integration is available, and running in CJS mode
       const shouldForce = options.force ?? shouldForceIntegration(client);
 
-      const processorOptions = { enableTruncation: options.enableTruncation };
       if (shouldForce) {
-        addVercelAiProcessors(client, processorOptions);
+        addVercelAiProcessors(client);
       } else {
-        instrumentation?.callWhenPatched(() => addVercelAiProcessors(client, processorOptions));
+        instrumentation?.callWhenPatched(() => addVercelAiProcessors(client));
       }
     },
   };