mirror of
https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git
synced 2025-08-20 12:29:17 +08:00
feat: limit config
This commit is contained in:
parent
944fea4cc9
commit
e9ce9c1f47
@ -0,0 +1,102 @@
|
|||||||
|
'use client'
|
||||||
|
import type { FC } from 'react'
|
||||||
|
import React, { useCallback } from 'react'
|
||||||
|
import { useTranslation } from 'react-i18next'
|
||||||
|
import type { Limit } from '../types'
|
||||||
|
import cn from '@/utils/classnames'
|
||||||
|
import Field from '@/app/components/workflow/nodes/_base/components/field'
|
||||||
|
import Switch from '@/app/components/base/switch'
|
||||||
|
import Slider from '@/app/components/base/slider'
|
||||||
|
|
||||||
|
const i18nPrefix = 'workflow.nodes.listFilter'
|
||||||
|
const LIMIT_SIZE_MIN = 1
|
||||||
|
const LIMIT_SIZE_MAX = 20
|
||||||
|
const LIMIT_SIZE_DEFAULT = 10
|
||||||
|
|
||||||
|
type Props = {
|
||||||
|
className?: string
|
||||||
|
readonly: boolean
|
||||||
|
config: Limit
|
||||||
|
onChange: (limit: Limit) => void
|
||||||
|
canSetRoleName?: boolean
|
||||||
|
}
|
||||||
|
|
||||||
|
const LIMIT_DEFAULT: Limit = {
|
||||||
|
enabled: false,
|
||||||
|
size: LIMIT_SIZE_DEFAULT,
|
||||||
|
}
|
||||||
|
|
||||||
|
const LimitConfig: FC<Props> = ({
|
||||||
|
className,
|
||||||
|
readonly,
|
||||||
|
config = LIMIT_DEFAULT,
|
||||||
|
onChange,
|
||||||
|
}) => {
|
||||||
|
const { t } = useTranslation()
|
||||||
|
const payload = config
|
||||||
|
|
||||||
|
const handleLimitEnabledChange = useCallback((enabled: boolean) => {
|
||||||
|
onChange({
|
||||||
|
...config,
|
||||||
|
enabled,
|
||||||
|
})
|
||||||
|
}, [config, onChange])
|
||||||
|
|
||||||
|
const handleLimitSizeChange = useCallback((size: number | string) => {
|
||||||
|
onChange({
|
||||||
|
...config,
|
||||||
|
size: parseInt(size as string),
|
||||||
|
})
|
||||||
|
}, [onChange, config])
|
||||||
|
|
||||||
|
const handleBlur = useCallback(() => {
|
||||||
|
const payload = config
|
||||||
|
if (!payload)
|
||||||
|
return
|
||||||
|
|
||||||
|
if (payload.size === undefined || payload.size === null)
|
||||||
|
handleLimitSizeChange(LIMIT_SIZE_DEFAULT)
|
||||||
|
}, [handleLimitSizeChange, config])
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className={cn(className)}>
|
||||||
|
<Field
|
||||||
|
title={t(`${i18nPrefix}.limit`)}
|
||||||
|
operations={
|
||||||
|
<Switch
|
||||||
|
defaultValue={payload.enabled}
|
||||||
|
onChange={handleLimitEnabledChange}
|
||||||
|
size='md'
|
||||||
|
disabled={readonly}
|
||||||
|
/>
|
||||||
|
}
|
||||||
|
>
|
||||||
|
{payload && (
|
||||||
|
<div className='flex justify-between items-center h-8 space-x-2'>
|
||||||
|
<input
|
||||||
|
value={(payload?.size || LIMIT_SIZE_DEFAULT) as number}
|
||||||
|
className='shrink-0 block pl-3 w-12 h-8 appearance-none outline-none rounded-lg bg-gray-100 text-[13px] text-gra-900'
|
||||||
|
type='number'
|
||||||
|
min={LIMIT_SIZE_MIN}
|
||||||
|
max={LIMIT_SIZE_MAX}
|
||||||
|
step={1}
|
||||||
|
onChange={e => handleLimitSizeChange(e.target.value)}
|
||||||
|
onBlur={handleBlur}
|
||||||
|
disabled={readonly || !payload?.enabled}
|
||||||
|
/>
|
||||||
|
<Slider
|
||||||
|
className='grow'
|
||||||
|
value={(payload?.size || LIMIT_SIZE_DEFAULT) as number}
|
||||||
|
min={LIMIT_SIZE_MIN}
|
||||||
|
max={LIMIT_SIZE_MAX}
|
||||||
|
step={1}
|
||||||
|
onChange={handleLimitSizeChange}
|
||||||
|
disabled={readonly || !payload?.enabled}
|
||||||
|
/>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</Field>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
export default React.memo(LimitConfig)
|
@ -5,6 +5,7 @@ import VarReferencePicker from '../_base/components/variable/var-reference-picke
|
|||||||
import OutputVars, { VarItem } from '../_base/components/output-vars'
|
import OutputVars, { VarItem } from '../_base/components/output-vars'
|
||||||
import useConfig from './use-config'
|
import useConfig from './use-config'
|
||||||
import type { ListFilterNodeType } from './types'
|
import type { ListFilterNodeType } from './types'
|
||||||
|
import LimitConfig from './components/limit-config'
|
||||||
import Field from '@/app/components/workflow/nodes/_base/components/field'
|
import Field from '@/app/components/workflow/nodes/_base/components/field'
|
||||||
import { type NodePanelProps } from '@/app/components/workflow/types'
|
import { type NodePanelProps } from '@/app/components/workflow/types'
|
||||||
|
|
||||||
@ -21,6 +22,7 @@ const Panel: FC<NodePanelProps<ListFilterNodeType>> = ({
|
|||||||
inputs,
|
inputs,
|
||||||
handleVarChanges,
|
handleVarChanges,
|
||||||
filterVar,
|
filterVar,
|
||||||
|
handleLimitChange,
|
||||||
} = useConfig(id, data)
|
} = useConfig(id, data)
|
||||||
|
|
||||||
return (
|
return (
|
||||||
@ -38,6 +40,12 @@ const Panel: FC<NodePanelProps<ListFilterNodeType>> = ({
|
|||||||
filterVar={filterVar}
|
filterVar={filterVar}
|
||||||
/>
|
/>
|
||||||
</Field>
|
</Field>
|
||||||
|
|
||||||
|
<LimitConfig
|
||||||
|
config={inputs.limit}
|
||||||
|
onChange={handleLimitChange}
|
||||||
|
readonly={readOnly}
|
||||||
|
/>
|
||||||
</div>
|
</div>
|
||||||
<div className='px-4 pt-4 pb-2'>
|
<div className='px-4 pt-4 pb-2'>
|
||||||
<OutputVars>
|
<OutputVars>
|
||||||
|
@ -5,6 +5,11 @@ export enum OrderBy {
|
|||||||
DESC = 'desc',
|
DESC = 'desc',
|
||||||
}
|
}
|
||||||
|
|
||||||
|
export type Limit = {
|
||||||
|
enabled: boolean
|
||||||
|
size?: number
|
||||||
|
}
|
||||||
|
|
||||||
export type ListFilterNodeType = CommonNodeType & {
|
export type ListFilterNodeType = CommonNodeType & {
|
||||||
variable: ValueSelector
|
variable: ValueSelector
|
||||||
filterBy: []
|
filterBy: []
|
||||||
@ -13,8 +18,5 @@ export type ListFilterNodeType = CommonNodeType & {
|
|||||||
key: ValueSelector | string
|
key: ValueSelector | string
|
||||||
value: OrderBy
|
value: OrderBy
|
||||||
}
|
}
|
||||||
limit: {
|
limit: Limit
|
||||||
enabled: boolean
|
|
||||||
value: number
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
@ -2,7 +2,7 @@ import { useCallback } from 'react'
|
|||||||
import produce from 'immer'
|
import produce from 'immer'
|
||||||
import type { ValueSelector, Var } from '../../types'
|
import type { ValueSelector, Var } from '../../types'
|
||||||
import { VarType } from '../../types'
|
import { VarType } from '../../types'
|
||||||
import { type ListFilterNodeType } from './types'
|
import type { Limit, type ListFilterNodeType } from './types'
|
||||||
import useNodeCrud from '@/app/components/workflow/nodes/_base/hooks/use-node-crud'
|
import useNodeCrud from '@/app/components/workflow/nodes/_base/hooks/use-node-crud'
|
||||||
import {
|
import {
|
||||||
useNodesReadOnly,
|
useNodesReadOnly,
|
||||||
@ -21,14 +21,22 @@ const useConfig = (id: string, payload: ListFilterNodeType) => {
|
|||||||
}, [inputs, setInputs])
|
}, [inputs, setInputs])
|
||||||
|
|
||||||
const filterVar = useCallback((varPayload: Var) => {
|
const filterVar = useCallback((varPayload: Var) => {
|
||||||
return varPayload.type !== VarType.file
|
return [VarType.arrayNumber, VarType.arrayString, VarType.arrayFile, VarType.arrayObject].includes(varPayload.type)
|
||||||
}, [])
|
}, [])
|
||||||
|
|
||||||
|
const handleLimitChange = useCallback((limit: Limit) => {
|
||||||
|
const newInputs = produce(inputs, (draft) => {
|
||||||
|
draft.limit = limit
|
||||||
|
})
|
||||||
|
setInputs(newInputs)
|
||||||
|
}, [inputs, setInputs])
|
||||||
|
|
||||||
return {
|
return {
|
||||||
readOnly,
|
readOnly,
|
||||||
inputs,
|
inputs,
|
||||||
filterVar,
|
filterVar,
|
||||||
handleVarChanges,
|
handleVarChanges,
|
||||||
|
handleLimitChange,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -501,6 +501,7 @@ const translation = {
|
|||||||
},
|
},
|
||||||
listFilter: {
|
listFilter: {
|
||||||
inputVar: 'Input Variable',
|
inputVar: 'Input Variable',
|
||||||
|
limit: 'Limit',
|
||||||
outputVars: {
|
outputVars: {
|
||||||
result: 'Filter result',
|
result: 'Filter result',
|
||||||
first_record: 'First record',
|
first_record: 'First record',
|
||||||
|
@ -501,6 +501,7 @@ const translation = {
|
|||||||
},
|
},
|
||||||
listFilter: {
|
listFilter: {
|
||||||
inputVar: '输入变量',
|
inputVar: '输入变量',
|
||||||
|
limit: '限制',
|
||||||
outputVars: {
|
outputVars: {
|
||||||
result: '过滤结果',
|
result: '过滤结果',
|
||||||
first_record: '第一条记录',
|
first_record: '第一条记录',
|
||||||
|
Loading…
x
Reference in New Issue
Block a user