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

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

@@ -1,19 +1,17 @@
 'use strict';
 
-var chunkYD7TO4PD_cjs = require('../chunk-YD7TO4PD.cjs');
-require('../chunk-VOZBYSWK.cjs');
-var chunk6J6F5LJI_cjs = require('../chunk-6J6F5LJI.cjs');
-var chunk255AHH2W_cjs = require('../chunk-255AHH2W.cjs');
-var chunk3STFITZF_cjs = require('../chunk-3STFITZF.cjs');
+var chunkWGWPCVYB_cjs = require('../chunk-WGWPCVYB.cjs');
+require('../chunk-26Q2QD5J.cjs');
+var chunkWR2SQLKP_cjs = require('../chunk-WR2SQLKP.cjs');
+var chunkZVVUZRG7_cjs = require('../chunk-ZVVUZRG7.cjs');
+var chunkQAFFDJFK_cjs = require('../chunk-QAFFDJFK.cjs');
 var chunkSWQOPFE4_cjs = require('../chunk-SWQOPFE4.cjs');
 require('../chunk-RBNCTHK3.cjs');
-var chunk3H2OJLC2_cjs = require('../chunk-3H2OJLC2.cjs');
+var chunkIHIS62VS_cjs = require('../chunk-IHIS62VS.cjs');
 
 // src/anthropic-agent-sdk/client-wrapper.ts
