panyy
/
UnisKB
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: improve error handling for video generation process

v3.2
wxg0103 2025-09-19 15:30:37 +08:00
parent 3b2c337e36
commit 60ce4bf70d
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
apps/models_provider/impl/aliyun_bai_lian_model_provider/model/ttv.py
View File

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