From eed5ba03fee776863607332ebae8e6b058c854b0 Mon Sep 17 00:00:00 2001 From: LzyGit <35184156+LzyGit@users.noreply.github.com> Date: Mon, 22 Apr 2024 15:41:26 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E5=8F=91=E9=80=81=E7=A9=BA?= =?UTF-8?q?=E5=AD=97=E7=AC=A6=E4=B8=B2system=20message=E7=9A=84=E5=AF=BC?= =?UTF-8?q?=E8=87=B4openai=E6=8E=A5=E5=8F=A3=E6=8A=A5=E9=94=99=E7=9A=84?= =?UTF-8?q?=E9=97=AE=E9=A2=98=20(#927)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: lzy-lenovo --- .../tencent/supersonic/common/util/ChatGptHelper.java | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/common/src/main/java/com/tencent/supersonic/common/util/ChatGptHelper.java b/common/src/main/java/com/tencent/supersonic/common/util/ChatGptHelper.java index 0a3d6b51f..23312f615 100644 --- a/common/src/main/java/com/tencent/supersonic/common/util/ChatGptHelper.java +++ b/common/src/main/java/com/tencent/supersonic/common/util/ChatGptHelper.java @@ -1,6 +1,7 @@ package com.tencent.supersonic.common.util; +import cn.hutool.core.util.StrUtil; import com.plexpt.chatgpt.ChatGPT; import com.plexpt.chatgpt.entity.chat.ChatCompletion; import com.plexpt.chatgpt.entity.chat.ChatCompletionResponse; @@ -14,6 +15,7 @@ import java.net.Proxy; import java.text.SimpleDateFormat; import java.util.Arrays; import java.util.Date; +import java.util.List; @Component @@ -47,9 +49,15 @@ public class ChatGptHelper { } public Message getChatCompletion(Message system, Message message) { + List messages ; + if (StrUtil.isBlank(system.getContent())) { + messages = Arrays.asList(message); + } else { + messages = Arrays.asList(system, message); + } ChatCompletion chatCompletion = ChatCompletion.builder() .model(ChatCompletion.Model.GPT_3_5_TURBO_16K.getName()) - .messages(Arrays.asList(system, message)) + .messages(messages) .maxTokens(10000) .temperature(0.9) .build();