Use shouldEnableTruncation in Anthropic AI integration

Author: andreiborza

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

@@ -1,8 +1,6 @@
-import { getClient } from '../../currentScopes';
 import { captureException } from '../../exports';
 import { SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN } from '../../semanticAttributes';
 import { SPAN_STATUS_ERROR } from '../../tracing';
-import { hasSpanStreamingEnabled } from '../../tracing/spans/hasSpanStreamingEnabled';
 import { startSpan, startSpanManual } from '../../tracing/trace';
 import type { Span, SpanAttributeValue } from '../../types-hoist/span';
 import {
@@ -27,6 +25,7 @@ import {
   buildMethodPath,
   resolveAIRecordingOptions,
   setTokenUsageAttributes,
+  shouldEnableTruncation,
   wrapPromiseWithMethods,
 } from '../ai/utils';
 import { ANTHROPIC_METHOD_REGISTRY } from './constants';
@@ -208,9 +207,7 @@ function handleStreamingRequest<T extends unknown[], R>(
       originalResult = originalMethod.apply(context, args) as Promise<R>;
 
       if (options.recordInputs && params) {
-        const client = getClient();
-        const enableTruncation = options.enableTruncation ?? !(client && hasSpanStreamingEnabled(client));
-        addPrivateRequestAttributes(span, params, enableTruncation);
+        addPrivateRequestAttributes(span, params, shouldEnableTruncation(options.enableTruncation));
       }
 
       return (async () => {
@@ -232,9 +229,7 @@ function handleStreamingRequest<T extends unknown[], R>(
     return startSpanManual(spanConfig, span => {
       try {
         if (options.recordInputs && params) {
-          const client = getClient();
-          const enableTruncation = options.enableTruncation ?? !(client && hasSpanStreamingEnabled(client));
-          addPrivateRequestAttributes(span, params, enableTruncation);
+          addPrivateRequestAttributes(span, params, shouldEnableTruncation(options.enableTruncation));
         }
         const messageStream = target.apply(context, args);
         return instrumentMessageStream(messageStream, span, options.recordOutputs ?? false);
@@ -295,9 +290,7 @@ function instrumentMethod<T extends unknown[], R>(
           originalResult = target.apply(context, args) as Promise<R>;
 
           if (options.recordInputs && params) {
-            const client = getClient();
-            const enableTruncation = options.enableTruncation ?? !(client && hasSpanStreamingEnabled(client));
-            addPrivateRequestAttributes(span, params, enableTruncation);
+            addPrivateRequestAttributes(span, params, shouldEnableTruncation(options.enableTruncation));
           }
 
           return originalResult.then(