"""对话 API

POST /api/chat/stream - SSE 流式对话
"""

import json
import logging
import traceback

from fastapi import APIRouter, Depends
from fastapi.responses import StreamingResponse
from pydantic import BaseModel

from app.core.deps import get_current_user
from app.db.error_logger import log_error
from app.llm.chat_agent import chat_stream

logger = logging.getLogger(__name__)
router = APIRouter(prefix="/api/chat", tags=["chat"])


class ChatMessage(BaseModel):
    role: str
    content: str


class ChatRequest(BaseModel):
    messages: list[ChatMessage]


@router.post("/stream")
async def chat_stream_endpoint(req: ChatRequest, current_user: dict = Depends(get_current_user)):
    """流式对话接口（SSE）"""
    messages = [{"role": m.role, "content": m.content} for m in req.messages]

    async def event_generator():
        try:
            async for msg in chat_stream(messages, current_user=current_user):
                data = json.dumps(msg, ensure_ascii=False)
                yield f"data: {data}\n\n"
            yield "data: [DONE]\n\n"
        except Exception as e:
            logger.error(f"Chat stream error: {e}")
            await log_error(
                "chat",
                f"Chat stream error: {e}",
                detail=traceback.format_exc(),
                context={"method": "POST", "path": "/api/chat/stream"},
            )
            error_data = json.dumps(
                {"type": "content", "content": f"出错了: {e}"},
                ensure_ascii=False,
            )
            yield f"data: {error_data}\n\n"
            yield "data: [DONE]\n\n"

    return StreamingResponse(
        event_generator(),
        media_type="text/event-stream",
        headers={
            "Cache-Control": "no-cache",
            "Connection": "keep-alive",
            "X-Accel-Buffering": "no",
        },
    )