diff --git a/full/src/main/java/apoc/vectordb/Pinecone.java b/full/src/main/java/apoc/vectordb/Pinecone.java index 3aca834ace..6baba680e8 100644 --- a/full/src/main/java/apoc/vectordb/Pinecone.java +++ b/full/src/main/java/apoc/vectordb/Pinecone.java @@ -184,7 +184,7 @@ private Stream getCommon( String url = "%s/vectors/fetch"; Map 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 index 7f1b2df673..bbba77e5c7 100644 --- 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 VectorEmbeddingConfig fromGet( - Map config, ProcedureCallContext procedureCallContext, List ids) { + Map config, ProcedureCallContext procedureCallContext, List ids, String collection) { List 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); } } }