@burtson-labs/bandit-engine 2.0.160 → 2.0.162

package/dist/{chunk-B3HAQC6F.mjs → chunk-Q6LXVYTC.mjs}

@@ -10,6 +10,36 @@ import { map as map2 } from "rxjs";
 
 // src/services/ollama/ollama.service.ts
 import { catchError, from, lastValueFrom, map, Observable, of, shareReplay, switchMap, throwError, timeout } from "rxjs";
+
+// src/services/stream/streamLineParser.ts
+var SKIP = { kind: "skip" };
+var StreamLineParser = class {
+  /** Set by `event: ping` until the frame's blank-line terminator. */
+  inHeartbeatFrame = false;
+  next(rawLine) {
+    const line = rawLine.trim();
+    if (!line) {
+      this.inHeartbeatFrame = false;
+      return SKIP;
+    }
+    if (line.startsWith(":")) return SKIP;
+    if (line.startsWith("event:")) {
+      this.inHeartbeatFrame = line.slice(6).trim().toLowerCase() === "ping";
+      return SKIP;
+    }
+    if (line.startsWith("id:") || line.startsWith("retry:")) return SKIP;
+    if (line.startsWith("data:")) {
+      if (this.inHeartbeatFrame) return SKIP;
+      const data = line.slice(5).trim();
+      if (!data) return SKIP;
+      if (data === "[DONE]") return { kind: "done" };
+      return { kind: "json", text: data };
+    }
+    return { kind: "json", text: line };
+  }
+};
+
+// src/services/ollama/ollama.service.ts
 var handleError = () => (obs) => obs.pipe(
   switchMap(
     (response) => response.ok ? of(response) : throwError(() => new Error(`Request failed: ${response.status} ${response.statusText}`))
@@ -100,13 +130,15 @@ var OllamaService = class {
         this._throwErrorIfNotOk(url, response);
         const reader = response.body?.getReader();
         const decoder = new TextDecoder();
+        const lineParser = new StreamLineParser();
         let buffer = "";
         const read = () => {
           reader?.read().then(({ done, value }) => {
             if (done) {
-              if (buffer.trim() !== "") {
+              const tail = lineParser.next(buffer);
+              if (tail.kind === "json") {
                 try {
-                  observer.next(JSON.parse(buffer));
+                  observer.next(JSON.parse(tail.text));
                 } catch (err) {
                   observer.error(err);
                   debugLogger.error("Final chunk parsing error (generate):", { buffer });
@@ -119,9 +151,10 @@ var OllamaService = class {
             const lines = buffer.split("\n");
             buffer = lines.pop() ?? "";
             for (const line of lines) {
-              if (line.trim()) {
+              const classified = lineParser.next(line);
+              if (classified.kind === "json") {
                 try {
-                  observer.next(JSON.parse(line));
+                  observer.next(JSON.parse(classified.text));
                 } catch (err) {
                   observer.error(err);
                   debugLogger.error("Error parsing JSON line (generate):", { line });
@@ -151,13 +184,15 @@ var OllamaService = class {
         this._throwErrorIfNotOk(url, response);
         const reader = response.body?.getReader();
         const decoder = new TextDecoder();
+        const lineParser = new StreamLineParser();
         let buffer = "";
         const read = () => {
           reader?.read().then(({ done, value }) => {
             if (done) {
-              if (buffer.trim() !== "") {
+              const tail = lineParser.next(buffer);
+              if (tail.kind === "json") {
                 try {
-                  observer.next(JSON.parse(buffer));
+                  observer.next(JSON.parse(tail.text));
                 } catch (err) {
                   observer.error(err);
                   debugLogger.error("Final chunk parsing error (chat):", { buffer });
@@ -170,9 +205,10 @@ var OllamaService = class {
             const lines = buffer.split("\n");
             buffer = lines.pop() ?? "";
             for (const line of lines) {
-              if (line.trim()) {
+              const classified = lineParser.next(line);
+              if (classified.kind === "json") {
                 try {
-                  observer.next(JSON.parse(line));
+                  observer.next(JSON.parse(classified.text));
                 } catch (err) {
                   observer.error(err);
                   debugLogger.error("Error parsing JSON line (chat):", { line });
@@ -1323,13 +1359,15 @@ var GatewayService = class {
       const handleStreamingResponse = async (response) => {
         const reader = response.body?.getReader();
         const decoder = new TextDecoder();
+        const lineParser = new StreamLineParser();
         let buffer = "";
         const read = () => {
           reader?.read().then(({ done, value }) => {
             if (done) {
-              if (buffer.trim() !== "") {
+              const tail = lineParser.next(buffer);
+              if (tail.kind === "json") {
                 try {
-                  const finalResponse = JSON.parse(buffer);
+                  const finalResponse = JSON.parse(tail.text);
                   observer.next(finalResponse);
                 } catch (err) {
                   debugLogger.error("GatewayService chat final chunk parsing error", { buffer, error: err });
@@ -1343,16 +1381,14 @@ var GatewayService = class {
             const lines = buffer.split("\n");
             buffer = lines.pop() ?? "";
             for (const line of lines) {
-              const trimmed = line.trim();
-              if (trimmed) {
-                let data = trimmed;
-                if (trimmed.startsWith("data: ")) {
-                  data = trimmed.slice(6);
-                  if (data === "[DONE]") {
-                    observer.complete();
-                    return;
-                  }
-                }
+              const classified = lineParser.next(line);
+              if (classified.kind === "done") {
+                observer.complete();
+                return;
+              }
+              if (classified.kind === "json") {
+                const trimmed = line.trim();
+                const data = classified.text;
                 try {
                   const parsed = JSON.parse(data);
                   if (request.provider === "ollama" && parsed.message) {
@@ -1519,13 +1555,15 @@ var GatewayService = class {
         }
         const reader = response.body?.getReader();
         const decoder = new TextDecoder();
+        const lineParser = new StreamLineParser();
         let buffer = "";
         const read = () => {
           reader?.read().then(({ done, value }) => {
             if (done) {
-              if (buffer.trim() !== "") {
+              const tail = lineParser.next(buffer);
+              if (tail.kind === "json") {
                 try {
-                  observer.next(JSON.parse(buffer));
+                  observer.next(JSON.parse(tail.text));
                 } catch (err) {
                   observer.error(err);
                   debugLogger.error("Final chunk parsing error (gateway generate):", { buffer });
@@ -1538,9 +1576,14 @@ var GatewayService = class {
             const lines = buffer.split("\n");
             buffer = lines.pop() ?? "";
             for (const line of lines) {
-              if (line.trim()) {
+              const classified = lineParser.next(line);
+              if (classified.kind === "done") {
+                observer.complete();
+                return;
+              }
+              if (classified.kind === "json") {
                 try {
-                  observer.next(JSON.parse(line));
+                  observer.next(JSON.parse(classified.text));
                 } catch (err) {
                   observer.error(err);
                   debugLogger.error("Error parsing JSON line (gateway generate):", { line });
@@ -3111,4 +3154,4 @@ export {
   AIProviderFactory,
   useAIProviderStore
 };
-//# sourceMappingURL=chunk-B3HAQC6F.mjs.map
+//# sourceMappingURL=chunk-Q6LXVYTC.mjs.map