zhangyong il y a 7 mois
Parent
commit
d914bf91c2
1 fichiers modifiés avec 3 ajouts et 0 suppressions
  1. 3 0
      job.py

+ 3 - 0
job.py

@@ -190,16 +190,19 @@ def run():
         return
     elif google_file_state != 'ACTIVE':
         logger.error('[+] 视频上传状态不为 ACTIVE, 跳过任务')
+        genai.delete_file(google_file)
         return
 
     model = create_model_cache(redis_task=redis_task)
     if isinstance(model, str):
         logger.error('[+] 创建模型失败, 跳过任务')
+        genai.delete_file(google_file)
         return
 
     response = analyze_video(model=model, google_file=google_file, prompt=prompt, redis_task=redis_task)
     if isinstance(response, str):
         logger.error('[+] 获取模型响应失败, 跳过任务')
+        genai.delete_file(google_file)
         return
 
     text = response.text.strip()