12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788 |
- import type { FC } from 'react'
- import React from 'react'
- import useSWR from 'swr'
- import { useTranslation } from 'react-i18next'
- import { useContext } from 'use-context-selector'
- import VarReferencePicker from '../_base/components/variable/var-reference-picker'
- import OutputVars, { VarItem } from '../_base/components/output-vars'
- import Split from '../_base/components/split'
- import { useNodeHelpLink } from '../_base/hooks/use-node-help-link'
- import useConfig from './use-config'
- import type { DocExtractorNodeType } from './types'
- import { fetchSupportFileTypes } from '@/service/datasets'
- import Field from '@/app/components/workflow/nodes/_base/components/field'
- import { BlockEnum, type NodePanelProps } from '@/app/components/workflow/types'
- import I18n from '@/context/i18n'
- import { LanguagesSupported } from '@/i18n/language'
- const i18nPrefix = 'workflow.nodes.docExtractor'
- const Panel: FC<NodePanelProps<DocExtractorNodeType>> = ({
- id,
- data,
- }) => {
- const { t } = useTranslation()
- const { locale } = useContext(I18n)
- const link = useNodeHelpLink(BlockEnum.DocExtractor)
- const { data: supportFileTypesResponse } = useSWR({ url: '/files/support-type' }, fetchSupportFileTypes)
- const supportTypes = supportFileTypesResponse?.allowed_extensions || []
- const supportTypesShowNames = (() => {
- const extensionMap: { [key: string]: string } = {
- md: 'markdown',
- pptx: 'pptx',
- htm: 'html',
- xlsx: 'xlsx',
- docx: 'docx',
- }
- return [...supportTypes]
- .map(item => extensionMap[item] || item) // map to standardized extension
- .map(item => item.toLowerCase()) // convert to lower case
- .filter((item, index, self) => self.indexOf(item) === index) // remove duplicates
- .join(locale !== LanguagesSupported[1] ? ', ' : '、 ')
- })()
- const {
- readOnly,
- inputs,
- handleVarChanges,
- filterVar,
- } = useConfig(id, data)
- return (
- <div className='mt-2'>
- <div className='px-4 pb-4 space-y-4'>
- <Field
- title={t(`${i18nPrefix}.inputVar`)}
- >
- <>
- <VarReferencePicker
- readonly={readOnly}
- nodeId={id}
- isShowNodeName
- value={inputs.variable_selector || []}
- onChange={handleVarChanges}
- filterVar={filterVar}
- typePlaceHolder='File | Array[File]'
- />
- <div className='mt-1 py-0.5 text-text-tertiary body-xs-regular'>
- {t(`${i18nPrefix}.supportFileTypes`, { types: supportTypesShowNames })}
- <a className='text-text-accent' href={link} target='_blank'>{t(`${i18nPrefix}.learnMore`)}</a>
- </div>
- </>
- </Field>
- </div>
- <Split />
- <div className='px-4 pt-4 pb-2'>
- <OutputVars>
- <VarItem
- name='text'
- type={inputs.is_array_file ? 'array[string]' : 'string'}
- description={t(`${i18nPrefix}.outputVars.text`)}
- />
- </OutputVars>
- </div>
- </div>
- )
- }
- export default React.memo(Panel)
|