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

add run name for query constructor (#18101)

Browse Source 
Co-authored-by: Eugene Yurtsev <eyurtsev@gmail.com>
This commit is contained in:
Harrison Chase 2024-02-26 08:17:05 -08:00 committed by GitHub
parent 719a1cde75
commit 935aefa8db
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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