From c7dc0e9af0ada61165f8c2ad49c175abf7cae509 Mon Sep 17 00:00:00 2001 From: pablodanswer Date: Sat, 19 Oct 2024 18:28:32 -0700 Subject: [PATCH] k --- backend/danswer/db/engine.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/backend/danswer/db/engine.py b/backend/danswer/db/engine.py index 1152ae458..af1dd3a0f 100644 --- a/backend/danswer/db/engine.py +++ b/backend/danswer/db/engine.py @@ -297,10 +297,13 @@ def get_session_with_tenant( ) -> Generator[Session, None, None]: """Generate a database session bound to a connection with the appropriate tenant schema set.""" engine = get_sqlalchemy_engine() - event.listen(engine, "checkout", set_search_path_on_checkout) if tenant_id is None: tenant_id = current_tenant_id.get() + else: + current_tenant_id.set(tenant_id) + + event.listen(engine, "checkout", set_search_path_on_checkout) if not is_valid_schema_name(tenant_id): raise HTTPException(status_code=400, detail="Invalid tenant ID")