Samples

Author: clun

astra-sdk-vector/src/main/java/com/dtsx/astra/sdk/vector/AstraVectorClient.java

@@ -207,6 +207,12 @@ public Stream<Database> findByName(String name) {
         return findAllDatabases().filter(db->name.equals(db.getInfo().getName()));
     }
 
+    public boolean isDatabaseExists(String name) {
+        return findByName(name).findFirst().isPresent();
+    }
+
+
+
     /**
      * Find a database from its id.
      *

astra-sdk-vector/src/main/java/com/dtsx/astra/sdk/vector/AstraVectorDatabaseClient.java

@@ -11,6 +11,7 @@
 import io.stargate.sdk.json.JsonCollectionClient;
 import io.stargate.sdk.json.JsonNamespaceClient;
 import io.stargate.sdk.json.domain.CollectionDefinition;
+import io.stargate.sdk.json.vector.JsonVectorStore;
 import io.stargate.sdk.json.vector.SimilarityMetric;
 import io.stargate.sdk.json.vector.VectorStore;
 import lombok.NonNull;
@@ -104,7 +105,7 @@ public void createVectorStore(String name, int dimension, SimilarityMetric metri
      * param dimension
      *      dimension
      */
-    public VectorStore<ObjectMap> createVectorStore(String name, int dimension) {
+    public JsonVectorStore createVectorStore(String name, int dimension) {
         nsClient.createCollection(name, dimension);
         return nsClient.vectorStore(name);
     }

astra-sdk-vector/src/test/java/com/dtsx/astra/sdk/vector/AstraVectorQuickStart.java

@@ -1,9 +1,17 @@
 package com.dtsx.astra.sdk.vector;
 
 import com.dtsx.astra.sdk.db.domain.CloudProviderType;
+import com.fasterxml.jackson.annotation.JsonProperty;
 import io.stargate.sdk.core.domain.ObjectMap;
+import io.stargate.sdk.json.domain.JsonDocument;
+import io.stargate.sdk.json.domain.odm.Document;
+import io.stargate.sdk.json.vector.JsonVectorStore;
 import io.stargate.sdk.json.vector.VectorStore;
+import lombok.AllArgsConstructor;
+import lombok.Data;
+import lombok.NoArgsConstructor;
 
+import java.util.Map;
 import java.util.UUID;
 
 public class AstraVectorQuickStart {
@@ -12,14 +20,66 @@ public void quickStartTest() {
         String databaseName = "vector_client_test";
         String astraToken = System.getenv("ASTRA_DB_APPLICATION_TOKEN");
 
-        UUID dbId = new AstraVectorClient(astraToken)
-                .createDatabase(databaseName, CloudProviderType.GCP, "us-east1");
+        AstraVectorClient vectorClient = new AstraVectorClient(astraToken);
+        if (!vectorClient.isDatabaseExists(databaseName)) {
+            vectorClient.createDatabase(databaseName);
+        }
 
-        AstraVectorDatabaseClient vectorDb = new AstraVectorClient(astraToken)
-                .database(databaseName);
+        // Without ODM Accessing the Vector DB with JSON-ISH
+        AstraVectorDatabaseClient vectorDb = vectorClient.database(databaseName);
+        JsonVectorStore jsonVectorStore =
+                vectorDb.createVectorStore("demo_product", 14);
 
-       DefaultVectorStore vectorStore = vectorDb.createVectorStore("demo_product", 14);
+        // ======== INSERTIONS =========
 
+        jsonVectorStore.insert(new JsonDocument("doc1")
+                .put("product_name", "HealthyFresh - Beef raw dog food")
+                .put("product_price", 12.99)
+                .vector(new float[]{1f, 0f, 1f, 1f, 1f, 1f, 0f, 0f, 0f, 0f, 0f, 0f, 0f, 0f}));
 
+        jsonVectorStore.insert(new JsonDocument("doc2")
+                .data(Map.of("product_name", "HealthyFresh - Chicken raw dog food"))
+                .vector(new float[]{1f, 1f, 1f, 1f, 1f, 0f, 0f, 0f, 0f, 0f, 0f, 0f, 0f, 0f}));
+
+        jsonVectorStore.insert(new JsonDocument("doc3")
+                .data("{"
+                        +"   \"product_name\": \"HealthyFresh - Chicken raw dog food\", "
+                        + "   \"product_price\": 9.99, "
+                        + "}")
+                .vector(new float[]{1f, 1f, 1f, 1f, 1f, 0f, 0f, 0f, 0f, 0f, 0f, 0f, 0f, 0f}));
+
+        //jsonVectorStore.insert("{"
+        //           + "   \"_id\":\"doc4\","
+        //           + "   \"$vector\":[1f, 1f, 1f, 1f, 1f, 0f, 0f, 0f, 0f, 0f, 0f, 0f, 0f, 0f],"
+        //           + "   \"product_name\": \"HealthyFresh - Chicken raw dog food\", "
+        //           + "   \"product_price\": 9.99, "
+        //           + "}");
+
+        // With ODM
+        VectorStore<Product> productVectorStore =
+                vectorDb.createVectorStore("demo_product", 14, Product.class);
+
+        // 3 fields: id, payload, vector
+        productVectorStore.insert("doc5",
+                new Product("HealthyFresh - Beef raw dog food", 12.99),
+                new float[]{1f, 1f, 1f, 1f, 1f, 0f, 0f, 0f, 0f, 0f, 0f, 0f, 0f, 0f});
+
+        // build the "document" and insert the document
+        Document<Product> doc6 = new Document<>("doc6",
+                new Product("HealthyFresh - Beef raw dog food", 12.99),
+                new float[]{1f, 1f, 1f, 1f, 1f, 0f, 0f, 0f, 0f, 0f, 0f, 0f, 0f, 0f});
+        productVectorStore.insert(doc6);
+
+    }
+
+
+    @Data
+    @NoArgsConstructor
+    @AllArgsConstructor
+    static class Product {
+        @JsonProperty("product_name")
+        private String name;
+        @JsonProperty("product_price")
+        private Double price;
     }
 }