@morsehq-dev/sdk 0.5.0-rc.4 → 0.5.0-rc.5

package/README.md

@@ -134,26 +134,31 @@ sibling `memory` span per call.
 
 ### `@morsehq-dev/sdk/anthropic-agent-sdk` — Claude Agent SDK
 
-Wraps `ClaudeSDKClient` from `@anthropic-ai/claude-agent-sdk`. Emits an
-outer **agent** span, per-call **llm** spans (with `context_segments`),
-**tool** spans, **subagent.spawn** spans for Task-tool fan-out, and
-**hook** spans for `pre_tool` / `post_tool` / `on_error` hooks.
+Wrap the package's top-level `query` and call the returned function in its
+place. Emits an outer **agent** span, per-call **llm** spans (with
+`context_segments`), **tool** spans, **subagent.spawn** spans for Task-tool
+fan-out, and **hook** spans.
 
 ```ts
-import { ClaudeSDKClient } from "@anthropic-ai/claude-agent-sdk";
-import { wrapClaudeAgentSDKWithFullInstrumentation } from "@morsehq-dev/sdk/anthropic-agent-sdk";
+import { query } from "@anthropic-ai/claude-agent-sdk";
+import { wrapClaudeAgentQuery } from "@morsehq-dev/sdk/anthropic-agent-sdk";
 
-const raw = new ClaudeSDKClient({
-  /* ... */
-});
-const { client, dispose } = wrapClaudeAgentSDKWithFullInstrumentation(raw);
+const tracedQuery = wrapClaudeAgentQuery(query);
 
-for await (const event of client.query("research the 2026 RLHF landscape")) {
+for await (const message of tracedQuery({ prompt: "research the 2026 RLHF landscape" })) {
   // your code unchanged — spans flow to Morse as a side-effect
 }
-dispose();
 ```
 
+The returned value is still a `Query`: `interrupt()`, `setPermissionMode()`
+and every other control method are forwarded to the real one, so wrapping
+costs you nothing at the call site.
+
+Why a wrapper rather than zero-config auto-detect: the package's `query`
+export is read-only and non-configurable, so there is nothing to patch in
+place — the same constraint the Vercel AI adapter hit. Auto-detect notices
+the package and logs a one-time hint pointing here.
+
 ### `@morsehq-dev/sdk/openai-agents` — OpenAI Agents SDK
 
 Wraps `Runner` from `@openai/agents`. Emits an outer **agent** span,

package/dist/anthropic/index.cjs

@@ -1,41 +1,41 @@
 'use strict';
 
-var chunkVM3QM22Z_cjs = require('../chunk-VM3QM22Z.cjs');
-require('../chunk-HZKXILBF.cjs');
-require('../chunk-RDFDPK2X.cjs');
+var chunkDPW4W7H7_cjs = require('../chunk-DPW4W7H7.cjs');
+require('../chunk-Z5MFR7UE.cjs');
+require('../chunk-RRE2LX2A.cjs');
 require('../chunk-KBWPNIH4.cjs');
 require('../chunk-HICEJ4LV.cjs');
