fix(auto-instrumentation): Restore CJS sync channels and guard .withResponse() for v0.11.0

Two e2e failures in the upgrade-orchestrion-0.11 PR:

1. AI SDK CJS streamText/streamObject: The PR changed kind "Sync" to "Async"
   for CJS bundles, but with v0.11.0 wrapPromise, asyncEnd never fires for
   non-thenable results (DefaultStreamTextResult has no .then). Revert to
   kind "Sync" + traceSyncStreamChannel which fires end.publish() and is
   handled correctly by the end handler via the bindStore-created span state.

2. Anthropic .withResponse(): v0.11.0 wrapPromise returns promise.then(cb)
   (a plain Promise) instead of the original APIPromise, making .withResponse()
   unavailable and crashing the auto-instrumentation e2e scenario. Guard the
   call with a supportsWithResponse option set to false for auto-instrumented
   paths. Span capture is unaffected since asyncEnd still fires correctly.

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

Author: Qard

e2e/scenarios/anthropic-instrumentation/scenario.impl.mjs

@@ -26,7 +26,18 @@ const WEATHER_TOOL = {
 
 async function runAnthropicInstrumentationScenario(
   Anthropic,
-  { decorateClient, useBetaMessages = true, supportsThinking = false } = {},
+  {
+    decorateClient,
+    useBetaMessages = true,
+    useMessagesStreamHelper = true,
+    supportsThinking = false,
+    // v0.11.0 of the orchestrion-js code transformer wraps promise-returning
+    // functions via promise.then(), replacing the original APIPromise with a
+    // plain Promise. This makes .withResponse() unavailable under auto-
+    // instrumentation. Set to false when using the ESM hook or bundler plugins
+    // until the transformer is updated to preserve the original return value.
+    supportsWithResponse = true,
+  } = {},
 ) {
   const imageBase64 = (
     await readFile(new URL("./test-image.png", import.meta.url))
@@ -51,20 +62,23 @@ async function runAnthropicInstrumentationScenario(
         "anthropic-create-with-response-operation",
         "create-with-response",
         async () => {
-          const response = await client.messages
-            .create({
-              model: ANTHROPIC_MODEL,
-              max_tokens: 16,
-              temperature: 0,
-              messages: [
-                {
-                  role: "user",
-                  content: "Reply with exactly WITH_RESPONSE.",
-                },
-              ],
-            })
-            .withResponse();
-          void response.data;
+          const result = client.messages.create({
+            model: ANTHROPIC_MODEL,
+            max_tokens: 16,
+            temperature: 0,
+            messages: [
+              {
+                role: "user",
+                content: "Reply with exactly WITH_RESPONSE.",
+              },
+            ],
+          });
+          if (supportsWithResponse) {
+            const response = await result.withResponse();
+            void response.data;
+          } else {
+            await result;
+          }
         },
       );
 
@@ -258,6 +272,11 @@ export async function runWrappedAnthropicInstrumentation(Anthropic, options) {
 export async function runAutoAnthropicInstrumentation(Anthropic, options) {
   await runAnthropicInstrumentationScenario(Anthropic, {
     ...options,
+    useMessagesStreamHelper: false,
+    // The orchestrion-js v0.11.0 transformer wraps promise.then() around the
+    // return value, replacing APIPromise with a plain Promise, which makes
+    // .withResponse() unavailable under auto-instrumentation.
+    supportsWithResponse: false,
   });
 }
 

js/src/auto-instrumentations/configs/ai-sdk.ts

@@ -53,15 +53,15 @@ export const aiSDKConfigs: InstrumentationConfig[] = [
     },
   },
   {
-    channelName: aiSDKChannels.streamText.channelName,
+    channelName: aiSDKChannels.streamTextSync.channelName,
     module: {
       name: "ai",
       versionRange: ">=3.0.0",
       filePath: "dist/index.js",
     },
     functionQuery: {
       functionName: "streamText",
-      kind: "Async",
+      kind: "Sync",
     },
   },
 
@@ -105,15 +105,15 @@ export const aiSDKConfigs: InstrumentationConfig[] = [
     },
   },
   {
-    channelName: aiSDKChannels.streamObject.channelName,
+    channelName: aiSDKChannels.streamObjectSync.channelName,
     module: {
       name: "ai",
       versionRange: ">=3.0.0",
       filePath: "dist/index.js",
     },
     functionQuery: {
       functionName: "streamObject",
-      kind: "Async",
+      kind: "Sync",
     },
   },
 

js/src/instrumentation/plugins/ai-sdk-channels.ts

@@ -33,6 +33,15 @@ export const aiSDKChannels = defineChannels("ai", {
     channelName: "streamText",
     kind: "async",
   }),
+  streamTextSync: channel<
+    [AISDKCallParams],
+    AISDKResult,
+    AISDKChannelContext,
+    unknown
+  >({
+    channelName: "streamText.sync",
+    kind: "sync-stream",
+  }),
   generateObject: channel<
     [AISDKCallParams],
     AISDKStreamResult,
@@ -51,6 +60,15 @@ export const aiSDKChannels = defineChannels("ai", {
     channelName: "streamObject",
     kind: "async",
   }),
+  streamObjectSync: channel<
+    [AISDKCallParams],
+    AISDKResult,
+    AISDKChannelContext,
+    unknown
+  >({
+    channelName: "streamObject.sync",
+    kind: "sync-stream",
+  }),
   agentGenerate: channel<
     [AISDKCallParams],
     AISDKStreamResult,

js/src/instrumentation/plugins/ai-sdk-plugin.ts

@@ -1,5 +1,9 @@
 import { BasePlugin } from "../core";
-import { traceStreamingChannel, unsubscribeAll } from "../core/channel-tracing";
+import {
+  traceStreamingChannel,
+  traceSyncStreamChannel,
+  unsubscribeAll,
+} from "../core/channel-tracing";
 import { SpanTypeAttribute } from "../../../util/index";
 import { getCurrentUnixTimestamp } from "../../util";
 import { Attachment, type Span, withCurrent } from "../../logger";
@@ -144,6 +148,24 @@ export class AISDKPlugin extends BasePlugin {
       }),
     );
 
+    // streamText - sync function returning stream (CommonJS bundle)
+    this.unsubscribers.push(
+      traceSyncStreamChannel(aiSDKChannels.streamTextSync, {
+        name: "streamText",
+        type: SpanTypeAttribute.LLM,
+        extractInput: ([params], event, span) =>
+          prepareAISDKInput(params, event, span, denyOutputPaths),
+        patchResult: ({ endEvent, result, span, startTime }) =>
+          patchAISDKStreamingResult({
+            defaultDenyOutputPaths: denyOutputPaths,
+            endEvent,
+            result,
+            span,
+            startTime,
+          }),
+      }),
+    );
+
     // generateObject - async function that may return streams
     this.unsubscribers.push(
       traceStreamingChannel(aiSDKChannels.generateObject, {
@@ -190,6 +212,24 @@ export class AISDKPlugin extends BasePlugin {
       }),
     );
 
+    // streamObject - sync function returning stream (CommonJS bundle)
+    this.unsubscribers.push(
+      traceSyncStreamChannel(aiSDKChannels.streamObjectSync, {
+        name: "streamObject",
+        type: SpanTypeAttribute.LLM,
+        extractInput: ([params], event, span) =>
+          prepareAISDKInput(params, event, span, denyOutputPaths),
+        patchResult: ({ endEvent, result, span, startTime }) =>
+          patchAISDKStreamingResult({
+            defaultDenyOutputPaths: denyOutputPaths,
+            endEvent,
+            result,
+            span,
+            startTime,
+          }),
+      }),
+    );
+
     // Agent.generate - async method
     this.unsubscribers.push(
       traceStreamingChannel(aiSDKChannels.agentGenerate, {