消息推送API追加用户积分数据

video-gen-api/app/api/v1/notifications.py

@@ -8,7 +8,7 @@ from sqlalchemy.ext.asyncio import AsyncSession
 from app.dependencies import get_db, get_current_user
 from app.models.user import User
 from app.models.notification import Notification
-from app.schemas.notification import NotificationOut, NotificationListOut, UnreadCountOut
+from app.schemas.notification import NotificationListOut, NotificationCreditsOut, UnreadCountOut
 from app.services.auth import decode_access_token
 from app.services.notification import (
     get_notifications,
@@ -112,7 +112,8 @@ async def list_notifications(
     db: AsyncSession = Depends(get_db),
 ):
     items, total = await get_notifications(db, current_user.id, page, page_size, is_read)
-    return {"items": items, "total": total}
+    credits = NotificationCreditsOut(balance=round(float(current_user.credits or 0.0), 2))
+    return {"items": items, "total": total, "credits": credits}
 
 
 @router.put("/{notification_id}/read")