(improvement)(chat) The history record only retains the query result of the first parse (#526)

Co-authored-by: jolunoluo
This commit is contained in:
LXW
2023-12-18 10:20:50 +08:00
committed by GitHub
parent 5016881ce3
commit 61641ecb00
3 changed files with 7 additions and 4 deletions

View File

@@ -53,7 +53,7 @@ public interface ChatService {
int updateQuery(ChatQueryDO chatQueryDO); int updateQuery(ChatQueryDO chatQueryDO);
Boolean updateQuery(Long questionId, QueryResult queryResult, ChatContext chatCtx); void updateQuery(Long questionId, int parseId, QueryResult queryResult, ChatContext chatCtx);
ChatParseDO getParseInfo(Long questionId, int parseId); ChatParseDO getParseInfo(Long questionId, int parseId);

View File

@@ -196,7 +196,11 @@ public class ChatServiceImpl implements ChatService {
} }
@Override @Override
public Boolean updateQuery(Long questionId, QueryResult queryResult, ChatContext chatCtx) { public void updateQuery(Long questionId, int parseId, QueryResult queryResult, ChatContext chatCtx) {
//The history record only retains the query result of the first parse
if (parseId > 1) {
return;
}
ChatQueryDO chatQueryDO = new ChatQueryDO(); ChatQueryDO chatQueryDO = new ChatQueryDO();
chatQueryDO.setQuestionId(questionId); chatQueryDO.setQuestionId(questionId);
chatQueryDO.setQueryResult(JsonUtil.toString(queryResult)); chatQueryDO.setQueryResult(JsonUtil.toString(queryResult));
@@ -204,7 +208,6 @@ public class ChatServiceImpl implements ChatService {
updateQuery(chatQueryDO); updateQuery(chatQueryDO);
chatRepository.updateLastQuestion(chatCtx.getChatId().longValue(), chatRepository.updateLastQuestion(chatCtx.getChatId().longValue(),
chatCtx.getQueryText(), getCurrentTime()); chatCtx.getQueryText(), getCurrentTime());
return true;
} }
@Override @Override

View File

@@ -209,7 +209,7 @@ public class QueryServiceImpl implements QueryService {
for (ExecuteResultProcessor executeResultProcessor : executeProcessors) { for (ExecuteResultProcessor executeResultProcessor : executeProcessors) {
executeResultProcessor.process(queryResult, parseInfo, queryReq); executeResultProcessor.process(queryResult, parseInfo, queryReq);
} }
chatService.updateQuery(queryReq.getQueryId(), queryResult, chatCtx); chatService.updateQuery(queryReq.getQueryId(), queryReq.getParseId(), queryResult, chatCtx);
} else { } else {
chatService.deleteChatQuery(queryReq.getQueryId()); chatService.deleteChatQuery(queryReq.getQueryId());
} }