@trustgraph/react-state 1.3.1 → 1.4.1

package/README.md

@@ -156,7 +156,7 @@ const handler: NotificationHandler = {
 - `useChat()` - Chat interface operations
 - `useChatQuery()` - Chat query management
 - `useStructuredQuery()` - Structured query operations
-- `useObjectsQuery()` - Object queries
+- `useRowsQuery()` - Row queries
 - `useNlpQuery()` - Natural language processing queries
 
 #### Configuration

package/dist/index.cjs

@@ -2159,12 +2159,80 @@ 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
+ */
+const useRowEmbeddingsQuery = () => {
+    // Socket connection for API calls
+    const socket = reactProvider.useSocket();
+    const connectionState = reactProvider.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 { settings } = useSettings();
+    // Only enable operations when socket is connected and ready
+    const isSocketReady = connectionState?.status === "authenticated" ||
+        connectionState?.status === "unauthenticated";
+    // Mutation for executing row embeddings queries
+    const rowEmbeddingsQueryMutation = reactQuery.useMutation({
+        mutationFn: async ({ query, schemaName, collection, indexName, limit, }) => {
+            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);
+        },
+        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}`);
+        },
+        onSuccess: (data) => {
+            if (data && data.length > 0) {
+                notify.success(`Found ${data.length} matching record${data.length !== 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
+    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
+        isReady: isSocketReady,
+    };
+};
+
 // React Query hooks for data fetching and mutation management
 /**
- * Custom hook for managing GraphQL objects queries
- * Provides functionality for executing GraphQL queries against structured data objects
+ * Custom hook for managing GraphQL rows queries
+ * Provides functionality for executing GraphQL queries against structured row data
  */
-const useObjectsQuery = () => {
+const useRowsQuery = () => {
     // Socket connection for API calls
     const socket = reactProvider.useSocket();
     const connectionState = reactProvider.useConnectionState();
@@ -2177,18 +2245,18 @@ const useObjectsQuery = () => {
     // Only enable operations when socket is connected and ready
     const isSocketReady = connectionState?.status === "authenticated" ||
         connectionState?.status === "unauthenticated";
-    // Mutation for executing GraphQL objects queries
-    const objectsQueryMutation = reactQuery.useMutation({
+    // Mutation for executing GraphQL rows queries
+    const rowsQueryMutation = reactQuery.useMutation({
         mutationFn: async ({ query, collection, variables, operationName, }) => {
             if (!isSocketReady) {
                 throw new Error("Socket connection not ready");
             }
             return socket
                 .flow(flowId)
-                .objectsQuery(query, collection || settings.collection, variables, operationName);
+                .rowsQuery(query, collection || settings.collection, variables, operationName);
         },
         onError: (err) => {
-            console.log("Objects query error:", err);
+            console.log("Rows query error:", err);
             const errorMessage = err instanceof Error ? err.message : String(err);
             notify.error(`GraphQL query failed: ${errorMessage}`);
         },
@@ -2197,18 +2265,18 @@ const useObjectsQuery = () => {
         },
     });
     // Show loading indicator for query operations
-    useActivity(objectsQueryMutation.isPending, "Executing GraphQL query");
+    useActivity(rowsQueryMutation.isPending, "Executing GraphQL query");
     // Return the public API for the hook
     return {
         // Query execution
-        executeQuery: objectsQueryMutation.mutate,
-        executeQueryAsync: objectsQueryMutation.mutateAsync,
+        executeQuery: rowsQueryMutation.mutate,
+        executeQueryAsync: rowsQueryMutation.mutateAsync,
         // Query state
-        isExecuting: objectsQueryMutation.isPending,
-        error: objectsQueryMutation.error,
-        data: objectsQueryMutation.data,
+        isExecuting: rowsQueryMutation.isPending,
+        error: rowsQueryMutation.error,
+        data: rowsQueryMutation.data,
         // Reset function to clear previous results
-        reset: objectsQueryMutation.reset,
+        reset: rowsQueryMutation.reset,
         // Socket readiness
         isReady: isSocketReady,
     };
@@ -4617,12 +4685,13 @@ exports.useMcpTools = useMcpTools;
 exports.useNlpQuery = useNlpQuery;
 exports.useNodeDetails = useNodeDetails;
 exports.useNotification = useNotification;
-exports.useObjectsQuery = useObjectsQuery;
 exports.useOntologies = useOntologies;
 exports.useParameterValidation = useParameterValidation;
 exports.useProcessing = useProcessing;
 exports.useProgressStateStore = useProgressStateStore;
 exports.usePrompts = usePrompts;
+exports.useRowEmbeddingsQuery = useRowEmbeddingsQuery;
+exports.useRowsQuery = useRowsQuery;
 exports.useSchemas = useSchemas;
 exports.useSearchStateStore = useSearchStateStore;
 exports.useSessionStore = useSessionStore;