improve: chat logic

modules/response_generator.py

@@ -1,340 +1,154 @@
 import json
-import os
 from .ai_model import AIModel
-from .knowledge_base import KnowledgeBase
+from .session_manager import SessionState # 假设 session_state 是一个类
 from utils.logger import log
+import re
 
 class ResponseGenerator:
-    def __init__(self, ai_model: AIModel, knowledge_base: KnowledgeBase):
+    def __init__(self, ai_model: AIModel):
         self.ai_model = ai_model
-        self.kb = knowledge_base
         self.personas = self._load_personas()
 
-    def _load_personas(self):
-        """加载personas配置"""
-        personas_path = "./config/personas.json"  # 简化路径
-        with open(personas_path, 'r', encoding='utf-8') as f:
+    def _load_personas(self) -> dict:
+        
+        with open("./config/personas.json", 'r', encoding='utf-8') as f:
             data = json.load(f)
+            log.info(f"✅ 成功加载 {len(data.get('personas', {}))} 个 persona。")
             return data.get('personas', {})
-
-    def generate(self, user_message: str, session_state: dict, extracted_info: dict) -> str:
-        """生成智能响应 - 修复重复询问问题"""
+    
+    def generate(self, session_state: SessionState) -> str:
+        """
+        根据当前会话状态，生成一个完全符合选定 Persona 风格的智能回复。
+        """
         try:
-            response_parts = []
-
-            # 1. 生成对新提取信息的确认反馈
-            acknowledgement = self._generate_acknowledgement(extracted_info)
-            if acknowledgement:
-                response_parts.append(acknowledgement)
+            persona_key = session_state.persona or "planner" # 如果未选择，则默认为 planner
+            persona_config = self.personas.get(persona_key, self.personas.get("planner"))
 
-            # 2. 检查信息完整性，按优先级询问缺失信息
-            next_question = self._get_next_question(session_state)
+            # 1. 检查信息完整性，并使用 persona 的风格来提问
+            next_question = self._get_next_question(session_state, persona_config)
             if next_question:
-                response_parts.append(next_question)
+                # 如果是第一个问题（目的地未知），直接返回问题
+                if not session_state.destination:
+                    return next_question
+                # 否则，先确认已知信息，再提问
+                else:
+                    ack = self._generate_acknowledgement(session_state, persona_config)
+                    return f"{ack} {next_question}"
+
+            # 2. 如果所有信息都齐全，则生成最终的旅行计划
+            log.info(f"✅ 所有信息已集齐，开始为 persona '{persona_key}' 生成最终计划。")
+            plan = self._generate_persona_enhanced_plan(session_state, persona_config)
+            return plan
 
-            # 3. 如果所有信息都齐全，生成旅行计划
-            if not next_question:
-                plan = self._generate_persona_enhanced_plan(user_message, session_state)
-                response_parts.append(plan)
-            
-            # 组合回复
-            return " ".join(response_parts) if response_parts else "我理解了，请继续告诉我您的需求。"
-        
         except Exception as e:
