diff --git a/backend/danswer/db/engine.py b/backend/danswer/db/engine.py index fbe8020dddca..5d4753e136a4 100644 --- a/backend/danswer/db/engine.py +++ b/backend/danswer/db/engine.py @@ -319,7 +319,9 @@ async def get_async_session_with_tenant( await session.execute(text(f'SET search_path = "{tenant_id}"')) if POSTGRES_IDLE_SESSIONS_TIMEOUT: await session.execute( - f"SET SESSION idle_in_transaction_session_timeout = {POSTGRES_IDLE_SESSIONS_TIMEOUT}" + text( + f"SET SESSION idle_in_transaction_session_timeout = {POSTGRES_IDLE_SESSIONS_TIMEOUT}" + ) ) except Exception: logger.exception("Error setting search_path.") @@ -380,7 +382,9 @@ def get_session_with_tenant( cursor.execute(f'SET search_path = "{tenant_id}"') if POSTGRES_IDLE_SESSIONS_TIMEOUT: cursor.execute( - f"SET SESSION idle_in_transaction_session_timeout = {POSTGRES_IDLE_SESSIONS_TIMEOUT}" + text( + f"SET SESSION idle_in_transaction_session_timeout = {POSTGRES_IDLE_SESSIONS_TIMEOUT}" + ) ) finally: cursor.close()