fix : recover file

modules/config_loader.py

@@ -0,0 +1,33 @@
+# modules/config_loader.py
+import json
+from pathlib import Path
+from utils.logger import log
+
+class ConfigLoader:
+    def __init__(self, config_dir: Path = Path("./config")):
+        self.config_dir = config_dir
+        self.cities = {}
+        self.personas = {}
+        self.interests = {}
+        try:
+            self._load_all()
+            log.info("✅ 所有配置文件加载成功")
+        except Exception as e:
+            log.error(f"❌ 配置文件加载失败: {e}", exc_info=True)
+            raise
+
+    def _load_all(self):
+        # 加载城市
+        with open(self.config_dir / "cities.json", 'r', encoding='utf-8') as f:
+            cities_data = json.load(f)
+            for city in cities_data['cities']:
+                for alias in [city['name']] + city.get('aliases', []):
+                    self.cities[alias.lower()] = city
+        
+        # 加载 personas
+        with open(self.config_dir / "personas.json", 'r', encoding='utf-8') as f:
+            self.personas = json.load(f)['personas']
+        
+        # 加载兴趣
+        with open(self.config_dir / "interests.json", 'r', encoding='utf-8') as f:
+            self.interests = json.load(f)['interests']

modules/knowledge_base.py

@@ -0,0 +1,29 @@
+# modules/knowledge_base.py
+import json
+from pathlib import Path
+from utils.logger import log
+
+class KnowledgeBase:
+    def __init__(self, file_path: Path = Path("./config/general_travelplan.json")):
+        self.knowledge = []
+        try:
+            with open(file_path, 'r', encoding='utf-8') as f:
+                self.knowledge = json.load(f).get('clean_knowledge', [])
+            log.info(f"✅ 知识库加载完成")
+        except Exception as e:
+            log.error(f"❌ 知识库加载失败: {e}", exc_info=True)
+            raise
+
+    def search(self, query: str) -> list:
+        relevant_knowledge = []
+        query_lower = query.lower()
+        
+        for item in self.knowledge:
+            # 简单实现：如果查询的城市在知识库的目的地中，则返回该知识
+            destinations = item.get('knowledge', {}).get('travel_knowledge', {}).get('destination_info', {}).get('primary_destinations', [])
+            for dest in destinations:
+                if dest.lower() in query_lower:
+                    if item not in relevant_knowledge:
+                        relevant_knowledge.append(item)
+                    break 
+        return relevant_knowledge

modules/travel_assistant.py

@@ -0,0 +1,45 @@
+# modules/travel_assistant.py
+from .config_loader import ConfigLoader
+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