opensource/sqlchat
1
0
Fork 0
mirror of https://github.com/sqlchat/sqlchat.git synced 2025-09-28 10:33:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: select as many tables up to the maximum token by default

Browse Source
This commit is contained in:
Tianzhou Chen
2023-06-08 01:23:34 +08:00
parent e774078613
commit 88f80aaefe
2 changed files with 19 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/components/ConnectionSidebar.tsx
View File

@ -102,6 +102,20 @@ const ConnectionSidebar = () => {
useEffect(() => {
const tableList = schemaList.find((schema) => schema.name === selectedSchemaName)?.tables || [];
updateTableList(tableList);
// By default, select as many tables up to the maximum token.
const defaultCheckedTableList = [];
if (selectedTableNameList.length === 0) {
let tokenCount = 0;
for (const table of tableList) {
tokenCount += countTextTokens(table?.structure || "");
if (tokenCount > maxToken) {
break;
}
defaultCheckedTableList.push(table.name);
}
conversationStore.updateSelectedTablesNameList(defaultCheckedTableList);
}
}, [selectedSchemaName, schemaList]);
useEffect(() => {
@ -144,15 +158,15 @@ const ConnectionSidebar = () => {
const handleTableCheckboxChange = async (tableName: string, value: boolean) => {
if (value) {
conversationStore.updateSelectedTablesName([...selectedTableNameList, tableName]);
conversationStore.updateSelectedTablesNameList([...selectedTableNameList, tableName]);
} else {
conversationStore.updateSelectedTablesName(selectedTableNameList.filter((name) => name !== tableName));
conversationStore.updateSelectedTablesNameList(selectedTableNameList.filter((name) => name !== tableName));
}
};
const handleSchemaNameSelect = async (schemaName: string) => {
// need to empty selectedTableNameList when schemaName changed. because selectedTableNameList may not exist in new schema.
conversationStore.updateSelectedTablesName([]);
conversationStore.updateSelectedTablesNameList([]);
conversationStore.updateSelectedSchemaName(schemaName);
};