observa-sdk 0.0.14 → 0.0.16

package/dist/index.cjs

@@ -451,7 +451,7 @@ function extractProviderFromModel(model) {
   }
   if (typeof model === "string") {
     const parts = model.split("/");
-    if (parts.length > 1) {
+    if (parts.length > 1 && parts[0]) {
       return parts[0].toLowerCase();
     }
     const modelLower = model.toLowerCase();
@@ -537,6 +537,53 @@ async function traceGenerateText(originalFn, args, options) {
     throw error;
   }
 }
+function wrapReadableStream(stream, onComplete, onError) {
+  const [userStream, trackingStream] = stream.tee();
+  const decoder = new TextDecoder();
+  let firstTokenTime = null;
+  const streamStartTime = Date.now();
+  const chunks = [];
+  (async () => {
+    try {
+      const reader = trackingStream.getReader();
+      while (true) {
+        const { done, value } = await reader.read();
+        if (done) break;
+        if (firstTokenTime === null && value !== null && value !== void 0) {
+          firstTokenTime = Date.now();
+        }
+        let text;
+        if (typeof value === "string") {
+          text = value;
+        } else if (value !== null && value !== void 0) {
+          try {
+            const testValue = value;
+            if (testValue instanceof Uint8Array || typeof ArrayBuffer !== "undefined" && typeof ArrayBuffer.isView === "function" && ArrayBuffer.isView(testValue)) {
+              text = decoder.decode(testValue, { stream: true });
+            } else {
+              text = String(value);
+            }
+          } catch {
+            text = String(value);
+          }
+        } else {
+          continue;
+        }
+        chunks.push(text);
+      }
+      const fullText = chunks.join("");
+      onComplete({
+        text: fullText,
+        timeToFirstToken: firstTokenTime ? firstTokenTime - streamStartTime : null,
+        streamingDuration: firstTokenTime ? Date.now() - firstTokenTime : null,
+        totalLatency: Date.now() - streamStartTime
+      });
+    } catch (error) {
+      onError(error);
+    }
+  })();
+  return userStream;
+}
 async function traceStreamText(originalFn, args, options) {
   const startTime = Date.now();
   const requestParams = args[0] || {};
@@ -546,43 +593,46 @@ async function traceStreamText(originalFn, args, options) {
   try {
     const result = await originalFn(...args);
     if (result.textStream) {
-      const wrappedStream = wrapStream(
-        result.textStream,
-        (fullResponse) => {
-          recordTrace3(
+      const originalTextStream = result.textStream;
+      const isReadableStream = originalTextStream && typeof originalTextStream.getReader === "function";
+      if (isReadableStream) {
+        const wrappedStream = wrapReadableStream(
+          originalTextStream,
+          (fullResponse) => {
+            recordTrace3(
+              {
+                model: modelIdentifier,
+                prompt: requestParams.prompt || requestParams.messages || null,
+                messages: requestParams.messages || null
+              },
+              fullResponse,
+              startTime,
+              options,
+              fullResponse.timeToFirstToken,
+              fullResponse.streamingDuration,
+              provider
+            );
+          },
+          (err) => recordError3(
             {
               model: modelIdentifier,
-              prompt: requestParams.prompt || requestParams.messages || null,
-              messages: requestParams.messages || null
+              prompt: requestParams.prompt || requestParams.messages || null
             },
-            fullResponse,
+            err,
             startTime,
-            options,
-            fullResponse.timeToFirstToken,
-            fullResponse.streamingDuration,
-            provider
-          );
-        },
-        (err) => recordError3(
-          {
-            model: modelIdentifier,
-            prompt: requestParams.prompt || requestParams.messages || null
-          },
-          err,
-          startTime,
-          options
-        ),
-        "vercel-ai"
-      );
-      const wrappedResult = Object.create(Object.getPrototypeOf(result));
-      Object.assign(wrappedResult, result);
-      Object.defineProperty(wrappedResult, "textStream", {
-        value: wrappedStream,
-        writable: true,
-        enumerable: true,
-        configurable: true
-      });
-      return wrappedResult;
+            options
+          )
+        );
+        const wrappedResult = Object.create(Object.getPrototypeOf(result));
+        Object.assign(wrappedResult, result);
+        Object.defineProperty(wrappedResult, "textStream", {
+          value: wrappedStream,
+          writable: true,
+          enumerable: true,
+          configurable: true
+        });
+        return wrappedResult;
+      }
     }
     recordTrace3(
       {

package/dist/index.js

@@ -431,7 +431,7 @@ function extractProviderFromModel(model) {
   }
   if (typeof model === "string") {
     const parts = model.split("/");
-    if (parts.length > 1) {
+    if (parts.length > 1 && parts[0]) {
       return parts[0].toLowerCase();
     }
     const modelLower = model.toLowerCase();
@@ -517,6 +517,53 @@ async function traceGenerateText(originalFn, args, options) {
     throw error;
   }
 }
+function wrapReadableStream(stream, onComplete, onError) {
+  const [userStream, trackingStream] = stream.tee();
+  const decoder = new TextDecoder();
+  let firstTokenTime = null;
+  const streamStartTime = Date.now();
+  const chunks = [];
+  (async () => {
+    try {
+      const reader = trackingStream.getReader();
+      while (true) {
+        const { done, value } = await reader.read();
+        if (done) break;
+        if (firstTokenTime === null && value !== null && value !== void 0) {
+          firstTokenTime = Date.now();
+        }
+        let text;
+        if (typeof value === "string") {
+          text = value;
+        } else if (value !== null && value !== void 0) {
+          try {
+            const testValue = value;
+            if (testValue instanceof Uint8Array || typeof ArrayBuffer !== "undefined" && typeof ArrayBuffer.isView === "function" && ArrayBuffer.isView(testValue)) {
+              text = decoder.decode(testValue, { stream: true });
+            } else {
+              text = String(value);
+            }
+          } catch {
+            text = String(value);
+          }
+        } else {
+          continue;
+        }
+        chunks.push(text);
+      }
+      const fullText = chunks.join("");
+      onComplete({
+        text: fullText,
+        timeToFirstToken: firstTokenTime ? firstTokenTime - streamStartTime : null,
+        streamingDuration: firstTokenTime ? Date.now() - firstTokenTime : null,
+        totalLatency: Date.now() - streamStartTime
+      });
+    } catch (error) {
+      onError(error);
+    }
+  })();
+  return userStream;
+}
 async function traceStreamText(originalFn, args, options) {
   const startTime = Date.now();
   const requestParams = args[0] || {};
@@ -526,43 +573,46 @@ async function traceStreamText(originalFn, args, options) {
   try {
     const result = await originalFn(...args);
     if (result.textStream) {
-      const wrappedStream = wrapStream(
-        result.textStream,
-        (fullResponse) => {
-          recordTrace3(
+      const originalTextStream = result.textStream;
+      const isReadableStream = originalTextStream && typeof originalTextStream.getReader === "function";
+      if (isReadableStream) {
+        const wrappedStream = wrapReadableStream(
+          originalTextStream,
+          (fullResponse) => {
+            recordTrace3(
+              {
+                model: modelIdentifier,
+                prompt: requestParams.prompt || requestParams.messages || null,
+                messages: requestParams.messages || null
+              },
+              fullResponse,
+              startTime,
+              options,
+              fullResponse.timeToFirstToken,
+              fullResponse.streamingDuration,
+              provider
+            );
+          },
+          (err) => recordError3(
             {
               model: modelIdentifier,
-              prompt: requestParams.prompt || requestParams.messages || null,
-              messages: requestParams.messages || null
+              prompt: requestParams.prompt || requestParams.messages || null
             },
-            fullResponse,
+            err,
             startTime,
-            options,
-            fullResponse.timeToFirstToken,
-            fullResponse.streamingDuration,
-            provider
-          );
-        },
-        (err) => recordError3(
-          {
-            model: modelIdentifier,
-            prompt: requestParams.prompt || requestParams.messages || null
-          },
-          err,
-          startTime,
-          options
-        ),
-        "vercel-ai"
-      );
-      const wrappedResult = Object.create(Object.getPrototypeOf(result));
-      Object.assign(wrappedResult, result);
-      Object.defineProperty(wrappedResult, "textStream", {
-        value: wrappedStream,
-        writable: true,
-        enumerable: true,
-        configurable: true
-      });
-      return wrappedResult;
+            options
+          )
+        );
+        const wrappedResult = Object.create(Object.getPrototypeOf(result));
+        Object.assign(wrappedResult, result);
+        Object.defineProperty(wrappedResult, "textStream", {
+          value: wrappedStream,
+          writable: true,
+          enumerable: true,
+          configurable: true
+        });
+        return wrappedResult;
+      }
     }
     recordTrace3(
       {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "observa-sdk",
-  "version": "0.0.14",
+  "version": "0.0.16",
   "description": "Enterprise-grade observability SDK for AI applications. Track and monitor LLM interactions with zero friction.",
   "type": "module",
   "main": "./dist/index.cjs",