feat: 完善Dify知识库管理召回测试模块，优化知识库上传文件时的分段配置设置

2025-12-03 12:03:09 +08:00
parent 0f49426a2e
commit a13f3b3635
29 changed files with 4016 additions and 1880 deletions
@@ -0,0 +1,92 @@
+import { useState, useEffect, useCallback } from 'react';
+import { message } from 'antd';
+import type { FormInstance } from 'antd';
+import type { Dataset } from '~/api/dify-dataset/type/datasetTypes';
+import { updateDatasetName } from '~/api/dify-dataset/api/datasetApi';
+
+/**
+ * 知识库设置状态管理 Hook
+ */
+export function useDatasetSettings(
+    dataset: Dataset | null,
+    form: FormInstance,
+    onDatasetUpdated: (dataset: Dataset) => void
+) {
+    const [saving, setSaving] = useState(false);
+    const [hasChanges, setHasChanges] = useState(false);
+
+    // 初始化表单数据
+    useEffect(() => {
+        if (dataset) {
+            form.setFieldsValue({
+                name: dataset.name,
+                description: dataset.description || '',
+            });
+            setHasChanges(false);
+        }
+    }, [dataset, form]);
+
+    /**
+     * 处理表单值变化
+     */
+    const handleValuesChange = useCallback(() => {
+        const values = form.getFieldsValue();
+        const changed =
+            values.name !== dataset?.name ||
+            values.description !== (dataset?.description || '');
+        setHasChanges(changed);
+    }, [form, dataset]);
+
+    /**
+     * 保存设置
+     */
+    const handleSave = useCallback(async () => {
+        if (!dataset) {
+            message.error('知识库不存在');
+            return;
+        }
+
+        try {
+            const values = await form.validateFields();
+            setSaving(true);
+
+            // 目前只支持修改名称
+            const updatedDataset = await updateDatasetName(dataset.id, values.name);
+
+            message.success('保存成功');
+            onDatasetUpdated(updatedDataset);
+            setHasChanges(false);
+        } catch (err: any) {
+            console.error('保存设置失败:', err);
+            message.error(err.message || '保存失败');
+        } finally {
+            setSaving(false);
+        }
+    }, [dataset, form, onDatasetUpdated]);
+
+    /**
+     * 重置表单
+     */
+    const handleReset = useCallback(() => {
+        if (dataset) {
+            form.setFieldsValue({
+                name: dataset.name,
+                description: dataset.description || '',
+            });
+            setHasChanges(false);
+        }
+    }, [dataset, form]);
+
+    return {
+        // 状态
+        saving,
+        hasChanges,
+        
+        // 方法
+        handleValuesChange,
+        handleSave,
+        handleReset,
+    };
+}
+
+export type UseDatasetSettingsReturn = ReturnType<typeof useDatasetSettings>;