dont error on sql import (#4647)

this makes it so we dont throw errors when importing langchain when
sqlalchemy==1.3.1

we dont really want to support 1.3.1 (seems like unneccessary maintance
cost) BUT we would like it to not terribly error should someone decide
to run on it
This commit is contained in:
Harrison Chase
2023-05-18 09:27:09 -07:00
committed by GitHub
parent c9a362e482
commit d5a0704544
3 changed files with 6 additions and 4 deletions
+2 -2
View File
@@ -13,7 +13,7 @@ from sqlalchemy import (
select,
text,
)
from sqlalchemy.engine import CursorResult, Engine
from sqlalchemy.engine import Engine
from sqlalchemy.exc import ProgrammingError, SQLAlchemyError
from sqlalchemy.schema import CreateTable
@@ -196,7 +196,7 @@ class SQLDatabase:
try:
# get the sample rows
with self._engine.connect() as connection:
sample_rows_result: CursorResult = connection.execute(command)
sample_rows_result = connection.execute(command) # type: ignore
# shorten values in the sample rows
sample_rows = list(
map(lambda ls: [str(i)[:100] for i in ls], sample_rows_result)