Nelze vybrat více než 25 témat Téma musí začínat písmenem nebo číslem, může obsahovat pomlčky („-“) a může být dlouhé až 35 znaků.

panel.tsx 4.0KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125
  1. import type { FC } from 'react'
  2. import { memo } from 'react'
  3. import type { NodePanelProps } from '../../types'
  4. import { AgentFeature, type AgentNodeType } from './types'
  5. import Field from '../_base/components/field'
  6. import { AgentStrategy } from '../_base/components/agent-strategy'
  7. import useConfig from './use-config'
  8. import { useTranslation } from 'react-i18next'
  9. import OutputVars, { VarItem } from '../_base/components/output-vars'
  10. import type { StrategyParamItem } from '@/app/components/plugins/types'
  11. import type { CredentialFormSchema } from '@/app/components/header/account-setting/model-provider-page/declarations'
  12. import { toType } from '@/app/components/tools/utils/to-form-schema'
  13. import { useStore } from '../../store'
  14. import Split from '../_base/components/split'
  15. import MemoryConfig from '../_base/components/memory-config'
  16. const i18nPrefix = 'workflow.nodes.agent'
  17. export function strategyParamToCredientialForm(param: StrategyParamItem): CredentialFormSchema {
  18. return {
  19. ...param as any,
  20. variable: param.name,
  21. show_on: [],
  22. type: toType(param.type),
  23. tooltip: param.help,
  24. }
  25. }
  26. const AgentPanel: FC<NodePanelProps<AgentNodeType>> = (props) => {
  27. const {
  28. inputs,
  29. setInputs,
  30. currentStrategy,
  31. formData,
  32. onFormChange,
  33. isChatMode,
  34. availableNodesWithParent,
  35. availableVars,
  36. readOnly,
  37. outputSchema,
  38. handleMemoryChange,
  39. } = useConfig(props.id, props.data)
  40. const { t } = useTranslation()
  41. const resetEditor = useStore(s => s.setControlPromptEditorRerenderKey)
  42. return <div className='my-2'>
  43. <Field
  44. required
  45. title={t('workflow.nodes.agent.strategy.label')}
  46. className='px-4 py-2'
  47. tooltip={t('workflow.nodes.agent.strategy.tooltip')} >
  48. <AgentStrategy
  49. strategy={inputs.agent_strategy_name ? {
  50. agent_strategy_provider_name: inputs.agent_strategy_provider_name!,
  51. agent_strategy_name: inputs.agent_strategy_name!,
  52. agent_strategy_label: inputs.agent_strategy_label!,
  53. agent_output_schema: inputs.output_schema,
  54. plugin_unique_identifier: inputs.plugin_unique_identifier!,
  55. } : undefined}
  56. onStrategyChange={(strategy) => {
  57. setInputs({
  58. ...inputs,
  59. agent_strategy_provider_name: strategy?.agent_strategy_provider_name,
  60. agent_strategy_name: strategy?.agent_strategy_name,
  61. agent_strategy_label: strategy?.agent_strategy_label,
  62. output_schema: strategy!.agent_output_schema,
  63. plugin_unique_identifier: strategy!.plugin_unique_identifier,
  64. })
  65. resetEditor(Date.now())
  66. }}
  67. formSchema={currentStrategy?.parameters?.map(strategyParamToCredientialForm) || []}
  68. formValue={formData}
  69. onFormValueChange={onFormChange}
  70. nodeOutputVars={availableVars}
  71. availableNodes={availableNodesWithParent}
  72. nodeId={props.id}
  73. />
  74. </Field>
  75. <div className='px-4 py-2'>
  76. {isChatMode && currentStrategy?.features?.includes(AgentFeature.HISTORY_MESSAGES) && (
  77. <>
  78. <Split />
  79. <MemoryConfig
  80. className='mt-4'
  81. readonly={readOnly}
  82. config={{ data: inputs.memory }}
  83. onChange={handleMemoryChange}
  84. canSetRoleName={false}
  85. />
  86. </>
  87. )}
  88. </div>
  89. <div>
  90. <OutputVars>
  91. <VarItem
  92. name='text'
  93. type='String'
  94. description={t(`${i18nPrefix}.outputVars.text`)}
  95. />
  96. <VarItem
  97. name='files'
  98. type='Array[File]'
  99. description={t(`${i18nPrefix}.outputVars.files.title`)}
  100. />
  101. <VarItem
  102. name='json'
  103. type='Array[Object]'
  104. description={t(`${i18nPrefix}.outputVars.json`)}
  105. />
  106. {outputSchema.map(({ name, type, description }) => (
  107. <VarItem
  108. key={name}
  109. name={name}
  110. type={type}
  111. description={description}
  112. />
  113. ))}
  114. </OutputVars>
  115. </div>
  116. </div>
  117. }
  118. AgentPanel.displayName = 'AgentPanel'
  119. export default memo(AgentPanel)