424 lines
12 KiB
TypeScript
424 lines
12 KiB
TypeScript
import DeleteModal from '@/components/delete-modal';
|
|
import DropdownButtons from '@/components/drop-down-buttons';
|
|
import PageTools from '@/components/page-tools';
|
|
import SealTable from '@/components/seal-table';
|
|
import SealColumn from '@/components/seal-table/components/seal-column';
|
|
import { PageAction } from '@/config';
|
|
import type { PageActionType } from '@/config/types';
|
|
import useExpandedRowKeys from '@/hooks/use-expanded-row-keys';
|
|
import useTableRowSelection from '@/hooks/use-table-row-selection';
|
|
import useTableSort from '@/hooks/use-table-sort';
|
|
import { handleBatchRequest } from '@/utils';
|
|
import {
|
|
DeleteOutlined,
|
|
EditOutlined,
|
|
PlusOutlined,
|
|
SyncOutlined,
|
|
WechatWorkOutlined
|
|
} from '@ant-design/icons';
|
|
import { PageContainer } from '@ant-design/pro-components';
|
|
import { Access, useAccess, useIntl, useNavigate } from '@umijs/max';
|
|
import { Button, Input, Space, Tag, message } from 'antd';
|
|
import dayjs from 'dayjs';
|
|
import _ from 'lodash';
|
|
import { useCallback, useRef, useState } from 'react';
|
|
import {
|
|
MODELS_API,
|
|
MODEL_INSTANCE_API,
|
|
createModel,
|
|
deleteModel,
|
|
deleteModelInstance,
|
|
queryModelInstancesList,
|
|
updateModel
|
|
} from '../apis';
|
|
import { modelSourceMap } from '../config';
|
|
import { FormData, ListItem, ModelInstanceListItem } from '../config/types';
|
|
import AddModal from './add-modal';
|
|
import InstanceItem from './instance-item';
|
|
import ViewLogsModal from './view-logs-modal';
|
|
|
|
interface ModelsProps {
|
|
handleSearch: (e: any) => void;
|
|
handleNameChange: (e: any) => void;
|
|
handleShowSizeChange?: (page: number, size: number) => void;
|
|
handlePageChange: (page: number, pageSize: number | undefined) => void;
|
|
queryParams: {
|
|
page: number;
|
|
perPage: number;
|
|
query?: string;
|
|
};
|
|
dataSource: ListItem[];
|
|
loading: boolean;
|
|
total: number;
|
|
}
|
|
|
|
const Models: React.FC<ModelsProps> = ({
|
|
dataSource,
|
|
handleNameChange,
|
|
handleSearch,
|
|
handlePageChange,
|
|
queryParams,
|
|
loading,
|
|
total
|
|
}) => {
|
|
console.log('model list====2');
|
|
const access = useAccess();
|
|
const intl = useIntl();
|
|
const navigate = useNavigate();
|
|
const rowSelection = useTableRowSelection();
|
|
const { handleExpandChange, updateExpandedRowKeys, expandedRowKeys } =
|
|
useExpandedRowKeys();
|
|
const { sortOrder, setSortOrder } = useTableSort({
|
|
defaultSortOrder: 'descend'
|
|
});
|
|
const [openLogModal, setOpenLogModal] = useState(false);
|
|
const [openAddModal, setOpenAddModal] = useState(false);
|
|
const [action, setAction] = useState<PageActionType>(PageAction.CREATE);
|
|
const [title, setTitle] = useState<string>('');
|
|
const [currentData, setCurrentData] = useState<ListItem | undefined>(
|
|
undefined
|
|
);
|
|
const [currentInstanceUrl, setCurrentInstanceUrl] = useState<string>('');
|
|
const modalRef = useRef<any>(null);
|
|
|
|
const ActionList = [
|
|
{
|
|
label: 'common.button.edit',
|
|
key: 'edit',
|
|
icon: <EditOutlined />
|
|
},
|
|
{
|
|
label: 'models.openinplayground',
|
|
key: 'chat',
|
|
icon: <WechatWorkOutlined />
|
|
},
|
|
{
|
|
label: 'common.button.delete',
|
|
key: 'delete',
|
|
props: {
|
|
danger: true
|
|
},
|
|
icon: <DeleteOutlined />
|
|
}
|
|
];
|
|
|
|
const setActionList = useCallback((record: ListItem) => {
|
|
return _.filter(ActionList, (action: any) => {
|
|
if (action.key === 'chat') {
|
|
return record.ready_replicas > 0 && !record.embedding_only;
|
|
}
|
|
return true;
|
|
});
|
|
}, []);
|
|
|
|
const handleOnSort = (dataIndex: string, order: any) => {
|
|
console.log('handleTableChange=======', dataIndex, order);
|
|
setSortOrder(order);
|
|
};
|
|
|
|
const handleAddModal = () => {
|
|
setOpenAddModal(true);
|
|
setAction(PageAction.CREATE);
|
|
setTitle(intl.formatMessage({ id: 'models.button.deploy' }));
|
|
};
|
|
|
|
const handleModalOk = useCallback(
|
|
async (data: FormData) => {
|
|
try {
|
|
if (action === PageAction.CREATE) {
|
|
await createModel({ data });
|
|
}
|
|
if (action === PageAction.EDIT) {
|
|
await updateModel({ data, id: currentData?.id as number });
|
|
}
|
|
setOpenAddModal(false);
|
|
message.success(intl.formatMessage({ id: 'common.message.success' }));
|
|
} catch (error) {}
|
|
},
|
|
[action, currentData]
|
|
);
|
|
|
|
const handleModalCancel = useCallback(() => {
|
|
console.log('handleModalCancel');
|
|
setOpenAddModal(false);
|
|
}, []);
|
|
|
|
const handleLogModalCancel = useCallback(() => {
|
|
setOpenLogModal(false);
|
|
}, []);
|
|
const handleDelete = async (row: any) => {
|
|
modalRef.current.show({
|
|
content: 'models.table.models',
|
|
name: row.name,
|
|
async onOk() {
|
|
await deleteModel(row.id);
|
|
updateExpandedRowKeys([row.id]);
|
|
}
|
|
});
|
|
};
|
|
const handleDeleteBatch = () => {
|
|
modalRef.current.show({
|
|
content: 'models.table.models',
|
|
selection: true,
|
|
async onOk() {
|
|
await handleBatchRequest(rowSelection.selectedRowKeys, deleteModel);
|
|
rowSelection.clearSelections();
|
|
updateExpandedRowKeys(rowSelection.selectedRowKeys);
|
|
}
|
|
});
|
|
};
|
|
|
|
const handleOpenPlayGround = (row: any) => {
|
|
navigate(`/playground?model=${row.name}`);
|
|
};
|
|
|
|
const handleViewLogs = async (row: any) => {
|
|
try {
|
|
setCurrentInstanceUrl(`${MODEL_INSTANCE_API}/${row.id}/logs`);
|
|
setOpenLogModal(true);
|
|
} catch (error) {
|
|
console.log('error:', error);
|
|
}
|
|
};
|
|
const handleDeleteInstace = (row: any, list: ModelInstanceListItem[]) => {
|
|
modalRef.current.show({
|
|
content: 'models.instances',
|
|
name: row.name,
|
|
async onOk() {
|
|
await deleteModelInstance(row.id);
|
|
if (list.length === 1) {
|
|
updateExpandedRowKeys([row.model_id]);
|
|
}
|
|
}
|
|
});
|
|
};
|
|
|
|
const getModelInstances = async (row: any) => {
|
|
const params = {
|
|
id: row.id,
|
|
page: 1,
|
|
perPage: 100
|
|
};
|
|
const data = await queryModelInstancesList(params);
|
|
return data.items || [];
|
|
};
|
|
|
|
const generateChildrenRequestAPI = (params: any) => {
|
|
return `${MODELS_API}/${params.id}/instances`;
|
|
};
|
|
|
|
const handleEdit = (row: ListItem) => {
|
|
setCurrentData(row);
|
|
setOpenAddModal(true);
|
|
setAction(PageAction.EDIT);
|
|
setTitle(intl.formatMessage({ id: 'models.title.edit' }));
|
|
};
|
|
|
|
const handleSelect = useCallback((val: any, row: ListItem) => {
|
|
if (val === 'edit') {
|
|
console.log('row=======', row);
|
|
handleEdit(row);
|
|
}
|
|
if (val === 'chat') {
|
|
handleOpenPlayGround(row);
|
|
}
|
|
if (val === 'delete') {
|
|
handleDelete(row);
|
|
}
|
|
}, []);
|
|
|
|
const handleChildSelect = useCallback(
|
|
(val: any, row: ModelInstanceListItem, list: ModelInstanceListItem[]) => {
|
|
if (val === 'delete') {
|
|
handleDeleteInstace(row, list);
|
|
}
|
|
if (val === 'viewlog') {
|
|
handleViewLogs(row);
|
|
}
|
|
},
|
|
[]
|
|
);
|
|
|
|
const renderChildren = useCallback((list: any) => {
|
|
return (
|
|
<InstanceItem
|
|
list={list}
|
|
handleChildSelect={handleChildSelect}
|
|
></InstanceItem>
|
|
);
|
|
}, []);
|
|
|
|
return (
|
|
<>
|
|
<PageContainer
|
|
ghost
|
|
header={{
|
|
title: intl.formatMessage({ id: 'models.title' })
|
|
}}
|
|
extra={[]}
|
|
>
|
|
<PageTools
|
|
marginBottom={22}
|
|
left={
|
|
<Space>
|
|
<Input
|
|
placeholder={intl.formatMessage({ id: 'common.filter.name' })}
|
|
style={{ width: 300 }}
|
|
size="large"
|
|
allowClear
|
|
onChange={handleNameChange}
|
|
></Input>
|
|
<Button
|
|
type="text"
|
|
style={{ color: 'var(--ant-color-text-tertiary)' }}
|
|
onClick={handleSearch}
|
|
icon={<SyncOutlined></SyncOutlined>}
|
|
></Button>
|
|
</Space>
|
|
}
|
|
right={
|
|
<Space size={20}>
|
|
<Button
|
|
icon={<PlusOutlined></PlusOutlined>}
|
|
type="primary"
|
|
onClick={handleAddModal}
|
|
>
|
|
{intl?.formatMessage?.({ id: 'models.button.deploy' })}
|
|
</Button>
|
|
<Access accessible={access.canDelete}>
|
|
<Button
|
|
icon={<DeleteOutlined />}
|
|
danger
|
|
onClick={handleDeleteBatch}
|
|
disabled={!rowSelection.selectedRowKeys.length}
|
|
>
|
|
{intl?.formatMessage?.({ id: 'common.button.delete' })}
|
|
</Button>
|
|
</Access>
|
|
</Space>
|
|
}
|
|
></PageTools>
|
|
<SealTable
|
|
dataSource={dataSource}
|
|
rowSelection={rowSelection}
|
|
expandedRowKeys={expandedRowKeys}
|
|
onExpand={handleExpandChange}
|
|
loading={loading}
|
|
rowKey="id"
|
|
childParentKey="model_id"
|
|
expandable={true}
|
|
onSort={handleOnSort}
|
|
pollingChildren={false}
|
|
watchChildren={true}
|
|
loadChildren={getModelInstances}
|
|
loadChildrenAPI={generateChildrenRequestAPI}
|
|
renderChildren={renderChildren}
|
|
pagination={{
|
|
showSizeChanger: true,
|
|
pageSize: queryParams.perPage,
|
|
current: queryParams.page,
|
|
total: total,
|
|
hideOnSinglePage: queryParams.perPage === 10,
|
|
onChange: handlePageChange
|
|
}}
|
|
>
|
|
<SealColumn
|
|
title={intl.formatMessage({ id: 'common.table.name' })}
|
|
dataIndex="name"
|
|
key="name"
|
|
width={400}
|
|
span={5}
|
|
render={(text, record: ListItem) => {
|
|
return (
|
|
<span>
|
|
<span className="m-r-5">{text}</span>
|
|
<span>
|
|
{record.embedding_only && (
|
|
<Tag style={{ marginTop: 6 }} color="geekblue">
|
|
Embedding Only
|
|
</Tag>
|
|
)}
|
|
</span>
|
|
</span>
|
|
);
|
|
}}
|
|
/>
|
|
<SealColumn
|
|
title={intl.formatMessage({ id: 'models.form.source' })}
|
|
dataIndex="source"
|
|
key="source"
|
|
span={6}
|
|
render={(text, record: ListItem) => {
|
|
return (
|
|
<span className="flex flex-column">
|
|
<span>
|
|
{record.source === modelSourceMap.huggingface_value
|
|
? `${modelSourceMap.huggingface} / ${record.huggingface_filename}`
|
|
: `${modelSourceMap.ollama_library} / ${record.ollama_library_model_name}`}
|
|
</span>
|
|
</span>
|
|
);
|
|
}}
|
|
/>
|
|
<SealColumn
|
|
title={intl.formatMessage({ id: 'models.form.replicas' })}
|
|
dataIndex="replicas"
|
|
key="replicas"
|
|
align="center"
|
|
span={4}
|
|
render={(text, record: ListItem) => {
|
|
return (
|
|
<span>
|
|
{record.ready_replicas} / {record.replicas}
|
|
</span>
|
|
);
|
|
}}
|
|
/>
|
|
<SealColumn
|
|
span={5}
|
|
title={intl.formatMessage({ id: 'common.table.createTime' })}
|
|
dataIndex="created_at"
|
|
key="created_at"
|
|
defaultSortOrder="descend"
|
|
sortOrder={sortOrder}
|
|
sorter={false}
|
|
render={(text, row) => {
|
|
return dayjs(text).format('YYYY-MM-DD HH:mm:ss');
|
|
}}
|
|
/>
|
|
<SealColumn
|
|
span={4}
|
|
title={intl.formatMessage({ id: 'common.table.operation' })}
|
|
key="operation"
|
|
dataIndex="operation"
|
|
render={(text, record) => {
|
|
return (
|
|
<DropdownButtons
|
|
items={setActionList(record)}
|
|
onSelect={(val) => handleSelect(val, record)}
|
|
></DropdownButtons>
|
|
);
|
|
}}
|
|
/>
|
|
</SealTable>
|
|
</PageContainer>
|
|
<AddModal
|
|
open={openAddModal}
|
|
action={action}
|
|
title={title}
|
|
data={currentData}
|
|
onCancel={handleModalCancel}
|
|
onOk={handleModalOk}
|
|
></AddModal>
|
|
<ViewLogsModal
|
|
url={currentInstanceUrl}
|
|
open={openLogModal}
|
|
onCancel={handleLogModalCancel}
|
|
></ViewLogsModal>
|
|
<DeleteModal ref={modalRef}></DeleteModal>
|
|
</>
|
|
);
|
|
};
|
|
|
|
export default Models;
|