import { get, post, put, remove, upload } from '@/utils/request'

export const knowledgeDatasets = (params = {}) => get('/router/rest/knowledge/base/datasets', params)

export const knowledgeDocuments = (params = {}) => {
    const { dataset_id, ...query } = params
    return get(`/router/rest/knowledge/document/${dataset_id}/documents`, params)
}

export const createKnowledge = (data = {}) => {
    const { dataset_id, ...query } = data
    return post(`/router/rest/knowledge/document/${dataset_id}/document/create_by_text`, data)
}
export const chatMessages = (params = {}) => get('/router/rest/chat/messages', params)

export const baseTraining = (params = {}) => get('/router/rest/knowledge/base/training/page', params)

export const creatTraining = (data = {}) => post('/router/rest/knowledge/base/training', data)

export const baseStatistics = (params = {}) => get('/router/rest/knowledge/base/statistics', params)

export const getAllChunks = (params = {}) => get('/router/rest/knowledge/chunk/getAllChunks', params)

export const getAllDocumnets = (params = {}) => {
    const query = { ...params }
    if ([undefined, null, ''].includes(query.dataset_id)) {
        delete query.dataset_id
    }
    return get('/router/rest/knowledge/document/getAllDocumnets', query)
}

export const getDocumentDetail = (params = {}) => {
    const { dataset_id,document_id, ...query } = params
    return get(`/router/rest/knowledge/document/${dataset_id}/documents/${document_id}`, params)
}

export const delDocuments = (params = {}) => {
    const { dataset_id, document_id, ...query } = params
    return remove(`/router/rest/knowledge/document/${dataset_id}/documents/${document_id}`, params)
}

export const createChunk = (data = {}) => {
    const { dataset_id,document_id, ...query } = data
    return post(`/router/rest/knowledge/chunk/${dataset_id}/${document_id}`, data)
}

export const delChunk = (params = {}) => {
    const { dataset_id, document_id,segment_id, ...query } = params
    return  remove(`/router/rest/knowledge/chunk/${dataset_id}/${document_id}/${segment_id}`, params)
}

export const editChunk = (data = {}) => {
    const { dataset_id,document_id,segment_id, ...query } = data
    return post(`/router/rest/knowledge/chunk/${dataset_id}/${document_id}/${segment_id}`, data)
}

export const blocking = (data = {}) => post('/router/rest/chat/blocking', data,{
    headers: {
        'Authorization': 'Bearer app-KpoSEqve0ybB2Ej5PIJiBu2k'
    }
})

export const downloadDocument = (params = {}) => {
    const { dataset_id,document_id, ...query } = params
    return get(`/router/rest/knowledge/document/${dataset_id}/documents/${document_id}/download`, params)
}