62 lines
2.0 KiB
TypeScript
62 lines
2.0 KiB
TypeScript
|
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 useNodeCrud from '../_base/hooks/use-node-crud'
|
||
|
|
import { AgentStrategy } from '../_base/components/agent-strategy'
|
||
|
|
|
||
|
|
const AgentPanel: FC<NodePanelProps<AgentNodeType>> = (props) => {
|
||
|
|
const { inputs, setInputs } = useNodeCrud(props.id, props.data)
|
||
|
|
const [iter, setIter] = [inputs.max_iterations, (value: number) => {
|
||
|
|
setInputs({
|
||
|
|
...inputs,
|
||
|
|
max_iterations: value,
|
||
|
|
})
|
||
|
|
}]
|
||
|
|
return <>
|
||
|
|
<Field title={'Strategy'} className='px-4' >
|
||
|
|
<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,
|
||
|
|
})
|
||
|
|
}}
|
||
|
|
formSchema={[]}
|
||
|
|
formValue={{}}
|
||
|
|
onFormValueChange={console.error}
|
||
|
|
/>
|
||
|
|
</Field>
|
||
|
|
<Field title={'tools'} className='px-4'>
|
||
|
|
|
||
|
|
</Field>
|
||
|
|
<Field title={'max iterations'} tooltip={'max iter'} inline className='px-4'>
|
||
|
|
<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>
|
||
|
|
</>
|
||
|
|
}
|
||
|
|
|
||
|
|
export default AgentPanel
|