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

!3 接入suno-api

Browse Source 
Merge pull request !3 from 小新/master-jdk21-ai
This commit is contained in:
芋道源码
2024-06-17 12:08:09 +00:00
committed by Gitee
parent a331ed546d f1292fb7fe
commit 5fabb0b757
17 changed files with 473 additions and 248 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/config/YudaoAiAutoConfiguration.java
View File

@ -151,7 +151,7 @@ public class YudaoAiAutoConfiguration {
@Bean
@ConditionalOnProperty(value = "yudao.ai.suno.enable", havingValue = "true")
public SunoApi sunoApi(YudaoAiProperties yudaoAiProperties) {
return new SunoApi(new SunoConfig(yudaoAiProperties.getSuno().getToken()));
return new SunoApi(new SunoConfig(yudaoAiProperties.getSuno().getBaseUrl()));
}
private static @NotNull MidjourneyConfig getMidjourneyConfig(ApplicationContext applicationContext,

4
yudao-module-ai/yudao-spring-boot-starter-ai/src/main/java/cn/iocoder/yudao/framework/ai/config/YudaoAiProperties.java
View File

@ -119,9 +119,9 @@ public class YudaoAiProperties {
private boolean enable = false;
/**
* token
* suno-api 服务的基本地址
*/
private String token;
private String baseUrl;
}

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

@ -13,7 +13,7 @@ import lombok.NoArgsConstructor;
@AllArgsConstructor
public class SunoConfig {
/**
* token信息
* suno-api服务的基本路径
*/
private String token;
private String baseUrl;
}

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

@ -4,112 +4,207 @@ import cn.iocoder.yudao.framework.ai.core.model.suno.SunoConfig;
import com.fasterxml.jackson.annotation.JsonInclude;
import com.fasterxml.jackson.annotation.JsonProperty;
import lombok.extern.slf4j.Slf4j;
import org.springframework.ai.openai.api.ApiUtils;
import org.springframework.core.ParameterizedTypeReference;
import org.springframework.http.HttpStatusCode;
import org.springframework.http.MediaType;
import org.springframework.web.reactive.function.client.ClientResponse;
import org.springframework.web.reactive.function.client.WebClient;
import reactor.core.publisher.Mono;
import java.util.List;
import java.util.function.Function;
import java.util.function.Predicate;
/**
* Suno API
* <br>
* 文档地址：https://platform.acedata.cloud/documents/d016ee3f-421b-4b6e-989a-8beba8701701
* 文档地址：https://github.com/status2xx/suno-api/blob/main/README_CN.md
*
* @Author xiaoxin
* @Date 2024/5/27
* @Date 2024/6/3
*/
@Slf4j
public class SunoApi {
public static final String DEFAULT_BASE_URL = "https://api.acedata.cloud/suno";
private final WebClient webClient;
private final Predicate<HttpStatusCode> STATUS_PREDICATE = status -> !status.is2xxSuccessful();
private final Function<ClientResponse, Mono<? extends Throwable>> EXCEPTION_FUNCTION = response -> response.bodyToMono(String.class)
.handle((respBody, sink) -> {
log.error("【suno-api】调用失败！resp: 【{}】", respBody);
sink.error(new IllegalStateException("【suno-api】调用失败！"));
});
public SunoApi(SunoConfig config) {
this.webClient = WebClient.builder()
.baseUrl(DEFAULT_BASE_URL)
.defaultHeaders(ApiUtils.getJsonContentHeaders(config.getToken()))
.baseUrl(config.getBaseUrl())
.defaultHeaders((headers) -> headers.setContentType(MediaType.APPLICATION_JSON))
.build();
}
// TODO @芋艿：方法名，要考虑下；
public SunoResp musicGen(SunoReq sunReq) {
public List<MusicData> generate(SunoApi.SunoReq sunReq) {
return this.webClient.post()
.uri("/audios")
.body(Mono.just(sunReq), SunoReq.class)
.uri("/api/generate")
.body(Mono.just(sunReq), SunoApi.SunoReq.class)
.retrieve()
.onStatus(status -> !status.is2xxSuccessful(),
response -> response.bodyToMono(String.class)
.handle((respBody, sink) -> {
log.error("【Suno】调用失败！resp: 【{}】", respBody);
sink.error(new IllegalStateException("【Suno】调用失败！"));
}))
.bodyToMono(SunoResp.class)
.onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION)
.bodyToMono(new ParameterizedTypeReference<List<MusicData>>() {
})
.block();
}
public List<MusicData> customGenerate(SunoApi.SunoReq sunReq) {
return this.webClient.post()
.uri("/api/custom_generate")
.body(Mono.just(sunReq), SunoApi.SunoReq.class)
.retrieve()
.onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION)
.bodyToMono(new ParameterizedTypeReference<List<MusicData>>() {
})
.block();
}
public List<MusicData> doChatCompletion(String prompt) {
return this.webClient.post()
.uri("/v1/chat/completions")
.body(Mono.just(new SunoReq(prompt)), SunoApi.SunoReq.class)
.retrieve()
.onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION)
.bodyToMono(new ParameterizedTypeReference<List<MusicData>>() {
})
.block();
}
public LyricsData generateLyrics(String prompt) {
return this.webClient.post()
.uri("/api/generate_lyrics")
.body(Mono.just(new SunoReq(prompt)), SunoApi.SunoReq.class)
.retrieve()
.onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION)
.bodyToMono(LyricsData.class)
.block();
}
public List<MusicData> selectById(String ids) {
return this.webClient.get()
.uri(uriBuilder -> uriBuilder
.path("/api/get")
.queryParam("ids", ids)
.build())
.retrieve()
.onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION)
.bodyToMono(new ParameterizedTypeReference<List<MusicData>>() {
})
.block();
}
public LimitData selectLimit() {
return this.webClient.get()
.uri("/api/get_limit")
.retrieve()
.onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION)
.bodyToMono(LimitData.class)
.block();
}
/**
* 请求数据对象，用于生成音乐音频。
* 根据提示生成音频
*
* @param prompt 用于生成音乐音频的提示
* @param lyric 用于生成音乐音频的歌词
* @param custom 指示音乐音频是否为定制，如果为 true，则从歌词生成，否则从提示生成
* @param title 音乐音频的标题
* @param style 音乐音频的风格
* @param callbackUrl 音乐音频生成后回调的 URL
* @param prompt 用于生成音乐音频的提示
* @param tags 音乐风格
* @param title 音乐名称
* @param mv 模型
* @param waitAudio false表示后台模式，仅返回音频任务信息，需要调用get API获取详细的音频信息。
* true表示同步模式，API最多等待100s，音频生成完毕后直接返回音频链接等信息，建议在GPT等agent中使用。
* @param makeInstrumental 指示音乐音频是否为定制，如果为 true，则从歌词生成，否则从提示生成
*/
@JsonInclude(value = JsonInclude.Include.NON_NULL)
public record SunoReq(
String prompt,
String lyric,
boolean custom,
String tags,
String title,
String style,
String callbackUrl
String mv,
@JsonProperty("wait_audio") boolean waitAudio,
@JsonProperty("make_instrumental") boolean makeInstrumental
) {
public SunoReq(String prompt) {
this(prompt, null, false, null, null, null);
this(prompt, null, null, null, false, false);
}
public SunoReq(String prompt, String mv, boolean makeInstrumental) {
this(prompt, null, null, mv, false, makeInstrumental);
}
public SunoReq(String prompt, String mv, String tags, String title) {
this(prompt, tags, title, mv, false, false);
}
}
/**
* SunoAPI 响应的数据。
* SunoAPI 响应的音频数据。
*
* @param success 表示请求是否成功
* @param taskId 任务 ID
* @param data 音乐数据列表
* @param id 音乐数据的 ID
* @param title 音乐音频的标题
* @param imageUrl 音乐音频的图片 URL
* @param lyric 音乐音频的歌词
* @param audioUrl 音乐音频的 URL
* @param videoUrl 音乐视频的 URL
* @param createdAt 音乐音频的创建时间
* @param modelName
* @param status submitted、queued、streaming、complete
* @param gptDescriptionPrompt
* @param prompt 生成音乐音频的提示
* @param type
* @param tags
*/
public record SunoResp(
boolean success,
@JsonProperty("task_id") String taskId,
List<MusicData> data
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,
@JsonProperty("model_name") String modelName,
String status,
@JsonProperty("gpt_description_prompt") String gptDescriptionPrompt,
String prompt,
String type,
String tags
) {
/**
* 单个音乐数据。
*
* @param id 音乐数据的 ID
* @param title 音乐音频的标题
* @param imageUrl 音乐音频的图片 URL
* @param lyric 音乐音频的歌词
* @param audioUrl 音乐音频的 URL
* @param videoUrl 音乐视频的 URL
* @param createdAt 音乐音频的创建时间
* @param model 使用的模型名称
* @param prompt 生成音乐音频的提示
* @param 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
) {
}
}
/**
* SunoAPI 响应的歌词数据。
*
* @param text 歌词
* @param title 标题
* @param status 状态
*/
public record LyricsData(
String text,
String title,
String status
) {
}
/**
* SunoAPI 响应的限额数据，目前每日免费50
*/
public record LimitData(
@JsonProperty("credits_left") Long creditsLeft,
String period,
@JsonProperty("monthly_limit") Long monthlyLimit,
@JsonProperty("monthly_usage") Long monthlyUsage
) {
}
}

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

