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

Add trace for db session creation

Browse Source
This commit is contained in:
Weves 2024-02-29 13:49:16 -08:00 committed by Chris Weaver
parent 31d3ae0e3e
commit 2331bf9b36
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
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
@ -69,8 +70,9 @@ def get_sqlalchemy_async_engine() -> AsyncEngine:
def get_session() -> Generator[Session, None, None]:
with Session(get_sqlalchemy_engine(), expire_on_commit=False) as session:
yield session
with tracer.trace("db.get_session"):
with Session(get_sqlalchemy_engine(), expire_on_commit=False) as session:
yield session
async def get_async_session() -> AsyncGenerator[AsyncSession, None]: