import type { BaseDTO } from '../base'

/**
 * 知识库配置
 */
export interface TbKnowledge extends BaseDTO {
    /** 知识库ID */
    knowledgeId?: string
    /** 知识库标题 */
    title?: string
    /** 知识库头像 */
    avatar?: string
    /** 知识库描述 */
    description?: string
    /** Dify知识库ID（Dataset ID） */
    difyDatasetId?: string
    /** Dify索引方式（high_quality/economy） */
    difyIndexingTechnique?: 'high_quality' | 'economy'
    /** 向量模型名称 */
    embeddingModel?: string
    /** 向量模型提供商 */
    embeddingModelProvider?: string
    /** Rerank模型名称 */
    rerankModel?: string
    /** Rerank模型提供商 */
    rerankModelProvider?: string
    /** 是否启用Rerank（0否 1是） */
    rerankingEnable?: number
    /** 检索Top K（返回前K个结果） */
    retrievalTopK?: number
    /** 检索分数阈值（0.00-1.00） */
    retrievalScoreThreshold?: number
    /** 文档数量 */
    documentCount?: number
    /** 总分段数 */
    totalChunks?: number
    /** 所属服务 workcase、bidding */
    service?: string
    /** bidding所属项目ID */
    projectId?: string
    /** 所属分类  workcase的内部知识库、外部知识库，其他服务default */
    category?: string
}

/**
 * 知识库文件
 */
export interface TbKnowledgeFile extends BaseDTO {
    /** 知识库ID */
    knowledgeId?: string
    /** 文件ID */
    fileId?: string
    /** 文件根ID */
    fileRootId?: string
    /** Dify文件ID */
    difyFileId?: string
    /** 文件版本 */
    version?: number
}

/**
 * 文档分段请求体
 */
export interface SegmentRequestBody {
    [key: string]: any
}

/**
 * 文档状态请求体
 */
export interface DocumentStatusRequestBody {
    [key: string]: any
}