diff --git a/libs/langchain/langchain/retrievers/self_query/base.py b/libs/langchain/langchain/retrievers/self_query/base.py index f631f4e547..b3cb8fc4c3 100644 --- a/libs/langchain/langchain/retrievers/self_query/base.py +++ b/libs/langchain/langchain/retrievers/self_query/base.py @@ -170,11 +170,12 @@ def _get_builtin_translator(vectorstore: VectorStore) -> Visitor: try: from langchain_postgres import PGVector + from langchain_postgres import PGVectorTranslator as NewPGVectorTranslator except ImportError: pass else: if isinstance(vectorstore, PGVector): - return PGVectorTranslator() + return NewPGVectorTranslator() raise ValueError( f"Self query retriever with Vector Store type {vectorstore.__class__}"