@bitfab/sdk 0.74.3 → 0.74.5

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

@@ -1,5 +1,5 @@
 const require_bamlPeer = require("./bamlPeer-5Ohpq8q9.cjs");
-const require_http = require("./http-BHgwaBjv.cjs");
+const require_http = require("./http-DNmzk2gf.cjs");
 const require_autoTrace = require("./autoTrace-DPWV1Xzr.cjs");
 const require_optionalPeer = require("./optionalPeer-CUH756lt.cjs");
 let superjson = require("superjson");
@@ -3735,6 +3735,7 @@ const RESUME_CONFLICTING_OPTIONS = [
 	"attempts",
 	"graderIds",
 	"onlyWithAssertions",
+	"categoryIds",
 	"dryRun",
 	"experimentalSelectiveReplay"
 ];
@@ -3842,6 +3843,7 @@ const REPLAY_OPTION_NAMES = [
 	"datasetIds",
 	"graderIds",
 	"onlyWithAssertions",
+	"categoryIds",
 	"skipAssertionJudging",
 	"judgeAssertions",
 	"adaptInputs",
@@ -4212,6 +4214,10 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
 		if (options.traceIds.length > 100) throw new require_http.BitfabError(`traceIds supports at most 100 trace IDs per replay (got ${options.traceIds.length}).`);
 	}
 	const datasetIds = resolveDatasetIds(options);
+	if (options?.categoryIds !== void 0) {
+		if (options.categoryIds.length === 0) throw new require_http.BitfabError("categoryIds must contain at least one category ID.");
+		if (options.traceIds === void 0 && datasetIds === void 0) throw new require_http.BitfabError("categoryIds narrows a trace ID or dataset selection. Pass it with traceIds, datasetId or datasetIds.");
+	}
 	if (options?.limit !== void 0 && options?.traceIds !== void 0) try {
 		console.warn("Bitfab: limit is ignored when traceIds is passed: the explicit trace ID list already determines how many traces replay.");
 	} catch {}
@@ -4223,7 +4229,7 @@ async function replay(httpClient, serviceUrl, traceFunctionKey, fn, options, reg
 	const includeDbBranchLease = dbBranchEnabled(options?.dbBranch) && options?.dryRun !== true;
 	let session;
 	if (options?.resume === void 0) {
-		const started = await httpClient.startReplay(traceFunctionKey, options?.traceIds ? void 0 : options?.limit ?? 5, options?.traceIds, options?.name, codeChangeDescription, codeChangeFiles, includeDbBranchLease, options?.experimentGroupId, datasetIds, options?.graderIds, resolveDbBranchSettings(options?.dbBranch), requestedAttempts, options?.adaptInputs !== void 0, options?.onlyWithAssertions, options?.skipAssertionJudging === true || options?.judgeAssertions === false, options?.experimentalSelectiveReplay, options?.notes, options?.metadata, options?.dryRun === true, options?.experimentGroupName, safeReplayInvocation(() => buildReplayInvocation(Object.entries(cliFlags(options) ?? explicitReplayOptions(options)))), codeChangeType);
+		const started = await httpClient.startReplay(traceFunctionKey, options?.traceIds ? void 0 : options?.limit ?? 5, options?.traceIds, options?.name, codeChangeDescription, codeChangeFiles, includeDbBranchLease, options?.experimentGroupId, datasetIds, options?.graderIds, resolveDbBranchSettings(options?.dbBranch), requestedAttempts, options?.adaptInputs !== void 0, options?.onlyWithAssertions, options?.skipAssertionJudging === true || options?.judgeAssertions === false, options?.experimentalSelectiveReplay, options?.notes, options?.metadata, options?.dryRun === true, options?.experimentGroupName, safeReplayInvocation(() => buildReplayInvocation(Object.entries(cliFlags(options) ?? explicitReplayOptions(options)))), codeChangeType, options?.categoryIds);
 		if (started.items.length === 0) try {
 			console.warn(`Bitfab: no traces matched "${traceFunctionKey}", so this replay ran nothing. Capture a trace, or seed one with seedTrace, before replaying.`);
 		} catch {}
@@ -4952,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 {
@@ -5001,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}`;
 }
@@ -5033,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"), {
@@ -7073,7 +7084,7 @@ var Bitfab = class {
 			await runWithSeedContext({ traceId }, async () => target(...args));
 		} finally {
 			try {
-				const { flushTraces } = await Promise.resolve().then(() => require("./http-BHgwaBjv.cjs")).then((n) => n.http_exports);
+				const { flushTraces } = await Promise.resolve().then(() => require("./http-DNmzk2gf.cjs")).then((n) => n.http_exports);
 				await flushTraces(3e4);
 			} finally {
 				unrecorded = activeTraceStates.delete(traceId);
@@ -7279,7 +7290,7 @@ async function seedFromRegistry(registry, pipeline, cases, options = {}) {
 		metadata: seedCase.metadata,
 		sessionId: seedCase.sessionId
 	}));
-	const { flushTraces } = await Promise.resolve().then(() => require("./http-BHgwaBjv.cjs")).then((n) => n.http_exports);
+	const { flushTraces } = await Promise.resolve().then(() => require("./http-DNmzk2gf.cjs")).then((n) => n.http_exports);
 	await flushTraces(3e4);
 	return {
 		pipeline,
@@ -7529,4 +7540,4 @@ Object.defineProperty(exports, "serializeReplayResult", {
 	}
 });
 
-//# sourceMappingURL=src-KAtu1Cmn.cjs.map
+//# sourceMappingURL=src-B7Xv4aR2.cjs.map