feat: 分组速率前端优化

This commit is contained in:
tbphp
2025-05-05 22:06:16 +08:00
parent 88ed83f419
commit 1cb4d750e4
2 changed files with 45 additions and 54 deletions

View File

@@ -13,7 +13,7 @@ const RateLimitSetting = () => {
ModelRequestRateLimitCount: 0, ModelRequestRateLimitCount: 0,
ModelRequestRateLimitSuccessCount: 1000, ModelRequestRateLimitSuccessCount: 1000,
ModelRequestRateLimitDurationMinutes: 1, ModelRequestRateLimitDurationMinutes: 1,
ModelRequestRateLimitGroup: '{}', ModelRequestRateLimitGroup: '',
}); });
let [loading, setLoading] = useState(false); let [loading, setLoading] = useState(false);
@@ -24,12 +24,14 @@ const RateLimitSetting = () => {
if (success) { if (success) {
let newInputs = {}; let newInputs = {};
data.forEach((item) => { data.forEach((item) => {
if (Object.prototype.hasOwnProperty.call(inputs, item.key)) { if (item.key === 'ModelRequestRateLimitGroup') {
if (item.key.endsWith('Enabled')) { item.value = JSON.stringify(JSON.parse(item.value), null, 2);
newInputs[item.key] = item.value === 'true'; }
} else {
newInputs[item.key] = item.value; if (item.key.endsWith('Enabled')) {
} newInputs[item.key] = item.value === 'true' ? true : false;
} else {
newInputs[item.key] = item.value;
} }
}); });

View File

@@ -6,6 +6,7 @@ import {
showError, showError,
showSuccess, showSuccess,
showWarning, showWarning,
verifyJSON,
} from '../../../helpers'; } from '../../../helpers';
import { useTranslation } from 'react-i18next'; import { useTranslation } from 'react-i18next';
@@ -18,7 +19,7 @@ export default function RequestRateLimit(props) {
ModelRequestRateLimitCount: -1, ModelRequestRateLimitCount: -1,
ModelRequestRateLimitSuccessCount: 1000, ModelRequestRateLimitSuccessCount: 1000,
ModelRequestRateLimitDurationMinutes: 1, ModelRequestRateLimitDurationMinutes: 1,
ModelRequestRateLimitGroup: '{}', ModelRequestRateLimitGroup: '',
}); });
const refForm = useRef(); const refForm = useRef();
const [inputsRow, setInputsRow] = useState(inputs); const [inputsRow, setInputsRow] = useState(inputs);
@@ -33,43 +34,32 @@ export default function RequestRateLimit(props) {
} else { } else {
value = inputs[item.key]; value = inputs[item.key];
} }
if (item.key === 'ModelRequestRateLimitGroup') {
try {
JSON.parse(value);
} catch (e) {
showError(t('用户组速率限制配置不是有效的 JSON 格式!'));
return Promise.reject('Invalid JSON format');
}
}
return API.put('/api/option/', { return API.put('/api/option/', {
key: item.key, key: item.key,
value, value,
}); });
}); });
const validRequests = requestQueue.filter(req => req !== null && req !== undefined && typeof req.then === 'function');
if (validRequests.length === 0 && requestQueue.length > 0) {
return;
}
setLoading(true); setLoading(true);
Promise.all(validRequests) Promise.all(requestQueue)
.then((res) => { .then((res) => {
if (validRequests.length === 1) { if (requestQueue.length === 1) {
if (res.includes(undefined)) return; if (res.includes(undefined)) return;
} else if (validRequests.length > 1) { } else if (requestQueue.length > 1) {
if (res.includes(undefined)) if (res.includes(undefined))
return showError(t('部分保存失败,请重试')); return showError(t('部分保存失败,请重试'));
} }
for (let i = 0; i < res.length; i++) {
if (!res[i].data.success) {
return showError(res[i].data.message);
}
}
showSuccess(t('保存成功')); showSuccess(t('保存成功'));
props.refresh(); props.refresh();
setInputsRow(structuredClone(inputs));
}) })
.catch((error) => { .catch(() => {
if (error !== 'Invalid JSON format') { showError(t('保存失败,请重试'));
showError(t('保存失败,请重试'));
}
}) })
.finally(() => { .finally(() => {
setLoading(false); setLoading(false);
@@ -79,15 +69,13 @@ export default function RequestRateLimit(props) {
useEffect(() => { useEffect(() => {
const currentInputs = {}; const currentInputs = {};
for (let key in props.options) { for (let key in props.options) {
if (Object.prototype.hasOwnProperty.call(inputs, key)) { // 使用 hasOwnProperty 检查 if (Object.keys(inputs).includes(key)) {
currentInputs[key] = props.options[key]; currentInputs[key] = props.options[key];
} }
} }
setInputs(currentInputs); setInputs(currentInputs);
setInputsRow(structuredClone(currentInputs)); setInputsRow(structuredClone(currentInputs));
if (refForm.current) {
refForm.current.setValues(currentInputs); refForm.current.setValues(currentInputs);
}
}, [props.options]); }, [props.options]);
return ( return (
@@ -168,40 +156,41 @@ export default function RequestRateLimit(props) {
/> />
</Col> </Col>
</Row> </Row>
<Row style={{ marginTop: 15 }}> <Row>
<Col span={24}> <Col xs={24} sm={16}>
<Form.TextArea <Form.TextArea
label={t('用户组速率限制 (JSON)')} label={t('组速率限制')}
field={'ModelRequestRateLimitGroup'}
placeholder={t( placeholder={t(
'请输入 JSON 格式的用户组限制,例如:\n{\n "default": [200, 100],\n "vip": [1000, 500]\n}', '{\n "default": [200, 100],\n "vip": [0, 1000]\n}',
)} )}
field={'ModelRequestRateLimitGroup'}
autosize={{ minRows: 5, maxRows: 15 }}
trigger='blur'
stopValidateWithError
rules={[
{
validator: (rule, value) => verifyJSON(value),
message: t('不是合法的 JSON 字符串'),
},
]}
extraText={ extraText={
<div> <div>
<p>{t('说明:')}</p> <p style={{ marginBottom: -15 }}>{t('说明')}</p>
<ul> <ul>
<li>{t('使用 JSON 对象格式,键为用户组名 (字符串),值为包含两个整数的数组 [总次数限制, 成功次数限制]。')}</li> <li>{t('使用 JSON 对象格式,格式为:{"组名": [最多请求次数, 最多请求完成次数]}')}</li>
<li>{t('总次数限制: 周期内允许的总请求次数 (含失败)0 代表不限制。')}</li> <li>{t('示例:{"default": [200, 100], "vip": [0, 1000]}。')}</li>
<li>{t('成功次数限制: 周期内允许的成功请求次数 (HTTP < 400),必须大于 0。')}</li> <li>{t('分组速率配置优先级高于全局速率限制。')}</li>
<li>{t('此配置将优先于上方的全局限制设置。')}</li>
<li>{t('未在此处配置的用户组将使用全局限制。')}</li>
<li>{t('限制周期统一使用上方配置的“限制周期”值。')}</li> <li>{t('限制周期统一使用上方配置的“限制周期”值。')}</li>
<li>{t('输入无效的 JSON 将无法保存。')}</li>
</ul> </ul>
</div> </div>
} }
autosize={{ minRows: 5, maxRows: 15 }}
style={{ fontFamily: 'monospace' }}
onChange={(value) => { onChange={(value) => {
setInputs({ setInputs({ ...inputs, ModelRequestRateLimitGroup: value });
...inputs,
ModelRequestRateLimitGroup: value,
});
}} }}
/> />
</Col> </Col>
</Row> </Row>
<Row style={{ marginTop: 15 }}> <Row>
<Button size='default' onClick={onSubmit}> <Button size='default' onClick={onSubmit}>
{t('保存模型速率限制')} {t('保存模型速率限制')}
</Button> </Button>
@@ -211,4 +200,4 @@ export default function RequestRateLimit(props) {
</Spin> </Spin>
</> </>
); );
} }