File tree Expand file tree Collapse file tree 1 file changed +12
-1
lines changed
app/resource_adapters/persistence/sqlmodel Expand file tree Collapse file tree 1 file changed +12
-1
lines changed Original file line number Diff line number Diff line change 33from loguru import logger
44from sqlalchemy import text
55from sqlalchemy .engine import Engine
6+ from sqlalchemy .pool import StaticPool
67from sqlmodel import Session , SQLModel , create_engine
78
89from config import settings
@@ -25,7 +26,17 @@ def get_engine(database_url: str | None = None) -> Engine:
2526 if database_url is None :
2627 database_url = settings .database_url
2728
28- _engine = create_engine (database_url , echo = True )
29+ # Configure engine based on database type
30+ engine_args = {"echo" : True }
31+
32+ # Add SQLite-specific settings for in-memory database
33+ if database_url == "sqlite://" :
34+ engine_args .update ({
35+ "connect_args" : {"check_same_thread" : False },
36+ "poolclass" : StaticPool
37+ })
38+
39+ _engine = create_engine (database_url , ** engine_args )
2940
3041 # Enable WAL mode if configured
3142 if settings .sqlite_wal_mode and database_url .startswith ("sqlite" ):
You can’t perform that action at this time.
0 commit comments