diff --git a/langchain/sql_database.py b/langchain/sql_database.py index a91c8c1a..fc0a5098 100644 --- a/langchain/sql_database.py +++ b/langchain/sql_database.py @@ -6,7 +6,6 @@ from typing import Any, Iterable, List, Optional import sqlalchemy from sqlalchemy import ( - CursorResult, MetaData, Table, create_engine, @@ -14,7 +13,7 @@ from sqlalchemy import ( select, text, ) -from sqlalchemy.engine import Engine +from sqlalchemy.engine import CursorResult, Engine from sqlalchemy.exc import ProgrammingError, SQLAlchemyError from sqlalchemy.schema import CreateTable diff --git a/pyproject.toml b/pyproject.toml index f6a5745f..f66b83c4 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -13,7 +13,7 @@ langchain-server = "langchain.server:main" [tool.poetry.dependencies] python = ">=3.8.1,<4.0" pydantic = "^1" -SQLAlchemy = ">=1.3,<3" +SQLAlchemy = ">=1.4,<3" requests = "^2" PyYAML = ">=5.4.1" numpy = "^1"