@bitfab/sdk 0.53.5 → 0.54.1

package/dist/index.cjs

@@ -42,7 +42,7 @@ var __version__, __packageName__;
 var init_version_generated = __esm({
   "src/version.generated.ts"() {
     "use strict";
-    __version__ = "0.53.5";
+    __version__ = "0.54.1";
     __packageName__ = "@bitfab/sdk";
   }
 });
@@ -186,6 +186,206 @@ var init_errors = __esm({
   }
 });
 
+// src/gitCommand.ts
+function isUnrefable(value) {
+  return typeof value === "object" && value !== null && "unref" in value && typeof value.unref === "function";
+}
+function unrefStream(stream) {
+  if (isUnrefable(stream)) {
+    stream.unref();
+  }
+}
+function terminate(child) {
+  if (child.pid && process.platform !== "win32") {
+    try {
+      process.kill(-child.pid, "SIGTERM");
+      return;
+    } catch {
+      child.kill("SIGTERM");
+      return;
+    }
+  }
+  child.kill("SIGTERM");
+}
+async function gitRunner(options) {
+  let spawn;
+  try {
+    ;
+    ({ spawn } = await import("child_process"));
+  } catch {
+    return null;
+  }
+  return (dir, args, env) => new Promise((resolve) => {
+    let child;
+    try {
+      child = spawn("git", args, {
+        cwd: dir,
+        detached: process.platform !== "win32",
+        stdio: ["ignore", "pipe", "pipe"],
+        ...env ? { env: { ...process.env, ...env } } : {}
+      });
+    } catch {
+      resolve(null);
+      return;
+    }
+    const chunks = [];
+    let size = 0;
+    let overflow = false;
+    let exitCode = null;
+    let done = false;
+    const timers = [];
+    const schedule = (callback, ms) => {
+      const timer = setTimeout(callback, ms);
+      if (!options.keepProcessAlive) {
+        timer.unref();
+      }
+      timers.push(timer);
+    };
+    const finish = () => {
+      if (done) {
+        return;
+      }
+      done = true;
+      for (const timer of timers) {
+        clearTimeout(timer);
+      }
+      child.stdout?.destroy();
+      child.stderr?.destroy();
+      resolve(
+        exitCode === 0 && !overflow ? Buffer.concat(chunks).toString("utf8") : null
+      );
+    };
+    child.stdout?.on("data", (chunk) => {
+      size += chunk.length;
+      if (size > options.maxBuffer) {
+        overflow = true;
+        terminate(child);
+        return;
+      }
+      chunks.push(chunk);
+    });
+    child.stderr?.resume();
+    child.on("error", () => {
+      exitCode = null;
+      finish();
+    });
+    child.on("exit", (code) => {
+      exitCode = code;
+      schedule(finish, EXIT_GRACE_MS);
+    });
+    child.on("close", (code) => {
+      exitCode = code ?? exitCode;
+      finish();
+    });
+    schedule(() => terminate(child), options.timeoutMs);
+    if (!options.keepProcessAlive) {
+      child.unref();
+      unrefStream(child.stdout);
+      unrefStream(child.stderr);
+    }
+  });
+}
+var EXIT_GRACE_MS;
+var init_gitCommand = __esm({
+  "src/gitCommand.ts"() {
+    "use strict";
+    EXIT_GRACE_MS = 500;
+  }
+});
+
+// src/gitState.ts
+function sha(value) {
+  const trimmed = value?.trim();
+  return trimmed && SHA_PATTERN.test(trimmed) ? trimmed : null;
+}
+function text(value) {
+  const trimmed = value?.trim();
+  return trimmed ? trimmed : null;
+}
+function isEmptyGitState(state) {
+  return Object.values(state).every((value) => value === null);
+}
+async function resolveGitState(cwd) {
+  const git = await gitRunner({
+    timeoutMs: GIT_TIMEOUT_MS,
+    maxBuffer: GIT_MAX_BUFFER,
+    keepProcessAlive: true
+  });
+  if (!git) {
+    return null;
+  }
+  const run = async (args, env) => (await git(cwd, args, env))?.trim() ?? null;
+  const [refs, email, branch] = await Promise.all([
+    run(["rev-parse", "--show-toplevel", "HEAD", "HEAD^{tree}"]),
+    run(["config", "user.email"]),
+    run(["symbolic-ref", "--short", "-q", "HEAD"])
+  ]);
+  const [root, commitSha, baseSha] = (refs ?? "").split("\n");
+  if (!root) {
+    return null;
+  }
+  const state = {
+    githubEmail: text(email),
+    branch: text(branch),
+    commitSha: sha(commitSha),
+    baseSha: sha(baseSha),
+    experimentSha: await resolveWorkingTreeSha(run, root.trim())
+  };
+  return isEmptyGitState(state) ? null : state;
+}
+async function resolveWorkingTreeSha(run, root) {
+  let indexPath;
+  let cleanup;
+  try {
+    const { mkdtemp, rm } = await import("fs/promises");
+    const { join } = await import("path");
+    const { tmpdir } = await import("os");
+    const dir = await mkdtemp(join(tmpdir(), "bitfab-git-"));
+    indexPath = join(dir, "index");
+    cleanup = () => rm(dir, { recursive: true, force: true });
+  } catch {
+    return null;
+  }
+  try {
+    const env = { GIT_INDEX_FILE: indexPath };
+    if (await run(["read-tree", "HEAD"], env) === null) {
+      return null;
+    }
+    if (await run(["add", "-A", "--", root], env) === null) {
+      return null;
+    }
+    return sha(await run(["write-tree"], env));
+  } catch {
+    return null;
+  } finally {
+    await cleanup().catch(() => {
+    });
+  }
+}
+function resolvedGitState() {
+  if (typeof process === "undefined" || process.env?.[DISABLE_ENV]) {
+    return Promise.resolve(null);
+  }
+  let cwd;
+  try {
+    cwd = process.cwd?.() ?? ".";
+  } catch {
+    return Promise.resolve(null);
+  }
+  return resolveGitState(cwd).catch(() => null);
+}
+var GIT_TIMEOUT_MS, GIT_MAX_BUFFER, DISABLE_ENV, SHA_PATTERN;
+var init_gitState = __esm({
+  "src/gitState.ts"() {
+    "use strict";
+    init_gitCommand();
+    GIT_TIMEOUT_MS = 3e4;
+    GIT_MAX_BUFFER = 1024 * 1024;
+    DISABLE_ENV = "BITFAB_DISABLE_GIT_STATE";
+    SHA_PATTERN = /^[0-9a-f]{40}$/;
+  }
+});
+
 // src/asyncStorage.ts
 function registerAsyncLocalStorageClass(cls) {
   if (!AsyncLocalStorageClass) {
@@ -1946,6 +2146,7 @@ var init_http = __esm({
     init_compress();
     init_constants();
     init_errors();
+    init_gitState();
     init_replayContext();
     init_serializePayload();
     init_simulationPlan();
@@ -2557,6 +2758,10 @@ var init_http = __esm({
         if (onlyWithAssertions) {
           payload.onlyWithAssertions = true;
         }
+        const git = await resolvedGitState();
+        if (git && !isEmptyGitState(git)) {
+          payload.git = git;
+        }
         const timeout = includeDbBranchLease ? REPLAY_DB_BRANCH_REQUEST_TIMEOUT_MS : 3e4;
         return this.request("/api/sdk/replay/start", payload, {
           timeout
@@ -2904,112 +3109,6 @@ var init_randomUuid = __esm({
   }
 });
 
-// src/gitCommand.ts
-function isUnrefable(value) {
-  return typeof value === "object" && value !== null && "unref" in value && typeof value.unref === "function";
-}
-function unrefStream(stream) {
-  if (isUnrefable(stream)) {
-    stream.unref();
-  }
-}
-function terminate(child) {
-  if (child.pid && process.platform !== "win32") {
-    try {
-      process.kill(-child.pid, "SIGTERM");
-      return;
-    } catch {
-      child.kill("SIGTERM");
-      return;
-    }
-  }
-  child.kill("SIGTERM");
-}
-async function gitRunner(options) {
-  let spawn;
-  try {
-    ;
-    ({ spawn } = await import("child_process"));
-  } catch {
-    return null;
-  }
-  return (dir, args) => new Promise((resolve) => {
-    let child;
-    try {
-      child = spawn("git", args, {
-        cwd: dir,
-        detached: process.platform !== "win32",
-        stdio: ["ignore", "pipe", "pipe"]
-      });
-    } catch {
-      resolve(null);
-      return;
-    }
-    const chunks = [];
-    let size = 0;
-    let overflow = false;
-    let exitCode = null;
-    let done = false;
-    const timers = [];
-    const schedule = (callback, ms) => {
-      const timer = setTimeout(callback, ms);
-      if (!options.keepProcessAlive) {
-        timer.unref();
-      }
-      timers.push(timer);
-    };
-    const finish = () => {
-      if (done) {
-        return;
-      }
-      done = true;
-      for (const timer of timers) {
-        clearTimeout(timer);
-      }
-      child.stdout?.destroy();
-      child.stderr?.destroy();
-      resolve(
-        exitCode === 0 && !overflow ? Buffer.concat(chunks).toString("utf8") : null
-      );
-    };
-    child.stdout?.on("data", (chunk) => {
-      size += chunk.length;
-      if (size > options.maxBuffer) {
-        overflow = true;
-        terminate(child);
-        return;
-      }
-      chunks.push(chunk);
-    });
-    child.stderr?.resume();
-    child.on("error", () => {
-      exitCode = null;
-      finish();
-    });
-    child.on("exit", (code) => {
-      exitCode = code;
-      schedule(finish, EXIT_GRACE_MS);
-    });
-    child.on("close", (code) => {
-      exitCode = code ?? exitCode;
-      finish();
-    });
-    schedule(() => terminate(child), options.timeoutMs);
-    if (!options.keepProcessAlive) {
-      child.unref();
-      unrefStream(child.stdout);
-      unrefStream(child.stderr);
-    }
-  });
-}
-var EXIT_GRACE_MS;
-var init_gitCommand = __esm({
-  "src/gitCommand.ts"() {
-    "use strict";
-    EXIT_GRACE_MS = 500;
-  }
-});
-
 // src/mockOverride.ts
 function resolveMockValue(value, ctx) {
   return typeof value === "function" ? value(ctx) : value;
@@ -4062,6 +4161,7 @@ var init_replay = __esm({
 // src/index.ts
 var index_exports = {};
 __export(index_exports, {
+  AssertionCategoriesClient: () => AssertionCategoriesClient,
   BITFAB_PROGRESS_PREFIX: () => BITFAB_PROGRESS_PREFIX,
   Bitfab: () => Bitfab,
   BitfabClaudeAgentHandler: () => BitfabClaudeAgentHandler,
@@ -4098,6 +4198,36 @@ __export(index_exports, {
 });
 module.exports = __toCommonJS(index_exports);
 
+// src/assertionCategories.ts
+var CATEGORIES_PATH = "/api/sdk/assertionCategories";
+var AssertionCategoriesClient = class {
+  constructor(httpClient) {
+    this.httpClient = httpClient;
+  }
+  /** Create a category, or update it by ID. Omitted descriptions are preserved. */
+  async save(params) {
+    const response = await this.httpClient.request(CATEGORIES_PATH, params);
+    return response.category;
+  }
+  /** Read one category in the API key's organization. */
+  async get(id) {
+    const response = await this.httpClient.get(
+      `${CATEGORIES_PATH}/${encodeURIComponent(id)}`
+    );
+    return response.category;
+  }
+  /** List the organization's categories ordered by title. */
+  async list() {
+    const response = await this.httpClient.get(CATEGORIES_PATH);
+    return response.categories;
+  }
+  /** Delete a category and clear its assignments, preserving assertions and verdicts. */
+  async delete(id) {
+    const response = await this.httpClient.request(`${CATEGORIES_PATH}/${encodeURIComponent(id)}`, {}, { method: "DELETE" });
+    return response.category;
+  }
+};
+
 // src/assertions.ts
 var ASSERTIONS_PATH = "/api/sdk/traces/assertions";
 function traceAssertionsPath(traceId, suffix = "") {
@@ -5280,9 +5410,9 @@ function assertSurfacesCompatible(requested, resolved2, parentSurface, traceFunc
 init_gitCommand();
 init_readEnv();
 var EXPLICIT_SHA_ENV = "BITFAB_COMMIT_SHA";
-var DISABLE_ENV = "BITFAB_DISABLE_COMMIT_REF";
-var GIT_TIMEOUT_MS = 2e3;
-var GIT_MAX_BUFFER = 1024 * 1024;
+var DISABLE_ENV2 = "BITFAB_DISABLE_COMMIT_REF";
+var GIT_TIMEOUT_MS2 = 2e3;
+var GIT_MAX_BUFFER2 = 1024 * 1024;
 var VERCEL_PROVIDER_HOSTS = {
   github: "github.com",
   gitlab: "gitlab.com",
@@ -5338,9 +5468,9 @@ var PLATFORM_ENVS = [
 ];
 var COMMIT_REF_ENV_NAMES = [
   EXPLICIT_SHA_ENV,
-  DISABLE_ENV,
+  DISABLE_ENV2,
   ...PLATFORM_ENVS.flatMap(
-    ([sha, branch]) => branch === null ? [sha] : [sha, branch]
+    ([sha2, branch]) => branch === null ? [sha2] : [sha2, branch]
   ),
   "GITHUB_SERVER_URL",
   "GITHUB_REPOSITORY",
@@ -5385,7 +5515,7 @@ function normalizeRemote(raw) {
 }
 function resolveCommitRefFromEnv(env) {
   const explicit = read(env, EXPLICIT_SHA_ENV);
-  const platform = PLATFORM_ENVS.find(([sha]) => read(env, sha) !== null);
+  const platform = PLATFORM_ENVS.find(([sha2]) => read(env, sha2) !== null);
   if (!platform) {
     if (explicit === null) {
       return null;
@@ -5409,16 +5539,16 @@ function resolveCommitRefFromEnv(env) {
 }
 async function resolveCommitRefFromGit(cwd) {
   const git = await gitRunner({
-    timeoutMs: GIT_TIMEOUT_MS,
-    maxBuffer: GIT_MAX_BUFFER,
+    timeoutMs: GIT_TIMEOUT_MS2,
+    maxBuffer: GIT_MAX_BUFFER2,
     keepProcessAlive: false
   });
   if (!git) {
     return null;
   }
   const run = async (args) => (await git(cwd, args))?.trim() ?? null;
-  const sha = await run(["rev-parse", "HEAD"]);
-  if (!sha) {
+  const sha2 = await run(["rev-parse", "HEAD"]);
+  if (!sha2) {
     return null;
   }
   const [branch, status, remote, roots] = await Promise.all([
@@ -5428,7 +5558,7 @@ async function resolveCommitRefFromGit(cwd) {
     run(["rev-list", "--max-parents=0", "HEAD"])
   ]);
   return {
-    sha,
+    sha: sha2,
     branch: branch || null,
     dirty: status === null ? null : status.length > 0,
     remote: normalizeRemote(remote),
@@ -5452,7 +5582,7 @@ function startCommitRefResolution() {
   if (resolved || gitStarted) {
     return;
   }
-  if (read(readEnv, DISABLE_ENV) !== null) {
+  if (read(readEnv, DISABLE_ENV2) !== null) {
     resolved = true;
     return;
   }
@@ -7033,14 +7163,14 @@ function modelLabel(model) {
 }
 function summarizeGenerate(result, model) {
   const content = Array.isArray(result.content) ? result.content : [];
-  const text = typeof result.text === "string" ? result.text : content.filter((p) => p.type === "text" && typeof p.text === "string").map((p) => p.text).join("");
+  const text2 = typeof result.text === "string" ? result.text : content.filter((p) => p.type === "text" && typeof p.text === "string").map((p) => p.text).join("");
   const toolCalls = content.filter((p) => p.type === "tool-call").map((p) => ({
     toolCallId: p.toolCallId,
     toolName: p.toolName,
     input: p.input ?? p.args
   }));
   const summary = {
-    text,
+    text: text2,
     toolCalls: toolCalls.length > 0 ? toolCalls : void 0,
     usage: result.usage,
     finishReason: result.finishReason
@@ -7052,7 +7182,7 @@ function summarizeGenerate(result, model) {
 }
 function accumulateStream(source, onComplete, model) {
   const reader = source.getReader();
-  let text = "";
+  let text2 = "";
   const toolCalls = [];
   let usage;
   let finishReason;
@@ -7074,7 +7204,7 @@ function accumulateStream(source, onComplete, model) {
       }
     }
     onComplete({
-      text,
+      text: text2,
       toolCalls: toolCalls.length > 0 ? toolCalls : void 0,
       usage,
       finishReason,
@@ -7119,7 +7249,7 @@ function accumulateStream(source, onComplete, model) {
           }
           try {
             if (part?.type === "text-delta") {
-              text += part.delta ?? part.textDelta ?? "";
+              text2 += part.delta ?? part.textDelta ?? "";
             } else if (part?.type === "tool-call") {
               toolCalls.push({
                 toolCallId: part.toolCallId,
@@ -7649,6 +7779,7 @@ var Bitfab = class {
     this.httpClient.releaseHeldExternalSpans = (timeoutMs) => this.simulationPlan.release(timeoutMs);
     this.httpClient.stopSimulationPlan = () => this.simulationPlan.stop();
     this.datasets = new DatasetsClient(this.httpClient);
+    this.assertionCategories = new AssertionCategoriesClient(this.httpClient);
     this.traces = new TracesClient(this.httpClient);
     this.labels = new LabelsClient(this.httpClient);
     this.graders = new GradersClient(this.httpClient);
@@ -9607,7 +9738,7 @@ async function settle(value) {
 }
 async function aiSdk(result) {
   const r = result ?? {};
-  const [text, usage, totalUsage, finishReason, toolCalls, toolResults] = await Promise.all([
+  const [text2, usage, totalUsage, finishReason, toolCalls, toolResults] = await Promise.all([
     settle(r.text),
     settle(r.usage),
     settle(r.totalUsage),
@@ -9616,7 +9747,7 @@ async function aiSdk(result) {
     settle(r.toolResults)
   ]);
   return {
-    text,
+    text: text2,
     usage: totalUsage ?? usage,
     finishReason,
     toolCalls,
@@ -9720,6 +9851,7 @@ function resolveTraceFunctionKey(registration) {
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  AssertionCategoriesClient,
   BITFAB_PROGRESS_PREFIX,
   Bitfab,
   BitfabClaudeAgentHandler,