refactor: merge category result after parsed

msgbyte · Feb 11, 2025 · 8d6ab5c · 8d6ab5c
1 parent a63240a
commit 8d6ab5c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/client/components/survey/SurveyAIBtn.tsx b/src/client/components/survey/SurveyAIBtn.tsx
@@ -28,6 +28,7 @@ import {
 import { Checkbox } from '../ui/checkbox';
 import { useWatch } from '@/hooks/useWatch';
 import { Alert, AlertDescription, AlertTitle } from '../ui/alert';
+import { uniq } from 'lodash-es';
 
 interface SurveyAIBtnProps {
   surveyId: string;
@@ -92,7 +93,7 @@ export const SurveyAIBtn: React.FC<SurveyAIBtnProps> = React.memo((props) => {
           suggestionCategory: category,
         });
 
-      setCategory(categorys);
+      setCategory((state) => uniq([...state, ...categorys]));
 
       setResultText([
         t('Analysis Count: {{num}}', { num: analysisCount }),