diff --git a/video-gen-api/app/services/upload_queue.py b/video-gen-api/app/services/upload_queue.py index 73ce61f0..ba9d51cf 100644 --- a/video-gen-api/app/services/upload_queue.py +++ b/video-gen-api/app/services/upload_queue.py @@ -15,6 +15,7 @@ from app.models.pre_test_template import PreTestTemplate from app.utils.id_gen import generate_id from app.utils.douyinApi import DouyinApi from app.utils.logger import get_logger +import os import hashlib import json @@ -27,7 +28,6 @@ class UploadQueue: def __init__(self): self.queue: asyncio.Queue[str] = asyncio.Queue() self.running = False - async def enqueue(self, task_id: str): """Add a task to the queue.""" await self.queue.put(task_id) @@ -622,4 +622,4 @@ async def _update_material_pre_test_status( ) -upload_queue = UploadQueue() \ No newline at end of file +upload_queue = UploadQueue()