@bitfab/sdk 0.76.2 → 0.76.4

package/dist/{src-B6zSnPpb.cjs → src-C9V1wPM_.cjs}

@@ -1,5 +1,5 @@
-const require_bamlPeer = require("./bamlPeer-5Ohpq8q9.cjs");
-const require_http = require("./http-DscPFMlr.cjs");
+const require_bamlPeer = require("./bamlPeer-BxJO7-pm.cjs");
+const require_http = require("./http-Dw6XzN7P.cjs");
 const require_autoTrace = require("./autoTrace-pZKu5Nk2.cjs");
 const require_optionalPeer = require("./optionalPeer-CUH756lt.cjs");
 let superjson = require("superjson");
@@ -5067,7 +5067,6 @@ var SpansClient = class {
 //#region src/traces.ts
 const ASSERTIONS_PATH = "/api/sdk/traces/assertions";
 const SEARCH_PATH = "/api/sdk/traces/search";
-const CODE_DRAFT_SAVE_TIMEOUT_MS = 3e5;
 const DEFAULT_GENERATION_TIMEOUT_MS = 6e5;
 const GENERATION_POLL_INTERVAL_MS = 2e3;
 var AssertionGenerationError = class extends Error {
@@ -5117,9 +5116,6 @@ function toWireAssertion({ category_assertion_id, ...assertion }) {
 		category_id: categoryId
 	};
 }
-function asksBitfabToWriteCode(assertion) {
-	return (assertion.kind === "code" || assertion.kind === "hybrid") && assertion.code === void 0;
-}
 function traceAssertionsPath(traceId, suffix = "") {
 	return `/api/sdk/traces/${encodeURIComponent(traceId)}/assertions${suffix}`;
 }
@@ -5160,7 +5156,7 @@ var TracesClient = class {
 			source: params.source ?? "agent",
 			guidedRewrite: params.skipGuidedRewrite !== true
 		};
-		return withCategoryIds((params.updates.some((update) => update.assertions.some(asksBitfabToWriteCode)) ? await this.httpClient.request(ASSERTIONS_PATH, body, { timeout: CODE_DRAFT_SAVE_TIMEOUT_MS }) : await this.httpClient.request(ASSERTIONS_PATH, body)).assertions);
+		return withCategoryIds((await this.httpClient.request(ASSERTIONS_PATH, body)).assertions);
 	}
 	async generateAssertions(traceId, options = {}) {
 		const started = await this.httpClient.request(traceAssertionsPath(traceId, "/generate"), {
@@ -5712,12 +5708,18 @@ function wrapAsyncGenerator(source, spanStack, sendSpan, recordYielded) {
 	return wrapped;
 }
 let cachedCollectorClass;
-async function loadCollectorClass() {
-	if (cachedCollectorClass !== void 0) return cachedCollectorClass;
-	const baml = await require_bamlPeer.loadBamlModule();
+async function loadCollectorClass(options) {
+	if (cachedCollectorClass !== void 0 && (cachedCollectorClass === null || options?.clientRuntime === void 0)) return cachedCollectorClass;
+	const baml = await require_bamlPeer.loadBamlModule(options);
 	if (!baml) return null;
-	cachedCollectorClass = baml.Collector;
-	return cachedCollectorClass;
+	const CollectorClass = baml.Collector;
+	if (require_bamlPeer.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 {
@@ -5837,13 +5839,19 @@ function invokeBamlWithCollector(invokeFn, CollectorClass) {
 	settle();
 	return result;
 }
+function warnIfAutoTracedBamlUntraced(CollectorClass) {
+	if (!CollectorClass) require_autoTrace.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 require_http.SPAN_TYPES.some((spanType) => spanType === value);
@@ -6118,7 +6126,7 @@ var Bitfab = class {
 	createAutoTraceRoot(traceFunctionKey, name, options, fn, rootFunctionName = "") {
 		const self = this;
 		self.simulationPlan.refresh();
-		loadCollectorClass();
+		loadCollectorClass({ searchLoadedModules: false });
 		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);
@@ -6488,11 +6496,12 @@ var Bitfab = class {
 		}
 		const methodName = method.name;
 		if (!methodName) throw new require_http.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) {
-				require_autoTrace.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.`);
+				require_autoTrace.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);
 			}
@@ -7228,7 +7237,7 @@ var Bitfab = class {
 			await runWithSeedContext({ traceId }, async () => target(...args));
 		} finally {
 			try {
-				const { flushTraces } = await Promise.resolve().then(() => require("./http-DscPFMlr.cjs")).then((n) => n.http_exports);
+				const { flushTraces } = await Promise.resolve().then(() => require("./http-Dw6XzN7P.cjs")).then((n) => n.http_exports);
 				await flushTraces(3e4);
 			} finally {
 				unrecorded = activeTraceStates.delete(traceId);
@@ -7446,7 +7455,7 @@ async function seedFromRegistry(registry, pipeline, cases, options = {}) {
 		metadata: seedCase.metadata,
 		sessionId: seedCase.sessionId
 	}));
-	const { flushTraces } = await Promise.resolve().then(() => require("./http-DscPFMlr.cjs")).then((n) => n.http_exports);
+	const { flushTraces } = await Promise.resolve().then(() => require("./http-Dw6XzN7P.cjs")).then((n) => n.http_exports);
 	await flushTraces(3e4);
 	return {
 		pipeline,
@@ -7696,4 +7705,4 @@ Object.defineProperty(exports, "serializeReplayResult", {
 	}
 });
 
-//# sourceMappingURL=src-B6zSnPpb.cjs.map
+//# sourceMappingURL=src-C9V1wPM_.cjs.map