add wenxinyiyan support

This commit is contained in:
robin
2023-11-06 17:36:24 +08:00
parent 37ced2023b
commit f461d9da80
3 changed files with 5 additions and 2 deletions

View File

@ -480,6 +480,9 @@ public class ChatController {
private SseEmitter chatWithWenxinAi(ChatQueryRequest queryRequest, SseEmitter sseEmitter, String uid) throws IOException {
String prompt = buildPrompt(queryRequest);
List<FastChatMessage> messages = getFastChatMessage(uid, prompt);
if (messages.size() >= 2 && messages.size() % 2 == 0) {
messages.remove(messages.size() - 1);
}
buildSseEmitter(sseEmitter, uid);