PersonalPublic/gpt_chat_bot
1
0
Fork 0
mirror of https://github.com/Balshgit/gpt_chat_bot.git synced 2025-09-11 22:30:41 +03:00
Code Issues Packages Projects Releases Wiki Activity

fix bot healthcheck endpoint (#22)

Browse Source
This commit is contained in:
Dmitry Afanasyev 2023-09-29 20:56:12 +03:00 committed by GitHub
parent 42f5191042
commit ae6cdb896a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
bot_microservice/api/bot/controllers.py
View File

@ -32,11 +32,12 @@ async def process_bot_updates(request: Request) -> None:
}, },
) )
async def gpt_healthcheck(response: Response) -> Response: async def gpt_healthcheck(response: Response) -> Response:
chat_gpt_service = ChatGptService(chat_gpt_model=settings.GPT_MODEL) chatgpt_service = ChatGptService(chat_gpt_model=settings.GPT_MODEL)
data = chat_gpt_service.build_request_data('Привет!') data = chatgpt_service.build_request_data("Привет!")
response.status_code = status.HTTP_200_OK
try: try:
gpt_response = await chat_gpt_service.do_request(data) chatgpt_response = await chatgpt_service.do_request(data)
if gpt_response.text == INVALID_GPT_MODEL_MESSAGE or response.status_code != status.HTTP_200_OK: if chatgpt_response.status_code != status.HTTP_200_OK or chatgpt_response.text == INVALID_GPT_MODEL_MESSAGE:
response.status_code = status.HTTP_500_INTERNAL_SERVER_ERROR response.status_code = status.HTTP_500_INTERNAL_SERVER_ERROR
except Exception: except Exception:
response.status_code = status.HTTP_500_INTERNAL_SERVER_ERROR response.status_code = status.HTTP_500_INTERNAL_SERVER_ERROR