🐛 fix: disabled autogenerate field icon when empty system role (#2076)

* fix: disabled autogenerate field icon when empty system role * 🐛 fix: update systemrole guard * 🐛 fix: revert action.ts file change * 🐛 fix: added space
lobehub · May 4, 2024 · 27095f5 · 27095f5
1 parent d6eeac0
commit 27095f5
Show file tree

Hide file tree

Showing 3 changed files with 36 additions and 27 deletions.
diff --git 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<AutoGenerateInputProps>(({ loading, onGenerate, ...props }) => {
-  const { t } = useTranslation('common');
-  const theme = useTheme();
+const AutoGenerateInput = memo<AutoGenerateInputProps>(
+  ({ loading, onGenerate, canAutoGenerate, ...props }) => {
+    const { t } = useTranslation('common');
+    const theme = useTheme();
 
-  return (
-    <Input
-      suffix={
-        onGenerate && (
-          <ActionIcon
-            active
-            icon={Wand2}
-            loading={loading}
-            onClick={onGenerate}
-            size={'small'}
-            style={{
-              color: theme.colorInfo,
-              marginRight: -4,
-            }}
-            title={t('autoGenerate')}
-          />
-        )
-      }
-      type={'block'}
-      {...props}
-    />
-  );
-});
+    return (
+      <Input
+        suffix={
+          onGenerate && (
+            <ActionIcon
+              active
+              disable={!canAutoGenerate}
+              icon={Wand2}
+              loading={loading}
+              onClick={onGenerate}
+              size="small"
+              style={{
+                color: theme.colorInfo,
+                marginRight: -4,
+              }}
+              title={t('autoGenerate')}
+            />
+          )
+        }
+        type="block"
+        {...props}
+      />
+    );
+  },
+);
 
 export default AutoGenerateInput;
diff --git 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<AutoGenerateInputProps>(
-  ({ loading, onGenerate, value, ...props }) => {
+  ({ loading, onGenerate, value, canAutoGenerate, ...props }) => {
     const { t } = useTranslation('common');
     const theme = useTheme();
 
@@ -25,6 +26,7 @@ const AutoGenerateSelect = memo<AutoGenerateInputProps>(
           onGenerate && (
             <ActionIcon
               active
+              disable={!canAutoGenerate}
               icon={Wand2}
               loading={loading}
               onClick={onGenerate}

diff --git a/src/features/AgentSetting/AgentMeta/index.tsx b/src/features/AgentSetting/AgentMeta/index.tsx
@@ -63,6 +63,8 @@ const AgentMeta = memo(() => {
     return {
       children: (
         <AutoGenerate
+          canAutoGenerate={hasSystemRole}
+          disabled={!hasSystemRole}
           loading={loading[item.key as keyof SessionLoadingState]}
           onChange={item.onChange}
           onGenerate={() => {