From 490d39f08164d0b017d7ad1e79fab0ffabc45780 Mon Sep 17 00:00:00 2001 From: Weves Date: Fri, 16 Jun 2023 19:02:50 -0600 Subject: [PATCH] Removing deprecated field --- backend/danswer/background/update.py | 2 +- backend/danswer/db/credentials.py | 2 +- backend/danswer/db/engine.py | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/backend/danswer/background/update.py b/backend/danswer/background/update.py index 75c079bf7..9ee4994c7 100755 --- a/backend/danswer/background/update.py +++ b/backend/danswer/background/update.py @@ -190,7 +190,7 @@ def update_loop(delay: int = 10) -> None: start = time.time() logger.info(f"Running update, current time: {time.ctime(start)}") try: - with Session(engine, future=True, expire_on_commit=False) as db_session: + with Session(engine, expire_on_commit=False) as db_session: create_indexing_jobs(db_session) run_indexing_jobs(db_session) except Exception as e: diff --git a/backend/danswer/db/credentials.py b/backend/danswer/db/credentials.py index 5bfd4b507..d748f793f 100644 --- a/backend/danswer/db/credentials.py +++ b/backend/danswer/db/credentials.py @@ -136,7 +136,7 @@ def create_initial_public_credential() -> None: "DB is not in a valid initial state." "There must exist an empty public credential for data connectors that do not require additional Auth." ) - with Session(build_engine(), future=True, expire_on_commit=False) as db_session: + with Session(build_engine(), expire_on_commit=False) as db_session: first_credential = fetch_credential_by_id(public_cred_id, None, db_session) if first_credential is not None: diff --git a/backend/danswer/db/engine.py b/backend/danswer/db/engine.py index 4d40d5f9d..02eb10479 100644 --- a/backend/danswer/db/engine.py +++ b/backend/danswer/db/engine.py @@ -60,12 +60,12 @@ def build_async_engine() -> AsyncEngine: def get_session() -> Generator[Session, None, None]: - with Session(build_engine(), future=True, expire_on_commit=False) as session: + with Session(build_engine(), expire_on_commit=False) as session: yield session async def get_async_session() -> AsyncGenerator[AsyncSession, None]: async with AsyncSession( - build_async_engine(), future=True, expire_on_commit=False + build_async_engine(), expire_on_commit=False ) as async_session: yield async_session