forked from Archives/langchain
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
# flake8: noqa
|
|
from langchain.prompts import PromptTemplate
|
|
from langchain.prompts.base import RegexParser
|
|
|
|
output_parser = RegexParser(
|
|
regex=r"(.*?)\nSOURCES: (.*)",
|
|
output_keys=["answer", "sources"],
|
|
default_output_key="answer",
|
|
)
|
|
|
|
DEFAULT_REFINE_PROMPT_TMPL = (
|
|
"The original question is as follows: {question}\n"
|
|
"We have provided an existing answer, including sources: {existing_answer}\n"
|
|
"We have the opportunity to refine the existing answer"
|
|
"(only if needed) with some more context below.\n"
|
|
"------------\n"
|
|
"{context_str}\n"
|
|
"------------\n"
|
|
"Given the new context, refine the original answer to better "
|
|
"answer the question. "
|
|
"If you do update it, please update the sources as well. "
|
|
"If the context isn't useful, return the original answer."
|
|
)
|
|
DEFAULT_REFINE_PROMPT = PromptTemplate(
|
|
input_variables=["question", "existing_answer", "context_str"],
|
|
template=DEFAULT_REFINE_PROMPT_TMPL,
|
|
output_parser=output_parser,
|
|
)
|
|
|
|
|
|
DEFAULT_TEXT_QA_PROMPT_TMPL = (
|
|
"Context information is below. \n"
|
|
"---------------------\n"
|
|
"{context_str}"
|
|
"\n---------------------\n"
|
|
"Given the context information and not prior knowledge, "
|
|
"answer the question: {question}\n"
|
|
)
|
|
DEFAULT_TEXT_QA_PROMPT = PromptTemplate(
|
|
input_variables=["context_str", "question"], template=DEFAULT_TEXT_QA_PROMPT_TMPL
|
|
)
|
|
|
|
EXAMPLE_PROMPT = PromptTemplate(
|
|
template="Content: {page_content}\nSource: {source}",
|
|
input_variables=["page_content", "source"],
|
|
)
|