github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-05-30 11:39:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

langchain[patch]: langchain-pinecone self query support (#20702)

Browse Source
This commit is contained in:
Bagatur 2024-04-21 15:42:39 -07:00 committed by GitHub
parent 5ae738c4fe
commit d0cee65cdc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 18 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
libs/langchain/langchain/retrievers/self_query/base.py
View File

@ -15,7 +15,6 @@ from langchain_community.vectorstores import (
MyScale, MyScale,
OpenSearchVectorSearch, OpenSearchVectorSearch,
PGVector, PGVector,
Pinecone,
Qdrant, Qdrant,
Redis, Redis,
SupabaseVectorStore, SupabaseVectorStore,
@ -27,6 +26,9 @@ from langchain_community.vectorstores import (
from langchain_community.vectorstores import ( from langchain_community.vectorstores import (
ElasticsearchStore as ElasticsearchStoreCommunity, ElasticsearchStore as ElasticsearchStoreCommunity,
) )
from langchain_community.vectorstores import (
Pinecone as CommunityPinecone,
)
from langchain_core.callbacks.manager import ( from langchain_core.callbacks.manager import (
AsyncCallbackManagerForRetrieverRun, AsyncCallbackManagerForRetrieverRun,
CallbackManagerForRetrieverRun, CallbackManagerForRetrieverRun,
@ -73,7 +75,7 @@ def _get_builtin_translator(vectorstore: VectorStore) -> Visitor:
BUILTIN_TRANSLATORS: Dict[Type[VectorStore], Type[Visitor]] = { BUILTIN_TRANSLATORS: Dict[Type[VectorStore], Type[Visitor]] = {
AstraDB: AstraDBTranslator, AstraDB: AstraDBTranslator,
PGVector: PGVectorTranslator, PGVector: PGVectorTranslator,
Pinecone: PineconeTranslator, CommunityPinecone: PineconeTranslator,
Chroma: ChromaTranslator, Chroma: ChromaTranslator,
DashVector: DashvectorTranslator, DashVector: DashvectorTranslator,
Dingo: DingoDBTranslator, Dingo: DingoDBTranslator,
@ -107,19 +109,27 @@ def _get_builtin_translator(vectorstore: VectorStore) -> Visitor:
else: else:
try: try:
from langchain_astradb.vectorstores import AstraDBVectorStore from langchain_astradb.vectorstores import AstraDBVectorStore
if isinstance(vectorstore, AstraDBVectorStore):
return AstraDBTranslator()
except ImportError: except ImportError:
pass pass
else:
if isinstance(vectorstore, AstraDBVectorStore):
return AstraDBTranslator()
try: try:
from langchain_elasticsearch.vectorstores import ElasticsearchStore from langchain_elasticsearch.vectorstores import ElasticsearchStore
if isinstance(vectorstore, ElasticsearchStore):
return ElasticsearchTranslator()
except ImportError: except ImportError:
pass pass
else:
if isinstance(vectorstore, ElasticsearchStore):
return ElasticsearchTranslator()
try:
from langchain_pinecone import Pinecone
except ImportError:
pass
else:
if isinstance(vectorstore, Pinecone):
return PineconeTranslator()
raise ValueError( raise ValueError(
f"Self query retriever with Vector Store type {vectorstore.__class__}" f"Self query retriever with Vector Store type {vectorstore.__class__}"