0.5 缩减重构

wolai-backend/app/routers/chat.py

@@ -1,26 +1,26 @@
-import asyncio
-from fastapi import APIRouter
-from fastapi.responses import StreamingResponse
-
-from app.deps import AuthDep
-from app.services.lightrag_service import lightrag_service
-
-router = APIRouter(prefix="/chat")
-
-
-@router.get("")
-async def chat(query: str, document_id: str, auth: AuthDep) -> StreamingResponse:  # noqa: ARG001
-    """
-    SSE 流式占位。后续会调用 LightRAG + OpenAI。
-    当前直接返回 mock 文字，确保前端链路可用。
-    """
-
-    async def event_stream() -> asyncio.AsyncGenerator[str, None]:
-        reply = await lightrag_service.query(query_text=query, user_id="placeholder-user")
-        chunks = [reply[: len(reply) // 2 or 1], reply[len(reply) // 2 or 1 :]]
-        for chunk in chunks:
-            yield f"data: {chunk}\n\n"
-            await asyncio.sleep(0.1)
-        yield "data: [DONE]\n\n"
-
-    return StreamingResponse(event_stream(), media_type="text/event-stream")
+import asyncio
+from fastapi import APIRouter
+from fastapi.responses import StreamingResponse
+
+from app.deps import AuthDep
+from app.services.lightrag_service import lightrag_service
+
+router = APIRouter(prefix="/chat")
+
+
+@router.get("")
+async def chat(query: str, document_id: str, auth: AuthDep) -> StreamingResponse:  # noqa: ARG001
+    """
+    SSE 流式占位。后续会调用 LightRAG + OpenAI。
+    当前直接返回 mock 文字，确保前端链路可用。
+    """
+
+    async def event_stream() -> asyncio.AsyncGenerator[str, None]:
+        reply = await lightrag_service.query(query_text=query, user_id="placeholder-user")
+        chunks = [reply[: len(reply) // 2 or 1], reply[len(reply) // 2 or 1 :]]
+        for chunk in chunks:
+            yield f"data: {chunk}\n\n"
+            await asyncio.sleep(0.1)
+        yield "data: [DONE]\n\n"
+
+    return StreamingResponse(event_stream(), media_type="text/event-stream")