Skip to content

Commit 1407652

Browse files
committed
nit
1 parent 2758ff7 commit 1407652

File tree

1 file changed

+5
-4
lines changed
  • backend/danswer/background/celery/apps

1 file changed

+5
-4
lines changed

backend/danswer/background/celery/apps/beat.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13,10 +13,11 @@
1313
from danswer.utils.logger import setup_logger
1414
from danswer.utils.variable_functionality import fetch_versioned_implementation
1515

16+
logger = setup_logger()
17+
1618

1719
celery_app = Celery(__name__)
1820
celery_app.config_from_object("danswer.background.celery.configs.beat")
19-
logger = setup_logger(__name__)
2021

2122

2223
class DynamicTenantScheduler(PersistentScheduler):
@@ -104,9 +105,6 @@ def _should_update_schedule(
104105
return current_tasks != new_tasks
105106

106107

107-
celery_app.conf.beat_scheduler = DynamicTenantScheduler
108-
109-
110108
@beat_init.connect
111109
def on_beat_init(sender: Any, **kwargs: Any) -> None:
112110
logger.info("beat_init signal received.")
@@ -122,3 +120,6 @@ def on_setup_logging(
122120
loglevel: Any, logfile: Any, format: Any, colorize: Any, **kwargs: Any
123121
) -> None:
124122
app_base.on_setup_logging(loglevel, logfile, format, colorize, **kwargs)
123+
124+
125+
celery_app.conf.beat_scheduler = DynamicTenantScheduler

0 commit comments

Comments
 (0)