package com.pharmacopoeia.nihaisha.entity;

import jakarta.persistence.*;
import lombok.*;

@Entity
@Table(name = "nih_retrieval_units")
@Data
@NoArgsConstructor
@AllArgsConstructor
@Builder
public class NihRetrievalUnit {

    @Id
    @Column(name = "unit_id")
    private String unitId;

    @Column(name = "paragraph_id", nullable = false)
    private String paragraphId;

    @Column(name = "doc_id", nullable = false)
    private String docId;

    @Column(name = "unit_type", nullable = false)
    private String unitType;

    @Column(nullable = false, columnDefinition = "TEXT")
    private String text;

    @Column(name = "text_for_embedding", nullable = false, columnDefinition = "TEXT")
    private String textForEmbedding;

    @Column(name = "sentence_start", nullable = false)
    private Integer sentenceStart;

    @Column(name = "sentence_end", nullable = false)
    private Integer sentenceEnd;

    @Column(nullable = false)
    @Builder.Default
    private Double weight = 1.0;
}