Skip to content

Commit 60ce4bf

Browse files
committed
fix: improve error handling for video generation process
1 parent 3b2c337 commit 60ce4bf

File tree

1 file changed

+6
-1
lines changed
  • apps/models_provider/impl/aliyun_bai_lian_model_provider/model

1 file changed

+6
-1
lines changed

apps/models_provider/impl/aliyun_bai_lian_model_provider/model/ttv.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,12 @@ def generate_video(self, prompt, negative_prompt=None, first_frame_url=None, las
104104
rsp = self._safe_call(VideoSynthesis.wait, task=rsp, api_key=self.api_key)
105105
if rsp.status_code == HTTPStatus.OK:
106106
maxkb_logger.info("视频生成完成!视频 URL:", rsp.output.video_url)
107-
return rsp.output.video_url
107+
if rsp.output.task_status == "SUCCEEDED":
108+
maxkb_logger.info("视频生成完成!视频 URL:", rsp.output.video_url)
109+
return rsp.output.video_url
110+
else:
111+
maxkb_logger.error("视频生成失败!")
112+
raise RuntimeError(f'生成失败, message: {rsp.output.message}')
108113
else:
109114
maxkb_logger.error(f'生成失败,status_code: {rsp.status_code}, code: {rsp.code}, message: {rsp.message}')
110115
raise RuntimeError(f'生成失败,status_code: {rsp.status_code}, code: {rsp.code}, message: {rsp.message}')

0 commit comments

Comments
 (0)