github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-26 00:23:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

community[minor] Pebblo: Support for new Pinecone class PineconeVectorStore (#28253)

Browse Source 
- **Description:** Support for new Pinecone class PineconeVectorStore in
PebbloRetrievalQA.
- **Issue:** NA
- **Dependencies:** NA
- **Tests:** -

Co-authored-by: Erick Friis <erick@langchain.dev>
This commit is contained in:
Rajendra Kadam 2024-12-10 01:03:54 +05:30 committed by GitHub
parent 85114b4f3a
commit 242fee11be
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
libs/community/langchain_community/chains/pebblo_retrieval/enforcement_filters.py
View File

@ -27,8 +27,9 @@ logger = logging.getLogger(__name__)
PINECONE = "Pinecone" PINECONE = "Pinecone"
QDRANT = "Qdrant" QDRANT = "Qdrant"
PGVECTOR = "PGVector" PGVECTOR = "PGVector"
PINECONE_VECTOR_STORE = "PineconeVectorStore"
SUPPORTED_VECTORSTORES = {PINECONE, QDRANT, PGVECTOR} SUPPORTED_VECTORSTORES = {PINECONE, QDRANT, PGVECTOR, PINECONE_VECTOR_STORE}
def clear_enforcement_filters(retriever: VectorStoreRetriever) -> None: def clear_enforcement_filters(retriever: VectorStoreRetriever) -> None:
@ -505,7 +506,7 @@ def _set_identity_enforcement_filter(
of the retriever based on the type of the vectorstore. of the retriever based on the type of the vectorstore.
""" """
search_kwargs = retriever.search_kwargs search_kwargs = retriever.search_kwargs
if retriever.vectorstore.__class__.__name__ == PINECONE: if retriever.vectorstore.__class__.__name__ in [PINECONE, PINECONE_VECTOR_STORE]:
_apply_pinecone_authorization_filter(search_kwargs, auth_context) _apply_pinecone_authorization_filter(search_kwargs, auth_context)
elif retriever.vectorstore.__class__.__name__ == QDRANT: elif retriever.vectorstore.__class__.__name__ == QDRANT:
_apply_qdrant_authorization_filter(search_kwargs, auth_context) _apply_qdrant_authorization_filter(search_kwargs, auth_context)