diff --git a/src/backend/langflow/interface/utilities/base.py b/src/backend/langflow/interface/utilities/base.py index 665143da2..bfc7cb11e 100644 --- a/src/backend/langflow/interface/utilities/base.py +++ b/src/backend/langflow/interface/utilities/base.py @@ -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 diff --git a/src/backend/langflow/interface/wrappers/base.py b/src/backend/langflow/interface/wrappers/base.py index c4399fb3e..de631101a 100644 --- a/src/backend/langflow/interface/wrappers/base.py +++ b/src/backend/langflow/interface/wrappers/base.py @@ -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 diff --git a/src/backend/langflow/services/database/manager.py b/src/backend/langflow/services/database/manager.py index 18364cd62..4cb715323 100644 --- a/src/backend/langflow/services/database/manager.py +++ b/src/backend/langflow/services/database/manager.py @@ -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: diff --git a/tests/test_setup_superuser.py b/tests/test_setup_superuser.py index 6cf4b3a3a..137c98763 100644 --- a/tests/test_setup_superuser.py +++ b/tests/test_setup_superuser.py @@ -12,9 +12,10 @@ from langflow.services.utils import ( @patch("langflow.services.getters.get_settings_service") @patch("langflow.services.utils.create_super_user") +@patch("langflow.services.utils.verify_password") @patch("langflow.services.getters.get_session") def test_setup_superuser( - mock_get_session, mock_create_super_user, mock_get_settings_service + mock_get_session, mock_create_super_user, mock_get_settings_service, mock_verify ): # Test when AUTO_LOGIN is True calls = []