opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-07-31 11:42:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'dev' into team

Browse Source
This commit is contained in:
JiaJu Zhuang
2023-09-09 18:59:45 +08:00
parent 97561bc669 32d3fd81d2
commit f0ebfbf18f
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/ai/listener/RestAIEventSourceListener.java
View File

@ -54,7 +54,9 @@ public class RestAIEventSourceListener extends EventSourceListener {
}
Message message = new Message();
if (StringUtils.isNotBlank(data)) {
message.setContent(data.replace("\"", ""));
data = data.replaceAll("^\"|\"$", "");
data = data.replaceAll("\\\\n", "\n");
message.setContent(data);
sseEmitter.send(SseEmitter.event()
.id(id)
.data(message)
@ -78,7 +80,6 @@ public class RestAIEventSourceListener extends EventSourceListener {
try {
if (Objects.isNull(response)) {
String message = t.getMessage();
message = message + ", AI无法正常访问, 请参考文章<https://github.com/chat2db/Chat2DB/blob/main/CHAT2DB_AI_SQL.md>进行配置";
Message sseMessage = new Message();
sseMessage.setContent(message);
sseEmitter.send(SseEmitter.event()
@ -94,9 +95,9 @@ public class RestAIEventSourceListener extends EventSourceListener {
String bodyString = null;
if (Objects.nonNull(body)) {
bodyString = body.string();
log.error("REST AI sse连接异常data：{}，异常：{}", bodyString, t);
log.error("REST AI sse body error：{}，exception：{}", bodyString, t);
} else {
log.error("REST AI sse连接异常data：{}，异常：{}", response, t);
log.error("REST AI sse response error：{}，exception：{}", response, t);
}
if (Objects.nonNull(eventSource)) {
eventSource.cancel();