TaoAi/src/main/java/io/github/qifan777/knowledge/ai/message/AiMessage.java

package io.github.qifan777.knowledge.ai.message;

import io.github.qifan777.knowledge.ai.session.AiSession;
import io.github.qifan777.knowledge.infrastructure.jimmer.BaseEntity;
import jakarta.validation.constraints.Null;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.experimental.Accessors;
import org.babyfish.jimmer.sql.*;
import org.springframework.ai.chat.messages.MessageType;

import java.util.List;

/**
 * 历史消息
 */

@Entity
public interface AiMessage extends BaseEntity {

    /**
     * 消息类型(用户/助手/系统)
     */
    MessageType type();

    /**
     * 消息内容
     */
    String textContent();

    @Serialized
    @Null
    List<Media> medias();

    @IdView
    String sessionId();

    /**
     * 会话
     */
    @ManyToOne
    @JoinColumn(name = "ai_session_id")
    @OnDissociate(DissociateAction.DELETE)
    AiSession session();

    @Data
    @Accessors(chain = true)
    class Media {
        public String type;
        public String data;
    }
}