From 196bdc51010c5abf4ba212d9cda9619503084ce0 Mon Sep 17 00:00:00 2001 From: Masen Furer Date: Tue, 22 Aug 2023 14:07:36 -0700 Subject: [PATCH] model: only pass "check_same_thread" arg for sqlite database (#1662) --- reflex/model.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/reflex/model.py b/reflex/model.py index 4bc55912c..2e9d275b5 100644 --- a/reflex/model.py +++ b/reflex/model.py @@ -43,12 +43,9 @@ def get_engine(url: Optional[str] = None): ) # Print the SQL queries if the log level is INFO or lower. echo_db_query = os.environ.get("SQLALCHEMY_ECHO") == "True" - return sqlmodel.create_engine( - url, - echo=echo_db_query, - # Needed for the admin dash. - connect_args={"check_same_thread": False}, - ) + # Needed for the admin dash on sqlite. + connect_args = {"check_same_thread": False} if url.startswith("sqlite") else {} + return sqlmodel.create_engine(url, echo=echo_db_query, connect_args=connect_args) class Model(Base, sqlmodel.SQLModel):