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

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

Browse Source 
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
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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