-            log.error(f"❌ 响应生成失败: {e}", exc_info=True)
-            return "抱歉，我在处理您的请求时遇到了问题，请稍后再试。"
-
-    def _get_next_question(self, session_state: dict) -> str:
-        """获取下一个需要询问的问题 - 修复版"""
-        
-        # 按顺序检查缺失信息
-        if not session_state.get("destination"):
-            return "你想去欧洲的哪个城市呢？比如巴黎、罗马、巴塞罗那？"
-        
-        if not session_state.get("duration"):
-            destination_name = session_state.get('destination', {}).get('name', '目的地')
-            return f"好的，{destination_name}是个很棒的选择！你计划玩几天呢？"
-        
-        if not session_state.get("budget"):
-            days = session_state.get('duration', {}).get('days', '几')
-            return f"了解！{days}天的行程。预算大概多少呢？（比如：2000欧元，或经济型/舒适型/豪华型）"
-        
-        # 所有信息都已收集完毕
-        return ""
-
-    def _generate_acknowledgement(self, extracted_info: dict) -> str:
-        """生成确认信息 - 简化版"""
-        if not extracted_info:
-            return ""
-        
-        ack_parts = []
-        
-        if "destination" in extracted_info:
-            name = extracted_info['destination'].get('name', '目的地')
-            ack_parts.append(f"目的地已设置为{name}")
-        
-        if "duration" in extracted_info:
-            days = extracted_info['duration'].get('days', '几')
-            ack_parts.append(f"行程天数已设置为{days}天")
-        
-        if "budget" in extracted_info:
-            budget_desc = self._format_budget_info(extracted_info['budget'])
-            ack_parts.append(f"预算已设置为{budget_desc}")
-        
-        if not ack_parts:
-            return ""
-        
-        return "好的，" + "，".join(ack_parts) + "。"
+            log.error(f"❌ 在响应生成过程中发生未知错误: {e}", exc_info=True)
+            return "抱歉，我的思绪好像出了一点小问题，我们能重新开始吗？"
+
+    def _get_next_question(self, session_state: SessionState, persona_config: dict) -> str:
+        """根据缺失信息，生成符合 persona 语气的下一个问题。"""
+        
+        persona_tone_style = f"语气({persona_config.get('tone', ['友好'])[0]})"
+
+        if not session_state.destination:
+            # 根据不同 persona 定制提问方式
+            if persona_config.get('name', '').startswith("高效"):
+                return "您好，为了开始规划，请首先提供您的目的地城市。"
+            elif persona_config.get('name', '').startswith("社交"):
+                return "哈喽！准备好开启一场超棒的旅行了吗？快告诉我你们想去哪里呀？目的地是哪里呀？🌍"
+            else: # 默认或体验型
+                return "您好！很高兴能为您服务。请问这次您渴望探索哪个目的地呢？"
+
+        if not session_state.duration:
+            dest_name = session_state.destination
+            if persona_config.get('name', '').startswith("高效"):
+                return f"目的地已确认为 {dest_name}。请提供计划的旅行总天数。"
+            elif persona_config.get('name', '').startswith("社交"):
+                return f"哇！{dest_name}！听起来就超赞的！你们打算一起玩几天呀？🥳"
+            else:
+                return f"{dest_name}，一个充满故事的地方。您希望在这片土地上沉浸几天呢？"
 
-    def _generate_persona_enhanced_plan(self, user_message: str, session_state: dict) -> str:
-        """生成最终旅行计划"""
-        
-        # 获取基本信息
-        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', '旅行者')
-        
-        # 如果AI可用，生成详细计划
-        if self.ai_model.is_available():
-            prompt = self._build_prompt(session_state)
-            return self.ai_model.generate(user_message, prompt)
+        if not session_state.budget:
+            if persona_config.get('name', '').startswith("高效"):
+                return "为了进行精确的预算分配，请明确您的预算等级（例如：经济型、舒适型、豪华型）或大致的总金额。"
+            elif persona_config.get('name', '').startswith("社交"):
+                return "最后一个问题！这次出去玩的预算大概是多少呀？是小资轻奢还是尽情嗨皮呢？💰"
+            else:
+                return "为了让体验更贴合您的期望，可以分享一下这次旅行的预算范围吗？"
+        
+        return "" # 所有信息都已收集完毕
+
+    def _generate_acknowledgement(self, session_state: SessionState, persona_config: dict) -> str:
+        """对刚刚更新的信息，生成一个符合 persona 语气的确认。"""
+        # 这个函数可以在 session_state 更新后被调用，以提供即时反馈。
+        # 在当前简化流程中，它被整合到提问环节。
+        dest = session_state.destination
+        days = session_state.duration
+        
+        if persona_config.get('name', '').startswith("高效"):
+            if days:
+                return f"已记录：行程共计 {days} 天。"
+            return f"收到。目的地：{dest}。"
+        elif persona_config.get('name', '').startswith("社交"):
+            if days:
+                return f"收到啦！{days} 天的行程，一定能玩得超开心！"
+            return f"好嘞！目的地 {dest} 已锁定！"
         else:
