com.dingtalk.baymax.framework.sdk.mercury.prompt.ChatPromptValue Maven / Gradle / Ivy

Go to download

package com.dingtalk.baymax.framework.sdk.mercury.prompt;

import com.dingtalk.baymax.framework.sdk.mercury.domain.BaseMessage;
import com.dingtalk.baymax.framework.sdk.mercury.util.BaseMessageUtils;

import java.util.List;

/**
 * ChatPromptValue
 *
 * @Author xiaoyan.wjw
 * @Date 2023-05-22 15:51:34
 */
public class ChatPromptValue extends PromptValue {
    private List messages;

    public List getMessages() {
        return messages;
    }

    public void setMessages(List messages) {
        this.messages = messages;
    }

    @Override
    public String toString() {
        return BaseMessageUtils.getBufferString(messages, "Human", "AI");
    }

    @Override
    public List toMessages() {
        return messages;
    }
}