Fixes #4080: Add Pinecone and Milvus support

neo4j-contrib · May 29, 2024 · e9a3a02 · e9a3a02
1 parent bc76a3d
commit e9a3a02
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/extended/src/main/java/apoc/vectordb/VectorEmbeddingHandler.java b/extended/src/main/java/apoc/vectordb/VectorEmbeddingHandler.java
@@ -1,11 +1,23 @@
 package apoc.vectordb;
 
 import apoc.ml.RestAPIConfig;
+import org.apache.commons.collections4.MapUtils;
+import org.apache.commons.lang3.StringUtils;
 import org.neo4j.internal.kernel.api.procs.ProcedureCallContext;
 
+import java.net.URL;
+import java.util.ArrayList;
+import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
-
+import java.util.stream.Collectors;
+
+import static apoc.ml.RestAPIConfig.BODY_KEY;
+import static apoc.ml.RestAPIConfig.ENDPOINT_KEY;
+import static apoc.ml.RestAPIConfig.HEADERS_KEY;
+import static apoc.ml.RestAPIConfig.JSON_PATH_KEY;
+import static apoc.ml.RestAPIConfig.METHOD_KEY;
+import static apoc.util.MapUtil.map;
 import static apoc.vectordb.VectorEmbeddingConfig.*;
 
 public interface VectorEmbeddingHandler {