upd: handle user message

epoch8 · Oct 25, 2023 · ea83e49 · ea83e49
1 parent e0dff72
commit ea83e49
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/rasa/core/channels/rest.py b/rasa/core/channels/rest.py
@@ -35,7 +35,7 @@ def name(cls) -> Text:
         return "rest"
 
     def __init__(self):
-        self._background_task = set()
+        self._background_tasks = set()
 
     @staticmethod
     async def on_message_wrapper(
@@ -130,7 +130,7 @@ def _logger_task_status(self, task):
             logger.error(f"{task.get_name()} Canceled Error!")
         except asyncio.InvalidStateError as e:
             logger.error(f"{task.get_name()} Invalid State Error!")
-        return self._background_task.discard
+        return self._background_tasks.discard
 
     def blueprint(
         self, on_new_message: Callable[[UserMessage], Awaitable[None]]
@@ -180,9 +180,9 @@ async def receive(request: Request) -> Union[ResponseStream, HTTPResponse]:
                             input_channel=input_channel,
                             metadata=metadata,
                         )
-                    task: asyncio.Task = asyncio.create_task(on_new_message(user_msg))
-                    self._background_tasks.add(task)
-                    task.add_done_callback(self._logger_task_status(task))
+                    # using the example on line 109
+                    task = asyncio.ensure_future(on_new_message(user_msg))
+                    await task
                 except CancelledError:
                     structlogger.error(
                         "rest.message.received.timeout", text=copy.deepcopy(text)