@absolutejs/rag 0.0.16 → 0.0.18

package/dist/react/index.js

@@ -2897,6 +2897,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),
@@ -4201,6 +4273,7 @@ var useRAGEvaluate = (path) => {
   const [suiteRuns, setSuiteRuns] = useState3([]);
   const [lastRequest, setLastRequest] = useState3(null);
   const [lastResponse, setLastResponse] = useState3(null);
+  const [streamProgress, setStreamProgress] = useState3(null);
   const evaluate = useCallback3(async (input) => {
     setIsEvaluating(true);
     setError(null);
@@ -4217,6 +4290,33 @@ var useRAGEvaluate = (path) => {
       setIsEvaluating(false);
     }
   }, [client]);
+  const evaluateStream = useCallback3(async (input, options) => {
+    setIsEvaluating(true);
+    setError(null);
+    setLastRequest(input);
+    setStreamProgress({ done: 0, total: input.cases.length });
+    try {
+      const response = await client.evaluateStream(input, {
+        onCase: (event) => {
+          setStreamProgress((current) => ({
+            done: (current?.done ?? 0) + 1,
+            total: event.total
+          }));
+          options?.onCase?.(event);
+        },
+        onStart: ({ total }) => setStreamProgress({ done: 0, total }),
+        signal: options?.signal
+      });
+      setLastResponse(response);
+      return response;
+    } catch (caught) {
+      const message = caught instanceof Error ? caught.message : String(caught);
+      setError(message);
+      throw caught;
+    } finally {
+      setIsEvaluating(false);
+    }
+  }, [client]);
   const saveSuite = useCallback3((suite) => {
     setSuites((current) => {
       const next = current.filter((entry) => entry.id !== suite.id);
@@ -4263,6 +4363,7 @@ var useRAGEvaluate = (path) => {
     clearRuns,
     error,
     evaluate,
+    evaluateStream,
     isEvaluating,
     lastRequest,
     lastResponse,
@@ -4271,6 +4372,7 @@ var useRAGEvaluate = (path) => {
     reset,
     runSuite,
     saveSuite,
+    streamProgress,
     suiteRuns,
     suites
   };
@@ -4941,5 +5043,5 @@ export {
   useRAG
 };
 
-//# debugId=A6BF1BC41FD6B14B64756E2164756E21
+//# debugId=2C44ECEE37502BBC64756E2164756E21
 //# sourceMappingURL=index.js.map