peihaoyu/forest
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🎨 openai model 切换

Browse Source
This commit is contained in:
ronger 2023-07-17 10:41:39 +08:00
parent 8ca74e13b5
commit 36721f4cfb
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/main/java/com/rymcu/forest/openai/OpenAiController.java
View File

@ -7,6 +7,7 @@ import com.rymcu.forest.entity.User;
import com.rymcu.forest.openai.entity.ChatMessageModel; import com.rymcu.forest.openai.entity.ChatMessageModel;
import com.rymcu.forest.openai.service.OpenAiService; import com.rymcu.forest.openai.service.OpenAiService;
import com.rymcu.forest.openai.service.SseService; import com.rymcu.forest.openai.service.SseService;
import com.rymcu.forest.util.Html2TextUtil;
import com.rymcu.forest.util.UserUtils; import com.rymcu.forest.util.UserUtils;
import com.theokanning.openai.completion.chat.ChatCompletionChoice; import com.theokanning.openai.completion.chat.ChatCompletionChoice;
import com.theokanning.openai.completion.chat.ChatCompletionChunk; import com.theokanning.openai.completion.chat.ChatCompletionChunk;
@ -72,7 +73,7 @@ public class OpenAiController {
list.add(message); list.add(message);
} }
messages.forEach(chatMessageModel -> { messages.forEach(chatMessageModel -> {
ChatMessage message = new ChatMessage(chatMessageModel.getRole(), chatMessageModel.getContent()); ChatMessage message = new ChatMessage(chatMessageModel.getRole(), Html2TextUtil.getContent(chatMessageModel.getContent()));
list.add(message); list.add(message);
}); });
return sendMessage(user, list); return sendMessage(user, list);
@ -82,7 +83,7 @@ public class OpenAiController {
private GlobalResult sendMessage(User user, List<ChatMessage> list) { private GlobalResult sendMessage(User user, List<ChatMessage> list) {
OpenAiService service = new OpenAiService(token, Duration.ofSeconds(180)); OpenAiService service = new OpenAiService(token, Duration.ofSeconds(180));
ChatCompletionRequest completionRequest = ChatCompletionRequest.builder() ChatCompletionRequest completionRequest = ChatCompletionRequest.builder()
.model("gpt-3.5-turbo") .model("gpt-3.5-turbo-16k-0613")
.stream(true) .stream(true)
.messages(list) .messages(list)
.build(); .build();