diff --git a/backend/danswer/server/models.py b/backend/danswer/server/models.py index b35711bd8..bf0ce3a3e 100644 --- a/backend/danswer/server/models.py +++ b/backend/danswer/server/models.py @@ -6,6 +6,10 @@ class ServerStatus(BaseModel): status: str +class UserRoleResponse(BaseModel): + role: str + + class QAQuestion(BaseModel): query: str collection: str diff --git a/backend/danswer/server/search_backend.py b/backend/danswer/server/search_backend.py index 2e56668b5..0eab6b5ae 100644 --- a/backend/danswer/server/search_backend.py +++ b/backend/danswer/server/search_backend.py @@ -17,6 +17,7 @@ from danswer.server.models import QAQuestion from danswer.server.models import QAResponse from danswer.server.models import ServerStatus from danswer.server.models import UserByEmail +from danswer.server.models import UserRoleResponse from danswer.utils.clients import TSClient from danswer.utils.logging import setup_logger from fastapi import APIRouter @@ -49,6 +50,11 @@ def test_endpoint(request: Request): print(request) +@router.get("/get-user-role", response_model=UserRoleResponse) +async def get_user_role(user: User = Depends(current_active_user)): + return UserRoleResponse(role=user.role) + + @router.get("/", response_model=ServerStatus) @router.get("/status", response_model=ServerStatus) def read_server_status():