用户系统

art-agent/backend/app/api/chat.py

@@ -3,11 +3,13 @@ import uuid
 from pathlib import Path
 from typing import Optional
 
-from fastapi import APIRouter, File, Form, UploadFile
+from fastapi import APIRouter, Depends, File, Form, UploadFile
 from sse_starlette.sse import EventSourceResponse
 
 from app.agent.loop import run_agent_loop
+from app.auth import get_current_user
 from app.config import get_image_models_list, get_default_image_model_id
+from app.db import User
 
 router = APIRouter()
 
@@ -27,6 +29,7 @@ async def _save_upload(file: UploadFile) -> str:
 @router.post("/upload-ref-image")
 async def upload_ref_image(
     file: UploadFile = File(...),
+    current_user: User = Depends(get_current_user),
 ):
     """
     独立的参考图上传端点。
@@ -38,7 +41,7 @@ async def upload_ref_image(
 
 
 @router.get("/models")
-async def list_models():
+async def list_models(current_user: User = Depends(get_current_user)):
     """返回可用的图像生成模型列表。"""
     return {
         "models": get_image_models_list(),
@@ -53,6 +56,7 @@ async def chat(
     ref_image_url: Optional[str] = Form(None),
     image_model: Optional[str] = Form(None),
     session_id: Optional[str] = Form(None),
+    current_user: User = Depends(get_current_user),
 ):
     """
     主对话端点。
@@ -78,6 +82,7 @@ async def chat(
             resolved_ref_url,
             image_model=image_model,
             session_id=session_id,
+            user_id=current_user.id,
         ):
             yield {
                 "event": event["type"],