@bitfab/sdk 0.76.2 → 0.76.4

package/dist/{src-CcYVtNMw.js → src-CFUOdE8n.js}

@@ -1,8 +1,8 @@
 import { t as __exportAll } from "./rolldown-runtime-D7D4PA-g.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-wadCG_ec.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-CQFVgQ4S.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";
@@ -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 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 {
@@ -5837,13 +5839,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);
@@ -6118,7 +6126,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);
@@ -6488,11 +6496,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);
 			}
@@ -7228,7 +7237,7 @@ var Bitfab = class {
 			await runWithSeedContext({ traceId }, async () => target(...args));
 		} finally {
 			try {
-				const { flushTraces } = await import("./http-wadCG_ec.js").then((n) => n.a);
+				const { flushTraces } = await import("./http-CQFVgQ4S.js").then((n) => n.a);
 				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 import("./http-wadCG_ec.js").then((n) => n.a);
+	const { flushTraces } = await import("./http-CQFVgQ4S.js").then((n) => n.a);
 	await flushTraces(3e4);
 	return {
 		pipeline,
@@ -7463,4 +7472,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-CcYVtNMw.js.map
+//# sourceMappingURL=src-CFUOdE8n.js.map