diff --git a/src/backend/langflow/__main__.py b/src/backend/langflow/__main__.py index 1c13c588c..cbfddf289 100644 --- a/src/backend/langflow/__main__.py +++ b/src/backend/langflow/__main__.py @@ -9,17 +9,19 @@ from typing import Optional import httpx import typer from dotenv import load_dotenv -from langflow.main import setup_app -from langflow.services.database.utils import session_getter -from langflow.services.deps import get_db_service, get_settings_service -from langflow.services.utils import initialize_services, initialize_settings_service -from langflow.utils.logger import configure, logger from multiprocess import Process, cpu_count # type: ignore from rich import box from rich import print as rprint from rich.console import Console from rich.panel import Panel from rich.table import Table +from sqlmodel import select + +from langflow.main import setup_app +from langflow.services.database.utils import session_getter +from langflow.services.deps import get_db_service, get_settings_service +from langflow.services.utils import initialize_services, initialize_settings_service +from langflow.utils.logger import configure, logger console = Console() @@ -338,7 +340,7 @@ def superuser( # Verify that the superuser was created from langflow.services.database.models.user.model import User - user: User = session.query(User).filter(User.username == username).first() + user: User = session.exec(select(User).where(User.username == username)).first() if user is None or not user.is_superuser: typer.echo("Superuser creation failed.") return