-require('../chunk-HA5RY4N5.cjs');
+require('../chunk-DQ4T2MJE.cjs');
 
 
 
 Object.defineProperty(exports, "MAX_SEGMENT_CONTENT_BYTES", {
   enumerable: true,
-  get: function () { return chunkVM3QM22Z_cjs.MAX_SEGMENT_CONTENT_BYTES; }
+  get: function () { return chunkDPW4W7H7_cjs.MAX_SEGMENT_CONTENT_BYTES; }
 });
 Object.defineProperty(exports, "_resetAnthropicInstallStateForTesting", {
   enumerable: true,
-  get: function () { return chunkVM3QM22Z_cjs._resetAnthropicInstallStateForTesting; }
+  get: function () { return chunkDPW4W7H7_cjs._resetAnthropicInstallStateForTesting; }
 });
 Object.defineProperty(exports, "extractAnthropicSegments", {
   enumerable: true,
-  get: function () { return chunkVM3QM22Z_cjs.extractAnthropicSegments; }
+  get: function () { return chunkDPW4W7H7_cjs.extractAnthropicSegments; }
 });
 Object.defineProperty(exports, "instrumentAnthropic", {
   enumerable: true,
-  get: function () { return chunkVM3QM22Z_cjs.instrumentAnthropic; }
+  get: function () { return chunkDPW4W7H7_cjs.instrumentAnthropic; }
 });
 Object.defineProperty(exports, "isAnthropicWrapped", {
   enumerable: true,
-  get: function () { return chunkVM3QM22Z_cjs.isAnthropicWrapped; }
+  get: function () { return chunkDPW4W7H7_cjs.isAnthropicWrapped; }
 });
 Object.defineProperty(exports, "uninstallAnthropic", {
   enumerable: true,
-  get: function () { return chunkVM3QM22Z_cjs.uninstallAnthropic; }
+  get: function () { return chunkDPW4W7H7_cjs.uninstallAnthropic; }
 });
 Object.defineProperty(exports, "wrapAnthropic", {
   enumerable: true,
-  get: function () { return chunkVM3QM22Z_cjs.wrapAnthropic; }
+  get: function () { return chunkDPW4W7H7_cjs.wrapAnthropic; }
 });
 //# sourceMappingURL=index.cjs.map
 //# sourceMappingURL=index.cjs.map

package/dist/anthropic/index.js

@@ -1,8 +1,8 @@
-export { MAX_SEGMENT_CONTENT_BYTES, _resetAnthropicInstallStateForTesting, extractAnthropicSegments, instrumentAnthropic, isAnthropicWrapped, uninstallAnthropic, wrapAnthropic } from '../chunk-2O2FFJLK.js';
-import '../chunk-YPP5DDPO.js';
-import '../chunk-Q6NQQ2LO.js';
+export { MAX_SEGMENT_CONTENT_BYTES, _resetAnthropicInstallStateForTesting, extractAnthropicSegments, instrumentAnthropic, isAnthropicWrapped, uninstallAnthropic, wrapAnthropic } from '../chunk-NR36TJMT.js';
+import '../chunk-QZDUGEIR.js';
+import '../chunk-YKPZQPEW.js';
 import '../chunk-XTKMUJWI.js';
 import '../chunk-4JPZV2VZ.js';
-import '../chunk-ELBY4SJ7.js';
+import '../chunk-ELVBOJWY.js';
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map

package/dist/anthropic-agent-sdk/index.cjs

@@ -1,19 +1,17 @@
 'use strict';
 
-var chunkVM3QM22Z_cjs = require('../chunk-VM3QM22Z.cjs');
-require('../chunk-HZKXILBF.cjs');
-var chunk7ZNGY236_cjs = require('../chunk-7ZNGY236.cjs');
-var chunkWLNADB4J_cjs = require('../chunk-WLNADB4J.cjs');
-var chunkRDFDPK2X_cjs = require('../chunk-RDFDPK2X.cjs');
+var chunkDPW4W7H7_cjs = require('../chunk-DPW4W7H7.cjs');
+require('../chunk-Z5MFR7UE.cjs');
+var chunkDGBQ3UI6_cjs = require('../chunk-DGBQ3UI6.cjs');
+var chunkSJ625SQS_cjs = require('../chunk-SJ625SQS.cjs');
+var chunkRRE2LX2A_cjs = require('../chunk-RRE2LX2A.cjs');
 var chunkKBWPNIH4_cjs = require('../chunk-KBWPNIH4.cjs');
 require('../chunk-HICEJ4LV.cjs');
-var chunkHA5RY4N5_cjs = require('../chunk-HA5RY4N5.cjs');
+var chunkDQ4T2MJE_cjs = require('../chunk-DQ4T2MJE.cjs');
 
 // src/anthropic-agent-sdk/client-wrapper.ts
