github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-18 10:43:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

updated sql_database.py for returning sorted table names. (#6692)

Browse Source 
Added code to get the tables info in sorted order in methods
get_usable_table_names and get_table_info.

Linked to Issue: #6640
This commit is contained in:
Sumanth Donthula 2023-06-25 14:04:24 -05:00 committed by GitHub
parent 9d1b3bab76
commit 3e30a5d967
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
langchain/sql_database.py
View File

@ -230,8 +230,8 @@ class SQLDatabase:
def get_usable_table_names(self) -> Iterable[str]: def get_usable_table_names(self) -> Iterable[str]:
"""Get names of tables available.""" """Get names of tables available."""
if self._include_tables: if self._include_tables:
return self._include_tables return sorted(self._include_tables)
return self._all_tables - self._ignore_tables return sorted(self._all_tables - self._ignore_tables)
def get_table_names(self) -> Iterable[str]: def get_table_names(self) -> Iterable[str]:
"""Get names of tables available.""" """Get names of tables available."""
@ -290,6 +290,7 @@ class SQLDatabase:
if has_extra_info: if has_extra_info:
table_info += "*/" table_info += "*/"
tables.append(table_info) tables.append(table_info)
tables.sort()
final_str = "\n\n".join(tables) final_str = "\n\n".join(tables)
return final_str return final_str