mirror of
https://github.com/danswer-ai/danswer.git
synced 2025-07-04 11:41:04 +02:00
@ -2,7 +2,6 @@ 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,7 +68,6 @@ def get_sqlalchemy_async_engine() -> AsyncEngine:
|
|||||||
return _ASYNC_ENGINE
|
return _ASYNC_ENGINE
|
||||||
|
|
||||||
|
|
||||||
@tracer.wrap("db.get_session")
|
|
||||||
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 Session(get_sqlalchemy_engine(), expire_on_commit=False) as session:
|
||||||
yield session
|
yield session
|
||||||
|
Reference in New Issue
Block a user