highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-04-11 05:19:52 +02:00
Code Issues Packages Projects Releases Wiki Activity

DAN-52 Check user role endpoint (#28)

Browse Source
This commit is contained in:
Yuhong Sun 2023-05-10 20:48:59 -07:00 committed by GitHub
parent 73d83b648e
commit 632a643b7a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend/danswer/server/models.py
View File

@ -6,6 +6,10 @@ class ServerStatus(BaseModel):
status: str
class UserRoleResponse(BaseModel):
role: str
class QAQuestion(BaseModel):
query: str
collection: str

6
backend/danswer/server/search_backend.py
View File

@ -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():