fix: improve error handling for video generation process

This commit is contained in:
wxg0103 2025-09-19 15:30:37 +08:00
parent 3b2c337e36
commit 60ce4bf70d

View File

@ -104,7 +104,12 @@ class GenerationVideoModel(MaxKBBaseModel, BaseGenerationVideo):
rsp = self._safe_call(VideoSynthesis.wait, task=rsp, api_key=self.api_key)
if rsp.status_code == HTTPStatus.OK:
maxkb_logger.info("视频生成完成!视频 URL:", rsp.output.video_url)
return rsp.output.video_url
if rsp.output.task_status == "SUCCEEDED":
maxkb_logger.info("视频生成完成!视频 URL:", rsp.output.video_url)
return rsp.output.video_url
else:
maxkb_logger.error("视频生成失败!")
raise RuntimeError(f'生成失败, message: {rsp.output.message}')
else:
maxkb_logger.error(f'生成失败status_code: {rsp.status_code}, code: {rsp.code}, message: {rsp.message}')
raise RuntimeError(f'生成失败status_code: {rsp.status_code}, code: {rsp.code}, message: {rsp.message}')