improve:classifier model

modules/intent_classifier.py

@@ -3,59 +3,31 @@ from .ai_model import AIModel
 
 class IntentClassifier:
     """
-    一个专门用于识别用户输入意图的分类器。
+    一个专门用于识别用户输入意uto的分类器。
     它使用LLM来判断用户的消息属于预定义的哪个类别。
-    【增强版】：内置了意图优先级规则，能处理混合意图。
     """
 
     def __init__(self, ai_model: AIModel):
         self.ai_model = ai_model
-        # 【修改】增加了 INQUIRY 意图，并调整了顺序以反映逻辑
-        self.VALID_INTENTS = ['PROVIDING_TRAVEL_INFO', 'INQUIRY', 'GREETING', 'OTHER']
+        self.VALID_INTENTS = ['PROVIDING_TRAVEL_INFO', 'GREETING', 'OTHER']
 
     def _build_prompt(self, message: str) -> str:
-        """
-        【修改】构建用于意图分类的、带有优先级规则和示例的Prompt。
-        """
+        """构建用于意图分类的、轻量级的Prompt。"""
         return f"""
-# 角色
-你是一个专业的旅行助手对话意图分类器。
+你的任务是分析用户的单条消息，并判断其意图。
+请从以下几个意图中选择一个最匹配的，并且只返回这个意图的字符串，不要添加任何其他解释或标点符号。
 
-# 任务
-你的任务是分析用户的单条消息，并从以下类别中选择最符合用户核心意图的一个类别。你必须严格遵循“意图优先级”规则。
-
-# 意图类别定义
-- PROVIDING_TRAVEL_INFO: 用户明确提供了关于旅行计划的具体信息，如目的地、预算、时长、旅行同伴、兴趣点等。这是需要提取信息的最高优先级意图。
-- INQUIRY: 用户在主动询问旅行建议、信息或请求帮助，但没有提供具体计划信息。例如："法国有什么好玩的？" 或 "帮我规划一下行程"。
-- GREETING: 用户主要在表达问候、打招呼或进行简单的寒暄。例如："你好", "hi", "在吗？"。
-- OTHER: 不属于以上任何一种的通用查询、闲聊或无法识别的意图。
-
-# 核心规则：意图优先级
-当用户输入包含多种意图时，必须遵循以下优先级顺序进行分类：
-PROVIDING_TRAVEL_INFO > INQUIRY > GREETING > OTHER
-
-这意味着：
-1. 只要用户提到了任何具体的旅行计划信息（地点、时间、预算等），意图就必须是 PROVIDING_TRAVEL_INFO。
-2. 如果没有提供具体信息，但有提问，意图就是 INQUIRY。
-3. 只有在几乎纯粹是打招呼时，意图才是 GREETING。
-
-# 示例
-- 用户输入: "你好，我想去东京玩" -> 分类: PROVIDING_TRAVEL_INFO
-- 用户输入: "Hi, 巴黎有什么推荐的吗？" -> 分类: INQUIRY
-- 用户输入: "周末愉快！" -> 分类: GREETING
-- 用户输入: "我们预算不多，大概3000元，目的地是柏林。" -> 分类: PROVIDING_TRAVEL_INFO
-- 用户输入: "你好在吗" -> 分类: GREETING
-- 用户输入: "随便聊聊" -> 分类: OTHER
-
-# 分析与决策
-请分析以下用户消息，并根据上述规则做出唯一的意图判断。
+可用意图:
+- PROVIDING_TRAVEL_INFO: 用户明确提到了旅行相关的具体信息，如目的地、时间、预算、人数等。例如："我想去巴黎玩一周" 或 "预算5000块钱"。
+- GREETING: 用户只是在打招呼或进行简单的问候。例如："你好", "hi", "在吗？"。
+- OTHER: 不属于以上任何一种的通用查询或闲聊。例如："法国有什么好玩的？" 或 "帮我订票"。
 
 用户消息:
 ---
 {message}
 ---
 
-你的判断结果 (请只返回一个意图字符串: PROVIDING_TRAVEL_INFO, INQUIRY, GREETING, 或 OTHER):
+你的判断结果 (请只返回一个意图字符串):
 """
 
     def classify(self, message: str) -> str:
@@ -65,6 +37,7 @@ PROVIDING_TRAVEL_INFO > INQUIRY > GREETING > OTHER
         
         try:
             response = self.ai_model.chat_completion(
+                model="gpt-3.5-turbo",
                 messages=[{"role": "user", "content": prompt}],
                 temperature=0.0,
                 max_tokens=10