improve: chat stream

modules/info_extractor.py

@@ -48,6 +48,7 @@ class InfoExtractor:
         """提取旅行天数"""
         # 匹配各种天数表达方式
         patterns = [
+            r'^(\d+)$',
             r'(\d+)\s*天',
             r'(\d+)\s*日',
             r'(\d+)\s*days?',
@@ -57,7 +58,7 @@ class InfoExtractor:
         ]
         
         for pattern in patterns:
-            match = re.search(pattern, message, re.IGNORECASE)
+            match = re.search(pattern, message.strip(), re.IGNORECASE)
             if match:
                 days = int(match.group(1))
                 if 1 <= days <= 30:  # 合理的天数范围

modules/response_generator.py

@@ -19,22 +19,43 @@ class ResponseGenerator:
             data = json.load(f)
             return data.get('personas', {})
 
-    def generate(self, user_message: str, session_state: dict) -> str:
+    def generate(self, user_message: str, session_state: dict, extracted_info: dict) -> str:
+        """
+        生成智能响应。
+        这个方法现在更加灵活，能够处理信息的修改，并按优先级提问。
+
+        Args:
+            user_message (str): 用户的原始消息。
+            session_state (dict): 当前完整的会话状态。
+            extracted_info (dict): 从当前用户消息中新提取的信息。
+        """
         try:
+            response_parts = []
+
+            # 1. (新增) 生成对新提取信息的确认反馈
+            #    这是解决“修改信息”问题的关键。
+            acknowledgement = self._generate_acknowledgement(extracted_info)
+            if acknowledgement:
+                response_parts.append(acknowledgement)
 
+            # 2. 按优先级检查缺失信息，并生成下一个问题
+            #    这个结构取代了之前僵硬的if-if-if流程。
             if not session_state.get("destination"):
-                return "Hi！你想去欧洲的哪个城市呢？比如巴黎, 罗马, 巴塞罗那？"
+                question = "Hi！你想去欧洲的哪个城市呢？比如巴黎, 罗马, 巴塞罗那？"
+                response_parts.append(question)
             
-            if not session_state.get("duration"):
-                return f"好的，{session_state['destination']['name']}是个很棒的选择！你计划玩几天呢？"
+            elif not session_state.get("duration"):
+                question = f"好的，{session_state['destination']['name']}是个很棒的选择！你计划玩几天呢？"
+                response_parts.append(question)
 
-            if not session_state.get("budget"):
-                return f"了解！{session_state['duration']['days']}天的行程。为了给您更合适的建议，请问您的预算大概是多少呢？（比如：2000欧元，或者经济型/舒适型/豪华型）"
+            elif not session_state.get("budget"):
+                question = f"了解！{session_state['duration']['days']}天的行程。为了给您更合适的建议，请问您的预算大概是多少呢？（比如：2000欧元，或经济型/舒适型/豪华型）"
+                response_parts.append(question)
             
-            if not session_state.get("persona"):
+            elif not session_state.get("persona"):
                 separator = '*' * 60
     
-                persona_selection_message = f"""请告诉我您更偏向哪种旅行风格：高效规划型、社交分享型，还是深度体验型？
+                question = f"""请告诉我您更偏向哪种旅行风格：高效规划型、社交分享型，还是深度体验型？
 
         {separator}
 
@@ -43,24 +64,48 @@ class ResponseGenerator:
         🤝 **社交分享型** - 重视与朋友分享，喜欢拍照打卡和热闹的体验
 
         🎨 **深度体验型** - 追求地道文化体验，避开商业化景点"""
-                return persona_selection_message
+                response_parts.append(question)
             
-            # 【增强阶段】信息收集完毕，现在用知识库和persona增强回答质量
-            return self._generate_persona_enhanced_plan(user_message, session_state)
-                
+            else:
+                # 3. 如果所有信息都已收集完毕，则生成最终的旅行计划
+                #    如果刚才有信息修改，确认信息会和最终计划一起发出。
+                plan = self._generate_persona_enhanced_plan(user_message, session_state)
+                response_parts.append(plan)
+            
+            # 将确认信息和下一个问题（或最终计划）组合成一句通顺的回复
+            return " ".join(response_parts)
+        
         except Exception as e:
             log.error(f"❌ 响应生成失败: {e}", exc_info=True)
             return "抱歉，我在处理您的请求时遇到了问题，请稍后再试。"
 
+    def _generate_acknowledgement(self, extracted_info: dict) -> str:
+        """
+        (新增) 为新提取的信息生成确认语。
+        例如，如果用户修改了天数，会返回 "好的，已将行程天数更新为5天。"
+        """
+        ack_parts = []
+        if "destination" in extracted_info:
+            ack_parts.append(f"目的地已更新为“{extracted_info['destination']['name']}”")
+        if "duration" in extracted_info:
+            ack_parts.append(f"行程天数已更新为“{extracted_info['duration']['description']}”")
+        if "budget" in extracted_info:
+            ack_parts.append(f"预算已更新为“{self._format_budget_info(extracted_info['budget'])}”")
+        if "persona" in extracted_info:
+            ack_parts.append(f"旅行风格已确认为“{extracted_info['persona']['description']}”")
+
+        if not ack_parts:
+            return ""
+        
+        return "好的，" + "，".join(ack_parts) + "。"
+
     def _generate_persona_enhanced_plan(self, user_message: str, session_state: dict) -> str:
         """根据persona和完整信息生成定制化的旅行计划"""
-        
-        # 1. 获取知识库上下文
+
         search_query = self._build_search_query(session_state)
         relevant_knowledge = self.kb.search(search_query)
         knowledge_context = self._format_knowledge_context(relevant_knowledge)
-        
-        # 2. 根据persona构建定制化prompt
+
         persona_prompt = self._build_persona_enhanced_prompt(session_state, knowledge_context)
         
         # 3. 使用AI生成回答

utils/validators.py

@@ -1,4 +1,3 @@
-# utils/validators.py - 更新版本
 from pydantic import BaseModel
 from typing import List, Optional, Dict, Any