Add Pinecone namespace support for add, search and delete operations

Author: adambchouti

vector-stores/spring-ai-pinecone/src/main/java/org/springframework/ai/vectorstore/PineconeVectorStore.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2023-2023 the original author or authors.
+ * Copyright 2023-2024 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -47,6 +47,7 @@
  * Pinecone index.
  *
  * @author Christian Tzolov
+ * @author Adam Bchouti
  */
 public class PineconeVectorStore implements VectorStore {
 
@@ -60,18 +61,9 @@ public class PineconeVectorStore implements VectorStore {
 
 	private final PineconeConnection pineconeConnection;
 
-	private final ObjectMapper objectMapper;
-
-	private String pineconeNamespace;
+	private final String pineconeNamespace;
 
-	/**
-	 * Change the Index Name.
-	 * @param pineconeNamespace The Azure VectorStore index name to use.
-	 */
-	public void setPineconeNamespace(String pineconeNamespace) {
-		Assert.hasText(pineconeNamespace, "The Pinecone namespace can not be empty.");
-		this.pineconeNamespace = pineconeNamespace;
-	}
+	private final ObjectMapper objectMapper;
 
 	/**
 	 * Configuration class for the PineconeVectorStore.
@@ -232,11 +224,11 @@ public PineconeVectorStore(PineconeVectorStoreConfig config, EmbeddingClient emb
 	}
 
 	/**
-	 * Adds a list of documents to the vector store.
+	 * Adds a list of documents to the vector store based on the namespace.
 	 * @param documents The list of documents to be added.
+	 * @param namespace The namespace to add the documents to
 	 */
-	@Override
-	public void add(List<Document> documents) {
+	public void add(List<Document> documents, String namespace) {
 
 		List<Vector> upsertVectors = documents.stream().map(document -> {
 			// Compute and assign an embedding to the document.
@@ -251,12 +243,21 @@ public void add(List<Document> documents) {
 
 		UpsertRequest upsertRequest = UpsertRequest.newBuilder()
 			.addAllVectors(upsertVectors)
-			.setNamespace(this.pineconeNamespace)
+			.setNamespace(namespace)
 			.build();
 
 		this.pineconeConnection.getBlockingStub().upsert(upsertRequest);
 	}
 
+	/**
+	 * Adds a list of documents to the vector store.
+	 * @param documents The list of documents to be added.
+	 */
+	@Override
+	public void add(List<Document> documents) {
+		add(documents, this.pineconeNamespace);
+	}
+
 	/**
 	 * Converts the document metadata to a Protobuf Struct.
 	 * @param document The document containing metadata.
@@ -286,15 +287,15 @@ private Value contentValue(Document document) {
 	}
 
 	/**
-	 * Deletes a list of documents by their IDs.
+	 * Deletes a list of documents by their IDs based on the namespace.
 	 * @param documentIds The list of document IDs to be deleted.
+	 * @param namespace The namespace of the document IDs.
 	 * @return An optional boolean indicating the deletion status.
 	 */
-	@Override
-	public Optional<Boolean> delete(List<String> documentIds) {
+	public Optional<Boolean> delete(List<String> documentIds, String namespace) {
 
 		DeleteRequest deleteRequest = DeleteRequest.newBuilder()
-			.setNamespace(this.pineconeNamespace) // ignored for free tier.
+			.setNamespace(namespace) // ignored for free tier.
 			.addAllIds(documentIds)
 			.setDeleteAll(false)
 			.build();
@@ -305,8 +306,17 @@ public Optional<Boolean> delete(List<String> documentIds) {
 		return Optional.of(true);
 	}
 
+	/**
+	 * Deletes a list of documents by their IDs.
+	 * @param documentIds The list of document IDs to be deleted.
+	 * @return An optional boolean indicating the deletion status.
+	 */
 	@Override
-	public List<Document> similaritySearch(SearchRequest request) {
+	public Optional<Boolean> delete(List<String> documentIds) {
+		return delete(documentIds, this.pineconeNamespace);
+	}
+
+	public List<Document> similaritySearch(SearchRequest request, String namespace) {
 
 		String nativeExpressionFilters = (request.getFilterExpression() != null)
 				? this.filterExpressionConverter.convertExpression(request.getFilterExpression()) : "";
@@ -317,7 +327,7 @@ public List<Document> similaritySearch(SearchRequest request) {
 			.addAllVector(toFloatList(queryEmbedding))
 			.setTopK(request.getTopK())
 			.setIncludeMetadata(true)
-			.setNamespace(this.pineconeNamespace);
+			.setNamespace(namespace);
 
 		if (StringUtils.hasText(nativeExpressionFilters)) {
 			queryRequestBuilder.setFilter(metadataFiltersToStruct(nativeExpressionFilters));
@@ -339,6 +349,11 @@ public List<Document> similaritySearch(SearchRequest request) {
 			.toList();
 	}
 
+	@Override
+	public List<Document> similaritySearch(SearchRequest request) {
+		return similaritySearch(request, this.pineconeNamespace);
+	}
+
 	private Struct metadataFiltersToStruct(String metadataFilters) {
 		try {
 			var structBuilder = Struct.newBuilder();