from docarray import Document
from fastapi import APIRouter, HTTPException
from now.executor.gateway.bff.app.v1.models.admin import (
UpdateApiKeysRequestModel,
UpdateEmailsRequestModel,
)
from now.executor.gateway.bff.app.v1.models.shared import BaseRequestModel
from now.executor.gateway.bff.app.v1.routers.helper import jina_client_post
router = APIRouter()
[docs]@router.post(
"/updateUserEmails",
summary='update user emails during runtime',
)
async def update_user_email(data: UpdateEmailsRequestModel):
"""
Update the list of emails for the security executor
"""
try:
await jina_client_post(
request_model=data,
docs=Document(),
endpoint='/admin/updateUserEmails',
parameters={'user_emails': data.user_emails},
)
except BaseException as e:
raise HTTPException(status_code=502, detail=str(e))
return 'SUCCESS'
[docs]@router.post(
"/updateApiKeys",
summary='update api keys during runtime',
)
async def update_api_keys(data: UpdateApiKeysRequestModel):
"""
Update the list of api keys for the security executor
"""
try:
await jina_client_post(
request_model=data,
docs=Document(),
endpoint='/admin/updateApiKeys',
parameters={'api_keys': data.api_keys},
)
except BaseException as e:
raise HTTPException(status_code=502, detail=str(e))
return 'SUCCESS'
[docs]@router.post(
"/getStatus",
summary='Get status of the flow during runtime',
)
async def get_host_status(data: BaseRequestModel):
"""
Get the status of the host in the request body
"""
try:
await jina_client_post(
request_model=data,
docs=Document(),
endpoint='/dry_run',
)
except Exception as e:
raise HTTPException(status_code=502, detail=str(e))
return 'SUCCESS'
