From e0331b55bb8a53cb169adbbceda7477284e7e8f9 Mon Sep 17 00:00:00 2001 From: Clark Date: Tue, 28 Mar 2023 14:45:50 +0800 Subject: [PATCH] fix(sql_database): related to #2020 (#2021) Fixed https://github.com/hwchase17/langchain/issues/2020 Co-authored-by: qianjun.wqj --- langchain/sql_database.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/langchain/sql_database.py b/langchain/sql_database.py index 18f76bd2..424f789a 100644 --- a/langchain/sql_database.py +++ b/langchain/sql_database.py @@ -124,7 +124,7 @@ class SQLDatabase: if self._sample_rows_in_table_info: # build the select command - command = select(table).limit(self._sample_rows_in_table_info) + command = select([table]).limit(self._sample_rows_in_table_info) # save the columns in string format columns_str = "\t".join([col.name for col in table.columns])