diff --git a/web/service/knowledge/use-segement.ts b/web/service/knowledge/use-segement.ts deleted file mode 100644 index e69de29bb2..0000000000 diff --git a/web/service/knowledge/use-segment.ts b/web/service/knowledge/use-segment.ts new file mode 100644 index 0000000000..ae2100f9c4 --- /dev/null +++ b/web/service/knowledge/use-segment.ts @@ -0,0 +1,38 @@ +import { useMutation } from '@tanstack/react-query' +import { del, patch } from '../base' +import type { CommonResponse } from '@/models/common' + +const NAME_SPACE = 'segment' + +export const useEnableSegment = () => { + return useMutation({ + mutationKey: [NAME_SPACE, 'enable'], + mutationFn: (payload: { datasetId: string; segmentIds: string[] }) => { + const { datasetId, segmentIds } = payload + const query = segmentIds.map(id => `segment_id=${id}`).join('&') + return patch(`/datasets/${datasetId}/segments/enable?${query}`) + }, + }) +} + +export const useDisableSegment = () => { + return useMutation({ + mutationKey: [NAME_SPACE, 'disable'], + mutationFn: (payload: { datasetId: string; segmentIds: string[] }) => { + const { datasetId, segmentIds } = payload + const query = segmentIds.map(id => `segment_id=${id}`).join('&') + return patch(`/datasets/${datasetId}/segments/disable?${query}`) + }, + }) +} + +export const useDeleteSegment = () => { + return useMutation({ + mutationKey: [NAME_SPACE, 'delete'], + mutationFn: (payload: { datasetId: string; documentId: string; segmentIds: string[] }) => { + const { datasetId, documentId, segmentIds } = payload + const query = segmentIds.map(id => `segment_id=${id}`).join('&') + return del(`/datasets/${datasetId}/documents/${documentId}/segments?${query}`) + }, + }) +}