Custom banner

backend/danswer/background/indexing/run_indexing.py

@@ -304,9 +304,7 @@ def _prepare_index_attempt(db_session: Session, index_attempt_id: int) -> IndexA
     return attempt
 
 
-def run_indexing_entrypoint(
-    index_attempt_id: int, is_ee: bool = False
-) -> None:
+def run_indexing_entrypoint(index_attempt_id: int, is_ee: bool = False) -> None:
     """Entrypoint for indexing run when using dask distributed.
     Wraps the actual logic in a `try` block so that we can catch any exceptions
     and mark the attempt as failed."""

backend/danswer/background/update.py

@@ -35,10 +35,10 @@ from danswer.db.models import IndexModelStatus
 from danswer.db.swap_index import check_index_swap
 from danswer.search.search_nlp_models import warm_up_encoders
 from danswer.utils.logger import setup_logger
+from danswer.utils.variable_functionality import global_version
 from shared_configs.configs import INDEXING_MODEL_SERVER_HOST
 from shared_configs.configs import LOG_LEVEL
 from shared_configs.configs import MODEL_SERVER_PORT
-from danswer.utils.variable_functionality import global_version
 
 logger = setup_logger()
 

backend/ee/danswer/server/enterprise_settings/models.py

@@ -9,5 +9,10 @@ class EnterpriseSettings(BaseModel):
     application_name: str | None = None
     use_custom_logo: bool = False
 
+    # custom Chat components
+    custom_header_content: str | None = None
+    custom_popup_header: str | None = None
+    custom_popup_content: str | None = None
+
     def check_validity(self) -> None:
         return