fix: recover file

modules/response_generator.py

@@ -0,0 +1,157 @@
+# modules/response_generator.py - 修复版本
+from .ai_model import AIModel
+from .knowledge_base import KnowledgeBase
+from utils.logger import log
+
+class ResponseGenerator:
+    def __init__(self, ai_model: AIModel, knowledge_base: KnowledgeBase):
+        self.ai_model = ai_model
+        self.kb = knowledge_base
+
+    def generate(self, user_message: str, session_state: dict) -> str:
+        try:
+            # 1. 优先使用 RAG (检索增强生成)
+            search_query = user_message
+            if session_state.get("destination"):
+                search_query += f" {session_state['destination']['name']}"
+                
+            relevant_knowledge = self.kb.search(search_query)
+            if relevant_knowledge:
+                context = self._format_knowledge_context(relevant_knowledge)
+                if self.ai_model.is_available():
+                    return self.ai_model.generate(user_message, context)
+
+            # 2. 如果没有知识库匹配，则使用基于规则的引导式对话
+            if not session_state.get("destination"):
+                return "听起来很棒！你想去欧洲的哪个城市呢？比如巴黎, 罗马, 巴塞罗那？"
+            if not session_state.get("duration"):
+                return f"好的，{session_state['destination']['name']}是个很棒的选择！你计划玩几天呢？"
+            if not session_state.get("persona"):
+                return "最后一个问题，这次旅行对你来说什么最重要呢？（例如：美食、艺术、购物、历史）"
+            
+            # 3. 如果信息都收集全了，但没触发RAG，让AI生成一个通用计划
+            if self.ai_model.is_available():
+                plan_prompt = self._build_plan_prompt(session_state)
+                return self.ai_model.generate(plan_prompt, context="用户需要一个详细的旅行计划。")
+            else:
+                return self._generate_fallback_plan(session_state)
+                
+        except Exception as e:
+            log.error(f"❌ 响应生成失败: {e}", exc_info=True)
+            return "抱歉，我在处理您的请求时遇到了问题，请稍后再试。"
+
+    def _build_plan_prompt(self, session_state: dict) -> str:
+        """构建AI生成计划的提示词 - 修复版本"""
+        destination = session_state.get("destination", {})
+        duration = session_state.get("duration", {})
+        persona = session_state.get("persona", {})
+    
+        prompt = f"请为用户生成一个在 {destination.get('name', '未知城市')} 的 "
+        prompt += f"{duration.get('days', '几')} 天旅行计划。"
+    
+        if persona:
+            # 使用name而不是description
+            persona_name = persona.get('name', '一般旅行者')
+            persona_style = persona.get('style', '')
+            prompt += f" 旅行风格: {persona_name}"
+            if persona_style:
+                prompt += f"（{persona_style}）"
+            prompt += "。"
+    
+        return prompt
+
+    def _generate_fallback_plan(self, session_state: dict) -> str:
+        """AI不可用时的备用计划生成"""
+        destination = session_state.get("destination", {})
+        duration = session_state.get("duration", {})
+        persona = session_state.get("persona", {})
+        
+        plan = f"为您推荐 {destination.get('name', '目的地')} {duration.get('days', '几')}天旅行计划：\n\n"
+        
+        # 使用cities.json中的信息
+        highlights = destination.get('highlights', '精彩景点等待您的探索')
+        plan += f"🎯 主要景点：{highlights}\n\n"
+        
+        best_season = destination.get('best_season', '全年适宜')
+        plan += f"🌤️ 最佳旅行时间：{best_season}\n\n"
+        
+        budget = destination.get('avg_daily_budget', 100)
+        plan += f"💰 预算参考：约{budget}欧元/天\n\n"
+        
+        if persona:
+            plan += f"🎨 根据您的{persona.get('name', '旅行')}风格，建议您重点关注相关的景点和体验。\n\n"
+        
+        plan += "详细行程规划请稍后重试，或告诉我您的具体需求！"
+        
+        return plan
+
+    def _format_knowledge_context(self, knowledge_items: list) -> str:
+        """格式化知识库内容作为上下文 - 完全安全版本"""
+        if not knowledge_items: 
+            return "没有特定的背景知识。"
+        
+        try:
+            # 只使用最相关的一条知识
+            item = knowledge_items[0]
+            
+            # 安全地获取嵌套数据
+            knowledge = item.get('knowledge', {})
+            travel_knowledge = knowledge.get('travel_knowledge', {})
+            
+            context_parts = []
+            
+            # 1. 获取目的地信息
+            destination_info = travel_knowledge.get('destination_info', {})
+            if destination_info:
+                primary_destinations = destination_info.get('primary_destinations', [])
+                if primary_destinations:
+                    context_parts.append(f"目的地: {', '.join(primary_destinations)}")
+                
+                recommended_duration = destination_info.get('recommended_duration')
+                if recommended_duration:
+                    context_parts.append(f"推荐天数: {recommended_duration}天")
+                
+                travel_theme = destination_info.get('travel_theme')
+                if travel_theme:
+                    context_parts.append(f"旅行主题: {travel_theme}")
+
+            # 2. 获取预算信息
+            budget_analysis = travel_knowledge.get('budget_analysis', {})
+            if budget_analysis:
+                total_budget = budget_analysis.get('total_budget_range')
+                if total_budget:
+                    context_parts.append(f"预算参考: {total_budget}")
+                
+                daily_average = budget_analysis.get('daily_average')
+                if daily_average:
+                    context_parts.append(f"日均预算: {daily_average}")
+
+            # 3. 获取专业见解 - 根据实际数据结构
+            professional_insights = travel_knowledge.get('professional_insights', {})
+            if professional_insights:
+                # 常见错误
+                common_mistakes = professional_insights.get('common_mistakes', [])
+                if common_mistakes:
+                    context_parts.append(f"注意事项: {', '.join(common_mistakes[:2])}")
+                
+                # 内行秘籍
+                insider_secrets = professional_insights.get('insider_secrets', [])
+                if insider_secrets:
+                    context_parts.append(f"内行贴士: {', '.join(insider_secrets[:2])}")
+                
+                # 最佳月份
+                seasonal_info = professional_insights.get('seasonal_considerations', {})
+                if seasonal_info:
+                    best_months = seasonal_info.get('best_months', [])
+                    if best_months:
+                        context_parts.append(f"最佳时间: {', '.join(best_months)}")
+
+            # 4. 拼接结果
+            if context_parts:
+                return "相关旅游知识：\n- " + "\n- ".join(context_parts)
+            else:
+                return "基于旅游知识库的一般建议。"
+                
+        except Exception as e:
+            log.error(f"❌ 格式化知识库上下文失败: {e}", exc_info=True)
+            return "知识库处理时发生错误，使用一般旅游建议。"