# Agent Types
PROMPT_TYPES = {
    "encyclopedia": {
        "name": "encyclopedia",
        "description": "A background information retriever for Evie",
    },
    "history": {
        "name": "history",
        "description": "Prompt to further detail a question based on the previous conversation",
    },
    "html_parse": {
        "name": "html_parse",
        "description": "An aid in transforming HTML-based inputs to markdown",
    },
    "pdf_parse": {
        "name": "pdf_parse",
        "description": "An assistant to parse PDF-content into markdown",
    },
    "rag": {
        "name": "rag",
        "description": "The Main RAG retriever",
    },
    "summary": {
        "name": "summary",
        "description": "An assistant to create a summary when multiple chunks are required for 1 file",
    },
    "transcript": {
        "name": "transcript",
        "description": "An assistant to transform a transcript to markdown.",
    },
    "translation_with_context": {
        "name": "translation_with_context",
        "description": "An assistant to translate text with context",
    },
    "translation_without_context": {
        "name": "translation_without_context",
        "description": "An assistant to translate text without context",
    },
    "check_affirmative_answer": {
        "name": "check_affirmative_answer",
        "description": "An assistant to check if the answer to a question is affirmative",
    },
    "check_additional_information": {
        "name": "check_additional_information",
        "description": "An assistant to check if the answer to a question includes additional information or questions",
    },
}
