| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128 |
- import type { FC } from 'react'
- import { memo } from 'react'
- import type { NodePanelProps } from '../../types'
- import { AgentFeature, type AgentNodeType } from './types'
- import Field from '../_base/components/field'
- import { AgentStrategy } from '../_base/components/agent-strategy'
- import useConfig from './use-config'
- import { useTranslation } from 'react-i18next'
- import OutputVars, { VarItem } from '../_base/components/output-vars'
- import type { StrategyParamItem } from '@/app/components/plugins/types'
- import type { CredentialFormSchema } from '@/app/components/header/account-setting/model-provider-page/declarations'
- import { toType } from '@/app/components/tools/utils/to-form-schema'
- import { useStore } from '../../store'
- import Split from '../_base/components/split'
- import MemoryConfig from '../_base/components/memory-config'
- const i18nPrefix = 'workflow.nodes.agent'
-
- export function strategyParamToCredientialForm(param: StrategyParamItem): CredentialFormSchema {
- return {
- ...param as any,
- variable: param.name,
- show_on: [],
- type: toType(param.type),
- tooltip: param.help,
- }
- }
-
- const AgentPanel: FC<NodePanelProps<AgentNodeType>> = (props) => {
- const {
- inputs,
- setInputs,
- currentStrategy,
- formData,
- onFormChange,
- isChatMode,
- availableNodesWithParent,
- availableVars,
- readOnly,
- outputSchema,
- handleMemoryChange,
- canChooseMCPTool,
- } = useConfig(props.id, props.data)
- const { t } = useTranslation()
-
- const resetEditor = useStore(s => s.setControlPromptEditorRerenderKey)
- return <div className='my-2'>
- <Field
- required
- title={t('workflow.nodes.agent.strategy.label')}
- className='px-4 py-2'
- tooltip={t('workflow.nodes.agent.strategy.tooltip')} >
- <AgentStrategy
- strategy={inputs.agent_strategy_name ? {
- agent_strategy_provider_name: inputs.agent_strategy_provider_name!,
- agent_strategy_name: inputs.agent_strategy_name!,
- agent_strategy_label: inputs.agent_strategy_label!,
- agent_output_schema: inputs.output_schema,
- plugin_unique_identifier: inputs.plugin_unique_identifier!,
- meta: inputs.meta,
- } : undefined}
- onStrategyChange={(strategy) => {
- setInputs({
- ...inputs,
- agent_strategy_provider_name: strategy?.agent_strategy_provider_name,
- agent_strategy_name: strategy?.agent_strategy_name,
- agent_strategy_label: strategy?.agent_strategy_label,
- output_schema: strategy!.agent_output_schema,
- plugin_unique_identifier: strategy!.plugin_unique_identifier,
- meta: strategy?.meta,
- })
- resetEditor(Date.now())
- }}
- formSchema={currentStrategy?.parameters?.map(strategyParamToCredientialForm) || []}
- formValue={formData}
- onFormValueChange={onFormChange}
- nodeOutputVars={availableVars}
- availableNodes={availableNodesWithParent}
- nodeId={props.id}
- canChooseMCPTool={canChooseMCPTool}
- />
- </Field>
- <div className='px-4 py-2'>
- {isChatMode && currentStrategy?.features?.includes(AgentFeature.HISTORY_MESSAGES) && (
- <>
- <Split />
- <MemoryConfig
- className='mt-4'
- readonly={readOnly}
- config={{ data: inputs.memory }}
- onChange={handleMemoryChange}
- canSetRoleName={false}
- />
- </>
- )}
- </div>
- <div>
- <OutputVars>
- <VarItem
- name='text'
- type='String'
- description={t(`${i18nPrefix}.outputVars.text`)}
- />
- <VarItem
- name='files'
- type='Array[File]'
- description={t(`${i18nPrefix}.outputVars.files.title`)}
- />
- <VarItem
- name='json'
- type='Array[Object]'
- description={t(`${i18nPrefix}.outputVars.json`)}
- />
- {outputSchema.map(({ name, type, description }) => (
- <VarItem
- key={name}
- name={name}
- type={type}
- description={description}
- />
- ))}
- </OutputVars>
- </div>
- </div>
- }
-
- AgentPanel.displayName = 'AgentPanel'
-
- export default memo(AgentPanel)
|