huangge1199/vue-pro
1
0
Fork 0
mirror of https://gitee.com/huangge1199_admin/vue-pro.git synced 2024-11-30 11:11:55 +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
Show Stats 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 字段 * 关联 AdminUserDO 的 userId 字段
*/ */
private Long userId; private Long userId;
@ -67,7 +67,7 @@ public class AiMusicDO extends BaseDO {
/** /**
* 生成模式 * 生成模式
* * <p>
* 枚举 {@link AiMusicGenerateModeEnum} * 枚举 {@link AiMusicGenerateModeEnum}
*/ */
private Integer generateMode; private Integer generateMode;
@ -113,4 +113,10 @@ public class AiMusicDO extends BaseDO {
*/ */
private String errorMessage; 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() return convertList(musicList, musicData -> new AiMusicDO()
.setTaskId(musicData.id()).setModel(musicData.modelName()) .setTaskId(musicData.id()).setModel(musicData.modelName())
.setPrompt(musicData.prompt()).setGptDescriptionPrompt(musicData.gptDescriptionPrompt()) .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)) .setTitle(musicData.title()).setLyric(musicData.lyric()).setTags(StrUtil.split(musicData.tags(), StrPool.COMMA))
.setStatus(Objects.equals("complete", musicData.status()) ? .setStatus(Objects.equals("complete", musicData.status()) ?
AiMusicStatusEnum.SUCCESS.getStatus() : AiMusicStatusEnum.IN_PROGRESS.getStatus())); AiMusicStatusEnum.SUCCESS.getStatus() : AiMusicStatusEnum.IN_PROGRESS.getStatus()));

16
yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/core/model/suno/api/SunoApi.java
View File

@ -117,7 +117,7 @@ public class SunoApi {
* @param prompt 用于生成音乐音频的提示 * @param prompt 用于生成音乐音频的提示
* @param tags 音乐风格 * @param tags 音乐风格
* @param title 音乐名称 * @param title 音乐名称
* @param mv 模型 * @param model 模型
* @param waitAudio false 表示后台模式，仅返回音频任务信息，需要调用 get API 获取详细的音频信息。 * @param waitAudio false 表示后台模式，仅返回音频任务信息，需要调用 get API 获取详细的音频信息。
* true 表示同步模式，API 最多等待 100s，音频生成完毕后直接返回音频链接等信息，建议在 GPT 等 agent 中使用。 * true 表示同步模式，API 最多等待 100s，音频生成完毕后直接返回音频链接等信息，建议在 GPT 等 agent 中使用。
* @param makeInstrumental 指示音乐音频是否为定制，如果为 true，则从歌词生成，否则从提示生成 * @param makeInstrumental 指示音乐音频是否为定制，如果为 true，则从歌词生成，否则从提示生成
@ -127,7 +127,7 @@ public class SunoApi {
String prompt, String prompt,
String tags, String tags,
String title, String title,
String mv, String model,
@JsonProperty("wait_audio") boolean waitAudio, @JsonProperty("wait_audio") boolean waitAudio,
@JsonProperty("make_instrumental") boolean makeInstrumental @JsonProperty("make_instrumental") boolean makeInstrumental
) { ) {
@ -136,12 +136,12 @@ public class SunoApi {
this(prompt, null, null, null, false, false); this(prompt, null, null, null, false, false);
} }
public MusicGenerateRequest(String prompt, String mv, boolean makeInstrumental) { public MusicGenerateRequest(String prompt, String model, boolean makeInstrumental) {
this(prompt, null, null, mv, false, makeInstrumental); this(prompt, null, null, model, false, makeInstrumental);
} }
public MusicGenerateRequest(String prompt, String mv, String tags, String title) { public MusicGenerateRequest(String prompt, String model, String tags, String title) {
this(prompt, tags, title, mv, false, false); this(prompt, tags, title, model, false, false);
} }
} }
@ -162,6 +162,7 @@ public class SunoApi {
* @param prompt 生成音乐音频的提示 * @param prompt 生成音乐音频的提示
* @param type 操作类型 * @param type 操作类型
* @param tags 音乐类型标签 * @param tags 音乐类型标签
* @param duration 音乐时长
*/ */
public record MusicData( public record MusicData(
String id, String id,
@ -176,7 +177,8 @@ public class SunoApi {
@JsonProperty("gpt_description_prompt") String gptDescriptionPrompt, @JsonProperty("gpt_description_prompt") String gptDescriptionPrompt,
String prompt, String prompt,
String type, String type,
String tags String tags,
Double duration
) { ) {
} }