Jelajahi Sumber

错误的时候添加了更多的信息

ChenYu 7 bulan lalu
induk
melakukan
359a488388
1 mengubah file dengan 6 tambahan dan 1 penghapusan
  1. 6 1
      app/service/task/impl/keling_image2video_task.py

+ 6 - 1
app/service/task/impl/keling_image2video_task.py

@@ -48,7 +48,12 @@ class KelingImage2VideoTask(BaseTask):
                 logger.info("[%s] Submit response: status=%d", self.get_type(), resp.status_code)
                 bean = resp.json()
         except Exception as e:
-            raise RuntimeError(f"Request failed: {e}")
+            if remaining > 0:
+                logger.warning("[%s] Request error: %s, retrying in 5s ... remaining=%d",
+                               self.get_type(), e, remaining - 1)
+                time.sleep(5)
+                return self._retry_execute(param, remaining - 1)
+            raise RuntimeError(f"Request failed after retries: {e}")
 
         code = bean.get("code", 0)
         if code in (1302, 1303) and remaining > 0: