Merge branch 'fix_tests' into session_id

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

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

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:

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 = []