diff --git a/src/features/AgentSetting/AgentMeta/AutoGenerateInput.tsx b/src/features/AgentSetting/AgentMeta/AutoGenerateInput.tsx index cebdde8c6c6..581f43e3ede 100644 --- a/src/features/AgentSetting/AgentMeta/AutoGenerateInput.tsx +++ b/src/features/AgentSetting/AgentMeta/AutoGenerateInput.tsx @@ -5,37 +5,42 @@ import { Wand2 } from 'lucide-react'; import { memo } from 'react'; import { useTranslation } from 'react-i18next'; + export interface AutoGenerateInputProps extends InputProps { + canAutoGenerate?: boolean; loading?: boolean; onGenerate?: () => void; } -const AutoGenerateInput = memo(({ loading, onGenerate, ...props }) => { - const { t } = useTranslation('common'); - const theme = useTheme(); +const AutoGenerateInput = memo( + ({ loading, onGenerate, canAutoGenerate, ...props }) => { + const { t } = useTranslation('common'); + const theme = useTheme(); - return ( - - ) - } - type={'block'} - {...props} - /> - ); -}); + return ( + + ) + } + type="block" + {...props} + /> + ); + }, +); export default AutoGenerateInput; diff --git a/src/features/AgentSetting/AgentMeta/AutoGenerateSelect.tsx b/src/features/AgentSetting/AgentMeta/AutoGenerateSelect.tsx index 980ffe9baff..a31ed496066 100644 --- a/src/features/AgentSetting/AgentMeta/AutoGenerateSelect.tsx +++ b/src/features/AgentSetting/AgentMeta/AutoGenerateSelect.tsx @@ -7,12 +7,13 @@ import { memo } from 'react'; import { useTranslation } from 'react-i18next'; export interface AutoGenerateInputProps extends SelectProps { + canAutoGenerate?: boolean; loading?: boolean; onGenerate?: () => void; } const AutoGenerateSelect = memo( - ({ loading, onGenerate, value, ...props }) => { + ({ loading, onGenerate, value, canAutoGenerate, ...props }) => { const { t } = useTranslation('common'); const theme = useTheme(); @@ -25,6 +26,7 @@ const AutoGenerateSelect = memo( onGenerate && ( { return { children: ( {