|
|
|
@ -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
|
|
|
|
|
)
|
|
|
|
|