-chunkHA5RY4N5_cjs.init_tracing();
+chunkDQ4T2MJE_cjs.init_tracing();
 var WRAPPED_CLIENT = /* @__PURE__ */ Symbol.for("morse.anthropic-agent-sdk.wrapped");
-var INSTALL_SENTINEL = /* @__PURE__ */ Symbol.for("morse.anthropic-agent-sdk.installed");
-var ORIGINALS_SENTINEL = /* @__PURE__ */ Symbol.for("morse.anthropic-agent-sdk.originals");
 function wrapClaudeAgentSDK(client, options = {}) {
   if (!client || typeof client !== "object") return client;
   if (client[WRAPPED_CLIENT]) return client;
@@ -36,48 +34,13 @@ function wrapClaudeAgentSDK(client, options = {}) {
   });
   return wrapper;
 }
-var instrumentClaudeAgentSDK = chunkRDFDPK2X_cjs.absorbErrorsAsync(
-  async (clientOrOptions = void 0, maybeOptions = {}) => {
-    if (clientOrOptions && isClient(clientOrOptions)) {
-      const opts2 = maybeOptions ?? {};
-      return wrapClaudeAgentSDK(clientOrOptions, opts2);
-    }
-    const opts = clientOrOptions ?? {};
-    const mod = await loadClaudeAgentSDKModule(opts.claudeAgentSdkModule);
-    if (!mod) return false;
-    const installed = mod[INSTALL_SENTINEL] === true;
-    if (installed) return true;
-    const carrier = findClientPrototypeCarrier(mod);
-    if (!carrier) return false;
-    const originalQuery = carrier.prototype.query;
-    if (typeof originalQuery !== "function") return false;
-    const patchedQuery = buildPatchedQuery(function(prompt, callOptions) {
-      return originalQuery.call(this, prompt, callOptions);
-    }, opts);
-    carrier.prototype.query = patchedQuery;
-    mod[ORIGINALS_SENTINEL] = {
-      protoQuery: originalQuery,
-      protoCarrier: carrier
-    };
-    mod[INSTALL_SENTINEL] = true;
-    return true;
+var instrumentClaudeAgentSDK = chunkRRE2LX2A_cjs.absorbErrorsAsync(
+  async (client = void 0, maybeOptions = {}) => {
+    if (client && isClient(client)) return wrapClaudeAgentSDK(client, maybeOptions);
+    return false;
   },
   { fingerprint: "morse.adapters.anthropic-agent-sdk.install" }
 );
-var uninstallClaudeAgentSDK = chunkRDFDPK2X_cjs.absorbErrorsSync(
-  (claudeAgentSdkModule) => {
-    if (!claudeAgentSdkModule || typeof claudeAgentSdkModule !== "object") return;
-    const mod = claudeAgentSdkModule;
-    if (!mod[INSTALL_SENTINEL]) return;
-    const originals = mod[ORIGINALS_SENTINEL];
-    if (originals?.protoCarrier && originals.protoQuery !== void 0) {
-      originals.protoCarrier.prototype.query = originals.protoQuery;
-    }
-    mod[ORIGINALS_SENTINEL] = void 0;
-    mod[INSTALL_SENTINEL] = false;
-  },
-  { fingerprint: "morse.adapters.anthropic-agent-sdk.uninstall" }
-);
 function isClaudeAgentSDKWrapped(client) {
   if (!client || typeof client !== "object") return false;
   return client[WRAPPED_CLIENT] === true;
@@ -87,36 +50,14 @@ function isClient(x) {
   const q = x.query;
   return typeof q === "function";
 }
-async function loadClaudeAgentSDKModule(explicit) {
-  if (explicit && typeof explicit === "object") return explicit;
-  try {
-    return await Function("m", "return import(m)")("@anthropic-ai/claude-agent-sdk");
-  } catch {
-    return null;
-  }
-}
-function findClientPrototypeCarrier(mod) {
-  if (!mod || typeof mod !== "object") return null;
-  const m = mod;
-  const candidates = [
-    m.ClaudeSDKClient,
-    m.default?.ClaudeSDKClient
-  ];
-  for (const c of candidates) {
-    if (c && typeof c === "function" && c.prototype && typeof c.prototype.query === "function") {
-      return c;
-    }
-  }
-  return null;
-}
 function buildPatchedQuery(original, options) {
   const defaultAgentName = options.defaultAgentName ?? "claude-agent-sdk";
   const defaultModel = options.defaultModel;
   return function patchedQuery(prompt, callOptions) {
     const boundOriginal = (p, o) => original.call(this, p, o);
     const model = (callOptions && typeof callOptions.model === "string" ? callOptions.model : void 0) ?? defaultModel;
-    async function* iterateInTrace() {
-      const handle = chunkWLNADB4J_cjs.emitAgentSpan({
+    async function* iterateInTrace2() {
+      const handle = chunkSJ625SQS_cjs.emitAgentSpan({
         adapter: "anthropic-agent-sdk",
         agentName: defaultAgentName,
         model
@@ -139,8 +80,8 @@ function buildPatchedQuery(original, options) {
       }
       handle?.end();
     }
-    if (chunkHA5RY4N5_cjs.getActiveTraceState() !== null) {
-      return iterateInTrace();
+    if (chunkDQ4T2MJE_cjs.getActiveTraceState() !== null) {
+      return iterateInTrace2();
     }
     return (async function* outerTrace() {
       const queue = [];
@@ -155,9 +96,9 @@ function buildPatchedQuery(original, options) {
           };
         }
       });
-      const runPromise = chunkHA5RY4N5_cjs.runAsync({ agentName: defaultAgentName }, async (runHandle) => {
+      const runPromise = chunkDQ4T2MJE_cjs.runAsync({ agentName: defaultAgentName }, async (runHandle) => {
         try {
-          for await (const event of iterateInTrace()) {
+          for await (const event of iterateInTrace2()) {
             queue.push({ kind: "value", value: event });
             notify?.();
           }
@@ -191,10 +132,10 @@ function buildPatchedQuery(original, options) {
 }
 
 // src/_agent-sdk-common/context-propagation.ts
-chunkHA5RY4N5_cjs.init_tracing();
-var captureParentForFanOut = chunkRDFDPK2X_cjs.absorbErrorsSync(
+chunkDQ4T2MJE_cjs.init_tracing();
+var captureParentForFanOut = chunkRRE2LX2A_cjs.absorbErrorsSync(
   () => {
-    const state = chunkHA5RY4N5_cjs.getActiveTraceState();
+    const state = chunkDQ4T2MJE_cjs.getActiveTraceState();
     if (state === null) return null;
     const top = state.spanStack[state.spanStack.length - 1];
     if (!top) return null;
@@ -205,13 +146,13 @@ var captureParentForFanOut = chunkRDFDPK2X_cjs.absorbErrorsSync(
   },
   { fingerprint: "agent-sdk-common.captureParentForFanOut" }
 );
-var emitWithPinnedParent = chunkRDFDPK2X_cjs.absorbErrorsSync(
+var emitWithPinnedParent = chunkRRE2LX2A_cjs.absorbErrorsSync(
   (opts) => {
-    const state = chunkHA5RY4N5_cjs.getActiveTraceState();
+    const state = chunkDQ4T2MJE_cjs.getActiveTraceState();
     if (state === null) return null;
     const snap = opts.snapshot;
     if (!snap || snap.traceId !== state.traceId) return null;
-    const handle = chunkHA5RY4N5_cjs.startSpan({
+    const handle = chunkDQ4T2MJE_cjs.startSpan({
       name: opts.name,
       type: opts.type,
       metadata: opts.metadata
@@ -233,13 +174,13 @@ function findSpanById(state, spanId) {
   }
   return void 0;
 }
-var emitAsToolSibling = chunkRDFDPK2X_cjs.absorbErrorsSync(
+var emitAsToolSibling = chunkRRE2LX2A_cjs.absorbErrorsSync(
   (opts) => {
-    const state = chunkHA5RY4N5_cjs.getActiveTraceState();
+    const state = chunkDQ4T2MJE_cjs.getActiveTraceState();
     if (state === null) return null;
     const tool = findSpanById(state, opts.toolSpanId);
     if (!tool) return null;
-    const handle = chunkHA5RY4N5_cjs.startSpan({
+    const handle = chunkDQ4T2MJE_cjs.startSpan({
       name: opts.name,
       type: opts.type,
       metadata: opts.metadata
@@ -250,9 +191,9 @@ var emitAsToolSibling = chunkRDFDPK2X_cjs.absorbErrorsSync(
   },
   { fingerprint: "agent-sdk-common.emitAsToolSibling" }
 );
-chunkRDFDPK2X_cjs.absorbErrorsSync(
+chunkRRE2LX2A_cjs.absorbErrorsSync(
   () => {
-    const state = chunkHA5RY4N5_cjs.getActiveTraceState();
+    const state = chunkDQ4T2MJE_cjs.getActiveTraceState();
     if (state === null) return null;
     return {
       traceId: state.traceId,
@@ -297,7 +238,7 @@ var _handleHookEvent = (payload, options) => {
       return;
     }
   }
-  const fallback = chunkWLNADB4J_cjs.emitHookSpan({
+  const fallback = chunkSJ625SQS_cjs.emitHookSpan({
     adapter: "anthropic-agent-sdk",
     hookName: name,
     hookKind: kind,
@@ -307,7 +248,7 @@ var _handleHookEvent = (payload, options) => {
     fallback.end();
   }
 };
-var handleHookEvent = chunkRDFDPK2X_cjs.absorbErrorsSync(_handleHookEvent, {
+var handleHookEvent = chunkRRE2LX2A_cjs.absorbErrorsSync(_handleHookEvent, {
   fingerprint: "morse.anthropic-agent-sdk.hook-instrumentation.handle"
 });
 function installHookInstrumentation(client, options) {
@@ -335,7 +276,7 @@ function installHookInstrumentation(client, options) {
 }
 
 // src/anthropic-agent-sdk/llm-instrumentation.ts
-chunkHA5RY4N5_cjs.init_diagnostics();
+chunkDQ4T2MJE_cjs.init_diagnostics();
 function readInputTokens(usage) {
   if (usage === void 0 || usage === null || typeof usage !== "object") return void 0;
   const base = typeof usage.input_tokens === "number" && Number.isFinite(usage.input_tokens) ? usage.input_tokens : 0;
@@ -402,7 +343,7 @@ async function* instrumentLlmCalls(events, options) {
           const reportedModel = typeof event.message?.model === "string" && event.message.model.length > 0 ? event.message.model : void 0;
           const model = reportedModel ?? fallbackModel;
           if (!model) return;
-          const handle = chunkWLNADB4J_cjs.emitLlmSpan({ adapter: "anthropic-agent-sdk", model });
+          const handle = chunkSJ625SQS_cjs.emitLlmSpan({ adapter: "anthropic-agent-sdk", model });
           if (!handle) return;
           const usage = event.message?.usage;
           const inputTokens = readInputTokens(usage);
@@ -419,7 +360,7 @@ async function* instrumentLlmCalls(events, options) {
           applyCost(handle, model, inputTokens, outputTokens, cacheWriteTokens, cachedInputTokens);
           if (requestParams !== void 0) {
             try {
-              const segments = chunkVM3QM22Z_cjs.extractAnthropicSegments(
+              const segments = chunkDPW4W7H7_cjs.extractAnthropicSegments(
                 requestParams,
                 event.message
               );
@@ -442,7 +383,7 @@ async function* instrumentLlmCalls(events, options) {
           if (model === void 0 || model.length === 0) {
             return;
           }
-          openSpan = chunkWLNADB4J_cjs.emitLlmSpan({
+          openSpan = chunkSJ625SQS_cjs.emitLlmSpan({
             adapter: "anthropic-agent-sdk",
             model
           });
@@ -484,7 +425,7 @@ async function* instrumentLlmCalls(events, options) {
             }
             if (requestParams !== void 0) {
               try {
-                const segments = chunkVM3QM22Z_cjs.extractAnthropicSegments(
+                const segments = chunkDPW4W7H7_cjs.extractAnthropicSegments(
                   requestParams,
                   event.message
                 );
@@ -512,13 +453,13 @@ async function* instrumentLlmCalls(events, options) {
 }
 function reportSwallowed(err, fingerprint) {
   try {
-    chunkHA5RY4N5_cjs.report(err, { fingerprint });
+    chunkDQ4T2MJE_cjs.report(err, { fingerprint });
   } catch {
   }
 }
 
 // src/anthropic-agent-sdk/tool-and-subagent.ts
-chunkHA5RY4N5_cjs.init_diagnostics();
+chunkDQ4T2MJE_cjs.init_diagnostics();
 var ADAPTER = "anthropic-agent-sdk";
 var TASK_TOOL_NAME = "Task";
 function safeCloseFailed(handle, message) {
@@ -564,7 +505,7 @@ async function* instrumentTools(events, options) {
         tool_use_id: toolUseId
       };
       if (taskPrompt !== void 0) {
-        metadata.task_description = chunk7ZNGY236_cjs.summarizeToolInput(taskPrompt, inputOpts);
+        metadata.task_description = chunkDGBQ3UI6_cjs.summarizeToolInput(taskPrompt, inputOpts);
       }
       let handle2 = null;
       try {
@@ -586,10 +527,10 @@ async function* instrumentTools(events, options) {
     }
     let handle = null;
     try {
-      handle = chunkWLNADB4J_cjs.emitToolSpan({
+      handle = chunkSJ625SQS_cjs.emitToolSpan({
         adapter: ADAPTER,
         toolName: name,
-        inputSummary: chunk7ZNGY236_cjs.summarizeToolInput(event.input, inputOpts),
+        inputSummary: chunkDGBQ3UI6_cjs.summarizeToolInput(event.input, inputOpts),
         attributes: { tool_use_id: toolUseId }
       });
     } catch {
@@ -606,7 +547,7 @@ async function* instrumentTools(events, options) {
     open.delete(toolUseId);
     const isError = event.is_error === true;
     try {
-      const outputSummary = chunk7ZNGY236_cjs.summarizeToolOutput(event.content, outputOpts);
+      const outputSummary = chunkDGBQ3UI6_cjs.summarizeToolOutput(event.content, outputOpts);
       entry.handle.span.metadata = {
         ...entry.handle.span.metadata ?? {},
         tool_output_summary: outputSummary
@@ -678,11 +619,124 @@ async function* instrumentTools(events, options) {
 }
 function reportSwallowed2(err, fingerprint) {
   try {
-    chunkHA5RY4N5_cjs.report(err, { fingerprint });
+    chunkDQ4T2MJE_cjs.report(err, { fingerprint });
   } catch {
   }
 }
 
+// src/anthropic-agent-sdk/trace-bridge.ts
+chunkDQ4T2MJE_cjs.init_tracing();
+function iterateInTrace(agentName, makeGenerator) {
+  if (chunkDQ4T2MJE_cjs.getActiveTraceState() !== null) {
+    return makeGenerator();
+  }
+  return (async function* outerTrace() {
+    const queue = [];
+    let notify = null;
+    const waitForItem = () => new Promise((resolve) => {
+      if (queue.length > 0) {
+        resolve();
+      } else {
+        notify = () => {
+          notify = null;
+          resolve();
+        };
+      }
+    });
+    const runPromise = chunkDQ4T2MJE_cjs.runAsync({ agentName }, async (runHandle) => {
+      try {
+        for await (const event of makeGenerator()) {
+          queue.push({ kind: "value", value: event });
+          notify?.();
+        }
+        runHandle.setOutcome(true);
+      } catch (err) {
+        queue.push({ kind: "error", error: err });
+        notify?.();
+        throw err;
+      } finally {
+        queue.push({ kind: "end" });
+        notify?.();
+      }
+      return null;
+    });
+    runPromise.catch(() => {
+    });
+    while (true) {
+      if (queue.length === 0) await waitForItem();
+      const item = queue.shift();
+      if (!item) continue;
+      if (item.kind === "value") {
+        yield item.value;
+      } else if (item.kind === "error") {
+        throw item.error;
+      } else {
+        return;
+      }
+    }
+  })();
+}
+
+// src/anthropic-agent-sdk/query-wrapper.ts
+var WRAPPED = /* @__PURE__ */ Symbol.for("morse.claudeAgentQuery.wrapped");
+function isClaudeAgentQueryWrapped(fn) {
+  return typeof fn === "function" && fn[WRAPPED] === true;
+}
+function wrapClaudeAgentQuery(query, options = {}) {
+  if (typeof query !== "function" || isClaudeAgentQueryWrapped(query)) return query;
+  const traced = function tracedQuery(params) {
+    const real = query(params);
+    const model = typeof params?.options?.model === "string" ? params.options.model : options.defaultModel;
+    let handle = null;
+    let opened = false;
+    async function* withAgentSpan() {
+      if (!opened) {
+        opened = true;
+        handle = chunkSJ625SQS_cjs.emitAgentSpan({
+          adapter: "anthropic-agent-sdk",
+          agentName: options.defaultAgentName ?? "claude-agent-sdk",
+          model
+        });
+      }
+      try {
+        for await (const message of real) {
+          yield message;
+        }
+      } catch (err) {
+        handle?.end(err);
+        handle = null;
+        throw err;
+      }
+      handle?.end();
+      handle = null;
+    }
+    const tapOpts = {
+      toolInputMaxBytes: options.toolInputMaxBytes,
+      toolOutputMaxBytes: options.toolOutputMaxBytes
+    };
+    const instrumented = iterateInTrace(
+      options.defaultAgentName ?? "claude-agent-sdk",
+      () => instrumentTools(
+        instrumentLlmCalls(withAgentSpan(), { fallbackModel: model }),
+        tapOpts
+      )
+    );
+    return new Proxy(real, {
+      get(target, prop, receiver) {
+        if (prop === Symbol.asyncIterator) return () => instrumented;
+        if (prop === "next" || prop === "return" || prop === "throw") {
+          const fn = instrumented[prop];
+          return typeof fn === "function" ? fn.bind(instrumented) : fn;
+        }
+        const value = Reflect.get(target, prop, receiver);
+        return typeof value === "function" ? value.bind(target) : value;
+      }
+    });
+  };
+  Object.defineProperty(traced, WRAPPED, { value: true, enumerable: false });
+  return traced;
+}
+
 // src/anthropic-agent-sdk/index.ts
 var FULL_WRAP_SENTINEL = /* @__PURE__ */ Symbol.for("morse.anthropic-agent-sdk.full-wrapped");
 function wrapClaudeAgentSDKWithFullInstrumentation(client, options = {}) {
@@ -754,8 +808,9 @@ exports.installHookInstrumentation = installHookInstrumentation;
 exports.instrumentClaudeAgentSDK = instrumentClaudeAgentSDK;
 exports.instrumentLlmCalls = instrumentLlmCalls;
 exports.instrumentTools = instrumentTools;
+exports.isClaudeAgentQueryWrapped = isClaudeAgentQueryWrapped;
 exports.isClaudeAgentSDKWrapped = isClaudeAgentSDKWrapped;
-exports.uninstallClaudeAgentSDK = uninstallClaudeAgentSDK;
+exports.wrapClaudeAgentQuery = wrapClaudeAgentQuery;
 exports.wrapClaudeAgentSDK = wrapClaudeAgentSDK;
 exports.wrapClaudeAgentSDKWithFullInstrumentation = wrapClaudeAgentSDKWithFullInstrumentation;
 //# sourceMappingURL=index.cjs.map