From ea1449fa7f4273e9953e15e126c50f7cdf3ad211 Mon Sep 17 00:00:00 2001 From: Daniel McKnight <34697904+NeonDaniel@users.noreply.github.com> Date: Fri, 17 Nov 2023 18:36:24 -0800 Subject: [PATCH] Fix context handling to move `routing_key` to `mq` context (#49) Co-authored-by: Daniel McKnight --- neon_messagebus_mq_connector/controller.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/neon_messagebus_mq_connector/controller.py b/neon_messagebus_mq_connector/controller.py index d52eb5a..0220150 100644 --- a/neon_messagebus_mq_connector/controller.py +++ b/neon_messagebus_mq_connector/controller.py @@ -129,16 +129,15 @@ def handle_neon_message(self, message: Message): if not message.data: message.data['msg'] = 'Failed to get response from Neon' message.context.setdefault('klat_data', {}) + message.context.setdefault('mq', {}) if message.msg_type == 'neon.get_tts.response': body = self.format_response(response_type=NeonResponseTypes.TTS, message=message) - message.context['klat_data'].setdefault('routing_key', - 'neon_tts_response') + message.context['mq'].setdefault('routing_key', 'neon_tts_response') elif message.msg_type == 'neon.get_stt.response': body = self.format_response(response_type=NeonResponseTypes.STT, message=message) - message.context['klat_data'].setdefault('routing_key', - 'neon_stt_response') + message.context['mq'].setdefault('routing_key', 'neon_stt_response') else: body = {'msg_type': message.msg_type, 'data': message.data, 'context': message.context} @@ -168,7 +167,7 @@ def handle_neon_profile_update(self, message: Message): to avoid publishing private profile values to a shared queue :param message: Message containing the updated user profile """ - if message.context.get('klat_data', {}).get('routing_key'): + if message.context.get('mq', {}).get('routing_key'): LOG.info(f"handling profile update for " f"user={message.data['profile']['user']['username']}") self.handle_neon_message(message)