31 lines
585 B
Java
31 lines
585 B
Java
package com.yutou.qqbot.data.baidu;
|
|
|
|
import lombok.Data;
|
|
|
|
@Data
|
|
public class Message {
|
|
private String role = "user";
|
|
private String content;
|
|
|
|
public Message() {
|
|
}
|
|
|
|
|
|
public boolean checkIsUser() {
|
|
return "user".equals(role);
|
|
}
|
|
|
|
public static Message create(String message) {
|
|
return create(message, false);
|
|
}
|
|
|
|
public static Message create(String message, boolean isGTP) {
|
|
Message msg = new Message();
|
|
msg.content = message;
|
|
if (isGTP) {
|
|
msg.role = "assistant";
|
|
}
|
|
return msg;
|
|
}
|
|
}
|