prompts/__init__.py

# -*-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
}