@trustgraph/react-state 1.4.2 → 1.4.4

package/dist/index.d.ts

@@ -31,6 +31,7 @@ export type { GraphRagOptions, GraphRagResult, GraphRagCallbacks, TextCompletion
 export { useChatSession, useChat } from "./state/chat-session";
 export { useStructuredQuery } from "./state/structured-query";
 export { useRowEmbeddingsQuery } from "./state/row-embeddings-query";
+export { useDocumentEmbeddingsQuery } from "./state/document-embeddings-query";
 export { useRowsQuery } from "./state/rows-query";
 export { useEmbeddings } from "./state/embeddings";
 export { useCollections } from "./state/collections";

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EACL,cAAc,EACd,aAAa,EACb,sBAAsB,GACvB,MAAM,4BAA4B,CAAC;AACpC,YAAY,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AAGtE,YAAY,EACV,OAAO,EACP,MAAM,EACN,IAAI,EACJ,OAAO,EACP,SAAS,EACT,WAAW,EACX,UAAU,EACV,eAAe,EACf,iBAAiB,EACjB,kBAAkB,GACnB,MAAM,oBAAoB,CAAC;AAG5B,OAAO,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AAC9D,YAAY,EAAE,yBAAyB,EAAE,MAAM,wBAAwB,CAAC;AACxE,YAAY,EAAE,mBAAmB,EAAE,MAAM,SAAS,CAAC;AAGnD,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAGlD,OAAO,EAAE,qBAAqB,EAAE,MAAM,kBAAkB,CAAC;AACzD,YAAY,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AACtD,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAClD,YAAY,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AACpD,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,YAAY,EAAE,iBAAiB,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AACxE,OAAO,EAAE,sBAAsB,EAAE,MAAM,mBAAmB,CAAC;AAC3D,YAAY,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AACjD,YAAY,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAC;AACrD,YAAY,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAGlD,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AACzC,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AACvD,OAAO,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAC9D,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,YAAY,EACV,eAAe,EACf,cAAc,EACd,iBAAiB,EACjB,uBAAuB,EACvB,cAAc,GACf,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,cAAc,EAAE,OAAO,EAAE,MAAM,sBAAsB,CAAC;AAC/D,OAAO,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAC9D,OAAO,EAAE,qBAAqB,EAAE,MAAM,8BAA8B,CAAC;AACrE,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,YAAY,EAAE,QAAQ,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AACrE,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAC5D,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,iBAAiB,EAAE,uBAAuB,EAAE,MAAM,yBAAyB,CAAC;AACrF,OAAO,EACL,iBAAiB,EACjB,sBAAsB,GACvB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAGtD,YAAY,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AAC7C,YAAY,EAAE,OAAO,EAAE,MAAM,iBAAiB,CAAC;AAC/C,YAAY,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AACvD,OAAO,EACL,gBAAgB,EAChB,oBAAoB,GACrB,MAAM,wBAAwB,CAAC;AAChC,YAAY,EAAE,iBAAiB,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAGxE,OAAO,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AACvE,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACrD,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AAC/E,YAAY,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAC1E,OAAO,EAAE,eAAe,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AACzE,YAAY,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EACL,cAAc,EACd,aAAa,EACb,sBAAsB,GACvB,MAAM,4BAA4B,CAAC;AACpC,YAAY,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AAGtE,YAAY,EACV,OAAO,EACP,MAAM,EACN,IAAI,EACJ,OAAO,EACP,SAAS,EACT,WAAW,EACX,UAAU,EACV,eAAe,EACf,iBAAiB,EACjB,kBAAkB,GACnB,MAAM,oBAAoB,CAAC;AAG5B,OAAO,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AAC9D,YAAY,EAAE,yBAAyB,EAAE,MAAM,wBAAwB,CAAC;AACxE,YAAY,EAAE,mBAAmB,EAAE,MAAM,SAAS,CAAC;AAGnD,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,WAAW,EAAE,MAAM,qBAAqB,CAAC;AAGlD,OAAO,EAAE,qBAAqB,EAAE,MAAM,kBAAkB,CAAC;AACzD,YAAY,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AACtD,OAAO,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AAClD,YAAY,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AACpD,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,YAAY,EAAE,iBAAiB,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AACxE,OAAO,EAAE,sBAAsB,EAAE,MAAM,mBAAmB,CAAC;AAC3D,YAAY,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACxD,OAAO,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AACjD,YAAY,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAC9C,OAAO,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAC;AACrD,YAAY,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAGlD,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AACzC,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AACvD,OAAO,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAC9D,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,YAAY,EACV,eAAe,EACf,cAAc,EACd,iBAAiB,EACjB,uBAAuB,EACvB,cAAc,GACf,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,cAAc,EAAE,OAAO,EAAE,MAAM,sBAAsB,CAAC;AAC/D,OAAO,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAC9D,OAAO,EAAE,qBAAqB,EAAE,MAAM,8BAA8B,CAAC;AACrE,OAAO,EAAE,0BAA0B,EAAE,MAAM,mCAAmC,CAAC;AAC/E,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAC;AACnD,YAAY,EAAE,QAAQ,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AACrE,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAC5D,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,iBAAiB,EAAE,uBAAuB,EAAE,MAAM,yBAAyB,CAAC;AACrF,OAAO,EACL,iBAAiB,EACjB,sBAAsB,GACvB,MAAM,yBAAyB,CAAC;AACjC,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAGtD,YAAY,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAC;AAC7C,YAAY,EAAE,OAAO,EAAE,MAAM,iBAAiB,CAAC;AAC/C,YAAY,EAAE,QAAQ,EAAE,MAAM,wBAAwB,CAAC;AACvD,OAAO,EACL,gBAAgB,EAChB,oBAAoB,GACrB,MAAM,wBAAwB,CAAC;AAChC,YAAY,EAAE,iBAAiB,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAGxE,OAAO,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,2BAA2B,CAAC;AACvE,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAC;AACrD,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AAC/E,YAAY,EAAE,WAAW,EAAE,aAAa,EAAE,MAAM,yBAAyB,CAAC;AAC1E,OAAO,EAAE,eAAe,EAAE,WAAW,EAAE,MAAM,2BAA2B,CAAC;AACzE,YAAY,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC"}

package/dist/index.esm.js

@@ -925,17 +925,19 @@ const useTriples = ({ flow, s, p, o, limit, collection }) => {
     const notify = useNotification();
     // Settings for default collection
     const { settings } = useSettings();
-    if (!flow)
-        flow = "default";
+    // Session state for default flow ID
+    const sessionFlowId = useSessionStore((state) => state.flowId);
+    // Use explicit param if provided, otherwise fall back to session state
+    const effectiveFlow = flow ?? sessionFlowId;
     /**
      * Query for fetching all token costs
      * Uses React Query for caching and background refetching
      */
     const query = useQuery({
-        queryKey: ["triples", { flow, s, p, o, limit }],
+        queryKey: ["triples", { flow: effectiveFlow, s, p, o, limit }],
         queryFn: () => {
             return socket
-                .flow(flow)
+                .flow(effectiveFlow)
                 .triplesQuery(s, p, o, limit, collection || settings.collection)
                 .then((x) => {
                 if (x["error"]) {
@@ -1301,13 +1303,17 @@ const updateSubgraphByRelationship = (socket, selectedNodeId, relationshipUri, d
  * Custom hook for managing graph visualization operations using React Query
  * Provides functionality for fetching and updating graph subgraphs
  * @param entityUri - The URI of the entity to build the graph around
- * @param flowId - The flow ID to use for the query
+ * @param flow - Optional flow ID to use for the query (defaults to session state)
  * @param collection - The collection to query
  * @returns {Object} Graph state and operations
  */
-const useGraphSubgraph = (entityUri, flowId, collection) => {
+const useGraphSubgraph = ({ entityUri, flow, collection, }) => {
     // WebSocket connection for communicating with the graph service
     const socket = useSocket();
+    // Session state for default flow ID
+    const sessionFlowId = useSessionStore((state) => state.flowId);
+    // Use explicit param if provided, otherwise fall back to session state
+    const effectiveFlow = flow ?? sessionFlowId;
     const addActivity = useProgressStateStore((state) => state.addActivity);
     const removeActivity = useProgressStateStore((state) => state.removeActivity);
     // Hook for displaying user notifications
@@ -1319,29 +1325,29 @@ const useGraphSubgraph = (entityUri, flowId, collection) => {
      * Uses React Query for caching and background refetching
      */
     const query = useQuery({
-        queryKey: ["graph-subgraph", { entityUri, flowId, collection }],
+        queryKey: ["graph-subgraph", { entityUri, flow: effectiveFlow, collection }],
         queryFn: async () => {
             if (!entityUri) {
                 throw new Error("Entity URI is required");
             }
             const sg = createSubgraph();
             // Use the existing updateSubgraph utility function for initial load
-            const api = socket.flow(flowId);
+            const api = socket.flow(effectiveFlow);
             return updateSubgraph(api, entityUri, sg, addActivity, removeActivity, collection);
         },
-        enabled: !!entityUri && !!flowId, // Only run query if both entityUri and flowId are available
+        enabled: !!entityUri && !!effectiveFlow, // Only run query if both entityUri and effectiveFlow are available
     });
     /**
      * Mutation for updating the graph subgraph when nodes are clicked
      */
     const updateMutation = useMutation({
         mutationFn: async ({ nodeId, currentGraph, }) => {
-            const api = socket.flow(flowId);
+            const api = socket.flow(effectiveFlow);
             return updateSubgraph(api, nodeId, currentGraph, addActivity, removeActivity);
         },
         onSuccess: (newGraph) => {
             // Update the cache with the new graph data
-            queryClient.setQueryData(["graph-subgraph", { entityUri, flowId, collection }], newGraph);
+            queryClient.setQueryData(["graph-subgraph", { entityUri, flow: effectiveFlow, collection }], newGraph);
         },
         onError: (err) => {
             console.log("Graph update error:", err);
@@ -1353,12 +1359,12 @@ const useGraphSubgraph = (entityUri, flowId, collection) => {
      */
     const relationshipNavigationMutation = useMutation({
         mutationFn: async ({ selectedNodeId, relationshipUri, direction, currentGraph, }) => {
-            const api = socket.flow(flowId);
+            const api = socket.flow(effectiveFlow);
             return updateSubgraphByRelationship(api, selectedNodeId, relationshipUri, direction, currentGraph, addActivity, removeActivity, collection);
         },
         onSuccess: (newGraph) => {
             // Update the cache with the new graph data
-            queryClient.setQueryData(["graph-subgraph", { entityUri, flowId, collection }], newGraph);
+            queryClient.setQueryData(["graph-subgraph", { entityUri, flow: effectiveFlow, collection }], newGraph);
         },
         onError: (err) => {
             console.log("Relationship navigation error:", err);
@@ -1392,51 +1398,36 @@ const useGraphSubgraph = (entityUri, flowId, collection) => {
 };
 
 /**
- * Custom hook for managing token cost operations
- * Provides functionality for fetching, deleting, and updating token costs
- * for AI models
- * @returns {Object} Token cost state and operations
+ * Custom hook for querying graph embeddings
+ * Finds graph entities similar to the provided embedding vectors
  */
-const useGraphEmbeddings = ({ flow, vecs, limit, collection }) => {
-    // WebSocket connection for communicating with the configuration service
+const useGraphEmbeddings = ({ flow, vecs, limit = 10, collection }) => {
     const socket = useSocket();
-    // Hook for displaying user notifications
     const notify = useNotification();
-    // Settings for default collection
     const { settings } = useSettings();
-    if (!flow)
-        flow = "default";
-    /**
-     * Query for fetching graph embeddings
-     * Uses React Query for caching and background refetching
-     */
+    const sessionFlowId = useSessionStore((state) => state.flowId);
+    const effectiveFlow = flow ?? sessionFlowId;
+    const effectiveCollection = collection ?? settings.collection;
     const query = useQuery({
-        queryKey: ["graph-embeddings", { vecs, limit }],
+        queryKey: ["graph-embeddings", { flow: effectiveFlow, vecs, limit, collection: effectiveCollection }],
+        enabled: !!vecs && vecs.length > 0 && !!effectiveFlow,
         queryFn: () => {
             return socket
-                .flow(flow)
-                .graphEmbeddingsQuery(vecs, limit, collection || settings.collection)
-                .then((x) => {
-                return x;
-            })
+                .flow(effectiveFlow)
+                .graphEmbeddingsQuery(vecs, limit, effectiveCollection)
                 .catch((err) => {
-                console.log("Error:", err);
                 const message = err instanceof Error ? err.message : String(err);
                 notify.error(message);
                 throw err;
             });
         },
     });
-    // Show loading indicators for long-running operations
     useActivity(query.isLoading, "Loading graph embeddings");
-    // Return token cost state and operations for use in components
     return {
-        // Token cost query state
         graphEmbeddings: query.data,
         isLoading: query.isLoading,
         isError: query.isError,
         error: query.error,
-        // Manual refetch function
         refetch: query.refetch,
     };
 };
@@ -1620,6 +1611,8 @@ const useVectorSearch = () => {
     const removeActivity = useProgressStateStore((state) => state.removeActivity);
     // Hook for displaying user notifications
     const notify = useNotification();
+    // Session state for default flow ID
+    const sessionFlowId = useSessionStore((state) => state.flowId);
     // State to track current search parameters
     const [searchParams, setSearchParams] = useState(null);
     /**
@@ -1631,7 +1624,7 @@ const useVectorSearch = () => {
         enabled: !!searchParams?.term,
         queryFn: () => {
             const { flow, term, limit, collection } = searchParams;
-            return vectorSearch(socket, flow || "default", addActivity, removeActivity, term, collection, limit)
+            return vectorSearch(socket, flow ?? sessionFlowId, addActivity, removeActivity, term, collection, limit)
                 .then((x) => {
                 if (x["error"]) {
                     console.log("Error:", x);
@@ -1653,7 +1646,7 @@ const useVectorSearch = () => {
             return;
         }
         setSearchParams({
-            flow: flow || "default",
+            flow: flow ?? sessionFlowId,
             term,
             limit: limit || 10,
             collection,
@@ -1677,13 +1670,17 @@ const useVectorSearch = () => {
  * Custom hook for managing entity detail operations using React Query
  * Provides functionality for fetching entity details and related triples
  * @param entityUri - The URI of the entity to fetch details for
- * @param flowId - The flow ID to use for the query
+ * @param flow - Optional flow ID to use for the query (defaults to session state)
  * @param collection - The collection to query
  * @returns {Object} Entity detail state and operations
  */
-const useEntityDetail = (entityUri, flowId, collection) => {
+const useEntityDetail = ({ entityUri, flow, collection, }) => {
     // WebSocket connection for communicating with the graph service
     const socket = useSocket();
+    // Session state for default flow ID
+    const sessionFlowId = useSessionStore((state) => state.flowId);
+    // Use explicit param if provided, otherwise fall back to session state
+    const effectiveFlow = flow ?? sessionFlowId;
     const addActivity = useProgressStateStore((state) => state.addActivity);
     const removeActivity = useProgressStateStore((state) => state.removeActivity);
     // Hook for displaying user notifications
@@ -1693,17 +1690,17 @@ const useEntityDetail = (entityUri, flowId, collection) => {
      * Uses React Query for caching and background refetching
      */
     const query = useQuery({
-        queryKey: ["entity-detail", { entityUri, flowId, collection }],
+        queryKey: ["entity-detail", { entityUri, flow: effectiveFlow, collection }],
         queryFn: async () => {
             if (!entityUri) {
                 throw new Error("Entity URI is required");
             }
             // Use the existing getTriples utility function
-            const api = socket.flow(flowId);
+            const api = socket.flow(effectiveFlow);
             return getTriples(api, entityUri, addActivity, removeActivity, undefined, collection);
         },
-        // Only run query if both entityUri and flowId are available
-        enabled: !!entityUri && !!flowId,
+        // Only run query if both entityUri and effectiveFlow are available
+        enabled: !!entityUri && !!effectiveFlow,
     });
     // Show loading indicators for long-running operations
     useActivity(query.isLoading, entityUri ? `Knowledge graph search: ${entityUri}` : "Loading entity");
@@ -1727,9 +1724,11 @@ const useEntityDetail = (entityUri, flowId, collection) => {
  * Hook providing low-level access to LLM inference services
  * No conversation state or side effects - just the API calls
  */
-const useInference = () => {
+const useInference = ({ flow } = {}) => {
     const socket = useSocket();
-    const flowId = useSessionStore((state) => state.flowId);
+    const sessionFlowId = useSessionStore((state) => state.flowId);
+    // Use explicit param if provided, otherwise fall back to session state
+    const effectiveFlow = flow ?? sessionFlowId;
     /**
      * Graph RAG inference with entity discovery
      */
@@ -1751,15 +1750,15 @@ const useInference = () => {
                         reject(new Error(error));
                     };
                     socket
-                        .flow(flowId)
+                        .flow(effectiveFlow)
                         .graphRagStreaming(input, onChunk, onError, options, collection);
                 })
-                : await socket.flow(flowId).graphRag(input, options || {}, collection);
+                : await socket.flow(effectiveFlow).graphRag(input, options || {}, collection);
             // Get embeddings for entity discovery
-            const embeddings = await socket.flow(flowId).embeddings(input);
+            const embeddings = await socket.flow(effectiveFlow).embeddings(input);
             // Query graph embeddings to find entities
             const entities = await socket
-                .flow(flowId)
+                .flow(effectiveFlow)
                 .graphEmbeddingsQuery(embeddings, options?.entityLimit || 10, collection);
             return { response, entities };
         },
@@ -1785,10 +1784,10 @@ const useInference = () => {
                         reject(new Error(error));
                     };
                     socket
-                        .flow(flowId)
+                        .flow(effectiveFlow)
                         .textCompletionStreaming(systemPrompt, input, onChunk, onError);
                 })
-                : await socket.flow(flowId).textCompletion(systemPrompt, input);
+                : await socket.flow(effectiveFlow).textCompletion(systemPrompt, input);
         },
     });
     /**
@@ -1816,7 +1815,7 @@ const useInference = () => {
                     reject(new Error(error));
                 };
                 socket
-                    .flow(flowId)
+                    .flow(effectiveFlow)
                     .agent(input, onThink, onObserve, onAnswer, onError);
             });
         },
@@ -1836,7 +1835,7 @@ const useInference = () => {
  * Combines conversation state with inference services
  * Handles routing, progress tracking, entity management, and notifications
  */
-const useChatSession = () => {
+const useChatSession = ({ flow } = {}) => {
     const socket = useSocket();
     const notify = useNotification();
     // Conversation state
@@ -1848,12 +1847,14 @@ const useChatSession = () => {
     const addActivity = useProgressStateStore((state) => state.addActivity);
     const removeActivity = useProgressStateStore((state) => state.removeActivity);
     // Session and workbench state
-    const flowId = useSessionStore((state) => state.flowId);
+    const sessionFlowId = useSessionStore((state) => state.flowId);
     const setEntities = useWorkbenchStateStore((state) => state.setEntities);
+    // Use explicit param if provided, otherwise fall back to session state
+    const effectiveFlow = flow ?? sessionFlowId;
     // Settings for GraphRAG configuration
     const { settings } = useSettings();
     // Inference services
-    const inference = useInference();
+    const inference = useInference({ flow });
     /**
      * Graph RAG chat handling with entity discovery
      */
@@ -1898,7 +1899,7 @@ const useChatSession = () => {
                 addActivity(labelActivity);
                 try {
                     const triples = await socket
-                        .flow(flowId)
+                        .flow(effectiveFlow)
                         .triplesQuery(entity, { t: "i", i: RDFS_LABEL }, undefined, 1, settings.collection);
                     removeActivity(labelActivity);
                     return triples;
@@ -2154,69 +2155,97 @@ const useStructuredQuery = () => {
 };
 
 // @ts-nocheck
-// React Query hooks for data fetching and mutation management
 /**
- * Custom hook for managing row embeddings query operations
- * Provides functionality for executing semantic searches on structured data indexes
- * First converts query text to embeddings, then searches for similar records
+ * Custom hook for querying row embeddings using vectors.
+ * Searches for similar records in structured data indexes.
  */
-const useRowEmbeddingsQuery = () => {
-    // Socket connection for API calls
+const useRowEmbeddingsQuery = ({ flow } = {}) => {
     const socket = useSocket();
     const connectionState = useConnectionState();
-    // Notification system for user feedback
     const notify = useNotification();
-    // Session state for current flow ID
-    const flowId = useSessionStore((state) => state.flowId);
-    // Settings for default collection
+    const sessionFlowId = useSessionStore((state) => state.flowId);
     const { settings } = useSettings();
-    // Only enable operations when socket is connected and ready
+    const effectiveFlow = flow ?? sessionFlowId;
     const isSocketReady = connectionState?.status === "authenticated" ||
         connectionState?.status === "unauthenticated";
-    // Mutation for executing row embeddings queries
-    const rowEmbeddingsQueryMutation = useMutation({
-        mutationFn: async ({ query, schemaName, collection, indexName, limit, }) => {
+    const mutation = useMutation({
+        mutationFn: async ({ vectors, schemaName, collection, indexName, limit = 10, }) => {
             if (!isSocketReady) {
                 throw new Error("Socket connection not ready");
             }
-            const flow = socket.flow(flowId);
-            // First, get embeddings for the query text
-            const vectors = await flow.embeddings(query);
-            // Then query row embeddings with those vectors
-            return flow.rowEmbeddingsQuery(vectors, schemaName, collection || settings.collection, indexName, limit || 10);
+            return socket.flow(effectiveFlow).rowEmbeddingsQuery(vectors, schemaName, collection ?? settings.collection, indexName, limit);
         },
         onError: (err) => {
-            console.log("Row embeddings query error:", err);
-            const errorMessage = err instanceof Error
-                ? err.message
-                : err?.toString() || "Unknown error";
-            notify.error(`Row embeddings query failed: ${errorMessage}`);
+            const message = err instanceof Error ? err.message : String(err);
+            notify.error(`Row embeddings query failed: ${message}`);
         },
         onSuccess: (data) => {
-            if (data && data.length > 0) {
-                notify.success(`Found ${data.length} matching record${data.length !== 1 ? 's' : ''}`);
+            const count = data?.length ?? 0;
+            if (count > 0) {
+                notify.success(`Found ${count} matching record${count !== 1 ? "s" : ""}`);
             }
             else {
                 notify.info("No matching records found");
             }
         },
     });
-    // Show loading indicator for row embeddings query operations
-    useActivity(rowEmbeddingsQueryMutation.isPending, "Executing row embeddings query");
-    // Return the public API for the hook
+    useActivity(mutation.isPending, "Querying row embeddings");
     return {
-        // Query execution
-        executeQuery: rowEmbeddingsQueryMutation.mutate,
-        executeQueryAsync: rowEmbeddingsQueryMutation.mutateAsync,
-        // Query state
-        isExecuting: rowEmbeddingsQueryMutation.isPending,
-        error: rowEmbeddingsQueryMutation.error,
-        // Results
-        matches: rowEmbeddingsQueryMutation.data || [],
-        hasResults: (rowEmbeddingsQueryMutation.data?.length || 0) > 0,
-        // Reset function to clear previous results
-        reset: rowEmbeddingsQueryMutation.reset,
-        // Socket readiness
+        executeQuery: mutation.mutate,
+        executeQueryAsync: mutation.mutateAsync,
+        isExecuting: mutation.isPending,
+        error: mutation.error,
+        matches: mutation.data ?? [],
+        hasResults: (mutation.data?.length ?? 0) > 0,
+        reset: mutation.reset,
+        isReady: isSocketReady,
+    };
+};
+
+// @ts-nocheck
+/**
+ * Custom hook for querying document chunks using vectors.
+ * Searches for document chunks with similar embeddings.
+ */
+const useDocumentEmbeddingsQuery = ({ flow } = {}) => {
+    const socket = useSocket();
+    const connectionState = useConnectionState();
+    const notify = useNotification();
+    const sessionFlowId = useSessionStore((state) => state.flowId);
+    const { settings } = useSettings();
+    const effectiveFlow = flow ?? sessionFlowId;
+    const isSocketReady = connectionState?.status === "authenticated" ||
+        connectionState?.status === "unauthenticated";
+    const mutation = useMutation({
+        mutationFn: async ({ vectors, user, collection, limit = 10, }) => {
+            if (!isSocketReady) {
+                throw new Error("Socket connection not ready");
+            }
+            return socket.flow(effectiveFlow).documentEmbeddingsQuery(vectors, user ?? settings.user, collection ?? settings.collection, limit);
+        },
+        onError: (err) => {
+            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");
+            }
+        },
+    });
+    useActivity(mutation.isPending, "Searching document chunks");
+    return {
+        executeQuery: mutation.mutate,
+        executeQueryAsync: mutation.mutateAsync,
+        isExecuting: mutation.isPending,
+        error: mutation.error,
+        results: mutation.data ?? [],
+        hasResults: Array.isArray(mutation.data) && mutation.data.length > 0,
+        reset: mutation.reset,
         isReady: isSocketReady,
     };
 };
@@ -2226,14 +2255,16 @@ const useRowEmbeddingsQuery = () => {
  * Custom hook for managing GraphQL rows queries
  * Provides functionality for executing GraphQL queries against structured row data
  */
-const useRowsQuery = () => {
+const useRowsQuery = ({ flow } = {}) => {
     // Socket connection for API calls
     const socket = useSocket();
     const connectionState = useConnectionState();
     // Notification system for user feedback
     const notify = useNotification();
     // Session state for current flow ID
-    const flowId = useSessionStore((state) => state.flowId);
+    const sessionFlowId = useSessionStore((state) => state.flowId);
+    // Use explicit param if provided, otherwise fall back to session state
+    const effectiveFlow = flow ?? sessionFlowId;
     // Settings for default collection
     const { settings } = useSettings();
     // Only enable operations when socket is connected and ready
@@ -2246,7 +2277,7 @@ const useRowsQuery = () => {
                 throw new Error("Socket connection not ready");
             }
             return socket
-                .flow(flowId)
+                .flow(effectiveFlow)
                 .rowsQuery(query, collection || settings.collection, variables, operationName);
         },
         onError: (err) => {
@@ -2292,18 +2323,20 @@ const useEmbeddings = ({ flow, term }) => {
         connectionState?.status === "unauthenticated";
     // Hook for displaying user notifications
     const notify = useNotification();
-    if (!flow)
-        flow = "default";
+    // Session state for default flow ID
+    const sessionFlowId = useSessionStore((state) => state.flowId);
+    // Use explicit param if provided, otherwise fall back to session state
+    const effectiveFlow = flow ?? sessionFlowId;
     /**
      * Query for fetching all token costs
      * Uses React Query for caching and background refetching
      */
     const query = useQuery({
-        queryKey: ["embeddings", { flow, term }],
-        enabled: isSocketReady && !!term && !!flow,
+        queryKey: ["embeddings", { flow: effectiveFlow, term }],
+        enabled: isSocketReady && !!term && !!effectiveFlow,
         queryFn: () => {
             return socket
-                .flow(flow)
+                .flow(effectiveFlow)
                 .embeddings(term)
                 .then((x) => {
                 if (x["error"]) {
@@ -2441,14 +2474,16 @@ const useCollections = () => {
  * Custom hook for managing NLP query operations
  * Provides functionality for converting natural language questions to GraphQL queries
  */
-const useNlpQuery = () => {
+const useNlpQuery = ({ flow } = {}) => {
     // Socket connection for API calls
     const socket = useSocket();
     const connectionState = useConnectionState();
     // Notification system for user feedback
     const notify = useNotification();
     // Session state for current flow ID
-    const flowId = useSessionStore((state) => state.flowId);
+    const sessionFlowId = useSessionStore((state) => state.flowId);
+    // Use explicit param if provided, otherwise fall back to session state
+    const effectiveFlow = flow ?? sessionFlowId;
     // Only enable operations when socket is connected and ready
     const isSocketReady = connectionState?.status === "authenticated" ||
         connectionState?.status === "unauthenticated";
@@ -2458,7 +2493,7 @@ const useNlpQuery = () => {
             if (!isSocketReady) {
                 throw new Error("Socket connection not ready");
             }
-            return socket.flow(flowId).nlpQuery(question, maxResults);
+            return socket.flow(effectiveFlow).nlpQuery(question, maxResults);
         },
         onError: (err) => {
             console.log("NLP query error:", err);
@@ -4634,5 +4669,5 @@ const useNodeDetails = (nodeId, flowId) => {
     };
 };
 
-export { DEFAULT_SETTINGS, NotificationProvider, RDFS_LABEL, SETTINGS_STORAGE_KEY, createDocId, fileToBase64, generateFlowBlueprintId, getTermValue$2 as getTermValue, getTriples, prepareMetadata, textToBase64, useActivity, useAgentTools, useChat, useChatSession, useCollections, useConversation, useEmbeddings, useEntityDetail, useFlowBlueprints, useFlowParameters, useFlows, useGraphEmbeddings, useGraphSubgraph, useInference, useKnowledgeCores, useLLMModels, useLibrary, useLoadStateStore, useMcpTools, useNlpQuery, useNodeDetails, useNotification, useOntologies, useParameterValidation, useProcessing, useProgressStateStore, usePrompts, useRowEmbeddingsQuery, useRowsQuery, useSchemas, useSearchStateStore, useSessionStore, useSettings, useStructuredQuery, useTokenCosts, useTriples, useVectorSearch, useWorkbenchStateStore, vectorSearch };
+export { DEFAULT_SETTINGS, NotificationProvider, RDFS_LABEL, SETTINGS_STORAGE_KEY, createDocId, fileToBase64, generateFlowBlueprintId, getTermValue$2 as getTermValue, getTriples, prepareMetadata, textToBase64, useActivity, useAgentTools, useChat, useChatSession, useCollections, useConversation, useDocumentEmbeddingsQuery, useEmbeddings, useEntityDetail, useFlowBlueprints, useFlowParameters, useFlows, useGraphEmbeddings, useGraphSubgraph, useInference, useKnowledgeCores, useLLMModels, useLibrary, useLoadStateStore, useMcpTools, useNlpQuery, useNodeDetails, useNotification, useOntologies, useParameterValidation, useProcessing, useProgressStateStore, usePrompts, useRowEmbeddingsQuery, useRowsQuery, useSchemas, useSearchStateStore, useSessionStore, useSettings, useStructuredQuery, useTokenCosts, useTriples, useVectorSearch, useWorkbenchStateStore, vectorSearch };
 //# sourceMappingURL=index.esm.js.map