first-update

easy-dataset-main/hooks/useTaskSettings.js

@@ -0,0 +1,63 @@
+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
+  };
+}