package com.xly.milvus.bean;

import lombok.AllArgsConstructor;
import lombok.Builder;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.util.List;
import java.util.Map;

/**
 * 相似度查询请求实体
 */
@Data
@Builder
@NoArgsConstructor
@AllArgsConstructor
public class SimilaritySearchRequest {
    private List<Float> queryVector;        // 查询向量
    private String queryText;                // 查询文本（如果有文本转向量服务）
    private Integer topK = 10;                // 返回数量
    private Double minScore;                  // 最小相似度得分
    private Double maxScore;                   // 最大相似度得分
    private String metricType = "IP";          // 距离类型: IP(内积), L2(欧氏距离), COSINE(余弦)
    private List<String> outputFields;         // 输出字段
    private Map<String, Object> filter;        // 过滤条件
    private String partitionName;               // 分区名称
    private Boolean withScore = true;           // 是否返回得分
    private Boolean withDistance = false;       // 是否返回距离
}