(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);
Boolean updateQuery(Long questionId, QueryResult queryResult, ChatContext chatCtx);
void updateQuery(Long questionId, int parseId, QueryResult queryResult, ChatContext chatCtx);
ChatParseDO getParseInfo(Long questionId, int parseId);

View File

@@ -196,7 +196,11 @@ public class ChatServiceImpl implements ChatService {
}
@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.setQuestionId(questionId);
chatQueryDO.setQueryResult(JsonUtil.toString(queryResult));
@@ -204,7 +208,6 @@ public class ChatServiceImpl implements ChatService {
updateQuery(chatQueryDO);
chatRepository.updateLastQuestion(chatCtx.getChatId().longValue(),
chatCtx.getQueryText(), getCurrentTime());
return true;
}
@Override

View File

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