From 16b11bda83c7f95a2692abeea72cf8aee89c1d83 Mon Sep 17 00:00:00 2001 From: Zhicheng Geng Date: Wed, 28 Jun 2023 20:06:10 -0500 Subject: [PATCH] Use `getLogger` instead of `basicConfig` in `multi_query.py` (#6891) Remove `logging.basicConfig`, which turns on logging. Use `getLogger` instead --- langchain/retrievers/multi_query.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/langchain/retrievers/multi_query.py b/langchain/retrievers/multi_query.py index 78a2624b673..925a7d0a9a7 100644 --- a/langchain/retrievers/multi_query.py +++ b/langchain/retrievers/multi_query.py @@ -9,7 +9,7 @@ from langchain.output_parsers.pydantic import PydanticOutputParser from langchain.prompts.prompt import PromptTemplate from langchain.schema import BaseRetriever, Document -logging.basicConfig(level=logging.INFO) +logger = logging.getLogger(__name__) class LineList(BaseModel): @@ -120,7 +120,7 @@ class MultiQueryRetriever(BaseRetriever): response = self.llm_chain({"question": question}) lines = getattr(response["text"], self.parser_key, []) if self.verbose: - logging.info(f"Generated queries: {lines}") + logger.info(f"Generated queries: {lines}") return lines def retrieve_documents(self, queries: List[str]) -> List[Document]: