[NOID] fix compile errors

neo4j-contrib · Jan 16, 2025 · 9d0e9b1 · 9d0e9b1
1 parent bb95ae2
commit 9d0e9b1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/full/src/main/java/apoc/vectordb/Pinecone.java b/full/src/main/java/apoc/vectordb/Pinecone.java
@@ -184,7 +184,7 @@ private Stream<VectorDbUtil.EmbeddingResult> getCommon(
         String url = "%s/vectors/fetch";
         Map<String, Object> config = getVectorDbInfo(hostOrKey, index, configuration, url);
 
-        VectorEmbeddingConfig conf = DB_HANDLER.getEmbedding().fromGet(config, procedureCallContext, ids);
+        VectorEmbeddingConfig conf = DB_HANDLER.getEmbedding().fromGet(config, procedureCallContext, ids, index);
 
         return getEmbeddingResultStream(conf, procedureCallContext, tx, v -> {
             Object vectors = ((Map) v).get("vectors");

diff --git a/full/src/main/java/apoc/vectordb/PineconeHandler.java b/full/src/main/java/apoc/vectordb/PineconeHandler.java
@@ -69,7 +69,7 @@ static class PineconeEmbeddingHandler implements VectorEmbeddingHandler {
          */
         @Override
         public <T> VectorEmbeddingConfig fromGet(
-                Map<String, Object> config, ProcedureCallContext procedureCallContext, List<T> ids) {
+                Map<String, Object> config, ProcedureCallContext procedureCallContext, List<T> ids, String collection) {
             List<String> fields = procedureCallContext.outputFields().collect(Collectors.toList());
 
             config.put(BODY_KEY, null);
@@ -118,7 +118,7 @@ private VectorEmbeddingConfig getVectorEmbeddingConfig(
             headers.remove(METHOD_KEY);
             apiConfig.setHeaders(headers);
 
-            return VectorEmbeddingHandler.populateApiBodyRequest(conf, additionalBodies);
+            return populateApiBodyRequest(conf, additionalBodies);
         }
     }
 }