Files
YG-Datasets/easy-dataset-main/hooks/useTaskSettings.js

64 lines
1.7 KiB
JavaScript

import { useState, useEffect } from 'react';
import { useTranslation } from 'react-i18next';
import { DEFAULT_SETTINGS } from '@/constant/setting';
export default function useTaskSettings(projectId) {
const { t } = useTranslation();
const [taskSettings, setTaskSettings] = useState({
...DEFAULT_SETTINGS
});
const [loading, setLoading] = useState(true);
const [error, setError] = useState(null);
const [success, setSuccess] = useState(false);
useEffect(() => {
async function fetchTaskSettings() {
try {
setLoading(true);
const response = await fetch(`/api/projects/${projectId}/tasks`);
if (!response.ok) {
throw new Error(t('settings.fetchTasksFailed'));
}
const data = await response.json();
// 如果没有配置,使用默认值
if (Object.keys(data).length === 0) {
setTaskSettings({
...DEFAULT_SETTINGS
});
} else {
// 确保所有默认值都被正确设置,特别是数字类型的字段
const mergedSettings = {
...DEFAULT_SETTINGS,
...data
};
// 确保 multiTurnRounds 是数字类型
if (mergedSettings.multiTurnRounds !== undefined) {
mergedSettings.multiTurnRounds = Number(mergedSettings.multiTurnRounds);
}
setTaskSettings(mergedSettings);
}
} catch (error) {
console.error('获取任务配置出错:', error);
setError(error.message);
} finally {
setLoading(false);
}
}
fetchTaskSettings();
}, [projectId, t]);
return {
taskSettings,
setTaskSettings,
loading,
error,
success,
setSuccess
};
}