@trustgraph/react-state 1.4.5 → 1.5.1

package/dist/index.cjs

@@ -1531,13 +1531,13 @@ const addRowEmbeddings = (socket, add, remove) => (entities) => {
         const act = "Embeddings " + text.substring(0, 20);
         add(act);
         return socket
-            .embeddings(text)
+            .embeddings([text])
             .then((x) => {
-            if (x && x.length > 0) {
+            if (x && x.length > 0 && x[0].length > 0) {
                 remove(act);
                 return {
                     ...ent,
-                    embeddings: x[0],
+                    embeddings: x[0][0],
                 };
             }
             else {
@@ -1575,7 +1575,8 @@ const vectorSearch = (socket, flowId, addActivity, removeActivity, term, collect
     const searchAct = "Search: " + term;
     addActivity(searchAct);
     return api
-        .embeddings(term)
+        .embeddings([term])
+        .then((vecs) => vecs[0])
         .then(getGraphEmbeddings(api, addActivity, removeActivity, limit || 10, collection))
         .then(addRowLabels(api, addActivity, removeActivity, collection))
         .then(addRowDefinitions(api, addActivity, removeActivity, collection))
@@ -1756,7 +1757,8 @@ const useInference = ({ flow } = {}) => {
                 })
                 : await socket.flow(effectiveFlow).graphRag(input, options || {}, collection);
             // Get embeddings for entity discovery
-            const embeddings = await socket.flow(effectiveFlow).embeddings(input);
+            const allEmbeddings = await socket.flow(effectiveFlow).embeddings([input]);
+            const embeddings = allEmbeddings[0];
             // Query graph embeddings to find entities
             const entities = await socket
                 .flow(effectiveFlow)
@@ -2120,16 +2122,7 @@ const useStructuredQuery = () => {
                 : err?.toString() || "Unknown error";
             notify.error(`Structured query failed: ${errorMessage}`);
         },
-        onSuccess: (data) => {
-            if (data.data) {
-                notify.success("Structured query executed successfully");
-            }
-            else if (data.errors && data.errors.length > 0) {
-                notify.warning("Query executed with errors");
-            }
-            else {
-                notify.warning("Query processed but returned no data");
-            }
+        onSuccess: () => {
         },
     });
     // Show loading indicator for structured query operations
@@ -2221,14 +2214,7 @@ const useDocumentEmbeddingsQuery = ({ flow } = {}) => {
             const message = err instanceof Error ? err.message : String(err);
             notify.error(`Document embeddings query failed: ${message}`);
         },
-        onSuccess: (data) => {
-            const count = Array.isArray(data) ? data.length : 0;
-            if (count > 0) {
-                notify.success(`Found ${count} matching chunk${count !== 1 ? "s" : ""}`);
-            }
-            else {
-                notify.info("No matching document chunks found");
-            }
+        onSuccess: () => {
         },
     });
     useActivity(mutation.isPending, "Searching document chunks");
@@ -2330,7 +2316,7 @@ const useEmbeddings = ({ flow, term }) => {
         queryFn: () => {
             return socket
                 .flow(effectiveFlow)
-                .embeddings(term)
+                .embeddings([term])
                 .then((x) => {
                 if (x["error"]) {
                     console.log("Error:", x);
@@ -2493,13 +2479,7 @@ const useNlpQuery = ({ flow } = {}) => {
             const errorMessage = err instanceof Error ? err.message : String(err);
             notify.error(`NLP query conversion failed: ${errorMessage}`);
         },
-        onSuccess: (data) => {
-            if (data.graphql_query) {
-                notify.success("Natural language converted to GraphQL successfully");
-            }
-            else {
-                notify.warning("Query processed but no GraphQL generated");
-            }
+        onSuccess: () => {
         },
     });
     // Show loading indicator for conversion operations
@@ -3497,8 +3477,6 @@ const useKnowledgeCores = () => {
             notify.error(err.toString());
         },
         onSuccess: () => {
-            // Show success notification
-            notify.success("Knowledge cores loaded successfully");
         },
     });
     // Show loading indicators for long-running operations