diff --git a/gemini_worker/src/service/rmq_msg_handler.py b/gemini_worker/src/service/rmq_msg_handler.py index ff71489..3a90dcc 100644 --- a/gemini_worker/src/service/rmq_msg_handler.py +++ b/gemini_worker/src/service/rmq_msg_handler.py @@ -22,13 +22,15 @@ async def handle_message_service( ): return else: - result = gemini_helper.create_request_ai( - messages=MESSAGES_STORAGE_SCHEMA - ) - sleep(60) - MESSAGES_STORAGE_SCHEMA.chats = [] - if result.success is not None: - await success_publisher.publish(result) - + try: + result = gemini_helper.create_request_ai( + messages=MESSAGES_STORAGE_SCHEMA + ) + sleep(60) + if result.success is not None: + await success_publisher.publish(result) + + finally: + MESSAGES_STORAGE_SCHEMA.chats = []