Add shouldEnableTruncation helper

Author: andreiborza

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

@@ -3,6 +3,7 @@
  */
 import { captureException } from '../../exports';
 import { getClient } from '../../currentScopes';
+import { hasSpanStreamingEnabled } from '../spans/hasSpanStreamingEnabled';
 import type { Span } from '../../types-hoist/span';
 import { isThenable } from '../../utils/is';
 import {
@@ -56,6 +57,16 @@ export function resolveAIRecordingOptions<T extends AIRecordingOptions>(options?
   } as T & Required<AIRecordingOptions>;
 }
 
+/**
+ * Resolves whether truncation should be enabled.
+ * If the user explicitly set `enableTruncation`, that value is used.
+ * Otherwise, truncation is disabled when span streaming is active.
+ */
+export function shouldEnableTruncation(enableTruncation: boolean | undefined): boolean {
+  const client = getClient();
+  return enableTruncation ?? !(client && hasSpanStreamingEnabled(client));
+}
+
 /**
  * Build method path from current traversal
  */

packages/core/src/tracing/openai/index.ts

@@ -1,4 +1,3 @@
-import { getClient } from '../../currentScopes';
 import { DEBUG_BUILD } from '../../debug-build';
 import { captureException } from '../../exports';
 import { SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN } from '../../semanticAttributes';
@@ -23,9 +22,9 @@ import {
   getJsonString,
   getTruncatedJsonString,
   resolveAIRecordingOptions,
+  shouldEnableTruncation,
   wrapPromiseWithMethods,
 } from '../ai/utils';
-import { hasSpanStreamingEnabled } from '../spans/hasSpanStreamingEnabled';
 import { OPENAI_METHOD_REGISTRY } from './constants';
 import { instrumentStream } from './streaming';
 import type { ChatCompletionChunk, OpenAiOptions, OpenAIStream, ResponseStreamingEvent } from './types';
@@ -172,9 +171,7 @@ function instrumentMethod<T extends unknown[], R>(
         originalResult = originalMethod.apply(context, args);
 
         if (options.recordInputs && params) {
-          const client = getClient();
-          const enableTruncation = options.enableTruncation ?? !(client && hasSpanStreamingEnabled(client));
-          addRequestAttributes(span, params, operationName, enableTruncation);
+          addRequestAttributes(span, params, operationName, shouldEnableTruncation(options.enableTruncation));
         }
 
         // Return async processing
@@ -212,9 +209,7 @@ function instrumentMethod<T extends unknown[], R>(
       originalResult = originalMethod.apply(context, args);
 
       if (options.recordInputs && params) {
-        const client = getClient();
-        const enableTruncation = options.enableTruncation ?? !(client && hasSpanStreamingEnabled(client));
-        addRequestAttributes(span, params, operationName, enableTruncation);
+        addRequestAttributes(span, params, operationName, shouldEnableTruncation(options.enableTruncation));
       }
 
       return originalResult.then(