Merge branch 'fix_tests' into session_id

This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-10-03 21:43:51 -03:00
commit d01d8c2945
4 changed files with 6 additions and 4 deletions

View file

@ -1,6 +1,6 @@
from typing import Dict, List, Optional, Type
from langchain import SQLDatabase, utilities
from langchain import utilities
from langflow.custom.customs import get_custom_nodes
from langflow.interface.base import LangChainTypeCreator
@ -32,7 +32,7 @@ class UtilityCreator(LangChainTypeCreator):
utility_name: import_class(f"langchain.utilities.{utility_name}")
for utility_name in utilities.__all__
}
self.type_dict["SQLDatabase"] = SQLDatabase
self.type_dict["SQLDatabase"] = utilities.SQLDatabase
# Filter according to settings.utilities
self.type_dict = {
name: utility

View file

@ -1,6 +1,6 @@
from typing import Dict, List, Optional
from langchain import requests, sql_database
from langchain.utilities import requests, sql_database
from langflow.interface.base import LangChainTypeCreator
from loguru import logger

View file

@ -171,6 +171,7 @@ class DatabaseService(Service):
# Now check if the table "flow" exists, if not, something went wrong
# and we need to create the tables again.
inspector = inspect(self.engine)
table_names = inspector.get_table_names()
for table in current_tables:
if table not in table_names: