chore: add watch for worker and gpu updating

This commit is contained in:
jialin
2025-05-26 16:17:17 +08:00
parent 9efacefba3
commit 4b4343a9fd
8 changed files with 38 additions and 15 deletions
+4 -2
View File
@@ -8,7 +8,7 @@ import { useIntl } from '@umijs/max';
import { ConfigProvider, Empty, Table } from 'antd';
import _ from 'lodash';
import React from 'react';
import { queryGpuDevicesList } from '../apis';
import { GPU_DEVICES_API, queryGpuDevicesList } from '../apis';
import { GPUDeviceItem } from '../config/types';
const { Column } = Table;
@@ -48,7 +48,9 @@ const GPUList: React.FC = () => {
handleSearch,
handleNameChange
} = useTableFetch<GPUDeviceItem>({
fetchAPI: queryGpuDevicesList
fetchAPI: queryGpuDevicesList,
watch: true,
API: GPU_DEVICES_API
});
const intl = useIntl();
@@ -547,7 +547,7 @@ const ModelFiles = () => {
}
};
const columns = [
const columns: any[] = [
{
title: intl.formatMessage({ id: 'models.form.source' }),
dataIndex: 'source',
+9 -2
View File
@@ -18,7 +18,12 @@ import { ConfigProvider, Empty, Table, Tooltip, message } from 'antd';
import _ from 'lodash';
import React, { useCallback, useState } from 'react';
import { useHotkeys } from 'react-hotkeys-hook';
import { deleteWorker, queryWorkersList, updateWorker } from '../apis';
import {
WORKERS_API,
deleteWorker,
queryWorkersList,
updateWorker
} from '../apis';
import { WorkerStatusMapValue, status } from '../config';
import { Filesystem, GPUDeviceItem, ListItem } from '../config/types';
import AddWorker from './add-worker';
@@ -100,7 +105,9 @@ const Workers: React.FC = () => {
} = useTableFetch<ListItem>({
fetchAPI: queryWorkersList,
deleteAPI: deleteWorker,
contentForDelete: 'worker'
contentForDelete: 'worker',
watch: true,
API: WORKERS_API
});
const intl = useIntl();