RateLimitSetting.js 1.9 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071
  1. import React, { useEffect, useState } from 'react';
  2. import { Card, Spin, Tabs } from '@douyinfe/semi-ui';
  3. import { API, showError, showSuccess } from '../helpers';
  4. import SettingsChats from '../pages/Setting/Operation/SettingsChats.js';
  5. import { useTranslation } from 'react-i18next';
  6. import RequestRateLimit from '../pages/Setting/RateLimit/SettingsRequestRateLimit.js';
  7. const RateLimitSetting = () => {
  8. const { t } = useTranslation();
  9. let [inputs, setInputs] = useState({
  10. ModelRequestRateLimitEnabled: false,
  11. ModelRequestRateLimitCount: 0,
  12. ModelRequestRateLimitSuccessCount: 1000,
  13. ModelRequestRateLimitDurationMinutes: 1,
  14. ModelRequestRateLimitGroup: '',
  15. });
  16. let [loading, setLoading] = useState(false);
  17. const getOptions = async () => {
  18. const res = await API.get('/api/option/');
  19. const { success, message, data } = res.data;
  20. if (success) {
  21. let newInputs = {};
  22. data.forEach((item) => {
  23. if (item.key === 'ModelRequestRateLimitGroup') {
  24. item.value = JSON.stringify(JSON.parse(item.value), null, 2);
  25. }
  26. if (item.key.endsWith('Enabled')) {
  27. newInputs[item.key] = item.value === 'true' ? true : false;
  28. } else {
  29. newInputs[item.key] = item.value;
  30. }
  31. });
  32. setInputs(newInputs);
  33. } else {
  34. showError(message);
  35. }
  36. };
  37. async function onRefresh() {
  38. try {
  39. setLoading(true);
  40. await getOptions();
  41. // showSuccess('刷新成功');
  42. } catch (error) {
  43. showError('刷新失败');
  44. } finally {
  45. setLoading(false);
  46. }
  47. }
  48. useEffect(() => {
  49. onRefresh();
  50. }, []);
  51. return (
  52. <>
  53. <Spin spinning={loading} size='large'>
  54. {/* AI请求速率限制 */}
  55. <Card style={{ marginTop: '10px' }}>
  56. <RequestRateLimit options={inputs} refresh={onRefresh} />
  57. </Card>
  58. </Spin>
  59. </>
  60. );
  61. };
  62. export default RateLimitSetting;