mirror of
https://github.com/Balshgit/gpt_chat_bot.git
synced 2025-09-10 17:20:41 +03:00
1e79c981c2
* remove fastapi users dependency * add user service to chatbot service * add user save on bot info command * add user model to admin * fix tests
50 lines
1.8 KiB
Python
50 lines
1.8 KiB
Python
from fastapi import APIRouter, Depends
|
|
from fastapi.responses import ORJSONResponse
|
|
from starlette import status
|
|
from starlette.responses import Response
|
|
|
|
from api.bot.deps import get_chatgpt_service
|
|
from api.exceptions import BaseAPIException
|
|
from constants import INVALID_GPT_REQUEST_MESSAGES
|
|
from core.bot.services import ChatGptService
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get(
|
|
"/healthcheck",
|
|
name="system:healthcheck",
|
|
status_code=status.HTTP_200_OK,
|
|
summary="Healthcheck service",
|
|
)
|
|
async def healthcheck() -> ORJSONResponse:
|
|
return ORJSONResponse(content=None, status_code=status.HTTP_200_OK)
|
|
|
|
|
|
@router.get(
|
|
"/bot-healthcheck",
|
|
name="system:gpt_healthcheck",
|
|
response_class=Response,
|
|
summary="Проверяет доступность моделей и если они недоступны, то возвращает код ответа 500",
|
|
responses={
|
|
status.HTTP_500_INTERNAL_SERVER_ERROR: {"description": "Request to chat gpt not success"},
|
|
status.HTTP_200_OK: {"description": "Successful Response"},
|
|
},
|
|
)
|
|
async def gpt_healthcheck(
|
|
response: Response,
|
|
chatgpt_service: ChatGptService = Depends(get_chatgpt_service),
|
|
) -> Response:
|
|
response.status_code = status.HTTP_200_OK
|
|
try:
|
|
chatgpt_response = await chatgpt_service.request_to_chatgpt_microservice(question="Привет!")
|
|
if chatgpt_response.status_code != status.HTTP_200_OK:
|
|
response.status_code = status.HTTP_500_INTERNAL_SERVER_ERROR
|
|
for message in INVALID_GPT_REQUEST_MESSAGES:
|
|
if message in chatgpt_response.text:
|
|
response.status_code = status.HTTP_500_INTERNAL_SERVER_ERROR
|
|
except BaseAPIException:
|
|
response.status_code = status.HTTP_500_INTERNAL_SERVER_ERROR
|
|
|
|
return Response(status_code=response.status_code, content=None)
|