highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-04-10 04:49:29 +02:00
Code Issues Packages Projects Releases Wiki Activity

Trace sqla get_session

Browse Source
This commit is contained in:
Kevin Shi 2024-02-29 15:45:53 -05:00 committed by Chris Weaver
parent c7d228e292
commit d07345c533
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
backend/danswer/db/engine.py
View File

@ -2,6 +2,7 @@ from collections.abc import AsyncGenerator
from collections.abc import Generator
from datetime import datetime
from ddtrace import tracer
from sqlalchemy import text
from sqlalchemy.engine import create_engine
from sqlalchemy.engine import Engine
@ -68,6 +69,7 @@ def get_sqlalchemy_async_engine() -> AsyncEngine:
return _ASYNC_ENGINE
@tracer.wrap("db.get_session")
def get_session() -> Generator[Session, None, None]:
with Session(get_sqlalchemy_engine(), expire_on_commit=False) as session:
yield session