fix model reference

meedan · Nov 8, 2023 · 6165fdc · 6165fdc
1 parent c02449a
commit 6165fdc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)