From ec3e008c9fbdfdf8d893558ee4f92e44f3b4853e Mon Sep 17 00:00:00 2001
From: 0007 <0007@qq.com>
Date: Wed, 27 Aug 2025 19:57:55 +0800
Subject: [PATCH] Add File
---
.../opensearch/OpenSearchVectorStoreTest.java | 97 +++++++++++++++++++
1 file changed, 97 insertions(+)
create mode 100644 agents-flex-store/agents-flex-store-opensearch/src/test/java/com/agentsflex/store/opensearch/OpenSearchVectorStoreTest.java
diff --git a/agents-flex-store/agents-flex-store-opensearch/src/test/java/com/agentsflex/store/opensearch/OpenSearchVectorStoreTest.java b/agents-flex-store/agents-flex-store-opensearch/src/test/java/com/agentsflex/store/opensearch/OpenSearchVectorStoreTest.java
new file mode 100644
index 0000000..37d9ff0
--- /dev/null
+++ b/agents-flex-store/agents-flex-store-opensearch/src/test/java/com/agentsflex/store/opensearch/OpenSearchVectorStoreTest.java
@@ -0,0 +1,97 @@
+/*
+ * Copyright (c) 2023-2025, Agents-Flex (fuhai999@gmail.com).
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.agentsflex.store.opensearch;
+
+import com.agentsflex.core.document.Document;
+import com.agentsflex.core.llm.embedding.EmbeddingModel;
+import com.agentsflex.core.llm.embedding.EmbeddingOptions;
+import com.agentsflex.core.store.SearchWrapper;
+import com.agentsflex.core.store.StoreOptions;
+import com.agentsflex.core.store.VectorData;
+import com.agentsflex.core.store.exception.StoreException;
+import org.junit.Test;
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
+
+/**
+ * @author songyinyin
+ * @since 2024/8/11 下午3:21
+ */
+public class OpenSearchVectorStoreTest {
+
+ @Test(expected = StoreException.class)
+ public void test01() {
+ OpenSearchVectorStore store = getOpenSearchVectorStore();
+
+ // https://opensearch.org/docs/latest/search-plugins/vector-search/#example
+ List list = new ArrayList<>();
+ Document doc1 = new Document();
+ doc1.setId(1);
+ doc1.setContent("test1");
+ doc1.setVector(new double[]{5.2, 4.4});
+ list.add(doc1);
+ Document doc2 = new Document();
+ doc2.setId(2);
+ 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);
+ Document doc4 = new Document();
+ doc4.setId(4);
+ doc4.setContent("test4");
+ doc4.setVector(new double[]{4.2, 4.6});
+ list.add(doc4);
+ Document doc5 = new Document();
+ doc5.setId(5);
+ doc5.setContent("test5");
+ doc5.setVector(new double[]{3.3, 4.5});
+ list.add(doc5);
+ store.storeInternal(list, StoreOptions.DEFAULT);
+
+ // 可能要等一会 才能查出结果
+ SearchWrapper searchWrapper = new SearchWrapper();
+ searchWrapper.setVector(new double[]{5, 4});
+ searchWrapper.setMaxResults(3);
+ List documents = store.searchInternal(searchWrapper, StoreOptions.DEFAULT);
+ for (Document document : documents) {
+ System.out.printf("id=%s, content=%s, vector=%s, metadata=%s\n",
+ document.getId(), document.getContent(), Arrays.toString(document.getVector()), document.getMetadataMap());
+ }
+
+ }
+
+ private static OpenSearchVectorStore getOpenSearchVectorStore() {
+ OpenSearchVectorStoreConfig config = new OpenSearchVectorStoreConfig();
+ config.setUsername("admin");
+ config.setPassword("4_Pa46WQczS?");
+ OpenSearchVectorStore store = new OpenSearchVectorStore(config);
+ store.setEmbeddingModel(new EmbeddingModel() {
+ @Override
+ public VectorData embed(Document document, EmbeddingOptions options) {
+ VectorData vectorData = new VectorData();
+ vectorData.setVector(new double[]{0, 0});
+ return vectorData;
+ }
+ });
+ return store;
+ }
+}