@trustgraph/react-state 1.5.1 → 1.5.2

package/dist/index.esm.js

@@ -1399,9 +1399,9 @@ const useGraphSubgraph = ({ entityUri, flow, collection, }) => {
 
 /**
  * Custom hook for querying graph embeddings
- * Finds graph entities similar to the provided embedding vectors
+ * Finds graph entities similar to the provided embedding vector
  */
-const useGraphEmbeddings = ({ flow, vecs, limit = 10, collection }) => {
+const useGraphEmbeddings = ({ flow, vec, limit = 10, collection }) => {
     const socket = useSocket();
     const notify = useNotification();
     const { settings } = useSettings();
@@ -1409,12 +1409,12 @@ const useGraphEmbeddings = ({ flow, vecs, limit = 10, collection }) => {
     const effectiveFlow = flow ?? sessionFlowId;
     const effectiveCollection = collection ?? settings.collection;
     const query = useQuery({
-        queryKey: ["graph-embeddings", { flow: effectiveFlow, vecs, limit, collection: effectiveCollection }],
-        enabled: !!vecs && vecs.length > 0 && !!effectiveFlow,
+        queryKey: ["graph-embeddings", { flow: effectiveFlow, vec, limit, collection: effectiveCollection }],
+        enabled: !!vec && vec.length > 0 && !!effectiveFlow,
         queryFn: () => {
             return socket
                 .flow(effectiveFlow)
-                .graphEmbeddingsQuery(vecs, limit, effectiveCollection)
+                .graphEmbeddingsQuery(vec, limit, effectiveCollection)
                 .catch((err) => {
                 const message = err instanceof Error ? err.message : String(err);
                 notify.error(message);
@@ -1432,24 +1432,23 @@ const useGraphEmbeddings = ({ flow, vecs, limit = 10, collection }) => {
     };
 };
 
-// Take the embeddings, and lookup entities using graph
+// Take the embedding vector, and lookup entities using graph
 // embeddings, add embedding to each entity row, just an easy
 // place to put it
 const getGraphEmbeddings = (socket, add, remove, limit, collection) => {
-    // Take the embeddings, and lookup entities using graph
-    // embeddings, add embedding to each entity row, just an easy
-    // place to put it
-    return (vecs) => {
+    // Take the embedding, and lookup entities using graph
+    // embeddings, add embedding to each entity row
+    return (vec) => {
         const act = "Graph embedding search";
         add(act);
         return socket
-            .graphEmbeddingsQuery(vecs, limit , collection)
-            .then((ents) => {
+            .graphEmbeddingsQuery(vec, limit , collection)
+            .then((matches) => {
             remove(act);
-            return ents
-                .filter((ent) => ent.t === "i")
-                .map((ent) => {
-                return { uri: ent.i, target: vecs[0] };
+            return matches
+                .filter((m) => m.entity !== null && m.entity.t === "i")
+                .map((m) => {
+                return { uri: m.entity.i, target: vec };
             });
         })
             .catch((err) => {
@@ -1532,11 +1531,11 @@ const addRowEmbeddings = (socket, add, remove) => (entities) => {
         return socket
             .embeddings([text])
             .then((x) => {
-            if (x && x.length > 0 && x[0].length > 0) {
+            if (x && x.length > 0) {
                 remove(act);
                 return {
                     ...ent,
-                    embeddings: x[0][0],
+                    embeddings: x[0],
                 };
             }
             else {
@@ -1896,7 +1895,10 @@ const useChatSession = ({ flow } = {}) => {
             // Start embeddings activity
             addActivity(embActivity);
             // Get labels for each entity
-            const labelPromises = result.entities.map(async (entity) => {
+            const labelPromises = result.entities
+                .filter((match) => match.entity !== null)
+                .map(async (match) => {
+                const entity = match.entity;
                 const labelActivity = "Label " + getTermValue$2(entity);
                 addActivity(labelActivity);
                 try {