Support SQLAlchemy 2.0 (#3310)

With executablebooks/jupyter-cache#93 merged and
`MyST-NB` updated, we can now support SQLAlchemy 2. Closes #1766

Author: jzluo

langchain/sql_database.py

@@ -180,7 +180,7 @@ def _get_table_indexes(self, table: Table) -> str:
 
     def _get_sample_rows(self, table: Table) -> str:
         # 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])

pyproject.toml

@@ -13,7 +13,7 @@ langchain-server = "langchain.server:main"
 [tool.poetry.dependencies]
 python = ">=3.8.1,<4.0"
 pydantic = "^1"
-SQLAlchemy = "^1"
+SQLAlchemy = ">1.3,<3"
 requests = "^2"
 PyYAML = ">=5.4.1"
 numpy = "^1"