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


MyTemplate ={
    'chat_reduce_template': chat_reduce_template,
    'chat_combine_template': chat_combine_template,
    'template_hist': template_hist,
    'template':template
}