chore: column settings

This commit is contained in:
jialin
2026-02-01 20:22:25 +08:00
parent 043218937f
commit 0849a2f114
16 changed files with 611 additions and 293 deletions
+1 -1
View File
@@ -113,7 +113,7 @@ const SimpleSelect: React.FC<SelectProps & { ref?: any; showTags?: boolean }> =
indeterminate={allSelection.indeterminate} indeterminate={allSelection.indeterminate}
onChange={handleOnCheckboxChange} onChange={handleOnCheckboxChange}
> >
{intl.formatMessage({ id: 'common.checbox.all' })} {intl.formatMessage({ id: 'common.checkbox.all' })}
</Checkbox> </Checkbox>
</SelectAllWrapper> </SelectAllWrapper>
)} )}
+9 -8
View File
@@ -40,20 +40,21 @@ export default {
'benchmark.table.filter.bygpu': 'Filter by GPU', 'benchmark.table.filter.bygpu': 'Filter by GPU',
'benchmark.table.filter.bymodel': 'Filter by Model', 'benchmark.table.filter.bymodel': 'Filter by Model',
'benchmark.table.filter.bydataset': 'Filter by Dataset', 'benchmark.table.filter.bydataset': 'Filter by Dataset',
'benchmark.table.avg': 'Average', 'benchmark.table.avg': 'Avg',
'benchmark.table.columnSettings': 'Column Settings', 'benchmark.table.columnSettings': 'Column Settings',
'benchmark.detail.summary.results': 'Test Results', 'benchmark.detail.summary.results': 'Test Results',
'benchmark.detail.summary.metadata': 'Metadata',
'benchmark.detail.summary.metrics': 'Test Metrics', 'benchmark.detail.summary.metrics': 'Test Metrics',
'benchmark.detail.summary.latency': 'Latency', 'benchmark.detail.summary.latency': 'Latency',
'benchmark.detail.summary.throughput': 'Throughput', 'benchmark.detail.summary.throughput': 'Throughput',
'benchmark.detail.summary.request': 'Requests', 'benchmark.detail.summary.request': 'Requests',
'benchmark.detail.avg.reqLatency': 'Average Request Latency', 'benchmark.detail.avg.reqLatency': 'Request Latency Avg',
'benchmark.detail.avg.ttft': 'Average TTFT', 'benchmark.detail.avg.ttft': 'TTFT Avg',
'benchmark.detail.avg.tpot': 'Average TPOT', 'benchmark.detail.avg.tpot': 'TPOT Avg',
'benchmark.detail.avg.itl': 'Average ITL', 'benchmark.detail.avg.itl': 'ITL Avg',
'benchmark.detail.throughput.totalToken': 'Total Token Throughput', 'benchmark.detail.throughput.totalToken': 'Total Throughput',
'benchmark.detail.throughput.inputToken': 'Input Token Throughput', 'benchmark.detail.throughput.inputToken': 'Input Throughput',
'benchmark.detail.throughput.outputToken': 'Output Token Throughput', 'benchmark.detail.throughput.outputToken': 'Output Throughput',
'benchmark.detail.requests.total': 'Total Requests', 'benchmark.detail.requests.total': 'Total Requests',
'benchmark.detail.requests.success': 'Successful Requests', 'benchmark.detail.requests.success': 'Successful Requests',
'benchmark.detail.requests.failed': 'Failed Requests', 'benchmark.detail.requests.failed': 'Failed Requests',
+1 -1
View File
@@ -97,7 +97,7 @@ export default {
'common.form.field.input.required': 'required', 'common.form.field.input.required': 'required',
'common.form.field.select.required': 'required', 'common.form.field.select.required': 'required',
'common.select.option': 'All', 'common.select.option': 'All',
'common.checbox.all': 'Select all', 'common.checkbox.all': 'Select all',
'common.select.all': 'All {type}', 'common.select.all': 'All {type}',
'common.data.unkonwn': 'Unknown', 'common.data.unkonwn': 'Unknown',
'common.data.none': 'No Data', 'common.data.none': 'No Data',
+9 -8
View File
@@ -40,20 +40,21 @@ export default {
'benchmark.table.filter.bygpu': 'Filter by GPU', 'benchmark.table.filter.bygpu': 'Filter by GPU',
'benchmark.table.filter.bymodel': 'Filter by Model', 'benchmark.table.filter.bymodel': 'Filter by Model',
'benchmark.table.filter.bydataset': 'Filter by Dataset', 'benchmark.table.filter.bydataset': 'Filter by Dataset',
'benchmark.table.avg': 'Average', 'benchmark.table.avg': 'Avg',
'benchmark.table.columnSettings': 'Column Settings', 'benchmark.table.columnSettings': 'Column Settings',
'benchmark.detail.summary.results': 'Test Results', 'benchmark.detail.summary.results': 'Test Results',
'benchmark.detail.summary.metadata': 'Metadata',
'benchmark.detail.summary.metrics': 'Test Metrics', 'benchmark.detail.summary.metrics': 'Test Metrics',
'benchmark.detail.summary.latency': 'Latency', 'benchmark.detail.summary.latency': 'Latency',
'benchmark.detail.summary.throughput': 'Throughput', 'benchmark.detail.summary.throughput': 'Throughput',
'benchmark.detail.summary.request': 'Requests', 'benchmark.detail.summary.request': 'Requests',
'benchmark.detail.avg.reqLatency': 'Average Request Latency', 'benchmark.detail.avg.reqLatency': 'Request Latency Avg',
'benchmark.detail.avg.ttft': 'Average TTFT', 'benchmark.detail.avg.ttft': 'TTFT Avg',
'benchmark.detail.avg.tpot': 'Average TPOT', 'benchmark.detail.avg.tpot': 'TPOT Avg',
'benchmark.detail.avg.itl': 'Average ITL', 'benchmark.detail.avg.itl': 'ITL Avg',
'benchmark.detail.throughput.totalToken': 'Total Token Throughput', 'benchmark.detail.throughput.totalToken': 'Total Throughput',
'benchmark.detail.throughput.inputToken': 'Input Token Throughput', 'benchmark.detail.throughput.inputToken': 'Input Throughput',
'benchmark.detail.throughput.outputToken': 'Output Token Throughput', 'benchmark.detail.throughput.outputToken': 'Output Throughput',
'benchmark.detail.requests.total': 'Total Requests', 'benchmark.detail.requests.total': 'Total Requests',
'benchmark.detail.requests.success': 'Successful Requests', 'benchmark.detail.requests.success': 'Successful Requests',
'benchmark.detail.requests.failed': 'Failed Requests', 'benchmark.detail.requests.failed': 'Failed Requests',
+1 -1
View File
@@ -97,7 +97,7 @@ export default {
'common.form.field.input.required': '必須', 'common.form.field.input.required': '必須',
'common.form.field.select.required': '必須', 'common.form.field.select.required': '必須',
'common.select.option': 'すべて', 'common.select.option': 'すべて',
'common.checbox.all': 'すべて', 'common.checkbox.all': 'すべて',
'common.select.all': 'すべての {type}', 'common.select.all': 'すべての {type}',
'common.data.unkonwn': '不明', 'common.data.unkonwn': '不明',
'common.data.none': 'データなし', 'common.data.none': 'データなし',
+9 -8
View File
@@ -40,20 +40,21 @@ export default {
'benchmark.table.filter.bygpu': 'Filter by GPU', 'benchmark.table.filter.bygpu': 'Filter by GPU',
'benchmark.table.filter.bymodel': 'Filter by Model', 'benchmark.table.filter.bymodel': 'Filter by Model',
'benchmark.table.filter.bydataset': 'Filter by Dataset', 'benchmark.table.filter.bydataset': 'Filter by Dataset',
'benchmark.table.avg': 'Average', 'benchmark.table.avg': 'Avg',
'benchmark.table.columnSettings': 'Column Settings', 'benchmark.table.columnSettings': 'Column Settings',
'benchmark.detail.summary.results': 'Test Results', 'benchmark.detail.summary.results': 'Test Results',
'benchmark.detail.summary.metadata': 'Metadata',
'benchmark.detail.summary.metrics': 'Test Metrics', 'benchmark.detail.summary.metrics': 'Test Metrics',
'benchmark.detail.summary.latency': 'Latency', 'benchmark.detail.summary.latency': 'Latency',
'benchmark.detail.summary.throughput': 'Throughput', 'benchmark.detail.summary.throughput': 'Throughput',
'benchmark.detail.summary.request': 'Requests', 'benchmark.detail.summary.request': 'Requests',
'benchmark.detail.avg.reqLatency': 'Average Request Latency', 'benchmark.detail.avg.reqLatency': 'Request Latency Avg',
'benchmark.detail.avg.ttft': 'Average TTFT', 'benchmark.detail.avg.ttft': 'TTFT Avg',
'benchmark.detail.avg.tpot': 'Average TPOT', 'benchmark.detail.avg.tpot': 'TPOT Avg',
'benchmark.detail.avg.itl': 'Average ITL', 'benchmark.detail.avg.itl': 'ITL Avg',
'benchmark.detail.throughput.totalToken': 'Total Token Throughput', 'benchmark.detail.throughput.totalToken': 'Total Throughput',
'benchmark.detail.throughput.inputToken': 'Input Token Throughput', 'benchmark.detail.throughput.inputToken': 'Input Throughput',
'benchmark.detail.throughput.outputToken': 'Output Token Throughput', 'benchmark.detail.throughput.outputToken': 'Output Throughput',
'benchmark.detail.requests.total': 'Total Requests', 'benchmark.detail.requests.total': 'Total Requests',
'benchmark.detail.requests.success': 'Successful Requests', 'benchmark.detail.requests.success': 'Successful Requests',
'benchmark.detail.requests.failed': 'Failed Requests', 'benchmark.detail.requests.failed': 'Failed Requests',
+1 -1
View File
@@ -97,7 +97,7 @@ export default {
'common.form.field.input.required': 'обязательно', 'common.form.field.input.required': 'обязательно',
'common.form.field.select.required': 'обязательно', 'common.form.field.select.required': 'обязательно',
'common.select.option': 'Все', 'common.select.option': 'Все',
'common.checbox.all': 'Все', 'common.checkbox.all': 'Все',
'common.select.all': 'Все {type}', 'common.select.all': 'Все {type}',
'common.data.unkonwn': 'Неизвестно', 'common.data.unkonwn': 'Неизвестно',
'common.data.none': 'Нет данных', 'common.data.none': 'Нет данных',
+8 -7
View File
@@ -43,17 +43,18 @@ export default {
'benchmark.table.avg': '均值', 'benchmark.table.avg': '均值',
'benchmark.table.columnSettings': '列设置', 'benchmark.table.columnSettings': '列设置',
'benchmark.detail.summary.results': '测试结果', 'benchmark.detail.summary.results': '测试结果',
'benchmark.detail.summary.metadata': '元数据',
'benchmark.detail.summary.metrics': '测试指标', 'benchmark.detail.summary.metrics': '测试指标',
'benchmark.detail.summary.latency': '延迟', 'benchmark.detail.summary.latency': '延迟',
'benchmark.detail.summary.throughput': '吞吐', 'benchmark.detail.summary.throughput': '吞吐',
'benchmark.detail.summary.request': '请求数', 'benchmark.detail.summary.request': '请求数',
'benchmark.detail.avg.reqLatency': '平均请求延迟', 'benchmark.detail.avg.reqLatency': '请求延迟 均值',
'benchmark.detail.avg.ttft': '平均 TTFT', 'benchmark.detail.avg.ttft': 'TTFT 均值',
'benchmark.detail.avg.tpot': '平均 TPOT', 'benchmark.detail.avg.tpot': 'TPOT 均值',
'benchmark.detail.avg.itl': '平均 ITL', 'benchmark.detail.avg.itl': 'ITL 均值',
'benchmark.detail.throughput.totalToken': '总 Token 吞吐量', 'benchmark.detail.throughput.totalToken': '总吞吐量',
'benchmark.detail.throughput.inputToken': '输入 Token 吞吐量', 'benchmark.detail.throughput.inputToken': '输入吞吐量',
'benchmark.detail.throughput.outputToken': '输出 Token 吞吐量', 'benchmark.detail.throughput.outputToken': '输出吞吐量',
'benchmark.detail.requests.total': '请求总数', 'benchmark.detail.requests.total': '请求总数',
'benchmark.detail.requests.success': '请求成功数', 'benchmark.detail.requests.success': '请求成功数',
'benchmark.detail.requests.failed': '请求失败数', 'benchmark.detail.requests.failed': '请求失败数',
+1 -1
View File
@@ -94,7 +94,7 @@ export default {
'common.data.unkonwn': '未知', 'common.data.unkonwn': '未知',
'common.data.none': '暂无', 'common.data.none': '暂无',
'common.action.none': '无', 'common.action.none': '无',
'common.checbox.all': '全选', 'common.checkbox.all': '全选',
'common.select.all': '全部{type}', 'common.select.all': '全部{type}',
'common.button.addfile': '添加文件', 'common.button.addfile': '添加文件',
'common.logout.msg': '登出成功', 'common.logout.msg': '登出成功',
+88 -22
View File
@@ -1,8 +1,12 @@
import OverlayScroller from '@/components/overlay-scroller'; import OverlayScroller from '@/components/overlay-scroller';
import {
readColumnSettings,
writeColumnSettings
} from '@/utils/localstore/index';
import { SettingOutlined } from '@ant-design/icons'; import { SettingOutlined } from '@ant-design/icons';
import { useIntl } from '@umijs/max'; import { useIntl } from '@umijs/max';
import { Button, Checkbox, Col, Popover, Row, Tooltip } from 'antd'; import { Button, Checkbox, Col, Popover, Row, Tooltip } from 'antd';
import React from 'react'; import React, { useEffect } from 'react';
import styled from 'styled-components'; import styled from 'styled-components';
const Container = styled.div` const Container = styled.div`
@@ -32,24 +36,53 @@ const Title = styled.div`
`; `;
const ColumnSettings: React.FC<{ const ColumnSettings: React.FC<{
width?: number;
fixedColumns?: string[];
tableName: string;
contentHeight: number; contentHeight: number;
columns: { columns: {
title: string; title: React.ReactNode;
dataIndex?: string; dataIndex?: string;
children?: { title: string; dataIndex?: string }[]; children?: { title: React.ReactNode; dataIndex?: string }[];
}[]; }[];
selectedColumns?: string[]; selectedColumns?: string[];
defaultSelectedColumns?: string[];
grouped?: boolean; grouped?: boolean;
onReset?: () => void;
onChange?: (selectedColumns: string[]) => void; onChange?: (selectedColumns: string[]) => void;
}> = (props) => { }> = (props) => {
const intl = useIntl(); const intl = useIntl();
const { contentHeight, columns, selectedColumns, grouped, onChange } = props; const {
tableName,
contentHeight,
width = 420,
columns,
selectedColumns,
defaultSelectedColumns,
grouped,
onReset,
onChange,
fixedColumns
} = props;
const [open, setOpen] = React.useState(false); const [open, setOpen] = React.useState(false);
const [innerSelectedColumns, setInnerSelectedColumns] = React.useState<
string[]
>(defaultSelectedColumns || []);
React.useEffect(() => {
if (open) {
setInnerSelectedColumns(selectedColumns ?? defaultSelectedColumns ?? []);
}
}, [open, selectedColumns, defaultSelectedColumns]);
const handleToggle = () => { const handleToggle = () => {
setOpen(!open); setOpen(!open);
}; };
const handleDraftChange = (columns: string[]) => {
setInnerSelectedColumns(columns);
};
const handleSelectAll = () => { const handleSelectAll = () => {
if (grouped) { if (grouped) {
const allCols: string[] = []; const allCols: string[] = [];
@@ -60,23 +93,55 @@ const ColumnSettings: React.FC<{
} }
}); });
}); });
onChange?.(allCols); handleDraftChange(allCols);
} else { } else {
const allCols = columns const allCols = columns
.map((col) => col.dataIndex) .map((col) => col.dataIndex)
.filter((dataIndex): dataIndex is string => Boolean(dataIndex)); .filter((dataIndex): dataIndex is string => Boolean(dataIndex));
onChange?.(allCols); handleDraftChange(allCols);
} }
}; };
const handleConfirm = () => { const handleConfirm = () => {
setOpen(false); setOpen(false);
writeColumnSettings(tableName, innerSelectedColumns);
onChange?.(innerSelectedColumns);
}; };
const handleReset = () => {
const resetCols = defaultSelectedColumns ?? [];
writeColumnSettings(tableName, resetCols);
onChange?.(resetCols);
setInnerSelectedColumns(resetCols);
onReset?.();
};
const handleOpenChange = (isOpen: boolean) => {
setOpen(isOpen);
};
useEffect(() => {
const initColumns = async () => {
const stored = await readColumnSettings(tableName);
console.log('stored columns:', stored);
if (stored && stored.length > 0) {
setInnerSelectedColumns(stored);
onChange?.(stored);
} else {
setInnerSelectedColumns(defaultSelectedColumns || []);
}
};
initColumns();
}, []);
const contentRender = () => { const contentRender = () => {
return ( return (
<Container> <Container>
{!grouped && <div className="title">Column Settings</div>} {!grouped && (
<div className="title">
{intl.formatMessage({ id: 'benchmark.table.columnSettings' })}
</div>
)}
<OverlayScroller <OverlayScroller
maxHeight={contentHeight} maxHeight={contentHeight}
styles={{ styles={{
@@ -86,10 +151,8 @@ const ColumnSettings: React.FC<{
}} }}
> >
<Checkbox.Group <Checkbox.Group
value={selectedColumns} value={innerSelectedColumns}
onChange={(checkedValues) => { onChange={handleDraftChange}
onChange?.(checkedValues as string[]);
}}
> >
<> <>
{grouped ? ( {grouped ? (
@@ -100,6 +163,9 @@ const ColumnSettings: React.FC<{
{row.children?.map((col) => ( {row.children?.map((col) => (
<Col key={col.dataIndex} span={12}> <Col key={col.dataIndex} span={12}>
<Checkbox <Checkbox
disabled={fixedColumns?.includes(
col.dataIndex || ''
)}
value={col.dataIndex} value={col.dataIndex}
style={{ marginBottom: 8 }} style={{ marginBottom: 8 }}
> >
@@ -115,6 +181,7 @@ const ColumnSettings: React.FC<{
{columns.map((col) => ( {columns.map((col) => (
<Col key={col.dataIndex} span={12}> <Col key={col.dataIndex} span={12}>
<Checkbox <Checkbox
disabled={fixedColumns?.includes(col.dataIndex || '')}
value={col.dataIndex} value={col.dataIndex}
style={{ marginBottom: 8 }} style={{ marginBottom: 8 }}
> >
@@ -129,20 +196,15 @@ const ColumnSettings: React.FC<{
</OverlayScroller> </OverlayScroller>
<div className="btn-wrapper"> <div className="btn-wrapper">
<Button <Button size="middle" onClick={handleReset}>
size="middle" {intl.formatMessage({ id: 'common.button.resetdefault' })}
onClick={() => {
onChange?.([]);
}}
>
Reset
</Button> </Button>
<div className="buttons"> <div className="buttons">
<Button size="middle" type="primary" onClick={handleSelectAll}> <Button size="middle" type="primary" onClick={handleSelectAll}>
Select All {intl.formatMessage({ id: 'common.checkbox.all' })}
</Button> </Button>
<Button size="middle" type="primary" onClick={handleConfirm}> <Button size="middle" type="primary" onClick={handleConfirm}>
Confirm {intl.formatMessage({ id: 'common.button.confirm' })}
</Button> </Button>
</div> </div>
</div> </div>
@@ -152,17 +214,21 @@ const ColumnSettings: React.FC<{
return ( return (
<Popover <Popover
open={open}
onOpenChange={handleOpenChange}
trigger={'click'} trigger={'click'}
arrow={false} arrow={false}
placement="bottomRight" placement="bottomRight"
content={contentRender()} content={contentRender()}
styles={{ styles={{
root: { root: {
width: '420px' width: width
} }
}} }}
> >
<Tooltip title="Column Settings"> <Tooltip
title={intl.formatMessage({ id: 'benchmark.table.columnSettings' })}
>
<Button onClick={handleToggle} icon={<SettingOutlined />}></Button> <Button onClick={handleToggle} icon={<SettingOutlined />}></Button>
</Tooltip> </Tooltip>
</Popover> </Popover>
@@ -68,7 +68,7 @@ const latencyColumns = [
const requestColumns = [ const requestColumns = [
{ {
label: 'benchmark.detail.requests.total', title: 'benchmark.detail.requests.total',
key: 'total_requests', key: 'total_requests',
dataIndex: 'total_requests', dataIndex: 'total_requests',
path: 'total_requests', path: 'total_requests',
@@ -77,7 +77,7 @@ const requestColumns = [
unit: '' unit: ''
}, },
{ {
label: 'benchmark.detail.requests.success', title: 'benchmark.detail.requests.success',
key: 'total_requests', key: 'total_requests',
dataIndex: 'successful_requests', dataIndex: 'successful_requests',
path: ['raw_metrics', 'benchmarks', '0'], path: ['raw_metrics', 'benchmarks', '0'],
@@ -88,7 +88,7 @@ const requestColumns = [
unit: '' unit: ''
}, },
{ {
label: 'benchmark.detail.requests.failed', title: 'benchmark.detail.requests.failed',
key: 'total_requests', key: 'total_requests',
dataIndex: 'failed_requests', dataIndex: 'failed_requests',
path: ['raw_metrics', 'benchmarks', '0'], path: ['raw_metrics', 'benchmarks', '0'],
@@ -99,7 +99,7 @@ const requestColumns = [
unit: '' unit: ''
}, },
{ {
label: 'benchmark.detail.requests.concurrency', title: 'benchmark.detail.requests.concurrency',
key: 'request_concurrency', key: 'request_concurrency',
dataIndex: 'request_concurrency', dataIndex: 'request_concurrency',
path: ['raw_metrics', 'benchmarks', '0'], path: ['raw_metrics', 'benchmarks', '0'],
@@ -180,9 +180,9 @@ const PercentileResult: React.FC = () => {
); );
const requestItems = requestColumns.map( const requestItems = requestColumns.map(
({ label, dataIndex, path, render, unit, color }) => ({ ({ title, dataIndex, path, render, unit, color }) => ({
key: dataIndex, key: dataIndex,
label: intl.formatMessage({ id: label }), label: intl.formatMessage({ id: title }),
children: unit ? ( children: unit ? (
<span className="flex-center" style={{ color: color }}> <span className="flex-center" style={{ color: color }}>
{render(_.get(detailData, path) ?? 0)}{' '} {render(_.get(detailData, path) ?? 0)}{' '}
@@ -1,22 +1,21 @@
// columns.ts // columns.ts
import AutoTooltip from '@/components/auto-tooltip'; import AutoTooltip from '@/components/auto-tooltip';
import StatusTag from '@/components/status-tag';
import { tableSorter } from '@/config/settings'; import { tableSorter } from '@/config/settings';
import { useIntl } from '@umijs/max'; import { useIntl } from '@umijs/max';
import { Progress, Typography } from 'antd'; import { Typography } from 'antd';
import { ColumnsType } from 'antd/es/table'; import { ColumnsType } from 'antd/es/table';
import _ from 'lodash';
import { useMemo } from 'react'; import { useMemo } from 'react';
import RowActions from '../components/row-actions'; import RowActions from '../components/row-actions';
import { BenchmarkStatus, BenchmarkStatusLabelMap } from '../config';
import { BenchmarkListItem as ListItem } from '../config/types'; import { BenchmarkListItem as ListItem } from '../config/types';
const useBenchmarkColumns = ( const useBenchmarkColumns = (params: {
sortOrder: string[], sortOrder: string[];
handleSelect: (val: string, record: ListItem) => void, columns: ColumnsType<ListItem>;
onCellClick?: (record: ListItem, dataIndex: string) => void handleSelect: (val: string, record: ListItem) => void;
): ColumnsType<ListItem> => { onCellClick?: (record: ListItem, dataIndex: string) => void;
}): ColumnsType<ListItem> => {
const intl = useIntl(); const intl = useIntl();
const { onCellClick, handleSelect, columns } = params;
return useMemo(() => { return useMemo(() => {
return [ return [
@@ -32,55 +31,10 @@ const useBenchmarkColumns = (
</AutoTooltip> </AutoTooltip>
) )
}, },
{ ...columns,
title: intl.formatMessage({ id: 'benchmark.table.model' }),
dataIndex: 'model_name',
sorter: tableSorter(1),
render: (text: string) => (
<AutoTooltip ghost minWidth={20}>
{text}
</AutoTooltip>
)
},
{
title: intl.formatMessage({ id: 'benchmark.table.dataset' }),
dataIndex: 'dataset_name',
sorter: tableSorter(1),
render: (text: string) => (
<AutoTooltip ghost minWidth={20}>
{text}
</AutoTooltip>
)
},
{
title: intl.formatMessage({ id: 'common.table.status' }),
dataIndex: 'state',
ellipsis: {
showTitle: false
},
width: 120,
render: (value: number, record: ListItem) => (
<span className="flex-center gap-8">
<StatusTag
statusValue={{
status: BenchmarkStatus[value],
text: BenchmarkStatusLabelMap[value],
message: record.state_message || undefined
}}
/>
{record.progress !== undefined && record.progress < 100 && (
<Progress
type="circle"
percent={_.round(record.progress, 0)}
size={20}
/>
)}
</span>
)
},
// { // {
// title: intl.formatMessage({ id: 'benchmark.table.requestRate' }), // title: intl.formatMessage({ id: 'benchmark.table.model' }),
// dataIndex: 'request_rate', // dataIndex: 'model_name',
// sorter: tableSorter(1), // sorter: tableSorter(1),
// render: (text: string) => ( // render: (text: string) => (
// <AutoTooltip ghost minWidth={20}> // <AutoTooltip ghost minWidth={20}>
@@ -88,88 +42,93 @@ const useBenchmarkColumns = (
// </AutoTooltip> // </AutoTooltip>
// ) // )
// }, // },
{
title: intl.formatMessage({ id: 'benchmark.table.gpu' }),
dataIndex: 'gpu_summary',
sorter: tableSorter(1),
render: (text: string) => (
<AutoTooltip ghost minWidth={20}>
{text}
</AutoTooltip>
)
},
{
title: (
<span className="flex-column">
<span>{intl.formatMessage({ id: 'benchmark.table.itl' })}</span>
<span>Avg (ms)</span>
</span>
),
dataIndex: 'inter_token_latency_mean',
sorter: tableSorter(1),
render: (text: string) => (
<AutoTooltip ghost minWidth={20}>
{_.round(text, 1)}
</AutoTooltip>
)
},
{
title: (
<span className="flex-column">
<span>{intl.formatMessage({ id: 'benchmark.table.tpot' })}</span>
<span>Avg (ms)</span>
</span>
),
dataIndex: 'time_per_output_token_mean',
sorter: tableSorter(1),
render: (text: string) => (
<AutoTooltip ghost minWidth={20}>
{_.round(text, 2)}
</AutoTooltip>
)
},
{
title: (
<span className="flex-column">
<span>{intl.formatMessage({ id: 'benchmark.table.ttft' })}</span>
<span>Avg (ms)</span>
</span>
),
dataIndex: 'time_to_first_token_mean',
sorter: tableSorter(1),
render: (text: string) => (
<AutoTooltip ghost minWidth={20}>
{_.round(text, 2)}
</AutoTooltip>
)
},
{
title: intl.formatMessage({ id: 'benchmark.table.rps' }),
dataIndex: 'requests_per_second_mean',
sorter: tableSorter(1),
render: (text: string) => (
<AutoTooltip ghost minWidth={20}>
{_.round(text, 0)}
</AutoTooltip>
)
},
{
title: intl.formatMessage({ id: 'benchmark.table.tps' }),
dataIndex: 'tokens_per_second_mean',
sorter: tableSorter(1),
render: (text: string) => (
<AutoTooltip ghost minWidth={20}>
{_.round(text, 2)}
</AutoTooltip>
)
},
// { // {
// title: intl.formatMessage({ id: 'common.table.createTime' }), // title: intl.formatMessage({ id: 'benchmark.table.dataset' }),
// dataIndex: 'created_at', // dataIndex: 'dataset_name',
// sorter: tableSorter(3), // sorter: tableSorter(1),
// render: (value: string) => ( // render: (text: string) => (
// <AutoTooltip ghost> // <AutoTooltip ghost minWidth={20}>
// {dayjs(value).format('YYYY-MM-DD HH:mm:ss')} // {text}
// </AutoTooltip>
// )
// },
// {
// title: intl.formatMessage({ id: 'common.table.status' }),
// dataIndex: 'state',
// ellipsis: {
// showTitle: false
// },
// width: 120,
// render: (value: number, record: ListItem) => (
// <span className="flex-center gap-8">
// <StatusTag
// statusValue={{
// status: BenchmarkStatus[value],
// text: BenchmarkStatusLabelMap[value],
// message: record.state_message || undefined
// }}
// />
// {record.progress !== undefined && record.progress < 100 && (
// <Progress
// type="circle"
// percent={_.round(record.progress, 0)}
// size={20}
// />
// )}
// </span>
// )
// },
// {
// title: intl.formatMessage({ id: 'benchmark.table.gpu' }),
// dataIndex: 'gpu_summary',
// sorter: tableSorter(1),
// render: (text: string) => (
// <AutoTooltip ghost minWidth={20}>
// {text}
// </AutoTooltip>
// )
// },
// {
// title: 'TPS',
// dataIndex: 'tokens_per_second_mean',
// sorter: tableSorter(1),
// render: (text: string) => (
// <AutoTooltip ghost minWidth={20}>
// {_.round(text, 2)}
// </AutoTooltip>
// )
// },
// {
// title: (
// <span className="flex-column">
// <span>{intl.formatMessage({ id: 'benchmark.table.ttft' })}</span>
// <span>
// {intl.formatMessage({ id: 'benchmark.table.avg' })} (ms)
// </span>
// </span>
// ),
// dataIndex: 'time_to_first_token_mean',
// sorter: tableSorter(1),
// render: (text: string) => (
// <AutoTooltip ghost minWidth={20}>
// {_.round(text, 2)}
// </AutoTooltip>
// )
// },
// {
// title: (
// <span className="flex-column">
// <span>{intl.formatMessage({ id: 'benchmark.table.tpot' })}</span>
// <span>
// {intl.formatMessage({ id: 'benchmark.table.avg' })} (ms)
// </span>
// </span>
// ),
// dataIndex: 'time_per_output_token_mean',
// sorter: tableSorter(1),
// render: (text: string) => (
// <AutoTooltip ghost minWidth={20}>
// {_.round(text, 2)}
// </AutoTooltip> // </AutoTooltip>
// ) // )
// }, // },
@@ -184,7 +143,7 @@ const useBenchmarkColumns = (
) )
} }
]; ];
}, [intl, onCellClick, handleSelect]); }, [intl, onCellClick, handleSelect, columns]);
}; };
export default useBenchmarkColumns; export default useBenchmarkColumns;
+364 -84
View File
@@ -1,112 +1,392 @@
import AutoTooltip from '@/components/auto-tooltip';
import StatusTag from '@/components/status-tag';
import { tableSorter } from '@/config/settings';
import ColumnSettings from '@/pages/_components/column-settings'; import ColumnSettings from '@/pages/_components/column-settings';
import { useIntl } from '@umijs/max'; import { useIntl } from '@umijs/max';
import { Progress } from 'antd';
import dayjs from 'dayjs';
import _, { round } from 'lodash';
import React from 'react'; import React from 'react';
import styled from 'styled-components';
import { BenchmarkStatus, BenchmarkStatusLabelMap } from '../config';
import { BenchmarkListItem as ListItem } from '../config/types';
const SubTitleWrapper = styled.span`
display: list-item;
overflow: hidden;
text-overflow: ellipsis;
white-space: nowrap;
`;
const defaultColumns: string[] = [
'model_name',
'dataset_name',
'state',
'gpu_summary',
'tokens_per_second_mean',
'time_to_first_token_mean',
'time_per_output_token_mean'
];
const fixedColumns: string[] = [];
const useColumnSettings = (options: { contentHeight: number }) => { const useColumnSettings = (options: { contentHeight: number }) => {
const intl = useIntl(); const intl = useIntl();
const { contentHeight } = options; const { contentHeight } = options;
const [selectedColumns, setSelectedColumns] = React.useState<string[]>([]); const [selectedColumns, setSelectedColumns] =
React.useState<string[]>(defaultColumns);
const renderTitle = (
title: React.ReactNode,
options?: { subTitle?: React.ReactNode }
): React.ReactNode => {
return (
<span>
<AutoTooltip
ghost
minWidth={20}
title={`${title} ${options?.subTitle || ''}`}
>
{title}
</AutoTooltip>
{options?.subTitle && (
<SubTitleWrapper>{options.subTitle}</SubTitleWrapper>
)}
</span>
);
};
const resultColumns = [
{
title: renderTitle(
`${intl.formatMessage({ id: 'benchmark.detail.summary.latency' })}`,
{
subTitle: `${intl.formatMessage({ id: 'benchmark.table.avg' })} (ms)`
}
),
pos: 11,
dataIndex: 'request_latency_mean',
path: 'request_latency_mean',
unit: 'ms',
sorter: tableSorter(1),
render: (value: number) => round(value, 2)
},
{
title: 'TPS',
pos: 12,
dataIndex: 'tokens_per_second_mean',
path: 'tokens_per_second_mean',
unit: 'Tokens/s',
sorter: tableSorter(1),
render: (text: number) => (
<AutoTooltip ghost minWidth={20}>
{_.round(text, 2)}
</AutoTooltip>
)
},
{
title: renderTitle('TTFT', {
subTitle: `${intl.formatMessage({ id: 'benchmark.table.avg' })} (ms)`
}),
pos: 13,
sorter: tableSorter(1),
dataIndex: 'time_to_first_token_mean',
path: 'time_to_first_token_mean',
unit: 'ms',
render: (text: number) => (
<AutoTooltip ghost minWidth={20}>
{_.round(text, 2)}
</AutoTooltip>
)
},
{
title: renderTitle('TPOT', {
subTitle: `${intl.formatMessage({ id: 'benchmark.table.avg' })} (ms)`
}),
pos: 14,
sorter: tableSorter(1),
dataIndex: 'time_per_output_token_mean',
path: 'time_per_output_token_mean',
unit: 'ms',
render: (text: number) => (
<AutoTooltip ghost minWidth={20}>
{_.round(text, 2)}
</AutoTooltip>
)
},
{
title: renderTitle('ITL', {
subTitle: `${intl.formatMessage({ id: 'benchmark.table.avg' })} (ms)`
}),
pos: 15,
sorter: tableSorter(1),
dataIndex: 'inter_token_latency_mean',
path: 'inter_token_latency_mean',
unit: 'ms',
render: (value: number) => round(value, 2)
},
{
title: 'RPS',
pos: 16,
dataIndex: 'requests_per_second_mean',
sorter: tableSorter(1),
render: (text: string) => (
<AutoTooltip ghost minWidth={20}>
{_.round(text, 0)}
</AutoTooltip>
)
},
{
title: renderTitle(
`${intl.formatMessage({
id: 'benchmark.detail.throughput.inputToken'
})}`,
{
subTitle: '(Tokens/s)'
}
),
pos: 17,
dataIndex: 'input_tokens_per_second_mean',
path: 'input_tokens_per_second_mean',
unit: 'Tokens/s',
render: (value: number) => (
<AutoTooltip ghost minWidth={20}>
{round(value, 2)}
</AutoTooltip>
)
},
{
title: renderTitle(
`${intl.formatMessage({
id: 'benchmark.detail.throughput.outputToken'
})}`,
{
subTitle: '(Tokens/s)'
}
),
pos: 18,
dataIndex: 'output_tokens_per_second_mean',
path: 'output_tokens_per_second_mean',
unit: 'Tokens/s',
render: (value: number) => (
<AutoTooltip ghost minWidth={20}>
{round(value, 2)}
</AutoTooltip>
)
},
{
title: renderTitle(
intl.formatMessage({ id: 'benchmark.detail.requests.total' })
),
pos: 19,
key: 'total_requests',
dataIndex: 'total_requests',
path: 'total_requests',
precision: 0,
render: (value: number) => (
<AutoTooltip ghost minWidth={20}>
{round(value, 0) || 0}
</AutoTooltip>
),
unit: ''
},
{
title: renderTitle(
intl.formatMessage({ id: 'benchmark.detail.requests.success' })
),
pos: 20,
key: 'total_requests',
dataIndex: 'successful_requests',
path: ['raw_metrics', 'benchmarks', '0'],
render: (value: number) =>
round(_.get(value, ['metrics', 'request_totals', 'successful']), 0) ||
0,
precision: 0,
color: 'var(--ant-color-success)',
unit: ''
},
{
title: renderTitle(
intl.formatMessage({ id: 'benchmark.detail.requests.failed' })
),
pos: 21,
key: 'total_requests',
dataIndex: 'failed_requests',
path: ['raw_metrics', 'benchmarks', '0'],
render: (value: number) =>
round(_.get(value, ['metrics', 'request_totals', 'errored']), 0) || 0,
precision: 0,
color: 'var(--ant-color-error)',
unit: ''
},
{
title: renderTitle(
intl.formatMessage({
id: 'benchmark.detail.requests.concurrency'
})
),
pos: 22,
key: 'request_concurrency',
dataIndex: 'request_concurrency',
path: ['raw_metrics', 'benchmarks', '0'],
render: (value: number) =>
round(_.get(value, 'metrics.request_concurrency.successful.mean'), 0) ||
0,
precision: 0,
unit: ''
}
];
const metadataColumns = [
{
title: renderTitle(intl.formatMessage({ id: 'clusters.title' })),
pos: 1,
dataIndex: 'cluster_id',
render: (text: string) => (
<AutoTooltip ghost minWidth={20}>
{text}
</AutoTooltip>
)
},
{
title: renderTitle(intl.formatMessage({ id: 'resources.worker' })),
pos: 2,
dataIndex: 'worker_id',
render: (text: string) => (
<AutoTooltip ghost minWidth={20}>
{text}
</AutoTooltip>
)
},
{
title: renderTitle(
intl.formatMessage({ id: 'benchmark.detail.modelName' })
),
pos: 3,
dataIndex: 'model_name',
render: (text: string) => (
<AutoTooltip ghost minWidth={20}>
{text}
</AutoTooltip>
)
},
{
title: renderTitle(intl.formatMessage({ id: 'benchmark.table.dataset' })),
pos: 4,
dataIndex: 'dataset_name',
render: (text: string) => (
<AutoTooltip ghost minWidth={20}>
{text}
</AutoTooltip>
)
},
{
title: renderTitle(intl.formatMessage({ id: 'benchmark.form.profile' })),
pos: 5,
dataIndex: 'profile',
render: (text: string) => (
<AutoTooltip ghost minWidth={20}>
{text}
</AutoTooltip>
)
},
{
title: renderTitle(intl.formatMessage({ id: 'benchmark.table.gpu' })),
pos: 6,
dataIndex: 'gpu_summary',
render: (text: string) => (
<AutoTooltip ghost minWidth={20}>
{text}
</AutoTooltip>
)
},
{
title: renderTitle(intl.formatMessage({ id: 'common.table.status' })),
pos: 7,
width: 120,
dataIndex: 'state',
render: (value: number, record: ListItem) => (
<span className="flex-center gap-8">
<StatusTag
statusValue={{
status: BenchmarkStatus[value],
text: BenchmarkStatusLabelMap[value],
message: record.state_message || undefined
}}
/>
{record.progress !== undefined && record.progress < 100 && (
<Progress
type="circle"
percent={_.round(record.progress, 0)}
size={20}
/>
)}
</span>
)
},
{
title: renderTitle(
intl.formatMessage({ id: 'benchmark.table.requestRate' })
),
pos: 8,
dataIndex: 'request_rate',
render: (text: string) => (
<AutoTooltip ghost minWidth={20}>
{text}
</AutoTooltip>
)
},
{
title: renderTitle(intl.formatMessage({ id: 'common.table.createTime' })),
pos: 23,
dataIndex: 'created_at',
sorter: tableSorter(6),
render: (value: string) => (
<AutoTooltip ghost minWidth={20}>
{dayjs(value).format('YYYY-MM-DD HH:mm:ss')}
</AutoTooltip>
)
}
];
const handleOnChange = (columns: string[]) => { const handleOnChange = (columns: string[]) => {
setSelectedColumns(columns); setSelectedColumns(columns);
console.log('selected columns:', columns); console.log('selected columns:', columns);
}; };
const resultColumns = [ const handleOnReset = () => {
{ setSelectedColumns(defaultColumns);
title: intl.formatMessage({ id: 'benchmark.table.rps' }), };
dataIndex: 'requests_per_second_mean'
},
{
title: 'Throughput',
dataIndex: 'throughput_mean'
},
{
title: 'Throughput request',
dataIndex: 'throughput_request_mean'
},
{
title: 'Generated Tokens',
dataIndex: 'generated_tokens_mean'
},
{
title: intl.formatMessage({ id: 'benchmark.table.tps' }),
dataIndex: 'tokens_per_second_mean'
},
{
title: intl.formatMessage({ id: 'benchmark.table.itl' }),
dataIndex: 'inter_token_latency_mean'
},
{
title: intl.formatMessage({ id: 'benchmark.table.tpot' }),
dataIndex: 'time_per_output_token_mean'
},
{
title: intl.formatMessage({ id: 'benchmark.table.ttft' }),
dataIndex: 'time_to_first_token_mean'
},
{
title: 'Latency Avg',
dataIndex: 'latency_mean'
}
];
const metadataColumns = [ const columns = React.useMemo(() => {
{ const allColumns = [...metadataColumns, ...resultColumns];
title: intl.formatMessage({ id: 'clusters.title' }), const selected = allColumns.filter((col) =>
dataIndex: 'cluster_id' selectedColumns.includes(col.dataIndex as string)
}, );
{ // Sort by pos
title: intl.formatMessage({ id: 'resources.worker' }), selected.sort((a, b) => (a.pos || 0) - (b.pos || 0));
dataIndex: 'worker_id' console.log('columns===', selected);
}, return selected;
{ }, [selectedColumns]);
title: intl.formatMessage({ id: 'common.table.name' }),
dataIndex: 'name'
},
{
title: intl.formatMessage({ id: 'benchmark.table.model' }),
dataIndex: 'model_name'
},
{
title: intl.formatMessage({ id: 'benchmark.table.dataset' }),
dataIndex: 'dataset_name'
},
{
title: intl.formatMessage({ id: 'common.table.status' }),
dataIndex: 'state'
},
{
title: intl.formatMessage({ id: 'benchmark.table.requestRate' }),
dataIndex: 'request_rate'
},
{
title: intl.formatMessage({ id: 'benchmark.table.gpu' }),
dataIndex: 'gpu_summary'
},
{
title: intl.formatMessage({ id: 'common.table.createTime' }),
dataIndex: 'created_at'
},
{
title: intl.formatMessage({ id: 'common.table.operation' }),
dataIndex: 'operations'
}
];
const SettingsButton = ( const SettingsButton = (
<ColumnSettings <ColumnSettings
tableName="benchmark"
contentHeight={contentHeight} contentHeight={contentHeight}
defaultSelectedColumns={selectedColumns}
selectedColumns={selectedColumns} selectedColumns={selectedColumns}
onChange={handleOnChange} onChange={handleOnChange}
onReset={handleOnReset}
fixedColumns={fixedColumns}
grouped={true} grouped={true}
columns={[ columns={[
{ {
title: 'Benchmark Results', title: intl.formatMessage({ id: 'benchmark.detail.summary.results' }),
children: resultColumns children: resultColumns
}, },
{ {
title: 'Metadata', title: intl.formatMessage({
id: 'benchmark.detail.summary.metadata'
}),
children: metadataColumns children: metadataColumns
} }
]} ]}
@@ -115,7 +395,7 @@ const useColumnSettings = (options: { contentHeight: number }) => {
return { return {
SettingsButton, SettingsButton,
selectedColumns columns
}; };
}; };
+5 -4
View File
@@ -63,7 +63,7 @@ const Benchmark: React.FC = () => {
}); });
const { openViewLogsModal, closeViewLogsModal, openViewLogsModalStatus } = const { openViewLogsModal, closeViewLogsModal, openViewLogsModalStatus } =
useViewLogs(); useViewLogs();
const { SettingsButton, selectedColumns } = useColumnSettings({ const { SettingsButton, columns: selectedColumns } = useColumnSettings({
contentHeight: 320 contentHeight: 320
}); });
const { handleStopBenchmark } = useStopBenchmark(); const { handleStopBenchmark } = useStopBenchmark();
@@ -162,11 +162,12 @@ const Benchmark: React.FC = () => {
); );
}; };
const columns = useBenchmarkColumns( const columns = useBenchmarkColumns({
sortOrder, sortOrder,
columns: selectedColumns,
handleSelect, handleSelect,
handleOnCellClick onCellClick: handleOnCellClick
); });
const handleExportData = () => { const handleExportData = () => {
exportData(rowSelection.selectedRowKeys); exportData(rowSelection.selectedRowKeys);
-2
View File
@@ -91,8 +91,6 @@ const Login = () => {
const [userInfo, setUserInfo] = useAtom(userAtom); const [userInfo, setUserInfo] = useAtom(userAtom);
const { initialState, setInitialState } = useModel('@@initialState') || {}; const { initialState, setInitialState } = useModel('@@initialState') || {};
console.log('useSettings==========', userSettings);
const gotoDefaultPage = async (userInfo: any) => { const gotoDefaultPage = async (userInfo: any) => {
if (!userInfo || userInfo?.require_password_change) { if (!userInfo || userInfo?.require_password_change) {
return; return;
+10
View File
@@ -37,14 +37,24 @@ const writeState = (key: string, data: any) => {
} }
}; };
const readColumnSettings = (key: string) => {
return readState(key);
};
const writeColumnSettings = (key: string, data: any) => {
writeState(key, data);
};
export { export {
CRYPT_TEXT, CRYPT_TEXT,
IS_FIRST_LOGIN, IS_FIRST_LOGIN,
REMEMBER_ME_KEY, REMEMBER_ME_KEY,
getRememberMe, getRememberMe,
readColumnSettings,
readState, readState,
rememberMe, rememberMe,
removeRememberMe, removeRememberMe,
writeColumnSettings,
writeState writeState
}; };