QuestionEmbedding.api.ts 1.8 KB
import { defHttp } from '/@/utils/http/axios';
import { useMessage } from '/@/hooks/web/useMessage';

const { createConfirm } = useMessage();

enum Api {
  list = '/question/embedding/list',
  save = '/question/embedding/add',
  edit = '/question/embedding/edit',
  deleteOne = '/question/embedding/delete',
  deleteBatch = '/question/embedding/deleteBatch',
  importExcel = '/question/embedding/importZip',
  uploadZip = '/question/embedding/uploadZip',
}

export const getExportUrl = Api.importExcel;
export const getImportZipUrl = Api.uploadZip;

export const list = async (params) => {
  try {
    const res = await defHttp.get({
      url: Api.list,
      params: { ...params, size: 1000 }
    });

    if (res?.records && Array.isArray(res.records)) {
      res.records = res.records.map(item => ({
        ...item,
        ...item.metadata,
        question: item.question || '',
        answer: item.answer || ''
      }));
    }
    return res;
  } catch (error) {
    console.error("Error fetching question embeddings:", error);
    return {
      records: [],
      total: 0,
      size: 10,
      current: 1,
      pages: 0
    };
  }
}

export const deleteOne = (params, handleSuccess) => {
  return defHttp.delete({ url: Api.deleteOne, params }, { joinParamsToUrl: true }).then(() => {
    handleSuccess();
  });
};

export const batchDelete = (params, handleSuccess) => {
  createConfirm({
    iconType: 'warning',
    title: '确认删除',
    content: '是否删除选中问答数据',
    okText: '确认',
    cancelText: '取消',
    onOk: () => {
      return defHttp.delete({ url: Api.deleteBatch, data: params }, { joinParamsToUrl: true }).then(() => {
        handleSuccess();
      });
    },
  });
};

export const saveOrUpdate = (params, isUpdate) => {
  const url = isUpdate ? Api.edit : Api.save;
  return defHttp.post({ url: url, params });
};