feat: 添加Docker配置、依赖管理和测试框架

- 添加Dockerfile支持容器化部署
- 完善requirements.txt包含所有必要依赖
- 创建完整的测试框架和单元测试
- 支持HuggingFace Spaces自动构建和部署
- 添加健康检查和测试覆盖

Dockerfile

@@ -0,0 +1,32 @@
+# Dockerfile
+FROM python:3.10-slim
+
+# 设置工作目录
+WORKDIR /app
+
+# 设置环境变量
+ENV PYTHONPATH=/app
+ENV PYTHONUNBUFFERED=1
+
+# 安装系统依赖
+RUN apt-get update && apt-get install -y \
+    git \
+    curl \
+    && rm -rf /var/lib/apt/lists/*
+
+# 复制requirements文件并安装Python依赖
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# 复制项目文件
+COPY . .
+
+# 暴露端口
+EXPOSE 7860
+
+# 健康检查
+HEALTHCHECK --interval=30s --timeout=30s --start-period=5s --retries=3 \
+    CMD curl -f http://localhost:7860/health || exit 1
+
+# 启动命令
+CMD ["python", "-m", "uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

requirements.txt

@@ -0,0 +1,29 @@
+# requirements.txt
+# FastAPI 和 Web 服务
+fastapi==0.104.1
+uvicorn[standard]==0.24.0
+pydantic==2.5.0
+python-multipart==0.0.6
+
+# AI 模型和机器学习
+torch==2.1.0
+transformers==4.54.1
+accelerate==0.25.0
+safetensors==0.4.1
+
+# 图像处理
+Pillow==10.1.0
+requests==2.31.0
+
+# 音频处理（为将来扩展准备）
+# librosa==0.10.1
+# soundfile==0.12.1
+
+# 工具库
+numpy==1.24.3
+typing-extensions==4.8.0
+
+# 测试框架
+pytest==7.4.3
+pytest-asyncio==0.21.1
+httpx==0.25.2

tests/__init__.py

@@ -0,0 +1,7 @@
+# tests/__init__.py
+"""
+测试模块包
+包含所有组件的单元测试和集成测试
+"""
+
+__version__ = '1.0.0'

tests/test_ai_model.py

@@ -0,0 +1,56 @@
+# tests/test_ai_model.py
+import pytest
+from unittest.mock import Mock, patch
+from modules.ai_model import AIModel
+
+class TestAIModel:
+    
+    @pytest.fixture
+    def mock_ai_model(self):
+        """创建模拟的AI模型"""
+        with patch('modules.ai_model.Gemma3nForConditionalGeneration') as mock_model, \
+             patch('modules.ai_model.AutoProcessor') as mock_processor:
+            
+            mock_model.from_pretrained.return_value = Mock()
+            mock_processor.from_pretrained.return_value = Mock()
+            
+            ai_model = AIModel()
+            yield ai_model
+    
+    def test_detect_input_type_text(self, mock_ai_model):
+        """测试文本输入检测"""
+        assert mock_ai_model.detect_input_type("我想去巴黎") == "text"
+    
+    def test_detect_input_type_image_url(self, mock_ai_model):
+        """测试图片URL检测"""
+        assert mock_ai_model.detect_input_type("https://example.com/image.jpg") == "image"
+    
+    def test_detect_input_type_image_path(self, mock_ai_model):
+        """测试图片路径检测"""
+        assert mock_ai_model.detect_input_type("./images/paris.png") == "image"
+    
+    def test_detect_input_type_audio(self, mock_ai_model):
+        """测试音频文件检测"""
+        assert mock_ai_model.detect_input_type("audio.mp3") == "audio"
+    
+    def test_is_available(self, mock_ai_model):
+        """测试模型可用性检查"""
+        assert mock_ai_model.is_available() == True
+    
+    @patch('modules.ai_model.Image')
+    @patch('modules.ai_model.requests')
+    def test_format_input_image_url(self, mock_requests, mock_image, mock_ai_model):
+        """测试图片URL格式化"""
+        mock_response = Mock()
+        mock_response.content = b"fake_image_data"
+        mock_requests.get.return_value = mock_response
+        mock_image.open.return_value.convert.return_value = "processed_image"
+        
+        input_type, formatted_data, processed_text = mock_ai_model.format_input(
+            "image", 
+            "https://example.com/image.jpg"
+        )
+        
+        assert input_type == "image"
+        assert formatted_data == "processed_image"
+        assert "请描述这张图片" in processed_text

tests/test_config_loader.py

@@ -0,0 +1,57 @@
+# tests/test_config_loader.py
+import pytest
+import json
+import tempfile
+from pathlib import Path
+from modules.config_loader import ConfigLoader
+
+class TestConfigLoader:
+    
+    @pytest.fixture
+    def temp_config_dir(self):
+        """创建临时配置目录和文件"""
+        with tempfile.TemporaryDirectory() as tmp_dir:
+            config_dir = Path(tmp_dir)
+            
+            # 创建测试配置文件
+            cities_data = {
+                "cities": [
+                    {"name": "巴黎", "country": "法国", "aliases": ["paris"]}
+                ]
+            }
+            with open(config_dir / "cities.json", 'w', encoding='utf-8') as f:
+                json.dump(cities_data, f, ensure_ascii=False)
+            
+            personas_data = {
+                "personas": {
+                    "planner": {"name": "规划型", "keywords": ["规划", "安排"]}
+                }
+            }
+            with open(config_dir / "personas.json", 'w', encoding='utf-8') as f:
+                json.dump(personas_data, f, ensure_ascii=False)
+            
+            interests_data = {
+                "interests": {"美食": ["美食", "餐厅"]}
+            }
+            with open(config_dir / "interests.json", 'w', encoding='utf-8') as f:
+                json.dump(interests_data, f, ensure_ascii=False)
+            
+            yield config_dir
+    
+    def test_load_cities(self, temp_config_dir):
+        """测试城市配置加载"""
+        loader = ConfigLoader(temp_config_dir)
+        assert "巴黎" in loader.cities
+        assert "paris" in loader.cities
+        assert loader.cities["巴黎"]["name"] == "巴黎"
+    
+    def test_load_personas(self, temp_config_dir):
+        """测试人格配置加载"""
+        loader = ConfigLoader(temp_config_dir)
+        assert "planner" in loader.personas
+        assert loader.personas["planner"]["name"] == "规划型"
+    
+    def test_load_interests(self, temp_config_dir):
+        """测试兴趣配置加载"""
+        loader = ConfigLoader(temp_config_dir)
+        assert "美食" in loader.interests

tests/test_travel_assistant.py

@@ -0,0 +1,52 @@
+# tests/test_travel_assistant.py
+import pytest
+from unittest.mock import Mock, patch
+from modules.travel_assistant import TravelAssistant
+
+class TestTravelAssistant:
+    
+    @pytest.fixture
+    def mock_travel_assistant(self):
+        """创建模拟的旅游助手"""
+        with patch('modules.travel_assistant.ConfigLoader') as mock_config, \
+             patch('modules.travel_assistant.KnowledgeBase') as mock_kb, \
+             patch('modules.travel_assistant.AIModel') as mock_ai:
+            
+            # 设置模拟对象
+            mock_config.return_value.cities = {"巴黎": {"name": "巴黎", "country": "法国"}}
+            mock_config.return_value.personas = {"planner": {"name": "规划型"}}
+            
+            assistant = TravelAssistant()
+            yield assistant
+    
+    def test_chat_basic_flow(self, mock_travel_assistant):
+        """测试基本聊天流程"""
+        reply, session_id, status_info, history = mock_travel_assistant.chat(
+            message="我想去巴黎旅游",
+            session_id=None,
+            history=[]
+        )
+        
+        assert isinstance(reply, str)
+        assert len(session_id) == 8  # UUID前8位
+        assert isinstance(status_info, str)
+        assert len(history) == 1
+        assert history[0][0] == "我想去巴黎旅游"
+    
+    def test_session_persistence(self, mock_travel_assistant):
+        """测试会话持久性"""
+        # 第一次对话
+        _, session_id, _, _ = mock_travel_assistant.chat(
+            message="我想去巴黎",
+            session_id=None,
+            history=[]
+        )
+        
+        # 第二次对话使用相同session_id
+        reply, same_session_id, _, _ = mock_travel_assistant.chat(
+            message="3天行程",
+            session_id=session_id,
+            history=[]
+        )
+        
+        assert session_id == same_session_id