From 468e6d57452f2082bef8bb745585d8fdc612023a Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 11 Dec 2023 12:23:38 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(=5F=5Fmain=5F=5F.py):=20repl?= =?UTF-8?q?ace=20deprecated=20session.query()=20method=20with=20session.ex?= =?UTF-8?q?ec()=20to=20fix=20superuser=20creation=20issue?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/__main__.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) 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