zhangyong 7 bulan lalu
induk
melakukan
d914bf91c2
1 mengubah file dengan 3 tambahan dan 0 penghapusan
  1. 3 0
      job.py

+ 3 - 0
job.py

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