🐛 fix(__main__.py): replace deprecated session.query() method with session.exec() to fix superuser creation issue
This commit is contained in:
parent
3169e2ecf5
commit
468e6d5745
1 changed files with 8 additions and 6 deletions
|
|
@ -9,17 +9,19 @@ from typing import Optional
|
||||||
import httpx
|
import httpx
|
||||||
import typer
|
import typer
|
||||||
from dotenv import load_dotenv
|
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 multiprocess import Process, cpu_count # type: ignore
|
||||||
from rich import box
|
from rich import box
|
||||||
from rich import print as rprint
|
from rich import print as rprint
|
||||||
from rich.console import Console
|
from rich.console import Console
|
||||||
from rich.panel import Panel
|
from rich.panel import Panel
|
||||||
from rich.table import Table
|
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()
|
console = Console()
|
||||||
|
|
||||||
|
|
@ -338,7 +340,7 @@ def superuser(
|
||||||
# Verify that the superuser was created
|
# Verify that the superuser was created
|
||||||
from langflow.services.database.models.user.model import User
|
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:
|
if user is None or not user.is_superuser:
|
||||||
typer.echo("Superuser creation failed.")
|
typer.echo("Superuser creation failed.")
|
||||||
return
|
return
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue