Do not use sqlite-only args when using non-sqlite

Shaunwei · Jul 16, 2023 · b1be209 · b1be209
1 parent 91dab3f
commit b1be209
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/realtime_ai_character/database/connection.py b/realtime_ai_character/database/connection.py
@@ -7,8 +7,11 @@
 
 SQLALCHEMY_DATABASE_URL = os.getenv("DATABASE_URL")
 
+connect_args = {"check_same_thread": False} if SQLALCHEMY_DATABASE_URL.startswith(
+    "sqlite") else {}
+
 engine = create_engine(
-    SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False}
+    SQLALCHEMY_DATABASE_URL, connect_args=connect_args
 )
 
 SessionLocal = sessionmaker(