FIX AutoDL调度修复暖池功能

This commit is contained in:
2025-04-17 12:00:50 +08:00
parent 1272a33718
commit bf4c66fc27
3 changed files with 27 additions and 12 deletions

View File

@@ -1,4 +1,5 @@
import time
import traceback
import uuid
from concurrent.futures.thread import ThreadPoolExecutor
from typing import Union
@@ -18,6 +19,8 @@ class RunningPool:
def run(self, instance_id, uid, base_url:str, video_file:Union[UploadFile,str], audio_file:Union[UploadFile,str]) -> Union[bool, None]:
try:
code = self._req(base_url, video_file, audio_file)
if not code:
raise Exception("[Code] is None")
except Exception as e:
loguru.logger.error("%s Task Submit Failed: %s" % (uid, e))
return None
@@ -41,6 +44,7 @@ class RunningPool:
return {"status": resp.json()['status'], "msg": resp.json()['msg'],
"instance_id": self.tasks[uid]["instance_id"]}
except Exception as e:
traceback.print_exc()
loguru.logger.error("%s Get Result Failed: %s" % (uid, e))
return None
else: