diff --git a/backend/danswer/db/engine.py b/backend/danswer/db/engine.py index 639f6addc75..6865d630716 100644 --- a/backend/danswer/db/engine.py +++ b/backend/danswer/db/engine.py @@ -312,7 +312,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.") @@ -373,7 +375,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()