-            # 备用计划
-            return f"""✈️ 为您制定{location}{days}天旅行计划
-
-👤 旅行风格：{persona_name}
-💰 预算：{budget_info}
-
-🎯 主要景点：{destination.get('highlights', '经典景点等您探索')}
-
-📍 建议路线：根据您的{persona_name}风格，为您推荐最适合的行程安排。
-
-如需详细规划，请稍后重试或告诉我具体需求！"""
-
-    def _build_prompt(self, session_state: dict) -> str:
-        """构建AI提示词"""
-        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)
-        
-        # 基础prompt
-        prompt = f"请为{location}制定{days}天旅行计划，预算{budget_info}。"
-        
-        # 添加persona风格
-        persona_key = persona.get('key')
-        if persona_key in self.personas:
-            persona_config = self.personas[persona_key]
-            template = persona_config.get('prompt_template', '')
-            if template:
-                try:
-                    prompt = template.format(
-                        location=location,
-                        days=days,
-                        budget=budget_info,
-                        date="近期",
-                        user_tags="",
-                        commercial_preference="适中",
-                        group_description="个人",
-                        tags=""
-                    )
-                except:
-                    # 如果模板格式化失败，使用基础prompt
-                    pass
-        
-        return 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('description'):
-            return budget['description']
-        
-        if budget.get('type'):
-            type_map = {
-                'economy': '经济型',
-                'comfortable': '舒适型', 
-                'luxury': '豪华型'
-            }
-            return type_map.get(budget['type'], budget['type'])
-        
-        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)
+            if days:
+                return f"明白了，一段为期 {days} 天的旅程。"
+            return f"好的，我们将围绕 {dest} 展开探索。"
+    
+    def _generate_persona_enhanced_plan(self, session_state: SessionState, persona_config: dict) -> str:
+        """调用AI模型，使用 persona 模板生成最终的旅行计划。"""
+        prompt = self._build_prompt(session_state, persona_config)
+        
+        log.info(f"🧠 正在使用 persona '{persona_config.get('name')}' 的模板生成计划...")
+        # log.debug(f"最终生成的Prompt:\n---\n{prompt}\n---") # 如果需要调试，可以取消此行注释
+
+        # 实际调用AI模型
+        # plan = self.ai_model.generate(prompt)
+        # return plan
+
+        # --- 当前用于演示的临时返回 ---
+        # 它会清晰地展示我们是如何根据persona构建prompt的
+        return f"✨ **计划生成预览** ✨\n\n我将使用下面的指令为您生成计划：\n\n```prompt\n{prompt}\n```"
+
+
+    def _build_prompt(self, session_state: SessionState, persona_config: dict) -> str:
+        """
+        【核心升级】
+        一个更健壮、更灵活的 prompt 构建方法。
+        它会安全地填充模板，即使 session_state 中的信息不完整。
+        """
+        template = persona_config.get('prompt_template')
+        if not template:
+            log.warning(f"Persona '{persona_config.get('name')}' 缺少 'prompt_template'，将使用通用模板。")
+            # 此处可以返回一个通用的后备 prompt
+            return f"请为我规划一个去 {session_state.destination} 的 {session_state.duration} 天行程。"
+
+        # 1. 从 session_state 收集所有可能用到的信息
+        context = {
+            "location": session_state.destination,
+            "days": session_state.duration,
+            "budget": session_state.budget,
+            "date": session_state.get_specific_info('date', '近期'), # 假设 session_state 有此方法
+            "user_tags": ", ".join(session_state.get_specific_info('tags', [])),
+            "commercial_preference": session_state.get_specific_info('commercial', '适中'),
+            "group_description": session_state.get_specific_info('group', '个人'),
+            "tags": ", ".join(session_state.get_specific_info('tags', []))
+        }
+
+        # 2. 使用正则表达式查找模板中所有的 {placeholder}
+        placeholders = re.findall(r'\{(\w+)\}', template)
+
+        # 3. 用 context 中的值替换模板中的占位符
+        # 这种方法比 .format(**context) 更安全，因为它只替换存在的占位符，
+        # 并且如果 context 中缺少某个键，它不会抛出 KeyError。
+        for key in placeholders:
+            value = context.get(key)
+            if value is not None:
+                template = template.replace(f'{{{key}}}', str(value))
             else:
-                return "基于专业旅游建议"
-                
-        except Exception as e:
-            log.error(f"❌ 格式化知识库上下文失败: {e}", exc_info=True)
-            return "基于一般旅游建议"
+                # 如果 context 中没有这个值，可以选择留空或使用默认值
+                template = template.replace(f'{{{key}}}', '（未提供）')
+        
+        return template