diff --git a/src/backend/langflow/services/database/manager.py b/src/backend/langflow/services/database/manager.py index 7f8afab6f..a89db8395 100644 --- a/src/backend/langflow/services/database/manager.py +++ b/src/backend/langflow/services/database/manager.py @@ -107,12 +107,10 @@ class DatabaseManager(Service): # We will check that all models are in the database # and that the database is up to date with all columns sql_models = [models.Flow, models.User, models.ApiKey] - results = [] - for sql_model in sql_models: - results.append( - TableResults(sql_model.__tablename__, self.check_table(sql_model)) - ) - return results + return [ + TableResults(sql_model.__tablename__, self.check_table(sql_model)) + for sql_model in sql_models + ] def check_table(self, model): results = [] diff --git a/src/backend/langflow/services/manager.py b/src/backend/langflow/services/manager.py index 34c58c648..dc901730c 100644 --- a/src/backend/langflow/services/manager.py +++ b/src/backend/langflow/services/manager.py @@ -1,5 +1,5 @@ from langflow.services.schema import ServiceType -from typing import TYPE_CHECKING, List, Optional +from typing import TYPE_CHECKING, Dict, List, Optional from loguru import logger if TYPE_CHECKING: