diff --git a/libs/langchain/langchain/chains/openai_functions/qa_with_structure.py b/libs/langchain/langchain/chains/openai_functions/qa_with_structure.py index dff17030a..770b3e9d7 100644 --- a/libs/langchain/langchain/chains/openai_functions/qa_with_structure.py +++ b/libs/langchain/langchain/chains/openai_functions/qa_with_structure.py @@ -28,6 +28,7 @@ def create_qa_with_structure_chain( schema: Union[dict, Type[BaseModel]], output_parser: str = "base", prompt: Optional[Union[PromptTemplate, ChatPromptTemplate]] = None, + verbose: bool = False, ) -> LLMChain: """Create a question answering chain that returns an answer with sources based on schema. @@ -87,18 +88,24 @@ def create_qa_with_structure_chain( prompt=prompt, llm_kwargs=llm_kwargs, output_parser=_output_parser, + verbose=verbose, ) return chain -def create_qa_with_sources_chain(llm: BaseLanguageModel, **kwargs: Any) -> LLMChain: +def create_qa_with_sources_chain( + llm: BaseLanguageModel, verbose: bool = False, **kwargs: Any +) -> LLMChain: """Create a question answering chain that returns an answer with sources. Args: llm: Language model to use for the chain. + verbose: Whether to print the details of the chain **kwargs: Keyword arguments to pass to `create_qa_with_structure_chain`. Returns: Chain (LLMChain) that can be used to answer questions with citations. """ - return create_qa_with_structure_chain(llm, AnswerWithSources, **kwargs) + return create_qa_with_structure_chain( + llm, AnswerWithSources, verbose=verbose, **kwargs + )