2025-04-25 17:24:47 +08:00
|
|
|
import type { NodeDefault } from '../../types'
|
|
|
|
import type { KnowledgeBaseNodeType } from './types'
|
2025-05-06 17:24:30 +08:00
|
|
|
import {
|
|
|
|
ChunkStructureEnum,
|
|
|
|
HybridSearchModeEnum,
|
|
|
|
IndexMethodEnum,
|
|
|
|
RetrievalSearchMethodEnum,
|
|
|
|
} from './types'
|
2025-04-25 17:24:47 +08:00
|
|
|
import { genNodeMetaData } from '@/app/components/workflow/utils'
|
|
|
|
import { BlockEnum } from '@/app/components/workflow/types'
|
|
|
|
|
2025-04-29 16:11:54 +08:00
|
|
|
const metaData = genNodeMetaData({
|
|
|
|
sort: 3.1,
|
|
|
|
type: BlockEnum.KnowledgeBase,
|
|
|
|
})
|
2025-04-25 17:24:47 +08:00
|
|
|
const nodeDefault: NodeDefault<KnowledgeBaseNodeType> = {
|
2025-04-29 16:11:54 +08:00
|
|
|
metaData,
|
2025-05-06 17:24:30 +08:00
|
|
|
defaultValue: {
|
|
|
|
index_chunk_variable_selector: [],
|
|
|
|
chunk_structure: ChunkStructureEnum.general,
|
|
|
|
indexing_technique: IndexMethodEnum.QUALIFIED,
|
|
|
|
keyword_number: 10,
|
|
|
|
retrieval_model: {
|
|
|
|
search_method: RetrievalSearchMethodEnum.hybrid,
|
|
|
|
top_k: 2,
|
|
|
|
score_threshold_enabled: false,
|
|
|
|
score_threshold: 0.5,
|
|
|
|
hybridSearchMode: HybridSearchModeEnum.WeightedScore,
|
|
|
|
},
|
|
|
|
},
|
2025-04-25 17:24:47 +08:00
|
|
|
checkValid() {
|
|
|
|
return {
|
|
|
|
isValid: true,
|
|
|
|
errorMessage: '',
|
|
|
|
}
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
export default nodeDefault
|