@bitfab/sdk 0.74.4 → 0.74.5

package/dist/{src-Dp8OX46R.js → src-Ch86hs9i.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-C_t7m9b_.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";
@@ -4958,6 +4958,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 +5008,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 +5043,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 +7084,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-C_t7m9b_.js").then((n) => n.a);
 				await flushTraces(3e4);
 			} finally {
 				unrecorded = activeTraceStates.delete(traceId);
@@ -7285,7 +7290,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-C_t7m9b_.js").then((n) => n.a);
 	await flushTraces(3e4);
 	return {
 		pipeline,
@@ -7302,4 +7307,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-Ch86hs9i.js.map