bitfab 0.64.0 → 0.64.2

package/cloudReplay.py

@@ -413,7 +413,7 @@ def configure_secrets(argv):
                 *(["--env", environment] if environment else []),
             ],
             input=values.get(name, values.get(targets[name])),
-            env=None if token is None else {**os.environ, "GH_TOKEN": token},
+            env={**os.environ, "GH_TOKEN": token},
         )
     result["set"] = [targets[name] for name in found]
     if missing or empty:
@@ -425,8 +425,9 @@ def configure_secrets(argv):
 def github_access():
     if shutil.which("gh") is not None:
         with contextlib.suppress(RuntimeError, OSError, subprocess.SubprocessError):
-            command(["gh", "auth", "status", "--hostname", "github.com"])
-            return {"source": "the GitHub CLI", "token": None}
+            token = command(["gh", "auth", "token", "--hostname", "github.com"])
+            if token:
+                return {"source": "the GitHub CLI", "token": token}
     for name in ("GH_TOKEN", "GITHUB_TOKEN"):
         if os.environ.get(name):
             return {"source": name, "token": os.environ[name]}
@@ -467,26 +468,6 @@ def api(repo, suffix, *, method="GET", payload=None, raw=False):
     path = f"repos/{repo}" + (f"/{suffix}" if suffix else "")
     access = github_access()
     data = None if payload is None else json.dumps(payload)
-    if access["token"] is None:
-        result = command(
-            [
-                "gh",
-                "api",
-                "--hostname",
-                "github.com",
-                "-H",
-                f"X-GitHub-Api-Version: {API_VERSION}",
-                *([] if raw else ["--jq", "tojson"]),
-                path,
-                "--method",
-                method,
-                *(["--input", "-"] if data is not None else []),
-            ],
-            input=data,
-        )
-        if raw:
-            return result
-        return json.loads(result) if result else None
     request = urllib.request.Request(
         f"https://api.github.com/{path}",
         method=method,
@@ -512,7 +493,7 @@ def api(repo, suffix, *, method="GET", payload=None, raw=False):
             raise CommandError(
                 "Python cannot verify GitHub's certificate because it has no certificate store. "
                 "On a python.org install, run Install Certificates.command from its Applications folder, "
-                "set SSL_CERT_FILE to a certificate bundle, or log in with the GitHub CLI (gh auth login), which cloud replay uses instead"
+                "or set SSL_CERT_FILE to a certificate bundle"
             ) from None
         raise CommandError(f"Could not reach api.github.com: {reason}") from None
     if raw:

package/dist/{chunk-ANMDH7EL.js → chunk-ENORQO5K.js}

@@ -17,7 +17,7 @@ import {
   serializeValue,
   toJsonSafe,
   toJsonSafeReport
-} from "./chunk-IYZRISE3.js";
+} from "./chunk-GLWAET3G.js";
 import {
   BitfabError,
   CONTENT_OFF_KEY,
@@ -36,7 +36,7 @@ import {
   readEnv,
   recordCallerTraceMetadata,
   retireTraceMetadata
-} from "./chunk-ZKP65VBL.js";
+} from "./chunk-XHUW3NY6.js";
 import {
   __privateAdd,
   __privateGet,
@@ -3906,6 +3906,22 @@ var BitfabVercelAiHandler = class {
 };
 
 // src/client.ts
+function markTracedFunction(target, traceFunctionKey, original) {
+  Object.defineProperty(target, "_bitfabTraceFunctionKey", {
+    value: traceFunctionKey
+  });
+  Object.defineProperty(target, "_bitfabWrappedFn", { value: original });
+  Object.defineProperty(target, "bind", {
+    value(thisArg, ...boundArgs) {
+      return markTracedFunction(
+        Function.prototype.bind.call(this, thisArg, ...boundArgs),
+        traceFunctionKey,
+        original
+      );
+    }
+  });
+  return target;
+}
 function resolveSpanExperimentId(options) {
   const { experimentId, testRunId } = options;
   if (experimentId !== void 0 && testRunId !== void 0 && experimentId !== testRunId) {
@@ -4756,11 +4772,7 @@ var Bitfab = class {
         }
       );
     };
-    Object.defineProperty(autoTraceRoot, "_bitfabTraceFunctionKey", {
-      value: traceFunctionKey
-    });
-    Object.defineProperty(autoTraceRoot, "_bitfabWrappedFn", { value: fn });
-    return autoTraceRoot;
+    return markTracedFunction(autoTraceRoot, traceFunctionKey, fn);
   }
   applySimulationPlan(payload, submit, drop) {
     this.simulationPlan.send(
@@ -5775,11 +5787,7 @@ var Bitfab = class {
       }
       return runWithSpanStack(newStack, executeWithContext);
     };
-    Object.defineProperty(wrappedFn, "_bitfabTraceFunctionKey", {
-      value: traceFunctionKey
-    });
-    Object.defineProperty(wrappedFn, "_bitfabWrappedFn", { value: fn });
-    return wrappedFn;
+    return markTracedFunction(wrappedFn, traceFunctionKey, fn);
   }
   /**
    * Create a standard ECMAScript method decorator that records each invocation
@@ -6245,7 +6253,7 @@ var Bitfab = class {
       await runWithSeedContext({ traceId }, async () => target(...args));
     } finally {
       try {
-        const { flushTraces: flushTraces2 } = await import("./http-LGRICPE6.js");
+        const { flushTraces: flushTraces2 } = await import("./http-MFXRJ6RS.js");
         await flushTraces2(3e4);
       } finally {
         unrecorded = activeTraceStates.delete(traceId);
@@ -6304,7 +6312,7 @@ var Bitfab = class {
         `Function is wrapped with trace function key '${wrappedKey}' but replay was called with '${traceFunctionKey}'. Pass matching keys, or pass the unwrapped function to replay it under the explicit key.`
       );
     }
-    const { replay: doReplay } = await import("./replay-4XNNKZ6T.js");
+    const { replay: doReplay } = await import("./replay-QZQYRTAO.js");
     return doReplay(
       this.httpClient,
       this.serviceUrl,
@@ -6574,8 +6582,34 @@ var finalizers = {
 
 // src/replayRegistry.ts
 function defineReplayRegistry(registry) {
+  validateReplayRegistry(registry);
   return registry;
 }
+var REGISTRATION_FIELDS = /* @__PURE__ */ new Set([
+  "client",
+  "fn",
+  "traceFunctionKey",
+  "options",
+  "optionsFactory"
+]);
+function validateReplayRegistry(registry) {
+  for (const [pipeline, registration] of Object.entries(registry)) {
+    if (typeof registration !== "object" || registration === null) {
+      throw new BitfabError(
+        `Replay registry entry '${pipeline}' must be an object with client and fn.`
+      );
+    }
+    const unknownFields = Object.keys(registration).filter(
+      (field) => !REGISTRATION_FIELDS.has(field)
+    );
+    if (unknownFields.length > 0) {
+      warnOnce(
+        `replay-registry-unknown-fields:${pipeline}:${unknownFields.join(",")}`,
+        `Replay registry entry '${pipeline}' has unknown field(s): ${unknownFields.join(", ")}. They are ignored. An entry takes client, fn, traceFunctionKey, options, and optionsFactory. Put replay options such as adaptInputs, mock, and mockOverride inside options. A later SDK release will reject unknown fields.`
+      );
+    }
+  }
+}
 async function reseedFromRegistry(registry, pipeline, traceIds) {
   const registration = registry[pipeline];
   if (registration === void 0) {
@@ -6624,7 +6658,7 @@ async function seedFromRegistry(registry, pipeline, cases, options = {}) {
       sessionId: seedCase.sessionId
     })
   );
-  const { flushTraces: flushTraces2 } = await import("./http-LGRICPE6.js");
+  const { flushTraces: flushTraces2 } = await import("./http-MFXRJ6RS.js");
   await flushTraces2(3e4);
   return { pipeline, traceFunctionKey, traceIds };
 }
@@ -6669,4 +6703,4 @@ export {
   reseedFromRegistry,
   seedFromRegistry
 };
-//# sourceMappingURL=chunk-ANMDH7EL.js.map
+//# sourceMappingURL=chunk-ENORQO5K.js.map