feat: 集成persona系统和预算收集

✨ 新功能:
- 完整persona支持（planner/social/experiential）
- 预算收集和约束处理
- 动态AI prompt构建

🔧 改进:
- 修正response_generator逻辑主次关系
- 优化信息收集流程
- 增强备用方案生成

📱 API更新:
- 支持persona_key参数传递
- 更新ChatRequest验证器

app.py

@@ -89,11 +89,14 @@ async def chat_endpoint(request: ChatRequest):
     
     try:
         log.info(f"收到聊天请求: {request.message[:50]}...")
+        if request.persona_key:
+            log.info(f"用户选择的persona: {request.persona_key}")
         
         reply, session_id, status_info, history = assistant.chat(
             request.message, 
             request.session_id,
-            request.history or []
+            request.history or [],
+            request.persona_key  # 传递persona参数
         )
         
         log.info(f"聊天响应生成成功，会话ID: {session_id}")
@@ -111,6 +114,13 @@ async def chat_endpoint(request: ChatRequest):
             status_code=500,
             detail="Internal Server Error: Failed to process chat request."
         )
+        
+    except Exception as e:
+        log.error(f"❌ Chat endpoint error: {e}", exc_info=True)
+        raise HTTPException(
+            status_code=500,
+            detail="Internal Server Error: Failed to process chat request."
+        )
 
 @app.post("/api/reset")
 async def reset_session(session_id: str):

modules/response_generator.py

@@ -1,4 +1,5 @@
-# modules/response_generator.py - 修复版本
+import json
+import os
 from .ai_model import AIModel
 from .knowledge_base import KnowledgeBase
 from utils.logger import log
@@ -7,94 +8,254 @@ class ResponseGenerator:
     def __init__(self, ai_model: AIModel, knowledge_base: KnowledgeBase):
         self.ai_model = ai_model
         self.kb = knowledge_base
+        self.personas = self._load_personas()
 
-    def generate(self, user_message: str, session_state: dict) -> str:
+    def _load_personas(self):
+        """加载personas配置"""
         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)
+            personas_path = os.path.join(os.path.dirname(__file__), '..', 'data', 'personas.json')
+            with open(personas_path, 'r', encoding='utf-8') as f:
+                data = json.load(f)
+                return data.get('personas', {})
+        except Exception as e:
+            log.error(f"❌ 加载personas.json失败: {e}")
+            return {}
 
-            # 2. 如果没有知识库匹配，则使用基于规则的引导式对话
+    def generate(self, user_message: str, session_state: dict) -> str:
+        try:
+            # 【主流程】按顺序收集信息：目的地 → 天数 → 预算 → persona（可选）
+            
+            # 1. 检查是否需要收集目的地
             if not session_state.get("destination"):
                 return "听起来很棒！你想去欧洲的哪个城市呢？比如巴黎, 罗马, 巴塞罗那？"
+            
+            # 2. 检查是否需要收集天数
             if not session_state.get("duration"):
                 return f"好的，{session_state['destination']['name']}是个很棒的选择！你计划玩几天呢？"
+            
+            # 3. 检查是否需要收集预算
+            if not session_state.get("budget"):
+                return f"了解！{session_state['duration']['days']}天的行程。为了给您更合适的建议，请问您的预算大概是多少呢？（比如：2000欧元，或者经济型/舒适型/豪华型）"
+            
+            # 4. persona由前端传入，这里检查是否存在
+            # 注意：如果前端没传persona，可以询问或使用默认值
             if not session_state.get("persona"):
-                return "最后一个问题，这次旅行对你来说什么最重要呢？（例如：美食、艺术、购物、历史）"
+                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)
+            # 【增强阶段】信息收集完毕，现在用知识库和persona增强回答质量
+            return self._generate_persona_enhanced_plan(user_message, session_state)
                 
         except Exception as e:
             log.error(f"❌ 响应生成失败: {e}", exc_info=True)
             return "抱歉，我在处理您的请求时遇到了问题，请稍后再试。"
 
