Add hybrid Synxt

datastax · clun · Apr 10, 2025 · Apr 4, 2025 · Apr 8, 2025 · Apr 8, 2025
commit e2c424fce41fe07b2e9a8607e27a52d5599af472
diff --git a/astra-db-java/src/main/java/com/datastax/astra/client/core/query/Sort.java b/astra-db-java/src/main/java/com/datastax/astra/client/core/query/Sort.java
@@ -236,4 +236,19 @@ public static Sort hybrid(Hybrid hybrid) {
                 .build();
     }
 
+    /**
+     * Build a sort clause with vectorize.
+     *
+     * @param hybrid
+     *      hybrid sort
+     * @return
+     *       sort instance.
+     */
+    public static Sort hybrid(String hybrid) {
+        return internalBuilder()
+                .field(DataAPIKeywords.HYBRID.getKeyword())
+                .hybrid(new Hybrid(hybrid))
+                .build();
+    }
+
 }
diff --git a/examples/src/main/java/com/datastax/astra/client/collections/findrerank/SampleFind.java b/examples/src/main/java/com/datastax/astra/client/collections/findrerank/SampleFind.java
@@ -0,0 +1,39 @@
+package com.datastax.astra.client.collections.findrerank;
+
+import com.datastax.astra.client.DataAPIClient;
+import com.datastax.astra.client.collections.Collection;
+import com.datastax.astra.client.collections.commands.options.CollectionFindAndRerankOptions;
+import com.datastax.astra.client.collections.definition.documents.Document;
+import com.datastax.astra.client.core.hybrid.Hybrid;
+import com.datastax.astra.client.core.query.Sort;
+import com.datastax.astra.client.core.rerank.RerankedResult;
+
+public class SampleFind {
+
+    public static void main(String[] args) {
+        Collection<Document> collection = new DataAPIClient("TOKEN")
+                .getDatabase("API_ENDPOINT")
+                .getCollection("COLLECTION_NAME");
+
+        // With Vectorize
+        for (RerankedResult<Document> results : collection
+                .findAndRerank(new CollectionFindAndRerankOptions()
+                        .sort(Sort.hybrid("A tree on a hill")))) {
+            System.out.println(results.getDocument());
+        }
+
+        // Without Vectorize
+        Hybrid hybrid = new Hybrid()
+                .vector(new float[]{0.25f, 0.25f, 0.25f, 0.25f, 0.25f})
+                .lexical("A tree on a hill");
+        for (RerankedResult<Document> results : collection
+                .findAndRerank(new CollectionFindAndRerankOptions()
+                        .sort(Sort.hybrid(hybrid))
+                        .rerankOn("$lexical")
+                        .rerankQuery("A house in the woods"))) {
+            System.out.println(results.getDocument());
+        }
+
+        // Use a different query in the reranking step
+    }
+}