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;

import java.util.List;

/**
 * 关系抽取请求 DTO
 *
 * @author lingyue
 * @since 2026-01-19
 */
@Data
@Builder
@NoArgsConstructor
@AllArgsConstructor
@Schema(description = "关系抽取请求")
public class RelationRequest {

    @Schema(description = "文档ID")
    private String documentId;

    @Schema(description = "原始文本内容", required = true)
    private String text;

    @Schema(description = "已提取的实体列表", required = true)
    private List<EntityInfo> entities;
}
