import { useCallback, useEffect, useMemo, useState } from 'react' import { useNavigate } from '@tanstack/react-router' import { ChevronLeft, ChevronRight, Database, Gauge, Loader2, RefreshCw, RotateCcw, Save, SlidersHorizontal, Sparkles, Trash2, Upload, } from 'lucide-react' import { CodeEditor } from '@/components' import { MemoryDeleteDialog } from '@/components/memory/MemoryDeleteDialog' import { MemoryConfigEditor } from '@/components/memory/MemoryConfigEditor' import { Alert, AlertDescription } from '@/components/ui/alert' import { Badge } from '@/components/ui/badge' import { Button } from '@/components/ui/button' import { Card, CardContent, CardDescription, CardHeader, CardTitle } from '@/components/ui/card' import { Checkbox } from '@/components/ui/checkbox' import { Dialog, DialogContent, DialogDescription, DialogFooter, DialogHeader, DialogTitle, } from '@/components/ui/dialog' import { Input } from '@/components/ui/input' import { Label } from '@/components/ui/label' import { Progress } from '@/components/ui/progress' import { ScrollArea } from '@/components/ui/scroll-area' import { Select, SelectContent, SelectItem, SelectTrigger, SelectValue, } from '@/components/ui/select' import { Table, TableBody, TableCell, TableHead, TableHeader, TableRow } from '@/components/ui/table' import { Tabs, TabsContent, TabsList, TabsTrigger } from '@/components/ui/tabs' import { Textarea } from '@/components/ui/textarea' import { useToast } from '@/hooks/use-toast' import { cn } from '@/lib/utils' import { cancelMemoryImportTask, createMemoryLpmmConvertImport, createMemoryLpmmOpenieImport, createMemoryMaibotMigrationImport, createMemoryRawScanImport, createMemoryTemporalBackfillImport, executeMemoryDelete, getMemoryFeedbackCorrection, getMemoryFeedbackCorrections, getMemoryImportPathAliases, getMemoryImportSettings, getMemoryImportTask, getMemoryImportTaskChunks, applyBestMemoryTuningProfile, createMemoryPasteImport, createMemoryTuningTask, createMemoryUploadImport, getMemoryConfig, getMemoryConfigRaw, getMemoryConfigSchema, getMemoryDeleteOperation, getMemoryDeleteOperations, getMemoryImportTasks, getMemoryRuntimeConfig, getMemorySources, getMemoryTuningProfile, getMemoryTuningTasks, type MemoryDeleteRequestPayload, type MemoryImportChunkListPayload, type MemoryImportInputMode, type MemoryImportSettings, type MemoryImportTaskKind, type MemoryImportTaskPayload, previewMemoryDelete, refreshMemoryRuntimeSelfCheck, rollbackMemoryFeedbackCorrection, resolveMemoryImportPath, retryMemoryImportTask, restoreMemoryDelete, updateMemoryConfig, updateMemoryConfigRaw, type MemoryConfigSchemaPayload, type MemoryDeleteExecutePayload, type MemoryDeleteOperationPayload, type MemoryFeedbackActionLogPayload, type MemoryFeedbackCorrectionDetailTaskPayload, type MemoryFeedbackCorrectionSummaryPayload, type MemorySourceItemPayload, type MemoryRuntimeConfigPayload, type MemoryTaskPayload, } from '@/lib/memory-api' const DELETE_OPERATION_FETCH_LIMIT = 100 const DELETE_OPERATION_PAGE_SIZE = 6 const DELETE_OPERATION_ITEM_PAGE_SIZE = 8 const FEEDBACK_CORRECTION_FETCH_LIMIT = 100 const FEEDBACK_CORRECTION_PAGE_SIZE = 6 const FEEDBACK_ACTION_LOG_PAGE_SIZE = 8 const IMPORT_CHUNK_PAGE_SIZE = 50 const RUNNING_IMPORT_STATUS = new Set(['preparing', 'running', 'cancel_requested']) const QUEUED_IMPORT_STATUS = new Set(['queued']) const IMPORT_STATUS_TEXT: Record = { queued: '排队中', preparing: '准备中', running: '运行中', cancel_requested: '取消中', cancelled: '已取消', completed: '已完成', completed_with_errors: '完成(有错误)', failed: '失败', } const IMPORT_STEP_TEXT: Record = { queued: '排队中', preparing: '准备中', running: '运行中', splitting: '分块中', extracting: '抽取中', writing: '写入中', saving: '保存中', backfilling: '回填中', converting: '转换中', verifying: '校验中', switching: '切换中', cancel_requested: '取消中', cancelled: '已取消', completed: '已完成', completed_with_errors: '完成(有错误)', failed: '失败', } const IMPORT_KIND_OPTIONS: Array<{ value: MemoryImportTaskKind; label: string; description: string }> = [ { value: 'upload', label: '上传文件', description: '从本地批量上传文本文件' }, { value: 'paste', label: '粘贴导入', description: '直接粘贴文本或 JSON 内容创建任务' }, { value: 'raw_scan', label: '本地扫描', description: '按路径别名和匹配规则批量扫描导入' }, { value: 'lpmm_openie', label: 'LPMM OpenIE', description: '读取 LPMM 数据并抽取关系' }, { value: 'lpmm_convert', label: 'LPMM 转换', description: '将 LPMM 数据转换到目标目录' }, { value: 'temporal_backfill', label: '时序回填', description: '对既有数据执行时间字段回填' }, { value: 'maibot_migration', label: 'MaiBot 迁移', description: '从 MaiBot 历史库迁移长期记忆数据' }, ] function normalizeProgress(value: number | string | null | undefined): number { const numeric = Number(value ?? 0) if (!Number.isFinite(numeric)) { return 0 } if (numeric < 0) { return 0 } if (numeric > 100) { return 100 } return numeric } function parseOptionalPositiveInt(input: string): number | undefined { const value = input.trim() if (!value) { return undefined } const parsed = Number(value) if (!Number.isInteger(parsed) || parsed <= 0) { return undefined } return parsed } function parseCommaSeparatedList(input: string): string[] { return input .split(',') .map((item) => item.trim()) .filter(Boolean) } function normalizeImportInputMode(value: string): MemoryImportInputMode { return value === 'json' ? 'json' : 'text' } function getImportStatusLabel(status: string): string { const normalized = String(status ?? '').trim() if (!normalized) { return '-' } return IMPORT_STATUS_TEXT[normalized] ?? normalized } function getImportStepLabel(step: string): string { const normalized = String(step ?? '').trim() if (!normalized) { return '-' } return IMPORT_STEP_TEXT[normalized] ?? normalized } function getImportStatusVariant(status: string): 'default' | 'secondary' | 'destructive' | 'outline' { if (status === 'failed') { return 'destructive' } if (status === 'completed') { return 'default' } if (status === 'completed_with_errors' || status === 'cancelled') { return 'secondary' } if (RUNNING_IMPORT_STATUS.has(status) || QUEUED_IMPORT_STATUS.has(status)) { return 'outline' } return 'secondary' } function formatImportTime(timestamp?: number | null): string { if (!timestamp) { return '-' } const normalized = timestamp > 1_000_000_000_000 ? timestamp : timestamp * 1000 const value = new Date(normalized) if (Number.isNaN(value.getTime())) { return '-' } return value.toLocaleString('zh-CN', { hour12: false, year: 'numeric', month: '2-digit', day: '2-digit', hour: '2-digit', minute: '2-digit', }) } function formatDeleteOperationMode(mode: string): string { switch (mode) { case 'entity': return '实体' case 'relation': return '关系' case 'paragraph': return '段落' case 'source': return '来源' case 'mixed': return '混合' default: return mode || '未知' } } function formatDeleteOperationStatus(status: string): string { switch (status) { case 'executed': return '已执行' case 'restored': return '已恢复' default: return status || '未知' } } function formatDeleteOperationTime(timestamp?: number | null): string { if (!timestamp) { return '未知时间' } const normalized = timestamp > 1_000_000_000_000 ? timestamp : timestamp * 1000 const value = new Date(normalized) if (Number.isNaN(value.getTime())) { return '未知时间' } return value.toLocaleString('zh-CN', { hour12: false, year: 'numeric', month: '2-digit', day: '2-digit', hour: '2-digit', minute: '2-digit', }) } function formatFeedbackDecision(decision: string): string { switch (decision) { case 'correct': return '纠正' case 'reject': return '否定' case 'confirm': return '确认' case 'supplement': return '补充' case 'none': return '无动作' default: return decision || '未知' } } function formatFeedbackTaskStatus(status: string): string { switch (status) { case 'pending': return '待处理' case 'running': return '处理中' case 'applied': return '已应用' case 'skipped': return '已跳过' case 'error': return '失败' default: return status || '未知' } } function formatFeedbackRollbackStatus(status: string): string { switch (status) { case 'none': return '未回退' case 'running': return '回退中' case 'rolled_back': return '已回退' case 'error': return '回退失败' default: return status || '未知' } } function getFeedbackStatusVariant( status: string, ): 'default' | 'secondary' | 'destructive' | 'outline' { if (status === 'applied' || status === 'rolled_back') { return 'default' } if (status === 'error') { return 'destructive' } if (status === 'running' || status === 'pending') { return 'outline' } return 'secondary' } function summarizeFeedbackActionPayload(value: Record | undefined): string { if (!value) { return '' } const hash = String(value.hash ?? '').trim() const subject = String(value.subject ?? '').trim() const predicate = String(value.predicate ?? '').trim() const object = String(value.object ?? '').trim() if (subject && predicate && object) { return formatDeleteRelationText(subject, predicate, object) } if (hash) { return hash } if (Array.isArray(value.target_hashes) && value.target_hashes.length > 0) { return `targets ${value.target_hashes.length}` } return trimDeleteItemText(JSON.stringify(value, null, 2), 120) } type DeleteOperationItem = NonNullable[number] function trimDeleteItemText(value: string, maxLength: number = 140): string { const normalized = String(value ?? '').trim().replace(/\s+/g, ' ') if (!normalized) { return '' } if (normalized.length <= maxLength) { return normalized } return `${normalized.slice(0, maxLength)}...` } function formatDeleteRelationText(subject: string, predicate: string, object: string): string { const left = String(subject ?? '').trim() const middle = String(predicate ?? '').trim() const right = String(object ?? '').trim() return [left, middle, right].filter(Boolean).join(' -> ') } function getDeleteOperationItemLabel(item: DeleteOperationItem): string { const payload = item.payload ?? {} if (item.item_type === 'entity') { const entity = (payload.entity ?? {}) as Record return String(entity.name ?? item.item_key ?? item.item_hash ?? '未命名实体') } if (item.item_type === 'relation') { const relation = (payload.relation ?? {}) as Record return ( formatDeleteRelationText( String(relation.subject ?? ''), String(relation.predicate ?? ''), String(relation.object ?? ''), ) || String(item.item_key ?? item.item_hash ?? '未命名关系') ) } if (item.item_type === 'paragraph') { const paragraph = (payload.paragraph ?? {}) as Record const source = String(paragraph.source ?? '').trim() return source || String(item.item_key ?? item.item_hash ?? '未命名段落') } return String(item.item_key ?? item.item_hash ?? '未命名对象') } function getDeleteOperationItemPreview(item: DeleteOperationItem): string { const payload = item.payload ?? {} if (item.item_type === 'entity') { const paragraphLinks = Array.isArray(payload.paragraph_links) ? payload.paragraph_links : [] if (paragraphLinks.length > 0) { return `关联段落 ${paragraphLinks.length} 个` } return '实体快照' } if (item.item_type === 'relation') { const relation = (payload.relation ?? {}) as Record const paragraphHashes = Array.isArray(payload.paragraph_hashes) ? payload.paragraph_hashes : [] const confidence = relation.confidence const parts = [] if (paragraphHashes.length > 0) { parts.push(`证据段落 ${paragraphHashes.length} 个`) } if (typeof confidence === 'number') { parts.push(`置信度 ${confidence.toFixed(2)}`) } return parts.join(',') || '关系快照' } if (item.item_type === 'paragraph') { const paragraph = (payload.paragraph ?? {}) as Record return trimDeleteItemText(String(paragraph.content ?? '')) } return '' } function getDeleteOperationItemSource(item: DeleteOperationItem): string { const payload = item.payload ?? {} if (item.item_type === 'paragraph') { const paragraph = (payload.paragraph ?? {}) as Record return String(paragraph.source ?? '').trim() } return String(payload.source ?? '').trim() } export function KnowledgeBasePage() { const navigate = useNavigate() const { toast } = useToast() const [loading, setLoading] = useState(true) const [saving, setSaving] = useState(false) const [refreshingCheck, setRefreshingCheck] = useState(false) const [creatingImport, setCreatingImport] = useState(false) const [creatingTuning, setCreatingTuning] = useState(false) const [rawMode, setRawMode] = useState(false) const [schemaPayload, setSchemaPayload] = useState(null) const [visualConfig, setVisualConfig] = useState>({}) const [rawConfig, setRawConfig] = useState('') const [rawConfigExists, setRawConfigExists] = useState(true) const [rawConfigUsingDefault, setRawConfigUsingDefault] = useState(false) const [runtimeConfig, setRuntimeConfig] = useState(null) const [selfCheckReport, setSelfCheckReport] = useState | null>(null) const [importSettings, setImportSettings] = useState({}) const [importPathAliases, setImportPathAliases] = useState>({}) const [importTasks, setImportTasks] = useState([]) const [selectedImportTaskId, setSelectedImportTaskId] = useState('') const [selectedImportTask, setSelectedImportTask] = useState(null) const [selectedImportTaskLoading, setSelectedImportTaskLoading] = useState(false) const [selectedImportFileId, setSelectedImportFileId] = useState('') const [importChunkOffset, setImportChunkOffset] = useState(0) const [importChunksPayload, setImportChunksPayload] = useState(null) const [importChunksLoading, setImportChunksLoading] = useState(false) const [importCreateMode, setImportCreateMode] = useState('upload') const [importAutoPolling, setImportAutoPolling] = useState(true) const [importErrorText, setImportErrorText] = useState('') const [importCommonFileConcurrency, setImportCommonFileConcurrency] = useState('2') const [importCommonChunkConcurrency, setImportCommonChunkConcurrency] = useState('4') const [importCommonLlmEnabled, setImportCommonLlmEnabled] = useState(true) const [importCommonStrategyOverride, setImportCommonStrategyOverride] = useState('auto') const [importCommonDedupePolicy, setImportCommonDedupePolicy] = useState('content_hash') const [importCommonChatLog, setImportCommonChatLog] = useState(false) const [importCommonChatReferenceTime, setImportCommonChatReferenceTime] = useState('') const [importCommonForce, setImportCommonForce] = useState(false) const [importCommonClearManifest, setImportCommonClearManifest] = useState(false) const [uploadInputMode, setUploadInputMode] = useState('text') const [uploadFiles, setUploadFiles] = useState([]) const [pasteName, setPasteName] = useState('') const [pasteMode, setPasteMode] = useState('text') const [pasteContent, setPasteContent] = useState('') const [rawAlias, setRawAlias] = useState('raw') const [rawRelativePath, setRawRelativePath] = useState('') const [rawGlob, setRawGlob] = useState('*') const [rawInputMode, setRawInputMode] = useState('text') const [rawRecursive, setRawRecursive] = useState(true) const [openieAlias, setOpenieAlias] = useState('lpmm') const [openieRelativePath, setOpenieRelativePath] = useState('') const [openieIncludeAllJson, setOpenieIncludeAllJson] = useState(false) const [convertAlias, setConvertAlias] = useState('lpmm') const [convertRelativePath, setConvertRelativePath] = useState('') const [convertTargetAlias, setConvertTargetAlias] = useState('plugin_data') const [convertTargetRelativePath, setConvertTargetRelativePath] = useState('') const [convertDimension, setConvertDimension] = useState('') const [convertBatchSize, setConvertBatchSize] = useState('1024') const [backfillAlias, setBackfillAlias] = useState('plugin_data') const [backfillRelativePath, setBackfillRelativePath] = useState('') const [backfillLimit, setBackfillLimit] = useState('100000') const [backfillDryRun, setBackfillDryRun] = useState(false) const [backfillNoCreatedFallback, setBackfillNoCreatedFallback] = useState(false) const [maibotSourceDb, setMaibotSourceDb] = useState('') const [maibotTimeFrom, setMaibotTimeFrom] = useState('') const [maibotTimeTo, setMaibotTimeTo] = useState('') const [maibotStartId, setMaibotStartId] = useState('') const [maibotEndId, setMaibotEndId] = useState('') const [maibotStreamIds, setMaibotStreamIds] = useState('') const [maibotGroupIds, setMaibotGroupIds] = useState('') const [maibotUserIds, setMaibotUserIds] = useState('') const [maibotReadBatchSize, setMaibotReadBatchSize] = useState('2000') const [maibotCommitWindowRows, setMaibotCommitWindowRows] = useState('20000') const [maibotEmbedWorkers, setMaibotEmbedWorkers] = useState('') const [maibotNoResume, setMaibotNoResume] = useState(false) const [maibotResetState, setMaibotResetState] = useState(false) const [maibotDryRun, setMaibotDryRun] = useState(false) const [maibotVerifyOnly, setMaibotVerifyOnly] = useState(false) const [pathResolveAlias, setPathResolveAlias] = useState('raw') const [pathResolveRelativePath, setPathResolveRelativePath] = useState('') const [pathResolveMustExist, setPathResolveMustExist] = useState(true) const [pathResolveOutput, setPathResolveOutput] = useState('') const [resolvingPath, setResolvingPath] = useState(false) const [tuningTasks, setTuningTasks] = useState([]) const [tuningProfile, setTuningProfile] = useState>({}) const [tuningProfileToml, setTuningProfileToml] = useState('') const [memorySources, setMemorySources] = useState([]) const [deleteOperations, setDeleteOperations] = useState([]) const [selectedOperationDetail, setSelectedOperationDetail] = useState(null) const [selectedOperationDetailLoading, setSelectedOperationDetailLoading] = useState(false) const [selectedOperationDetailError, setSelectedOperationDetailError] = useState('') const [sourceSearch, setSourceSearch] = useState('') const [operationSearch, setOperationSearch] = useState('') const [operationModeFilter, setOperationModeFilter] = useState('all') const [operationStatusFilter, setOperationStatusFilter] = useState('all') const [operationPage, setOperationPage] = useState(1) const [selectedOperationId, setSelectedOperationId] = useState('') const [selectedOperationItemSearch, setSelectedOperationItemSearch] = useState('') const [selectedOperationItemPage, setSelectedOperationItemPage] = useState(1) const [selectedSources, setSelectedSources] = useState([]) const [deleteDialogOpen, setDeleteDialogOpen] = useState(false) const [deleteDialogTitle, setDeleteDialogTitle] = useState('删除预览') const [deleteDialogDescription, setDeleteDialogDescription] = useState('') const [deletePreview, setDeletePreview] = useState> | null>(null) const [deletePreviewError, setDeletePreviewError] = useState(null) const [deletePreviewLoading, setDeletePreviewLoading] = useState(false) const [deleteExecuting, setDeleteExecuting] = useState(false) const [deleteRestoring, setDeleteRestoring] = useState(false) const [deleteResult, setDeleteResult] = useState(null) const [pendingDeleteRequest, setPendingDeleteRequest] = useState(null) const [feedbackCorrections, setFeedbackCorrections] = useState([]) const [feedbackSearch, setFeedbackSearch] = useState('') const [feedbackStatusFilter, setFeedbackStatusFilter] = useState('all') const [feedbackRollbackFilter, setFeedbackRollbackFilter] = useState('all') const [feedbackPage, setFeedbackPage] = useState(1) const [selectedFeedbackTaskId, setSelectedFeedbackTaskId] = useState(0) const [selectedFeedbackTaskDetail, setSelectedFeedbackTaskDetail] = useState(null) const [selectedFeedbackTaskLoading, setSelectedFeedbackTaskLoading] = useState(false) const [selectedFeedbackTaskError, setSelectedFeedbackTaskError] = useState('') const [feedbackActionLogSearch, setFeedbackActionLogSearch] = useState('') const [feedbackActionLogPage, setFeedbackActionLogPage] = useState(1) const [feedbackRollbackDialogOpen, setFeedbackRollbackDialogOpen] = useState(false) const [feedbackRollbackReason, setFeedbackRollbackReason] = useState('') const [feedbackRollingBack, setFeedbackRollingBack] = useState(false) const [tuningObjective, setTuningObjective] = useState('precision_priority') const [tuningIntensity, setTuningIntensity] = useState('standard') const [tuningSampleSize, setTuningSampleSize] = useState('24') const [tuningTopKEval, setTuningTopKEval] = useState('20') const loadPage = useCallback(async () => { try { setLoading(true) const [ schema, configPayload, rawPayload, runtimePayload, importSettingsPayload, pathAliasPayload, importTaskPayload, tuningProfilePayload, tuningTaskPayload, sourcePayload, deleteOperationPayload, feedbackCorrectionPayload, ] = await Promise.all([ getMemoryConfigSchema(), getMemoryConfig(), getMemoryConfigRaw(), getMemoryRuntimeConfig(), getMemoryImportSettings(), getMemoryImportPathAliases(), getMemoryImportTasks(20), getMemoryTuningProfile(), getMemoryTuningTasks(20), getMemorySources(), getMemoryDeleteOperations(DELETE_OPERATION_FETCH_LIMIT), getMemoryFeedbackCorrections({ limit: FEEDBACK_CORRECTION_FETCH_LIMIT }), ]) setSchemaPayload(schema) setVisualConfig(configPayload.config ?? {}) setRawConfig(rawPayload.config ?? '') setRawConfigExists(rawPayload.exists ?? true) setRawConfigUsingDefault(rawPayload.using_default ?? false) setRuntimeConfig(runtimePayload) setImportSettings(importSettingsPayload.settings ?? {}) setImportPathAliases(pathAliasPayload.path_aliases ?? {}) setImportTasks(importTaskPayload.items ?? []) setTuningProfile(tuningProfilePayload.profile ?? {}) setTuningProfileToml(tuningProfilePayload.toml ?? '') setTuningTasks(tuningTaskPayload.items ?? []) setMemorySources(sourcePayload.items ?? []) setDeleteOperations(deleteOperationPayload.items ?? []) setFeedbackCorrections(feedbackCorrectionPayload.items ?? []) if (!selectedImportTaskId && (importTaskPayload.items ?? []).length > 0) { const initialTaskId = String(importTaskPayload.items?.[0]?.task_id ?? '') if (initialTaskId) { setSelectedImportTaskId(initialTaskId) } } if (!maibotSourceDb && String(importSettingsPayload.settings?.maibot_source_db_default ?? '').trim()) { setMaibotSourceDb(String(importSettingsPayload.settings?.maibot_source_db_default ?? '').trim()) } if (!pathResolveAlias) { const aliasKeys = Object.keys(pathAliasPayload.path_aliases ?? {}) if (aliasKeys.length > 0) { setPathResolveAlias(aliasKeys[0]) } } } catch (error) { toast({ title: '加载长期记忆控制台失败', description: error instanceof Error ? error.message : '未知错误', variant: 'destructive', }) } finally { setLoading(false) } }, [maibotSourceDb, pathResolveAlias, selectedImportTaskId, toast]) useEffect(() => { void loadPage() }, [loadPage]) const configPath = schemaPayload?.path ?? 'config/a_memorix.toml' const schema = schemaPayload?.schema const runtimeBadges = useMemo(() => { if (!runtimeConfig) { return [] } return [ { label: '运行状态', value: runtimeConfig.runtime_ready ? '就绪' : '未就绪' }, { label: 'Embedding 维度', value: String(runtimeConfig.embedding_dimension) }, { label: '自动保存', value: runtimeConfig.auto_save ? '开启' : '关闭' }, { label: '数据目录', value: runtimeConfig.data_dir }, ] }, [runtimeConfig]) const importPollInterval = useMemo( () => Math.max(200, Number(importSettings.poll_interval_ms ?? 1000)), [importSettings.poll_interval_ms], ) const importAliasKeys = useMemo( () => Object.keys(importPathAliases).sort((left, right) => left.localeCompare(right)), [importPathAliases], ) const runningImportTasks = useMemo( () => importTasks.filter((task) => RUNNING_IMPORT_STATUS.has(String(task.status ?? '').trim())), [importTasks], ) const queuedImportTasks = useMemo( () => importTasks.filter((task) => QUEUED_IMPORT_STATUS.has(String(task.status ?? '').trim())), [importTasks], ) const recentImportTasks = useMemo( () => importTasks.filter((task) => { const status = String(task.status ?? '').trim() return !RUNNING_IMPORT_STATUS.has(status) && !QUEUED_IMPORT_STATUS.has(status) }), [importTasks], ) const selectedImportTaskSummary = useMemo(() => { if (!selectedImportTaskId) { return null } return importTasks.find((task) => task.task_id === selectedImportTaskId) ?? null }, [importTasks, selectedImportTaskId]) const selectedImportFiles = useMemo(() => { return Array.isArray(selectedImportTask?.files) ? selectedImportTask.files : [] }, [selectedImportTask?.files]) const selectedImportChunks = useMemo(() => { return Array.isArray(importChunksPayload?.items) ? importChunksPayload.items : [] }, [importChunksPayload?.items]) const selectedImportTaskResolved = selectedImportTask ?? selectedImportTaskSummary const selectedImportTaskErrorText = String(selectedImportTaskResolved?.error ?? '').trim() const selectedImportRetrySummary = selectedImportTaskResolved?.retry_summary const importChunkTotal = Number(importChunksPayload?.total ?? 0) const canImportChunkPrev = importChunkOffset > 0 const canImportChunkNext = importChunkOffset + IMPORT_CHUNK_PAGE_SIZE < importChunkTotal const buildCommonImportPayload = useCallback((): Record => { const payload: Record = { llm_enabled: importCommonLlmEnabled, strategy_override: importCommonStrategyOverride, dedupe_policy: importCommonDedupePolicy, chat_log: importCommonChatLog, force: importCommonForce, clear_manifest: importCommonClearManifest, } const fileConcurrency = parseOptionalPositiveInt(importCommonFileConcurrency) const chunkConcurrency = parseOptionalPositiveInt(importCommonChunkConcurrency) if (fileConcurrency !== undefined) { payload.file_concurrency = fileConcurrency } if (chunkConcurrency !== undefined) { payload.chunk_concurrency = chunkConcurrency } if (importCommonChatReferenceTime.trim()) { payload.chat_reference_time = importCommonChatReferenceTime.trim() } return payload }, [ importCommonChatLog, importCommonChatReferenceTime, importCommonChunkConcurrency, importCommonClearManifest, importCommonDedupePolicy, importCommonFileConcurrency, importCommonForce, importCommonLlmEnabled, importCommonStrategyOverride, ]) const refreshImportQueue = useCallback(async (silent: boolean = false) => { try { const [taskPayload, settingsPayload, pathAliasPayload] = await Promise.all([ getMemoryImportTasks(20), getMemoryImportSettings(), getMemoryImportPathAliases(), ]) const nextTasks = taskPayload.items ?? [] setImportTasks(nextTasks) setImportSettings(settingsPayload.settings ?? {}) setImportPathAliases(pathAliasPayload.path_aliases ?? {}) setImportErrorText('') if (nextTasks.length <= 0) { setSelectedImportTaskId('') setSelectedImportTask(null) setSelectedImportFileId('') setImportChunksPayload(null) return } if (!selectedImportTaskId || !nextTasks.some((item) => item.task_id === selectedImportTaskId)) { setSelectedImportTaskId(nextTasks[0].task_id) } } catch (error) { const message = error instanceof Error ? error.message : '刷新导入任务失败' setImportErrorText(message) if (!silent) { toast({ title: '刷新导入任务失败', description: message, variant: 'destructive', }) } } }, [selectedImportTaskId, toast]) const loadImportChunks = useCallback( async ( taskId: string, fileId: string, offset: number = 0, silent: boolean = false, ) => { if (!taskId || !fileId) { setImportChunksPayload(null) return } try { setImportChunksLoading(true) const payload = await getMemoryImportTaskChunks(taskId, fileId, offset, IMPORT_CHUNK_PAGE_SIZE) if (!payload.success) { throw new Error(payload.error || '加载分块详情失败') } setImportChunksPayload(payload) setImportErrorText('') } catch (error) { const message = error instanceof Error ? error.message : '加载分块详情失败' setImportChunksPayload(null) setImportErrorText(message) if (!silent) { toast({ title: '加载分块详情失败', description: message, variant: 'destructive', }) } } finally { setImportChunksLoading(false) } }, [toast], ) const loadImportTaskDetail = useCallback( async (taskId: string, silent: boolean = false) => { if (!taskId) { setSelectedImportTask(null) setSelectedImportFileId('') setImportChunksPayload(null) return } try { if (!silent) { setSelectedImportTaskLoading(true) } const payload = await getMemoryImportTask(taskId, false) if (!payload.success || !payload.task) { throw new Error(payload.error || '任务不存在') } const task = payload.task setSelectedImportTask(task) setImportErrorText('') const files = Array.isArray(task.files) ? task.files : [] const keepCurrentFile = files.some((file) => file.file_id === selectedImportFileId) const nextFileId = keepCurrentFile ? selectedImportFileId : String(files[0]?.file_id ?? '') const nextOffset = keepCurrentFile ? importChunkOffset : 0 if (!keepCurrentFile) { setImportChunkOffset(0) } setSelectedImportFileId(nextFileId) if (nextFileId) { await loadImportChunks(taskId, nextFileId, nextOffset, silent) } else { setImportChunksPayload(null) } } catch (error) { const message = error instanceof Error ? error.message : '加载导入任务详情失败' setSelectedImportTask(null) setSelectedImportFileId('') setImportChunksPayload(null) setImportErrorText(message) if (!silent) { toast({ title: '加载导入任务详情失败', description: message, variant: 'destructive', }) } } finally { if (!silent) { setSelectedImportTaskLoading(false) } } }, [importChunkOffset, loadImportChunks, selectedImportFileId, toast], ) const afterImportTaskCreated = useCallback( async (taskId: string, successTitle: string) => { await refreshImportQueue(true) if (taskId) { setSelectedImportTaskId(taskId) await loadImportTaskDetail(taskId, true) } toast({ title: successTitle, description: taskId ? `任务 ${taskId.slice(0, 12)} 已加入导入队列` : '导入任务已加入队列', }) }, [loadImportTaskDetail, refreshImportQueue, toast], ) const submitUploadImport = useCallback(async () => { if (uploadFiles.length <= 0) { toast({ title: '请选择上传文件', description: '至少选择一个 txt/md/json 文件后再提交', variant: 'destructive', }) return } try { setCreatingImport(true) const payload = { ...buildCommonImportPayload(), input_mode: uploadInputMode, } const result = await createMemoryUploadImport(uploadFiles, payload) if (!result.success) { throw new Error(result.error || '创建上传导入任务失败') } const taskId = String(result.task?.task_id ?? '') setUploadFiles([]) await afterImportTaskCreated(taskId, '上传导入任务已创建') } catch (error) { const message = error instanceof Error ? error.message : '创建上传导入任务失败' setImportErrorText(message) toast({ title: '创建上传导入任务失败', description: message, variant: 'destructive', }) } finally { setCreatingImport(false) } }, [afterImportTaskCreated, buildCommonImportPayload, toast, uploadFiles, uploadInputMode]) const submitPasteImport = useCallback(async () => { if (!pasteContent.trim()) { toast({ title: '粘贴内容不能为空', description: '请填写导入内容后再提交', variant: 'destructive', }) return } try { setCreatingImport(true) const result = await createMemoryPasteImport({ ...buildCommonImportPayload(), name: pasteName || undefined, content: pasteContent, input_mode: pasteMode, }) if (!result.success) { throw new Error(result.error || '创建粘贴导入任务失败') } const taskId = String(result.task?.task_id ?? '') setPasteContent('') setPasteName('') await afterImportTaskCreated(taskId, '粘贴导入任务已创建') } catch (error) { const message = error instanceof Error ? error.message : '创建粘贴导入任务失败' setImportErrorText(message) toast({ title: '创建粘贴导入任务失败', description: message, variant: 'destructive', }) } finally { setCreatingImport(false) } }, [afterImportTaskCreated, buildCommonImportPayload, pasteContent, pasteMode, pasteName, toast]) const submitRawScanImport = useCallback(async () => { try { setCreatingImport(true) const result = await createMemoryRawScanImport({ ...buildCommonImportPayload(), alias: rawAlias, relative_path: rawRelativePath, glob: rawGlob, recursive: rawRecursive, input_mode: rawInputMode, }) if (!result.success) { throw new Error(result.error || '创建本地扫描任务失败') } await afterImportTaskCreated(String(result.task?.task_id ?? ''), '本地扫描任务已创建') } catch (error) { const message = error instanceof Error ? error.message : '创建本地扫描任务失败' setImportErrorText(message) toast({ title: '创建本地扫描任务失败', description: message, variant: 'destructive', }) } finally { setCreatingImport(false) } }, [ afterImportTaskCreated, buildCommonImportPayload, rawAlias, rawGlob, rawInputMode, rawRecursive, rawRelativePath, toast, ]) const submitOpenieImport = useCallback(async () => { try { setCreatingImport(true) const result = await createMemoryLpmmOpenieImport({ ...buildCommonImportPayload(), alias: openieAlias, relative_path: openieRelativePath, include_all_json: openieIncludeAllJson, }) if (!result.success) { throw new Error(result.error || '创建 LPMM OpenIE 任务失败') } await afterImportTaskCreated(String(result.task?.task_id ?? ''), 'LPMM OpenIE 任务已创建') } catch (error) { const message = error instanceof Error ? error.message : '创建 LPMM OpenIE 任务失败' setImportErrorText(message) toast({ title: '创建 LPMM OpenIE 任务失败', description: message, variant: 'destructive', }) } finally { setCreatingImport(false) } }, [ afterImportTaskCreated, buildCommonImportPayload, openieAlias, openieIncludeAllJson, openieRelativePath, toast, ]) const submitConvertImport = useCallback(async () => { try { setCreatingImport(true) const result = await createMemoryLpmmConvertImport({ alias: convertAlias, relative_path: convertRelativePath, target_alias: convertTargetAlias, target_relative_path: convertTargetRelativePath, dimension: parseOptionalPositiveInt(convertDimension), batch_size: parseOptionalPositiveInt(convertBatchSize), }) if (!result.success) { throw new Error(result.error || '创建 LPMM 转换任务失败') } await afterImportTaskCreated(String(result.task?.task_id ?? ''), 'LPMM 转换任务已创建') } catch (error) { const message = error instanceof Error ? error.message : '创建 LPMM 转换任务失败' setImportErrorText(message) toast({ title: '创建 LPMM 转换任务失败', description: message, variant: 'destructive', }) } finally { setCreatingImport(false) } }, [ afterImportTaskCreated, convertAlias, convertBatchSize, convertDimension, convertRelativePath, convertTargetAlias, convertTargetRelativePath, toast, ]) const submitBackfillImport = useCallback(async () => { try { setCreatingImport(true) const result = await createMemoryTemporalBackfillImport({ alias: backfillAlias, relative_path: backfillRelativePath, limit: parseOptionalPositiveInt(backfillLimit), dry_run: backfillDryRun, no_created_fallback: backfillNoCreatedFallback, }) if (!result.success) { throw new Error(result.error || '创建时序回填任务失败') } await afterImportTaskCreated(String(result.task?.task_id ?? ''), '时序回填任务已创建') } catch (error) { const message = error instanceof Error ? error.message : '创建时序回填任务失败' setImportErrorText(message) toast({ title: '创建时序回填任务失败', description: message, variant: 'destructive', }) } finally { setCreatingImport(false) } }, [ afterImportTaskCreated, backfillAlias, backfillDryRun, backfillLimit, backfillNoCreatedFallback, backfillRelativePath, toast, ]) const submitMaibotMigrationImport = useCallback(async () => { try { setCreatingImport(true) const result = await createMemoryMaibotMigrationImport({ source_db: maibotSourceDb || undefined, time_from: maibotTimeFrom || undefined, time_to: maibotTimeTo || undefined, start_id: parseOptionalPositiveInt(maibotStartId), end_id: parseOptionalPositiveInt(maibotEndId), stream_ids: parseCommaSeparatedList(maibotStreamIds), group_ids: parseCommaSeparatedList(maibotGroupIds), user_ids: parseCommaSeparatedList(maibotUserIds), read_batch_size: parseOptionalPositiveInt(maibotReadBatchSize), commit_window_rows: parseOptionalPositiveInt(maibotCommitWindowRows), embed_workers: parseOptionalPositiveInt(maibotEmbedWorkers), no_resume: maibotNoResume, reset_state: maibotResetState, dry_run: maibotDryRun, verify_only: maibotVerifyOnly, }) if (!result.success) { throw new Error(result.error || '创建 MaiBot 迁移任务失败') } await afterImportTaskCreated(String(result.task?.task_id ?? ''), 'MaiBot 迁移任务已创建') } catch (error) { const message = error instanceof Error ? error.message : '创建 MaiBot 迁移任务失败' setImportErrorText(message) toast({ title: '创建 MaiBot 迁移任务失败', description: message, variant: 'destructive', }) } finally { setCreatingImport(false) } }, [ afterImportTaskCreated, maibotCommitWindowRows, maibotDryRun, maibotEmbedWorkers, maibotEndId, maibotGroupIds, maibotNoResume, maibotReadBatchSize, maibotResetState, maibotSourceDb, maibotStartId, maibotStreamIds, maibotTimeFrom, maibotTimeTo, maibotUserIds, maibotVerifyOnly, toast, ]) const cancelSelectedImportTask = useCallback(async () => { if (!selectedImportTaskId) { return } try { const payload = await cancelMemoryImportTask(selectedImportTaskId) if (!payload.success) { throw new Error(payload.error || '取消导入任务失败') } await refreshImportQueue(true) await loadImportTaskDetail(selectedImportTaskId, true) toast({ title: '已请求取消任务', description: `任务 ${selectedImportTaskId.slice(0, 12)} 正在取消`, }) } catch (error) { const message = error instanceof Error ? error.message : '取消导入任务失败' setImportErrorText(message) toast({ title: '取消导入任务失败', description: message, variant: 'destructive', }) } }, [loadImportTaskDetail, refreshImportQueue, selectedImportTaskId, toast]) const retrySelectedImportTask = useCallback(async () => { if (!selectedImportTaskId) { return } try { const payload = await retryMemoryImportTask(selectedImportTaskId, { overrides: buildCommonImportPayload(), }) if (!payload.success) { throw new Error(payload.error || '重试失败项失败') } const nextTaskId = String(payload.task?.task_id ?? '') await refreshImportQueue(true) if (nextTaskId) { setSelectedImportTaskId(nextTaskId) await loadImportTaskDetail(nextTaskId, true) } else { await loadImportTaskDetail(selectedImportTaskId, true) } toast({ title: '重试任务已创建', description: nextTaskId ? `重试任务 ${nextTaskId.slice(0, 12)} 已进入队列` : '失败项已提交重试', }) } catch (error) { const message = error instanceof Error ? error.message : '重试失败项失败' setImportErrorText(message) toast({ title: '重试失败项失败', description: message, variant: 'destructive', }) } }, [buildCommonImportPayload, loadImportTaskDetail, refreshImportQueue, selectedImportTaskId, toast]) const resolveImportPath = useCallback(async () => { if (!pathResolveAlias.trim()) { return } try { setResolvingPath(true) const payload = await resolveMemoryImportPath({ alias: pathResolveAlias, relative_path: pathResolveRelativePath, must_exist: pathResolveMustExist, }) const lines = [ `路径别名: ${payload.alias}`, `相对路径: ${payload.relative_path || '(空)'}`, `解析结果: ${payload.resolved_path}`, `是否存在: ${String(payload.exists)}`, `是否文件: ${String(payload.is_file)}`, `是否目录: ${String(payload.is_dir)}`, ] setPathResolveOutput(lines.join('\n')) } catch (error) { const message = error instanceof Error ? error.message : '路径解析失败' setPathResolveOutput(`解析失败:${message}`) } finally { setResolvingPath(false) } }, [pathResolveAlias, pathResolveMustExist, pathResolveRelativePath]) const selectImportTask = useCallback( async (taskId: string) => { setSelectedImportTaskId(taskId) setImportChunkOffset(0) await loadImportTaskDetail(taskId) }, [loadImportTaskDetail], ) const selectImportFile = useCallback( async (fileId: string) => { if (!selectedImportTaskId) { return } setSelectedImportFileId(fileId) setImportChunkOffset(0) await loadImportChunks(selectedImportTaskId, fileId, 0) }, [loadImportChunks, selectedImportTaskId], ) const moveImportChunkPage = useCallback( async (direction: -1 | 1) => { if (!selectedImportTaskId || !selectedImportFileId) { return } const nextOffset = direction < 0 ? Math.max(0, importChunkOffset - IMPORT_CHUNK_PAGE_SIZE) : importChunkOffset + IMPORT_CHUNK_PAGE_SIZE if (nextOffset === importChunkOffset) { return } setImportChunkOffset(nextOffset) await loadImportChunks(selectedImportTaskId, selectedImportFileId, nextOffset) }, [importChunkOffset, loadImportChunks, selectedImportFileId, selectedImportTaskId], ) useEffect(() => { if (importAliasKeys.length <= 0) { return } const pickAlias = (current: string, preferred: string): string => { if (current && importAliasKeys.includes(current)) { return current } if (importAliasKeys.includes(preferred)) { return preferred } return importAliasKeys[0] } setRawAlias((current) => pickAlias(current, 'raw')) setOpenieAlias((current) => pickAlias(current, 'lpmm')) setConvertAlias((current) => pickAlias(current, 'lpmm')) setConvertTargetAlias((current) => pickAlias(current, 'plugin_data')) setBackfillAlias((current) => pickAlias(current, 'plugin_data')) setPathResolveAlias((current) => pickAlias(current, 'raw')) }, [importAliasKeys]) useEffect(() => { const defaultFileConcurrency = String(importSettings.default_file_concurrency ?? '').trim() const defaultChunkConcurrency = String(importSettings.default_chunk_concurrency ?? '').trim() if (defaultFileConcurrency && importCommonFileConcurrency === '2') { setImportCommonFileConcurrency(defaultFileConcurrency) } if (defaultChunkConcurrency && importCommonChunkConcurrency === '4') { setImportCommonChunkConcurrency(defaultChunkConcurrency) } const defaultSourceDb = String(importSettings.maibot_source_db_default ?? '').trim() if (defaultSourceDb && !maibotSourceDb.trim()) { setMaibotSourceDb(defaultSourceDb) } }, [ importCommonChunkConcurrency, importCommonFileConcurrency, importSettings.default_chunk_concurrency, importSettings.default_file_concurrency, importSettings.maibot_source_db_default, maibotSourceDb, ]) useEffect(() => { if (!selectedImportTaskId && importTasks.length > 0) { void selectImportTask(importTasks[0].task_id) } }, [importTasks, selectImportTask, selectedImportTaskId]) useEffect(() => { if (!selectedImportTaskId) { setSelectedImportTask(null) setSelectedImportFileId('') setImportChunksPayload(null) return } if (!importTasks.some((task) => task.task_id === selectedImportTaskId) && importTasks.length > 0) { void selectImportTask(importTasks[0].task_id) return } void loadImportTaskDetail(selectedImportTaskId, true) }, [importTasks, loadImportTaskDetail, selectImportTask, selectedImportTaskId]) useEffect(() => { if (!importAutoPolling) { return } const timerId = window.setInterval(() => { void refreshImportQueue(true) if (selectedImportTaskId) { void loadImportTaskDetail(selectedImportTaskId, true) } }, importPollInterval) return () => { window.clearInterval(timerId) } }, [importAutoPolling, importPollInterval, loadImportTaskDetail, refreshImportQueue, selectedImportTaskId]) const filteredSources = useMemo(() => { const keyword = sourceSearch.trim().toLowerCase() if (!keyword) { return memorySources } return memorySources.filter((item) => String(item.source ?? '').toLowerCase().includes(keyword)) }, [memorySources, sourceSearch]) const filteredDeleteOperations = useMemo(() => { const keyword = operationSearch.trim().toLowerCase() return deleteOperations.filter((operation) => { const mode = String(operation.mode ?? '').trim() const status = String(operation.status ?? '').trim() const summary = operation.summary ?? {} const sources = Array.isArray(summary.sources) ? summary.sources : [] if (operationModeFilter !== 'all' && mode !== operationModeFilter) { return false } if (operationStatusFilter !== 'all' && status !== operationStatusFilter) { return false } if (!keyword) { return true } return [ operation.operation_id, operation.reason, operation.requested_by, mode, status, ...sources.map((item) => String(item)), ] .map((item) => String(item ?? '').toLowerCase()) .some((item) => item.includes(keyword)) }) }, [deleteOperations, operationModeFilter, operationSearch, operationStatusFilter]) const deleteOperationPageCount = Math.max(1, Math.ceil(filteredDeleteOperations.length / DELETE_OPERATION_PAGE_SIZE)) const pagedDeleteOperations = useMemo(() => { const start = (operationPage - 1) * DELETE_OPERATION_PAGE_SIZE return filteredDeleteOperations.slice(start, start + DELETE_OPERATION_PAGE_SIZE) }, [filteredDeleteOperations, operationPage]) const selectedDeleteOperation = useMemo( () => filteredDeleteOperations.find((operation) => operation.operation_id === selectedOperationId) ?? pagedDeleteOperations[0] ?? null, [filteredDeleteOperations, pagedDeleteOperations, selectedOperationId], ) useEffect(() => { setOperationPage(1) }, [operationSearch, operationModeFilter, operationStatusFilter]) useEffect(() => { if (operationPage > deleteOperationPageCount) { setOperationPage(deleteOperationPageCount) } }, [deleteOperationPageCount, operationPage]) useEffect(() => { if (!selectedDeleteOperation) { if (selectedOperationId) { setSelectedOperationId('') } setSelectedOperationDetail(null) setSelectedOperationDetailError('') return } if (selectedDeleteOperation.operation_id !== selectedOperationId) { setSelectedOperationId(selectedDeleteOperation.operation_id) } }, [selectedDeleteOperation, selectedOperationId]) useEffect(() => { const operationId = selectedDeleteOperation?.operation_id if (!operationId) { setSelectedOperationDetail(null) setSelectedOperationDetailError('') return } let cancelled = false setSelectedOperationDetailLoading(true) setSelectedOperationDetailError('') void getMemoryDeleteOperation(operationId) .then((payload) => { if (cancelled) { return } if (!payload.success || !payload.operation) { setSelectedOperationDetail(null) setSelectedOperationDetailError(payload.error || '未能加载删除操作详情') return } setSelectedOperationDetail(payload.operation) }) .catch((error) => { if (cancelled) { return } setSelectedOperationDetail(null) setSelectedOperationDetailError(error instanceof Error ? error.message : '未能加载删除操作详情') }) .finally(() => { if (!cancelled) { setSelectedOperationDetailLoading(false) } }) return () => { cancelled = true } }, [selectedDeleteOperation?.operation_id]) const toggleSourceSelection = useCallback((source: string, checked: boolean) => { setSelectedSources((current) => { if (checked) { return current.includes(source) ? current : [...current, source] } return current.filter((item) => item !== source) }) }, []) const openSourceDeletePreview = useCallback(async () => { if (selectedSources.length <= 0) { toast({ title: '请选择来源', description: '至少选择一个来源后再进行删除预览', variant: 'destructive', }) return } const request: MemoryDeleteRequestPayload = { mode: 'source', selector: { sources: selectedSources }, reason: 'knowledge_base_source_delete', requested_by: 'knowledge_base', } setDeleteDialogTitle('批量删除来源') setDeleteDialogDescription('删除来源只会删除该来源下的段落,以及失去全部证据的关系,不会自动删除实体') setPendingDeleteRequest(request) setDeletePreview(null) setDeleteResult(null) setDeletePreviewError(null) setDeleteDialogOpen(true) setDeletePreviewLoading(true) try { const preview = await previewMemoryDelete(request) setDeletePreview(preview) } catch (error) { setDeletePreviewError(error instanceof Error ? error.message : '删除预览失败') } finally { setDeletePreviewLoading(false) } }, [selectedSources, toast]) const executePendingDelete = useCallback(async () => { if (!pendingDeleteRequest) { return } try { setDeleteExecuting(true) const result = await executeMemoryDelete(pendingDeleteRequest) setDeleteResult(result) toast({ title: result.success ? '删除成功' : '删除失败', description: result.success ? `操作 ${result.operation_id} 已完成` : result.error || '未能执行删除', variant: result.success ? 'default' : 'destructive', }) if (result.success) { const [sourcePayload, deleteOperationPayload] = await Promise.all([ getMemorySources(), getMemoryDeleteOperations(DELETE_OPERATION_FETCH_LIMIT), ]) setMemorySources(sourcePayload.items ?? []) setDeleteOperations(deleteOperationPayload.items ?? []) setSelectedSources([]) } } catch (error) { setDeletePreviewError(error instanceof Error ? error.message : '删除失败') toast({ title: '删除失败', description: error instanceof Error ? error.message : '未知错误', variant: 'destructive', }) } finally { setDeleteExecuting(false) } }, [pendingDeleteRequest, toast]) const restoreDeleteOperation = useCallback(async (operationId: string) => { try { setDeleteRestoring(true) await restoreMemoryDelete({ operation_id: operationId, requested_by: 'knowledge_base' }) toast({ title: '恢复成功', description: `删除操作 ${operationId} 已恢复`, }) setDeleteDialogOpen(false) const [sourcePayload, deleteOperationPayload] = await Promise.all([ getMemorySources(), getMemoryDeleteOperations(DELETE_OPERATION_FETCH_LIMIT), ]) setMemorySources(sourcePayload.items ?? []) setDeleteOperations(deleteOperationPayload.items ?? []) } catch (error) { toast({ title: '恢复失败', description: error instanceof Error ? error.message : '未知错误', variant: 'destructive', }) } finally { setDeleteRestoring(false) } }, [toast]) const closeDeleteDialog = useCallback((open: boolean) => { if (!open) { setDeleteDialogOpen(false) setDeletePreview(null) setDeleteResult(null) setDeletePreviewError(null) setPendingDeleteRequest(null) return } setDeleteDialogOpen(true) }, []) const filteredFeedbackCorrections = useMemo(() => { const keyword = feedbackSearch.trim().toLowerCase() return feedbackCorrections.filter((item) => { const taskStatus = String(item.task_status ?? '').trim().toLowerCase() const rollbackStatus = String(item.rollback_status ?? '').trim().toLowerCase() if (feedbackStatusFilter !== 'all' && taskStatus !== feedbackStatusFilter) { return false } if (feedbackRollbackFilter !== 'all' && rollbackStatus !== feedbackRollbackFilter) { return false } if (!keyword) { return true } return [ item.query_tool_id, item.session_id, item.query_text, item.decision, item.task_status, item.rollback_status, ] .map((value) => String(value ?? '').toLowerCase()) .some((value) => value.includes(keyword)) }) }, [feedbackCorrections, feedbackRollbackFilter, feedbackSearch, feedbackStatusFilter]) const feedbackPageCount = Math.max(1, Math.ceil(filteredFeedbackCorrections.length / FEEDBACK_CORRECTION_PAGE_SIZE)) const pagedFeedbackCorrections = useMemo(() => { const start = (feedbackPage - 1) * FEEDBACK_CORRECTION_PAGE_SIZE return filteredFeedbackCorrections.slice(start, start + FEEDBACK_CORRECTION_PAGE_SIZE) }, [feedbackPage, filteredFeedbackCorrections]) const selectedFeedbackCorrection = useMemo( () => filteredFeedbackCorrections.find((item) => item.task_id === selectedFeedbackTaskId) ?? pagedFeedbackCorrections[0] ?? null, [filteredFeedbackCorrections, pagedFeedbackCorrections, selectedFeedbackTaskId], ) useEffect(() => { setFeedbackPage(1) }, [feedbackSearch, feedbackStatusFilter, feedbackRollbackFilter]) useEffect(() => { if (feedbackPage > feedbackPageCount) { setFeedbackPage(feedbackPageCount) } }, [feedbackPage, feedbackPageCount]) useEffect(() => { if (!selectedFeedbackCorrection) { if (selectedFeedbackTaskId) { setSelectedFeedbackTaskId(0) } setSelectedFeedbackTaskDetail(null) setSelectedFeedbackTaskError('') return } if (selectedFeedbackCorrection.task_id !== selectedFeedbackTaskId) { setSelectedFeedbackTaskId(selectedFeedbackCorrection.task_id) } }, [selectedFeedbackCorrection, selectedFeedbackTaskId]) useEffect(() => { const taskId = selectedFeedbackCorrection?.task_id if (!taskId) { setSelectedFeedbackTaskDetail(null) setSelectedFeedbackTaskError('') return } let cancelled = false setSelectedFeedbackTaskLoading(true) setSelectedFeedbackTaskError('') void getMemoryFeedbackCorrection(taskId) .then((payload) => { if (cancelled) { return } if (!payload.success || !payload.task) { setSelectedFeedbackTaskDetail(null) setSelectedFeedbackTaskError(payload.error || '未能加载纠错任务详情') return } setSelectedFeedbackTaskDetail(payload.task) }) .catch((error) => { if (cancelled) { return } setSelectedFeedbackTaskDetail(null) setSelectedFeedbackTaskError(error instanceof Error ? error.message : '未能加载纠错任务详情') }) .finally(() => { if (!cancelled) { setSelectedFeedbackTaskLoading(false) } }) return () => { cancelled = true } }, [selectedFeedbackCorrection?.task_id]) const selectedFeedbackResolved = useMemo(() => { if (!selectedFeedbackCorrection) { return null } if (selectedFeedbackTaskDetail?.task_id === selectedFeedbackCorrection.task_id) { return { ...selectedFeedbackCorrection, ...selectedFeedbackTaskDetail, } satisfies MemoryFeedbackCorrectionDetailTaskPayload } return selectedFeedbackTaskDetail ?? selectedFeedbackCorrection }, [selectedFeedbackCorrection, selectedFeedbackTaskDetail]) const selectedFeedbackActionLogs = Array.isArray(selectedFeedbackResolved?.action_logs) ? selectedFeedbackResolved.action_logs : [] const filteredFeedbackActionLogs = useMemo(() => { const keyword = feedbackActionLogSearch.trim().toLowerCase() if (!keyword) { return selectedFeedbackActionLogs } return selectedFeedbackActionLogs.filter((item) => [ item.action_type, item.target_hash, item.reason, summarizeFeedbackActionPayload(item.before_payload), summarizeFeedbackActionPayload(item.after_payload), ] .map((value) => String(value ?? '').toLowerCase()) .some((value) => value.includes(keyword)), ) }, [feedbackActionLogSearch, selectedFeedbackActionLogs]) const feedbackActionLogPageCount = Math.max( 1, Math.ceil(filteredFeedbackActionLogs.length / FEEDBACK_ACTION_LOG_PAGE_SIZE), ) const pagedFeedbackActionLogs = useMemo(() => { const start = (feedbackActionLogPage - 1) * FEEDBACK_ACTION_LOG_PAGE_SIZE return filteredFeedbackActionLogs.slice(start, start + FEEDBACK_ACTION_LOG_PAGE_SIZE) }, [feedbackActionLogPage, filteredFeedbackActionLogs]) useEffect(() => { setFeedbackActionLogPage(1) }, [selectedFeedbackTaskId, feedbackActionLogSearch]) useEffect(() => { if (feedbackActionLogPage > feedbackActionLogPageCount) { setFeedbackActionLogPage(feedbackActionLogPageCount) } }, [feedbackActionLogPage, feedbackActionLogPageCount]) const openFeedbackRollbackDialog = useCallback(() => { setFeedbackRollbackReason('') setFeedbackRollbackDialogOpen(true) }, []) const executeFeedbackRollback = useCallback(async () => { const taskId = selectedFeedbackResolved?.task_id if (!taskId) { return } try { setFeedbackRollingBack(true) const payload = await rollbackMemoryFeedbackCorrection(taskId, { requested_by: 'knowledge_base', reason: feedbackRollbackReason.trim(), }) if (!payload.success) { throw new Error(payload.error || '回退失败') } toast({ title: payload.already_rolled_back ? '该纠错已回退' : '纠错回退成功', description: `任务 ${taskId} 的回退结果已写入日志`, }) setFeedbackRollbackDialogOpen(false) const [listPayload, detailPayload] = await Promise.all([ getMemoryFeedbackCorrections({ limit: FEEDBACK_CORRECTION_FETCH_LIMIT }), getMemoryFeedbackCorrection(taskId), ]) setFeedbackCorrections(listPayload.items ?? []) setSelectedFeedbackTaskDetail(detailPayload.task ?? null) const [sourcePayload, runtimePayload] = await Promise.all([ getMemorySources(), getMemoryRuntimeConfig(), ]) setMemorySources(sourcePayload.items ?? []) setRuntimeConfig(runtimePayload) } catch (error) { toast({ title: '纠错回退失败', description: error instanceof Error ? error.message : '未知错误', variant: 'destructive', }) } finally { setFeedbackRollingBack(false) } }, [feedbackRollbackReason, selectedFeedbackResolved?.task_id, toast]) const selectedOperationResolved = useMemo(() => { if (!selectedDeleteOperation) { return null } if (selectedOperationDetail?.operation_id === selectedDeleteOperation.operation_id) { return { ...selectedDeleteOperation, ...selectedOperationDetail, } satisfies MemoryDeleteOperationPayload } return selectedDeleteOperation }, [selectedDeleteOperation, selectedOperationDetail]) const selectedOperationSummaryResolved = ((selectedOperationResolved?.summary ?? {}) as Record) const selectedOperationCounts = ((selectedOperationSummaryResolved.counts as Record | undefined) ?? {}) const selectedOperationSources = Array.isArray(selectedOperationSummaryResolved.sources) ? selectedOperationSummaryResolved.sources.map((item) => String(item)).filter(Boolean) : [] const selectedOperationItems = Array.isArray(selectedOperationResolved?.items) ? selectedOperationResolved.items : [] const filteredSelectedOperationItems = useMemo(() => { const keyword = selectedOperationItemSearch.trim().toLowerCase() if (!keyword) { return selectedOperationItems } return selectedOperationItems.filter((item) => { const payload = item.payload ?? {} const source = String(payload.source ?? '').trim() return [ item.item_type, item.item_hash, item.item_key, source, ] .map((value) => String(value ?? '').toLowerCase()) .some((value) => value.includes(keyword)) }) }, [selectedOperationItemSearch, selectedOperationItems]) const selectedOperationItemPageCount = Math.max( 1, Math.ceil(filteredSelectedOperationItems.length / DELETE_OPERATION_ITEM_PAGE_SIZE), ) const pagedSelectedOperationItems = useMemo(() => { const start = (selectedOperationItemPage - 1) * DELETE_OPERATION_ITEM_PAGE_SIZE return filteredSelectedOperationItems.slice(start, start + DELETE_OPERATION_ITEM_PAGE_SIZE) }, [filteredSelectedOperationItems, selectedOperationItemPage]) useEffect(() => { setSelectedOperationItemPage(1) }, [selectedOperationId, selectedOperationItemSearch]) useEffect(() => { if (selectedOperationItemPage > selectedOperationItemPageCount) { setSelectedOperationItemPage(selectedOperationItemPageCount) } }, [selectedOperationItemPage, selectedOperationItemPageCount]) const saveVisualConfig = useCallback(async () => { try { setSaving(true) await updateMemoryConfig(visualConfig) const [nextConfig, nextRaw, nextRuntime] = await Promise.all([ getMemoryConfig(), getMemoryConfigRaw(), getMemoryRuntimeConfig(), ]) setVisualConfig(nextConfig.config) setRawConfig(nextRaw.config) setRawConfigExists(nextRaw.exists ?? true) setRawConfigUsingDefault(nextRaw.using_default ?? false) setRuntimeConfig(nextRuntime) toast({ title: '配置已保存', description: '长期记忆配置已经应用到运行时' }) } catch (error) { toast({ title: '保存配置失败', description: error instanceof Error ? error.message : '未知错误', variant: 'destructive', }) } finally { setSaving(false) } }, [toast, visualConfig]) const saveRaw = useCallback(async () => { try { setSaving(true) await updateMemoryConfigRaw(rawConfig) const [nextConfig, nextRaw, nextRuntime] = await Promise.all([ getMemoryConfig(), getMemoryConfigRaw(), getMemoryRuntimeConfig(), ]) setVisualConfig(nextConfig.config) setRawConfig(nextRaw.config ?? '') setRawConfigExists(nextRaw.exists ?? true) setRawConfigUsingDefault(nextRaw.using_default ?? false) setRuntimeConfig(nextRuntime) toast({ title: '原始 TOML 已保存', description: '长期记忆配置已经重新加载' }) } catch (error) { toast({ title: '保存原始配置失败', description: error instanceof Error ? error.message : '未知错误', variant: 'destructive', }) } finally { setSaving(false) } }, [rawConfig, toast]) const refreshSelfCheck = useCallback(async () => { try { setRefreshingCheck(true) const payload = await refreshMemoryRuntimeSelfCheck() setSelfCheckReport((payload.report ?? null) as Record | null) const nextRuntime = await getMemoryRuntimeConfig() setRuntimeConfig(nextRuntime) toast({ title: payload.success ? '自检通过' : '自检未通过', description: payload.success ? '运行时状态正常' : '请检查 embedding 配置和外部服务连通性', variant: payload.success ? 'default' : 'destructive', }) } catch (error) { toast({ title: '运行时自检失败', description: error instanceof Error ? error.message : '未知错误', variant: 'destructive', }) } finally { setRefreshingCheck(false) } }, [toast]) const submitImportByMode = useCallback(async () => { if (creatingImport) { return } switch (importCreateMode) { case 'upload': await submitUploadImport() break case 'paste': await submitPasteImport() break case 'raw_scan': await submitRawScanImport() break case 'lpmm_openie': await submitOpenieImport() break case 'lpmm_convert': await submitConvertImport() break case 'temporal_backfill': await submitBackfillImport() break case 'maibot_migration': await submitMaibotMigrationImport() break default: break } }, [ creatingImport, importCreateMode, submitBackfillImport, submitConvertImport, submitMaibotMigrationImport, submitOpenieImport, submitPasteImport, submitRawScanImport, submitUploadImport, ]) const submitTuningTask = useCallback(async () => { try { setCreatingTuning(true) await createMemoryTuningTask({ objective: tuningObjective, intensity: tuningIntensity, sample_size: Number(tuningSampleSize), top_k_eval: Number(tuningTopKEval), }) const tasks = await getMemoryTuningTasks(20) setTuningTasks(tasks.items ?? []) toast({ title: '调优任务已创建', description: '新的检索调优任务已经进入队列' }) } catch (error) { toast({ title: '创建调优任务失败', description: error instanceof Error ? error.message : '未知错误', variant: 'destructive', }) } finally { setCreatingTuning(false) } }, [toast, tuningIntensity, tuningObjective, tuningSampleSize, tuningTopKEval]) const applyBestTask = useCallback(async (taskId: string) => { try { await applyBestMemoryTuningProfile(taskId) const [profilePayload, runtimePayload, tuningTaskPayload] = await Promise.all([ getMemoryTuningProfile(), getMemoryRuntimeConfig(), getMemoryTuningTasks(20), ]) setTuningProfile(profilePayload.profile ?? {}) setTuningProfileToml(profilePayload.toml ?? '') setRuntimeConfig(runtimePayload) setTuningTasks(tuningTaskPayload.items ?? []) toast({ title: '最佳参数已应用', description: `任务 ${taskId} 的最佳轮次已经写入运行时` }) } catch (error) { toast({ title: '应用最佳参数失败', description: error instanceof Error ? error.message : '未知错误', variant: 'destructive', }) } }, [toast]) if (loading) { return (
正在加载长期记忆控制台...
) } return (

长期记忆控制台

A_Memorix 的配置、自检、导入和检索调优,都在这里!

{runtimeBadges.map((item) => ( {item.label} {item.value} ))}
概览 配置 导入 调优 删除 纠错历史
运行时自检 用于确认 embedding、向量库与运行时状态是否一致
当前配置文件路径:{configPath}
当前运行态摘要 这里展示运行态重点指标,方便先判断是否需要导入或调优
{runtimeConfig?.runtime_ready ? '运行就绪' : '运行未就绪'} {runtimeConfig?.embedding_degraded ? 'Embedding 已退化' : 'Embedding 正常'}
待补回段落向量
{runtimeConfig?.paragraph_vector_backfill_pending ?? 0}
失败补回任务
{runtimeConfig?.paragraph_vector_backfill_failed ?? 0}
当前调优配置
                        {JSON.stringify(tuningProfile, null, 2)}
                      
长期记忆配置 常用字段可视化编辑,长尾高级项继续通过原始 TOML 维护
当前配置文件:{configPath} {schema?._note ? `;${schema._note}` : ''} {!rawConfigExists || rawConfigUsingDefault ? ( 检测到配置文件尚未保存,当前展示的是默认模板内容点击“保存”后相关配置文件会自动创建 {' '} {configPath} ) : null} {rawMode ? ( ) : schema ? ( ) : (
当前未能加载配置 schema,请先刷新页面或检查后端日志
)}
创建导入任务 同页完成模式选择、参数设置与任务创建,不再切换到旧导入页 setImportCreateMode(value as MemoryImportTaskKind)} className="space-y-4" >
{IMPORT_KIND_OPTIONS.map((item) => ( {item.label} ))}
公共参数
所有导入模式共用,创建任务时会自动并入请求参数
同时处理的文件数量
setImportCommonFileConcurrency(event.target.value)} />
单文件内并行分块数量
setImportCommonChunkConcurrency(event.target.value)} />
setImportCommonLlmEnabled(Boolean(value))} /> 启用 LLM 抽取
setImportCommonChatLog(Boolean(value))} /> 按聊天日志解析
高级参数
setImportCommonStrategyOverride(event.target.value)} />
setImportCommonDedupePolicy(event.target.value)} />
setImportCommonChatReferenceTime(event.target.value)} />
setImportCommonForce(Boolean(value))} /> 强制导入
setImportCommonClearManifest(Boolean(value))} /> 清空任务清单
上传本地文件,适合批量导入
setUploadFiles(Array.from(event.target.files ?? []))} />
已选择 {uploadFiles.length} 个文件
粘贴少量文本,适合临时导入
setPasteName(event.target.value)} />