@bitfab/sdk 0.76.1 → 0.76.3

package/dist/{src-BJjbR3Nq.js → src-CYs1uXoX.js}

@@ -1,8 +1,8 @@
 import { t as __exportAll } from "./rolldown-runtime-D7D4PA-g.js";
-import { C as BitfabError, E as __version__, S as gitRunner, T as readEnv$1, _ as MAX_SPAN_CARRIER_BYTES, b as replayContextReady, c as recordCallerTraceMetadata, f as ROOT_TRACE_FUNCTION_KEY_FIELD, g as MAX_COMPRESSIBLE_SPAN_CARRIER_BYTES, h as makeSpanOrigin, i as flushTraces, l as retireTraceMetadata, m as isFrameworkInstrumentation, n as SPAN_TYPES, o as unrefTimer, p as SimulationPlan, r as experimentIdWireFields, s as callerTraceMetadata, t as HttpClient, v as byteLength, w as MixedTracingError, x as runWithReplayContext, y as getReplayContext } from "./http-DS1QjBvx.js";
+import { C as BitfabError, D as readEnv$1, E as serializeError, O as __version__, S as gitRunner, T as describeError, _ as MAX_SPAN_CARRIER_BYTES, b as replayContextReady, c as recordCallerTraceMetadata, f as ROOT_TRACE_FUNCTION_KEY_FIELD, g as MAX_COMPRESSIBLE_SPAN_CARRIER_BYTES, h as makeSpanOrigin, i as flushTraces, l as retireTraceMetadata, m as isFrameworkInstrumentation, n as SPAN_TYPES, o as unrefTimer, p as SimulationPlan, r as experimentIdWireFields, s as callerTraceMetadata, t as HttpClient, v as byteLength, w as MixedTracingError, x as runWithReplayContext, y as getReplayContext } from "./http-BA7vmvuC.js";
 import { A as recordWatchedClientCall, B as installLangChainRunScopes, C as OPENAI_RESPONSES, D as clientCallStreams, E as clientCallProvider, F as mapOpenAiAgentsTrace, G as isCaptureSuppressed, I as openAiAgentsParentSpanId, J as asyncStorageReady, K as runWithCaptureSuppressed, L as openAiAgentsSpanChain, M as watchClientCall, N as StreamFinalizationError, O as frameworkRecordsClientCall, P as endOpenAiAgentsSpan, R as unmapOpenAiAgentsTrace, S as OPENAI_EMBEDDINGS, T as OPENROUTER, U as warnOnce, V as markLangChainRunRecorded, W as writeOnce, X as isAsyncStorageInitDone, Y as createAsyncLocalStorage, _ as runWithAutoTraceNodeConfiguration, b as GEMINI, d as currentAutoTraceEntries, g as runWithAutoTraceFramesAt, h as runWithAutoTraceEntry, j as sharedClientCallStreamRecording, k as recordAutoTracedClientCall, m as runAsAutoTraceRootFunction, p as invokeThroughAutoTraceEntries, s as __bitfabAutoTraceActive, u as clearAutoTraceNodeConfiguration, v as runWithAutoTraceRootContext, w as OPENAI_TRANSCRIPTIONS, x as OPENAI_CHAT, y as ANTHROPIC, z as forgetLangChainRun } from "./autoTrace-BgN0zCQf.js";
 import { t as importOptionalPeer } from "./optionalPeer-CczrzrXx.js";
-import { t as loadBamlModule } from "./bamlPeer-Dyeb3e5o.js";
+import { n as loadBamlModule, t as isSharedBamlModule } from "./bamlPeer-CWPgBSlJ.js";
 import superjson from "superjson";
 //#region src/categories.ts
 const CATEGORIES_PATH = "/api/sdk/categories";
@@ -4022,22 +4022,10 @@ function errorMessage(error) {
 }
 function replayItemErrorMessage(error) {
 	if (error instanceof DbBranchReplayError) return `Replay requested a database branch for trace ${error.originalTraceId} but it could not be resolved (${error.code}): ${error.message}. The function was not run, because replaying it against the live database would produce a result that looks valid but did not use the historical data you asked for.`;
-	return errorMessage(error);
+	return describeError(error);
 }
 function replayJsonReplacer(_key, value) {
-	if (value instanceof Error) {
-		const serialized = {
-			name: value.name,
-			message: value.message,
-			stack: value.stack
-		};
-		if (value instanceof DbBranchReplayError) {
-			serialized.code = value.code;
-			serialized.originalTraceId = value.originalTraceId;
-			if (value.cause !== void 0) serialized.cause = value.cause;
-		}
-		return serialized;
-	}
+	if (value instanceof Error) return serializeError(value, (error) => error instanceof DbBranchReplayError ? { originalTraceId: error.originalTraceId } : {});
 	return value;
 }
 function serializeReplayResult(result) {
@@ -4216,7 +4204,7 @@ async function processItem(httpClient, serverItem, fn, experimentId, mockStrateg
 		} catch (e) {
 			if (replayStarted !== null) replayDurationMs = Math.round(performance.now() - replayStarted);
 			traceError = e;
-			error = errorMessage(e);
+			error = describeError(e);
 		}
 	} catch (e) {
 		if (replayStarted !== null) replayDurationMs = Math.round(performance.now() - replayStarted);
@@ -5700,7 +5688,7 @@ function wrapAsyncGenerator(source, spanStack, sendSpan, recordYielded) {
 			} else yielded.push(result.value);
 			return result;
 		} catch (error) {
-			finalize(error instanceof Error ? error.message : String(error));
+			finalize(describeError(error));
 			throw error;
 		}
 	};
@@ -5724,12 +5712,18 @@ function wrapAsyncGenerator(source, spanStack, sendSpan, recordYielded) {
 	return wrapped;
 }
 let cachedCollectorClass;
