diff --git a/app/main.py b/app/main.py index f4911f5..1982dc7 100644 --- a/app/main.py +++ b/app/main.py @@ -13,7 +13,7 @@ from app.utils import get_password_hash, verify_password import app.serializers as serializers # Import all serializers (detailed in __init__.py) # Import models -from app.models import User, Pin, Friend, Token, TokenData, HTTPError +from app.models import User, Friend, Token, TokenData, HTTPError # Import all DTOs (detailed in __init__.py) from app.dto import FriendAddDTO, FriendListDTO, UserDTO, UserRegisterDTO, PinDTO @@ -218,6 +218,7 @@ async def add_friend(friend_to_add: FriendAddDTO, current_user: User = Depends(g ) friend["user_id"] = current_user.uid + friend["status"] = "pending" friend_id = friends_collection.insert_one(friend).inserted_id return {"id": str(friend_id)}