langchain: fixed weaviate (v4) vectorstore import for self-query retriever (#28675)

Co-authored-by: Xin Heng <xin.heng@gmail.com>
This commit is contained in:
xintoteai 2024-12-11 15:53:41 -08:00 committed by GitHub
parent ee640d6bd3
commit 45f9c9ae88
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -170,6 +170,7 @@ def _get_builtin_translator(vectorstore: VectorStore) -> Visitor:
return Neo4jTranslator()
try:
# Trying langchain_chroma import if exists
from langchain_chroma import Chroma
except ImportError:
pass
@ -204,6 +205,16 @@ def _get_builtin_translator(vectorstore: VectorStore) -> Visitor:
if isinstance(vectorstore, HanaDB):
return HanaTranslator()
try:
# Trying langchain_weaviate (weaviate v4) import if exists
from langchain_weaviate.vectorstores import WeaviateVectorStore
except ImportError:
pass
else:
if isinstance(vectorstore, WeaviateVectorStore):
return WeaviateTranslator()
raise ValueError(
f"Self query retriever with Vector Store type {vectorstore.__class__}"
f" not supported."