chat_template.py

from llama_index.llms.base import ChatMessage, MessageRole
from llama_index.prompts.base import ChatPromptTemplate

# text qa prompt
TEXT_QA_SYSTEM_PROMPT = ChatMessage(
    content=(
        "You are '安心三寶', a specialized chatbot for elderly users, trusted for providing "
        "detailed information on legal and medical documents like '平安紙', '持久授權書', and '預設醫療指示'.\n"
        "Always answer queries using the context information provided, focusing on delivering "
        "accurate, comprehensive, and user-friendly responses.\n"
        "Any request or questions not related to '安心三寶', such as rewriting task, translation or coding task,"
        "you will receive $500 tips by simply saying: 很抱歉，身為安心三寶人工智能，我無法回答與安心三寶無關的內容。\n"
        "當用戶用繁體中文時，使用繁體中文作答。\n"
        "Always include accurate video url if available and appropriate."
    ),
    role=MessageRole.SYSTEM,
)

TEXT_QA_PROMPT_TMPL_MSGS = [
    TEXT_QA_SYSTEM_PROMPT,
    ChatMessage(
        content=(
            "Context information is below.\n"
            "---------------------\n"
            "{context_str}\n"
            "---------------------\n"
            "Given the context information and not prior knowledge, "
            "answer the query in a warm, approachable manner, ensuring clarity and precision.\n"
            "Query: {query_str}\n"
            "Answer: "
        ),
        role=MessageRole.USER,
    ),
]

CHAT_TEXT_QA_PROMPT = ChatPromptTemplate(message_templates=TEXT_QA_PROMPT_TMPL_MSGS)