package cn.iocoder.yudao.module.ai.framework.ai.core.model.xinghuo;
|
|
import cn.hutool.core.util.StrUtil;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.ai.chat.model.ChatModel;
|
import org.springframework.ai.chat.model.ChatResponse;
|
import org.springframework.ai.chat.prompt.ChatOptions;
|
import org.springframework.ai.chat.prompt.Prompt;
|
import org.springframework.ai.deepseek.DeepSeekChatModel;
|
import org.springframework.ai.deepseek.DeepSeekChatOptions;
|
import org.springframework.ai.deepseek.api.DeepSeekApi;
|
import reactor.core.publisher.Flux;
|
|
/**
|
* 讯飞星火 {@link ChatModel} 实现类
|
*
|
* @author fansili
|
*/
|
@Slf4j
|
public class XingHuoChatModel implements ChatModel {
|
|
/**
|
* 星火 X2
|
*
|
* @see <a href="https://spark-api-open.xf-yun.com/x2/chat/completions">接口地址</a>
|
*/
|
public static final String MODEL_X2 = "x2";
|
|
/**
|
* 星火 X2 Flash
|
*
|
* @see <a href="https://spark-api-open.xf-yun.com/agent/v1/chat/completions">接口地址</a>
|
*/
|
public static final String MODEL_X2_FLASH = "x2-flash";
|
|
private static final String BASE_URL_X2 = "https://spark-api-open.xf-yun.com/x2";
|
|
private static final String BASE_URL_X2_FLASH = "https://spark-api-open.xf-yun.com/agent/v1";
|
|
public static final String MODEL_DEFAULT = MODEL_X2_FLASH;
|
|
private static String getBaseUrl(String model) {
|
if (MODEL_X2_FLASH.equals(model)) {
|
return BASE_URL_X2_FLASH;
|
}
|
return BASE_URL_X2;
|
}
|
|
private final String apiKey;
|
|
private final DeepSeekChatOptions options;
|
|
/**
|
* 兼容 OpenAI 接口，进行复用
|
*/
|
private final ChatModel chatModelX2;
|
|
private final ChatModel chatModelX2Flash;
|
|
private XingHuoChatModel(String apiKey, DeepSeekChatOptions options) {
|
this.apiKey = apiKey;
|
this.options = options;
|
this.chatModelX2 = buildChatModel(MODEL_X2);
|
this.chatModelX2Flash = buildChatModel(MODEL_X2_FLASH);
|
}
|
|
public static Builder builder() {
|
return new Builder();
|
}
|
|
@Override
|
public ChatResponse call(Prompt prompt) {
|
return getChatModel(prompt).call(buildApiPrompt(prompt));
|
}
|
|
@Override
|
public Flux<ChatResponse> stream(Prompt prompt) {
|
return getChatModel(prompt).stream(buildApiPrompt(prompt));
|
}
|
|
@Override
|
public ChatOptions getOptions() {
|
return options;
|
}
|
|
@Override
|
@Deprecated(forRemoval = true)
|
@SuppressWarnings("removal")
|
public ChatOptions getDefaultOptions() {
|
return getOptions();
|
}
|
|
private ChatModel getChatModel(Prompt prompt) {
|
String model = options.getModel();
|
ChatOptions options = prompt.getOptions();
|
if (options != null && isBusinessModel(options.getModel())) {
|
model = options.getModel();
|
}
|
return getChatModel(model);
|
}
|
|
private ChatModel getChatModel(String model) {
|
if (MODEL_X2_FLASH.equals(model)) {
|
return chatModelX2Flash;
|
}
|
return chatModelX2;
|
}
|
|
private ChatModel buildChatModel(String model) {
|
return DeepSeekChatModel.builder()
|
.deepSeekApi(DeepSeekApi.builder()
|
.baseUrl(getBaseUrl(model))
|
.apiKey(apiKey)
|
.build())
|
.options(options.mutate().model("spark-x").build())
|
.build();
|
}
|
|
private static Prompt buildApiPrompt(Prompt prompt) {
|
ChatOptions options = prompt.getOptions();
|
if (options == null) {
|
return prompt;
|
}
|
return Prompt.builder()
|
.messages(prompt.getInstructions())
|
.chatOptions(options.mutate().model("spark-x").build())
|
.build();
|
}
|
|
private static boolean isBusinessModel(String model) {
|
return MODEL_X2.equals(model) || MODEL_X2_FLASH.equals(model);
|
}
|
|
public static final class Builder {
|
|
private String apiKey;
|
|
private DeepSeekChatOptions options;
|
|
public Builder apiKey(String apiKey) {
|
this.apiKey = apiKey;
|
return this;
|
}
|
|
public Builder options(DeepSeekChatOptions options) {
|
this.options = options;
|
return this;
|
}
|
|
public XingHuoChatModel build() {
|
DeepSeekChatOptions options = this.options != null ? this.options : DeepSeekChatOptions.builder().build();
|
if (StrUtil.isEmpty(options.getModel())) {
|
options = options.mutate().model(MODEL_DEFAULT).build();
|
}
|
return new XingHuoChatModel(apiKey, options);
|
}
|
|
}
|
|
}
|
