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

@ -2,7 +2,7 @@ export enum AIType {
CHAT2DBAI = 'CHAT2DBAI',
ZHIPUAI = 'ZHIPUAI',
BAICHUANAI='BAICHUANAI',
// WENXINAI='WENXINAI',
WENXINAI='WENXINAI',
// TONGYIQIANWENAI='TONGYIQIANWENAI',
OPENAI = 'OPENAI',
AZUREAI = 'AZUREAI',

View File

@ -73,7 +73,7 @@ chatgpt:
# 可以选择GPT3或GPT35
version: GPT35
context:
length: 4
length: 1
# Print the HTTP log
logbook:

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);