github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-27 08:58:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use getLogger instead of basicConfig in multi_query.py (#6891)

Browse Source 
Remove `logging.basicConfig`, which turns on logging. Use `getLogger`
instead
This commit is contained in:
Zhicheng Geng 2023-06-28 20:06:10 -05:00 committed by GitHub
parent f07dd02b50
commit 16b11bda83
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
langchain/retrievers/multi_query.py
View File

@ -9,7 +9,7 @@ from langchain.output_parsers.pydantic import PydanticOutputParser
from langchain.prompts.prompt import PromptTemplate from langchain.prompts.prompt import PromptTemplate
from langchain.schema import BaseRetriever, Document from langchain.schema import BaseRetriever, Document
logging.basicConfig(level=logging.INFO) logger = logging.getLogger(__name__)
class LineList(BaseModel): class LineList(BaseModel):
@ -120,7 +120,7 @@ class MultiQueryRetriever(BaseRetriever):
response = self.llm_chain({"question": question}) response = self.llm_chain({"question": question})
lines = getattr(response["text"], self.parser_key, []) lines = getattr(response["text"], self.parser_key, [])
if self.verbose: if self.verbose:
logging.info(f"Generated queries: {lines}") logger.info(f"Generated queries: {lines}")
return lines return lines
def retrieve_documents(self, queries: List[str]) -> List[Document]: def retrieve_documents(self, queries: List[str]) -> List[Document]: