from llama_index.core import PromptTemplate

text_qa_template_str = (
    "以下为上下文信息\n"
    "---------------------\n"
    "{context_str}\n"
    "---------------------\n"
    "请根据上下文信息而非先前知识回答我的问题或回复我的指令。前面的上下文信息可能有用，也可能没用，你需要从我给出的上下文信息中选出与我的问题最相关的那些，来为你的回答提供依据。回答一定要忠于原文，简洁但不丢信息，不要胡乱编造。我的问题或指令是什么语种，你就用什么语种回复。\n"
    "问题：{query_str}\n"
    "你的回复： "
)


text_qa_template = PromptTemplate(text_qa_template_str)

refine_template_str = (
    "这是原本的问题： {query_str}\n"
    "我们已经提供了回答: {existing_answer}\n"
    "现在我们有机会改进这个回答 "
    "使用以下更多上下文（仅当需要用时）\n"
    "------------\n"
    "{context_msg}\n"
    "------------\n"
    "根据新的上下文, 请改进原来的回答。"
    "如果新的上下文没有用, 直接返回原本的回答。\n"
    "改进的回答: "
)
refine_template = PromptTemplate(refine_template_str)

summary_template_str = (
    "来自多个来源的上下文信息如下。\n"
    "---------------------\n"
    "{context_str}\n"
    "---------------------\n"
    "鉴于来自多个来源的信息而非先验知识， "
    "回答查询。\n"
    "Query: {query_str}\n"
    "Answer: "
)
summary_template = PromptTemplate(summary_template_str)

simple_template_str = (
    "{query_str}"
)
simple_template = PromptTemplate(simple_template_str)