fix queue compute node

2023-09-27 17:53:03 +08:00
parent 01f05285c7
commit e13e31aae9
1 changed files with 2 additions and 2 deletions
@@ -56,7 +56,7 @@ class Queue_ComputeNode(ComputeNode):

        return result

-    async def start(self):
+    def start(self):
        if self.is_start is True:
            return
        self.is_start = True
@@ -65,7 +65,7 @@ class Queue_ComputeNode(ComputeNode):
            while True:
                task = await self.task_queue.get()
                logger.info(f"openai_node get task: {task.display()}")
-                self._run_task(task)
+                await self._run_task(task)

        asyncio.create_task(_run_task_loop())