-chunk3H2OJLC2_cjs.init_tracing();
+chunkIHIS62VS_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 = chunk3STFITZF_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 = chunkQAFFDJFK_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 = chunk3STFITZF_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 = chunk255AHH2W_cjs.emitAgentSpan({
+    async function* iterateInTrace2() {
+      const handle = chunkZVVUZRG7_cjs.emitAgentSpan({
         adapter: "anthropic-agent-sdk",
         agentName: defaultAgentName,
         model
@@ -139,8 +80,8 @@ function buildPatchedQuery(original, options) {
       }
       handle?.end();
     }
-    if (chunk3H2OJLC2_cjs.getActiveTraceState() !== null) {
-      return iterateInTrace();
+    if (chunkIHIS62VS_cjs.getActiveTraceState() !== null) {
+      return iterateInTrace2();
     }
     return (async function* outerTrace() {
       const queue = [];
@@ -155,9 +96,9 @@ function buildPatchedQuery(original, options) {
           };
         }
       });
-      const runPromise = chunk3H2OJLC2_cjs.runAsync({ agentName: defaultAgentName }, async (runHandle) => {
+      const runPromise = chunkIHIS62VS_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
-chunk3H2OJLC2_cjs.init_tracing();
-var captureParentForFanOut = chunk3STFITZF_cjs.absorbErrorsSync(
+chunkIHIS62VS_cjs.init_tracing();
+var captureParentForFanOut = chunkQAFFDJFK_cjs.absorbErrorsSync(
   () => {
-    const state = chunk3H2OJLC2_cjs.getActiveTraceState();
+    const state = chunkIHIS62VS_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 = chunk3STFITZF_cjs.absorbErrorsSync(
   },
   { fingerprint: "agent-sdk-common.captureParentForFanOut" }
 );
-var emitWithPinnedParent = chunk3STFITZF_cjs.absorbErrorsSync(
+var emitWithPinnedParent = chunkQAFFDJFK_cjs.absorbErrorsSync(
   (opts) => {
-    const state = chunk3H2OJLC2_cjs.getActiveTraceState();
+    const state = chunkIHIS62VS_cjs.getActiveTraceState();
     if (state === null) return null;
     const snap = opts.snapshot;
     if (!snap || snap.traceId !== state.traceId) return null;
-    const handle = chunk3H2OJLC2_cjs.startSpan({
+    const handle = chunkIHIS62VS_cjs.startSpan({
       name: opts.name,
       type: opts.type,
       metadata: opts.metadata
@@ -233,13 +174,13 @@ function findSpanById(state, spanId) {
   }
   return void 0;
 }
-var emitAsToolSibling = chunk3STFITZF_cjs.absorbErrorsSync(
+var emitAsToolSibling = chunkQAFFDJFK_cjs.absorbErrorsSync(
   (opts) => {
-    const state = chunk3H2OJLC2_cjs.getActiveTraceState();
+    const state = chunkIHIS62VS_cjs.getActiveTraceState();
     if (state === null) return null;
     const tool = findSpanById(state, opts.toolSpanId);
     if (!tool) return null;
-    const handle = chunk3H2OJLC2_cjs.startSpan({
+    const handle = chunkIHIS62VS_cjs.startSpan({
       name: opts.name,
       type: opts.type,
       metadata: opts.metadata
@@ -250,9 +191,9 @@ var emitAsToolSibling = chunk3STFITZF_cjs.absorbErrorsSync(
   },
   { fingerprint: "agent-sdk-common.emitAsToolSibling" }
 );
-chunk3STFITZF_cjs.absorbErrorsSync(
+chunkQAFFDJFK_cjs.absorbErrorsSync(
   () => {
-    const state = chunk3H2OJLC2_cjs.getActiveTraceState();
+    const state = chunkIHIS62VS_cjs.getActiveTraceState();
     if (state === null) return null;
     return {
       traceId: state.traceId,
@@ -297,7 +238,7 @@ var _handleHookEvent = (payload, options) => {
       return;
     }
   }
-  const fallback = chunk255AHH2W_cjs.emitHookSpan({
+  const fallback = chunkZVVUZRG7_cjs.emitHookSpan({
     adapter: "anthropic-agent-sdk",
     hookName: name,
     hookKind: kind,
@@ -307,7 +248,7 @@ var _handleHookEvent = (payload, options) => {
     fallback.end();
   }
 };
-var handleHookEvent = chunk3STFITZF_cjs.absorbErrorsSync(_handleHookEvent, {
+var handleHookEvent = chunkQAFFDJFK_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
-chunk3H2OJLC2_cjs.init_diagnostics();
+chunkIHIS62VS_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 = chunk255AHH2W_cjs.emitLlmSpan({ adapter: "anthropic-agent-sdk", model });
+          const handle = chunkZVVUZRG7_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 = chunkYD7TO4PD_cjs.extractAnthropicSegments(
+              const segments = chunkWGWPCVYB_cjs.extractAnthropicSegments(
                 requestParams,
                 event.message
               );
@@ -442,7 +383,7 @@ async function* instrumentLlmCalls(events, options) {
           if (model === void 0 || model.length === 0) {
             return;
           }
-          openSpan = chunk255AHH2W_cjs.emitLlmSpan({
+          openSpan = chunkZVVUZRG7_cjs.emitLlmSpan({
             adapter: "anthropic-agent-sdk",
             model
           });
@@ -484,7 +425,7 @@ async function* instrumentLlmCalls(events, options) {
             }
             if (requestParams !== void 0) {
               try {
-                const segments = chunkYD7TO4PD_cjs.extractAnthropicSegments(
+                const segments = chunkWGWPCVYB_cjs.extractAnthropicSegments(
                   requestParams,
                   event.message
                 );
@@ -512,13 +453,13 @@ async function* instrumentLlmCalls(events, options) {
 }
 function reportSwallowed(err, fingerprint) {
   try {
-    chunk3H2OJLC2_cjs.report(err, { fingerprint });
+    chunkIHIS62VS_cjs.report(err, { fingerprint });
   } catch {
   }
 }
 
 // src/anthropic-agent-sdk/tool-and-subagent.ts
-chunk3H2OJLC2_cjs.init_diagnostics();
+chunkIHIS62VS_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 = chunk6J6F5LJI_cjs.summarizeToolInput(taskPrompt, inputOpts);
+        metadata.task_description = chunkWR2SQLKP_cjs.summarizeToolInput(taskPrompt, inputOpts);
       }
       let handle2 = null;
       try {
@@ -586,10 +527,10 @@ async function* instrumentTools(events, options) {
     }
     let handle = null;
     try {
-      handle = chunk255AHH2W_cjs.emitToolSpan({
+      handle = chunkZVVUZRG7_cjs.emitToolSpan({
         adapter: ADAPTER,
         toolName: name,
-        inputSummary: chunk6J6F5LJI_cjs.summarizeToolInput(event.input, inputOpts),
+        inputSummary: chunkWR2SQLKP_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 = chunk6J6F5LJI_cjs.summarizeToolOutput(event.content, outputOpts);
+      const outputSummary = chunkWR2SQLKP_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 {
-    chunk3H2OJLC2_cjs.report(err, { fingerprint });
+    chunkIHIS62VS_cjs.report(err, { fingerprint });
   } catch {
   }
 }
 
+// src/anthropic-agent-sdk/trace-bridge.ts
+chunkIHIS62VS_cjs.init_tracing();
+function iterateInTrace(agentName, makeGenerator) {
+  if (chunkIHIS62VS_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 = chunkIHIS62VS_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 = chunkZVVUZRG7_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