bitfab 0.20.0 → 0.21.1

package/dist/{chunk-IUZIGC6T.js → chunk-75LZO6JS.js}

@@ -10,7 +10,7 @@ import {
 } from "./chunk-EQI6ZJC3.js";
 
 // src/version.generated.ts
-var __version__ = "0.20.0";
+var __version__ = "0.21.1";
 
 // src/constants.ts
 var DEFAULT_SERVICE_URL = "https://bitfab.ai";
@@ -1929,9 +1929,15 @@ var BitfabOpenAITracingProcessor = class {
    * Extract and add input/response to serialized span, updating errors list.
    */
   extractSpanInputResponse(span, serializedSpan, errors) {
+    if (span.spanData?.type !== "response") {
+      return;
+    }
     const spanData = serializedSpan.span_data;
     try {
-      spanData.input = span.spanData?._input || [];
+      const input = span.spanData?._input;
+      if (input !== void 0) {
+        spanData.input = input;
+      }
     } catch (error) {
       errors.push({
         source: "sdk",
@@ -1940,7 +1946,10 @@ var BitfabOpenAITracingProcessor = class {
       });
     }
     try {
-      spanData.response = span.spanData?._response || null;
+      const response = span.spanData?._response;
+      if (response !== void 0) {
+        spanData.response = response;
+      }
     } catch (error) {
       errors.push({
         source: "sdk",
@@ -2654,11 +2663,11 @@ var Bitfab = class {
         startedAt,
         spanType: options.type ?? "custom"
       };
-      const sendSpan = async (params) => {
+      const sendSpan = async (params, spanOpts) => {
         const replayCtx = getReplayContext();
         const persistenceCollector = isRootSpan ? replayCtx?.pendingPersistence : void 0;
         let resolvePersistence;
-        if (persistenceCollector) {
+        if (persistenceCollector && !spanOpts?.skipPersistenceRegistration) {
           persistenceCollector.push(
             new Promise((resolve) => {
               resolvePersistence = resolve;
@@ -2758,11 +2767,41 @@ var Bitfab = class {
           }
         }
       }
+      const recordSpan = (result) => {
+        if (options.finalize) {
+          const replayCtx = getReplayContext();
+          const persistenceCollector = isRootSpan ? replayCtx?.pendingPersistence : void 0;
+          let resolvePersistence;
+          if (persistenceCollector) {
+            persistenceCollector.push(
+              new Promise((resolve) => {
+                resolvePersistence = resolve;
+              })
+            );
+          }
+          void Promise.resolve().then(() => options.finalize(result)).then(
+            (output) => sendSpan(
+              { result: output },
+              { skipPersistenceRegistration: true }
+            )
+          ).catch(
+            (error) => sendSpan(
+              {
+                result: void 0,
+                error: error instanceof Error ? `finalize failed: ${error.message}` : `finalize failed: ${String(error)}`
+              },
+              { skipPersistenceRegistration: true }
+            )
+          ).finally(() => resolvePersistence?.());
+        } else {
+          void sendSpan({ result });
+        }
+      };
       const executeWithContext = () => {
         const result = fn(...args);
         if (result instanceof Promise) {
           return result.then((resolvedResult) => {
-            void sendSpan({ result: resolvedResult });
+            recordSpan(resolvedResult);
             return resolvedResult;
           }).catch((error) => {
             void sendSpan({
@@ -2775,7 +2814,7 @@ var Bitfab = class {
         if (isAsyncGenerator(result)) {
           return wrapAsyncGenerator(result, newStack, sendSpan);
         }
-        void sendSpan({ result });
+        recordSpan(result);
         return result;
       };
       return runWithSpanStack(newStack, executeWithContext);
@@ -3059,6 +3098,54 @@ var BitfabFunction = class {
   }
 };
 
+// src/finalizers.ts
+async function settle(value) {
+  try {
+    return await value;
+  } catch {
+    return void 0;
+  }
+}
+async function aiSdk(result) {
+  const r = result ?? {};
+  const [text, usage, totalUsage, finishReason, toolCalls, toolResults] = await Promise.all([
+    settle(r.text),
+    settle(r.usage),
+    settle(r.totalUsage),
+    settle(r.finishReason),
+    settle(r.toolCalls),
+    settle(r.toolResults)
+  ]);
+  return {
+    text,
+    usage: totalUsage ?? usage,
+    finishReason,
+    toolCalls,
+    toolResults
+  };
+}
+async function readableStream(stream, onLive) {
+  const [live, copy] = stream.tee();
+  onLive(live);
+  const chunks = [];
+  const reader = copy.getReader();
+  try {
+    for (; ; ) {
+      const { done, value } = await reader.read();
+      if (done) {
+        break;
+      }
+      chunks.push(value);
+    }
+  } catch {
+  }
+  return { chunks };
+}
+var finalizers = {
+  aiSdk,
+  readableStream
+};
+
 export {
   __version__,
   DEFAULT_SERVICE_URL,
@@ -3071,6 +3158,7 @@ export {
   getCurrentSpan,
   getCurrentTrace,
   Bitfab,
-  BitfabFunction
+  BitfabFunction,
+  finalizers
 };
-//# sourceMappingURL=chunk-IUZIGC6T.js.map
+//# sourceMappingURL=chunk-75LZO6JS.js.map