opensource/sqlchat
1
0
Fork 0
mirror of https://github.com/sqlchat/sqlchat.git synced 2025-07-25 15:56:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: display default prompt in the schema drawer if general bot is selected

Browse Source
This commit is contained in:
Tianzhou Chen
2023-06-07 23:44:42 +08:00
parent 8434d83133
commit 3b276ce9a6
1 changed files with 10 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
src/components/SchemaDrawer.tsx
View File

@ -22,26 +22,20 @@ const SchemaDrawer = (props: Props) => {
const getPrompt = async () => {
if (!currentConversation) return;
if (!connectionStore.currentConnectionCtx?.database) return;
const promptGenerator = getPromptGeneratorOfAssistant(getAssistantById(currentConversation.assistantId)!);
let dbPrompt = promptGenerator();
const maxToken = getModel(settingStore.setting.openAIApiConfig?.model || "").max_token;
const schemaList = await connectionStore.getOrFetchDatabaseSchema(connectionStore.currentConnectionCtx?.database);
if (connectionStore.currentConnectionCtx?.database) {
try {
dbPrompt = generateDbPromptFromContext(
promptGenerator,
schemaList,
currentConversation.selectedSchemaName || "",
currentConversation.selectedTablesName || [],
maxToken
);
setPrompt(dbPrompt);
} catch (error: any) {
toast.error(error.message);
}
const maxToken = getModel(settingStore.setting.openAIApiConfig?.model || "").max_token;
const schemaList = await connectionStore.getOrFetchDatabaseSchema(connectionStore.currentConnectionCtx?.database);
dbPrompt = generateDbPromptFromContext(
promptGenerator,
schemaList,
currentConversation.selectedSchemaName || "",
currentConversation.selectedTablesName || [],
maxToken
);
}
setPrompt(dbPrompt);
};
useEffect(() => {