phase1-final

backend/python/app/api/profile.py

@@ -1,3 +1,4 @@
+from app.api.chat_service import get_user_balance
 from fastapi import APIRouter, Depends, HTTPException, Query
 from sqlalchemy.orm import Session
 from sqlalchemy import and_, desc, asc
@@ -38,6 +39,12 @@ class UpdateApiKeyRequest(BaseModel):
     status: str  # enabled, disabled, revoked
 
 
+
+@router.get("/balance")
+async def get_balance(user=Depends(get_current_user)):
+    balance = await get_user_balance(user["id"])
+    return {"code": 0, "data": balance, "message": "ok"}
+
 @router.get("/")
 def get_profile(
     db: Session = Depends(get_db),