highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-06-27 16:30:56 +02:00
Code Issues Packages Projects Releases Wiki Activity

use text()

Browse Source
This commit is contained in:
Richard Kuo (Danswer) 2024-11-13 15:03:21 -08:00
parent 485f3f72fa
commit 65a205d488
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
backend/danswer/db/engine.py
View File

@ -312,7 +312,9 @@ async def get_async_session_with_tenant(
await session.execute(text(f'SET search_path = "{tenant_id}"')) await session.execute(text(f'SET search_path = "{tenant_id}"'))
if POSTGRES_IDLE_SESSIONS_TIMEOUT: if POSTGRES_IDLE_SESSIONS_TIMEOUT:
await session.execute( 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: except Exception:
logger.exception("Error setting search_path.") logger.exception("Error setting search_path.")
@ -373,7 +375,9 @@ def get_session_with_tenant(
cursor.execute(f'SET search_path = "{tenant_id}"') cursor.execute(f'SET search_path = "{tenant_id}"')
if POSTGRES_IDLE_SESSIONS_TIMEOUT: if POSTGRES_IDLE_SESSIONS_TIMEOUT:
cursor.execute( 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: finally:
cursor.close() cursor.close()