Skip to content

Commit 0ab2cf3

Browse files
committed
Fixes session factory
1 parent 7876f75 commit 0ab2cf3

File tree

2 files changed

+10
-7
lines changed

2 files changed

+10
-7
lines changed

‎dialog_lib/db/session.py‎

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,12 +5,15 @@
55

66
from contextlib import contextmanager
77

8-
engine = sa.create_engine(os.environ.get("DATABASE_URL"))
9-
Session = sessionmaker(bind=engine)
8+
from functools import cache
9+
10+
@cache
11+
def get_engine():
12+
return sa.create_engine(os.environ.get("DATABASE_URL"))
1013

1114
@contextmanager
1215
def session_scope():
13-
with Session(bind=engine) as session:
16+
with Session(bind=get_engine()) as session:
1417
try:
1518
yield session
1619
session.commit()
@@ -22,4 +25,4 @@ def session_scope():
2225

2326
def get_session():
2427
with session_scope() as session:
25-
return session
28+
return session

‎docker-compose.yml‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,11 @@
1-
version: '3.3'
1+
version: "3.3"
22
services:
33
db:
44
image: pgvector/pgvector:pg15
55
restart: always
66
volumes:
7-
- ./etc/db-ext-vector-test.sql:/docker-entrypoint-initdb.d/init.sql
8-
- postgres_data:/var/lib/postgresql/data/
7+
- ./etc/db-ext-vector-test.sql:/docker-entrypoint-initdb.d/init.sql
8+
- postgres_data:/var/lib/postgresql/data/
99
environment:
1010
POSTGRES_USER: talkdai
1111
POSTGRES_PASSWORD: talkdai

0 commit comments

Comments
 (0)