add run name for query constructor (langchain-ai#18101)

Co-authored-by: Eugene Yurtsev <eyurtsev@gmail.com>
FalkorDB · Mar 30, 2024 · e5d2f1d · e5d2f1d
1 parent de27200
commit e5d2f1d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/libs/langchain/langchain/retrievers/self_query/base.py b/libs/langchain/langchain/retrievers/self_query/base.py
@@ -54,6 +54,7 @@
 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 @@ def from_llm(
             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,