From 935aefa8dba18edc0453e677925c17dc71444e56 Mon Sep 17 00:00:00 2001
From: Harrison Chase <hw.chase.17@gmail.com>
Date: Mon, 26 Feb 2024 08:17:05 -0800
Subject: [PATCH] add run name for query constructor (#18101)

Co-authored-by: Eugene Yurtsev <eyurtsev@gmail.com>
---
 libs/langchain/langchain/retrievers/self_query/base.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/libs/langchain/langchain/retrievers/self_query/base.py b/libs/langchain/langchain/retrievers/self_query/base.py
index 94a2fb96104..e5819a64a1a 100644
--- 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.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,