From 8ad20f1e9b6a0f0c42c09805f474a2f4264245f4 Mon Sep 17 00:00:00 2001 From: Roger Gonzalez Date: Sun, 9 Apr 2023 16:24:44 -0300 Subject: Added error handling to chatgpt and dall-e --- bot.py | 29 ++++++++++++++++++++--------- 1 file changed, 20 insertions(+), 9 deletions(-) diff --git a/bot.py b/bot.py index 7f7f3f1..36348d2 100644 --- a/bot.py +++ b/bot.py @@ -214,9 +214,16 @@ async def chatgpt(room, message): return {"role": "user", "content": message} personal_conversation.append(format_message(message_content)) - completion = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=personal_conversation) - personal_conversation.append(completion.choices[0].message) - await bot.api.send_text_message(room_id, completion.choices[0].message.content) + + try: + completion = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=personal_conversation) + response = completion.choices[0].message.content + personal_conversation.append(completion.choices[0].message) + except Exception as e: + print(f"Error: {e}") + response = "There was a problem with your prompt" + + await bot.api.send_text_message(room_id, response) @bot.listener.on_message_event @@ -257,12 +264,16 @@ async def dall_e(room, message): print(f"Room: {room_id}, User: {user}, Message: dalle") await bot.api.send_text_message(room_id, "Generating image...") - image = openai.Image.create(prompt=message) - image_url = image["data"][0]["url"] - image_filename = wget.download(image_url) - - await bot.api.send_image_message(room_id, image_filename) - os.remove(image_filename) + try: + image = openai.Image.create(prompt=message) + image_url = image["data"][0]["url"] + image_filename = wget.download(image_url) + + await bot.api.send_image_message(room_id, image_filename) + os.remove(image_filename) + except Exception as e: + print(f"Error sending image: {e}") + await bot.api.send_text_message(room_id, f"Error sending image: {e}") bot.run() -- cgit v1.2.3