opensource/Chat2DB
1
0
Fork 0
mirror of https://github.com/CodePhiliaX/Chat2DB.git synced 2025-07-29 18:53:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

add wenxinyiyan support

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

2
chat2db-client/src/typings/ai.ts
View File

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

2
chat2db-server/chat2db-server-start/src/main/resources/application.yml
View File

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

3
chat2db-server/chat2db-server-web/chat2db-server-web-api/src/main/java/ai/chat2db/server/web/api/controller/ai/ChatController.java
View File

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