PingChuan
124 lines
4.1 KiB
TypeScript
124 lines
4.1 KiB
TypeScript
import { message } from 'antd';
|
|
import { useCallback, useEffect, useState } from 'react';
|
|
import { updateDocumentWithSettings } from '~/api/dify-dataset/api/documentApi';
|
|
import { fetchSegments } from '~/api/dify-dataset/api/segmentApi';
|
|
import type { Segment } from '~/api/dify-dataset/type';
|
|
import type { Document } from '~/api/dify-dataset/type/documentTypes';
|
|
import type { DocumentDetailSegmentationSettings } from '~/types/dify-dataset-manager/document-detail';
|
|
import { DEFAULT_DOCUMENT_DETAIL_SETTINGS } from '~/types/dify-dataset-manager/document-detail';
|
|
|
|
/**
|
|
* 文档详情状态管理 Hook
|
|
*/
|
|
export function useDocumentDetail(datasetId: string, document: Document | null) {
|
|
// 分段设置状态
|
|
const [settings, setSettings] = useState<DocumentDetailSegmentationSettings>(DEFAULT_DOCUMENT_DETAIL_SETTINGS);
|
|
|
|
// 预览状态
|
|
const [previewSegments, setPreviewSegments] = useState<Segment[]>([]);
|
|
const [previewLoading, setPreviewLoading] = useState(false);
|
|
const [showPreview, setShowPreview] = useState(false);
|
|
|
|
// 保存状态
|
|
const [saving, setSaving] = useState(false);
|
|
|
|
// 当文档变化时重置设置
|
|
useEffect(() => {
|
|
if (document) {
|
|
// 可以从文档中读取已有的设置，这里使用默认值
|
|
setSettings(DEFAULT_DOCUMENT_DETAIL_SETTINGS);
|
|
setPreviewSegments([]);
|
|
setShowPreview(false);
|
|
}
|
|
}, [document?.id]);
|
|
|
|
/**
|
|
* 更新设置
|
|
*/
|
|
const updateSettings = useCallback((key: keyof DocumentDetailSegmentationSettings, value: any) => {
|
|
setSettings(prev => ({ ...prev, [key]: value }));
|
|
}, []);
|
|
|
|
/**
|
|
* 重置设置
|
|
*/
|
|
const handleReset = useCallback(() => {
|
|
setSettings(DEFAULT_DOCUMENT_DETAIL_SETTINGS);
|
|
setPreviewSegments([]);
|
|
setShowPreview(false);
|
|
}, []);
|
|
|
|
/**
|
|
* 预览分段
|
|
*/
|
|
const handlePreview = useCallback(async () => {
|
|
if (!document) return;
|
|
|
|
setPreviewLoading(true);
|
|
setShowPreview(true);
|
|
try {
|
|
// 获取当前文档的分段作为预览
|
|
const response = await fetchSegments(datasetId, document.id, 1, 50);
|
|
setPreviewSegments(response.data || []);
|
|
if (response.data?.length === 0) {
|
|
message.info('该文档暂无分段数据');
|
|
}
|
|
} catch (err: any) {
|
|
console.error('预览分段失败:', err);
|
|
message.error(err.message || '预览失败');
|
|
} finally {
|
|
setPreviewLoading(false);
|
|
}
|
|
}, [datasetId, document]);
|
|
|
|
/**
|
|
* 保存并处理
|
|
*/
|
|
const handleSaveAndProcess = useCallback(async () => {
|
|
if (!document) return;
|
|
|
|
setSaving(true);
|
|
try {
|
|
await updateDocumentWithSettings(datasetId, document.id, {
|
|
indexing_technique: 'high_quality',
|
|
process_rule: {
|
|
mode: 'custom',
|
|
rules: {
|
|
pre_processing_rules: [
|
|
{ id: 'remove_extra_spaces', enabled: settings.removeExtraSpaces },
|
|
{ id: 'remove_urls_emails', enabled: settings.removeUrlsEmails },
|
|
],
|
|
segmentation: {
|
|
separator: settings.separator.replace(/\\n/g, '\n'),
|
|
max_tokens: settings.maxTokens,
|
|
},
|
|
},
|
|
},
|
|
});
|
|
message.success('设置已保存，文档正在重新处理...');
|
|
} catch (err: any) {
|
|
console.error('保存设置失败:', err);
|
|
message.error(err.message || '保存失败');
|
|
} finally {
|
|
setSaving(false);
|
|
}
|
|
}, [datasetId, document, settings]);
|
|
|
|
return {
|
|
// 状态
|
|
settings,
|
|
previewSegments,
|
|
previewLoading,
|
|
showPreview,
|
|
saving,
|
|
|
|
// 方法
|
|
updateSettings,
|
|
handleReset,
|
|
handlePreview,
|
|
handleSaveAndProcess,
|
|
};
|
|
}
|
|
|
|
export type UseDocumentDetailReturn = ReturnType<typeof useDocumentDetail>;
|