@bitfab/sdk 0.76.5 → 0.76.6

package/dist/node.js

@@ -1,6 +1,6 @@
-import { A as VerdictsClient, C as OrganizationMembersClient, D as BitfabLangGraphCallbackHandler, E as BitfabLangGraphIntegration, F as GraderRerunTimeoutError, I as BitfabClaudeAgentHandler, L as DEFAULT_SERVICE_URL, M as SUPPORTED_PROVIDERS, N as DatasetsClient, O as GradersClient, P as GraderRerunError, R as AssertionCategoriesClient, S as BitfabPiIntegration, T as NO_MOCK_OVERRIDE, _ as BITFAB_PROGRESS_PREFIX, a as Bitfab, b as reportReplayProgress, c as getCurrentSpan, d as BitfabOpenAITracingProcessor, f as AssertionGenerationError, g as BitfabRivetIntegration, h as SpansClient, i as finalizers, j as ExperimentsClient, k as LabelsClient, l as getCurrentTrace, m as TracesClient, n as reseedFromRegistry, o as BitfabFunction, p as AssertionGenerationTimeoutError, r as seedFromRegistry, s as getCurrentReplayBranch, t as defineReplayRegistry, u as BitfabVercelAiHandler, v as DbBranchReplayError, w as BitfabOpenAIAgentHandler, x as serializeReplayResult, y as ReplayError, z as CategoriesClient } from "./src-B_e7xNzm.js";
-import { C as BitfabError, O as __version__, i as flushTraces, t as HttpClient, w as MixedTracingError } from "./http-BLyaXOOY.js";
-import { Z as registerAsyncLocalStorageClass, q as assertAsyncStorageRegistered } from "./autoTrace-BgN0zCQf.js";
+import { A as VerdictsClient, C as OrganizationMembersClient, D as BitfabLangGraphCallbackHandler, E as BitfabLangGraphIntegration, F as GraderRerunTimeoutError, I as BitfabClaudeAgentHandler, L as DEFAULT_SERVICE_URL, M as SUPPORTED_PROVIDERS, N as DatasetsClient, O as GradersClient, P as GraderRerunError, R as AssertionCategoriesClient, S as BitfabPiIntegration, T as NO_MOCK_OVERRIDE, _ as BITFAB_PROGRESS_PREFIX, a as Bitfab, b as reportReplayProgress, c as getCurrentSpan, d as BitfabOpenAITracingProcessor, f as AssertionGenerationError, g as BitfabRivetIntegration, h as SpansClient, i as finalizers, j as ExperimentsClient, k as LabelsClient, l as getCurrentTrace, m as TracesClient, n as reseedFromRegistry, o as BitfabFunction, p as AssertionGenerationTimeoutError, r as seedFromRegistry, s as getCurrentReplayBranch, t as defineReplayRegistry, u as BitfabVercelAiHandler, v as DbBranchReplayError, w as BitfabOpenAIAgentHandler, x as serializeReplayResult, y as ReplayError, z as CategoriesClient } from "./src-DIJP4CNn.js";
+import { C as BitfabError, O as __version__, i as flushTraces, t as HttpClient, w as MixedTracingError } from "./http-DZlsxUgc.js";
+import { J as assertAsyncStorageRegistered, Q as registerAsyncLocalStorageClass } from "./autoTrace-g3y0sLPq.js";
 import { AsyncLocalStorage } from "async_hooks";
 //#region src/asyncStorageNode.ts
 registerAsyncLocalStorageClass(AsyncLocalStorage);

package/dist/{registryCommand-M1u4H9PI.js → registryCommand-UEUDL6JV.js}

@@ -641,6 +641,21 @@ function geminiUsage(body, response) {
 		reasoningTokens: thoughts
 	};
 }
+function isGeminiEmbeddingResponse(response) {
+	return Array.isArray(field(response, "embeddings"));
+}
+function geminiEmbeddingPrompt(body) {
+	return renderedPrompt(geminiEntries(field(body, "contents")));
+}
+function geminiEmbeddingUsage(body, response) {
+	const embeddings = field(response, "embeddings");
+	const inputTokens = sumOfPresent((Array.isArray(embeddings) ? embeddings : []).map((embedding) => numberField(field(embedding, "statistics"), "tokenCount")));
+	return {
+		model: geminiModel(body, response),
+		inputTokens,
+		outputTokens: inputTokens === void 0 ? void 0 : 0
+	};
+}
 function geminiChunkText(chunk) {
 	const candidates = field(chunk, "candidates");
 	return Array.isArray(candidates) ? geminiContentText(field(candidates[0], "content")) ?? "" : "";
@@ -754,6 +769,19 @@ const CLIENT_CALL_PROVIDERS = [
 		prompt: geminiPrompt,
 		usage: geminiUsage,
 		streamSummary: geminiStreamSummary
+	},
+	{
+		key: "gemini-embeddings",
+		provider: "gemini",
+		library: "@google/genai",
+		endpoint: "models",
+		methods: ["embedContent"],
+		streamMethods: [],
+		streamEvents: [],
+		matchesReceiver: isGeminiModels,
+		isResponse: isGeminiEmbeddingResponse,
+		prompt: geminiEmbeddingPrompt,
+		usage: geminiEmbeddingUsage
 	}
 ];
 new Map(CLIENT_CALL_PROVIDERS.map((provider) => [provider.key, provider]));
@@ -1650,4 +1678,4 @@ function runIfEntrypoint(moduleUrl, main) {
 //#endregion
 export { runRegistryMain as a, runSeedCli as c, runIfEntrypoint as i, parseRegistryCommandArgs as n, checkReplayCliArgs as o, resolveRegistry as r, runReplayCli as s, importRegistry as t };
 
-//# sourceMappingURL=registryCommand-M1u4H9PI.js.map
+//# sourceMappingURL=registryCommand-UEUDL6JV.js.map