github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-11 05:45:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

feature: add verbosity to create_qa_with_sources_chain (#9742)

Browse Source 
Adds a verbose parameter to the create_qa_with_sources_chain and
create_qa_with_structure_chain functions
This commit is contained in:
Sajal Sharma 2023-09-04 11:42:20 +08:00 committed by GitHub
parent 68f2363f5d
commit 0b6993987f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
libs/langchain/langchain/chains/openai_functions/qa_with_structure.py
View File

@ -28,6 +28,7 @@ def create_qa_with_structure_chain(
schema: Union[dict, Type[BaseModel]], schema: Union[dict, Type[BaseModel]],
output_parser: str = "base", output_parser: str = "base",
prompt: Optional[Union[PromptTemplate, ChatPromptTemplate]] = None, prompt: Optional[Union[PromptTemplate, ChatPromptTemplate]] = None,
verbose: bool = False,
) -> LLMChain: ) -> LLMChain:
"""Create a question answering chain that returns an answer with sources """Create a question answering chain that returns an answer with sources
based on schema. based on schema.
@ -87,18 +88,24 @@ def create_qa_with_structure_chain(
prompt=prompt, prompt=prompt,
llm_kwargs=llm_kwargs, llm_kwargs=llm_kwargs,
output_parser=_output_parser, output_parser=_output_parser,
verbose=verbose,
) )
return chain 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. """Create a question answering chain that returns an answer with sources.
Args: Args:
llm: Language model to use for the chain. 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`. **kwargs: Keyword arguments to pass to `create_qa_with_structure_chain`.
Returns: Returns:
Chain (LLMChain) that can be used to answer questions with citations. 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
)