bitfab 0.68.0 → 0.68.1

package/dist/{src-Dze8o7iU.cjs → src-BYXxPnpE.cjs}

@@ -1,6 +1,6 @@
 const require_bamlPeer = require("./bamlPeer-BUoZgNxZ.cjs");
-const require_http = require("./http-psFF1ptn.cjs");
-const require_autoTrace = require("./autoTrace-BikML0lX.cjs");
+const require_http = require("./http-r1r6injY.cjs");
+const require_autoTrace = require("./autoTrace-D3-OZcmR.cjs");
 let superjson = require("superjson");
 superjson = require_bamlPeer.__toESM(superjson, 1);
 //#region src/categories.ts
@@ -4798,7 +4798,7 @@ function extractContextFromCollector(collector) {
 }
 const BOUND_FUNCTION_NAME_PREFIX = /^bound /;
 function discoveredSpanType(definition, nodeConfiguration) {
-	const recordsModelCall = definition.baml === true || definition.jev === true;
+	const recordsModelCall = definition.baml === true || definition.jev === true || definition.openrouter === true;
 	const ownType = recordsModelCall ? "llm" : "function";
 	if (nodeConfiguration === void 0) return ownType;
 	if (recordsModelCall && nodeConfiguration.defaultType === true) return ownType;
@@ -5199,7 +5199,7 @@ var Bitfab = class {
 						...nodeConfiguration?.finalize !== void 0 && { finalize: nodeConfiguration.finalize }
 					};
 					const isAsync = definition.async === true;
-					return openChild(entry, childOptions, inputs, isAsync, definition.baml === true ? () => invokeAutoTracedBaml(invokeFn, isAsync) : definition.jev === true ? () => recordAutoTracedJev(invokeFn, getCurrentSpan()) : invokeFn);
+					return openChild(entry, childOptions, inputs, isAsync, definition.baml === true ? () => invokeAutoTracedBaml(invokeFn, isAsync) : definition.jev === true ? () => recordAutoTracedJev(invokeFn, getCurrentSpan()) : definition.openrouter === true ? () => require_autoTrace.recordAutoTracedOpenRouter(invokeFn, inputs[0], getCurrentSpan()) : invokeFn);
 				},
 				invokeNestedRoot(nestedName, nestedFunctionName, inputs, invokeFn, entry, link, spanDetails) {
 					if (excluded.has(nestedName) || excluded.has(nestedFunctionName) || !recordsSpan(entry, nestedName, true)) return invokeFn();
@@ -5503,6 +5503,26 @@ var Bitfab = class {
 		});
 		return wrappedFn;
 	}
+	wrapOpenRouter(method, options) {
+		const methodName = method.name.replace(BOUND_FUNCTION_NAME_PREFIX, "");
+		if (!methodName) throw new require_http.BitfabError("wrapOpenRouter requires a named function (e.g., openai.chat.completions.create.bind(openai.chat.completions)).");
+		const wrappedFn = async (...args) => {
+			const span = getCurrentSpan();
+			const startedAt = Date.now();
+			const { watch, result } = require_autoTrace.watchOpenRouterUsage(() => method(...args));
+			const response = await result;
+			require_autoTrace.recordWatchedOpenRouterCompletion(span, watch, args[0], response, Date.now() - startedAt);
+			try {
+				options?.onResponse?.(response);
+			} catch {}
+			return response;
+		};
+		Object.defineProperty(wrappedFn, "name", {
+			configurable: true,
+			value: methodName
+		});
+		return wrappedFn;
+	}
 	withSpan(traceFunctionKey, optionsOrFn, maybeFn) {
 		this.simulationPlan.refresh();
 		const options = typeof optionsOrFn === "function" ? {} : optionsOrFn;
@@ -6139,7 +6159,7 @@ var Bitfab = class {
 			await runWithSeedContext({ traceId }, async () => target(...args));
 		} finally {
 			try {
-				const { flushTraces } = await Promise.resolve().then(() => require("./http-psFF1ptn.cjs")).then((n) => n.http_exports);
+				const { flushTraces } = await Promise.resolve().then(() => require("./http-r1r6injY.cjs")).then((n) => n.http_exports);
 				await flushTraces(3e4);
 			} finally {
 				unrecorded = activeTraceStates.delete(traceId);
@@ -6228,6 +6248,9 @@ var BitfabFunction = class {
 	wrapJev(method, options) {
 		return this.client.wrapJev(method, options);
 	}
+	wrapOpenRouter(method, options) {
+		return this.client.wrapOpenRouter(method, options);
+	}
 };
 //#endregion
 //#region src/finalizers.ts
@@ -6330,7 +6353,7 @@ async function seedFromRegistry(registry, pipeline, cases, options = {}) {
 		metadata: seedCase.metadata,
 		sessionId: seedCase.sessionId
 	}));
-	const { flushTraces } = await Promise.resolve().then(() => require("./http-psFF1ptn.cjs")).then((n) => n.http_exports);
+	const { flushTraces } = await Promise.resolve().then(() => require("./http-r1r6injY.cjs")).then((n) => n.http_exports);
 	await flushTraces(3e4);
 	return {
 		pipeline,
@@ -6574,4 +6597,4 @@ Object.defineProperty(exports, "serializeReplayResult", {
 	}
 });
 
-//# sourceMappingURL=src-Dze8o7iU.cjs.map
+//# sourceMappingURL=src-BYXxPnpE.cjs.map