hhyykk/ipms-sjy
1
1
Fork 0
mirror of https://gitee.com/hhyykk/ipms-sjy.git synced 2025-07-25 00:15:06 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

【优化】AI 音乐：生成支持指定模型生成，支持回写音乐时长

Browse Source
This commit is contained in:
xiaoxin
2024-06-28 13:46:56 +08:00
parent bdd7d0c1fd
commit 3acd058635
3 changed files with 18 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/dal/dataobject/music/AiMusicDO.java
View File

@ -30,7 +30,7 @@ public class AiMusicDO extends BaseDO {
/**
* 用户编号
*
* <p>
* 关联 AdminUserDO 的 userId 字段
*/
private Long userId;
@ -67,7 +67,7 @@ public class AiMusicDO extends BaseDO {
/**
* 生成模式
*
* <p>
* 枚举 {@link AiMusicGenerateModeEnum}
*/
private Integer generateMode;
@ -113,4 +113,10 @@ public class AiMusicDO extends BaseDO {
*/
private String errorMessage;
/**
* 音乐时长
*/
private Double duration;
}

2
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/music/AiMusicServiceImpl.java
View File

@ -132,7 +132,7 @@ public class AiMusicServiceImpl implements AiMusicService {
return convertList(musicList, musicData -> new AiMusicDO()
.setTaskId(musicData.id()).setModel(musicData.modelName())
.setPrompt(musicData.prompt()).setGptDescriptionPrompt(musicData.gptDescriptionPrompt())
.setAudioUrl(createFile(musicData.audioUrl())).setVideoUrl(createFile(musicData.videoUrl())).setImageUrl(createFile(musicData.imageUrl()))
.setAudioUrl(createFile(musicData.audioUrl())).setVideoUrl(createFile(musicData.videoUrl())).setImageUrl(createFile(musicData.imageUrl())).setDuration(musicData.duration())
.setTitle(musicData.title()).setLyric(musicData.lyric()).setTags(StrUtil.split(musicData.tags(), StrPool.COMMA))
.setStatus(Objects.equals("complete", musicData.status()) ?
AiMusicStatusEnum.SUCCESS.getStatus() : AiMusicStatusEnum.IN_PROGRESS.getStatus()));