@bitfab/sdk 0.38.5 → 0.38.7

package/dist/{chunk-Z3GIZGS5.js → chunk-RZ3P6ICN.js}

@@ -5,7 +5,7 @@ import {
   runWithAutoTraceContext,
   runWithAutoTraceNodeConfiguration,
   runWithAutoTraceRootContext
-} from "./chunk-J47KPS77.js";
+} from "./chunk-ZUD7OFYB.js";
 import {
   BitfabError,
   DEFAULT_SERVICE_URL,
@@ -19,7 +19,7 @@ import {
   toJsonSafe,
   toJsonSafeReport,
   warnOnce
-} from "./chunk-VVI573UU.js";
+} from "./chunk-7E6KCBTH.js";
 import {
   __privateAdd,
   __privateGet,
@@ -1183,6 +1183,7 @@ var BitfabLangGraphCallbackHandler = class {
     });
     this.traceFunctionKey = config.traceFunctionKey;
     this.getActiveSpanContext = config.getActiveSpanContext ?? null;
+    this.captureTools = config.captureTools ?? true;
   }
   /**
    * Flush and release the span transport this handler started. A no-op when
@@ -1491,6 +1492,9 @@ var BitfabLangGraphCallbackHandler = class {
   }
   // ── tool callbacks ────────────────────────────────────────────
   async handleToolStart(tool, input, runId, parentRunId, tags, metadata, runName) {
+    if (!this.captureTools) {
+      return;
+    }
     try {
       const serialized = tool ?? {};
       const name = runName ?? serialized.name ?? "tool";
@@ -1507,12 +1511,18 @@ var BitfabLangGraphCallbackHandler = class {
     }
   }
   async handleToolEnd(output, runId) {
+    if (!this.captureTools) {
+      return;
+    }
     try {
       this.completeSpan(runId, output);
     } catch {
     }
   }
   async handleToolError(error, runId) {
+    if (!this.captureTools) {
+      return;
+    }
     try {
       this.completeSpan(
         runId,
@@ -1557,6 +1567,256 @@ var BitfabLangGraphCallbackHandler = class {
   }
 };
 
+// src/langgraphIntegration.ts
+var TOOL_RESULT_TAG = "__bitfabLangGraphToolResult";
+function isRecord(value) {
+  return typeof value === "object" && value !== null;
+}
+function isToolMessage(value) {
+  return isRecord(value) && value.lc_direct_tool_output === true && value.type === "tool" && (typeof value.content === "string" || Array.isArray(value.content));
+}
+function isCommand(value) {
+  return isRecord(value) && value.lg_name === "Command";
+}
+function encodeNested(value) {
+  if (isToolMessage(value) || isCommand(value)) {
+    return encodeNativeToolResult(value);
+  }
+  if (Array.isArray(value)) {
+    return value.map(encodeNested);
+  }
+  if (isRecord(value)) {
+    return Object.fromEntries(
+      Object.entries(value).map(([key, entry]) => [key, encodeNested(entry)])
+    );
+  }
+  return value;
+}
+function encodeNativeToolResult(value) {
+  if (isToolMessage(value)) {
+    return {
+      [TOOL_RESULT_TAG]: "tool-message",
+      content: value.content,
+      name: typeof value.name === "string" ? value.name : void 0,
+      id: typeof value.id === "string" ? value.id : void 0,
+      status: value.status === "success" || value.status === "error" ? value.status : void 0,
+      artifact: encodeNested(value.artifact),
+      metadata: encodeNested(value.metadata),
+      additionalKwargs: encodeNested(value.additional_kwargs),
+      responseMetadata: encodeNested(value.response_metadata)
+    };
+  }
+  return {
+    [TOOL_RESULT_TAG]: "command",
+    graph: isRecord(value) && typeof value.graph === "string" ? value.graph : void 0,
+    update: isRecord(value) ? encodeNested(value.update) : void 0,
+    resume: isRecord(value) ? encodeNested(value.resume) : void 0,
+    goto: isRecord(value) ? encodeNested(value.goto) : void 0
+  };
+}
+function finalizeToolResult(value) {
+  return isToolMessage(value) || isCommand(value) ? encodeNativeToolResult(value) : value;
+}
+function isEncodedToolResult(value) {
+  return isRecord(value) && typeof value[TOOL_RESULT_TAG] === "string";
+}
+async function loadLangChainCore() {
+  try {
+    return await importOptionalPeer([
+      "@langchain",
+      "core",
+      "messages"
+    ]);
+  } catch {
+    throw new BitfabError(
+      "LangGraph tool replay requires @langchain/core. Install @langchain/langgraph before using getLangGraphIntegration().",
+      "https://docs.bitfab.ai/frameworks/langgraph"
+    );
+  }
+}
+async function loadLangGraph() {
+  try {
+    return await importOptionalPeer([
+      "@langchain",
+      "langgraph"
+    ]);
+  } catch {
+    throw new BitfabError(
+      "Replaying a LangGraph Command requires @langchain/langgraph.",
+      "https://docs.bitfab.ai/frameworks/langgraph"
+    );
+  }
+}
+async function reviveNested(value, toolCallId) {
+  if (isEncodedToolResult(value)) {
+    return reviveToolResult(value, toolCallId);
+  }
+  if (Array.isArray(value)) {
+    return Promise.all(value.map((entry) => reviveNested(entry, toolCallId)));
+  }
+  if (isRecord(value)) {
+    const entries = await Promise.all(
+      Object.entries(value).map(async ([key, entry]) => [
+        key,
+        await reviveNested(entry, toolCallId)
+      ])
+    );
+    return Object.fromEntries(entries);
+  }
+  return value;
+}
+async function reviveToolResult(value, toolCallId) {
+  if (value[TOOL_RESULT_TAG] === "tool-message") {
+    const { ToolMessage } = await loadLangChainCore();
+    return new ToolMessage({
+      content: value.content,
+      tool_call_id: toolCallId,
+      ...typeof value.name === "string" && { name: value.name },
+      ...typeof value.id === "string" && { id: value.id },
+      ...(value.status === "success" || value.status === "error") && {
+        status: value.status
+      },
+      ...value.artifact !== void 0 && {
+        artifact: await reviveNested(value.artifact, toolCallId)
+      },
+      ...isRecord(value.metadata) && {
+        metadata: await reviveNested(value.metadata, toolCallId)
+      },
+      ...isRecord(value.additionalKwargs) && {
+        additional_kwargs: await reviveNested(
+          value.additionalKwargs,
+          toolCallId
+        )
+      },
+      ...isRecord(value.responseMetadata) && {
+        response_metadata: await reviveNested(
+          value.responseMetadata,
+          toolCallId
+        )
+      }
+    });
+  }
+  const { Command } = await loadLangGraph();
+  return new Command({
+    ...typeof value.graph === "string" && { graph: value.graph },
+    ...value.update !== void 0 && {
+      update: await reviveNested(value.update, toolCallId)
+    },
+    ...value.resume !== void 0 && {
+      resume: await reviveNested(value.resume, toolCallId)
+    },
+    ...value.goto !== void 0 && {
+      goto: await reviveNested(value.goto, toolCallId)
+    }
+  });
+}
+var BitfabLangGraphIntegration = class {
+  constructor(config) {
+    this.client = config.client;
+    this.traceFunctionKey = config.traceFunctionKey;
+    this.callbackHandler = config.callbackHandler;
+    this.mockToolsOnReplay = config.options?.mockToolsOnReplay ?? true;
+  }
+  /**
+   * Wrap tools before passing the same returned array to both
+   * `model.bindTools()` and `new ToolNode()`. Each invocation becomes an
+   * independently mockable child span.
+   */
+  wrapTools(tools) {
+    return tools.map((tool) => this.wrapTool(tool));
+  }
+  /**
+   * Create the normal graph entry point. The returned function adds Bitfab's
+   * callback handler, preserves invocation config, and records only the graph
+   * input as the replayable root input.
+   *
+   * @experimental This API may change before it is stable.
+   */
+  createInvoker(graph) {
+    const configuredGraph = graph.withConfig({
+      callbacks: [this.callbackHandler]
+    });
+    return (input, config) => {
+      const invoke = this.wrapInvoke(
+        (rootInput) => configuredGraph.invoke(rootInput, config)
+      );
+      return invoke(input);
+    };
+  }
+  wrapTool(tool) {
+    const toolName = tool.name;
+    if (typeof toolName !== "string" || toolName.length === 0) {
+      throw new BitfabError(
+        "LangGraph replayable tools must have a name.",
+        "https://docs.bitfab.ai/frameworks/langgraph"
+      );
+    }
+    const mockToolsOnReplay = this.mockToolsOnReplay;
+    const shouldMock = typeof mockToolsOnReplay === "boolean" ? mockToolsOnReplay : mockToolsOnReplay.includes(toolName);
+    const originalInvoke = tool.invoke.bind(tool);
+    return new Proxy(tool, {
+      get: (target, property) => {
+        if (property === "invoke") {
+          return async (input, ...rest) => {
+            const toolCallId = isRecord(input) && typeof input.id === "string" ? input.id : "";
+            const args = isRecord(input) && "args" in input ? input.args : input;
+            this.assertReplayToolResultExists(toolName, shouldMock);
+            const execute = this.client.withSpan(
+              this.traceFunctionKey,
+              {
+                name: toolName,
+                type: "function",
+                captureWhen: "nested",
+                mockOnReplay: shouldMock,
+                finalize: finalizeToolResult
+              },
+              async (_args) => await originalInvoke(input, ...rest)
+            );
+            const result = await execute(args);
+            return isEncodedToolResult(result) ? await reviveToolResult(result, toolCallId) : result;
+          };
+        }
+        const value = Reflect.get(target, property, target);
+        return typeof value === "function" ? value.bind(target) : value;
+      }
+    });
+  }
+  assertReplayToolResultExists(toolName, shouldMock) {
+    const replayContext = getReplayContext();
+    if (!replayContext?.mockTree) {
+      return;
+    }
+    const counterKey = `${this.traceFunctionKey}:${toolName}`;
+    const callIndex = replayContext.callCounters?.get(counterKey) ?? 0;
+    const mockSpan = replayContext.mockTree.spans.get(
+      `${counterKey}:${callIndex}`
+    );
+    const hasMatchingOverride = replayContext.mockOverrides?.some(
+      (override) => override.match({
+        traceFunctionKey: this.traceFunctionKey,
+        spanName: toolName,
+        type: "function",
+        originalSpanId: mockSpan?.sourceSpanId
+      })
+    );
+    const expectsRecordedOutput = replayContext.mockStrategy === "all" || replayContext.mockStrategy === "marked" && shouldMock;
+    if (hasMatchingOverride !== true && expectsRecordedOutput && !mockSpan) {
+      throw new BitfabError(
+        `No recorded LangGraph tool result for "${toolName}" at call ${callIndex + 1}; refusing to execute the live tool during replay.`,
+        "https://docs.bitfab.ai/frameworks/langgraph"
+      );
+    }
+  }
+  /** Wrap the function that invokes the compiled graph as the replay root. */
+  wrapInvoke(fn) {
+    return this.client.withSpan(
+      this.traceFunctionKey,
+      { name: this.traceFunctionKey, type: "agent" },
+      fn
+    );
+  }
+};
+
 // src/openaiAgentSdk.ts
 var BitfabOpenAIAgentHandler = class {
   constructor(config) {
@@ -2393,8 +2653,9 @@ var Bitfab = class {
    * Decorate a class method as an automatically expanded trace root.
    *
    * Build instrumentation turns repository functions called beneath this
-   * method into nested spans. Every generated span preserves structure; only
-   * function IDs selected by the capture policy include inputs and output.
+   * method into nested spans that capture inputs, outputs, and errors by
+   * default. A confirmed capture policy can narrow rich capture to selected
+   * function IDs.
    * Without a compatible build transform, this still records the decorated
    * method as a normal rich root span but cannot discover child calls.
    *
@@ -2604,7 +2865,7 @@ var Bitfab = class {
               type: nodeConfiguration?.type ?? "function",
               captureWhen: "nested",
               functionId: definition.id,
-              captureContent: nodeConfiguration !== void 0 || capturePolicy.has(definition.id),
+              captureContent: nodeConfiguration !== void 0 || capturePolicy === void 0 || capturePolicy.has(definition.id),
               autoTraceDefinition: definition,
               ...nodeConfiguration?.testRunId !== void 0 && {
                 testRunId: nodeConfiguration.testRunId
@@ -2669,7 +2930,11 @@ var Bitfab = class {
       const functionIds = Array.isArray(policy.functionIds) ? policy.functionIds.filter(
         (id) => typeof id === "string" && id.startsWith(`${AUTO_TRACE_PROTOCOL}:`)
       ).slice(0, DEFAULT_AUTO_TRACE_MAX_SPANS) : [];
-      __setBitfabAutoTraceCapturePolicy(this, traceFunctionKey, functionIds);
+      __setBitfabAutoTraceCapturePolicy(
+        this,
+        traceFunctionKey,
+        policy.revision === null ? void 0 : functionIds
+      );
       state.refreshAfter = Date.now() + AUTO_TRACE_POLICY_REFRESH_MS;
     }).catch(() => {
       state.refreshAfter = Date.now() + AUTO_TRACE_POLICY_RETRY_MS;
@@ -2928,6 +3193,37 @@ var Bitfab = class {
   getLangChainCallbackHandler(traceFunctionKey) {
     return this.getLangGraphCallbackHandler(traceFunctionKey);
   }
+  /**
+   * Get the first-class LangGraph integration for tracing and replaying tools
+   * executed by `ToolNode`.
+   *
+   * The integration combines `wrapTools()` for per-tool replay interception
+   * with `createInvoker()` for a callback-configured replayable graph entry
+   * point. Lower-level callback and root wrappers remain available.
+   *
+   * @param traceFunctionKey - Groups traces under this key in Bitfab
+   * @param options - Controls which tools are marked for replay mocking
+   * @experimental This API may change before it is stable.
+   */
+  getLangGraphIntegration(traceFunctionKey, options) {
+    const callbackHandler = new BitfabLangGraphCallbackHandler({
+      apiKey: this.resolveApiKey(),
+      traceFunctionKey,
+      serviceUrl: this.serviceUrl,
+      getActiveSpanContext: () => {
+        const stack = getSpanStack();
+        return stack[stack.length - 1] ?? null;
+      },
+      captureTools: false,
+      _httpClient: this.httpClient
+    });
+    return new BitfabLangGraphIntegration({
+      client: this,
+      traceFunctionKey,
+      callbackHandler,
+      options
+    });
+  }
   /**
    * Get a Claude Agent SDK handler for tracing.
    *
@@ -3806,7 +4102,7 @@ var Bitfab = class {
         `Function is wrapped with trace function key '${wrappedKey}' but replay was called with '${traceFunctionKey}'. Pass matching keys, or pass the unwrapped function to replay it under the explicit key.`
       );
     }
-    const { replay: doReplay } = await import("./replay-Z6DON4SP.js");
+    const { replay: doReplay } = await import("./replay-DIRRDWI2.js");
     return doReplay(
       this.httpClient,
       this.serviceUrl,
@@ -3958,6 +4254,14 @@ var BitfabFunction = class {
   getLangChainCallbackHandler() {
     return this.client.getLangChainCallbackHandler(this.traceFunctionKey);
   }
+  /**
+   * Get the first-class LangGraph tool replay integration bound to this key.
+   *
+   * @experimental This API may change before it is stable.
+   */
+  getLangGraphIntegration(options) {
+    return this.client.getLangGraphIntegration(this.traceFunctionKey, options);
+  }
   /**
    * Wrap a BAML client method to automatically capture prompt and LLM metadata.
    * Delegates to the parent client's wrapBAML method.
@@ -4039,6 +4343,7 @@ export {
   BitfabClaudeAgentHandler,
   SUPPORTED_PROVIDERS,
   BitfabLangGraphCallbackHandler,
+  BitfabLangGraphIntegration,
   BitfabOpenAIAgentHandler,
   BitfabOpenAITracingProcessor,
   BitfabVercelAiHandler,
@@ -4050,4 +4355,4 @@ export {
   finalizers,
   defineReplayRegistry
 };
-//# sourceMappingURL=chunk-Z3GIZGS5.js.map
+//# sourceMappingURL=chunk-RZ3P6ICN.js.map