@trustgraph/react-state 1.6.1 → 1.7.1

package/dist/index.cjs

@@ -1844,20 +1844,20 @@ const useInference = ({ flow } = {}) => {
      * Agent inference with streaming callbacks and optional explainability
      */
     const agentMutation = reactQuery.useMutation({
-        mutationFn: async ({ input, callbacks, }) => {
+        mutationFn: async ({ input, collection, callbacks, }) => {
             const wantsExplainability = !!callbacks?.onExplain;
             const explainEvents = wantsExplainability ? [] : undefined;
             return new Promise((resolve, reject) => {
                 let fullAnswer = "";
-                const onThink = (thought, complete) => {
-                    callbacks?.onThink?.(thought, complete);
+                const onThink = (thought, complete, messageId) => {
+                    callbacks?.onThink?.(thought, complete, messageId);
                 };
-                const onObserve = (observation, complete) => {
-                    callbacks?.onObserve?.(observation, complete);
+                const onObserve = (observation, complete, messageId) => {
+                    callbacks?.onObserve?.(observation, complete, messageId);
                 };
-                const onAnswer = (answer, complete) => {
+                const onAnswer = (answer, complete, messageId) => {
                     fullAnswer += answer;
-                    callbacks?.onAnswer?.(answer, complete);
+                    callbacks?.onAnswer?.(answer, complete, messageId);
                     if (complete) {
                         resolve(fullAnswer);
                     }
@@ -1874,7 +1874,7 @@ const useInference = ({ flow } = {}) => {
                     : undefined;
                 socket
                     .flow(effectiveFlow)
-                    .agent(input, onThink, onObserve, onAnswer, onError, onExplain);
+                    .agent(input, onThink, onObserve, onAnswer, onError, onExplain, collection);
             });
         },
     });
@@ -3061,11 +3061,9 @@ const useEmbeddings = ({ flow, term }) => {
     };
 };
 
-// @ts-nocheck
 /**
  * Custom hook for managing collection operations
  * Provides functionality for fetching, creating, updating, and deleting collections
- * @returns {Object} Collections state and operations
  */
 const useCollections = () => {
     // WebSocket connection for communicating with the collection management service