fix: wrong module name

modules/response_generator.py

@@ -2,6 +2,7 @@ import json
 import random
 from .ai_model import AIModel
 from .knowledge_base import KnowledgeBase
+from .session_manager import SessionManager
 from utils.logger import log
 
 class ResponseGenerator:
@@ -100,7 +101,7 @@ class ResponseGenerator:
             log.error(f"❌ 响应生成失败: {e}", exc_info=True)
             return "抱歉，我在处理您的请求时遇到了问题，请稍后再试。"
         
-    def _generate_vivid_acknowledgement(self, extracted_info: dict, session_state: SessionState) -> str:
+    def _generate_vivid_acknowledgement(self, extracted_info: dict, session_state: SessionManager) -> str:
         """
         【新增实现】根据最新提取的信息，生成一句生动的、非模板化的确认语。
         """
@@ -126,7 +127,7 @@ class ResponseGenerator:
         # 如果没有提取到新信息，返回空字符串
         return ""
 
-    def _get_dynamic_next_question(self, session_state: SessionState) -> str:
+    def _get_dynamic_next_question(self, session_state: SessionManager) -> str:
         """
         【新增实现】根据当前缺失的信息，生成引导性的下一个问题。
         """
@@ -140,6 +141,39 @@ class ResponseGenerator:
             # 这里可以从persona配置中动态生成问题，暂时使用硬编码
             return "您更喜欢哪种旅行风格呢？\n- 🌍 文化探索型\n- 🎨 艺术历史型\n- 🍽️ 美食购物型\n- 🏞️ 自然风光型"
         return "" # 所有信息都已收集
+    
+
+    def _get_destination_name(self, session_state: dict) -> str:
+
+        destination_info = session_state.get('destination')
+        if destination_info and isinstance(destination_info, dict):
+            return destination_info.get('name', '未知目的地')
+        return '未知目的地'
+
+    def _get_duration_days(self, session_state: dict) -> int:
+
+        duration_info = session_state.get('duration')
+        if duration_info and isinstance(duration_info, dict):
+            # 确保即使 'days' 键不存在，也返回一个数字
+            return duration_info.get('days', 0)
+        # 关键：返回 0 而不是 None
+        return 0
+
+    def _format_budget_info(self, budget_data: dict | None) -> str:
+    
+        if budget_data and isinstance(budget_data, dict):
+            # 优先使用现成的描述，因为它最准确
+            if budget_data.get('description'):
+                return budget_data['description']
+        
+            # 如果没有描述，则动态创建一个
+            amount = budget_data.get('amount', '')
+            currency = budget_data.get('currency', '')
+            if amount: # 仅在有金额时才显示
+                return f"{amount} {currency}".strip()
+    
+        # 如果没有预算信息或信息不完整，返回默认字符串
+        return '预算未设定'
 
     def _generate_knowledge_enhanced_plan(self, user_message: str, session_state: dict) -> str:
         """生成融合知识库信息的旅行计划"""