import { create } from 'domain';
import api from './api';

// Api参数类型
export type LoginApiParams = {
    userName: string,
    password: string,
};

export type FetchAppListApiParams = {
    pageNumber: number,
    pageSize: number,
};

export type FetchKnowledgeLibListApiParams = {
    pageNumber: number,
    pageSize: number,
};

export type CreateOrModifyKnowledgeLibApiParams = {
    name: string,
    embeddingId: string,
    description: string,
};

export type FetchAppIndexParams = {
    pageNumber: number,
    pageSize: number,
};

export type CreateOrModifyApplicationApiParams = {
    name: string,
    desc: string,
    prompt: string,
    temperature: number,
    top_p: number,
    knowledge_ids: string[],
    param_desc: string,
    max_token: number,
    knowledge_info: {
        model: string,
        knowledge_ids: string[],
        slice_config_type: string,
        recall_method: string,
        slice_count: number,
        rerank_status: number,
        rerank_model_name: string,
        show_recall_result: boolean,
        recall_slice_splicing_method: string,
    },
    questionList: [
        {
            question: string,
        }
    ]
};

export type FetchDocumentLibListApiParams = {
    knowledge_id: string,
    page: number,
    size: number,
}

export type ModifyDocumentApiParams = {
    id: string,
    name: string,
    url: string,
    length: number,
    sentence_size: number,
    knowledge_type: number,
    custom_separator: string[],
    embedding_stat: number,
    word_num: number,
    parse_image: number
};

export type FetchChatHistoryListApiParams = Partial<{
    appId: string,
    pageNumber: number,
    pageSize: number,
}>;

// Api函数类型
export type LoginApi = (data: LoginApiParams) => Promise<any>;
export type LogoutApi = () => Promise<any>;
export type FetchAppListApi = (data: FetchAppListApiParams) => Promise<any>;
export type FetchKnowledgeLibListApi = (data: FetchKnowledgeLibListApiParams) => Promise<any>;
export type FetchEmbeddingListApi = () => Promise<any>;
export type CreateKnowledgeLibApi = (data: CreateOrModifyKnowledgeLibApiParams) => Promise<any>;
export type ModifyKnowledgeLibApi = (knowledgeId: string, data: CreateOrModifyKnowledgeLibApiParams) => Promise<any>;
export type DeleteKnowledgeLibApi = (knowledgeId: string) => Promise<any>;
export type FetchKnowledgeLibDetailApi = (knowledgeId: string) => Promise<any>;
export type FetchAppIndexApi = (data: FetchAppIndexParams) => Promise<any>;
export type FetchApplicationDetailApi = (appId: string) => Promise<any>;
export type FetchModelListApi = () => Promise<any>;
export type FetchKnowledgeApiListApi = () => Promise<any>;
export type ModifyApplicationLibApi = (appId: string, data: CreateOrModifyApplicationApiParams) => Promise<any>;
export type DeleteApplicationLibApi = (appId: string) => Promise<any>;
export type CreateApplicationLibApi = (data: CreateOrModifyApplicationApiParams) => Promise<any>;
export type uploadFileLibApi = (konwledgeId: string) => Promise<any>;
export type FetchDocumentLibListApi = (data: FetchDocumentLibListApiParams) => Promise<any>;
export type FetchDocumentLibApi = (id: string) => Promise<any>;
export type ModifyDocumentLibApi = (id: string, data: ModifyDocumentApiParams) => Promise<any>;
export type DeleteDocumentLibApi = (id: string) => Promise<any>;
export type FetchApplicationListApi = () => Promise<any>;
export type FetchChatHistoryListApi = (data: FetchChatHistoryListApiParams) => Promise<any>;
export type ExportChatHistoryApi = (id: string) => Promise<any>;

// 登录
const loginApi: LoginApi = async (data) => {
    return api.post('/login', data);
};

// 登出
const logoutApi: LogoutApi = async () => {
    return api.post('/logout');
};

// 获取应用列表
const fetchAppListApi: FetchAppListApi = async (data) => {
    return api.post('/bigmodel/api/getApplicationList', data);
};

// 获取知识库列表
const fetchKnowledgeLibListApi: FetchKnowledgeLibListApi = async (data) => {
    return api.post('bigmodel/api/knowledgeList', data);
};

// 获取向量化模型列表
const fetchEmbeddingListApi: FetchEmbeddingListApi = async () => {
    return api.get('bigmodel/api/embedding');
};

// 创建知识库
const createKnowledgeLibApi: CreateKnowledgeLibApi = async (data) => {
    return api.post('bigmodel/api/createKnowledge', data);
};

// 修改知识库
const modifyKnowledgeLibApi: ModifyKnowledgeLibApi = async (knowledgeId, data) => {
    return api.put(`bigmodel/api/updateKnowledge/${knowledgeId}`, data);
};

// 删除知识库
const deleteKnowledgeLibApi: DeleteKnowledgeLibApi = async (knowledgeId) => {
    return api.delete(`bigmodel/api/delKnowledge/${knowledgeId}`);
};

// 获取知识库详情
const fetchKnowledgeLibDetailApi: FetchKnowledgeLibDetailApi = async (knowledgeId) => {
    return api.get(`bigmodel/api/detailKnowledge/${knowledgeId}`);
};

// 获取首页信息
const fetchIndexApi: FetchAppIndexApi = async (data) => {
    return api.post('/bigmodel/api/index', data);
};

// 获取应用详情
const fetchApplicationDetailApi: FetchApplicationDetailApi = async (appId) => {
    return api.get(`bigmodel/api/selectApplication/${appId}`);
};

// 模型列表
const fetchModelListApi: FetchModelListApi = async () => {
    return api.get('/bigmodel/api/model/list');
};

// 知识库列表
const fetchKnowledgeListApi: FetchKnowledgeApiListApi = async () => {
    return api.get('/bigmodel/api/applicationList');
};

// 编辑应用
const modifyApplicationLibApi: ModifyApplicationLibApi = async (appId, data) => {
    return api.put(`bigmodel/api/updateApplication/${appId}`, data);
};

// 删除应用
const deleteApplicationLibApi: DeleteApplicationLibApi = async (appId) => {
    return api.delete(`bigmodel/api/delApplication/${appId}`);
};

// 创建应用
const createApplicationLibApi: CreateApplicationLibApi = async (data) => {
    return api.post('bigmodel/api/createApplication', data);
};

// 上传知识文件
const uploadFileLibApi: uploadFileLibApi = async (konwledgeId) => {
    return api.post(`bigmodel/api/uploadDocument/${konwledgeId}`);
};

// 获取知识列表
const fetchDocumentLibListApi: FetchDocumentLibListApi = async (data) => {
    return api.post('bigmodel/api/documentList', data);
};

// 知识详情
const fetchDocumentLibApi: FetchDocumentLibApi = async (id) => {
    return api.get(`bigmodel/api/documentDetail/${id}`);
};

// 修改知识
const modifyDocumentApi: ModifyDocumentLibApi = async (id, data) => {
    return api.put(`bigmodel/api/updateDocument/${id}`, data);
};

// 删除知识文件
const deleteDocumentApi: DeleteDocumentLibApi = async (id) => {
    return api.delete(`bigmodel/api/delDocument/${id}`);
};

// 获取应用列表
const fetchApplicationListApi: FetchApplicationListApi = async () => {
    return api.get('bigmodel/api/application/list');
};

// 获取聊天记录列表
const fetchChatHistoryListApi: FetchChatHistoryListApi = async (data) => {
    return api.post('bigmodel/api/chatHistory/list', data);
};

// 导出聊天记录
const exportChatHistoryApi: ExportChatHistoryApi = async (id) => {
    return api.post(`bigmodel/api/dialog/export/${id}`, {}, { responseType: 'blob' });
};

export const apis = {
    login: loginApi,
    logout: logoutApi,
    fetchAppList: fetchAppListApi,
    fetchKnowledgeLibList: fetchKnowledgeLibListApi,
    fetchEmbeddingList: fetchEmbeddingListApi,
    createKnowledgeLib: createKnowledgeLibApi,
    modifyKnowledgeLib: modifyKnowledgeLibApi,
    deleteKnowledgeLib: deleteKnowledgeLibApi,
    fetchKnowledgeLibDetail: fetchKnowledgeLibDetailApi,
    fetchIndexCount: fetchIndexApi,
    fetchApplicationDetail: fetchApplicationDetailApi,
    fetchModelList: fetchModelListApi,
    fetchKnowledgeList: fetchKnowledgeListApi,
    modifyApplicationApi: modifyApplicationLibApi,
    deleteApplicationApi: deleteApplicationLibApi,
    createApplicationApi: createApplicationLibApi,
    uploadFileLibApi: uploadFileLibApi,
    fetchDocumentLibListApi: fetchDocumentLibListApi,
    fetchDocumentDetailLibApi: fetchDocumentLibApi,
    modifyDocumentLibApi: modifyDocumentApi,
    deleteDocumentLibApi: deleteDocumentApi,
    fetchApplicationList: fetchApplicationListApi,
    fetchChatHistoryList: fetchChatHistoryListApi,
    exportChatHistory: exportChatHistoryApi,
};