Add control to speed up marshalling

Author: clun

astra-db-java/src/main/java/com/datastax/astra/internal/utils/JsonUtils.java

@@ -154,7 +154,11 @@ public static String marshall(Object o) {
      * @param <T>
      *     current type
      */
+    @SuppressWarnings("unchecked")
     public static <T> T convertValue(Object bean, Class<T> clazz) {
+        if (bean.getClass() == clazz) {
+            return (T) bean;
+        }
         return  getDataApiObjectMapper().convertValue(bean, clazz);
     }
 

astra-db-java/src/test/java/com/datastax/astra/test/integration/collection/VectorizePreviewITTest.java

@@ -29,7 +29,7 @@ class VectorizePreviewITTest {
     static final String COLLECTION_VECTORIZE = "collection_vectorize";
     static final String NVIDIA_PROVIDER = "nvidia";
     static final String NVIDIA_MODEL = "NV-Embed-QA";
-    static final Integer NVIDIA_DIMENSION= 1024;
+    static final Integer NVIDIA_DIMENSION = 1024;
     private static Database db;
     private static Collection<Document> collectionVectorize;
 
@@ -46,11 +46,10 @@ static void setup() {
     void shouldCreateACollectionWithNvidia() {
         Collection<Document> collection = db.createCollection(COLLECTION_VECTORIZE,
                 CollectionOptions.builder()
-                .vectorDimension(NVIDIA_DIMENSION)
-                .vectorSimilarity(SimilarityMetric.COSINE)
-                .vectorize(NVIDIA_PROVIDER, NVIDIA_MODEL)
-                .build());
-
+                        .vectorDimension(NVIDIA_DIMENSION)
+                        .vectorSimilarity(SimilarityMetric.COSINE)
+                        .vectorize(NVIDIA_PROVIDER, NVIDIA_MODEL)
+                        .build());
         assertThat(collection).isNotNull();
         assertThat(db.listCollectionNames().collect(Collectors.toList())).contains(COLLECTION_VECTORIZE);
     }
@@ -68,8 +67,9 @@ void shouldInsertOneDocumentWithVectorize() {
 
     @Test
     void testFindVectorize() {
-        List<Document> doclist = collectionVectorize
-                .find(sort("Life is too short for Javascript"))
+        List<Document> doclist = collectionVectorize.find(
+                        // FindOptions.Builder.sort(...), some options
+                        sort("Life is too short for Javascript"))
                 .all();
         assertThat(doclist).isNotNull().isNotEmpty();
     }

examples/src/main/java/Quickstart.java

@@ -0,0 +1,65 @@
+import com.datastax.astra.client.Collection;
+import com.datastax.astra.client.DataAPIClient;
+import com.datastax.astra.client.Database;
+import com.datastax.astra.client.model.Document;
+import com.datastax.astra.client.model.FindIterable;
+
+import static com.datastax.astra.client.model.SimilarityMetric.COSINE;
+
+public class Quickstart {
+
+  public static void main(String[] args) {
+    // Loading Arguments
+    String astraToken = System.getenv("ASTRA_DB_APPLICATION_TOKEN");
+    String astraApiEndpoint = System.getenv("ASTRA_DB_API_ENDPOINT");
+
+    // Initialize the client. The keyspace parameter is optional if you use
+    // "default_keyspace".
+    DataAPIClient client = new DataAPIClient(astraToken);
+    System.out.println("Connected to AstraDB");
+
+    Database db = client.getDatabase(astraApiEndpoint, "default_keyspace");
+    System.out.println("Connected to Database.");
+// end::init[]
+
+// tag::collection[]
+    // Create a collection. The default similarity metric is cosine.
+    Collection<Document> collection = db
+            .createCollection("vector_test", 5, COSINE);
+    System.out.println("Created a collection");
+// end::collection[]
+
+// tag::data[]
+    // Insert documents into the collection
+    collection.insertMany(
+            new Document("1")
+                    .append("text", "ChatGPT integrated sneakers that talk to you")
+                    .vector(new float[]{0.1f, 0.15f, 0.3f, 0.12f, 0.05f}),
+            new Document("2")
+                    .append("text", "An AI quilt to help you sleep forever")
+                    .vector(new float[]{0.45f, 0.09f, 0.01f, 0.2f, 0.11f}),
+            new Document("3")
+                    .append("text", "A deep learning display that controls your mood")
+                    .vector(new float[]{0.1f, 0.05f, 0.08f, 0.3f, 0.6f}));
+    System.out.println("Inserted documents into the collection");
+// end::data[]
+
+// tag::search[]
+    // Perform a similarity search
+    FindIterable<Document> resultsSet = collection.find(
+            new float[]{0.15f, 0.1f, 0.1f, 0.35f, 0.55f},
+            10
+    );
+    resultsSet.forEach(System.out::println);
+// end::search[]
+
+// tag::cleanup[]
+    // Delete the collection
+    collection.drop();
+    System.out.println("Deleted the collection");
+// end::cleanup[]
+
+// tag::end[]
+  }
+}
+// end::end[]

examples/src/main/java/com/datastax/astra/client/QuickStart.java

@@ -2,23 +2,43 @@
 
 import com.datastax.astra.client.Collection;
 import com.datastax.astra.client.DataAPIClient;
+import com.datastax.astra.client.DataAPIOptions;
 import com.datastax.astra.client.Database;
+import com.datastax.astra.client.model.CollectionIdTypes;
+import com.datastax.astra.client.model.CollectionOptions;
 import com.datastax.astra.client.model.Document;
+import com.datastax.astra.client.model.FindOptions;
+import com.datastax.astra.client.model.Projections;
+import com.datastax.astra.client.model.SimilarityMetric;
+import com.datastax.astra.internal.command.LoggingCommandObserver;
 import com.datastax.astra.tool.csv.CsvLoader;
 import com.datastax.astra.tool.csv.CsvRowMapper;
 import lombok.extern.slf4j.Slf4j;
 
+import static com.datastax.astra.client.model.CollectionIdTypes.OBJECT_ID;
+import static com.datastax.astra.client.model.Filters.and;
+import static com.datastax.astra.client.model.Filters.eq;
+import static com.datastax.astra.client.model.Projections.include;
+
 /**
  * Load a CSV to Astra
  */
 @Slf4j
 public class CsvLoaderListing {
 
+    private static final String ASTRA_TOKEN  = "<replace_me>";
+    private static final String API_ENDPOINT = "<replace_me>";
+
     public static void main(String[] args) throws Exception {
         // Get an empty Collection
-        DataAPIClient        client      = new DataAPIClient("<replace_me>");
-        Database             db         = client.getDatabase("<replace_me>");
+        DataAPIClient        client      = new DataAPIClient(ASTRA_TOKEN);
+
+        Database             db         = client.getDatabase(API_ENDPOINT);
+
         Collection<Document> collection = db.createCollection("airbnb");
+
+        db.getCollection("airbnb").deleteAll();
+        collection.registerListener("logger", new LoggingCommandObserver(CsvLoaderListing.class));
         collection.deleteAll();
 
         // Zou !