@absolutejs/rag 0.0.16 → 0.0.18

package/dist/vue/index.js

@@ -194,6 +194,78 @@ var createRAGClient = (options) => {
       }
       return parseJson(response);
     },
+    async evaluateStream(input, options2) {
+      const response = await fetchImpl(`${basePath}/evaluate/stream`, {
+        body: JSON.stringify(input),
+        headers: jsonHeaders,
+        method: "POST",
+        signal: options2?.signal
+      });
+      if (!response.ok || !response.body) {
+        throw new Error(await toErrorMessage(response));
+      }
+      const reader = response.body.getReader();
+      const decoder = new TextDecoder;
+      let buffer = "";
+      let result = null;
+      let streamError = null;
+      const handleEvent = (rawEvent) => {
+        let eventName = "message";
+        const dataLines = [];
+        for (const line of rawEvent.split(`
+`)) {
+          if (line.startsWith("event:")) {
+            eventName = line.slice("event:".length).trim();
+          } else if (line.startsWith("data:")) {
+            dataLines.push(line.slice("data:".length).trim());
+          }
+        }
+        if (dataLines.length === 0) {
+          return;
+        }
+        const payload = JSON.parse(dataLines.join(`
+`));
+        if (eventName === "start") {
+          options2?.onStart?.(payload);
+        } else if (eventName === "case") {
+          options2?.onCase?.(payload);
+        } else if (eventName === "result") {
+          result = payload;
+        } else if (eventName === "error") {
+          streamError = payload.error ?? "RAG evaluation stream failed";
+        }
+      };
+      for (;; ) {
+        const { done, value } = await reader.read();
+        if (done) {
+          break;
+        }
+        buffer += decoder.decode(value, { stream: true });
+        let separator = buffer.indexOf(`
+
+`);
+        while (separator !== -1) {
+          const rawEvent = buffer.slice(0, separator);
+          buffer = buffer.slice(separator + 2);
+          if (rawEvent.trim().length > 0) {
+            handleEvent(rawEvent);
+          }
+          separator = buffer.indexOf(`
+
+`);
+        }
+      }
+      if (buffer.trim().length > 0) {
+        handleEvent(buffer);
+      }
+      if (streamError) {
+        throw new Error(streamError);
+      }
+      if (!result) {
+        throw new Error("RAG evaluation stream ended without a result");
+      }
+      return result;
+    },
     async compareRetrievals(input) {
       const response = await fetchImpl(`${basePath}/compare/retrieval`, {
         body: JSON.stringify(input),
@@ -4904,5 +4976,5 @@ export {
   useRAG
 };
 
-//# debugId=0F73B22F9985247C64756E2164756E21
+//# debugId=AEB60E9CDC7E85E464756E2164756E21
 //# sourceMappingURL=index.js.map