Add File

agents-flex-store/agents-flex-store-qdrant/src/test/java/com/agentsflex/store/qdrant/QdrantVectorStoreTest.java

@@ -0,0 +1,65 @@
+package com.agentsflex.store.qdrant;
+
+import com.agentsflex.core.document.Document;
+import com.agentsflex.core.store.SearchWrapper;
+import com.agentsflex.core.store.StoreOptions;
+import org.junit.Test;
+
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+
+public class QdrantVectorStoreTest {
+
+    @Test
+    public void testSaveVectors() throws Exception {
+        QdrantVectorStore db = getDb();
+        StoreOptions options = new StoreOptions();
+        options.setCollectionName("test_collection1");
+        List<Document> list = new ArrayList<>();
+        Document doc1 = new Document();
+        doc1.setId(1L);
+        doc1.setContent("test1");
+        doc1.setVector(new double[]{5.2, 4.4});
+        list.add(doc1);
+        Document doc2 = new Document();
+        doc2.setId(2L);
+        doc2.setContent("test2");
+        doc2.setVector(new double[]{5.2, 3.9});
+        list.add(doc2);
+        Document doc3 = new Document();
+        doc3.setId(3);
+        doc3.setContent("test3");
+        doc3.setVector(new double[]{4.9, 3.4});
+        list.add(doc3);
+        db.store(list, options);
+    }
+
+    @Test
+    public void testQuery() throws Exception {
+        QdrantVectorStore db = getDb();;
+        StoreOptions options = new StoreOptions();
+        options.setCollectionName("test_collection1");
+        SearchWrapper search = new SearchWrapper();
+        search.setVector(new double[]{5.2, 3.9});
+        //search.setText("test1");
+        search.setMaxResults(1);
+        List<Document> record = db.search(search, options);
+        System.out.println(record);
+    }
+
+    @Test
+    public void testDelete() throws Exception {
+        QdrantVectorStore db = getDb();
+        StoreOptions options = new StoreOptions();
+        options.setCollectionName("test_collection1");
+        db.delete(Collections.singletonList(3L), options);
+    }
+
+    private QdrantVectorStore getDb() throws Exception {
+        QdrantVectorStoreConfig config = new QdrantVectorStoreConfig();
+        config.setUri("localhost");
+        config.setDefaultCollectionName("test_collection1");
+        return new QdrantVectorStore(config);
+    }
+}