fix: fix bug

JerryFan626 · JerryFan626 · commit 102a8af9f7a9 · 2023-10-16T14:38:06.000+08:00
diff --git a/chat2db-client/src/components/Console/ChatInput/index.tsx b/chat2db-client/src/components/Console/ChatInput/index.tsx
@@ -23,7 +23,6 @@ interface IProps {
   aiType: AiSqlSourceType;
   remainingBtnLoading: boolean;
   disabled?: boolean;
-  defaultSelectedSyncModel?: SyncModelType;
   onPressEnter: (value: string) => void;
   onSelectTableSyncModel: (model: number) => void;
   onSelectTables?: (tables: string[]) => void;
@@ -50,7 +49,6 @@ const ChatInput = (props: IProps) => {
     return (
       <div className={styles.aiSelectedTable}>
         <Radio.Group
-          defaultValue={props.defaultSelectedSyncModel ?? SyncModelType.MANUAL}
           onChange={(v) => onSelectTableSyncModel(v.target.value)}
           value={syncTableModel}
           style={{ marginBottom: '8px' }}
@@ -102,7 +100,7 @@ const ChatInput = (props: IProps) => {
           title={<span style={{ color: window._AppThemePack.colorText }}>{i18n('chat.input.syncTable.tempTips')}</span>}
           defaultOpen={!hasBubble}
           color={window._AppThemePack.colorBgBase}
-          trigger={'click'}
+          trigger={'contextMenu'}
           onOpenChange={() => {
             localStorage.setItem('syncTableBubble', 'true');
           }}
diff --git a/chat2db-client/src/components/Console/index.tsx b/chat2db-client/src/components/Console/index.tsx
@@ -21,7 +21,7 @@ import configService from '@/service/config';
 // import NewEditor from './NewMonacoEditor';
 import styles from './index.less';
 import indexedDB from '@/indexedDB';
-import { isEmpty } from 'lodash';
+import { isEmpty, set } from 'lodash';
 
 enum IPromptType {
   NL_2_SQL = 'NL_2_SQL',
@@ -114,6 +114,10 @@ function Console(props: IProps) {
     [aiModel.aiConfig?.aiSqlSource],
   );
 
+  useEffect(() => {
+    handleSelectTableSyncModel();
+  }, [aiModel.hasWhite, localStorage.getItem('syncTableModel')]);
+
   useEffect(() => {
     if (appendValue) {
       editorRef?.current?.setValue(appendValue.text, appendValue.range);
@@ -430,14 +434,11 @@ function Console(props: IProps) {
     const syncModel: SyncModelType | null = Number(localStorage.getItem('syncTableModel')) ?? null;
     const hasAiAccess = aiModel.hasWhite;
 
-    if (!hasAiAccess) {
-      return SyncModelType.MANUAL;
+    if (!hasAiAccess || isEmpty(syncModel)) {
+      setSyncTableModel(SyncModelType.MANUAL);
     }
 
-    if (isEmpty(syncModel)) {
-      return SyncModelType.MANUAL;
-    }
-    return syncModel;
+    setSyncTableModel(syncModel);
   };
 
   return (
@@ -465,7 +466,6 @@ function Console(props: IProps) {
             }}
             onClickRemainBtn={handleClickRemainBtn}
             syncTableModel={syncTableModel}
-            defaultSelectedSyncModel={handleSelectTableSyncModel()}
             onSelectTableSyncModel={(model: number) => {
               setSyncTableModel(model);
               localStorage.setItem('syncTableModel', String(model));
