package cn.iocoder.yudao.module.ai.service.knowledge;

import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.module.ai.controller.admin.knowledge.vo.segment.*;
import cn.iocoder.yudao.module.ai.dal.dataobject.knowledge.AiKnowledgeSegmentDO;
import cn.iocoder.yudao.module.ai.service.knowledge.bo.AiKnowledgeSegmentSearchReqBO;
import cn.iocoder.yudao.module.ai.service.knowledge.bo.AiKnowledgeSegmentSearchRespBO;

import java.util.List;

public interface AiKnowledgeSegmentService {

    AiKnowledgeSegmentDO getSegment(Long id);

    AiKnowledgeSegmentDO validateSegmentExists(Long id);

    PageResult<AiKnowledgeSegmentDO> getSegmentPage(AiKnowledgeSegmentPageReqVO pageReqVO);

    Long createSegment(AiKnowledgeSegmentSaveReqVO saveReqVO);

    void updateSegment(AiKnowledgeSegmentSaveReqVO saveReqVO);

    void deleteSegment(Long id);

    void updateSegmentStatus(AiKnowledgeSegmentUpdateStatusReqVO updateStatusReqVO);

    void deleteSegmentsByDocumentId(Long documentId);

    void saveSegments(List<AiKnowledgeSegmentDO> segments, Long knowledgeId, Long embeddingModelId);

    List<AiKnowledgeSegmentSearchRespBO> searchSegments(AiKnowledgeSegmentSearchReqBO searchReqBO);

}