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.
48 lines
1.9 KiB
Python
48 lines
1.9 KiB
Python
# flake8: noqa
|
|
from langchain.prompts.prompt import PromptTemplate
|
|
|
|
web_search_template = """Please write a passage to answer the question
|
|
Question: {QUESTION}
|
|
Passage:"""
|
|
web_search = PromptTemplate(template=web_search_template, input_variables=["QUESTION"])
|
|
sci_fact_template = """Please write a scientific paper passage to support/refute the claim
|
|
Claim: {Claim}
|
|
Passage:"""
|
|
sci_fact = PromptTemplate(template=sci_fact_template, input_variables=["Claim"])
|
|
arguana_template = """Please write a counter argument for the passage
|
|
Passage: {PASSAGE}
|
|
Counter Argument:"""
|
|
arguana = PromptTemplate(template=arguana_template, input_variables=["PASSAGE"])
|
|
trec_covid_template = """Please write a scientific paper passage to answer the question
|
|
Question: {QUESTION}
|
|
Passage:"""
|
|
trec_covid = PromptTemplate(template=trec_covid_template, input_variables=["QUESTION"])
|
|
fiqa_template = """Please write a financial article passage to answer the question
|
|
Question: {QUESTION}
|
|
Passage:"""
|
|
fiqa = PromptTemplate(template=fiqa_template, input_variables=["QUESTION"])
|
|
dbpedia_entity_template = """Please write a passage to answer the question.
|
|
Question: {QUESTION}
|
|
Passage:"""
|
|
dbpedia_entity = PromptTemplate(
|
|
template=dbpedia_entity_template, input_variables=["QUESTION"]
|
|
)
|
|
trec_news_template = """Please write a news passage about the topic.
|
|
Topic: {TOPIC}
|
|
Passage:"""
|
|
trec_news = PromptTemplate(template=trec_news_template, input_variables=["TOPIC"])
|
|
mr_tydi_template = """Please write a passage in Swahili/Korean/Japanese/Bengali to answer the question in detail.
|
|
Question: {QUESTION}
|
|
Passage:"""
|
|
mr_tydi = PromptTemplate(template=mr_tydi_template, input_variables=["QUESTION"])
|
|
PROMPT_MAP = {
|
|
"web_search": web_search,
|
|
"sci_fact": sci_fact,
|
|
"arguana": arguana,
|
|
"trec_covid": trec_covid,
|
|
"fiqa": fiqa,
|
|
"dbpedia_entity": dbpedia_entity,
|
|
"trec_news": trec_news,
|
|
"mr_tydi": mr_tydi,
|
|
}
|