RateLimitSetting.js 1.8 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667
  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.endsWith('Enabled')) {
  24. newInputs[item.key] = item.value === 'true' ? true : false;
  25. } else {
  26. newInputs[item.key] = item.value;
  27. }
  28. });
  29. setInputs(newInputs);
  30. } else {
  31. showError(message);
  32. }
  33. };
  34. async function onRefresh() {
  35. try {
  36. setLoading(true);
  37. await getOptions();
  38. // showSuccess('刷新成功');
  39. } catch (error) {
  40. showError('刷新失败');
  41. } finally {
  42. setLoading(false);
  43. }
  44. }
  45. useEffect(() => {
  46. onRefresh();
  47. }, []);
  48. return (
  49. <>
  50. <Spin spinning={loading} size='large'>
  51. {/* AI请求速率限制 */}
  52. <Card style={{ marginTop: '10px' }}>
  53. <RequestRateLimit options={inputs} refresh={onRefresh} />
  54. </Card>
  55. </Spin>
  56. </>
  57. );
  58. };
  59. export default RateLimitSetting;