-    def _build_plan_prompt(self, session_state: dict) -> str:
-        """构建AI生成计划的提示词 - 修复版本"""
+    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生成回答
+        if self.ai_model.is_available():
+            return self.ai_model.generate(user_message, persona_prompt)
+        else:
+            return self._generate_fallback_plan(session_state, knowledge_context)
+
+    def _build_persona_enhanced_prompt(self, session_state: dict, knowledge_context: str = "") -> str:
+        """根据persona构建增强的prompt"""
         destination = session_state.get("destination", {})
         duration = session_state.get("duration", {})
+        budget = session_state.get("budget", {})
         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 += "。"
-    
+        
+        # 基础信息
+        location = destination.get('name', '目的地')
+        days = duration.get('days', '几')
+        budget_info = self._format_budget_info(budget)
+        
+        # 获取persona配置
+        persona_key = persona.get('key')
+        if not persona_key or persona_key not in self.personas:
+            # 如果没有有效persona，使用通用prompt
+            return self._build_generic_prompt(session_state, knowledge_context)
+        
+        persona_config = self.personas[persona_key]
+        
+        # 使用personas.json中的prompt_template
+        persona_template = persona_config.get('prompt_template', '')
+        
+        # 替换模板中的变量
+        enhanced_prompt = persona_template.format(
+            location=location,
+            days=days,
+            date="近期",  # 可以从session中获取更具体的日期
+            user_tags="",  # 可以从用户偏好中提取
+            commercial_preference="适中",  # 可以从session中获取
+            group_description="个人/朋友",  # 可以从session中获取
+            budget=budget_info,  # 使用格式化的预算信息
+            tags=""  # 可以从session中获取
+        )
+        
+        # 添加知识库上下文
+        if knowledge_context:
+            enhanced_prompt += f"\n\n【背景知识】\n{knowledge_context}"
+        
+        # 添加预算约束信息
+        enhanced_prompt += f"\n\n【预算约束】\n用户预算：{budget_info}，请确保所有推荐都在预算范围内。"
+        
+        # 添加persona特定的指导原则
+        characteristics = persona_config.get('characteristics', [])
+        if characteristics:
+            enhanced_prompt += f"\n\n【用户特征】\n" + "\n".join([f"- {char}" for char in characteristics])
+        
+        recommendation_strategy = persona_config.get('recommendation_strategy', [])
+        if recommendation_strategy:
+            enhanced_prompt += f"\n\n【推荐策略】\n" + "\n".join([f"- {strategy}" for strategy in recommendation_strategy])
+        
+        tone_guidelines = persona_config.get('tone', [])
+        if tone_guidelines:
+            enhanced_prompt += f"\n\n【语言风格】\n" + "\n".join([f"- {tone}" for tone in tone_guidelines])
+        
+        return enhanced_prompt
+
+    def _format_budget_info(self, budget: dict) -> str:
+        """格式化预算信息"""
+        if not budget:
+            return "中等预算"
+        
+        # 如果有具体金额
+        if budget.get('amount') and budget.get('currency'):
+            return f"{budget['amount']}{budget['currency']}"
+        
+        # 如果有预算类型
+        if budget.get('type'):
+            budget_type_map = {
+                'economy': '经济型预算',
+                'comfortable': '舒适型预算', 
+                'luxury': '豪华型预算'
+            }
+            return budget_type_map.get(budget['type'], budget['type'])
+        
+        # 如果有预算范围
+        if budget.get('range'):
+            return budget['range']
+        
+        return "中等预算"
+
+    def _build_generic_prompt(self, session_state: dict, knowledge_context: str = "") -> str:
+        """构建通用prompt（当没有persona时使用）"""
+        destination = session_state.get("destination", {})
+        duration = session_state.get("duration", {})
+        budget = session_state.get("budget", {})
+        
+        location = destination.get('name', '目的地')
+        days = duration.get('days', '几')
+        budget_info = self._format_budget_info(budget)
+        
+        prompt = f"""你是一个专业的旅游助手。请为用户生成一个详细的旅行计划。
+
+【基本信息】
+- 目的地：{location}
+- 旅行天数：{days}天
+- 预算：{budget_info}
+
+【要求】
+- 提供具体的景点推荐和路线安排
+- 包含交通、住宿、餐饮建议
+- 确保所有推荐都在预算范围内
+- 提供实用的旅行贴士"""
+        
+        if knowledge_context:
+            prompt += f"\n\n【背景信息】\n{knowledge_context}"
+        
+        prompt += "\n\n请生成一份实用、详细的旅行计划。"
+        
         return prompt
 
-    def _generate_fallback_plan(self, session_state: dict) -> str:
-        """AI不可用时的备用计划生成"""
+    def _build_search_query(self, session_state: dict) -> str:
+        """构建知识库搜索查询"""
+        destination = session_state.get("destination", {})
+        persona = session_state.get("persona", {})
+        budget = session_state.get("budget", {})
+        
+        query_parts = []
+        
+        # 添加目的地
+        if destination.get('name'):
+            query_parts.append(destination['name'])
+        
+        # 添加persona类型
+        if persona.get('key'):
+            query_parts.append(persona['key'])
+        
+        # 添加预算类型
+        if budget.get('type'):
+            query_parts.append(budget['type'])
+        
+        return " ".join(query_parts) if query_parts else "欧洲旅行"
+
+    def _generate_fallback_plan(self, session_state: dict, knowledge_context: str = "") -> str:
+        """AI不可用时的备用计划生成（现在包含预算和persona信息）"""
         destination = session_state.get("destination", {})
         duration = session_state.get("duration", {})
+        budget = session_state.get("budget", {})
         persona = session_state.get("persona", {})
         
-        plan = f"为您推荐 {destination.get('name', '目的地')} {duration.get('days', '几')}天旅行计划：\n\n"
+        location = destination.get('name', '目的地')
+        days = duration.get('days', '几')
+        budget_info = self._format_budget_info(budget)
+        persona_name = persona.get('name', '旅行者')
+        
+        plan = f"为您推荐 {location} {days}天旅行计划：\n\n"
+        
+        # 用户信息概述
+        plan += f"👤 旅行者类型：{persona_name}\n"
+        plan += f"💰 预算范围：{budget_info}\n\n"
         
-        # 使用cities.json中的信息
+        # 优先使用知识库信息
+        if knowledge_context and "相关旅游知识" in knowledge_context:
+            plan += f"📚 {knowledge_context}\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"
+        avg_budget = destination.get('avg_daily_budget', 100)
+        plan += f"💰 日均预算参考：约{avg_budget}欧元/天\n\n"
         
-        if persona:
-            plan += f"🎨 根据您的{persona.get('name', '旅行')}风格，建议您重点关注相关的景点和体验。\n\n"
+        # 根据persona调整建议风格
+        if persona.get('key') == 'planner':
+            plan += "📋 建议制定详细的每日行程表，包含具体时间安排和备选方案。\n\n"
+        elif persona.get('key') == 'social':
+            plan += "📸 推荐寻找热门打卡点，准备好相机记录美好时光！\n\n"
+        elif persona.get('key') == 'experiential':
+            plan += "🎨 建议深入当地社区，寻找地道的文化体验和小众景点。\n\n"
         
-        plan += "详细行程规划请稍后重试，或告诉我您的具体需求！"
+        # 预算相关建议
+        if budget.get('type') == 'economy':
+            plan += "💡 经济型贴士：考虑使用公共交通，选择青旅或民宿，寻找当地平价美食。\n\n"
+        elif budget.get('type') == 'luxury':
+            plan += "✨ 豪华体验：可以考虑五星酒店、私人导览和米其林餐厅。\n\n"
         
+        plan += "如需更详细的个性化规划，请告诉我您的具体需求！"
         return plan
 
     def _format_knowledge_context(self, knowledge_items: list) -> str:
-        """格式化知识库内容作为上下文 - 完全安全版本"""
+        """格式化知识库内容作为上下文"""
         if not knowledge_items: 
-            return "没有特定的背景知识。"
+            return "基于一般旅游建议"
         
         try:
             # 只使用最相关的一条知识
             item = knowledge_items[0]
-            
-            # 安全地获取嵌套数据
             knowledge = item.get('knowledge', {})
             travel_knowledge = knowledge.get('travel_knowledge', {})
             
@@ -110,10 +271,6 @@ class ResponseGenerator:
                 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', {})
@@ -126,20 +283,17 @@ class ResponseGenerator:
                 if daily_average:
                     context_parts.append(f"日均预算: {daily_average}")
 
