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

【解决todo】SunoResp、SunoReq改为record简化

Browse Source
This commit is contained in:
xiaoxin
2024-05-30 16:56:49 +08:00
parent a858eb84e5
commit c0de6cc508
6 changed files with 98 additions and 120 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -3,7 +3,6 @@ package cn.iocoder.yudao.framework.ai.core.model.suno.api;
import cn.iocoder.yudao.framework.ai.core.model.suno.SunoConfig;
import com.fasterxml.jackson.annotation.JsonInclude;
import com.fasterxml.jackson.annotation.JsonProperty;
import lombok.Data;
import lombok.extern.slf4j.Slf4j;
import org.springframework.ai.openai.api.ApiUtils;
import org.springframework.web.reactive.function.client.WebClient;
@ -48,128 +47,69 @@ public class SunoApi {
.block();
}
// TODO @xiaoxin：看看是不是使用 record 特性，简化下；
/**
* 请求数据对象，用于生成音乐音频
* 请求数据对象，用于生成音乐音频。
*
* @param prompt 用于生成音乐音频的提示
* @param lyric 用于生成音乐音频的歌词
* @param custom 指示音乐音频是否为定制，如果为 true，则从歌词生成，否则从提示生成
* @param title 音乐音频的标题
* @param style 音乐音频的风格
* @param callbackUrl 音乐音频生成后回调的 URL
*/
@Data
@JsonInclude(value = JsonInclude.Include.NON_NULL)
public static class SunoReq {
/**
* 用于生成音乐音频的提示
*/
private String prompt;
public record SunoReq(
String prompt,
String lyric,
boolean custom,
String title,
String style,
String callbackUrl
) {
public SunoReq(String prompt) {
this(prompt, null, false, null, null, null);
}
/**
* 用于生成音乐音频的歌词
*/
private String lyric;
/**
* 指示音乐音频是否为定制，如果为 true，则从歌词生成，否则从提示生成
*/
private boolean custom;
/**
* 音乐音频的标题
*/
private String title;
/**
* 音乐音频的风格
*/
private String style;
/**
* 音乐音频生成后回调的 URL
*/
private String callbackUrl;
}
// TODO @xiaoxin：看看是不是使用 record 特性，简化下；
/**
* SunoAPI 响应的数据
* SunoAPI 响应的数据。
*
* @param success 表示请求是否成功
* @param taskId 任务 ID
* @param data 音乐数据列表
*/
@Data
public static class SunoResp {
public record SunoResp(
boolean success,
@JsonProperty("task_id") String taskId,
List<MusicData> data
) {
/**
* 表示请求是否成功
* 单个音乐数据。
*
* @param id 音乐数据的 ID
* @param title 音乐音频的标题
* @param imageUrl 音乐音频的图片 URL
* @param lyric 音乐音频的歌词
* @param audioUrl 音乐音频的 URL
* @param videoUrl 音乐视频的 URL
* @param createdAt 音乐音频的创建时间
* @param model 使用的模型名称
* @param prompt 生成音乐音频的提示
* @param style 音乐音频的风格
*/
private boolean success;
/**
* 任务 ID
*/
@JsonProperty("task_id")
private String taskId;
/**
* 音乐数据列表
*/
private List<MusicData> data;
/**
* 表示单个音乐数据的类
*/
@Data
static class MusicData {
/**
* 音乐数据的 ID
*/
private String id;
/**
* 音乐音频的标题
*/
private String title;
/**
* 音乐音频的图片 URL
*/
@JsonProperty("image_url")
private String imageUrl;
/**
* 音乐音频的歌词
*/
private String lyric;
/**
* 音乐音频的 URL
*/
@JsonProperty("audio_url")
private String audioUrl;
/**
* 音乐视频的 URL
*/
@JsonProperty("video_url")
private String videoUrl;
/**
* 音乐音频的创建时间
*/
@JsonProperty("created_at")
private String createdAt;
/**
* 使用的模型名称
*/
private String model;
/**
* 生成音乐音频的提示
*/
private String prompt;
/**
* 音乐音频的风格
*/
private String style;
public record MusicData(
String id,
String title,
@JsonProperty("image_url") String imageUrl,
String lyric,
@JsonProperty("audio_url") String audioUrl,
@JsonProperty("video_url") String videoUrl,
@JsonProperty("created_at") String createdAt,
String model,
String prompt,
String style
) {
}
}
}

4
yudao-module-ai/yudao-spring-boot-starter-ai/src/test/java/cn/iocoder/yudao/framework/ai/suno/SunoTests.java
View File

@ -22,8 +22,8 @@ public class SunoTests {
@Test
public void generateMusic() {
SunoApi sunoApi = new SunoApi(sunoConfig);
SunoApi.SunoReq sunoReq = new SunoApi.SunoReq()
.setPrompt("创作一首带有轻松吉他旋律的流行歌曲，[verse] 描述夏日海滩的宁静，[chorus] 节奏加快，表达对自由的向往。");
SunoApi.SunoReq sunoReq = new SunoApi.SunoReq("创作一首带有轻松吉他旋律的流行歌曲，[verse] 描述夏日海滩的宁静，[chorus] 节奏加快，表达对自由的向往。");
SunoApi.SunoResp sunoResp = sunoApi.musicGen(sunoReq);
System.out.println(sunoResp);
}