@bitfab/sdk 0.74.4 → 0.74.6

package/dist/{src-Cr4ZJLeB.cjs → src-atyS216N.cjs}

@@ -1,5 +1,5 @@
 const require_bamlPeer = require("./bamlPeer-5Ohpq8q9.cjs");
-const require_http = require("./http-VtGRzeNI.cjs");
+const require_http = require("./http-C_f9vk_C.cjs");
 const require_autoTrace = require("./autoTrace-DPWV1Xzr.cjs");
 const require_optionalPeer = require("./optionalPeer-CUH756lt.cjs");
 let superjson = require("superjson");
@@ -51,7 +51,135 @@ function unserializableStub(value, reason) {
 function serializeValue(value) {
 	return require_autoTrace.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.default.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.default.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) {
-			require_autoTrace.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) {
+	require_autoTrace.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 Promise.resolve().then(() => require("./http-VtGRzeNI.cjs")).then((n) => n.http_exports);
+				const { flushTraces } = await Promise.resolve().then(() => require("./http-C_f9vk_C.cjs")).then((n) => n.http_exports);
 				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 Promise.resolve().then(() => require("./http-VtGRzeNI.cjs")).then((n) => n.http_exports);
+	const { flushTraces } = await Promise.resolve().then(() => require("./http-C_f9vk_C.cjs")).then((n) => n.http_exports);
 	await flushTraces(3e4);
 	return {
 		pipeline,
@@ -7535,4 +7670,4 @@ Object.defineProperty(exports, "serializeReplayResult", {
 	}
 });
 
-//# sourceMappingURL=src-Cr4ZJLeB.cjs.map
+//# sourceMappingURL=src-atyS216N.cjs.map