Skip to content

Commit 4b19f2b

Browse files
author
xiongbinbin
committed
fix message role compare
1 parent 6a3843a commit 4b19f2b

2 files changed

Lines changed: 3 additions & 3 deletions

File tree

src/main/java/com/alibaba/dashscope/aigc/imagegeneration/ImageGeneration.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -314,7 +314,7 @@ private void preprocessInput(ImageGenerationParam param)
314314
boolean hasUpload = false;
315315
OSSUploadCertificate certificate = null;
316316
for (ImageGenerationMessage msg : param.getMessages()) {
317-
if (msg.getRole().equals(Role.USER)) {
317+
if (msg.getRole().equals(Role.USER.getValue())) {
318318
boolean isUpload;
319319
PreprocessMessageInput.PreprocessResult result =
320320
PreprocessMessageInput.preProcessMultiModalMessageInputs(

src/main/java/com/alibaba/dashscope/aigc/multimodalconversation/MultiModalConversation.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -236,7 +236,7 @@ private void preprocessInput(MultiModalConversationParam param)
236236
boolean isUpload = false;
237237
if (msg instanceof MultiModalConversationMessage) {
238238
MultiModalConversationMessage realMsg = (MultiModalConversationMessage)msg;
239-
if (realMsg.getRole().equals(Role.USER)) {
239+
if (realMsg.getRole().equals(Role.USER.getValue())) {
240240
PreprocessMessageInput.PreprocessResult result =
241241
PreprocessMessageInput.preProcessMessageInputs(
242242
param.getModel(),
@@ -248,7 +248,7 @@ private void preprocessInput(MultiModalConversationParam param)
248248
}
249249
} else {
250250
MultiModalMessage realMsg = (MultiModalMessage) msg;
251-
if (realMsg.getRole().equals(Role.USER)) {
251+
if (realMsg.getRole().equals(Role.USER.getValue())) {
252252
PreprocessMessageInput.PreprocessResult result =
253253
PreprocessMessageInput.preProcessMultiModalMessageInputs(
254254
param.getModel(), (MultiModalMessage) msg, param.getApiKey(), certificate);

0 commit comments

Comments
 (0)