2024-12-24 14:20:15 +08:00
|
|
|
import type { FC } from 'react'
|
|
|
|
import type { NodePanelProps } from '../../types'
|
|
|
|
import type { AgentNodeType } from './types'
|
|
|
|
import Field from '../_base/components/field'
|
|
|
|
import { InputNumber } from '@/app/components/base/input-number'
|
|
|
|
import Slider from '@/app/components/base/slider'
|
|
|
|
import { AgentStrategy } from '../_base/components/agent-strategy'
|
2024-12-25 14:52:11 +08:00
|
|
|
import useConfig from './use-config'
|
|
|
|
import { useTranslation } from 'react-i18next'
|
2024-12-25 15:06:58 +08:00
|
|
|
import { type CredentialFormSchema, FormTypeEnum } from '@/app/components/header/account-setting/model-provider-page/declarations'
|
2024-12-24 14:20:15 +08:00
|
|
|
|
2024-12-25 15:06:58 +08:00
|
|
|
// @ts-expect-error fuck
|
2024-12-25 11:24:24 +08:00
|
|
|
const mockSchema = [
|
|
|
|
{
|
|
|
|
name: 'format',
|
|
|
|
label: {
|
|
|
|
en_US: 'Format',
|
|
|
|
zh_Hans: '格式',
|
|
|
|
pt_BR: 'Format',
|
|
|
|
ja_JP: 'Format',
|
|
|
|
},
|
2024-12-25 15:06:58 +08:00
|
|
|
placeholder: undefined,
|
|
|
|
scope: undefined,
|
2024-12-25 11:24:24 +08:00
|
|
|
required: false,
|
|
|
|
default: '%Y-%m-%d %H:%M:%S',
|
|
|
|
options: [],
|
|
|
|
type: 'text-input',
|
|
|
|
form: 'form',
|
|
|
|
llm_description: null,
|
|
|
|
variable: 'format',
|
|
|
|
_type: 'string',
|
|
|
|
show_on: [],
|
|
|
|
tooltip: {
|
|
|
|
en_US: 'Time format in strftime standard.',
|
|
|
|
zh_Hans: 'strftime 标准的时间格式。',
|
|
|
|
},
|
|
|
|
},
|
|
|
|
{
|
2024-12-25 15:06:58 +08:00
|
|
|
name: 'model',
|
|
|
|
type: FormTypeEnum.modelSelector,
|
2024-12-25 11:24:24 +08:00
|
|
|
label: {
|
2024-12-25 15:06:58 +08:00
|
|
|
en_US: 'Model',
|
|
|
|
zh_Hans: '模型',
|
2024-12-25 11:24:24 +08:00
|
|
|
},
|
2024-12-25 15:12:33 +08:00
|
|
|
variable: 'model',
|
2024-12-25 15:06:58 +08:00
|
|
|
scope: 'all',
|
2024-12-25 11:24:24 +08:00
|
|
|
},
|
2024-12-25 15:06:58 +08:00
|
|
|
] as Array<CredentialFormSchema & { name: string }>
|
2024-12-25 11:24:24 +08:00
|
|
|
|
2024-12-24 14:20:15 +08:00
|
|
|
const AgentPanel: FC<NodePanelProps<AgentNodeType>> = (props) => {
|
2024-12-25 14:52:11 +08:00
|
|
|
const { inputs, setInputs } = useConfig(props.id, props.data)
|
|
|
|
const { t } = useTranslation()
|
2024-12-24 14:20:15 +08:00
|
|
|
const [iter, setIter] = [inputs.max_iterations, (value: number) => {
|
|
|
|
setInputs({
|
|
|
|
...inputs,
|
|
|
|
max_iterations: value,
|
|
|
|
})
|
|
|
|
}]
|
2024-12-25 11:24:24 +08:00
|
|
|
return <div className='space-y-2 my-2'>
|
2024-12-25 14:52:11 +08:00
|
|
|
<Field title={t('workflow.nodes.agent.strategy.label')} className='px-4' >
|
2024-12-24 14:20:15 +08:00
|
|
|
<AgentStrategy
|
|
|
|
strategy={inputs.agent_strategy_name ? {
|
|
|
|
agent_strategy_provider_name: inputs.agent_strategy_provider_name!,
|
|
|
|
agent_strategy_name: inputs.agent_strategy_name!,
|
|
|
|
agent_parameters: inputs.agent_parameters,
|
|
|
|
} : undefined}
|
|
|
|
onStrategyChange={(strategy) => {
|
|
|
|
setInputs({
|
|
|
|
...inputs,
|
|
|
|
agent_strategy_provider_name: strategy?.agent_strategy_provider_name,
|
|
|
|
agent_strategy_name: strategy?.agent_strategy_name,
|
|
|
|
agent_parameters: strategy?.agent_parameters,
|
|
|
|
})
|
|
|
|
}}
|
2024-12-25 11:24:24 +08:00
|
|
|
formSchema={mockSchema as any}
|
|
|
|
formValue={inputs.agent_parameters || {}}
|
|
|
|
onFormValueChange={value => setInputs({
|
|
|
|
...inputs,
|
|
|
|
agent_parameters: value,
|
|
|
|
})}
|
2024-12-24 14:20:15 +08:00
|
|
|
/>
|
|
|
|
</Field>
|
2024-12-25 14:52:11 +08:00
|
|
|
<Field title={t('workflow.nodes.agent.tools')} className='px-4'>
|
2024-12-24 14:20:15 +08:00
|
|
|
|
|
|
|
</Field>
|
2024-12-25 14:52:11 +08:00
|
|
|
<Field title={t('workflow.nodes.agent.maxIterations')} tooltip={'max iter'} inline className='px-4'>
|
2024-12-24 14:20:15 +08:00
|
|
|
<div className='flex w-[200px] items-center gap-3'>
|
|
|
|
<Slider value={iter} onChange={setIter} className='w-full' min={1} max={10} />
|
|
|
|
<InputNumber
|
|
|
|
value={iter}
|
|
|
|
// TODO: maybe empty, handle this
|
|
|
|
onChange={setIter as any}
|
|
|
|
defaultValue={3}
|
|
|
|
size='sm'
|
|
|
|
min={1}
|
|
|
|
max={10}
|
|
|
|
className='w-12'
|
|
|
|
placeholder=''
|
|
|
|
/>
|
|
|
|
</div>
|
|
|
|
</Field>
|
2024-12-25 11:24:24 +08:00
|
|
|
</div>
|
2024-12-24 14:20:15 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
export default AgentPanel
|