@bitfab/sdk 0.74.4 → 0.74.6

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-Bm2-Peq8.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-DdNn4qGL.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-Bm2-Peq8.js";
+import { a as runRegistryMain, c as runSeedCli, i as runIfEntrypoint, n as parseRegistryCommandArgs, r as resolveRegistry, t as importRegistry } from "./registryCommand-DdNn4qGL.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-Dp8OX46R.js → src-CRrvnhok.js}

@@ -1,5 +1,5 @@
 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-BM1TSXDd.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-Br-jOeI6.js";
 import { A as unmapOpenAiAgentsTrace, B as asyncStorageReady, C as recordAutoTracedClientCall, D as mapOpenAiAgentsTrace, E as endOpenAiAgentsSpan, F as warnOnce, H as isAsyncStorageInitDone, I as writeOnce, L as isCaptureSuppressed, M as installLangChainRunScopes, N as markLangChainRunRecorded, O as openAiAgentsParentSpanId, R as runWithCaptureSuppressed, S as frameworkRecordsClientCall, T as watchClientCall, V as createAsyncLocalStorage, _ as runWithAutoTraceRootContext, b as OPENROUTER, f as invokeThroughAutoTraceEntries, g as runWithAutoTraceNodeConfiguration, h as runWithAutoTraceFramesAt, j as forgetLangChainRun, k as openAiAgentsSpanChain, l as clearAutoTraceNodeConfiguration, m as runWithAutoTraceEntry, o as __bitfabAutoTraceActive, p as runAsAutoTraceRootFunction, u as currentAutoTraceEntries, v as ANTHROPIC, w as recordWatchedClientCall, x as clientCallProvider, y as OPENAI_RESPONSES } from "./autoTrace-kHOBg49M.js";
 import { t as importOptionalPeer } from "./optionalPeer-CczrzrXx.js";
 import { t as loadBamlModule } from "./bamlPeer-Dyeb3e5o.js";
