@trustgraph/react-state 1.4.6 → 1.5.1

package/dist/index.esm.js

@@ -1530,13 +1530,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 {
@@ -1574,7 +1574,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))
@@ -1755,7 +1756,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)
@@ -2313,7 +2315,7 @@ const useEmbeddings = ({ flow, term }) => {
         queryFn: () => {
             return socket
                 .flow(effectiveFlow)
-                .embeddings(term)
+                .embeddings([term])
                 .then((x) => {
                 if (x["error"]) {
                     console.log("Error:", x);