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

【代码优化】AI：完善 SunoApiTests 单测，方便大家快速体验

Browse Source
This commit is contained in:
YunaiV
2024-07-06 23:13:55 +08:00
parent 62d5ed8395
commit fa230cea8e
3 changed files with 84 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -68,17 +68,6 @@ public class SunoApi {
.block();
}
public List<MusicData> chatCompletion(String prompt) {
return this.webClient.post()
.uri("/v1/chat/completions")
.body(Mono.just(new MusicGenerateRequest(prompt)), MusicGenerateRequest.class)
.retrieve()
.onStatus(STATUS_PREDICATE, EXCEPTION_FUNCTION.apply(prompt))
.bodyToMono(new ParameterizedTypeReference<List<MusicData>>() {
})
.block();
}
public LyricsData generateLyrics(String prompt) {
return this.webClient.post()
.uri("/api/generate_lyrics")
@ -198,7 +187,7 @@ public class SunoApi {
}
/**
* Suno API 响应的限额数据，目前每日免费50
* Suno API 响应的限额数据，目前每日免费 50
*/
public record LimitUsageData(
@JsonProperty("credits_left") Long creditsLeft,