@@ -51,7 +51,135 @@ function unserializableStub(value, reason) {
 function serializeValue(value) {
 	return runWithCaptureSuppressed(() => serializeWithSuperjson(value));
 }
+function exceedsSerializedBudget(value, budget, frameworkWalk = false, dropped) {
+	let total = 0;
+	let steps = 0;
+	const maxSteps = 4 * budget + 1024;
+	const giveUp = /* @__PURE__ */ new Error("size walk step limit");
+	const step = () => {
+		steps += 1;
+		if (steps > maxSteps) throw giveUp;
+	};
+	const found = [];
+	const ancestors = /* @__PURE__ */ new Set();
+	const visit = (node, depth, jsonRules) => {
+		if (total > budget) return;
+		step();
+		if (node === null) {
+			total += 4;
+			return;
+		}
+		switch (typeof node) {
+			case "string":
+				total += node.length + 2;
+				return;
+			case "number":
+				total += 1;
+				return;
+			case "boolean":
+				total += 4;
+				return;
+			case "undefined": return;
+		}
+		if (frameworkWalk && depth > MAX_SAFE_DEPTH) {
+			found.push(describeValue(node));
+			total += 2;
+			return;
+		}
+		if (typeof node === "bigint") {
+			total += 1;
+			return;
+		}
+		if (typeof node !== "object") {
+			found.push(describeValue(node));
+			return;
+		}
+		if (ancestors.has(node)) {
+			found.push(describeValue(node));
+			total += 4;
+			return;
+		}
+		if (!jsonRules && registeredOnSuperjson(node)) {
+			total += 1;
+			return;
+		}
+		if (ArrayBuffer.isView(node)) {
+			total += node instanceof DataView ? 2 : Math.max(2, 2 * node.length);
+			return;
+		}
+		if (typeof node.toJSON === "function") {
+			total += 2;
+			return;
+		}
+		if (!jsonRules && node instanceof Error) {
+			total += node.name.length + node.message.length + 4;
+			return;
+		}
+		total += 2;
+		const childRules = jsonRules || !walkedBySuperjson(node);
+		ancestors.add(node);
+		let written = 0;
+		const separate = () => {
+			if (written > 0) total += 1;
+			written += 1;
+		};
+		if (Array.isArray(node)) for (let index = 0; index < node.length && total <= budget; index += 1) {
+			if (!jsonRules && !(index in node)) {
+				step();
+				continue;
+			}
+			separate();
+			visit(node[index], depth + 1, childRules);
+		}
+		else if (jsonRules && (node instanceof Map || node instanceof Set)) {} else if (node instanceof Map) for (const [key, entry] of node) {
+			if (total > budget) break;
+			separate();
+			total += 3;
+			visit(key, depth + 1, childRules);
+			visit(entry, depth + 1, childRules);
+		}
+		else if (node instanceof Set) for (const entry of node) {
+			if (total > budget) break;
+			separate();
+			visit(entry, depth + 1, childRules);
+		}
+		else for (const key of Object.keys(node)) {
+			if (total > budget) break;
+			if (frameworkWalk && key.startsWith("_")) continue;
+			const entry = node[key];
+			if (entry !== void 0 && typeof entry !== "function" && typeof entry !== "symbol") {
+				separate();
+				total += key.length + 3;
+			}
+			visit(entry, depth + 1, childRules);
+		}
+		ancestors.delete(node);
+	};
+	try {
+		visit(value, 0, frameworkWalk);
+	} catch {
+		return false;
+	}
+	if (total <= budget) return false;
+	dropped?.push(...found);
+	return true;
+}
+const superjsonDefault = superjson.defaultInstance;
+function registeredOnSuperjson(node) {
+	if (superjsonDefault === void 0) return false;
+	const ownConstructor = node.constructor;
+	return superjsonDefault.customTransformerRegistry.findApplicable(node) !== void 0 || typeof ownConstructor === "function" && superjsonDefault.classRegistry.getIdentifier(ownConstructor) !== void 0;
+}
+function walkedBySuperjson(node) {
+	if (Array.isArray(node) || node instanceof Map || node instanceof Set) return true;
+	const prototype = Object.getPrototypeOf(node);
+	return prototype === null || prototype === Object.prototype;
+}
+function tooLargeReason(budget) {
+	return `too_large_over_${budget}_bytes`;
+}
 function serializeWithSuperjson(value) {
+	if (exceedsSerializedBudget(value, MAX_SERIALIZED_BYTES)) return unserializableStub(value, tooLargeReason(MAX_SERIALIZED_BYTES));
 	try {
 		const { json, meta } = superjson.serialize(value);
 		let size;
@@ -89,22 +217,24 @@ function toJsonSafeReport(value) {
 }
 function jsonSafeReport(value) {
 	const dropped = [];
+	if (exceedsSerializedBudget(value, MAX_FRAMEWORK_SERIALIZED_BYTES, true, dropped)) return frameworkTooLarge(tooLargeReason(MAX_FRAMEWORK_SERIALIZED_BYTES), dropped);
 	const safe = toJsonSafeInner(value, 0, /* @__PURE__ */ new WeakSet(), dropped);
 	try {
 		const size = JSON.stringify(safe)?.length ?? 0;
-		if (size > MAX_FRAMEWORK_SERIALIZED_BYTES) {
-			warnOnce("toJsonSafe:too_large", `a framework payload exceeded ${MAX_FRAMEWORK_SERIALIZED_BYTES} bytes and was replaced with a placeholder so the span still ships. The captured state for this span is incomplete.`);
-			return {
-				safe: `<unserializable: too_large_${size}_bytes>`,
-				dropped: [...dropped, `too_large_${size}_bytes`]
-			};
-		}
+		if (size > MAX_FRAMEWORK_SERIALIZED_BYTES) return frameworkTooLarge(`too_large_${size}_bytes`, dropped);
 	} catch {}
 	return {
 		safe,
 		dropped
 	};
 }
+function frameworkTooLarge(reason, dropped) {
+	warnOnce("toJsonSafe:too_large", `a framework payload exceeded ${MAX_FRAMEWORK_SERIALIZED_BYTES} bytes and was replaced with a placeholder so the span still ships. The captured state for this span is incomplete.`);
+	return {
+		safe: `<unserializable: ${reason}>`,
+		dropped: [...dropped, "too_large"]
+	};
+}
 function toJsonSafeInner(value, depth, seen, dropped) {
 	if (value === null || value === void 0) return value;
 	if (typeof value === "string" || typeof value === "number" || typeof value === "boolean") return value;
@@ -4958,6 +5088,7 @@ 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 {
@@ -5007,6 +5138,9 @@ 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}`;
 }
@@ -5039,14 +5173,15 @@ var TracesClient = class {
 	}
 	async saveAssertionsAll(params) {
 		if (params.updates.length === 0) return [];
-		return withCategoryIds((await this.httpClient.request(ASSERTIONS_PATH, {
+		const body = {
 			updates: params.updates.map((update) => ({
 				...update,
 				assertions: update.assertions.map(toWireAssertion)
 			})),
 			source: params.source ?? "agent",
 			guidedRewrite: params.skipGuidedRewrite !== true
-		})).assertions);
+		};
+		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);
 	}
 	async generateAssertions(traceId, options = {}) {
 		const started = await this.httpClient.request(traceAssertionsPath(traceId, "/generate"), {
@@ -7079,7 +7214,7 @@ var Bitfab = class {
 			await runWithSeedContext({ traceId }, async () => target(...args));
 		} finally {
 			try {
-				const { flushTraces } = await import("./http-BM1TSXDd.js").then((n) => n.a);
+				const { flushTraces } = await import("./http-Br-jOeI6.js").then((n) => n.a);
 				await flushTraces(3e4);
 			} finally {
 				unrecorded = activeTraceStates.delete(traceId);
@@ -7285,7 +7420,7 @@ async function seedFromRegistry(registry, pipeline, cases, options = {}) {
 		metadata: seedCase.metadata,
 		sessionId: seedCase.sessionId
 	}));
-	const { flushTraces } = await import("./http-BM1TSXDd.js").then((n) => n.a);
+	const { flushTraces } = await import("./http-Br-jOeI6.js").then((n) => n.a);
 	await flushTraces(3e4);
 	return {
 		pipeline,
@@ -7302,4 +7437,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-Dp8OX46R.js.map
+//# sourceMappingURL=src-CRrvnhok.js.map