merge
This commit is contained in:
hzywhite
parent
482a09d397
commit
e07d4bb70b
8 changed files with 39 additions and 11 deletions
|
|
@ -256,6 +256,7 @@ export type Scheme = {
|
|||
config: {
|
||||
framework: 'lightrag' | 'raganything';
|
||||
extractor?: 'mineru' | 'docling' | undefined; // Optional extractor field
|
||||
modelSource?: 'huggingface' | 'modelscope' | 'local' | undefined; // Optional model source field
|
||||
};
|
||||
};
|
||||
|
||||
|
|
@ -379,9 +380,9 @@ export const getDocuments = async (): Promise<DocsStatusesResponse> => {
|
|||
return response.data
|
||||
}
|
||||
|
||||
export const scanNewDocuments = async (framework?: string): Promise<ScanResponse> => {
|
||||
export const scanNewDocuments = async (schemeConfig: any): Promise<ScanResponse> => {
|
||||
const response = await axiosInstance.post('/documents/scan', {
|
||||
framework
|
||||
schemeConfig
|
||||
})
|
||||
return response.data
|
||||
}
|
||||
|
|
|
|||
|
|
@ -24,6 +24,7 @@ import { useTranslation } from 'react-i18next';
|
|||
interface SchemeConfig {
|
||||
framework: 'lightrag' | 'raganything';
|
||||
extractor?: 'mineru' | 'docling';
|
||||
modelSource?: 'huggingface' | 'modelscope' | 'local';
|
||||
}
|
||||
|
||||
const SchemeManagerDialog = () => {
|
||||
|
|
@ -95,7 +96,7 @@ const SchemeManagerDialog = () => {
|
|||
try {
|
||||
const newScheme = await addScheme({
|
||||
name: trimmedName,
|
||||
config: { framework: 'lightrag', extractor: undefined },
|
||||
config: { framework: 'lightrag', extractor: undefined, modelSource: undefined },
|
||||
});
|
||||
|
||||
// 更新方案列表
|
||||
|
|
@ -136,6 +137,7 @@ const SchemeManagerDialog = () => {
|
|||
...updates,
|
||||
framework: updates.framework ?? selectedScheme.config?.framework ?? 'lightrag',
|
||||
extractor: updates.extractor || selectedScheme.config?.extractor || (updates.framework === 'raganything' ? 'mineru' : undefined),
|
||||
modelSource: updates.modelSource || selectedScheme.config?.modelSource || (updates.extractor === 'mineru' ? 'huggingface' : undefined),
|
||||
},
|
||||
};
|
||||
|
||||
|
|
@ -265,6 +267,21 @@ const SchemeManagerDialog = () => {
|
|||
</select>
|
||||
</div>
|
||||
)}
|
||||
|
||||
{selectedScheme.config?.extractor === "mineru" && (
|
||||
<div>
|
||||
<label className="block text-sm mb-1">{t('schemeManager.modelSource')}</label>
|
||||
<select
|
||||
value={selectedScheme.config?.modelSource || "huggingface"}
|
||||
onChange={(e) => handleConfigChange({ modelSource: e.target.value as 'huggingface' | 'modelscope' | 'local' })}
|
||||
className="w-full px-3 py-1.5 border rounded-md focus:outline-none"
|
||||
>
|
||||
<option value="huggingface">huggingface</option>
|
||||
<option value="modelscope">modelscope</option>
|
||||
<option value="local">local</option>
|
||||
</select>
|
||||
</div>
|
||||
)}
|
||||
</div>
|
||||
) : (
|
||||
<div className="flex flex-col items-center justify-center h-[70%] text-gray-500">
|
||||
|
|
|
|||
|
|
@ -636,9 +636,9 @@ export default function DocumentManager() {
|
|||
return;
|
||||
}
|
||||
|
||||
const framework = selectedScheme.config?.framework;
|
||||
const schemeConfig = selectedScheme.config
|
||||
|
||||
const { status, message, track_id: _track_id } = await scanNewDocuments(framework); // eslint-disable-line @typescript-eslint/no-unused-vars
|
||||
const { status, message, track_id: _track_id } = await scanNewDocuments(schemeConfig); // eslint-disable-line @typescript-eslint/no-unused-vars
|
||||
|
||||
// Check again if component is still mounted after the request completes
|
||||
if (!isMountedRef.current) return;
|
||||
|
|
@ -664,7 +664,7 @@ export default function DocumentManager() {
|
|||
} catch (err) {
|
||||
// Only show error if component is still mounted
|
||||
if (isMountedRef.current) {
|
||||
toast.error(t('documentPanel.documentManager.errors.scanFailed', { error: errorMessage(err) }));
|
||||
toast.error(t('documentPanel.documentManager.errors.scanFiled', { error: errorMessage(err) }));
|
||||
}
|
||||
}
|
||||
}, [t, startPollingInterval, currentTab, health, statusCounts, selectedScheme])
|
||||
|
|
|
|||
|
|
@ -147,7 +147,8 @@
|
|||
"errors": {
|
||||
"loadFailed": "فشل تحميل المستندات\n{{error}}",
|
||||
"scanFailed": "فشل مسح المستندات\n{{error}}",
|
||||
"scanProgressFailed": "فشل الحصول على تقدم المسح\n{{error}}"
|
||||
"scanProgressFailed": "فشل الحصول على تقدم المسح\n{{error}}",
|
||||
"missingSchemeId": "الحل هو في عداد المفقودين ، حدد الحل"
|
||||
},
|
||||
"fileNameLabel": "اسم الملف",
|
||||
"showButton": "عرض",
|
||||
|
|
@ -429,6 +430,7 @@
|
|||
"selectSchemePrompt": "يرجى تحديد أو إنشاء مخطط أولاً",
|
||||
"processingFramework": "إطار المعالجة",
|
||||
"extractionTool": "أداة الاستخراج",
|
||||
"modelSource": "مصدر النموذج",
|
||||
"errors": {
|
||||
"loadFailed": "فشل تحميل المخططات",
|
||||
"nameEmpty": "لا يمكن أن يكون اسم المخطط فارغًا",
|
||||
|
|
|
|||
|
|
@ -147,7 +147,8 @@
|
|||
"errors": {
|
||||
"loadFailed": "Failed to load documents\n{{error}}",
|
||||
"scanFailed": "Failed to scan documents\n{{error}}",
|
||||
"scanProgressFailed": "Failed to get scan progress\n{{error}}"
|
||||
"scanProgressFailed": "Failed to get scan progress\n{{error}}",
|
||||
"missingSchemeId": "Lack of solution, please select a solution"
|
||||
},
|
||||
"fileNameLabel": "File Name",
|
||||
"showButton": "Show",
|
||||
|
|
@ -429,6 +430,7 @@
|
|||
"selectSchemePrompt": "Please select or create a scheme first",
|
||||
"processingFramework": "Processing Framework",
|
||||
"extractionTool": "Extraction Tool",
|
||||
"modelSource": "Model Source",
|
||||
"errors": {
|
||||
"loadFailed": "Failed to load schemes",
|
||||
"nameEmpty": "Scheme name cannot be empty",
|
||||
|
|
|
|||
|
|
@ -147,7 +147,8 @@
|
|||
"errors": {
|
||||
"loadFailed": "Échec du chargement des documents\n{{error}}",
|
||||
"scanFailed": "Échec de la numérisation des documents\n{{error}}",
|
||||
"scanProgressFailed": "Échec de l'obtention de la progression de la numérisation\n{{error}}"
|
||||
"scanProgressFailed": "Échec de l'obtention de la progression de la numérisation\n{{error}}",
|
||||
"missingSchemeId": "Schéma de traitement manquant, veuillez sélectionner un schéma de traitement"
|
||||
},
|
||||
"fileNameLabel": "Nom du fichier",
|
||||
"showButton": "Afficher",
|
||||
|
|
@ -429,6 +430,7 @@
|
|||
"selectSchemePrompt": "Veuillez d'abord sélectionner ou créer un schéma",
|
||||
"processingFramework": "Framework de traitement",
|
||||
"extractionTool": "Outil d'extraction",
|
||||
"modelSource": "Source du modèle",
|
||||
"errors": {
|
||||
"loadFailed": "Échec du chargement des schémas",
|
||||
"nameEmpty": "Le nom du schéma ne peut pas être vide",
|
||||
|
|
|
|||
|
|
@ -147,7 +147,8 @@
|
|||
"errors": {
|
||||
"loadFailed": "加载文档失败\n{{error}}",
|
||||
"scanFailed": "扫描文档失败\n{{error}}",
|
||||
"scanProgressFailed": "获取扫描进度失败\n{{error}}"
|
||||
"scanProgressFailed": "获取扫描进度失败\n{{error}}",
|
||||
"missingSchemeId": "缺少处理方案，请选择处理方案"
|
||||
},
|
||||
"fileNameLabel": "文件名",
|
||||
"showButton": "显示",
|
||||
|
|
@ -429,6 +430,7 @@
|
|||
"selectSchemePrompt": "请先选择或创建一个方案",
|
||||
"processingFramework": "处理框架",
|
||||
"extractionTool": "提取工具",
|
||||
"modelSource": "模型源",
|
||||
"errors": {
|
||||
"loadFailed": "加载方案失败",
|
||||
"nameEmpty": "方案名称不能为空",
|
||||
|
|
|
|||
|
|
@ -147,7 +147,8 @@
|
|||
"errors": {
|
||||
"loadFailed": "載入文件失敗\n{{error}}",
|
||||
"scanFailed": "掃描文件失敗\n{{error}}",
|
||||
"scanProgressFailed": "取得掃描進度失敗\n{{error}}"
|
||||
"scanProgressFailed": "取得掃描進度失敗\n{{error}}",
|
||||
"missingSchemeId": "缺少處理方案，請選擇處理方案"
|
||||
},
|
||||
"fileNameLabel": "檔案名稱",
|
||||
"showButton": "顯示",
|
||||
|
|
@ -429,6 +430,7 @@
|
|||
"selectSchemePrompt": "請先選擇或建立一個方案",
|
||||
"processingFramework": "處理框架",
|
||||
"extractionTool": "提取工具",
|
||||
"modelSource": "模型源",
|
||||
"errors": {
|
||||
"loadFailed": "載入方案失敗",
|
||||
"nameEmpty": "方案名稱不能為空",
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue