From 7084d69ea7e1f120f1728aa4cbf901daf69f1084 Mon Sep 17 00:00:00 2001 From: Harrison Chase Date: Mon, 24 Apr 2023 22:16:07 -0700 Subject: [PATCH] Harrison/verbose conv ret (#3492) Co-authored-by: makretch --- langchain/chains/conversational_retrieval/base.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/langchain/chains/conversational_retrieval/base.py b/langchain/chains/conversational_retrieval/base.py index 900d8e7c..8bd948cf 100644 --- a/langchain/chains/conversational_retrieval/base.py +++ b/langchain/chains/conversational_retrieval/base.py @@ -173,6 +173,7 @@ class ConversationalRetrievalChain(BaseConversationalRetrievalChain): retriever: BaseRetriever, condense_question_prompt: BasePromptTemplate = CONDENSE_QUESTION_PROMPT, chain_type: str = "stuff", + verbose: bool = False, combine_docs_chain_kwargs: Optional[Dict] = None, **kwargs: Any, ) -> BaseConversationalRetrievalChain: @@ -181,9 +182,12 @@ class ConversationalRetrievalChain(BaseConversationalRetrievalChain): doc_chain = load_qa_chain( llm, chain_type=chain_type, + verbose=verbose, **combine_docs_chain_kwargs, ) - condense_question_chain = LLMChain(llm=llm, prompt=condense_question_prompt) + condense_question_chain = LLMChain( + llm=llm, prompt=condense_question_prompt, verbose=verbose + ) return cls( retriever=retriever, combine_docs_chain=doc_chain,