highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-06-30 09:40:50 +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
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 collections.abc import Generator
from datetime import datetime from datetime import datetime
from ddtrace import tracer
from sqlalchemy import text from sqlalchemy import text
from sqlalchemy.engine import create_engine from sqlalchemy.engine import create_engine
from sqlalchemy.engine import Engine from sqlalchemy.engine import Engine
@ -69,8 +70,9 @@ def get_sqlalchemy_async_engine() -> AsyncEngine:
def get_session() -> Generator[Session, None, None]: def get_session() -> Generator[Session, None, None]:
with Session(get_sqlalchemy_engine(), expire_on_commit=False) as session: with tracer.trace("db.get_session"):
yield session with Session(get_sqlalchemy_engine(), expire_on_commit=False) as session:
yield session
async def get_async_session() -> AsyncGenerator[AsyncSession, None]: async def get_async_session() -> AsyncGenerator[AsyncSession, None]: