Convert message preprocessing to gather

definitive-io · Jan 5, 2024 · 13dd245 · 13dd245
1 parent 56b521f
commit 13dd245
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/packages/openassistants/openassistants/core/assistant.py b/packages/openassistants/openassistants/core/assistant.py
@@ -348,11 +348,13 @@ async def convert_list_message(self, messages, message, idx):
         )
 
     async def pre_process_messages(self, messages):
-        for idx, message in enumerate(messages):
-            if isinstance(message, OpasUserMessage) and isinstance(
-                message.content, list
-            ):
-                self.convert_list_message(messages, message, idx)
+        tasks = [
+            self.convert_list_message(messages, message, idx)
+            for idx, message in enumerate(messages)
+            if isinstance(message, OpasUserMessage)
+            and isinstance(message.content, list)
+        ]
+        await asyncio.gather(*tasks)
         return messages
 
     async def run_chat(