ai 2.1.14 → 2.1.15

package/dist/index.d.ts

@@ -91,7 +91,7 @@ declare function streamToResponse(res: ReadableStream, response: ServerResponse,
 
 declare function HuggingFaceStream(res: AsyncGenerator<any>, callbacks?: AIStreamCallbacks): ReadableStream;
 
-declare function CohereStream(res: AsyncGenerator<any>, callbacks?: AIStreamCallbacks): ReadableStream;
+declare function CohereStream(reader: Response, callbacks?: AIStreamCallbacks): ReadableStream;
 
 declare function AnthropicStream(res: Response, cb?: AIStreamCallbacks): ReadableStream;
 

package/dist/index.js

@@ -244,27 +244,56 @@ function HuggingFaceStream(res, callbacks) {
 }
 
 // streams/cohere-stream.ts
+var utf8Decoder = new TextDecoder("utf-8");
+function processLines(lines, controller) {
+  return __async(this, null, function* () {
+    for (const line of lines) {
+      const { text, is_finished } = JSON.parse(line);
+      if (is_finished === true) {
+        controller.close();
+      } else {
+        controller.enqueue(text);
+      }
+    }
+  });
+}
+function readAndProcessLines(reader, controller) {
+  return __async(this, null, function* () {
+    let segment = "";
+    while (true) {
+      const { value: chunk, done } = yield reader.read();
+      if (done) {
+        break;
+      }
+      segment += utf8Decoder.decode(chunk, { stream: true });
+      const linesArray = segment.split(/\r\n|\n|\r/g);
+      segment = linesArray.pop() || "";
+      yield processLines(linesArray, controller);
+    }
+    if (segment) {
+      const linesArray = [segment];
+      yield processLines(linesArray, controller);
+    }
+    controller.close();
+  });
+}
 function createParser3(res) {
+  var _a;
+  const reader = (_a = res.body) == null ? void 0 : _a.getReader();
   return new ReadableStream({
-    pull(controller) {
+    start(controller) {
       return __async(this, null, function* () {
-        const { value, done } = yield res.next();
-        if (done) {
+        if (!reader) {
           controller.close();
           return;
         }
-        const { text, is_finished } = JSON.parse(value);
-        if (is_finished === true) {
-          controller.close();
-        } else {
-          controller.enqueue(text);
-        }
+        yield readAndProcessLines(reader, controller);
       });
     }
   });
 }
-function CohereStream(res, callbacks) {
-  return createParser3(res).pipeThrough(createCallbacksTransformer(callbacks));
+function CohereStream(reader, callbacks) {
+  return createParser3(reader).pipeThrough(createCallbacksTransformer(callbacks));
 }
 
 // streams/anthropic-stream.ts

package/dist/index.mjs

@@ -211,27 +211,56 @@ function HuggingFaceStream(res, callbacks) {
 }
 
 // streams/cohere-stream.ts
+var utf8Decoder = new TextDecoder("utf-8");
+function processLines(lines, controller) {
+  return __async(this, null, function* () {
+    for (const line of lines) {
+      const { text, is_finished } = JSON.parse(line);
+      if (is_finished === true) {
+        controller.close();
+      } else {
+        controller.enqueue(text);
+      }
+    }
+  });
+}
+function readAndProcessLines(reader, controller) {
+  return __async(this, null, function* () {
+    let segment = "";
+    while (true) {
+      const { value: chunk, done } = yield reader.read();
+      if (done) {
+        break;
+      }
+      segment += utf8Decoder.decode(chunk, { stream: true });
+      const linesArray = segment.split(/\r\n|\n|\r/g);
+      segment = linesArray.pop() || "";
+      yield processLines(linesArray, controller);
+    }
+    if (segment) {
+      const linesArray = [segment];
+      yield processLines(linesArray, controller);
+    }
+    controller.close();
+  });
+}
 function createParser3(res) {
+  var _a;
+  const reader = (_a = res.body) == null ? void 0 : _a.getReader();
   return new ReadableStream({
-    pull(controller) {
+    start(controller) {
       return __async(this, null, function* () {
-        const { value, done } = yield res.next();
-        if (done) {
+        if (!reader) {
           controller.close();
           return;
         }
-        const { text, is_finished } = JSON.parse(value);
-        if (is_finished === true) {
-          controller.close();
-        } else {
-          controller.enqueue(text);
-        }
+        yield readAndProcessLines(reader, controller);
       });
     }
   });
 }
-function CohereStream(res, callbacks) {
-  return createParser3(res).pipeThrough(createCallbacksTransformer(callbacks));
+function CohereStream(reader, callbacks) {
+  return createParser3(reader).pipeThrough(createCallbacksTransformer(callbacks));
 }
 
 // streams/anthropic-stream.ts

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai",
-  "version": "2.1.14",
+  "version": "2.1.15",
   "license": "Apache-2.0",
   "sideEffects": false,
   "main": "./dist/index.js",