@ -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>
* 文档地址： platform.acedata.cloud/documents/d016ee3f-421b-4b6e-989a-8beba8701701
* 文档地址： 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】调用失败！ " , respBody ) ;
sink . error ( new IllegalStateException ( " 【suno-api】调用失败！ " ) ) ;
} ) ;
public SunoApi ( SunoConfig config ) {
this . webClient = WebClient . builder ( )
. baseUrl ( DEFAULT_BASE_URL )
. defaultHeaders ( ApiUtil s. g etJson ContentHeaders ( config . getToken ( ) ) )
. baseUrl ( config . getBaseUrl ( ) )
. defaultHeaders ( ( headers ) - > header s. s etContentType ( 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 . . class )
. retrieve ( )
. onStatus ( status - > ! status . is2xxSuccessful ( ) ,
response - > response . bodyToMono ( String . class )
. handle ( ( respBody , sink ) - > {
log . error ( " 【Suno】调用失败！ " , 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表示后台模式， ，
* true表示同步模式， ， ，
* @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 响应的限额数据，
*/
public record LimitData (
@JsonProperty ( " credits_left " ) Long creditsLeft ,
String period ,
@JsonProperty ( " monthly_limit " ) Long monthlyLimit ,
@JsonProperty ( " monthly_usage " ) Long monthlyUsage
) {
}
}
芋道源码