package com.ruoyi.approve.service;

import com.baomidou.mybatisplus.extension.service.IService;
import com.ruoyi.approve.dto.KnowledgeBaseVectorVO;
import com.ruoyi.approve.pojo.KnowledgeBaseVector;

import java.util.List;

/**
 * 知识库文件向量记录 Service
 */
public interface KnowledgeBaseVectorService extends IService<KnowledgeBaseVector> {

    /**
     * 查询知识库的文件向量状态列表
     */
    List<KnowledgeBaseVectorVO> getVectorStatusByKnowledgeBaseId(Long knowledgeBaseId);

    /**
     * 创建向量记录并触发异步向量化
     */
    KnowledgeBaseVector createVectorRecord(Long knowledgeBaseId, Long storageBlobId, String fileName, String fileType);

    /**
     * 更新向量状态
     */
    void updateVectorStatus(Long id, Integer status, Integer chunkCount, String error);

    /**
     * 重新处理向量化
     */
    void reprocessVector(Long id);

    /**
     * 删除向量记录及相关向量数据
     */
    void deleteVector(Long id);

    /**
     * 批量删除向量记录
     */
    void deleteVectors(List<Long> ids);
}