package com.lingyue.ai.dto.ner;

import io.swagger.v3.oas.annotations.media.Schema;
import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;

/**
 * 关系信息
 *
 * @author lingyue
 * @since 2026-01-19
 */
@Data
@Builder
@NoArgsConstructor
@AllArgsConstructor
@Schema(description = "关系信息")
public class RelationInfo {

    @Schema(description = "源实体名称")
    private String fromEntity;

    @Schema(description = "源实体临时ID")
    private String fromEntityId;

    @Schema(description = "目标实体名称")
    private String toEntity;

    @Schema(description = "目标实体临时ID")
    private String toEntityId;

    @Schema(description = "关系类型", example = "负责/属于/包含/位于/关联")
    private String relationType;

    @Schema(description = "置信度", example = "0.92")
    private Double confidence;
}
