add run name for query constructor (#18101)

Co-authored-by: Eugene Yurtsev <eyurtsev@gmail.com>
pull/18049/head^2
Harrison Chase 4 months ago committed by GitHub
parent 719a1cde75
commit 935aefa8db
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -54,6 +54,7 @@ from langchain.retrievers.self_query.vectara import VectaraTranslator
from langchain.retrievers.self_query.weaviate import WeaviateTranslator
logger = logging.getLogger(__name__)
QUERY_CONSTRUCTOR_RUN_NAME = "query_constructor"
def _get_builtin_translator(vectorstore: VectorStore) -> Visitor:
@ -244,6 +245,9 @@ class SelfQueryRetriever(BaseRetriever):
enable_limit=enable_limit,
**chain_kwargs,
)
query_constructor = query_constructor.with_config(
run_name=QUERY_CONSTRUCTOR_RUN_NAME
)
return cls(
query_constructor=query_constructor,
vectorstore=vectorstore,

Loading…
Cancel
Save