from langchain.schema import HumanMessage, SystemMessage


def get_FAQ_prompt():
    system_prompt = SystemMessage(
        content=(
            f"""
            You are an AI assistant that creates FAQs.
            Given the document, generate 5 question-answer pairs.
            """
        )
    )
    return system_prompt


def get_FAQ_user_prompt(RAG_DOC):
    user_prompt = HumanMessage(
        content=f"Extract Question Answer Pair from the given doc, {RAG_DOC}"
    )
    return user_prompt