File size: 936 Bytes
6c945f2 d067a6a 6c945f2 d067a6a 6c945f2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
# -*-coding:utf-8 -*-
# load prompt template
with open("prompts/combine_prompt.txt", "r") as f:
template = f.read()
with open("prompts/combine_prompt_hist.txt", "r") as f:
template_hist = f.read()
with open("prompts/chat_combine_prompt.txt", "r") as f:
chat_combine_template = f.read()
with open("prompts/chat_reduce_prompt.txt", "r") as f:
chat_reduce_template = f.read()
with open("prompts/qa_sys_prompt.txt", "r") as f:
qa_sys_prompt = f.read()
with open("prompts/qa_user_prompt.txt", "r") as f:
qa_user_prompt = f.read()
with open("prompts/summary_prompt.txt", "r") as f:
summary_prompt = f.read()
MyTemplate ={
'chat_reduce_template': chat_reduce_template,
'chat_combine_template': chat_combine_template,
'template_hist': template_hist,
'template':template,
'qa_sys_template': qa_sys_prompt,
'qa_user_template': qa_user_prompt,
'summary_template': summary_prompt
} |