highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-09-28 04:49:21 +02:00
Code Issues Packages Projects Releases Wiki Activity

multitenant setup (#2845)

Browse Source
This commit is contained in:
pablodanswer
2024-10-17 17:54:02 -07:00
committed by GitHub
parent 61424de531
commit 6e54c97326
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
backend/danswer/main.py
View File

@@ -84,6 +84,7 @@ from danswer.server.token_rate_limits.api import (
router as token_rate_limit_settings_router, router as token_rate_limit_settings_router,
) )
from danswer.setup import setup_danswer from danswer.setup import setup_danswer
from danswer.setup import setup_multitenant_danswer
from danswer.utils.logger import setup_logger from danswer.utils.logger import setup_logger
from danswer.utils.telemetry import get_or_generate_uuid from danswer.utils.telemetry import get_or_generate_uuid
from danswer.utils.telemetry import optional_telemetry from danswer.utils.telemetry import optional_telemetry
@@ -182,6 +183,8 @@ async def lifespan(app: FastAPI) -> AsyncGenerator:
# If we are multi-tenant, we need to only set up initial public tables # If we are multi-tenant, we need to only set up initial public tables
with Session(engine) as db_session: with Session(engine) as db_session:
setup_danswer(db_session) setup_danswer(db_session)
else:
setup_multitenant_danswer()
optional_telemetry(record_type=RecordType.VERSION, data={"version": __version__}) optional_telemetry(record_type=RecordType.VERSION, data={"version": __version__})
yield yield
@@ -312,6 +315,7 @@ def get_application() -> FastAPI:
prefix="/auth/oauth", prefix="/auth/oauth",
tags=["auth"], tags=["auth"],
) )
# Need basic auth router for `logout` endpoint # Need basic auth router for `logout` endpoint
include_router_with_global_prefix_prepended( include_router_with_global_prefix_prepended(
application, application,