feat: Refactor file upload configuration and validation logic

This commit is contained in:
twwu
2025-04-24 13:46:50 +08:00
parent 93f83086c1
commit d768094376
13 changed files with 407 additions and 28 deletions

View File

@ -272,7 +272,7 @@ export const checkWatercrawlTaskStatus: Fetcher<CommonResponse, string> = (jobId
})
}
type FileTypesRes = {
export type FileTypesRes = {
allowed_extensions: string[]
}

View File

@ -5,13 +5,15 @@ import type {
StructuredOutputRulesResponse,
} from '@/models/common'
import { useMutation, useQuery } from '@tanstack/react-query'
import type { FileTypesRes } from './datasets'
const NAME_SPACE = 'common'
export const useFileUploadConfig = () => {
export const useFileUploadConfig = (enabled?: true) => {
return useQuery<FileUploadConfigResponse>({
queryKey: [NAME_SPACE, 'file-upload-config'],
queryFn: () => get<FileUploadConfigResponse>('/files/upload'),
enabled,
})
}
@ -26,3 +28,10 @@ export const useGenerateStructuredOutputRules = () => {
},
})
}
export const useFileSupportTypes = () => {
return useQuery<FileTypesRes>({
queryKey: [NAME_SPACE, 'file-types'],
queryFn: () => get<FileTypesRes>('/files/support-type'),
})
}