V2NIMAIModelCallMessage.java

浏览该文件的文档.

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

import com.netease.nimlib.log.NimLog;
import com.netease.nimlib.sdk.v2.ai.enums.V2NIMAIModelRoleType;
import com.netease.nimlib.util.EncryptUtil;
import java.io.Serializable;

/**
 * 请求调用上下文内容，
 */
public class V2NIMAIModelCallMessage implements Serializable {

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

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

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

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

    public V2NIMAIModelCallMessage(V2NIMAIModelRoleType role, String msg, Integer type) {
        this.role = role;
        this.msg = msg;
        this.type = type;
    }

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

    /**
     * 设置上下文内容的角色
     * @param role 上下文内容的角色
     */
    public void setRole(V2NIMAIModelRoleType 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() {
        if (NimLog.isDebugLog()){
            final StringBuilder sb = new StringBuilder("V2NIMAIModelCallMessage{");
            sb.append("role=").append(role);
            sb.append(", msg='").append(msg).append('\'');
            sb.append(", type=").append(type);
            sb.append('}');
            return sb.toString();
        }else {
            final StringBuilder sb = new StringBuilder("V2NIMAIModelCallMessage{");
            sb.append("role=").append(role);
            sb.append(", msg='").append(EncryptUtil.encryptBase64(msg)).append('\'');
            sb.append(", type=").append(type);
            sb.append('}');
            return sb.toString();
        }

    }

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

    public String getInvalidMsg() {
        if (type == null) {
            return "type is null";
        }
        if (role == null) {
            return "role is null";
        }
        return null;
    }

    /**
     * 判断 msg 是否需要被过滤掉（msg 为 null 或 trim 后为空字符串）
     * @return 如果需要被过滤掉返回 true，否则返回 false
     */
    public boolean shouldBeFiltered() {
        return msg == null || msg.trim().isEmpty();
    }
}