improve: intent classifier

modules/info_extractor.py

@@ -10,10 +10,15 @@ class InfoExtractor:
 
     def _build_prompt_template(self) -> str:
 
-        return """你是一个顶级的旅游信息提取专家。
-你的任务是从用户的输入中，精准地提取出【目的地】、【旅行天数】和【预算】这三项核心信息。
-请严格按照下面的JSON格式返回结果，不要添加任何额外的解释或说明。
-如果某项信息不存在，请将对应的值设为 null。
+        return """你是一个专业的旅游信息提取AI。
+你的任务是仔细阅读用户的请求，并从中提取出关键的旅行信息。
+
+请严格按照以下嵌套的JSON格式返回
+---
+**重要规则**
+1. 如果某个信息在用户请求中没有明确提及，请将对应的值设为 null。
+2. **如果用户的请求只是简单的问候 (例如 "hi", "你好")，或者完全不包含任何目的地、时间、预算等旅行信息，请必须返回一个空的JSON对象，即 `{}`。**
+---
 
 {{
   "destination": {{

modules/intent_classifier.py

@@ -0,0 +1,55 @@
+import logging
+from .ai_model import AIModel
+
+class IntentClassifier:
+    """
+    一个专门用于识别用户输入意uto的分类器。
+    它使用LLM来判断用户的消息属于预定义的哪个类别。
+    """
+
+    def __init__(self, ai_model: AIModel):
+        self.ai_model = ai_model
+        self.VALID_INTENTS = ['PROVIDING_TRAVEL_INFO', 'GREETING', 'OTHER']
+
+    def _build_prompt(self, message: str) -> str:
+        """构建用于意图分类的、轻量级的Prompt。"""
+        return f"""
+你的任务是分析用户的单条消息，并判断其意图。
+请从以下几个意图中选择一个最匹配的，并且只返回这个意图的字符串，不要添加任何其他解释或标点符号。
+
+可用意图:
+- PROVIDING_TRAVEL_INFO: 用户明确提到了旅行相关的具体信息，如目的地、时间、预算、人数等。例如："我想去巴黎玩一周" 或 "预算5000块钱"。
+- GREETING: 用户只是在打招呼或进行简单的问候。例如："你好", "hi", "在吗？"。
+- OTHER: 不属于以上任何一种的通用查询或闲聊。例如："法国有什么好玩的？" 或 "帮我订票"。
+
+用户消息:
+---
+{message}
+---
+
+你的判断结果 (请只返回一个意图字符串):
+"""
+
+    def classify(self, message: str) -> str:
+        """对给定的消息进行意图分类。"""
+        log.info(f"🚀 开始对消息进行意图分类: '{message[:30]}...'")
+        prompt = self._build_prompt(message)
+        
+        try:
+            response = self.ai_model.chat_completion(
+                model="gpt-3.5-turbo",
+                messages=[{"role": "user", "content": prompt}],
+                temperature=0.0,
+                max_tokens=10
+            )
+            intent = response.strip().replace("'", "").replace("\"", "")
+            
+            if intent in self.VALID_INTENTS:
+                log.info(f"✅ 意图分类成功: {intent}")
+                return intent
+            else:
+                log.warning(f"⚠️ LLM返回了未知的意图 '{intent}'，将回退到 'OTHER'")
+                return 'OTHER'
+        except Exception as e:
+            log.error(f"❌ 意图分类时发生错误: {e}", exc_info=True)
+            return 'OTHER'

modules/travel_assistant.py

@@ -2,6 +2,7 @@
 from .config_loader import ConfigLoader
 from .ai_model import AIModel
 from .knowledge_base import KnowledgeBase
+from .intent_classifier import IntentClassifier
 from .info_extractor import InfoExtractor
 from .session_manager import SessionManager
 from .response_generator import ResponseGenerator
@@ -15,7 +16,8 @@ class TravelAssistant:
         self.kb = KnowledgeBase()
         self.ai_model = AIModel()
         self.session_manager = SessionManager()
-        self.info_extractor = InfoExtractor(self.ai_model) 
+        self.info_extractor = InfoExtractor(self.ai_model)
+        self.intent_classifier = IntentClassifier(self.ai_model) 
         self.response_generator = ResponseGenerator(self.ai_model, self.kb)
         log.info("✅ Travel Assistant 核心模块全部初始化完成！")
 
@@ -43,17 +45,32 @@ class TravelAssistant:
             session_state = self.session_manager.get_or_create_session(current_session_id)
             log.info(f"✅ 设置persona: {persona_info['name']}")
 
-        # 3. 提取信息
-        extracted_info = self.info_extractor.extract(message)
-        log.info(f"🔍 提取信息: {list(extracted_info.keys()) if extracted_info else '无'}")
+        # 3. 意图识别 (前置守卫)
+        intent = self.intent_classifier.classify(message)
+        log.info(f"🔍 用户意图识别结果: '{intent}'")
+
+        extracted_info = {}
         
-        # 4. 更新会话状态
-        if extracted_info:
-            self.session_manager.update_session(current_session_id, extracted_info)
-            session_state = self.session_manager.get_or_create_session(current_session_id)
+        # 4.: 根据意图进行逻辑分流
+        if intent == 'PROVIDING_TRAVEL_INFO':
+            # 场景A: 用户提供了旅行信息，执行完整的信息提取
+            extracted_info = self.info_extractor.extract(message)
+            if extracted_info:
+                self.session_manager.update_session(current_session_id, extracted_info)
+                session_state = self.session_manager.get_or_create_session(current_session_id)
+            
+            # 无论是否提取成功，都让 response_generator 来生成上下文感知的回复
+            bot_response = self.response_generator.generate(message, session_state, extracted_info)
+
+        else:
+            # 场景B: 用户意图是问候或其它，直接生成引导性回复，完全绕过信息提取
+            log.info(f"💬 意图为 '{intent}'，绕过信息提取，直接生成引导性回复。")
+            if intent == 'GREETING':
+                bot_response = "您好！很高兴能为您规划旅程。请问您想去哪里，玩几天，预算大概是多少呢？"
+            else: # 'OTHER'
+                # 对于其它问题，可以调用通用的生成器，让它决定如何回复
+                bot_response = self.response_generator.generate(message, session_state, {})
 
-        # 5. 生成回复
-        bot_response = self.response_generator.generate(message, session_state, extracted_info)
         
         # 6. 返回结果
         status_info = self.session_manager.format_session_info(session_state)