NIMAIModelCallMessage.java

浏览该文件的文档.

package com.netease.nimlib.sdk.ai.params;

import android.text.TextUtils;
import com.netease.nimlib.log.NimLog;
import com.netease.nimlib.sdk.ai.enums.NIMAIModelRoleType;
import java.io.Serializable;

public class NIMAIModelCallMessage implements Serializable {

    private static final String             TAG = "NIMAIModelCallMessage";
    /** 上下文内容的角色 */
    private              NIMAIModelRoleType role;

    /** 上下文的内容 */
    private String msg;

    /** 类型, 暂时只有 0, 代表文本，预留扩展能力 */
    private Integer type;

    public NIMAIModelCallMessage() {
        this(null, null, null);
    }

    public NIMAIModelCallMessage(NIMAIModelRoleType role, String msg, Integer type) {
        this.role = role;
        this.msg = msg;
        this.type = type;
    }

    /**
     * 获取上下文内容的角色
     * @return 上下文内容的角色
     */
    public NIMAIModelRoleType getRole() {
        return role;
    }

    /**
     * 设置上下文内容的角色
     * @param role 上下文内容的角色
     */
    public void setRole(NIMAIModelRoleType role) {
        this.role = role;
    }

    /**
     * 获取上下文的内容
     * @return 上下文的内容
     */
    public String getMsg() {
        return msg;
    }

    /**
     * 设置上下文的内容
     * @param msg 上下文的内容
     */
    public void setMsg(String msg) {
        this.msg = msg;
    }

    /**
     * 获取类型
     * @return 类型
     */
    public Integer getType() {
        return type;
    }

    /**
     * 设置类型
     * @param type 类型
     */
    public void setType(Integer type) {
        this.type = type;
    }

    @Override
    public String toString() {
        final StringBuilder sb = new StringBuilder("NIMAIModelCallMessage{");
        sb.append("role=").append(role);
        sb.append(", msg='").append(msg).append('\'');
        sb.append(", type=").append(type);
        sb.append('}');
        return sb.toString();
    }

    public boolean isValid() {
        if (TextUtils.isEmpty(msg)) {
            NimLog.e(TAG, "msg is empty");
            return false;
        }
        if (type == null) {
            NimLog.e(TAG, "type is null");
            return false;
        }
        if (role == null) {
            NimLog.e(TAG, "role is null");
            return false;
        }
        return true;
    }
}