package cn.iocoder.yudao.module.llm.service.modelservice;

import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.module.llm.controller.admin.modelservice.vo.ModelServicePageReqVO;
import cn.iocoder.yudao.module.llm.controller.admin.modelservice.vo.ModelServiceRespVO;
import cn.iocoder.yudao.module.llm.controller.admin.modelservice.vo.ModelServiceSaveReqVO;
import cn.iocoder.yudao.module.llm.dal.dataobject.modelservice.ModelServiceDO;

import javax.validation.Valid;
import java.util.List;
import java.util.Map;

/**
 * 模型服务 Service 接口
 *
 * @author 华大大模型
 */
public interface ModelServiceService {

    /**
     * 创建模型服务
     *
     * @param createReqVO 创建信息
     * @return 编号
     */
    Long createModelService (@Valid ModelServiceSaveReqVO createReqVO);

    /**
     * 更新模型服务
     *
     * @param updateReqVO 更新信息
     */
    void updateModelService (@Valid ModelServiceSaveReqVO updateReqVO);

    /**
     * 删除模型服务
     *
     * @param id 编号
     */
    void deleteModelService (Long id);

    /**
     * 获得模型服务
     *
     * @param id 编号
     * @return 模型服务
     */
    ModelServiceDO getModelService (Long id);

    /**
     * 获得模型服务分页
     *
     * @param pageReqVO 分页查询
     * @return 模型服务分页
     */
    PageResult<ModelServiceDO> getModelServicePage (ModelServicePageReqVO pageReqVO);

    /**
     * 查询所有模型服务
     *
     * @return
     */
    List<ModelServiceDO> getModelServiceList ();

    void active (ModelServiceSaveReqVO updateReqVO);

    void unactive (ModelServiceSaveReqVO updateReqVO);

    PageResult<ModelServiceRespVO> getModelServicePage1 (ModelServicePageReqVO pageReqVO);

    ModelServiceDO getByApiKey (String apiKey);

    List<String> getDownLoadList (Long id);

    /**
     * 获取模型服务管理数量
     *
     * @return 模型服务管理数量
     */
    Long getCount ();


    /**
     * 根据 微调任务 id 获取 模型服务
     *
     * @param taskId 知识库 id
     * @return 模型服务的 Map 集合
     */
    Map<Long, String> getModelServiceByTaskId (Long taskId);
}
