package com.ruoyi.approve.mapper;
|
|
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
|
import com.ruoyi.approve.dto.KnowledgeBaseVectorVO;
|
import com.ruoyi.approve.pojo.KnowledgeBaseVector;
|
import org.apache.ibatis.annotations.Mapper;
|
import org.apache.ibatis.annotations.Param;
|
import org.apache.ibatis.annotations.Select;
|
|
import java.util.List;
|
|
/**
|
* 知识库文件向量记录 Mapper
|
*/
|
@Mapper
|
public interface KnowledgeBaseVectorMapper extends BaseMapper<KnowledgeBaseVector> {
|
|
/**
|
* 查询知识库的文件向量状态列表
|
*/
|
@Select("SELECT v.*, b.path as previewUrl " +
|
"FROM knowledge_base_vector v " +
|
"LEFT JOIN storage_blob b ON v.storage_blob_id = b.id " +
|
"WHERE v.knowledge_base_id = #{knowledgeBaseId} " +
|
"ORDER BY v.create_time DESC")
|
List<KnowledgeBaseVectorVO> selectByKnowledgeBaseId(@Param("knowledgeBaseId") Long knowledgeBaseId);
|
|
/**
|
* 统计知识库的文件数量
|
*/
|
@Select("SELECT COUNT(*) FROM knowledge_base_vector WHERE knowledge_base_id = #{knowledgeBaseId}")
|
int countByKnowledgeBaseId(@Param("knowledgeBaseId") Long knowledgeBaseId);
|
|
/**
|
* 统计知识库的总切片数量
|
*/
|
@Select("SELECT COALESCE(SUM(chunk_count), 0) FROM knowledge_base_vector WHERE knowledge_base_id = #{knowledgeBaseId} AND vector_status = 2")
|
int sumChunkCountByKnowledgeBaseId(@Param("knowledgeBaseId") Long knowledgeBaseId);
|
}
|
