feat: improve chat feedback navigation

ai_knowledge_base_v2/apps/backend/app/api/feedback.py

@@ -1,7 +1,8 @@
 from __future__ import annotations
 
-from datetime import UTC, date, datetime, timedelta
+from datetime import date, timedelta
 from io import BytesIO
+from typing import Annotated
 
 from fastapi import APIRouter, Depends, HTTPException, Query, status
 from fastapi.responses import StreamingResponse
@@ -18,13 +19,14 @@ from app.api.pagination import page_result
 from app.core.database import get_db
 from app.core.dependencies import get_current_admin, get_current_user
 from app.core.responses import api_success
-from app.core.time_utils import business_day_boundary, to_business_naive
+from app.core.time_utils import business_day_boundary, to_business_naive, utc_now_naive
 from app.models.admin import Admin
 from app.models.chat import ChatMessage, ChatSession
 from app.models.feedback import MessageFeedback
 from app.models.user import User
 from app.services.admin_service import OperationLogService
 from app.services.admin_permission_service import require_permission
+from app.services.chat_message_navigation_service import ChatMessageNavigationService
 
 router = APIRouter()
 
@@ -96,7 +98,13 @@ def export_feedback(
 
 
 @router.get("/admin/{feedback_id}")
-def feedback_detail(feedback_id: int, db: Session = Depends(get_db), admin: Admin = Depends(get_current_admin)) -> dict:
+def feedback_detail(
+    feedback_id: int,
+    messagePage: Annotated[int | None, Query(ge=1)] = None,
+    messagePageSize: Annotated[int, Query(ge=10, le=100)] = 20,
+    db: Session = Depends(get_db),
+    admin: Admin = Depends(get_current_admin),
+) -> dict:
     require_permission(admin, "feedback.detail")
     row = db.execute(select(MessageFeedback, User, ChatMessage, ChatSession).join(User, User.id == MessageFeedback.user_id).join(ChatMessage, ChatMessage.id == MessageFeedback.message_id).join(ChatSession, ChatSession.id == MessageFeedback.session_id).where(MessageFeedback.id == feedback_id)).first()
     if row is None:
@@ -105,10 +113,43 @@ def feedback_detail(feedback_id: int, db: Session = Depends(get_db), admin: Admi
     if not feedback.is_read:
         feedback.is_read = 1
         feedback.read_by = admin.id
-        feedback.read_at = datetime.now(UTC).replace(tzinfo=None)
+        feedback.read_at = utc_now_naive()
         db.commit()
-    messages = db.scalars(select(ChatMessage).where(ChatMessage.session_id == feedback.session_id, ChatMessage.id <= target.id).order_by(ChatMessage.id.asc()).limit(200)).all()
-    return api_success({**_summary(feedback, user, target), "sessionTitle": session.title, "messages": [{"id": m.id, "role": m.role, "content": m.content, "createdAt": m.created_at, "isTarget": m.id == target.id} for m in messages]})
+    message_query = select(ChatMessage).where(ChatMessage.session_id == feedback.session_id)
+    message_total = db.scalar(select(func.count()).select_from(message_query.subquery())) or 0
+    resolved_page = messagePage or ChatMessageNavigationService.page_for_message(
+        db,
+        session_id=feedback.session_id,
+        message=target,
+        page_size=messagePageSize,
+    )
+    messages = db.scalars(
+        message_query
+        .order_by(ChatMessage.created_at.asc(), ChatMessage.id.asc())
+        .offset((resolved_page - 1) * messagePageSize)
+        .limit(messagePageSize)
+    ).all()
+    serialized_messages = [
+        {
+            "id": message.id,
+            "role": message.role,
+            "content": message.content,
+            "createdAt": message.created_at,
+            "isTarget": message.id == target.id,
+        }
+        for message in messages
+    ]
+    return api_success({
+        **_summary(feedback, user, target),
+        "sessionTitle": session.title,
+        "messages": serialized_messages,
+        "messagesPage": page_result(
+            serialized_messages,
+            total=message_total,
+            page=resolved_page,
+            page_size=messagePageSize,
+        ),
+    })
 
 
 @router.delete("/admin/{feedback_id}")