Skip to content

Commit 85e4b62

Browse files
JerryFan626JerryFan626
committed
feat: Add tips about auto sync table to AI
1 parent 0de30f1 commit 85e4b62

5 files changed

Lines changed: 53 additions & 33 deletions

File tree

‎chat2db-client/src/components/Console/ChatInput/index.less‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@
4343
.tableSelectBlock {
4444
// margin-right: 8px;
4545
cursor: pointer;
46-
padding: 4px;
46+
padding: 8px;
4747
border-radius: 4px 12px;
4848

4949
&:hover {

‎chat2db-client/src/components/Console/ChatInput/index.tsx‎

Lines changed: 42 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import React, { ChangeEvent, useEffect, useState } from 'react';
22
import styles from './index.less';
33
import AIImg from '@/assets/img/ai.svg';
4-
import { Button, Checkbox, Dropdown, Input, Modal, Popover, Select, Spin } from 'antd';
4+
import { Button, Checkbox, Dropdown, Input, Modal, Popover, Select, Spin, Tooltip, Radio } from 'antd';
55
import i18n from '@/i18n/';
66
import Iconfont from '@/components/Iconfont';
77
import { WarningOutlined } from '@ant-design/icons';
@@ -12,12 +12,14 @@ interface IProps {
1212
value?: string;
1313
result?: string;
1414
tables?: string[];
15+
syncTableModel: number;
1516
selectedTables?: string[];
1617
remainingUse?: IRemainingUse;
1718
aiType: AiSqlSourceType;
1819
remainingBtnLoading: boolean;
1920
disabled?: boolean;
2021
onPressEnter: (value: string) => void;
22+
onSelectTableSyncModel: (model: number) => void;
2123
onSelectTables?: (tables: string[]) => void;
2224
onClickRemainBtn: Function;
2325
}
@@ -37,31 +39,38 @@ const ChatInput = (props: IProps) => {
3739
};
3840

3941
const renderSelectTable = () => {
40-
const { tables, selectedTables, onSelectTables } = props;
42+
const { tables, syncTableModel, onSelectTableSyncModel, selectedTables, onSelectTables } = props;
4143
const options = (tables || []).map((t) => ({ value: t, label: t }));
4244
return (
4345
<div className={styles.aiSelectedTable}>
44-
<span className={styles.aiSelectedTableTips}>
45-
{/* <WarningOutlined style={{color: 'yellow'}}/> */}
46-
{i18n('chat.input.remain.tooltip')}
47-
</span>
48-
<Select
49-
showSearch
50-
mode="multiple"
51-
allowClear
52-
options={options}
53-
placeholder={i18n('chat.input.tableSelect.placeholder')}
54-
value={selectedTables}
55-
onChange={(v) => {
56-
onSelectTables && onSelectTables(v);
57-
}}
58-
/>
46+
<Radio.Group onChange={(v) => onSelectTableSyncModel(v.target.value)} value={syncTableModel}>
47+
<Radio value={0}>{i18n('chat.input.syncTable.tips')}</Radio>
48+
<Radio value={1} style={{ marginTop: '8px', display: 'flex' }}>
49+
<>
50+
<span className={styles.aiSelectedTableTips}>
51+
{/* <WarningOutlined style={{color: 'yellow'}}/> */}
52+
{i18n('chat.input.remain.tooltip')}
53+
</span>
54+
<Select
55+
showSearch
56+
mode="multiple"
57+
allowClear
58+
options={options}
59+
placeholder={i18n('chat.input.tableSelect.placeholder')}
60+
value={selectedTables}
61+
onChange={(v) => {
62+
onSelectTables && onSelectTables(v);
63+
}}
64+
/>
65+
</>
66+
</Radio>
67+
</Radio.Group>
5968
</div>
6069
);
6170
};
6271

6372
const renderSuffix = () => {
64-
const remainCnt = props?.remainingUse?.remainingUses ?? '-';
73+
// const remainCnt = props?.remainingUse?.remainingUses ?? '-';
6574
return (
6675
<div className={styles.suffixBlock}>
6776
<Button
@@ -75,11 +84,20 @@ const ChatInput = (props: IProps) => {
7584
>
7685
<Iconfont code="&#xe643;" className={styles.enterIcon} />
7786
</Button>
78-
<div className={styles.tableSelectBlock}>
79-
<Popover content={renderSelectTable()} placement="bottom">
80-
<Iconfont code="&#xe618;" />
81-
</Popover>
82-
</div>
87+
<Tooltip
88+
title="🎉上线自动同步所有表功能"
89+
open={!localStorage.getItem('syncTableBubble')}
90+
onOpenChange={() => {
91+
localStorage.setItem('syncTableBubble', 'true');
92+
}}
93+
>
94+
<div className={styles.tableSelectBlock}>
95+
<Popover content={renderSelectTable()} placement="bottom">
96+
<Iconfont code="&#xe618;" />
97+
</Popover>
98+
</div>
99+
</Tooltip>
100+
83101
{/* {props.aiType === AiSqlSourceType.CHAT2DBAI && (
84102
<Spin spinning={!!props.remainingBtnLoading} size="small">
85103
<div
@@ -110,6 +128,6 @@ const ChatInput = (props: IProps) => {
110128
/>
111129
</div>
112130
);
113-
}
131+
};
114132

115133
export default ChatInput;

‎chat2db-client/src/components/Console/index.tsx‎

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -93,6 +93,7 @@ function Console(props: IProps) {
9393
const chatResult = useRef('');
9494
const editorRef = useRef<IExportRefFunction>();
9595
const [selectedTables, setSelectedTables] = useState<string[]>([]);
96+
const [syncTableModel, setSyncTableModel] = useState<number>(0);
9697
const [isLoading, setIsLoading] = useState(false);
9798
const [aiContent, setAiContent] = useState('');
9899
const [isAiDrawerOpen, setIsAiDrawerOpen] = useState(false);
@@ -233,7 +234,7 @@ function Console(props: IProps) {
233234
dataSourceId,
234235
databaseName,
235236
schemaName,
236-
tableNames: selectedTables,
237+
tableNames: syncTableModel ? selectedTables : null,
237238
});
238239

239240
const handleMessage = (message: string) => {
@@ -429,6 +430,8 @@ function Console(props: IProps) {
429430
setSelectedTables(tables);
430431
}}
431432
onClickRemainBtn={handleClickRemainBtn}
433+
syncTableModel={syncTableModel}
434+
onSelectTableSyncModel={(model: number) => setSyncTableModel(model)}
432435
/>
433436
)}
434437
{/* <div key={uuid()}>{chatContent.current}</div> */}

‎chat2db-client/src/i18n/en-us/chat.ts‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
export default {
22
'chat.input.remain': '{1} remaining',
3-
'chat.input.remain.tooltip': 'Table schema added to prompt.',
43
'chat.input.tableSelect.placeholder':'Please choose tables',
54
'chat.input.tableSelect.error.TooManyTable':'You can only select up to 8 tables',
65
'chat.input.remain.dialog.tips':
7-
'Subscribe our official WeChat account, send 推广 to get more chances to experience.',
8-
6+
'Subscribe our official WeChat account, send 推广 to get more chances to experience.',
7+
"chat.input.syncTable.tips":'Mode One: Automatically synchronize all tables to the AI context',
8+
'chat.input.remain.tooltip': 'Mode Two: Table schema added to prompt.',
99
};

‎chat2db-client/src/i18n/zh-cn/chat.ts‎

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,8 @@
11
export default {
22
'chat.input.remain': '剩余 {1} 次',
3-
'chat.input.remain.tooltip': '选中表的schema将会被添加到prompt中',
43
'chat.input.tableSelect.placeholder': '请选择表',
54
'chat.input.tableSelect.error.TooManyTable': '最多选择8张表',
6-
'chat.input.remain.dialog.tips':
7-
'关注公众号，发送"推广"获取更多体验次数',
8-
5+
'chat.input.remain.dialog.tips': '关注公众号，发送"推广"获取更多体验次数',
6+
'chat.input.syncTable.tips': '模式一: 自动同步所有表给AI上下文',
7+
'chat.input.remain.tooltip': '模式二: 选中表的schema将会被添加到prompt中',
98
};

0 commit comments

Comments
 (0)