mirror of
https://github.com/QuivrHQ/quivr.git
synced 2024-12-18 11:51:41 +03:00
8d54187713
# Description Please include a summary of the changes and the related issue. Please also include relevant motivation and context. ## Checklist before requesting a review Please delete options that are not relevant. - [ ] My code follows the style guidelines of this project - [ ] I have performed a self-review of my code - [ ] I have commented hard-to-understand areas - [ ] I have ideally added tests that prove my fix is effective or that my feature works - [ ] New and existing unit tests pass locally with my changes - [ ] Any dependent changes have been merged ## Screenshots (if appropriate): --------- Co-authored-by: Stan Girard <girard.stanislas@gmail.com>
114 lines
3.5 KiB
Python
114 lines
3.5 KiB
Python
from fastapi import APIRouter, Depends, Request
|
|
from middlewares.auth import AuthBearer, get_current_user
|
|
from modules.brain.service.brain_user_service import BrainUserService
|
|
from modules.user.dto.inputs import UserUpdatableProperties
|
|
from modules.user.entity.user_identity import UserIdentity
|
|
from modules.user.repository.users import Users
|
|
from modules.user.service.user_usage import UserUsage
|
|
|
|
user_router = APIRouter()
|
|
brain_user_service = BrainUserService()
|
|
user_repository = Users()
|
|
|
|
|
|
@user_router.get("/user", dependencies=[Depends(AuthBearer())], tags=["User"])
|
|
async def get_user_endpoint(
|
|
request: Request, current_user: UserIdentity = Depends(get_current_user)
|
|
):
|
|
"""
|
|
Get user information and statistics.
|
|
|
|
- `current_user`: The current authenticated user.
|
|
- Returns the user's email, maximum brain size, current brain size, maximum requests number, requests statistics, and the current date.
|
|
|
|
This endpoint retrieves information and statistics about the authenticated user. It includes the user's email, maximum brain size,
|
|
current brain size, maximum requests number, requests statistics, and the current date. The brain size is calculated based on the
|
|
user's uploaded vectors, and the maximum brain size is obtained from the environment variables. The requests statistics provide
|
|
information about the user's API usage.
|
|
"""
|
|
|
|
user_daily_usage = UserUsage(
|
|
id=current_user.id,
|
|
email=current_user.email,
|
|
)
|
|
user_settings = user_daily_usage.get_user_settings()
|
|
max_brain_size = user_settings.get("max_brain_size", 1000000000)
|
|
|
|
monthly_chat_credit = user_settings.get("monthly_chat_credit", 10)
|
|
|
|
user_daily_usage = UserUsage(id=current_user.id)
|
|
|
|
return {
|
|
"email": current_user.email,
|
|
"max_brain_size": max_brain_size,
|
|
"current_brain_size": 0,
|
|
"monthly_chat_credit": monthly_chat_credit,
|
|
"models": user_settings.get("models", []),
|
|
"id": current_user.id,
|
|
"is_premium": user_settings["is_premium"],
|
|
}
|
|
|
|
|
|
@user_router.put(
|
|
"/user/identity",
|
|
dependencies=[Depends(AuthBearer())],
|
|
tags=["User"],
|
|
)
|
|
def update_user_identity_route(
|
|
user_identity_updatable_properties: UserUpdatableProperties,
|
|
current_user: UserIdentity = Depends(get_current_user),
|
|
) -> UserIdentity:
|
|
"""
|
|
Update user identity.
|
|
"""
|
|
return user_repository.update_user_properties(
|
|
current_user.id, user_identity_updatable_properties
|
|
)
|
|
|
|
|
|
@user_router.get(
|
|
"/user/identity",
|
|
dependencies=[Depends(AuthBearer())],
|
|
tags=["User"],
|
|
)
|
|
def get_user_identity_route(
|
|
current_user: UserIdentity = Depends(get_current_user),
|
|
) -> UserIdentity:
|
|
"""
|
|
Get user identity.
|
|
"""
|
|
return user_repository.get_user_identity(current_user.id)
|
|
|
|
@user_router.delete(
|
|
"/user_data",
|
|
dependencies=[Depends(AuthBearer())],
|
|
tags=["User"],
|
|
)
|
|
async def delete_user_data_route(
|
|
current_user: UserIdentity = Depends(get_current_user),
|
|
):
|
|
"""
|
|
Delete a user.
|
|
|
|
- `user_id`: The ID of the user to delete.
|
|
|
|
This endpoint deletes a user from the system.
|
|
"""
|
|
|
|
user_repository.delete_user_data(current_user.id)
|
|
|
|
return {"message": "User deleted successfully"}
|
|
|
|
@user_router.get(
|
|
"/user/credits",
|
|
dependencies=[Depends(AuthBearer())],
|
|
tags=["User"],
|
|
)
|
|
def get_user_credits(
|
|
current_user: UserIdentity = Depends(get_current_user),
|
|
) -> int:
|
|
"""
|
|
Get user remaining credits.
|
|
"""
|
|
return user_repository.get_user_credits(current_user.id)
|