File size: 6,117 Bytes
0c83b5e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 | from typing import Dict, List
GLOBAL_INTENTS = [
"restart",
"new_topic",
"complaint",
"direct_support",
"courses_info",
"children_courses",
"adults_courses",
"new_student",
"current_student",
"unclear",
]
STATE_ALLOWED_INTENTS: Dict[str, List[str]] = {
"WAITING_MAIN_MENU": [
"new_student",
"current_student",
"courses_info",
"complaint",
"direct_support",
],
"WAITING_USER_TYPE": [
"new_student",
"current_student",
],
"WAITING_AUDIENCE": [
"adults",
"children",
],
"WAITING_PRIOR_STUDY": [
"prior_study_yes",
"prior_study_no",
],
"WAITING_BEGINNER_SCHEDULE_CHOICE": [
"confirm_schedule_reviewed",
"proceed_booking",
"switch_to_prior_study_true",
"switch_to_prior_study_false",
"support_needed",
],
"WAITING_PDF_102_CONFIRMATION": [
"confirm_pdf_reviewed",
"switch_to_prior_study_true",
"switch_to_prior_study_false",
"support_needed",
],
"WAITING_PLACEMENT_TEST_CONFIRMATION": [
"confirm_placement_test_reviewed",
"switch_to_prior_study_true",
"switch_to_prior_study_false",
"support_needed",
],
"WAITING_CURRENT_STUDENT_ACTION": [
"current_student_support",
"current_student_next_level",
],
"WAITING_SUPPORT_QUESTION": [
"support_question_text",
],
"WAITING_LEVEL_SELECTION": [
"level_selected",
"support_needed",
],
"WAITING_PAYMENT_METHOD": [
"payment_method_selected",
"support_needed",
],
"WAITING_COMPLAINT_FORM": [
"complaint_form_submitted",
],
"HANDOFF_DONE": [
"thanks",
"courses_info",
"children_courses",
"adults_courses",
"new_student",
"current_student",
"direct_support",
"complaint",
"restart",
"new_topic",
],
}
INTENT_DESCRIPTIONS: Dict[str, str] = {
"restart": "The user wants to restart the conversation from the beginning.",
"new_topic": "The user explicitly wants to ask about something else or start a new topic.",
"complaint": "The user is making or asking to make a complaint.",
"direct_support": "The user wants to talk to customer support or service directly.",
"courses_info": "The user is asking generally about available courses.",
"children_courses": "The user is asking about children courses.",
"adults_courses": "The user is asking about adult courses.",
"new_student": "The user indicates they are a new student.",
"current_student": "The user indicates they are a current student.",
"adults": "Direct answer: adults.",
"children": "Direct answer: children.",
"prior_study_yes": "Direct answer: user studied German before.",
"prior_study_no": "Direct answer: user did not study German before.",
"confirm_schedule_reviewed": "The user confirms they reviewed the beginner schedule.",
"proceed_booking": "The user wants to proceed with booking.",
"switch_to_prior_study_true": "The user changes to the path where they studied German before.",
"switch_to_prior_study_false": "The user changes to the beginner path.",
"support_needed": "The user needs help or wants support within the current topic.",
"confirm_pdf_reviewed": "The user confirms they reviewed the PDF/details file.",
"confirm_placement_test_reviewed": "The user confirms they reviewed placement test details.",
"current_student_support": "Current student wants support or has a question.",
"current_student_next_level": "Current student wants to book the next level.",
"support_question_text": "The message itself is the support question text.",
"level_selected": "The user selected a course level.",
"payment_method_selected": "The user selected a payment method.",
"complaint_form_submitted": "The user says they submitted the complaint form.",
"thanks": "The user is thanking the bot.",
"unclear": "The message is unclear and should not be confidently routed.",
}
def get_allowed_intents_for_state(state: str) -> List[str]:
state_specific = STATE_ALLOWED_INTENTS.get(state, [])
final = []
for item in GLOBAL_INTENTS + state_specific:
if item not in final:
final.append(item)
return final
def _summarize_flow_data(flow_data: dict) -> str:
if not flow_data:
return "No extra context."
fields = []
for key in [
"customer_type",
"audience",
"prior_study",
"selected_level",
"payment_method",
"gender",
]:
if key in flow_data and flow_data[key] is not None:
fields.append(f"{key}={flow_data[key]}")
return ", ".join(fields) if fields else "No extra context."
def build_system_prompt(state: str, flow_data: dict, allowed_intents: List[str]) -> str:
descriptions = []
for intent in allowed_intents:
desc = INTENT_DESCRIPTIONS.get(intent, "")
descriptions.append(f"- {intent}: {desc}")
description_block = "\n".join(descriptions)
context_summary = _summarize_flow_data(flow_data)
return f"""
You are a state-aware intent classifier for an Arabic WhatsApp bot.
Your task:
- Read the user's Arabic message.
- Understand the CURRENT conversation state.
- Choose exactly ONE intent label from the allowed list.
- Return ONLY the label.
- Do not explain.
- Do not add punctuation.
- Do not add extra words.
Current state:
{state}
Known context:
{context_summary}
Allowed intent labels:
{", ".join(allowed_intents)}
Intent descriptions:
{description_block}
Decision rules:
- If the message is a direct answer to the current question, choose the direct answer label.
- If the user changes path but stays in the same broad topic, choose the appropriate state-switch label.
- If the user changes to a new topic, choose the correct topic-switch label.
- If you are not confident, return: unclear
Return only one label from the allowed list.
""".strip() |