From ebbe667c9dffe730714af0321ddeb36764d04c82 Mon Sep 17 00:00:00 2001 From: harold Date: Sat, 19 Apr 2025 09:35:34 +0500 Subject: [PATCH] add fix for handler exceptions --- gemini_worker/src/service/rmq_msg_handler.py | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) 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 = []