bitfab 0.68.3 → 0.68.5

package/dist/replayCli.js

@@ -1,5 +1,5 @@
 #!/usr/bin/env node
-import { a as runRegistryMain, c as runSeedCli, i as runIfEntrypoint, n as parseRegistryCommandArgs, o as checkReplayCliArgs, r as resolveRegistry, s as runReplayCli, t as importRegistry } from "./registryCommand-Dk4Y1ZgW.js";
+import { a as runRegistryMain, c as runSeedCli, i as runIfEntrypoint, n as parseRegistryCommandArgs, o as checkReplayCliArgs, r as resolveRegistry, s as runReplayCli, t as importRegistry } from "./registryCommand-D7bfo0Au.js";
 //#region src/cloudReplayCli.ts
 const CLOUD_REPLAY_HINT = "Add --cloud to run the same replay on GitHub Actions; bitfab-replay --cloud --help explains it.";
 function isCloudReplayCommand(argv) {

package/dist/seedCli.js

@@ -1,5 +1,5 @@
 #!/usr/bin/env node
-import { a as runRegistryMain, c as runSeedCli, i as runIfEntrypoint, n as parseRegistryCommandArgs, r as resolveRegistry, t as importRegistry } from "./registryCommand-Dk4Y1ZgW.js";
+import { a as runRegistryMain, c as runSeedCli, i as runIfEntrypoint, n as parseRegistryCommandArgs, r as resolveRegistry, t as importRegistry } from "./registryCommand-D7bfo0Au.js";
 //#region src/seedCli.ts
 const USAGE = "Usage: bitfab-seed --registry <path> <pipeline> --from-trace <id>[,<id>...]\n       bitfab-seed --registry <path> <pipeline> --cases <cases.jsonl> [--run]\nRun bitfab-seed --registry <path> <pipeline> --help for details.";
 async function runSeedCommand(argv, io = {}, importer = importRegistry) {

package/dist/{src-BRR42NiJ.cjs → src-CxELko3f.cjs}

@@ -1,6 +1,7 @@
-const require_bamlPeer = require("./bamlPeer-BUoZgNxZ.cjs");
-const require_http = require("./http-bBRhl6b9.cjs");
-const require_autoTrace = require("./autoTrace-D3-OZcmR.cjs");
+const require_bamlPeer = require("./bamlPeer-5Ohpq8q9.cjs");
+const require_http = require("./http-DFqFW_hG.cjs");
+const require_autoTrace = require("./autoTrace-DUmR1KJ8.cjs");
+const require_optionalPeer = require("./optionalPeer-CUH756lt.cjs");
 let superjson = require("superjson");
 superjson = require_bamlPeer.__toESM(superjson, 1);
 //#region src/categories.ts
@@ -1696,6 +1697,7 @@ var BitfabLangGraphCallbackHandler = class {
 		this.getActiveSpanContext = config.getActiveSpanContext ?? null;
 		this.shouldRecord = config.shouldRecord ?? (() => this.httpClient.hasApiKey());
 		this.captureTools = config.captureTools ?? true;
+		require_autoTrace.installLangChainRunScopes();
 	}
 	async close(timeoutMs) {
 		return this.ownsHttpClient ? this.httpClient.close(timeoutMs) : true;
@@ -1907,6 +1909,7 @@ var BitfabLangGraphCallbackHandler = class {
 			const converted = messages.map((batch) => batch.map(convertMessage));
 			const spanInfo = this.startSpan(runId, parentRunId, String(name), "llm", converted, metadata, tags);
 			spanInfo.model = model;
+			if (this.shouldRecord()) require_autoTrace.markLangChainRunRecorded(runId);
 		} catch {}
 	}
 	async handleLLMStart(llm, prompts, runId, parentRunId, _extraParams, tags, metadata, runName) {
@@ -1917,10 +1920,12 @@ var BitfabLangGraphCallbackHandler = class {
 			const name = runName ?? model ?? idArr?.[idArr.length - 1] ?? "llm";
 			const spanInfo = this.startSpan(runId, parentRunId, String(name), "llm", prompts, metadata, tags);
 			spanInfo.model = model;
+			if (this.shouldRecord()) require_autoTrace.markLangChainRunRecorded(runId);
 		} catch {}
 	}
 	async handleLLMEnd(output, runId) {
 		try {
+			require_autoTrace.forgetLangChainRun(runId);
 			let llmOutput;
 			const generations = output.generations;
 			if (generations?.length && generations[generations.length - 1]?.length) {
@@ -1938,6 +1943,7 @@ var BitfabLangGraphCallbackHandler = class {
 	}
 	async handleLLMError(error, runId) {
 		try {
+			require_autoTrace.forgetLangChainRun(runId);
 			this.completeSpan(runId, void 0, error instanceof Error ? error.message : String(error));
 		} catch {}
 	}
@@ -2043,7 +2049,7 @@ function fromBase64(text) {
 }
 async function loadLangGraphPeer() {
 	try {
-		return await require_bamlPeer.importOptionalPeer(["@langchain", "langgraph"]);
+		return await require_optionalPeer.importOptionalPeer(["@langchain", "langgraph"]);
 	} catch {
 		throw new require_http.BitfabError("Restoring a LangGraph thread during replay requires @langchain/langgraph.", "https://docs.bitfab.ai/frameworks/langgraph");
 	}
@@ -2285,14 +2291,14 @@ function isEncodedToolResult(value) {
 }
 async function loadPeer(specifierParts, packageName, docsPath) {
 	try {
-		return await require_bamlPeer.importOptionalPeer(specifierParts);
+		return await require_optionalPeer.importOptionalPeer(specifierParts);
 	} catch {
 		throw new require_http.BitfabError(`This LangGraph integration method requires ${packageName}. Install it to use it.`, `https://docs.bitfab.ai${docsPath}`);
 	}
 }
 async function loadLangChainCore() {
 	try {
-		return await require_bamlPeer.importOptionalPeer([
+		return await require_optionalPeer.importOptionalPeer([
 			"@langchain",
 			"core",
 			"messages"
@@ -2303,7 +2309,7 @@ async function loadLangChainCore() {
 }
 async function loadLangGraph() {
 	try {
-		return await require_bamlPeer.importOptionalPeer(["@langchain", "langgraph"]);
+		return await require_optionalPeer.importOptionalPeer(["@langchain", "langgraph"]);
 	} catch {
 		throw new require_http.BitfabError("Replaying a LangGraph Command requires @langchain/langgraph.", "https://docs.bitfab.ai/frameworks/langgraph");
 	}
@@ -2548,7 +2554,7 @@ var BitfabOpenAIAgentHandler = class {
 		this.getActiveSpanContext = config.getActiveSpanContext;
 	}
 	async wrapRun(agent, input, options) {
-		const { run } = await require_bamlPeer.importOptionalPeer(["@openai", "agents"]);
+		const { run } = await require_optionalPeer.importOptionalPeer(["@openai", "agents"]);
 		if (this.getActiveSpanContext?.() != null) return run(agent, input, options);
 		const isStreaming = options?.stream === true;
 		const finalize = async (result) => {
@@ -2570,64 +2576,6 @@ var BitfabOpenAIAgentHandler = class {
 	}
 };
 //#endregion
-//#region src/openaiAgentsSpanContext.ts
-const OPENAI_AGENTS_CONTEXT_STORAGE = Symbol.for("openai.agents.core.asyncLocalStorage");
-const MAX_OPENAI_AGENTS_SPAN_CHAIN = 64;
-const spanGlobal = globalThis;
-const state = spanGlobal.__bitfabOpenAiAgentsSpansV1 ?? {
-	bitfabTraceIds: /* @__PURE__ */ new Map(),
-	endedSpanIds: /* @__PURE__ */ new Map()
-};
-spanGlobal.__bitfabOpenAiAgentsSpansV1 = state;
-function mapOpenAiAgentsTrace(agentsTraceId, bitfabTraceId) {
-	state.bitfabTraceIds.set(agentsTraceId, bitfabTraceId);
-}
-function unmapOpenAiAgentsTrace(agentsTraceId) {
-	state.bitfabTraceIds.delete(agentsTraceId);
-	state.endedSpanIds.delete(agentsTraceId);
-}
-function endOpenAiAgentsSpan(agentsTraceId, spanId) {
-	if (!state.bitfabTraceIds.has(agentsTraceId)) return;
-	const ended = state.endedSpanIds.get(agentsTraceId) ?? /* @__PURE__ */ new Set();
-	ended.add(spanId);
-	state.endedSpanIds.set(agentsTraceId, ended);
-}
-function isOpenAiAgentsContextStorage(value) {
-	return value !== null && typeof value === "object" && typeof value.getStore === "function";
-}
-function currentOpenAiAgentsSpan() {
-	const storage = spanGlobal[OPENAI_AGENTS_CONTEXT_STORAGE];
-	if (!isOpenAiAgentsContextStorage(storage)) return;
-	try {
-		const store = storage.getStore();
-		return store?.active === true ? store.span ?? void 0 : void 0;
-	} catch {
-		return;
-	}
-}
-function currentOpenAiAgentsSpans() {
-	const spans = [];
-	let span = currentOpenAiAgentsSpan();
-	while (span !== void 0 && spans.length < MAX_OPENAI_AGENTS_SPAN_CHAIN) {
-		spans.push(span);
-		span = span.previousSpan ?? void 0;
-	}
-	return spans;
-}
-function openAiAgentsSpanChain() {
-	if (state.bitfabTraceIds.size === 0) return [];
-	return currentOpenAiAgentsSpans().map((span) => span.spanId).filter((spanId) => typeof spanId === "string");
-}
-function openAiAgentsParentSpanId(bitfabTraceId, enclosingSpanIds) {
-	if (state.bitfabTraceIds.size === 0) return;
-	for (const span of currentOpenAiAgentsSpans()) {
-		if (typeof span.spanId !== "string" || typeof span.traceId !== "string") return;
-		if (state.bitfabTraceIds.get(span.traceId) !== bitfabTraceId) return;
-		if (enclosingSpanIds?.includes(span.spanId)) return;
-		if (!state.endedSpanIds.get(span.traceId)?.has(span.spanId)) return span.spanId;
-	}
-}
-//#endregion
 //#region src/organizationMembers.ts
 const MEMBERS_PATH = "/api/sdk/organizationMembers";
 var OrganizationMembersClient = class {
@@ -4688,7 +4636,7 @@ var BitfabOpenAITracingProcessor = class {
 		const activeContext = this.getActiveSpanContext?.();
 		if (activeContext) {
 			this.activeSpanMappings[trace.traceId] = activeContext;
-			if (this.shouldRecord()) mapOpenAiAgentsTrace(trace.traceId, activeContext.traceId);
+			if (this.shouldRecord()) require_autoTrace.mapOpenAiAgentsTrace(trace.traceId, activeContext.traceId);
 		}
 		const canonicalTraceId = activeContext?.traceId ?? this.getCanonicalTraceId(trace.traceId);
 		this.canonicalTraceIds[trace.traceId] = canonicalTraceId;
@@ -4713,7 +4661,7 @@ var BitfabOpenAITracingProcessor = class {
 		if (!this.endedTraces.has(traceId) || [...this.startedSpans.values()].includes(traceId)) return;
 		this.endedTraces.delete(traceId);
 		delete this.activeSpanMappings[traceId];
-		unmapOpenAiAgentsTrace(traceId);
+		require_autoTrace.unmapOpenAiAgentsTrace(traceId);
 		delete this.canonicalTraceIds[traceId];
 		delete this.activeTraces[traceId];
 	}
@@ -4727,7 +4675,7 @@ var BitfabOpenAITracingProcessor = class {
 	async onSpanEnd(span) {
 		const originalTraceId = span.traceId ?? "";
 		const traceId = this.activeSpanMappings[originalTraceId]?.traceId ?? originalTraceId;
-		if (span.spanId !== void 0) endOpenAiAgentsSpan(originalTraceId, span.spanId);
+		if (span.spanId !== void 0) require_autoTrace.endOpenAiAgentsSpan(originalTraceId, span.spanId);
 		try {
 			this.sendSpan(span);
 		} finally {
@@ -4742,7 +4690,7 @@ var BitfabOpenAITracingProcessor = class {
 		await this.httpClient.waitForPendingRequests();
 	}
 	async shutdown(timeout) {
-		for (const traceId of Object.keys(this.activeSpanMappings)) unmapOpenAiAgentsTrace(traceId);
+		for (const traceId of Object.keys(this.activeSpanMappings)) require_autoTrace.unmapOpenAiAgentsTrace(traceId);
 		this.activeTraces = {};
 		this.activeSpanMappings = {};
 		this.canonicalTraceIds = {};
@@ -5220,7 +5168,7 @@ function extractContextFromCollector(collector) {
 }
 const BOUND_FUNCTION_NAME_PREFIX = /^bound /;
 function discoveredSpanType(definition, nodeConfiguration) {
-	const recordsModelCall = definition.baml === true || definition.jev === true || definition.openrouter === true;
+	const recordsModelCall = definition.baml === true || definition.jev === true || definition.clientCall !== void 0;
 	const ownType = recordsModelCall ? "llm" : "function";
 	if (nodeConfiguration === void 0) return ownType;
 	if (recordsModelCall && nodeConfiguration.defaultType === true) return ownType;
@@ -5541,6 +5489,7 @@ var Bitfab = class {
 		const self = this;
 		self.simulationPlan.refresh();
 		loadCollectorClass();
+		require_autoTrace.installLangChainRunScopes();
 		const runRootFunction = (receiver, args) => require_autoTrace.runAsAutoTraceRootFunction(rootFunctionName, () => fn.apply(receiver, args));
 		const maxDepth = autoTraceLimit(options.maxDepth, DEFAULT_AUTO_TRACE_MAX_DEPTH);
 		const maxCapturedSubtreeSpans = autoTraceLimit(options.maxCapturedSubtreeSpans, DEFAULT_AUTO_TRACE_MAX_CAPTURED_SUBTREE_SPANS);
@@ -5629,7 +5578,8 @@ var Bitfab = class {
 						...nodeConfiguration?.finalize !== void 0 && { finalize: nodeConfiguration.finalize }
 					};
 					const isAsync = definition.async === true;
-					return openChild(entry, childOptions, inputs, isAsync, definition.baml === true ? () => invokeAutoTracedBaml(invokeFn, isAsync) : definition.jev === true ? () => recordAutoTracedJev(invokeFn, getCurrentSpan()) : definition.openrouter === true ? () => require_autoTrace.recordAutoTracedOpenRouter(invokeFn, inputs[0], getCurrentSpan()) : invokeFn);
+					const clientCall = require_autoTrace.clientCallProvider(definition.clientCall);
+					return openChild(entry, childOptions, inputs, isAsync, definition.baml === true ? () => invokeAutoTracedBaml(invokeFn, isAsync) : definition.jev === true ? () => recordAutoTracedJev(invokeFn, getCurrentSpan()) : clientCall !== void 0 ? () => require_autoTrace.recordAutoTracedClientCall(clientCall, invokeFn, inputs[0], getCurrentSpan()) : invokeFn);
 				},
 				invokeNestedRoot(nestedName, nestedFunctionName, inputs, invokeFn, entry, link, spanDetails) {
 					if (excluded.has(nestedName) || excluded.has(nestedFunctionName) || !recordsSpan(entry, nestedName, true)) return invokeFn();
@@ -5941,14 +5891,30 @@ var Bitfab = class {
 		return wrappedFn;
 	}
 	wrapOpenRouter(method, options) {
+		return this.wrapClientCall(require_autoTrace.OPENROUTER, "wrapOpenRouter requires a named function (e.g., openai.chat.completions.create.bind(openai.chat.completions)).", method, options);
+	}
+	wrapAnthropic(method, options) {
+		return this.wrapClientCall(require_autoTrace.ANTHROPIC, "wrapAnthropic requires a named function (e.g., anthropic.messages.create.bind(anthropic.messages)).", method, options);
+	}
+	wrapOpenAIResponses(method, options) {
+		return this.wrapClientCall(require_autoTrace.OPENAI_RESPONSES, "wrapOpenAIResponses requires a named function (e.g., openai.responses.create.bind(openai.responses)).", method, options);
+	}
+	wrapClientCall(provider, anonymousError, method, options) {
 		const methodName = method.name.replace(BOUND_FUNCTION_NAME_PREFIX, "");
-		if (!methodName) throw new require_http.BitfabError("wrapOpenRouter requires a named function (e.g., openai.chat.completions.create.bind(openai.chat.completions)).");
+		if (!methodName) throw new require_http.BitfabError(anonymousError);
 		const wrappedFn = async (...args) => {
 			const span = getCurrentSpan();
 			const startedAt = Date.now();
-			const { watch, result } = require_autoTrace.watchOpenRouterUsage(() => method(...args));
+			if (require_autoTrace.frameworkRecordsClientCall()) {
+				const response = await method(...args);
+				try {
+					options?.onResponse?.(response);
+				} catch {}
+				return response;
+			}
+			const { watch, result } = require_autoTrace.watchClientCall(() => method(...args));
 			const response = await result;
-			require_autoTrace.recordWatchedOpenRouterCompletion(span, watch, args[0], response, Date.now() - startedAt);
+			require_autoTrace.recordWatchedClientCall(provider, span, watch, args[0], response, Date.now() - startedAt);
 			try {
 				options?.onResponse?.(response);
 			} catch {}
@@ -6011,7 +5977,7 @@ var Bitfab = class {
 				const seedCtxForTraceId = parentContext ? null : getSeedContext();
 				const traceId = parentContext?.traceId ?? replayCtxForTraceId?.traceId ?? seedCtxForTraceId?.traceId ?? randomUuid();
 				const spanId = randomUuid();
-				const openAiAgentsParent = parentContext !== void 0 && options.instrumentation === "trace" ? openAiAgentsParentSpanId(parentContext.traceId, parentContext.openAiAgentsSpanIds) : void 0;
+				const openAiAgentsParent = parentContext !== void 0 && options.instrumentation === "trace" ? require_autoTrace.openAiAgentsParentSpanId(parentContext.traceId, parentContext.openAiAgentsSpanIds) : void 0;
 				const frameworkParent = parentContext !== void 0 ? options.frameworkParentSpanId : void 0;
 				const parentSpanId = openAiAgentsParent ?? frameworkParent ?? parentContext?.spanId ?? null;
 				const isRootSpan = parentSpanId === null;
@@ -6022,7 +5988,7 @@ var Bitfab = class {
 				const requestedSurface = options.surface ?? "opt-in";
 				const surface = resolveSurface(requestedSurface, parentContext?.surface);
 				assertSurfacesCompatible(requestedSurface, surface, parentContext?.surface, traceFunctionKey);
-				const openAiAgentsSpanIds = openAiAgentsSpanChain();
+				const openAiAgentsSpanIds = require_autoTrace.openAiAgentsSpanChain();
 				const functionNameAtOpen = fn.name !== "" ? fn.name : void 0;
 				const newContext = {
 					traceId,
@@ -6598,7 +6564,7 @@ var Bitfab = class {
 			await runWithSeedContext({ traceId }, async () => target(...args));
 		} finally {
 			try {
-				const { flushTraces } = await Promise.resolve().then(() => require("./http-bBRhl6b9.cjs")).then((n) => n.http_exports);
+				const { flushTraces } = await Promise.resolve().then(() => require("./http-DFqFW_hG.cjs")).then((n) => n.http_exports);
 				await flushTraces(3e4);
 			} finally {
 				unrecorded = activeTraceStates.delete(traceId);
@@ -6693,6 +6659,12 @@ var BitfabFunction = class {
 	wrapOpenRouter(method, options) {
 		return this.client.wrapOpenRouter(method, options);
 	}
+	wrapAnthropic(method, options) {
+		return this.client.wrapAnthropic(method, options);
+	}
+	wrapOpenAIResponses(method, options) {
+		return this.client.wrapOpenAIResponses(method, options);
+	}
 };
 //#endregion
 //#region src/finalizers.ts
@@ -6795,7 +6767,7 @@ async function seedFromRegistry(registry, pipeline, cases, options = {}) {
 		metadata: seedCase.metadata,
 		sessionId: seedCase.sessionId
 	}));
-	const { flushTraces } = await Promise.resolve().then(() => require("./http-bBRhl6b9.cjs")).then((n) => n.http_exports);
+	const { flushTraces } = await Promise.resolve().then(() => require("./http-DFqFW_hG.cjs")).then((n) => n.http_exports);
 	await flushTraces(3e4);
 	return {
 		pipeline,
@@ -7039,4 +7011,4 @@ Object.defineProperty(exports, "serializeReplayResult", {
 	}
 });
 
-//# sourceMappingURL=src-BRR42NiJ.cjs.map
+//# sourceMappingURL=src-CxELko3f.cjs.map