Update code in endpoints/login.post.py

endpoints/login.post.py

@@ -2,17 +2,17 @@ from fastapi import APIRouter, Depends, HTTPException, status
 from sqlalchemy.orm import Session
 from core.database import get_db
 from models.user import User
-from schemas.user import UserSchema, UserLogin
+from schemas.user import UserLogin, TokenResponse
 from helpers.auth_helpers import authenticate_user, create_access_token
 
 router = APIRouter()
 
-@router.post("/login", status_code=200)
+@router.post("/login", status_code=200, response_model=TokenResponse)
 async def login(
-    user_credentials: UserLogin,
+    credentials: UserLogin,
     db: Session = Depends(get_db)
 ):
-    user = authenticate_user(db, user_credentials.email, user_credentials.password)
+    user = authenticate_user(db, credentials.email, credentials.password)
     if not user:
         raise HTTPException(
             status_code=status.HTTP_401_UNAUTHORIZED,
@@ -21,9 +21,4 @@ async def login(
         )
     
     access_token = create_access_token(data={"sub": user.email})
-    
+    return {"access_token": access_token, "token_type": "bearer"}
-    return {
-        "access_token": access_token,
-        "token_type": "bearer",
-        "user": user
-    }