-            # 3. 获取专业见解 - 根据实际数据结构
+            # 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])}")
+                    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', [])
@@ -150,8 +304,8 @@ class ResponseGenerator:
             if context_parts:
                 return "相关旅游知识：\n- " + "\n- ".join(context_parts)
             else:
-                return "基于旅游知识库的一般建议。"
+                return "基于专业旅游建议"
                 
         except Exception as e:
             log.error(f"❌ 格式化知识库上下文失败: {e}", exc_info=True)
-            return "知识库处理时发生错误，使用一般旅游建议。"
+            return "基于一般旅游建议"

modules/travel_assistant.py

@@ -1,45 +1,323 @@
-# modules/travel_assistant.py
-from .config_loader import ConfigLoader
+# modules/response_generator.py - 完整版本（含persona+预算）
+import json
+import os
 from .ai_model import AIModel
 from .knowledge_base import KnowledgeBase
-from .info_extractor import InfoExtractor
-from .session_manager import SessionManager
-from .response_generator import ResponseGenerator
 from utils.logger import log
 
-class TravelAssistant:
-    def __init__(self):
-        # 依赖注入：在这里实例化所有需要的模块
-        log.info("开始初始化 Travel Assistant 核心模块...")
-        self.config = ConfigLoader()
-        self.kb = KnowledgeBase()
-        self.ai_model = AIModel()
-        self.session_manager = SessionManager()
-        self.info_extractor = InfoExtractor(self.config)
-        self.response_generator = ResponseGenerator(self.ai_model, self.kb)
-        log.info("✅ Travel Assistant 核心模块全部初始化完成！")
-
-    def chat(self, message: str, session_id: str, history: list):
-        # 1. 获取或创建会话
-        session_state = self.session_manager.get_or_create_session(session_id)
-        current_session_id = session_state['session_id']
-
-        # 2. 从用户输入中提取信息
-        extracted_info = self.info_extractor.extract(message)
-        
-        # 3. 更新会话状态
-        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. 生成回复
-        bot_response = self.response_generator.generate(message, session_state)
-        
-        # 5. 格式化状态信息用于前端显示
-        status_info = self.session_manager.format_session_info(session_state)
-        
-        # 6. 更新对话历史
-        new_history = history + [[message, bot_response]]
-        
-        return bot_response, current_session_id, status_info, new_history
+class ResponseGenerator:
+    def __init__(self, ai_model: AIModel, knowledge_base: KnowledgeBase):
+        self.ai_model = ai_model
+        self.kb = knowledge_base
+        self.personas = self._load_personas()
+
+    def _load_personas(self):
+        """加载personas配置"""
+        try:
+            personas_path = os.path.join(os.path.dirname(__file__), '..', 'data', 'personas.json')
+            with open(personas_path, 'r', encoding='utf-8') as f:
+                data = json.load(f)
+                return data.get('personas', {})
+        except Exception as e:
+            log.error(f"❌ 加载personas.json失败: {e}")
+            return {}
+
+    def generate(self, user_message: str, session_state: dict) -> str:
+        try:
+            # 【主流程】按顺序收集信息：目的地 → 天数 → 预算 → persona（可选）
+            
+            # 1. 检查是否需要收集目的地
+            if not session_state.get("destination"):
+                return "Hi！你想去欧洲的哪个城市呢？比如巴黎, 罗马, 巴塞罗那？"
+            
+            # 2. 检查是否需要收集天数
+            if not session_state.get("duration"):
+                return f"好的，{session_state['destination']['name']}是个很棒的选择！你计划玩几天呢？"
+            
+            # 3. 检查是否需要收集预算
+            if not session_state.get("budget"):
+                return f"了解！{session_state['duration']['days']}天的行程。为了给您更合适的建议，请问您的预算大概是多少呢？（比如：2000欧元，或者经济型/舒适型/豪华型）"
+            
+            # 4. persona由前端传入，这里检查是否存在
+            # 注意：如果前端没传persona，可以询问或使用默认值
+            if not session_state.get("persona"):
+                separator = '*' * 60
+    
+                persona_selection_message = f"""请告诉我您更偏向哪种旅行风格：高效规划型、社交分享型，还是深度体验型？
+
+        {separator}
+
+        🗓️ **高效规划型** - 注重时间安排和预算控制，喜欢详细的行程规划
+
+        🤝 **社交分享型** - 重视与朋友分享，喜欢拍照打卡和热闹的体验
+
+        🎨 **深度体验型** - 追求地道文化体验，避开商业化景点"""
+                return persona_selection_message
+            
+            # 【增强阶段】信息收集完毕，现在用知识库和persona增强回答质量
+            return self._generate_persona_enhanced_plan(user_message, session_state)
+                
+        except Exception as e:
+            log.error(f"❌ 响应生成失败: {e}", exc_info=True)
+            return "抱歉，我在处理您的请求时遇到了问题，请稍后再试。"
+
+    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生成回答
+        if self.ai_model.is_available():
+            return self.ai_model.generate(user_message, persona_prompt)
+        else:
+            return self._generate_fallback_plan(session_state, knowledge_context)
+
+    def _build_persona_enhanced_prompt(self, session_state: dict, knowledge_context: str = "") -> str:
+        """根据persona构建增强的prompt"""
+        destination = session_state.get("destination", {})
+        duration = session_state.get("duration", {})
+        budget = session_state.get("budget", {})
+        persona = session_state.get("persona", {})
+        
+        # 基础信息
+        location = destination.get('name', '目的地')
+        days = duration.get('days', '几')
+        budget_info = self._format_budget_info(budget)
+        
+        # 获取persona配置
+        persona_key = persona.get('key')
+        if not persona_key or persona_key not in self.personas:
+            # 如果没有有效persona，使用通用prompt
+            return self._build_generic_prompt(session_state, knowledge_context)
+        
+        persona_config = self.personas[persona_key]
+        
+        # 使用personas.json中的prompt_template
+        persona_template = persona_config.get('prompt_template', '')
+        
+        # 替换模板中的变量
+        enhanced_prompt = persona_template.format(
+            location=location,
+            days=days,
+            date="近期",  # 可以从session中获取更具体的日期
+            user_tags="",  # 可以从用户偏好中提取
+            commercial_preference="适中",  # 可以从session中获取
+            group_description="个人/朋友",  # 可以从session中获取
+            budget=budget_info,  # 使用格式化的预算信息
+            tags=""  # 可以从session中获取
+        )
+        
+        # 添加知识库上下文
+        if knowledge_context:
+            enhanced_prompt += f"\n\n【背景知识】\n{knowledge_context}"
+        
+        # 添加预算约束信息
+        enhanced_prompt += f"\n\n【预算约束】\n用户预算：{budget_info}，请确保所有推荐都在预算范围内。"
+        
+        # 添加persona特定的指导原则
+        characteristics = persona_config.get('characteristics', [])
+        if characteristics:
+            enhanced_prompt += f"\n\n【用户特征】\n" + "\n".join([f"- {char}" for char in characteristics])
+        
+        recommendation_strategy = persona_config.get('recommendation_strategy', [])
+        if recommendation_strategy:
+            enhanced_prompt += f"\n\n【推荐策略】\n" + "\n".join([f"- {strategy}" for strategy in recommendation_strategy])
+        
+        tone_guidelines = persona_config.get('tone', [])
+        if tone_guidelines:
+            enhanced_prompt += f"\n\n【语言风格】\n" + "\n".join([f"- {tone}" for tone in tone_guidelines])
+        
+        return enhanced_prompt
+
+    def _format_budget_info(self, budget: dict) -> str:
+        """格式化预算信息"""
+        if not budget:
+            return "中等预算"
+        
+        # 如果有具体金额
+        if budget.get('amount') and budget.get('currency'):
+            return f"{budget['amount']}{budget['currency']}"
+        
+        # 如果有预算类型
+        if budget.get('type'):
+            budget_type_map = {
+                'economy': '经济型预算',
+                'comfortable': '舒适型预算', 
+                'luxury': '豪华型预算'
+            }
+            return budget_type_map.get(budget['type'], budget['type'])
+        
+        # 如果有预算范围
+        if budget.get('range'):
+            return budget['range']
+        
+        return "中等预算"
+
+    def _build_generic_prompt(self, session_state: dict, knowledge_context: str = "") -> str:
+        """构建通用prompt（当没有persona时使用）"""
+        destination = session_state.get("destination", {})
+        duration = session_state.get("duration", {})
+        budget = session_state.get("budget", {})
+        
+        location = destination.get('name', '目的地')
+        days = duration.get('days', '几')
+        budget_info = self._format_budget_info(budget)
+        
+        prompt = f"""你是一个专业的旅游助手。请为用户生成一个详细的旅行计划。
+
+【基本信息】
+- 目的地：{location}
+- 旅行天数：{days}天
+- 预算：{budget_info}
+
+【要求】
+- 提供具体的景点推荐和路线安排
+- 包含交通、住宿、餐饮建议
+- 确保所有推荐都在预算范围内
+- 提供实用的旅行贴士"""
+        
+        if knowledge_context:
+            prompt += f"\n\n【背景信息】\n{knowledge_context}"
+        
+        prompt += "\n\n请生成一份实用、详细的旅行计划。"
+        
+        return prompt
+
+    def _build_search_query(self, session_state: dict) -> str:
+        """构建知识库搜索查询"""
+        destination = session_state.get("destination", {})
+        persona = session_state.get("persona", {})
+        budget = session_state.get("budget", {})
+        
+        query_parts = []
+        
+        # 添加目的地
+        if destination.get('name'):
+            query_parts.append(destination['name'])
+        
+        # 添加persona类型
+        if persona.get('key'):
+            query_parts.append(persona['key'])
+        
+        # 添加预算类型
+        if budget.get('type'):
+            query_parts.append(budget['type'])
+        
+        return " ".join(query_parts) if query_parts else "欧洲旅行"
+
+    def _generate_fallback_plan(self, session_state: dict, knowledge_context: str = "") -> str:
+        """AI不可用时的备用计划生成（现在包含预算和persona信息）"""
+        destination = session_state.get("destination", {})
+        duration = session_state.get("duration", {})
+        budget = session_state.get("budget", {})
+        persona = session_state.get("persona", {})
+        
+        location = destination.get('name', '目的地')
+        days = duration.get('days', '几')
+        budget_info = self._format_budget_info(budget)
+        persona_name = persona.get('name', '旅行者')
+        
+        plan = f"为您推荐 {location} {days}天旅行计划：\n\n"
+        
+        # 用户信息概述
+        plan += f"👤 旅行者类型：{persona_name}\n"
+        plan += f"💰 预算范围：{budget_info}\n\n"
+        
+        # 优先使用知识库信息
+        if knowledge_context and "相关旅游知识" in knowledge_context:
+            plan += f"📚 {knowledge_context}\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"
+        
+        avg_budget = destination.get('avg_daily_budget', 100)
+        plan += f"💰 日均预算参考：约{avg_budget}欧元/天\n\n"
+        
+        # 根据persona调整建议风格
+        if persona.get('key') == 'planner':
+            plan += "📋 建议制定详细的每日行程表，包含具体时间安排和备选方案。\n\n"
+        elif persona.get('key') == 'social':
+            plan += "📸 推荐寻找热门打卡点，准备好相机记录美好时光！\n\n"
+        elif persona.get('key') == 'experiential':
+            plan += "🎨 建议深入当地社区，寻找地道的文化体验和小众景点。\n\n"
+        
+        # 预算相关建议
+        if budget.get('type') == 'economy':
+            plan += "💡 经济型贴士：考虑使用公共交通，选择青旅或民宿，寻找当地平价美食。\n\n"
+        elif budget.get('type') == 'luxury':
+            plan += "✨ 豪华体验：可以考虑五星酒店、私人导览和米其林餐厅。\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}天")
+
+            # 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 "基于一般旅游建议"

utils/validators.py

@@ -1,14 +1,14 @@
-# utils/validators.py
 from pydantic import BaseModel
-from typing import List, Optional
+from typing import List, Optional, Dict, Any
 
 class ChatRequest(BaseModel):
     message: str
     session_id: Optional[str] = None
     history: Optional[List[List[str]]] = []
+    persona_key: Optional[str] = None  # 新增：接收前端传来的persona
 
 class ChatResponse(BaseModel):
     reply: str
     session_id: str
-    status_info: str
-    history: List[List[str]]
+    status_info: Dict[str, Any]
+    history: List[List[str]]