添加聊天管理功能

Author: User

backend/logging_config.py

@@ -62,7 +62,23 @@ def setup_logging():
     root_logger.addHandler(console_handler)
 
     # 设置第三方库的日志级别
+    # 配置Uvicorn日志
     logging.getLogger("uvicorn").setLevel(logging.INFO)
+    logging.getLogger("uvicorn.access").setLevel(logging.INFO)
+    logging.getLogger("uvicorn.error").setLevel(logging.INFO)
+    
+    # 过滤掉无效HTTP请求的警告（通常来自扫描器或格式错误的请求）
+    class InvalidRequestFilter(logging.Filter):
+        """过滤无效HTTP请求的警告"""
+        def filter(self, record):
+            # 过滤掉"Invalid HTTP request received"警告
+            if "Invalid HTTP request received" in str(record.getMessage()):
+                return False
+            return True
+    
+    # 应用过滤器到uvicorn.error日志记录器
+    invalid_request_filter = InvalidRequestFilter()
+    logging.getLogger("uvicorn.error").addFilter(invalid_request_filter)
     logging.getLogger("fastapi").setLevel(logging.INFO)
     logging.getLogger("sqlalchemy.engine").setLevel(logging.WARNING)
     logging.getLogger("chromadb").setLevel(logging.WARNING)

backend/main.py

@@ -1,7 +1,9 @@
-from fastapi import FastAPI, HTTPException, Depends, UploadFile, File, Form
+from fastapi import FastAPI, HTTPException, Depends, UploadFile, File, Form, Request
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import JSONResponse
+from fastapi.responses import JSONResponse, Response
 from fastapi.staticfiles import StaticFiles
+from starlette.middleware.base import BaseHTTPMiddleware
+from starlette.types import Message
 import sys
 import os
 import json
@@ -59,6 +61,30 @@
     allow_headers=["*"],
 )
 
+# 添加中间件来静默处理常见请求，减少日志噪音
+class SilentCommonRequestsMiddleware(BaseHTTPMiddleware):
+    """静默处理常见请求（favicon、robots.txt等），减少日志噪音"""
+    
+    async def dispatch(self, request: Request, call_next):
+        path = request.url.path
+        
+        # 静默处理的路径列表
+        silent_paths = [
+            "/favicon.ico",
+            "/robots.txt",
+            "/.well-known/security.txt",
+            "/.well-known/",
+        ]
+        
+        # 如果是静默路径，直接返回空响应
+        if any(path.startswith(silent) for silent in silent_paths):
+            return Response(status_code=204)  # No Content
+        
+        # 继续处理其他请求
+        return await call_next(request)
+
+app.add_middleware(SilentCommonRequestsMiddleware)
+
 # 文件存储配置（使用项目根目录）
 UPLOAD_DIR = Path(project_root) / "uploads"
 UPLOAD_DIR.mkdir(exist_ok=True)
@@ -165,6 +191,29 @@ async def root():
         "plugins": plugin_stats
     }
 
+@app.get("/favicon.ico")
+async def favicon():
+    """处理favicon请求，返回空响应"""
+    return Response(status_code=204)
+
+@app.get("/robots.txt")
+async def robots():
+    """处理robots.txt请求"""
+    return Response(
+        content="User-agent: *\nDisallow: /",
+        media_type="text/plain",
+        status_code=200
+    )
+
+@app.get("/.well-known/security.txt")
+async def security_txt():
+    """处理security.txt请求"""
+    return Response(
+        content="# Security Policy\nContact: security@example.com\n",
+        media_type="text/plain",
+        status_code=200
+    )
+
 @app.post("/chat", response_model=ChatResponse)
 async def chat(request: ChatRequest):
     """聊天接口"""

backend/routers/chat.py

@@ -177,6 +177,33 @@ async def get_user_sessions(user_id: str, limit: int = 50):
         raise HTTPException(status_code=500, detail=str(e))
 
 
+@router.get("/users/{user_id}/sessions/search")
+async def search_user_sessions(user_id: str, keyword: str = "", limit: int = 50):
+    """搜索用户会话"""
+    try:
+        result = await chat_service.search_user_sessions(user_id, keyword, limit)
+        return result
+    except Exception as e:
+        logger.error(f"搜索用户会话错误: {e}")
+        raise HTTPException(status_code=500, detail=str(e))
+
+
+@router.post("/sessions/batch-delete")
+async def delete_sessions_batch(session_ids: List[str]):
+    """批量删除会话"""
+    try:
+        if not session_ids:
+            raise HTTPException(status_code=400, detail="会话ID列表不能为空")
+        
+        result = await chat_service.delete_sessions_batch(session_ids)
+        return result
+    except HTTPException:
+        raise
+    except Exception as e:
+        logger.error(f"批量删除会话错误: {e}")
+        raise HTTPException(status_code=500, detail=str(e))
+
+
 @router.delete("/sessions/{session_id}")
 async def delete_session(session_id: str):
     """删除会话"""

backend/services/chat_service.py

@@ -534,6 +534,141 @@ async def delete_session(self, session_id: str) -> bool:
             print(f"删除会话失败: {e}")
             return False
 
+    async def search_user_sessions(
+        self,
+        user_id: str,
+        keyword: str = "",
+        limit: int = 50
+    ) -> Dict[str, Any]:
+        """
+        搜索用户会话
+        
+        Args:
+            user_id: 用户ID
+            keyword: 搜索关键词
+            limit: 限制数量
+            
+        Returns:
+            会话列表
+        """
+        try:
+            with DatabaseManager() as db:
+                from backend.database import ChatMessage
+                
+                sessions = db.get_user_sessions(user_id, limit * 2)  # 获取更多以便筛选
+                
+                session_list = []
+                keyword_lower = keyword.lower() if keyword else ""
+                
+                for session in sessions:
+                    # 检查会话是否有消息
+                    message_count = db.db.query(ChatMessage)\
+                        .filter(ChatMessage.session_id == session.session_id)\
+                        .count()
+                    
+                    if message_count == 0:
+                        continue
+                    
+                    # 获取会话的第一条消息作为标题
+                    first_message = db.db.query(ChatMessage)\
+                        .filter(ChatMessage.session_id == session.session_id)\
+                        .filter(ChatMessage.role == 'user')\
+                        .order_by(ChatMessage.created_at.asc())\
+                        .first()
+                    
+                    # 获取会话的最后一条消息作为预览
+                    last_message = db.db.query(ChatMessage)\
+                        .filter(ChatMessage.session_id == session.session_id)\
+                        .order_by(ChatMessage.created_at.desc())\
+                        .first()
+                    
+                    title = first_message.content[:30] + "..." if first_message and len(first_message.content) > 30 else (first_message.content if first_message else "新对话")
+                    
+                    # 生成预览文本
+                    preview = ""
+                    if last_message:
+                        preview = last_message.content[:50] + "..." if len(last_message.content) > 50 else last_message.content
+                    
+                    # 如果有关键词，进行搜索过滤
+                    if keyword_lower:
+                        title_lower = title.lower()
+                        preview_lower = preview.lower()
+                        if keyword_lower not in title_lower and keyword_lower not in preview_lower:
+                            continue
+                    
+                    session_list.append({
+                        "session_id": session.session_id,
+                        "title": title,
+                        "preview": preview,
+                        "message_count": message_count,
+                        "created_at": session.created_at.isoformat() if session.created_at else None,
+                        "updated_at": session.updated_at.isoformat() if session.updated_at else None
+                    })
+                    
+                    # 限制返回数量
+                    if len(session_list) >= limit:
+                        break
+                
+                return {
+                    "user_id": user_id,
+                    "sessions": session_list,
+                    "total": len(session_list),
+                    "keyword": keyword
+                }
+        except Exception as e:
+            print(f"搜索用户会话失败: {e}")
+            return {
+                "user_id": user_id,
+                "sessions": [],
+                "total": 0,
+                "keyword": keyword,
+                "error": str(e)
+            }
+    
+    async def delete_sessions_batch(self, session_ids: List[str]) -> Dict[str, Any]:
+        """
+        批量删除会话
+        
+        Args:
+            session_ids: 会话ID列表
+            
+        Returns:
+            删除结果
+        """
+        try:
+            success_count = 0
+            failed_count = 0
+            failed_sessions = []
+            
+            for session_id in session_ids:
+                try:
+                    success = await self.delete_session(session_id)
+                    if success:
+                        success_count += 1
+                    else:
+                        failed_count += 1
+                        failed_sessions.append(session_id)
+                except Exception as e:
+                    failed_count += 1
+                    failed_sessions.append(session_id)
+                    print(f"删除会话 {session_id} 失败: {e}")
+            
+            return {
+                "success_count": success_count,
+                "failed_count": failed_count,
+                "failed_sessions": failed_sessions,
+                "total": len(session_ids)
+            }
+        except Exception as e:
+            print(f"批量删除会话失败: {e}")
+            return {
+                "success_count": 0,
+                "failed_count": len(session_ids),
+                "failed_sessions": session_ids,
+                "total": len(session_ids),
+                "error": str(e)
+            }
+    
     async def get_user_emotion_trends(self, user_id: str, days: int = 7) -> Dict[str, Any]:
         """
         获取用户情绪趋势

frontend/src/App.js

@@ -5,6 +5,7 @@ import ChatContainer from './components/ChatContainer';
 import FeedbackModal from './components/FeedbackModal';
 import PersonalizationPanel from './components/PersonalizationPanel';
 import StyleComparison from './components/StyleComparison';
+import HistoryManagementModal from './components/HistoryManagementModal';
 import { motion } from 'framer-motion';
 import { X } from 'lucide-react';
 import { useChat, useFileUpload, useKeyboard, useSession, useFeedback, useURLDetection } from './hooks';
@@ -25,6 +26,7 @@ function App() {
   const [inputValue, setInputValue] = useState('');
   const [showPersonalizationPanel, setShowPersonalizationPanel] = useState(false);
   const [showStyleComparison, setShowStyleComparison] = useState(false);
+  const [showHistoryManagement, setShowHistoryManagement] = useState(false);
 
   // Refs
   const inputRef = useRef(null);
@@ -112,6 +114,24 @@ function App() {
     deleteConversationHook(targetSessionId, sessionId, setMessages, setSessionId, setSuggestions);
   }, [deleteConversationHook, sessionId, setSessionId, setMessages, setSuggestions]);
 
+  // 处理历史消息管理中的会话选择
+  const handleHistorySessionSelect = useCallback((targetSessionId) => {
+    loadSessionHistory(targetSessionId, setMessages, setSuggestions);
+    setSessionId(targetSessionId);
+  }, [loadSessionHistory, setMessages, setSuggestions, setSessionId]);
+
+  // 处理批量删除后的回调
+  const handleSessionsDeleted = useCallback((deletedSessionIds) => {
+    // 如果当前会话被删除，清空消息
+    if (deletedSessionIds.includes(sessionId)) {
+      setMessages([]);
+      setSessionId(null);
+      setSuggestions([]);
+    }
+    // 刷新历史会话列表
+    loadHistorySessions();
+  }, [sessionId, setMessages, setSessionId, setSuggestions, loadHistorySessions]);
+
   // 键盘处理
   const { handleKeyPress, handleTabNavigation } = useKeyboard(
     startNewChat,
@@ -185,6 +205,7 @@ function App() {
         onDeleteSession={handleDeleteSession}
         onOpenPersonalization={() => setShowPersonalizationPanel(true)}
         onOpenStyleComparison={() => setShowStyleComparison(true)}
+        onOpenHistoryManagement={() => setShowHistoryManagement(true)}
       />
 
       <ChatContainer
@@ -226,6 +247,14 @@ function App() {
         onCommentChange={setFeedbackComment}
         onSubmit={submitFeedback}
       />
+
+      <HistoryManagementModal
+        show={showHistoryManagement}
+        onClose={() => setShowHistoryManagement(false)}
+        userId={currentUserId}
+        onSessionSelect={handleHistorySessionSelect}
+        onSessionsDeleted={handleSessionsDeleted}
+      />
     </AppContainer>
   );
 }