github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-21 14:18:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added filter argument to pinecone queries, fixes #600 (#601)

Browse Source 
Added filter argument to similarity_search() and
similarity_search_with_score()

Co-authored-by: Sam Cartford (MBP) <cartford@hey.com>
This commit is contained in:
babbldev 2023-01-12 23:15:51 -06:00 committed by GitHub
parent c4c6bf6e6e
commit b5eb91536a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
langchain/vectorstores/pinecone.py
View File

@ -80,6 +80,7 @@ class Pinecone(VectorStore):
self,
query: str,
k: int = 5,
filter: Optional[dict] = None,
namespace: Optional[str] = None,
) -> List[Tuple[Document, float]]:
"""Return pinecone documents most similar to query, along with scores.
@ -87,6 +88,7 @@ class Pinecone(VectorStore):
Args:
query: Text to look up documents similar to.
k: Number of Documents to return. Defaults to 4.
filter: Dictionary of argument(s) to filter on metadata
namespace: Namespace to search in. Default will search in '' namespace.
Returns:
@ -95,7 +97,11 @@ class Pinecone(VectorStore):
query_obj = self._embedding_function(query)
docs = []
results = self._index.query(
[query_obj], top_k=k, include_metadata=True, namespace=namespace
[query_obj],
top_k=k,
include_metadata=True,
namespace=namespace,
filter=filter,
)
for res in results["matches"]:
metadata = res["metadata"]
@ -107,6 +113,7 @@ class Pinecone(VectorStore):
self,
query: str,
k: int = 5,
filter: Optional[dict] = None,
namespace: Optional[str] = None,
) -> List[Document]:
"""Return pinecone documents most similar to query.
@ -114,6 +121,7 @@ class Pinecone(VectorStore):
Args:
query: Text to look up documents similar to.
k: Number of Documents to return. Defaults to 4.
filter: Dictionary of argument(s) to filter on metadata
namespace: Namespace to search in. Default will search in '' namespace.
Returns:
@ -122,7 +130,11 @@ class Pinecone(VectorStore):
query_obj = self._embedding_function(query)
docs = []
results = self._index.query(
[query_obj], top_k=k, include_metadata=True, namespace=namespace
[query_obj],
top_k=k,
include_metadata=True,
namespace=namespace,
filter=filter,
)
for res in results["matches"]:
metadata = res["metadata"]