From 65a205d488d4a3e45d28cf0a8c62671af7ee2b3c Mon Sep 17 00:00:00 2001 From: "Richard Kuo (Danswer)" Date: Wed, 13 Nov 2024 15:03:21 -0800 Subject: [PATCH] use text() --- backend/danswer/db/engine.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/backend/danswer/db/engine.py b/backend/danswer/db/engine.py index 639f6addc75f..6865d6307162 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()