highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-07-28 05:43:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

Add global prefix to EE endpoints

Browse Source
This commit is contained in:
Weves
2024-01-17 14:26:19 -08:00
committed by Chris Weaver
parent 8bf483904d
commit fe975c3357
1 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
backend/ee/danswer/main.py
View File

@@ -13,6 +13,7 @@ from danswer.configs.app_configs import SECRET
from danswer.configs.app_configs import WEB_DOMAIN from danswer.configs.app_configs import WEB_DOMAIN
from danswer.configs.constants import AuthType from danswer.configs.constants import AuthType
from danswer.main import get_application from danswer.main import get_application
from danswer.main import include_router_with_global_prefix_prepended
from danswer.utils.logger import setup_logger from danswer.utils.logger import setup_logger
from danswer.utils.variable_functionality import global_version from danswer.utils.variable_functionality import global_version
from ee.danswer.configs.app_configs import OPENID_CONFIG_URL from ee.danswer.configs.app_configs import OPENID_CONFIG_URL
@@ -36,7 +37,8 @@ def get_ee_application() -> FastAPI:
application = get_application() application = get_application()
if AUTH_TYPE == AuthType.OIDC: if AUTH_TYPE == AuthType.OIDC:
application.include_router( include_router_with_global_prefix_prepended(
application,
fastapi_users.get_oauth_router( fastapi_users.get_oauth_router(
OpenID(OAUTH_CLIENT_ID, OAUTH_CLIENT_SECRET, OPENID_CONFIG_URL), OpenID(OAUTH_CLIENT_ID, OAUTH_CLIENT_SECRET, OPENID_CONFIG_URL),
auth_backend, auth_backend,
@@ -49,24 +51,25 @@ def get_ee_application() -> FastAPI:
tags=["auth"], tags=["auth"],
) )
# need basic auth router for `logout` endpoint # need basic auth router for `logout` endpoint
application.include_router( include_router_with_global_prefix_prepended(
application,
fastapi_users.get_auth_router(auth_backend), fastapi_users.get_auth_router(auth_backend),
prefix="/auth", prefix="/auth",
tags=["auth"], tags=["auth"],
) )
elif AUTH_TYPE == AuthType.SAML: elif AUTH_TYPE == AuthType.SAML:
application.include_router(saml_router) include_router_with_global_prefix_prepended(application, saml_router)
# RBAC / group access control # RBAC / group access control
application.include_router(user_group_router) include_router_with_global_prefix_prepended(application, user_group_router)
# Analytics endpoints # Analytics endpoints
application.include_router(analytics_router) include_router_with_global_prefix_prepended(application, analytics_router)
application.include_router(query_history_router) include_router_with_global_prefix_prepended(application, query_history_router)
# Api key management # Api key management
application.include_router(api_key_router) include_router_with_global_prefix_prepended(application, api_key_router)
# EE only backend APIs # EE only backend APIs
application.include_router(chat_query_router) include_router_with_global_prefix_prepended(application, chat_query_router)
return application return application