From 6165fdcc601abcfa0625772dc0c41f45e87bb757 Mon Sep 17 00:00:00 2001 From: Devin Gaffney Date: Wed, 8 Nov 2023 13:32:02 -0800 Subject: [PATCH] fix model reference --- lib/queue/worker.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/queue/worker.py b/lib/queue/worker.py index 3b0f2df..b45bbcb 100644 --- a/lib/queue/worker.py +++ b/lib/queue/worker.py @@ -52,7 +52,7 @@ def safely_respond(self, model: Model) -> List[schemas.Message]: if messages_with_queues: logger.debug(f"About to respond to: ({messages_with_queues})") try: - responses = model.respond([schemas.Message(**{**json.loads(message.body), **{"model_name": self.model.model_name}}) for message, queue in messages_with_queues]) + responses = model.respond([schemas.Message(**{**json.loads(message.body), **{"model_name": model.model_name}}) for message, queue in messages_with_queues]) except Exception as e: logger.error(e) self.delete_messages(messages_with_queues)