2024-04-08 18:51:46 +08:00
|
|
|
import type { FC } from 'react'
|
2024-07-24 12:50:48 +08:00
|
|
|
import {
|
|
|
|
memo,
|
|
|
|
useCallback,
|
2025-03-18 11:01:06 +08:00
|
|
|
useMemo,
|
2024-07-24 12:50:48 +08:00
|
|
|
} from 'react'
|
2025-03-18 11:01:06 +08:00
|
|
|
import { intersectionBy } from 'lodash-es'
|
2024-04-08 18:51:46 +08:00
|
|
|
import { useTranslation } from 'react-i18next'
|
|
|
|
import VarReferencePicker from '../_base/components/variable/var-reference-picker'
|
|
|
|
import useConfig from './use-config'
|
|
|
|
import RetrievalConfig from './components/retrieval-config'
|
|
|
|
import AddKnowledge from './components/add-dataset'
|
|
|
|
import DatasetList from './components/dataset-list'
|
2025-03-18 11:01:06 +08:00
|
|
|
import MetadataFilter from './components/metadata/metadata-filter'
|
2024-04-08 18:51:46 +08:00
|
|
|
import type { KnowledgeRetrievalNodeType } from './types'
|
|
|
|
import Field from '@/app/components/workflow/nodes/_base/components/field'
|
|
|
|
import Split from '@/app/components/workflow/nodes/_base/components/split'
|
|
|
|
import OutputVars, { VarItem } from '@/app/components/workflow/nodes/_base/components/output-vars'
|
2025-06-24 09:10:30 +08:00
|
|
|
import type { NodePanelProps } from '@/app/components/workflow/types'
|
2024-04-08 18:51:46 +08:00
|
|
|
|
|
|
|
const i18nPrefix = 'workflow.nodes.knowledgeRetrieval'
|
|
|
|
|
|
|
|
const Panel: FC<NodePanelProps<KnowledgeRetrievalNodeType>> = ({
|
|
|
|
id,
|
|
|
|
data,
|
|
|
|
}) => {
|
|
|
|
const { t } = useTranslation()
|
|
|
|
|
|
|
|
const {
|
|
|
|
readOnly,
|
|
|
|
inputs,
|
|
|
|
handleQueryVarChange,
|
|
|
|
filterVar,
|
|
|
|
handleModelChanged,
|
|
|
|
handleCompletionParamsChange,
|
|
|
|
handleRetrievalModeChange,
|
|
|
|
handleMultipleRetrievalConfigChange,
|
|
|
|
selectedDatasets,
|
2025-03-18 11:01:06 +08:00
|
|
|
selectedDatasetsLoaded,
|
2024-04-08 18:51:46 +08:00
|
|
|
handleOnDatasetsChange,
|
2024-07-24 12:50:48 +08:00
|
|
|
rerankModelOpen,
|
|
|
|
setRerankModelOpen,
|
2025-03-18 11:01:06 +08:00
|
|
|
handleAddCondition,
|
|
|
|
handleMetadataFilterModeChange,
|
|
|
|
handleRemoveCondition,
|
|
|
|
handleToggleConditionLogicalOperator,
|
|
|
|
handleUpdateCondition,
|
|
|
|
handleMetadataModelChange,
|
|
|
|
handleMetadataCompletionParamsChange,
|
|
|
|
availableStringVars,
|
|
|
|
availableStringNodesWithParent,
|
|
|
|
availableNumberVars,
|
|
|
|
availableNumberNodesWithParent,
|
2024-04-08 18:51:46 +08:00
|
|
|
} = useConfig(id, data)
|
|
|
|
|
2024-07-24 12:50:48 +08:00
|
|
|
const handleOpenFromPropsChange = useCallback((openFromProps: boolean) => {
|
|
|
|
setRerankModelOpen(openFromProps)
|
|
|
|
}, [setRerankModelOpen])
|
|
|
|
|
2025-03-18 11:01:06 +08:00
|
|
|
const metadataList = useMemo(() => {
|
|
|
|
return intersectionBy(...selectedDatasets.filter((dataset) => {
|
|
|
|
return !!dataset.doc_metadata
|
|
|
|
}).map((dataset) => {
|
|
|
|
return dataset.doc_metadata!
|
|
|
|
}), 'name')
|
|
|
|
}, [selectedDatasets])
|
|
|
|
|
2024-04-08 18:51:46 +08:00
|
|
|
return (
|
2024-12-11 14:21:38 +08:00
|
|
|
<div className='pt-2'>
|
2025-03-21 17:41:03 +08:00
|
|
|
<div className='space-y-4 px-4 pb-2'>
|
2024-04-08 18:51:46 +08:00
|
|
|
{/* {JSON.stringify(inputs, null, 2)} */}
|
|
|
|
<Field
|
|
|
|
title={t(`${i18nPrefix}.queryVariable`)}
|
2025-04-27 16:12:02 +08:00
|
|
|
required
|
2024-04-08 18:51:46 +08:00
|
|
|
>
|
|
|
|
<VarReferencePicker
|
|
|
|
nodeId={id}
|
|
|
|
readonly={readOnly}
|
|
|
|
isShowNodeName
|
|
|
|
value={inputs.query_variable_selector}
|
|
|
|
onChange={handleQueryVarChange}
|
|
|
|
filterVar={filterVar}
|
|
|
|
/>
|
|
|
|
</Field>
|
|
|
|
|
|
|
|
<Field
|
|
|
|
title={t(`${i18nPrefix}.knowledge`)}
|
2025-04-27 16:12:02 +08:00
|
|
|
required
|
2024-04-08 18:51:46 +08:00
|
|
|
operations={
|
|
|
|
<div className='flex items-center space-x-1'>
|
|
|
|
<RetrievalConfig
|
|
|
|
payload={{
|
|
|
|
retrieval_mode: inputs.retrieval_mode,
|
|
|
|
multiple_retrieval_config: inputs.multiple_retrieval_config,
|
|
|
|
single_retrieval_config: inputs.single_retrieval_config,
|
|
|
|
}}
|
|
|
|
onRetrievalModeChange={handleRetrievalModeChange}
|
|
|
|
onMultipleRetrievalConfigChange={handleMultipleRetrievalConfigChange}
|
|
|
|
singleRetrievalModelConfig={inputs.single_retrieval_config?.model}
|
|
|
|
onSingleRetrievalModelChange={handleModelChanged as any}
|
|
|
|
onSingleRetrievalModelParamsChange={handleCompletionParamsChange}
|
2024-07-24 12:50:48 +08:00
|
|
|
readonly={readOnly || !selectedDatasets.length}
|
|
|
|
openFromProps={rerankModelOpen}
|
|
|
|
onOpenFromPropsChange={handleOpenFromPropsChange}
|
|
|
|
selectedDatasets={selectedDatasets}
|
2024-04-08 18:51:46 +08:00
|
|
|
/>
|
2025-04-10 17:15:48 +08:00
|
|
|
{!readOnly && (<div className='h-3 w-px bg-divider-regular'></div>)}
|
2024-04-08 18:51:46 +08:00
|
|
|
{!readOnly && (
|
|
|
|
<AddKnowledge
|
|
|
|
selectedIds={inputs.dataset_ids}
|
|
|
|
onChange={handleOnDatasetsChange}
|
|
|
|
/>
|
|
|
|
)}
|
|
|
|
</div>
|
|
|
|
}
|
|
|
|
>
|
|
|
|
<DatasetList
|
|
|
|
list={selectedDatasets}
|
|
|
|
onChange={handleOnDatasetsChange}
|
|
|
|
readonly={readOnly}
|
|
|
|
/>
|
|
|
|
</Field>
|
|
|
|
</div>
|
2025-03-18 11:01:06 +08:00
|
|
|
<div className='mb-2 py-2'>
|
|
|
|
<MetadataFilter
|
|
|
|
metadataList={metadataList}
|
|
|
|
selectedDatasetsLoaded={selectedDatasetsLoaded}
|
|
|
|
metadataFilterMode={inputs.metadata_filtering_mode}
|
|
|
|
metadataFilteringConditions={inputs.metadata_filtering_conditions}
|
|
|
|
handleAddCondition={handleAddCondition}
|
|
|
|
handleMetadataFilterModeChange={handleMetadataFilterModeChange}
|
|
|
|
handleRemoveCondition={handleRemoveCondition}
|
|
|
|
handleToggleConditionLogicalOperator={handleToggleConditionLogicalOperator}
|
|
|
|
handleUpdateCondition={handleUpdateCondition}
|
|
|
|
metadataModelConfig={inputs.metadata_model_config}
|
|
|
|
handleMetadataModelChange={handleMetadataModelChange}
|
|
|
|
handleMetadataCompletionParamsChange={handleMetadataCompletionParamsChange}
|
|
|
|
availableStringVars={availableStringVars}
|
|
|
|
availableStringNodesWithParent={availableStringNodesWithParent}
|
|
|
|
availableNumberVars={availableNumberVars}
|
|
|
|
availableNumberNodesWithParent={availableNumberNodesWithParent}
|
|
|
|
/>
|
|
|
|
</div>
|
2024-04-08 18:51:46 +08:00
|
|
|
<Split />
|
2024-12-11 14:21:38 +08:00
|
|
|
<div>
|
2024-04-08 18:51:46 +08:00
|
|
|
<OutputVars>
|
|
|
|
<>
|
|
|
|
<VarItem
|
|
|
|
name='result'
|
|
|
|
type='Array[Object]'
|
|
|
|
description={t(`${i18nPrefix}.outputVars.output`)}
|
|
|
|
subItems={[
|
|
|
|
{
|
|
|
|
name: 'content',
|
|
|
|
type: 'string',
|
|
|
|
description: t(`${i18nPrefix}.outputVars.content`),
|
|
|
|
},
|
|
|
|
// url, title, link like bing search reference result: link, link page title, link page icon
|
|
|
|
{
|
|
|
|
name: 'title',
|
|
|
|
type: 'string',
|
|
|
|
description: t(`${i18nPrefix}.outputVars.title`),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: 'url',
|
|
|
|
type: 'string',
|
|
|
|
description: t(`${i18nPrefix}.outputVars.url`),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: 'icon',
|
|
|
|
type: 'string',
|
|
|
|
description: t(`${i18nPrefix}.outputVars.icon`),
|
|
|
|
},
|
|
|
|
{
|
|
|
|
name: 'metadata',
|
|
|
|
type: 'object',
|
|
|
|
description: t(`${i18nPrefix}.outputVars.metadata`),
|
|
|
|
},
|
|
|
|
]}
|
|
|
|
/>
|
|
|
|
|
|
|
|
</>
|
|
|
|
</OutputVars>
|
|
|
|
</div>
|
|
|
|
</div>
|
|
|
|
)
|
|
|
|
}
|
|
|
|
|
2024-07-24 12:50:48 +08:00
|
|
|
export default memo(Panel)
|