From 4e7d0c115be32a82b2157a9ef491b957b16d250e Mon Sep 17 00:00:00 2001 From: mrkhalil6 <59417186+mrkhalil6@users.noreply.github.com> Date: Sat, 8 Jul 2023 00:03:59 +0500 Subject: [PATCH] Add support for filters and namespaces in similarity search in Pinecone similarity_score_threshold (#7301) At the moment, pinecone vectorStore does not support filters and namespaces when using similarity_score_threshold search type. In this PR, I've implemented that. It passes all the kwargs except "score_threshold" as that is not a supported argument for method "similarity_search_with_score". --------- Co-authored-by: Bagatur --- langchain/vectorstores/pinecone.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/langchain/vectorstores/pinecone.py b/langchain/vectorstores/pinecone.py index 552fa2ef1e..a577e9af4d 100644 --- a/langchain/vectorstores/pinecone.py +++ b/langchain/vectorstores/pinecone.py @@ -166,7 +166,8 @@ class Pinecone(VectorStore): k: int = 4, **kwargs: Any, ) -> List[Tuple[Document, float]]: - return self.similarity_search_with_score(query, k) + kwargs.pop("score_threshold", None) + return self.similarity_search_with_score(query, k, **kwargs) def max_marginal_relevance_search_by_vector( self,