opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-07-29 18:53:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #872 from evrentan/reafactor/refactor-Chat2dbAIClient-class

Browse Source 
refactor(Chat2dbAIClient): refactor Chat2dbAIClient class
This commit is contained in:
Chat2DB
2024-10-28 14:06:18 +08:00
committed by GitHub
parent 666c27af5e 4494f95d4f
commit c4ca7ff9cb
1 changed files with 43 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/ai/chat2db/client/Chat2dbAIClient.java
View File

@ -4,11 +4,10 @@ package ai.chat2db.server.web.api.controller.ai.chat2db.client;
import ai.chat2db.server.domain.api.model.Config;
import ai.chat2db.server.domain.api.service.ConfigService;
import ai.chat2db.server.web.api.util.ApplicationContextUtil;
import com.google.common.collect.Lists;
import com.unfbx.chatgpt.OpenAiStreamClient;
import com.unfbx.chatgpt.constant.OpenAIConst;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import java.util.Objects;
/**
* @author jipengfei
@ -35,7 +34,7 @@ public class Chat2dbAIClient {
public static final String CHAT2DB_EMBEDDING_MODEL= "fastchat.embedding.model";
private static Chat2DBAIStreamClient CHAT2DB_AI_STREAM_CLIENT;
private static volatile Chat2DBAIStreamClient CHAT2DB_AI_STREAM_CLIENT;
public static Chat2DBAIStreamClient getInstance() {
if (CHAT2DB_AI_STREAM_CLIENT != null) {
@ -57,35 +56,57 @@ public class Chat2dbAIClient {
}
public static void refresh() {
String apikey;
String apiHost = ApplicationContextUtil.getProperty(CHAT2DB_OPENAI_HOST);
if (StringUtils.isBlank(apiHost)) {
apiHost = OpenAIConst.OPENAI_HOST;
}
ConfigService configService = ApplicationContextUtil.getBean(ConfigService.class);
CHAT2DB_AI_STREAM_CLIENT = Chat2DBAIStreamClient.builder().apiHost(getApiHost(configService))
.apiKey(getApiKey(configService)).model(getModel(configService)).build();
}
private static String getApiHost(ConfigService configService) {
Config apiHostConfig = configService.find(CHAT2DB_OPENAI_HOST).getData();
if (apiHostConfig != null) {
apiHost = apiHostConfig.getContent();
if (Objects.nonNull(apiHostConfig)) {
return apiHostConfig.getContent();
}
String apiHost = ApplicationContextUtil.getProperty(CHAT2DB_OPENAI_HOST);
if (apiHost.isBlank()) {
return OpenAIConst.OPENAI_HOST;
}
return apiHost;
}
private static String getApiKey(ConfigService configService) {
String apiKey;
Config config = configService.find(CHAT2DB_OPENAI_KEY).getData();
if (config != null) {
apikey = config.getContent();
if (Objects.nonNull(config)) {
apiKey = config.getContent();
} else {
apikey = ApplicationContextUtil.getProperty(CHAT2DB_OPENAI_KEY);
apiKey = ApplicationContextUtil.getProperty(CHAT2DB_OPENAI_KEY);
}
log.info("refresh chat2db apikey:{}", maskApiKey(apiKey));
return apiKey;
}
private static String getModel(ConfigService configService) {
Config modelConfig = configService.find(CHAT2DB_OPENAI_MODEL).getData();
String model = "";
if (modelConfig != null) {
model = modelConfig.getContent();
if (Objects.nonNull(modelConfig)) {
return modelConfig.getContent();
}
log.info("refresh chat2db apikey:{}", maskApiKey(apikey));
CHAT2DB_AI_STREAM_CLIENT = Chat2DBAIStreamClient.builder().apiHost(apiHost)
.apiKey(apikey).model(model).build();
return null;
}
private static String maskApiKey(String input) {
if (input == null) {
return input;
if (Objects.isNull(input)) {
return null;
}
StringBuilder maskedString = new StringBuilder(input);