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

Revert "本地连接mysql"

Browse Source 
This reverts commit 2284c32b8fd12505c582e59f736290ee6def920e.
This commit is contained in:
robin
2023-11-03 13:34:22 +08:00
parent 2284c32b8f
commit b13ffbfc8b
2 changed files with 5 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
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,7 +196,6 @@ public class Chat2DBAIStreamClient {
try {
ChatCompletion chatCompletion = ChatCompletion.builder()
.messages(chatMessages)
.model(this.model)
.stream(true)
.build();

49
chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/ai/openai/listener/OpenAIEventSourceListener.java
View File

@ -2,19 +2,8 @@ package ai.chat2db.server.web.api.controller.ai.openai.listener;
import java.util.Objects;
import ai.chat2db.server.domain.api.enums.AiSqlSourceEnum;
import ai.chat2db.server.domain.api.model.Config;
import ai.chat2db.server.domain.api.service.ConfigService;
import ai.chat2db.server.tools.base.wrapper.result.DataResult;
import ai.chat2db.server.web.api.controller.ai.baichuan.model.BaichuanChatCompletions;
import ai.chat2db.server.web.api.controller.ai.baichuan.model.BaichuanChatMessage;
import ai.chat2db.server.web.api.controller.ai.chat2db.client.Chat2dbAIClient;
import ai.chat2db.server.web.api.controller.ai.fastchat.model.FastChatMessage;
import ai.chat2db.server.web.api.controller.ai.response.ChatCompletionResponse;
import ai.chat2db.server.web.api.controller.ai.zhipu.model.ZhipuChatCompletions;
import ai.chat2db.server.web.api.util.ApplicationContextUtil;
import com.fasterxml.jackson.databind.DeserializationFeature;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.unfbx.chatgpt.entity.chat.Message;
import lombok.SneakyThrows;
@ -65,50 +54,16 @@ public class OpenAIEventSourceListener extends EventSourceListener {
return;
}
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()) ? 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
String text = completionResponse.getChoices().get(0).getDelta() == null
? completionResponse.getChoices().get(0).getText()
: completionResponse.getChoices().get(0).getDelta().getContent();
completionId = completionResponse.getId();
break;
}
Message message = new Message();
if (text != null) {
message.setContent(text);
sseEmitter.send(SseEmitter.event()
.id(completionId)
.id(completionResponse.getId())
.data(message)
.reconnectTime(3000));
}