Sadeep Sachintha commited on
Commit
0116ebe
·
1 Parent(s): d24d9cd

fix: resolve create_engine type error and use dynamic prepared statement names

Browse files
Files changed (1) hide show
  1. db/session.py +3 -3
db/session.py CHANGED
@@ -4,6 +4,7 @@ from core.config import settings
4
  from db.models import Base
5
 
6
  import ssl
 
7
 
8
  logger = logging.getLogger(__name__)
9
 
@@ -14,7 +15,7 @@ if "supabase" in settings.database_url or "postgres" in settings.database_url or
14
  ssl_context.verify_mode = ssl.CERT_NONE
15
  connect_args["ssl"] = ssl_context
16
  connect_args["statement_cache_size"] = 0
17
- connect_args["prepared_statement_cache_size"] = 0
18
 
19
  # Initialize the async engine
20
  engine = create_async_engine(
@@ -22,8 +23,7 @@ engine = create_async_engine(
22
  echo=(settings.log_level == "DEBUG"),
23
  future=True,
24
  pool_pre_ping=True,
25
- connect_args=connect_args,
26
- prepared_statement_cache_size=0
27
  )
28
 
29
  # Create an async session factory
 
4
  from db.models import Base
5
 
6
  import ssl
7
+ from uuid import uuid4
8
 
9
  logger = logging.getLogger(__name__)
10
 
 
15
  ssl_context.verify_mode = ssl.CERT_NONE
16
  connect_args["ssl"] = ssl_context
17
  connect_args["statement_cache_size"] = 0
18
+ connect_args["prepared_statement_name_func"] = lambda: f"__asyncpg_{uuid4().hex}__"
19
 
20
  # Initialize the async engine
21
  engine = create_async_engine(
 
23
  echo=(settings.log_level == "DEBUG"),
24
  future=True,
25
  pool_pre_ping=True,
26
+ connect_args=connect_args
 
27
  )
28
 
29
  # Create an async session factory