chore: add model api
This commit is contained in:
@@ -1,8 +1,9 @@
|
||||
import { request } from '@umijs/max';
|
||||
import { GPUDeviceItem, ListItem } from '../config/types';
|
||||
import { GPUDeviceItem, ListItem, ModelFile } from '../config/types';
|
||||
|
||||
export const WORKERS_API = '/workers';
|
||||
export const GPU_DEVICES_API = '/gpu-devices';
|
||||
export const MODEL_FILES_API = '/model-files';
|
||||
|
||||
export async function queryWorkersList(params: Global.SearchParams) {
|
||||
return request<Global.PageResponse<ListItem>>(`${WORKERS_API}`, {
|
||||
@@ -36,3 +37,39 @@ export async function updateWorker(id: string | number, data: any) {
|
||||
data
|
||||
});
|
||||
}
|
||||
|
||||
export async function queryModelFilesList(params: Global.SearchParams) {
|
||||
return request<Global.PageResponse<ModelFile>>(MODEL_FILES_API, {
|
||||
method: 'GET',
|
||||
params
|
||||
});
|
||||
}
|
||||
|
||||
export async function deleteModelFile(id: string | number) {
|
||||
return request<Global.PageResponse<ModelFile>>(`${MODEL_FILES_API}/${id}`, {
|
||||
method: 'DELETE'
|
||||
});
|
||||
}
|
||||
|
||||
export async function updateModelFile(id: string | number, data: any) {
|
||||
return request<Global.PageResponse<ModelFile>>(`${MODEL_FILES_API}/${id}`, {
|
||||
method: 'PUT',
|
||||
data
|
||||
});
|
||||
}
|
||||
|
||||
export async function downloadModelFile(data: any) {
|
||||
return request<Global.PageResponse<ModelFile>>(MODEL_FILES_API, {
|
||||
method: 'POST',
|
||||
data
|
||||
});
|
||||
}
|
||||
|
||||
export async function retryDownloadModelFile(id: string | number) {
|
||||
return request<Global.PageResponse<ModelFile>>(
|
||||
`${MODEL_FILES_API}/${id}/reset`,
|
||||
{
|
||||
method: 'POST'
|
||||
}
|
||||
);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user