opensource/sqlchat
1
0
Fork 0
mirror of https://github.com/sqlchat/sqlchat.git synced 2025-08-02 14:01:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: include non-public schema in the prompt

Browse Source
This commit is contained in:
Tianzhou Chen
2023-12-17 23:34:41 +08:00
parent eb9ec23838
commit f2cc5e8076
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/lib/connectors/postgres/index.ts
View File

@ -128,7 +128,9 @@ const getTableSchema = async (connection: Connection, databaseName: string): Pro
`${row["column_name"]} ${row["data_type"].toUpperCase()} ${String(row["is_nullable"]).toUpperCase() === "NO" ? "NOT NULL" : ""}` `${row["column_name"]} ${row["data_type"].toUpperCase()} ${String(row["is_nullable"]).toUpperCase() === "NO" ? "NOT NULL" : ""}`
); );
} }
table.structure = `CREATE TABLE \`${table.name}\` (\n${columnList.join(",\n")}\n);`;
let fullTableName = schema.name == "public" ? `"${table.name}"` : `"${schema.name}"."${table.name}"`;
table.structure = `CREATE TABLE ${fullTableName} (\n${columnList.join(",\n")}\n);`;
} }
} }