1a8888211f
- Rename files to follow camelCase naming convention: • auth-header.js → authUtils.js • other.js → logUtils.js • rehypeSplitWordsIntoSpans.js → textAnimationUtils.js - Update import paths in affected components: • Update exports in helpers/index.js • Fix import in LogsTable.js for logUtils • Fix import in MarkdownRenderer.js for textAnimationUtils - Remove old files after successful migration - Improve file naming clarity: • authUtils.js better describes authentication utilities • logUtils.js clearly indicates log processing functions • textAnimationUtils.js concisely describes text animation functionality This refactoring enhances code maintainability and follows consistent naming patterns throughout the helpers directory.
70 lines
2.0 KiB
JavaScript
70 lines
2.0 KiB
JavaScript
import { useCallback, useEffect } from 'react';
|
|
import { useTranslation } from 'react-i18next';
|
|
import { API } from '../helpers/api';
|
|
import { API_ENDPOINTS } from '../constants/playground.constants';
|
|
import { processModelsData, processGroupsData } from '../helpers/apiUtils';
|
|
|
|
export const useDataLoader = (
|
|
userState,
|
|
inputs,
|
|
handleInputChange,
|
|
setModels,
|
|
setGroups
|
|
) => {
|
|
const { t } = useTranslation();
|
|
|
|
const loadModels = useCallback(async () => {
|
|
try {
|
|
const res = await API.get(API_ENDPOINTS.USER_MODELS);
|
|
const { success, message, data } = res.data;
|
|
|
|
if (success) {
|
|
const { modelOptions, selectedModel } = processModelsData(data, inputs.model);
|
|
setModels(modelOptions);
|
|
|
|
if (selectedModel !== inputs.model) {
|
|
handleInputChange('model', selectedModel);
|
|
}
|
|
} else {
|
|
showError(t(message));
|
|
}
|
|
} catch (error) {
|
|
showError(t('加载模型失败'));
|
|
}
|
|
}, [inputs.model, handleInputChange, setModels, t]);
|
|
|
|
const loadGroups = useCallback(async () => {
|
|
try {
|
|
const res = await API.get(API_ENDPOINTS.USER_GROUPS);
|
|
const { success, message, data } = res.data;
|
|
|
|
if (success) {
|
|
const userGroup = userState?.user?.group || JSON.parse(localStorage.getItem('user'))?.group;
|
|
const groupOptions = processGroupsData(data, userGroup);
|
|
setGroups(groupOptions);
|
|
|
|
const hasCurrentGroup = groupOptions.some(option => option.value === inputs.group);
|
|
if (!hasCurrentGroup) {
|
|
handleInputChange('group', groupOptions[0]?.value || '');
|
|
}
|
|
} else {
|
|
showError(t(message));
|
|
}
|
|
} catch (error) {
|
|
showError(t('加载分组失败'));
|
|
}
|
|
}, [userState, inputs.group, handleInputChange, setGroups, t]);
|
|
|
|
// 自动加载数据
|
|
useEffect(() => {
|
|
if (userState?.user) {
|
|
loadModels();
|
|
loadGroups();
|
|
}
|
|
}, [userState?.user, loadModels, loadGroups]);
|
|
|
|
return {
|
|
loadModels,
|
|
loadGroups
|
|
};
|
|
};
|