@ -5,27 +5,54 @@ import cn.iocoder.yudao.framework.ai.core.model.suno.api.SunoApi;
import org.junit.Before;
import org.junit.Test;
import java.util.List;
/**
* @Author xiaoxin
* @Date 2024/5/27
*/
public class SunoTests {
private SunoConfig sunoConfig;
SunoApi sunoApi;
@Before
public void setup() {
String token = "16b4356581984d538652354b60d69ff0";
this.sunoConfig = new SunoConfig(token);
String url = "https://suno-ix9nve79x-status2xxs-projects.vercel.app";
this.sunoApi = new SunoApi(new SunoConfig(url));
}
@Test
public void generateMusic() {
SunoApi sunoApi = new SunoApi(sunoConfig);
SunoApi.SunoReq sunoReq = new SunoApi.SunoReq("创作一首带有轻松吉他旋律的流行歌曲，[verse] 描述夏日海滩的宁静，[chorus] 节奏加快，表达对自由的向往。");
public void selectById() {
System.out.println(sunoApi.selectById("d460ddda-7c87-4f34-b751-419b08a590ca,ff90ea66-49cd-4fd2-b44c-44267dfd5551"));
SunoApi.SunoResp sunoResp = sunoApi.musicGen(sunoReq);
System.out.println(sunoResp);
}
@Test
public void generate() {
List<SunoApi.MusicData> generate = sunoApi.generate(new SunoApi.SunoReq("创作一首带有轻松吉他旋律的流行歌曲，[verse] 描述夏日海滩的宁静，[chorus] 节奏加快，表达对自由的向往。"));
System.out.println(generate);
}
@Test
public void doChatCompletion() {
List<SunoApi.MusicData> generate = sunoApi.doChatCompletion("创作一首带有轻松吉他旋律的流行歌曲，[verse] 描述夏日海滩的宁静，[chorus] 节奏加快，表达对自由的向往。");
System.out.println(generate);
}
@Test
public void generateLyrics() {
SunoApi.LyricsData lyricsData = sunoApi.generateLyrics("A soothing lullaby");
System.out.println(lyricsData);
}
@Test
public void selectLimit() {
SunoApi.LimitData limitData = sunoApi.selectLimit();
System.out.println(limitData);
}
}