github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-08 22:42:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add option to ignore or restrict to SQL tables (#151)

Browse Source 
`SQLDatabase` now accepts two `init` arguments:
1. `ignore_tables` to pass in a list of tables to not search over
2. `include_tables` to restrict to a list of tables to consider
This commit is contained in:
Nicholas Larus-Stone
2022-11-16 22:04:50 -08:00
committed by GitHub
parent d2f9288be6
commit ca4b10bb74
2 changed files with 41 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/unit_tests/test_sql_database.py
View File

@@ -28,11 +28,11 @@ def test_table_info() -> None:
db = SQLDatabase(engine)
output = db.table_info
expected_output = (
"The 'company' table has columns: company_id (INTEGER), "
"company_location (VARCHAR).\n"
"The 'user' table has columns: user_id (INTEGER), user_name (VARCHAR(16))."
"Table 'company' has columns: company_id (INTEGER), "
"company_location (VARCHAR).",
"Table 'user' has columns: user_id (INTEGER), user_name (VARCHAR(16)).",
)
assert output == expected_output
assert sorted(output.split("\n")) == sorted(expected_output)
def test_sql_database_run() -> None: