diff --git a/app/core/worker.py b/app/core/worker.py index b546b70..fd32e34 100644 --- a/app/core/worker.py +++ b/app/core/worker.py @@ -28,21 +28,30 @@ async def on_shutdown(ctx: dict[str, Any]) -> None: logger.info("ARQ worker shutting down...") -class WorkerSettings: - """ARQ worker settings. +# Import job functions directly so ARQ registers them by __name__ +from app.plugins.builtins.unified_search.jobs import ( + index_mails, + index_file, + index_contact, + index_company, + index_event, + reindex, + embedding_batch, +) +from app.plugins.builtins.ai_proactive.jobs import deep_analysis - functions must be dotted module paths to async callables, - e.g. 'app.plugins.builtins.unified_search.jobs.reindex' - """ + +class WorkerSettings: + """ARQ worker settings.""" functions = [ - "app.plugins.builtins.unified_search.jobs.index_mails", - "app.plugins.builtins.unified_search.jobs.index_file", - "app.plugins.builtins.unified_search.jobs.index_contact", - "app.plugins.builtins.unified_search.jobs.index_company", - "app.plugins.builtins.unified_search.jobs.index_event", - "app.plugins.builtins.unified_search.jobs.reindex", - "app.plugins.builtins.unified_search.jobs.embedding_batch", - "app.plugins.builtins.ai_proactive.jobs.deep_analysis", + index_mails, + index_file, + index_contact, + index_company, + index_event, + reindex, + embedding_batch, + deep_analysis, ] redis_settings = _get_redis_settings() on_startup = on_startup