| | |
| | | import cn.iocoder.yudao.module.ai.dal.dataobject.knowledge.AiKnowledgeDocumentDO; |
| | | import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper; |
| | | import org.apache.ibatis.annotations.Mapper; |
| | | |
| | | import java.util.Collection; |
| | | import java.util.List; |
| | | |
| | | /** |
| | | * AI 知识库文档 Mapper |
| | | * |
| | | * @author xiaoxin |
| | | */ |
| | | @Mapper |
| | | public interface AiKnowledgeDocumentMapper extends BaseMapperX<AiKnowledgeDocumentDO> { |
| | | |
| | |
| | | update(new LambdaUpdateWrapper<AiKnowledgeDocumentDO>() |
| | | .setSql(" retrieval_count = retrieval_count + 1") |
| | | .in(AiKnowledgeDocumentDO::getId, ids)); |
| | | } |
| | | |
| | | default List<AiKnowledgeDocumentDO> selectListByStatus(Integer status) { |
| | | return selectList(AiKnowledgeDocumentDO::getStatus, status); |
| | | } |
| | | |
| | | default List<AiKnowledgeDocumentDO> selectListByKnowledgeId(Long knowledgeId) { |