40 lines
1.0 KiB
TypeScript
Raw Normal View History

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