fix 文档管理模块 & 法规对话模块

backend/app/api/routes/rag.py

@@ -9,73 +9,74 @@ from typing import AsyncGenerator
 from fastapi import APIRouter
 from fastapi.responses import StreamingResponse
 
+from app.config.settings import settings
 from app.schemas.rag import RagChatRequest, QuickQuestionsResponse, QuickQuestion
-from app.services.mock_data import (
-    get_mock_quick_questions,
-    get_mock_retrieval,
-    get_mock_rag_answer,
-)
-# Keep route handlers close to their transport-layer wiring for easier auditing.
+from app.shared.bootstrap import get_agent_conversation_service
 
 
 router = APIRouter(prefix="/rag", tags=["RAG问答"])
 
+_DEFAULT_QUICK_QUESTIONS = [
+    {"id": "1", "question": "请总结最新入库法规对电池安全的核心要求", "category": "法规解读"},
+    {"id": "2", "question": "我上传的制度文档与新能源法规有哪些潜在冲突？", "category": "差距分析"},
+    {"id": "3", "question": "请给出法规依据，并按条款列出整改建议", "category": "整改建议"},
+    {"id": "4", "question": "请解释 UN-ECE 与 GB 标准在网络安全方面的差异", "category": "标准对比"},
+    {"id": "5", "question": "IATF 16949 对供应商质量管理有哪些强制要求？", "category": "法规解读"},
+    {"id": "6", "question": "ISO 45001 与 AQ 标准在职业健康安全方面的主要差异是什么？", "category": "标准对比"},
+]
+
 
 @router.post("/chat")
 async def rag_chat(request: RagChatRequest):
-    """Handle rag chat."""
+    """Stream RAG Q&A using the real agent service."""
+    _, event_stream = get_agent_conversation_service().stream_chat(
+        query=request.query,
+        session_id=request.session_id,
+        filters=request.filters,
+        top_k=request.top_k or settings.rag_top_k,
+    )
 
     async def generate() -> AsyncGenerator[str, None]:
-        # Keep route handlers close to their transport-layer wiring for easier auditing.
-        """Handle generate."""
-        yield f"event: message\ndata: {json.dumps({'type': 'retrieving'}, ensure_ascii=False)}\n\n"
-
-        # Keep route handlers close to their transport-layer wiring for easier auditing.
-        await asyncio.sleep(0.3)
-
-        # Keep route handlers close to their transport-layer wiring for easier auditing.
-        docs = get_mock_retrieval(request.query, top_k=request.top_k)
-
-        retrieved_data = [
-            {
-                "id": d["id"],
-                "score": d["score"],
-                "preview": d["preview"],
-                "doc_name": d.get("doc_name", ""),
-                "clause": d.get("clause", ""),
-            }
-            for d in docs
-        ]
-        yield f"event: message\ndata: {json.dumps({'type': 'retrieved', 'docs': retrieved_data}, ensure_ascii=False)}\n\n"
-
-        # Keep route handlers close to their transport-layer wiring for easier auditing.
-        yield (
-            f"event: message\ndata: "
-            f"{json.dumps({'type': 'generating', 'text': '正在生成答案...'}, ensure_ascii=False)}\n\n"
-        )
-
-        # Keep route handlers close to their transport-layer wiring for easier auditing.
-        await asyncio.sleep(0.2)
-
-        # Keep route handlers close to their transport-layer wiring for easier auditing.
-        answer = get_mock_rag_answer(request.query)
-
-        # Keep route handlers close to their transport-layer wiring for easier auditing.
-        sentences = answer.split("\n\n")
-        for sentence in sentences:
-            if sentence.strip():
-                # Keep route handlers close to their transport-layer wiring for easier auditing.
-                chunks = sentence.split("\n")
-                for chunk in chunks:
-                    if chunk.strip():
-                        await asyncio.sleep(0.05)  # Keep route handlers close to their transport-layer wiring for easier auditing.
-                        yield (
-                            "event: message\n"
-                            f"data: {json.dumps({'type': 'chunk', 'text': chunk + chr(10)}, ensure_ascii=False)}\n\n"
-                        )
-
-        # Keep route handlers close to their transport-layer wiring for easier auditing.
-        yield f"event: message\ndata: {json.dumps({'type': 'done'}, ensure_ascii=False)}\n\n"
+        """Translate agent SSE events to rag format."""
+        for event in event_stream:
+            event_type = event.get("event", "")
+            data = event.get("data", "")
+            if event_type == "sources":
+                docs = [
+                    {
+                        "id": str(s.get("chunk_id") or s.get("doc_id") or idx + 1),
+                        "score": s.get("score", 0),
+                        "preview": s.get("content", "")[:200],
+                        "doc_name": s.get("doc_name", ""),
+                        "clause": s.get("section_title", "法规片段"),
+                        "doc_id": s.get("doc_id"),
+                        "download_url": (
+                            f"/api/v1/documents/download/{s['doc_id']}" if s.get("doc_id") else None
+                        ),
+                    }
+                    for idx, s in enumerate(data if isinstance(data, list) else [])
+                ]
+                yield (
+                    "event: message\n"
+                    f"data: {json.dumps({'type': 'retrieved', 'docs': docs}, ensure_ascii=False)}\n\n"
+                )
+            elif event_type == "content":
+                if data:
+                    yield (
+                        "event: message\n"
+                        f"data: {json.dumps({'type': 'chunk', 'text': data}, ensure_ascii=False)}\n\n"
+                    )
+            elif event_type == "done":
+                yield (
+                    "event: message\n"
+                    f"data: {json.dumps({'type': 'done'}, ensure_ascii=False)}\n\n"
+                )
+            elif event_type == "status":
+                yield (
+                    "event: message\n"
+                    f"data: {json.dumps({'type': 'status', 'text': data}, ensure_ascii=False)}\n\n"
+                )
+            await asyncio.sleep(0)
 
     return StreamingResponse(
         generate(),
@@ -86,9 +87,13 @@ async def rag_chat(request: RagChatRequest):
 
 @router.get("/quick-questions", response_model=QuickQuestionsResponse)
 async def get_quick_questions():
-    """Return quick questions."""
-    questions = [
-        QuickQuestion(id=q["id"], question=q["question"], category=q["category"])
-        for q in get_mock_quick_questions()
-    ]
+    """Return configurable quick questions from settings or defaults."""
+    raw = getattr(settings, "rag_quick_questions", None)
+    if raw and isinstance(raw, list):
+        questions = [
+            QuickQuestion(id=str(i + 1), question=q if isinstance(q, str) else q.get("question", ""), category=q.get("category", "法规问答") if isinstance(q, dict) else "法规问答")
+            for i, q in enumerate(raw)
+        ]
+    else:
+        questions = [QuickQuestion(**q) for q in _DEFAULT_QUICK_QUESTIONS]
     return QuickQuestionsResponse(questions=questions)