-async function loadCollectorClass() {
-	if (cachedCollectorClass !== void 0) return cachedCollectorClass;
-	const baml = await loadBamlModule();
+async function loadCollectorClass(options) {
+	if (cachedCollectorClass !== void 0 && (cachedCollectorClass === null || options?.clientRuntime === void 0)) return cachedCollectorClass;
+	const baml = await loadBamlModule(options);
 	if (!baml) return null;
-	cachedCollectorClass = baml.Collector;
-	return cachedCollectorClass;
+	const CollectorClass = baml.Collector;
+	if (isSharedBamlModule(baml)) cachedCollectorClass = CollectorClass;
+	return CollectorClass;
+}
+function bamlRuntimeOf(bamlClient) {
+	if (typeof bamlClient !== "object" || bamlClient === null) return;
+	const runtime = bamlClient.runtime;
+	return typeof runtime === "object" && runtime !== null ? runtime : void 0;
 }
 function extractPromptFromCollector(collector) {
 	try {
@@ -5849,13 +5843,19 @@ function invokeBamlWithCollector(invokeFn, CollectorClass) {
 	settle();
 	return result;
 }
+function warnIfAutoTracedBamlUntraced(CollectorClass) {
+	if (!CollectorClass) warnOnce("autoTrace-baml-collector-unavailable", `@boundaryml/baml could not be loaded, so BAML calls under a trace root are recorded without their prompt, model, or token usage. If @boundaryml/baml is installed but your bundler hides it, add \`import "@bitfab/sdk/baml"\` once in your server code.`);
+}
 function invokeAutoTracedBaml(invokeFn, isAsync) {
 	if (cachedCollectorClass) return invokeBamlWithCollector(invokeFn, cachedCollectorClass);
 	if (cachedCollectorClass === null || !isAsync) {
-		loadCollectorClass();
+		loadCollectorClass().then(warnIfAutoTracedBamlUntraced).catch(() => {});
 		return invokeFn();
 	}
-	return loadCollectorClass().then((CollectorClass) => CollectorClass ? invokeBamlWithCollector(invokeFn, CollectorClass) : invokeFn());
+	return loadCollectorClass().then((CollectorClass) => {
+		warnIfAutoTracedBamlUntraced(CollectorClass);
+		return CollectorClass ? invokeBamlWithCollector(invokeFn, CollectorClass) : invokeFn();
+	});
 }
 function isSpanType(value) {
 	return SPAN_TYPES.some((spanType) => spanType === value);
@@ -6130,7 +6130,7 @@ var Bitfab = class {
 	createAutoTraceRoot(traceFunctionKey, name, options, fn, rootFunctionName = "") {
 		const self = this;
 		self.simulationPlan.refresh();
-		loadCollectorClass();
+		loadCollectorClass({ searchLoadedModules: false });
 		installLangChainRunScopes();
 		const runRootFunction = (receiver, args) => runAsAutoTraceRootFunction(rootFunctionName, () => fn.apply(receiver, args));
 		const maxDepth = autoTraceLimit(options.maxDepth, DEFAULT_AUTO_TRACE_MAX_DEPTH);
@@ -6500,11 +6500,12 @@ var Bitfab = class {
 		}
 		const methodName = method.name;
 		if (!methodName) throw new BitfabError("wrapBAML requires a named function (e.g., b.ClassifyText).");
-		loadCollectorClass();
+		const collectorOptions = { clientRuntime: bamlRuntimeOf(bamlClient) };
+		loadCollectorClass(collectorOptions);
 		const wrappedFn = async (...args) => {
-			const CollectorClass = await loadCollectorClass();
+			const CollectorClass = await loadCollectorClass(collectorOptions);
 			if (!CollectorClass) {
-				warnOnce("wrapBAML-collector-unavailable", `@boundaryml/baml could not be loaded, so "${methodName}" and every other wrapBAML call run untraced: no prompt, model, or token usage is recorded and onCollector is not called. If @boundaryml/baml is installed and your app is bundled (Next.js with Turbopack, for example), add \`import "@bitfab/sdk/baml"\` once in your server code.`);
+				warnOnce("wrapBAML-collector-unavailable", `@boundaryml/baml could not be loaded, so "${methodName}" and every other wrapBAML call run untraced: no prompt, model, or token usage is recorded and onCollector is not called. If @boundaryml/baml is installed but your bundler hides it, add \`import "@bitfab/sdk/baml"\` once in your server code.`);
 				wrappedFn.collector = null;
 				return await bamlClient[methodName](...args);
 			}
@@ -6796,7 +6797,7 @@ var Bitfab = class {
 					} catch (error) {
 						sendSpan({
 							result: void 0,
-							error: error instanceof Error ? error.message : String(error)
+							error: describeError(error)
 						});
 						throw error;
 					}
@@ -6806,7 +6807,7 @@ var Bitfab = class {
 					}).catch((error) => {
 						sendSpan({
 							result: void 0,
-							error: error instanceof Error ? error.message : String(error)
+							error: describeError(error)
 						});
 						throw error;
 					});
@@ -7240,7 +7241,7 @@ var Bitfab = class {
 			await runWithSeedContext({ traceId }, async () => target(...args));
 		} finally {
 			try {
-				const { flushTraces } = await import("./http-DS1QjBvx.js").then((n) => n.a);
+				const { flushTraces } = await import("./http-BA7vmvuC.js").then((n) => n.a);
 				await flushTraces(3e4);
 			} finally {
 				unrecorded = activeTraceStates.delete(traceId);
@@ -7458,7 +7459,7 @@ async function seedFromRegistry(registry, pipeline, cases, options = {}) {
 		metadata: seedCase.metadata,
 		sessionId: seedCase.sessionId
 	}));
-	const { flushTraces } = await import("./http-DS1QjBvx.js").then((n) => n.a);
+	const { flushTraces } = await import("./http-BA7vmvuC.js").then((n) => n.a);
 	await flushTraces(3e4);
 	return {
 		pipeline,
@@ -7475,4 +7476,4 @@ function resolveTraceFunctionKey(registration) {
 //#endregion
 export { VerdictsClient as A, OrganizationMembersClient as C, BitfabLangGraphCallbackHandler as D, BitfabLangGraphIntegration as E, GraderRerunTimeoutError as F, BitfabClaudeAgentHandler as I, DEFAULT_SERVICE_URL as L, SUPPORTED_PROVIDERS as M, DatasetsClient as N, GradersClient as O, GraderRerunError as P, AssertionCategoriesClient as R, BitfabPiIntegration as S, NO_MOCK_OVERRIDE as T, BITFAB_PROGRESS_PREFIX as _, Bitfab as a, reportReplayProgress as b, getCurrentSpan as c, BitfabOpenAITracingProcessor as d, AssertionGenerationError as f, BitfabRivetIntegration as g, SpansClient as h, finalizers as i, ExperimentsClient as j, LabelsClient as k, getCurrentTrace as l, TracesClient as m, reseedFromRegistry as n, BitfabFunction as o, AssertionGenerationTimeoutError as p, seedFromRegistry as r, getCurrentReplayBranch as s, defineReplayRegistry as t, BitfabVercelAiHandler as u, DbBranchReplayError as v, BitfabOpenAIAgentHandler as w, serializeReplayResult as x, ReplayError as y, CategoriesClient as z };
 
-//# sourceMappingURL=src-BJjbR3Nq.js.map
+//# sourceMappingURL=src-CYs1uXoX.js.map