@bitfab/sdk 0.75.0 → 0.76.1

package/dist/{src-BJ5a-chk.cjs → src-CLkegn-P.cjs}

@@ -1,6 +1,6 @@
 const require_bamlPeer = require("./bamlPeer-5Ohpq8q9.cjs");
-const require_http = require("./http-BkL24D7v.cjs");
-const require_autoTrace = require("./autoTrace-DPWV1Xzr.cjs");
+const require_http = require("./http-ykHGNcju.cjs");
+const require_autoTrace = require("./autoTrace-pZKu5Nk2.cjs");
 const require_optionalPeer = require("./optionalPeer-CUH756lt.cjs");
 let superjson = require("superjson");
 superjson = require_bamlPeer.__toESM(superjson, 1);
@@ -2670,15 +2670,6 @@ function normalizeMockOverrides(mockOverride) {
 	} : override);
 }
 //#endregion
-//#region src/streamFinalizationError.ts
-var StreamFinalizationError = class extends Error {
-	constructor(message, output) {
-		super(message);
-		this.output = output;
-		this.name = "StreamFinalizationError";
-	}
-};
-//#endregion
 //#region src/openaiAgentSdk.ts
 var BitfabOpenAIAgentHandler = class {
 	constructor(config) {
@@ -2695,7 +2686,7 @@ var BitfabOpenAIAgentHandler = class {
 			if (isStreaming && res?.completed) try {
 				await res.completed;
 			} catch (error) {
-				throw new StreamFinalizationError(error instanceof Error ? error.message : String(error), res.finalOutput);
+				throw new require_autoTrace.StreamFinalizationError(error instanceof Error ? error.message : String(error), res.finalOutput);
 			}
 			return res?.finalOutput;
 		};
@@ -3063,7 +3054,7 @@ var BitfabPiIntegration = class {
 			finalize: (result) => {
 				const finalMessage = result;
 				const text = messageText(finalMessage);
-				if (failed(finalMessage)) throw new StreamFinalizationError(finalMessage?.errorMessage ?? `Agent run ${finalMessage?.stopReason}`, text);
+				if (failed(finalMessage)) throw new require_autoTrace.StreamFinalizationError(finalMessage?.errorMessage ?? `Agent run ${finalMessage?.stopReason}`, text);
 				return text;
 			}
 		}, callReturningFinalMessage)(...args).then(() => void 0);
@@ -3082,7 +3073,7 @@ var BitfabPiIntegration = class {
 			const recordFinalMessage = async (result) => {
 				const message = await result.result();
 				llmSpan?.addContext(usageContext(message, model));
-				if (failed(message)) throw new StreamFinalizationError(message.errorMessage ?? `Model request ${message.stopReason}`, message);
+				if (failed(message)) throw new require_autoTrace.StreamFinalizationError(message.errorMessage ?? `Model request ${message.stopReason}`, message);
 				return message;
 			};
 			const traced = this.host.withSpan(this.traceFunctionKey, {
@@ -5557,7 +5548,7 @@ var BitfabVercelAiHandler = class {
 					type: "llm",
 					finalize: async () => {
 						const output = await summary;
-						if (output.error !== void 0) throw new StreamFinalizationError(output.error, output);
+						if (output.error !== void 0) throw new require_autoTrace.StreamFinalizationError(output.error, output);
 						return output;
 					},
 					surface: "inherit",
@@ -5654,6 +5645,9 @@ function runWithSpanStack(stack, fn) {
 		throw error;
 	}
 }
+function finalizedLater(finalize, result) {
+	return Promise.resolve().then(() => finalize(result));
+}
 function isAsyncIterable(value) {
 	return value !== null && typeof value === "object" && typeof value[Symbol.asyncIterator] === "function";
 }
@@ -5799,6 +5793,15 @@ function extractContextFromCollector(collector) {
 	}
 }
 const BOUND_FUNCTION_NAME_PREFIX = /^bound /;
+function unboundName(method) {
+	const name = method.name.replace(BOUND_FUNCTION_NAME_PREFIX, "");
+	return name === "" ? void 0 : name;
+}
+function requiredMethodName(method, error) {
+	const name = unboundName(method);
+	if (name === void 0) throw new require_http.BitfabError(error);
+	return name;
+}
 function discoveredSpanType(definition, nodeConfiguration) {
 	const recordsModelCall = definition.baml === true || definition.jev === true || definition.clientCall !== void 0;
 	const ownType = recordsModelCall ? "llm" : "function";
@@ -5860,10 +5863,12 @@ function isSpanType(value) {
 const noOpSpan = {
 	id: "",
 	traceId: "",
+	traceFunctionKeys: [],
 	addContext() {},
 	setPrompt() {}
 };
 const noOpTrace = {
+	traceFunctionKey: "",
 	setSessionId() {},
 	setName() {},
 	setMetadata() {},
@@ -5878,6 +5883,9 @@ function currentTraceRootSpan() {
 		if (context?.isRoot && context.traceId === traceId) return context;
 	}
 }
+function activeTraceFunctionKeys(stack) {
+	return [.../* @__PURE__ */ new Set([...require_autoTrace.currentAutoTraceEntries().map((entry) => entry.context.traceFunctionKey), ...stack.map((context) => context.traceFunctionKey)])];
+}
 function getCurrentSpan() {
 	const stack = getSpanStack();
 	const current = stack[stack.length - 1];
@@ -5885,6 +5893,7 @@ function getCurrentSpan() {
 	return {
 		id: current.spanId,
 		traceId: current.traceId,
+		traceFunctionKeys: activeTraceFunctionKeys(stack),
 		addContext(context) {
 			try {
 				if (typeof context !== "object" || context === null) return;
@@ -5924,6 +5933,7 @@ function getCurrentTrace() {
 		return traceState;
 	};
 	return {
+		traceFunctionKey: current.rootTraceFunctionKey,
 		setSessionId(sessionId) {
 			try {
 				const traceState = getOrCreateTraceState();
@@ -6169,9 +6179,9 @@ var Bitfab = class {
 				const innermost = stack[stack.length - 1];
 				return innermost === void 0 ? traceFunctionKey : rootTraceFunctionKeyOfTrace(innermost.traceId, traceFunctionKey);
 			};
-			const recordsSpan = (entry, spanName, declared) => {
+			const recordsSpan = (entry, spanName, declared, llmCall = false) => {
 				const contentOff = !declared && self.simulationPlan.contentOffNow(rootTraceFunctionKeyOfEntry(entry), spanName);
-				const countsTowardCapturedSubtree = !declared && !contentOff;
+				const countsTowardCapturedSubtree = !declared && !contentOff && !llmCall;
 				const refusedBy = entry.depth >= maxDepth ? "max_depth" : countsTowardCapturedSubtree && capturedSubtreeCount >= maxCapturedSubtreeSpans ? "max_captured_subtree_spans" : void 0;
 				if (refusedBy !== void 0) {
 					truncate(refusedBy);
@@ -6190,7 +6200,10 @@ var Bitfab = class {
 					if (nodeConfiguration?.capture === false) return require_autoTrace.clearAutoTraceNodeConfiguration(invokeFn);
 					const spanName = nodeConfiguration?.name ?? definition.name;
 					const declaredNode = nodeConfiguration !== void 0;
-					if (!recordsSpan(entry, spanName, declaredNode)) return invokeWithoutNode();
+					const clientCall = require_autoTrace.clientCallProvider(definition.clientCall);
+					const llmCall = definition.baml === true || definition.jev === true || clientCall !== void 0;
+					if (!recordsSpan(entry, spanName, declaredNode, llmCall)) return invokeWithoutNode();
+					const streamRecording = clientCall !== void 0 && require_autoTrace.clientCallStreams(clientCall, definition.clientCallMethod ?? simpleName ?? "", inputs[0]) ? require_autoTrace.sharedClientCallStreamRecording(spanDetails, clientCall, inputs[0]) : void 0;
 					const mockOnReplay = nodeConfiguration?.mockOnReplay ?? options.mockOnReplayDefault;
 					const nodeExperimentId = nodeConfiguration === void 0 ? void 0 : resolveSpanExperimentId(nodeConfiguration);
 					const childOptions = {
@@ -6207,11 +6220,14 @@ var Bitfab = class {
 						...nodeExperimentId !== void 0 && { experimentId: nodeExperimentId },
 						...mockOnReplay !== void 0 && { mockOnReplay },
 						...nodeConfiguration?.replayReusable !== void 0 && { replayReusable: nodeConfiguration.replayReusable },
-						...nodeConfiguration?.finalize !== void 0 && { finalize: nodeConfiguration.finalize }
+						...nodeConfiguration?.finalize !== void 0 && { finalize: nodeConfiguration.finalize },
+						...streamRecording !== void 0 && {
+							finalize: () => streamRecording.done,
+							alwaysRunsOnReplay: true
+						}
 					};
 					const isAsync = definition.async === true;
-					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);
+					return openChild(entry, childOptions, inputs, isAsync, definition.baml === true ? () => invokeAutoTracedBaml(invokeFn, isAsync) : definition.jev === true ? () => recordAutoTracedJev(invokeFn, getCurrentSpan()) : streamRecording !== void 0 ? () => streamRecording.start(getCurrentSpan(), invokeFn) : 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();
@@ -6521,8 +6537,7 @@ var Bitfab = class {
 		return wrappedFn;
 	}
 	wrapJev(method, options) {
-		const methodName = method.name.replace(BOUND_FUNCTION_NAME_PREFIX, "");
-		if (!methodName) throw new require_http.BitfabError("wrapJev requires a named function (e.g., jev.systemOne.bind(jev)).");
+		const name = requiredMethodName(method, "wrapJev requires a named function (e.g., jev.systemOne.bind(jev)).");
 		const wrappedFn = async (...args) => {
 			const span = getCurrentSpan();
 			const startedAt = Date.now();
@@ -6536,22 +6551,32 @@ var Bitfab = class {
 		};
 		Object.defineProperty(wrappedFn, "name", {
 			configurable: true,
-			value: methodName
+			value: name
 		});
 		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);
+		return this.wrapClientCall(require_autoTrace.OPENROUTER, method, options, requiredMethodName(method, "wrapOpenRouter requires a named function (e.g., openai.chat.completions.create.bind(openai.chat.completions))."));
 	}
 	wrapAnthropic(method, options) {
-		return this.wrapClientCall(require_autoTrace.ANTHROPIC, "wrapAnthropic requires a named function (e.g., anthropic.messages.create.bind(anthropic.messages)).", method, options);
+		return this.wrapClientCall(require_autoTrace.ANTHROPIC, method, options, requiredMethodName(method, "wrapAnthropic requires a named function (e.g., anthropic.messages.create.bind(anthropic.messages))."));
 	}
 	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);
+		return this.wrapClientCall(require_autoTrace.OPENAI_RESPONSES, method, options, requiredMethodName(method, "wrapOpenAIResponses requires a named function (e.g., openai.responses.create.bind(openai.responses))."));
+	}
+	wrapOpenAIChat(method, options) {
+		return this.wrapClientCall(require_autoTrace.OPENAI_CHAT, method, options, requiredMethodName(method, "wrapOpenAIChat requires a named function (e.g., openai.chat.completions.create.bind(openai.chat.completions))."));
+	}
+	wrapOpenAIEmbeddings(method, options) {
+		return this.wrapClientCall(require_autoTrace.OPENAI_EMBEDDINGS, method, options, requiredMethodName(method, "wrapOpenAIEmbeddings requires a named function (e.g., openai.embeddings.create.bind(openai.embeddings))."));
 	}
-	wrapClientCall(provider, anonymousError, method, options) {
-		const methodName = method.name.replace(BOUND_FUNCTION_NAME_PREFIX, "");
-		if (!methodName) throw new require_http.BitfabError(anonymousError);
+	wrapOpenAITranscriptions(method, options) {
+		return this.wrapClientCall(require_autoTrace.OPENAI_TRANSCRIPTIONS, method, options, requiredMethodName(method, "wrapOpenAITranscriptions requires a named function (e.g., openai.audio.transcriptions.create.bind(openai.audio.transcriptions))."));
+	}
+	wrapGemini(method, options) {
+		return this.wrapClientCall(require_autoTrace.GEMINI, method, options, unboundName(method) ?? "generateContent");
+	}
+	wrapClientCall(provider, method, options, name) {
 		const wrappedFn = async (...args) => {
 			const span = getCurrentSpan();
 			const startedAt = Date.now();
@@ -6572,7 +6597,7 @@ var Bitfab = class {
 		};
 		Object.defineProperty(wrappedFn, "name", {
 			configurable: true,
-			value: methodName
+			value: name
 		});
 		return wrappedFn;
 	}
@@ -6645,6 +6670,7 @@ var Bitfab = class {
 					spanId,
 					details: options.spanDetails ?? { contexts: [] },
 					spanName: options.name ?? options.nameFor?.(this) ?? qualifiedSpanName(this, functionNameAtOpen) ?? traceFunctionKey,
+					traceFunctionKey,
 					rootTraceFunctionKey: rootTraceFunctionKeyOfTrace(traceId, traceFunctionKey),
 					instrumentation: options.instrumentation ?? "span",
 					declaredNode: options.declaredNode === true,
@@ -6749,9 +6775,9 @@ var Bitfab = class {
 					}
 				};
 				const recordSpan = (result) => {
-					if (options.finalize) self.httpClient.trackDeferred(Promise.resolve().then(() => options.finalize(result)).then((output) => sendSpan({ result: output })).catch((error) => sendSpan({
-						result: error instanceof StreamFinalizationError ? error.output : void 0,
-						error: error instanceof StreamFinalizationError ? error.message : error instanceof Error ? `finalize failed: ${error.message}` : `finalize failed: ${String(error)}`
+					if (options.finalize) self.httpClient.trackDeferred(finalizedLater(options.finalize, result).then((output) => sendSpan({ result: output })).catch((error) => sendSpan({
+						result: error instanceof require_autoTrace.StreamFinalizationError ? error.output : void 0,
+						error: error instanceof require_autoTrace.StreamFinalizationError ? error.message : error instanceof Error ? `finalize failed: ${error.message}` : `finalize failed: ${String(error)}`
 					})));
 					else sendSpan({ result });
 				};
@@ -6800,7 +6826,7 @@ var Bitfab = class {
 						reuseAllowed: options.replayReusable === true,
 						canReuseOutput: options.finalize === void 0 && options.replayOutput === void 0 && !fn.constructor.name.includes("Generator")
 					});
-					if (decision.mock) {
+					if (decision.mock && options.alwaysRunsOnReplay !== true) {
 						sendSpan({
 							result: decision.output,
 							mocked: true,
@@ -6851,8 +6877,8 @@ var Bitfab = class {
 						return output;
 					};
 					const resolveBaseOutput = () => options.replayOutput !== void 0 ? options.replayOutput() : resolveRecordedOutput();
-					const shouldMockWithBaseStrategy = replayCtxForMock.mockStrategy === "all" || replayCtxForMock.mockStrategy === "marked" && options.mockOnReplay === true;
-					if (replayCtxForMock.mockOverrides?.length) {
+					const shouldMockWithBaseStrategy = options.alwaysRunsOnReplay !== true && (replayCtxForMock.mockStrategy === "all" || replayCtxForMock.mockStrategy === "marked" && options.mockOnReplay === true);
+					if (replayCtxForMock.mockOverrides?.length && options.alwaysRunsOnReplay !== true) {
 						const nodeMeta = {
 							traceFunctionKey,
 							spanName: baseSpanParams.spanName,
@@ -7214,7 +7240,7 @@ var Bitfab = class {
 			await runWithSeedContext({ traceId }, async () => target(...args));
 		} finally {
 			try {
-				const { flushTraces } = await Promise.resolve().then(() => require("./http-BkL24D7v.cjs")).then((n) => n.http_exports);
+				const { flushTraces } = await Promise.resolve().then(() => require("./http-ykHGNcju.cjs")).then((n) => n.http_exports);
 				await flushTraces(3e4);
 			} finally {
 				unrecorded = activeTraceStates.delete(traceId);
@@ -7318,6 +7344,18 @@ var BitfabFunction = class {
 	wrapOpenAIResponses(method, options) {
 		return this.client.wrapOpenAIResponses(method, options);
 	}
+	wrapOpenAIChat(method, options) {
+		return this.client.wrapOpenAIChat(method, options);
+	}
+	wrapOpenAIEmbeddings(method, options) {
+		return this.client.wrapOpenAIEmbeddings(method, options);
+	}
+	wrapOpenAITranscriptions(method, options) {
+		return this.client.wrapOpenAITranscriptions(method, options);
+	}
+	wrapGemini(method, options) {
+		return this.client.wrapGemini(method, options);
+	}
 };
 //#endregion
 //#region src/finalizers.ts
@@ -7420,7 +7458,7 @@ async function seedFromRegistry(registry, pipeline, cases, options = {}) {
 		metadata: seedCase.metadata,
 		sessionId: seedCase.sessionId
 	}));
-	const { flushTraces } = await Promise.resolve().then(() => require("./http-BkL24D7v.cjs")).then((n) => n.http_exports);
+	const { flushTraces } = await Promise.resolve().then(() => require("./http-ykHGNcju.cjs")).then((n) => n.http_exports);
 	await flushTraces(3e4);
 	return {
 		pipeline,
@@ -7670,4 +7708,4 @@ Object.defineProperty(exports, "serializeReplayResult", {
 	}
 });
 
-//# sourceMappingURL=src-BJ5a-chk.cjs.map
+//# sourceMappingURL=src-CLkegn-P.cjs.map