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

ai result update

Browse Source
This commit is contained in:
robin
2023-11-03 18:51:11 +08:00
parent dd2d198d3f
commit d1337f7671
2 changed files with 6 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/ai/chat2db/client/Chat2DBAIStreamClient.java
View File

@ -196,26 +196,11 @@ public class Chat2DBAIStreamClient {
log.error("param error：ChatEventSourceListener cannot be empty");
throw new ParamBusinessException();
}
log.info("Chat AI, prompt:{}", chatMessages.get(chatMessages.size() - 1).getContent());
try {
ChatCompletion chatCompletion = ChatCompletion.builder()
.messages(chatMessages)
.stream(true)
.build();
ConfigService configService = ApplicationContextUtil.getBean(ConfigService.class);
DataResult<Config> chat2dbModel = configService.find(Chat2dbAIClient.CHAT2DB_OPENAI_MODEL);
String model = Objects.nonNull(chat2dbModel.getData()) && StringUtils.isNotBlank(chat2dbModel.getData().getContent()) ? chat2dbModel.getData().getContent() : AiSqlSourceEnum.OPENAI.getCode();
AiSqlSourceEnum aiSqlSourceEnum = AiSqlSourceEnum.getByName(model);
switch (aiSqlSourceEnum) {
case BAICHUANAI:
chatCompletion = ChatCompletion.builder().messages(chatMessages).model("Baichuan2-53B").build();
break;
case ZHIPUAI:
chatCompletion = ChatCompletion.builder().messages(chatMessages).model("chatglm_turbo").build();
break;
default:
break;
}
EventSource.Factory factory = EventSources.createFactory(this.okHttpClient);
ObjectMapper mapper = new ObjectMapper();

43
chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/ai/chat2db/listener/Chat2dbAIEventSourceListener.java
View File

@ -66,43 +66,12 @@ public class Chat2dbAIEventSourceListener extends EventSourceListener {
}
ObjectMapper mapper = new ObjectMapper();
mapper.configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false);
ConfigService configService = ApplicationContextUtil.getBean(ConfigService.class);
DataResult<Config> chat2dbModel = configService.find(Chat2dbAIClient.CHAT2DB_OPENAI_MODEL);
String model = Objects.nonNull(chat2dbModel.getData()) && StringUtils.isNotBlank(chat2dbModel.getData().getContent()) ? chat2dbModel.getData().getContent() : AiSqlSourceEnum.OPENAI.getCode();
AiSqlSourceEnum aiSqlSourceEnum = AiSqlSourceEnum.getByName(model);
String text = "";
String completionId = null;
// 读取Json
switch (aiSqlSourceEnum) {
case BAICHUANAI:
BaichuanChatCompletions chatCompletions = mapper.readValue(data, BaichuanChatCompletions.class);
for (BaichuanChatMessage message : chatCompletions.getData().getMessages()) {
if (message != null) {
if (message.getContent() != null) {
text = message.getContent();
}
}
}
break;
case ZHIPUAI:
ZhipuChatCompletions zhipuChatCompletions = mapper.readValue(data, ZhipuChatCompletions.class);
text = zhipuChatCompletions.getData();
if (Objects.isNull(text)) {
for (FastChatMessage message : zhipuChatCompletions.getBody().getChoices()) {
if (message != null && message.getContent() != null) {
text = message.getContent();
}
}
}
break;
default:
ChatCompletionResponse completionResponse = mapper.readValue(data, ChatCompletionResponse.class);
text = completionResponse.getChoices().get(0).getDelta() == null
? completionResponse.getChoices().get(0).getText()
: completionResponse.getChoices().get(0).getDelta().getContent();
completionId = completionResponse.getId();
break;
}
ChatCompletionResponse completionResponse = mapper.readValue(data, ChatCompletionResponse.class);
String text = completionResponse.getChoices().get(0).getDelta() == null
? completionResponse.getChoices().get(0).getText()
: completionResponse.getChoices().get(0).getDelta().getContent();
String completionId = completionResponse.getId();
Message message = new Message();
if (text != null) {
message.setContent(text);