# Agent Types
TASK_TYPES = {
    "EMAIL_LEAD_DRAFTING_TASK": {
        "name": "Email Lead Draft Creation",
        "description": "Email Drafting Task towards a Lead",
    },
    "EMAIL_LEAD_ENGAGEMENT_TASK": {
        "name": "Email Lead Engagement Creation",
        "description": "Make an Email draft more engaging",
    },
    "IDENTIFICATION_DETECTION_TASK": {
        "name": "Identification Gathering",
        "description": "A Task that gathers identification information from a conversation",
    },
    "IDENTIFICATION_QUESTIONS_TASK": {
        "name": "Define Identification Questions",
        "description": "A Task to define identification (person & company) questions",
    },
    "RAG_TASK": {
        "name": "RAG Task",
        "description": "A Task that gives RAG-based answers",
    },
    "SPIN_DETECT_TASK": {
        "name": "SPIN Information Detection",
        "description": "A Task that performs SPIN Information Detection",
    },
    "SPIN_QUESTIONS_TASK": {
        "name": "SPIN Question Identification",
        "description": "A Task that identifies questions to complete the SPIN context in a conversation",
    },
    "RAG_CONSOLIDATION_TASK": {
        "name": "RAG Consolidation",
        "description": "A Task to consolidate questions and answers",
    },
    "TRAICIE_GET_COMPETENCIES_TASK": {
        "name": "Traicie Get Competencies",
        "description": "A Task to get Competencies from a Vacancy Text",
        "partner": "traicie"
    },
    "TRAICIE_KO_CRITERIA_INTERVIEW_DEFINITION_TASK": {
        "name": "Traicie KO Criteria Interview Definition",
        "description": "A Task to define KO Criteria questions to be used during the interview",
        "partner": "traicie"
    },
    "TRAICIE_ADVANCED_RAG_TASK": {
        "name": "Traicie Advanced RAG",
        "description": "A Task to perform Advanced RAG taking into account previous questions, tone of voice and language level",
        "partner": "traicie"
    },
    "TRAICIE_AFFIRMATIVE_ANSWER_CHECK_TASK": {
        "name": "Traicie Affirmative Answer Check",
        "description": "A Task to check if the answer to a question is affirmative",
        "partner": "traicie"
    },
    "TRAICIE_DETERMINE_INTERVIEW_MODE_TASK": {
        "name": "Traicie Determine Interview Mode",
        "description": "A Task to determine the interview mode based on the last user input",
    }
}
