@trustgraph/react-state 1.5.4 → 1.6.0

package/dist/index.cjs

@@ -1808,10 +1808,45 @@ const useInference = ({ flow } = {}) => {
         },
     });
     /**
-     * Agent inference with streaming callbacks
+     * Document RAG inference with optional explainability
+     */
+    const documentRagMutation = reactQuery.useMutation({
+        mutationFn: async ({ input, collection, docLimit, callbacks, }) => {
+            const wantsExplainability = !!callbacks?.onExplain;
+            const explainEvents = wantsExplainability ? [] : undefined;
+            const response = await new Promise((resolve, reject) => {
+                let accumulated = "";
+                const onChunk = (chunk, complete) => {
+                    accumulated += chunk;
+                    callbacks?.onChunk?.(chunk, complete);
+                    if (complete) {
+                        resolve(accumulated);
+                    }
+                };
+                const onError = (error) => {
+                    callbacks?.onError?.(error);
+                    reject(new Error(error));
+                };
+                const onExplain = wantsExplainability
+                    ? (event) => {
+                        explainEvents.push(event);
+                        callbacks.onExplain(event);
+                    }
+                    : undefined;
+                socket
+                    .flow(effectiveFlow)
+                    .documentRagStreaming(input, onChunk, onError, docLimit, collection, onExplain);
+            });
+            return { response, explainEvents };
+        },
+    });
+    /**
+     * Agent inference with streaming callbacks and optional explainability
      */
     const agentMutation = reactQuery.useMutation({
         mutationFn: async ({ input, callbacks, }) => {
+            const wantsExplainability = !!callbacks?.onExplain;
+            const explainEvents = wantsExplainability ? [] : undefined;
             return new Promise((resolve, reject) => {
                 let fullAnswer = "";
                 const onThink = (thought, complete) => {
@@ -1831,17 +1866,25 @@ const useInference = ({ flow } = {}) => {
                     callbacks?.onError?.(error);
                     reject(new Error(error));
                 };
+                const onExplain = wantsExplainability
+                    ? (event) => {
+                        explainEvents.push(event);
+                        callbacks.onExplain(event);
+                    }
+                    : undefined;
                 socket
                     .flow(effectiveFlow)
-                    .agent(input, onThink, onObserve, onAnswer, onError);
+                    .agent(input, onThink, onObserve, onAnswer, onError, onExplain);
             });
         },
     });
     return {
         graphRag: graphRagMutation.mutateAsync,
+        documentRag: documentRagMutation.mutateAsync,
         textCompletion: textCompletionMutation.mutateAsync,
         agent: agentMutation.mutateAsync,
         isLoading: graphRagMutation.isPending ||
+            documentRagMutation.isPending ||
             textCompletionMutation.isPending ||
             agentMutation.isPending,
     };