@bitfab/sdk 0.76.5 → 0.76.7

package/dist/{src-B_e7xNzm.js → src-Cm9Jdgfh.js}

@@ -1,6 +1,6 @@
 import { t as __exportAll } from "./rolldown-runtime-D7D4PA-g.js";
-import { C as BitfabError, D as readEnv$1, E as serializeError, O as __version__, S as gitRunner, T as describeError, _ 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-BLyaXOOY.js";
-import { A as recordWatchedClientCall, B as installLangChainRunScopes, C as OPENAI_RESPONSES, D as clientCallStreams, E as clientCallProvider, F as mapOpenAiAgentsTrace, G as isCaptureSuppressed, I as openAiAgentsParentSpanId, J as asyncStorageReady, K as runWithCaptureSuppressed, L as openAiAgentsSpanChain, M as watchClientCall, N as StreamFinalizationError, O as frameworkRecordsClientCall, P as endOpenAiAgentsSpan, R as unmapOpenAiAgentsTrace, S as OPENAI_EMBEDDINGS, T as OPENROUTER, U as warnOnce, V as markLangChainRunRecorded, W as writeOnce, X as isAsyncStorageInitDone, Y as createAsyncLocalStorage, _ as runWithAutoTraceNodeConfiguration, b as GEMINI, d as currentAutoTraceEntries, g as runWithAutoTraceFramesAt, h as runWithAutoTraceEntry, j as sharedClientCallStreamRecording, k as recordAutoTracedClientCall, m as runAsAutoTraceRootFunction, p as invokeThroughAutoTraceEntries, s as __bitfabAutoTraceActive, u as clearAutoTraceNodeConfiguration, v as runWithAutoTraceRootContext, w as OPENAI_TRANSCRIPTIONS, x as OPENAI_CHAT, y as ANTHROPIC, z as forgetLangChainRun } from "./autoTrace-BgN0zCQf.js";
+import { C as BitfabError, D as readEnv$1, E as serializeError, O as __version__, S as gitRunner, T as describeError, _ 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-DoS4PUm6.js";
+import { $ as asyncStorageReady, A as frameworkRecordsClientCall, B as mapOpenAiAgentsTrace, C as OPENAI_EMBEDDINGS, D as clientCallProvider, E as OPENROUTER, F as sharedClientCallStreamRecording, G as installLangChainRunScopes, H as openAiAgentsSpanChain, I as watchClientCall, J as warnOnce, K as markLangChainRunRecorded, L as StreamFinalizationError, M as isOpenRouterBaseUrl, N as recordAutoTracedClientCall, O as clientCallProviderForUsage, P as recordWatchedClientCall, R as MODEL_CALL_SPAN_TYPES, S as OPENAI_CHAT, T as OPENAI_TRANSCRIPTIONS, U as unmapOpenAiAgentsTrace, V as openAiAgentsParentSpanId, W as forgetLangChainRun, X as isCaptureSuppressed, Y as writeOnce, Z as runWithCaptureSuppressed, _ as runWithAutoTraceNodeConfiguration, b as GEMINI, d as currentAutoTraceEntries, et as createAsyncLocalStorage, g as runWithAutoTraceFramesAt, h as runWithAutoTraceEntry, j as isAnthropicBaseUrl, k as clientCallStreams, m as runAsAutoTraceRootFunction, p as invokeThroughAutoTraceEntries, s as __bitfabAutoTraceActive, tt as isAsyncStorageInitDone, u as clearAutoTraceNodeConfiguration, v as runWithAutoTraceRootContext, w as OPENAI_RESPONSES, x as GEMINI_EMBEDDINGS, y as ANTHROPIC, z as endOpenAiAgentsSpan } from "./autoTrace-RWUOk1WF.js";
 import { t as importOptionalPeer } from "./optionalPeer-CczrzrXx.js";
 import { n as loadBamlModule, t as isSharedBamlModule } from "./bamlPeer-CWPgBSlJ.js";
 import superjson from "superjson";
@@ -1545,11 +1545,11 @@ var GradersClient = class {
 const JEV_PROVIDER = "typesafe";
 const JEV_WATCH_STORAGE_SYMBOL = Symbol.for("bitfab.jevWatchStorage");
 const repliesRecordedByAutoTrace = /* @__PURE__ */ new WeakSet();
-function isRecord$5(value) {
+function isRecord$6(value) {
 	return typeof value === "object" && value !== null;
 }
 function isJevReply(value) {
-	return isRecord$5(value) && typeof value.model === "string" && isRecord$5(value.usage);
+	return isRecord$6(value) && typeof value.model === "string" && isRecord$6(value.usage);
 }
 function tokenCount(usage, key) {
 	const count = usage[key];
@@ -1608,7 +1608,7 @@ function recordWatchedJevReply(span, watch, reply, durationMs) {
 	markEnclosingWatches(watch);
 }
 function isRecordedByAutoTrace(reply) {
-	return isRecord$5(reply) && repliesRecordedByAutoTrace.has(reply);
+	return isRecord$6(reply) && repliesRecordedByAutoTrace.has(reply);
 }
 function recordAutoTracedJev(invokeFn, span) {
 	const startedAt = Date.now();
@@ -2121,7 +2121,7 @@ var BitfabLangGraphCallbackHandler = class {
 //#region src/integrationFixture.ts
 const INTEGRATION_FIXTURE_CONTEXT_KEY = "__bitfabIntegrationFixture";
 const INTEGRATION_FIXTURE_VERSION = 1;
-function isRecord$4(value) {
+function isRecord$5(value) {
 	return typeof value === "object" && value !== null && !Array.isArray(value);
 }
 function createIntegrationFixtureContext(name, value) {
@@ -2135,8 +2135,8 @@ function createIntegrationFixtureContext(name, value) {
 }
 function hasIntegrationFixture(contexts, name) {
 	return contexts.some((context) => {
-		const fixture = isRecord$4(context) ? context[INTEGRATION_FIXTURE_CONTEXT_KEY] : void 0;
-		return isRecord$4(fixture) && fixture.version === INTEGRATION_FIXTURE_VERSION && fixture.name === name;
+		const fixture = isRecord$5(context) ? context[INTEGRATION_FIXTURE_CONTEXT_KEY] : void 0;
+		return isRecord$5(fixture) && fixture.version === INTEGRATION_FIXTURE_VERSION && fixture.name === name;
 	});
 }
 function extractIntegrationFixtures(spanData) {
@@ -2144,9 +2144,9 @@ function extractIntegrationFixtures(spanData) {
 	if (!Array.isArray(contexts)) return;
 	const fixtures = /* @__PURE__ */ new Map();
 	for (const context of contexts) {
-		if (!isRecord$4(context)) continue;
+		if (!isRecord$5(context)) continue;
 		const fixture = context[INTEGRATION_FIXTURE_CONTEXT_KEY];
-		if (!isRecord$4(fixture) || fixture.version !== INTEGRATION_FIXTURE_VERSION || typeof fixture.name !== "string" || fixtures.has(fixture.name)) continue;
+		if (!isRecord$5(fixture) || fixture.version !== INTEGRATION_FIXTURE_VERSION || typeof fixture.name !== "string" || fixtures.has(fixture.name)) continue;
 		fixtures.set(fixture.name, deserializeValue({
 			json: fixture.value,
 			...fixture.valueMeta !== void 0 && { meta: fixture.valueMeta }
@@ -2165,7 +2165,7 @@ const THREAD_CONFIG_KEYS = [
 	"checkpoint_ns",
 	"checkpoint_id"
 ];
-function isRecord$3(value) {
+function isRecord$4(value) {
 	return typeof value === "object" && value !== null;
 }
 function toBase64(bytes) {
@@ -2192,12 +2192,12 @@ async function newMemorySaver() {
 	return new MemorySaver();
 }
 async function unstoredDeltaChannels(graph, tuple) {
-	const channels = isRecord$3(graph) ? graph.channels : void 0;
-	if (!isRecord$3(channels)) return [];
+	const channels = isRecord$4(graph) ? graph.channels : void 0;
+	if (!isRecord$4(channels)) return [];
 	const { DeltaChannel } = await loadLangGraphPeer();
 	if (DeltaChannel === void 0) return [];
-	const versions = isRecord$3(tuple.checkpoint.channel_versions) ? tuple.checkpoint.channel_versions : {};
-	const values = isRecord$3(tuple.checkpoint.channel_values) ? tuple.checkpoint.channel_values : {};
+	const versions = isRecord$4(tuple.checkpoint.channel_versions) ? tuple.checkpoint.channel_versions : {};
+	const values = isRecord$4(tuple.checkpoint.channel_values) ? tuple.checkpoint.channel_values : {};
 	return Object.entries(channels).filter(([name, spec]) => spec instanceof DeltaChannel && name in versions && !(name in values)).map(([name]) => name);
 }
 async function selfContained(graph, checkpointer, tuple) {
@@ -2210,27 +2210,27 @@ async function selfContained(graph, checkpointer, tuple) {
 		checkpoint: {
 			...tuple.checkpoint,
 			channel_values: {
-				...isRecord$3(tuple.checkpoint.channel_values) ? tuple.checkpoint.channel_values : {},
+				...isRecord$4(tuple.checkpoint.channel_values) ? tuple.checkpoint.channel_values : {},
 				...Object.fromEntries(names.filter((name) => name in values).map((name) => [name, values[name]]))
 			}
 		}
 	};
 }
 function savingCheckpointer(graph) {
-	const own = isRecord$3(graph) ? graph.checkpointer : void 0;
-	const inner = isRecord$3(graph) && isRecord$3(graph.graph) ? graph.graph.checkpointer : void 0;
+	const own = isRecord$4(graph) ? graph.checkpointer : void 0;
+	const inner = isRecord$4(graph) && isRecord$4(graph.graph) ? graph.graph.checkpointer : void 0;
 	const checkpointer = own ?? inner;
-	return isRecord$3(checkpointer) && typeof checkpointer.getTuple === "function" ? checkpointer : null;
+	return isRecord$4(checkpointer) && typeof checkpointer.getTuple === "function" ? checkpointer : null;
 }
 function threadConfig(config) {
-	const configurable = isRecord$3(config?.configurable) ? config.configurable : {};
+	const configurable = isRecord$4(config?.configurable) ? config.configurable : {};
 	return Object.fromEntries(THREAD_CONFIG_KEYS.filter((key) => key in configurable).map((key) => [key, configurable[key]]));
 }
 function withConfigurable(config, values) {
 	return {
 		...config ?? {},
 		configurable: {
-			...isRecord$3(config?.configurable) ? config.configurable : {},
+			...isRecord$4(config?.configurable) ? config.configurable : {},
 			...values
 		}
 	};
@@ -2367,7 +2367,7 @@ var GraphRun = class {
 	}
 	async restoreThread(replay) {
 		const { saver, thread } = await restoredCheckpointer(replay.integrationFixtures?.get(this.fixtureName) ?? null);
-		const configurable = isRecord$3(this.config?.configurable) ? this.config.configurable : {};
+		const configurable = isRecord$4(this.config?.configurable) ? this.config.configurable : {};
 		return {
 			saver,
 			thread: {
@@ -2381,19 +2381,19 @@ var GraphRun = class {
 //#region src/langgraphIntegration.ts
 const TOOL_RESULT_TAG = "__bitfabLangGraphToolResult";
 const AGENT_MIDDLEWARE_BRAND = Symbol.for("AgentMiddleware");
-function isRecord$2(value) {
+function isRecord$3(value) {
 	return typeof value === "object" && value !== null;
 }
 function isToolMessage(value) {
-	return isRecord$2(value) && value.lc_direct_tool_output === true && value.type === "tool" && (typeof value.content === "string" || Array.isArray(value.content));
+	return isRecord$3(value) && value.lc_direct_tool_output === true && value.type === "tool" && (typeof value.content === "string" || Array.isArray(value.content));
 }
 function isCommand(value) {
-	return isRecord$2(value) && value.lg_name === "Command";
+	return isRecord$3(value) && value.lg_name === "Command";
 }
 function encodeNested(value) {
 	if (isToolMessage(value) || isCommand(value)) return encodeNativeToolResult(value);
 	if (Array.isArray(value)) return value.map(encodeNested);
-	if (isRecord$2(value)) return Object.fromEntries(Object.entries(value).map(([key, entry]) => [key, encodeNested(entry)]));
+	if (isRecord$3(value)) return Object.fromEntries(Object.entries(value).map(([key, entry]) => [key, encodeNested(entry)]));
 	return value;
 }
 function encodeNativeToolResult(value) {
@@ -2410,17 +2410,17 @@ function encodeNativeToolResult(value) {
 	};
 	return {
 		[TOOL_RESULT_TAG]: "command",
-		graph: isRecord$2(value) && typeof value.graph === "string" ? value.graph : void 0,
-		update: isRecord$2(value) ? encodeNested(value.update) : void 0,
-		resume: isRecord$2(value) ? encodeNested(value.resume) : void 0,
-		goto: isRecord$2(value) ? encodeNested(value.goto) : void 0
+		graph: isRecord$3(value) && typeof value.graph === "string" ? value.graph : void 0,
+		update: isRecord$3(value) ? encodeNested(value.update) : void 0,
+		resume: isRecord$3(value) ? encodeNested(value.resume) : void 0,
+		goto: isRecord$3(value) ? encodeNested(value.goto) : void 0
 	};
 }
 function finalizeToolResult(value) {
 	return isToolMessage(value) || isCommand(value) ? encodeNativeToolResult(value) : value;
 }
 function isEncodedToolResult(value) {
-	return isRecord$2(value) && typeof value[TOOL_RESULT_TAG] === "string";
+	return isRecord$3(value) && typeof value[TOOL_RESULT_TAG] === "string";
 }
 async function loadPeer(specifierParts, packageName, docsPath) {
 	try {
@@ -2450,7 +2450,7 @@ async function loadLangGraph() {
 async function reviveNested(value, toolCallId) {
 	if (isEncodedToolResult(value)) return reviveToolResult(value, toolCallId);
 	if (Array.isArray(value)) return Promise.all(value.map((entry) => reviveNested(entry, toolCallId)));
-	if (isRecord$2(value)) {
+	if (isRecord$3(value)) {
 		const entries = await Promise.all(Object.entries(value).map(async ([key, entry]) => [key, await reviveNested(entry, toolCallId)]));
 		return Object.fromEntries(entries);
 	}
@@ -2466,9 +2466,9 @@ async function reviveToolResult(value, toolCallId) {
 			...typeof value.id === "string" && { id: value.id },
 			...(value.status === "success" || value.status === "error") && { status: value.status },
 			...value.artifact !== void 0 && { artifact: await reviveNested(value.artifact, toolCallId) },
-			...isRecord$2(value.metadata) && { metadata: await reviveNested(value.metadata, toolCallId) },
-			...isRecord$2(value.additionalKwargs) && { additional_kwargs: await reviveNested(value.additionalKwargs, toolCallId) },
-			...isRecord$2(value.responseMetadata) && { response_metadata: await reviveNested(value.responseMetadata, toolCallId) }
+			...isRecord$3(value.metadata) && { metadata: await reviveNested(value.metadata, toolCallId) },
+			...isRecord$3(value.additionalKwargs) && { additional_kwargs: await reviveNested(value.additionalKwargs, toolCallId) },
+			...isRecord$3(value.responseMetadata) && { response_metadata: await reviveNested(value.responseMetadata, toolCallId) }
 		});
 	}
 	const { Command } = await loadLangGraph();
@@ -2545,7 +2545,7 @@ var BitfabLangGraphIntegration = class {
 		return middleware.includes(this.middleware) ? [...middleware] : [...middleware, this.middleware];
 	}
 	withSubagentMiddleware(spec) {
-		if (!isRecord$2(spec) || "runnable" in spec || "graphId" in spec) return spec;
+		if (!isRecord$3(spec) || "runnable" in spec || "graphId" in spec) return spec;
 		return {
 			...spec,
 			middleware: this.withMiddleware(Array.isArray(spec.middleware) ? spec.middleware : [])
@@ -2562,7 +2562,7 @@ var BitfabLangGraphIntegration = class {
 		return await this.runClaimedToolCall(traceId, toolCallId, toolName, shouldMock, request.toolCall.args, async () => await handler(request), stepSpanId);
 	}
 	stepSpanIdFor(config, traceId) {
-		const namespace = isRecord$2(config) && isRecord$2(config.configurable) ? config.configurable.checkpoint_ns : void 0;
+		const namespace = isRecord$3(config) && isRecord$3(config.configurable) ? config.configurable.checkpoint_ns : void 0;
 		return typeof namespace === "string" ? this.callbackHandler.visibleStepSpanId(namespace, traceId) : null;
 	}
 	recordingTraceId(toolName) {
@@ -2609,8 +2609,8 @@ var BitfabLangGraphIntegration = class {
 		const originalInvoke = tool.invoke.bind(tool);
 		return new Proxy(tool, { get: (target, property) => {
 			if (property === "invoke") return async (input, ...rest) => {
-				const toolCallId = isRecord$2(input) && typeof input.id === "string" ? input.id : "";
-				const args = isRecord$2(input) && "args" in input ? input.args : input;
+				const toolCallId = isRecord$3(input) && typeof input.id === "string" ? input.id : "";
+				const args = isRecord$3(input) && "args" in input ? input.args : input;
 				this.assertReplayToolResultExists(toolName, shouldMock);
 				const traceId = this.getActiveSpan()?.traceId ?? null;
 				const run = async () => await originalInvoke(input, ...rest);
@@ -2728,7 +2728,7 @@ function transcriptFixtureName(traceFunctionKey) {
 	return `pi-transcript:${traceFunctionKey}`;
 }
 function isTranscriptFixture(value) {
-	return isRecord$1(value) && (value.method === "prompt" || value.method === "continue") && Array.isArray(value.messages);
+	return isRecord$2(value) && (value.method === "prompt" || value.method === "continue") && Array.isArray(value.messages);
 }
 function conversationMessages(messages) {
 	return messages.filter((message) => message?.role !== "system");
@@ -2788,17 +2788,17 @@ async function rethrowingKept(work) {
 		throw error;
 	}
 }
-function isRecord$1(value) {
+function isRecord$2(value) {
 	return typeof value === "object" && value !== null;
 }
 function isAgentLike(value) {
-	return isRecord$1(value) && typeof value.prompt === "function" && typeof value.continue === "function" && isRecord$1(value.state) && Array.isArray(value.state.messages);
+	return isRecord$2(value) && typeof value.prompt === "function" && typeof value.continue === "function" && isRecord$2(value.state) && Array.isArray(value.state.messages);
 }
 function isSessionLike(value) {
-	return isRecord$1(value) && typeof value.prompt === "function" && isAgentLike(value.agent);
+	return isRecord$2(value) && typeof value.prompt === "function" && isAgentLike(value.agent);
 }
 function isEventStream(value) {
-	return isRecord$1(value) && typeof value.result === "function" && typeof value[Symbol.asyncIterator] === "function";
+	return isRecord$2(value) && typeof value.result === "function" && typeof value[Symbol.asyncIterator] === "function";
 }
 function failed(message) {
 	return typeof message?.stopReason === "string" && FAILED_STOP_REASONS.has(message.stopReason);
@@ -2827,7 +2827,7 @@ function usageContext(message, model) {
 	const modelName = message.model ?? model.id;
 	if (typeof modelName === "string") context.model = modelName;
 	const usage = message.usage;
-	if (!isRecord$1(usage)) return context;
+	if (!isRecord$2(usage)) return context;
 	const entries = {
 		inputTokens: wholePromptTokens(usage),
 		outputTokens: asCount(usage.output),
@@ -2838,7 +2838,7 @@ function usageContext(message, model) {
 	return context;
 }
 function toolDeclarations(tools) {
-	return tools.filter(isRecord$1).map((tool) => ({
+	return tools.filter(isRecord$2).map((tool) => ({
 		name: tool.name,
 		description: tool.description,
 		parameters: tool.parameters
@@ -2919,7 +2919,7 @@ var BitfabPiIntegration = class {
 			});
 			const value = Reflect.get(target, property, proxy);
 			if (STREAM_FN_PROPERTY_NAMES_BY_RELEASE.has(property) && typeof value === "function") return this.wrapStreamFn(value);
-			if (STATE_PROPERTIES.has(property) && isRecord$1(value)) return this.toolWrappingState(value);
+			if (STATE_PROPERTIES.has(property) && isRecord$2(value)) return this.toolWrappingState(value);
 			if (HOOKS_RETURNING_CONTEXT.has(property) && typeof value === "function") return this.toolWrappingHook(value);
 			return value;
 		} });
@@ -2943,7 +2943,7 @@ var BitfabPiIntegration = class {
 		if (existing !== void 0) return existing;
 		const wrapped = async (...args) => {
 			const update = await hook(...args);
-			if (!isRecord$1(update) || !isRecord$1(update.context)) return update;
+			if (!isRecord$2(update) || !isRecord$2(update.context)) return update;
 			const tools = update.context.tools;
 			if (!Array.isArray(tools)) return update;
 			return {
@@ -3101,7 +3101,7 @@ var BitfabPiIntegration = class {
 		return this.originalTools.get(tool) ?? tool;
 	}
 	wrapTool(tool) {
-		if (!isRecord$1(tool) || typeof tool.execute !== "function" || wrappersByAnyIntegration.has(tool)) return tool;
+		if (!isRecord$2(tool) || typeof tool.execute !== "function" || wrappersByAnyIntegration.has(tool)) return tool;
 		const existing = this.wrappedTools.get(tool);
 		if (existing !== void 0) return existing;
 		const execute = tool.execute;
@@ -4785,7 +4785,7 @@ const QUEUE_MESSAGE_EFFECTS = /* @__PURE__ */ new Set([
 	"queue.tryNext",
 	"queue.tryNextBatch"
 ]);
-function isRecord(value) {
+function isRecord$1(value) {
 	return typeof value === "object" && value !== null;
 }
 function readProperty(value, property) {
@@ -4814,7 +4814,7 @@ function snapshotContext(context) {
 	if (typeof aborted === "boolean") snapshot.aborted = aborted;
 	else {
 		const abortSignal = readProperty(source, "abortSignal");
-		if (isRecord(abortSignal)) {
+		if (isRecord$1(abortSignal)) {
 			const signalAborted = readProperty(abortSignal, "aborted");
 			if (typeof signalAborted === "boolean") snapshot.aborted = signalAborted;
 		}
@@ -4855,7 +4855,7 @@ function fixtureName(traceFunctionKey) {
 }
 function snapshotFromReplay(traceFunctionKey, replay) {
 	const fixture = replay.integrationFixtures?.get(fixtureName(traceFunctionKey));
-	if (!isRecord(fixture)) throw new BitfabError(`Trace "${traceFunctionKey}" has no Rivet context fixture. Capture a fresh trace with the Rivet integration before replaying it.`, RIVET_DOCS_URL);
+	if (!isRecord$1(fixture)) throw new BitfabError(`Trace "${traceFunctionKey}" has no Rivet context fixture. Capture a fresh trace with the Rivet integration before replaying it.`, RIVET_DOCS_URL);
 	return fixture;
 }
 var BitfabRivetIntegration = class {
@@ -4866,7 +4866,7 @@ var BitfabRivetIntegration = class {
 		this.markedEffects = new Set(options.mockOnReplay ?? []);
 	}
 	withContext(context, fn) {
-		if (!isRecord(context)) throw new BitfabError("Rivet withContext requires a context object");
+		if (!isRecord$1(context)) throw new BitfabError("Rivet withContext requires a context object");
 		return runWithRivetContext({
 			context,
 			fixtureContext: createIntegrationFixtureContext(fixtureName(this.traceFunctionKey), snapshotContext(context)),
@@ -4932,7 +4932,7 @@ var BitfabRivetIntegration = class {
 			if (!binding) return;
 			const context = binding.context;
 			const target = readProperty(context, group);
-			return isRecord(target) ? target : void 0;
+			return isRecord$1(target) ? target : void 0;
 		};
 		const proxy = this.createMethodProxy(targetProvider, group, RIVET_GROUP_EFFECT_METHODS[group], replay);
 		if (group !== "queue") return proxy;
@@ -4981,10 +4981,10 @@ var BitfabRivetIntegration = class {
 			if (replay) return await callback(this.createMethodProxy(() => void 0, "db.transaction", ["execute"], replay));
 			const context = binding?.context;
 			const db = context ? readProperty(context, "db") : void 0;
-			const transaction = isRecord(db) ? readProperty(db, "transaction") : void 0;
-			if (!isRecord(db) || typeof transaction !== "function") throw unavailableCapability("db.transaction");
+			const transaction = isRecord$1(db) ? readProperty(db, "transaction") : void 0;
+			if (!isRecord$1(db) || typeof transaction !== "function") throw unavailableCapability("db.transaction");
 			return await transaction.call(db, async (transactionDb) => {
-				if (!isRecord(transactionDb)) throw unavailableCapability("db.transaction.execute");
+				if (!isRecord$1(transactionDb)) throw unavailableCapability("db.transaction.execute");
 				return await callback(this.createMethodProxy(() => transactionDb, "db.transaction", ["execute"], replay));
 			}, options);
 		};
@@ -4995,7 +4995,7 @@ var BitfabRivetIntegration = class {
 		return this.decorateQueueMessage(value, replay);
 	}
 	decorateQueueMessage(value, replay) {
-		if (!isRecord(value)) return value;
+		if (!isRecord$1(value)) return value;
 		const originalComplete = readProperty(value, "complete");
 		const complete = this.createEffect("queue.message.complete", () => typeof originalComplete === "function" ? { complete: originalComplete.bind(value) } : void 0, "complete", replay);
 		return {
@@ -5005,7 +5005,7 @@ var BitfabRivetIntegration = class {
 	}
 	queueCallIsCompletable(effect, args) {
 		const options = effect === "queue.waitForNames" || (effect === "queue.next" || effect === "queue.nextBatch") && typeof args[0] === "string" ? args[1] : args[0];
-		return isRecord(options) && options.completable === true;
+		return isRecord$1(options) && options.completable === true;
 	}
 	createQueueIterator(queue, options) {
 		return { async *[Symbol.asyncIterator]() {
@@ -5197,6 +5197,56 @@ function withoutDataRecords(edit) {
 	};
 }
 //#endregion
+//#region src/openaiAgentsUsage.ts
+const TOKEN_CONTEXT_KEYS = [
+	"inputTokens",
+	"outputTokens",
+	"cachedInputTokens",
+	"cacheReadTokens",
+	"costUsd"
+];
+function isRecord(value) {
+	return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+function modelResponses(spanData) {
+	const firstOutput = Array.isArray(spanData.output) ? spanData.output[0] : void 0;
+	return [
+		spanData,
+		spanData.response,
+		firstOutput
+	].filter(isRecord);
+}
+function modelCallModel(spanData) {
+	return modelResponses(spanData).map((candidate) => candidate.model).find((model) => typeof model === "string" && model !== "");
+}
+function modelCallProvider(spanData, call) {
+	const baseUrl = isRecord(spanData.model_config) ? spanData.model_config.base_url : void 0;
+	if (isOpenRouterBaseUrl(baseUrl)) return "openrouter";
+	return isAnthropicBaseUrl(baseUrl) ? "anthropic" : call.provider;
+}
+function hasTokenUsage(context) {
+	return isRecord(context) && TOKEN_CONTEXT_KEYS.some((key) => typeof context[key] === "number" && Number.isFinite(context[key]));
+}
+function openAiAgentsUsageContext(spanData) {
+	if (!isRecord(spanData) || !MODEL_CALL_SPAN_TYPES.has(String(spanData.type))) return null;
+	const response = modelResponses(spanData).find((candidate) => isRecord(candidate.usage));
+	if (response === void 0 || !isRecord(response.usage)) return null;
+	const call = clientCallProviderForUsage(response.usage);
+	const usage = Object.fromEntries(Object.entries(call.usage({ model: modelCallModel(spanData) }, response)).filter(([, value]) => value !== void 0));
+	if (usage.inputTokens === void 0 && usage.outputTokens === void 0) return null;
+	return {
+		provider: modelCallProvider(spanData, call),
+		...usage
+	};
+}
+function addOpenAiAgentsUsageContext(spanData) {
+	const existing = Array.isArray(spanData.contexts) ? spanData.contexts : [];
+	if (existing.some(hasTokenUsage)) return;
+	const context = openAiAgentsUsageContext(spanData);
+	if (context === null) return;
+	spanData.contexts = [...existing, context];
+}
+//#endregion
 //#region src/tracing.ts
 var BitfabOpenAITracingProcessor = class {
 	getCanonicalTraceId(sourceTraceId) {
@@ -5376,6 +5426,7 @@ var BitfabOpenAITracingProcessor = class {
 		const [serializedSpan, exportErrors] = this.exportSpan(span);
 		errors.push(...exportErrors);
 		this.extractSpanInputResponse(span, serializedSpan, errors);
+		addOpenAiAgentsUsageContext(serializedSpan.span_data);
 		this.applySpanOverrides(serializedSpan, span.traceId ?? "");
 		const payload = this.buildSpanPayload(serializedSpan, errors);
 		const canonicalTraceId = span.traceId ? this.getCanonicalTraceId(span.traceId) : void 0;
@@ -6573,6 +6624,9 @@ var Bitfab = class {
 	wrapGemini(method, options) {
 		return this.wrapClientCall(GEMINI, method, options, unboundName(method) ?? "generateContent");
 	}
+	wrapGeminiEmbeddings(method, options) {
+		return this.wrapClientCall(GEMINI_EMBEDDINGS, method, options, unboundName(method) ?? "embedContent");
+	}
 	wrapClientCall(provider, method, options, name) {
 		const wrappedFn = async (...args) => {
 			const span = getCurrentSpan();
@@ -7237,7 +7291,7 @@ var Bitfab = class {
 			await runWithSeedContext({ traceId }, async () => target(...args));
 		} finally {
 			try {
-				const { flushTraces } = await import("./http-BLyaXOOY.js").then((n) => n.a);
+				const { flushTraces } = await import("./http-DoS4PUm6.js").then((n) => n.a);
 				await flushTraces(3e4);
 			} finally {
 				unrecorded = activeTraceStates.delete(traceId);
@@ -7353,6 +7407,9 @@ var BitfabFunction = class {
 	wrapGemini(method, options) {
 		return this.client.wrapGemini(method, options);
 	}
+	wrapGeminiEmbeddings(method, options) {
+		return this.client.wrapGeminiEmbeddings(method, options);
+	}
 };
 //#endregion
 //#region src/finalizers.ts
@@ -7455,7 +7512,7 @@ async function seedFromRegistry(registry, pipeline, cases, options = {}) {
 		metadata: seedCase.metadata,
 		sessionId: seedCase.sessionId
 	}));
-	const { flushTraces } = await import("./http-BLyaXOOY.js").then((n) => n.a);
+	const { flushTraces } = await import("./http-DoS4PUm6.js").then((n) => n.a);
 	await flushTraces(3e4);
 	return {
 		pipeline,
@@ -7472,4 +7529,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-B_e7xNzm.js.map
+//# sourceMappingURL=src-Cm9Jdgfh.js.map