From 6bd610a7736ba02cb40bc91b4e0d8278bd43b712 Mon Sep 17 00:00:00 2001 From: Edwin Jose Date: Thu, 4 Sep 2025 15:22:13 -0400 Subject: [PATCH] making tasks of init sequential --- src/main.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/src/main.py b/src/main.py index 39235541..90988a24 100644 --- a/src/main.py +++ b/src/main.py @@ -199,7 +199,7 @@ async def init_index_when_ready(): async def ingest_default_documents_when_ready(services): """Scan the local documents folder and ingest files like a non-auth upload.""" try: - + logger.info("Ingesting default documents when ready") base_dir = os.path.abspath(os.path.join(os.getcwd(), "documents")) if not os.path.isdir(base_dir): logger.info("Default documents directory not found; skipping ingestion", base_dir=base_dir) @@ -261,6 +261,11 @@ async def ingest_default_documents_when_ready(services): except Exception as e: logger.error("Default documents ingestion failed", error=str(e)) +async def startup_tasks(services): + """Startup tasks""" + logger.info("Starting startup tasks") + await init_index() + await ingest_default_documents_when_ready(services) async def initialize_services(): """Initialize all services and their dependencies""" @@ -703,13 +708,9 @@ async def create_app(): @app.on_event("startup") async def startup_event(): # Start index initialization in background to avoid blocking OIDC endpoints - t1 = asyncio.create_task(init_index_when_ready()) + t1 = asyncio.create_task(startup_tasks(services)) app.state.background_tasks.add(t1) t1.add_done_callback(app.state.background_tasks.discard) - # Start default documents ingestion in background - t2 = asyncio.create_task(ingest_default_documents_when_ready(services)) - app.state.background_tasks.add(t2) - t2.add_done_callback(app.state.background_tasks.discard) # Add shutdown event handler @app.on_event("shutdown")