🤖 Refactor Database Initialization for Celery Workers

src/celery_app/app.py

@@ -23,7 +23,13 @@ def setup_celery_entrypoint(app: Celery):
 def init_celery_app(*args: Any, sender: Celery, config: CeleryConfig = injected, **kwargs: Any):
     for k, v in config.items():
         setattr(sender.conf, k, v)
-    bootup(start_model_loading=False, integrations=[CeleryIntegration(propagate_traces=True)])
+
+    # Initialize without database setup for Celery workers
+    bootup(
+        start_model_loading=False,
+        integrations=[CeleryIntegration(propagate_traces=True)],
+        skip_db_init=True
+    )
     from celery_app.tasks import setup_periodic_tasks
 
     sender.on_after_finalize.connect(setup_periodic_tasks)

src/seer/bootup.py

@@ -1,5 +1,6 @@
 import logging
 
+from flask import has_app_context, current_app
 import sentry_sdk
 from psycopg import Connection
 from sentry_sdk.integrations import Integration
@@ -31,18 +32,23 @@ class DisablePreparedStatementConnection(Connection):
     pass
 
 
+
 @inject
 def bootup(
-    *, start_model_loading: bool, integrations: list[Integration], config: AppConfig = injected
+    *, start_model_loading: bool, integrations: list[Integration], config: AppConfig = injected, skip_db_init: bool = False
 ):
     initialize_sentry_sdk(integrations)
+
+
     with sentry_sdk.metrics.timing(key="seer_bootup_time"):
         initialize_logs(["seer.", "celery_app."])
         config.do_validation()
-        initialize_database()
+
+        # Only initialize database if we have an app context
+        if has_app_context():
+            initialize_database()
         initialize_models(start_model_loading)
 
-
 @inject
 def initialize_sentry_sdk(integrations: list[Integration], config: AppConfig = injected) -> None:
     def before_send(event: Event, hint: dict) -> Event | None:

src/seer/db.py

@@ -7,6 +7,7 @@
 import sqlalchemy
 from flask import Flask
 from flask_migrate import Migrate
+from flask import current_app
 from flask_sqlalchemy import SQLAlchemy
 from pgvector.sqlalchemy import Vector  # type: ignore
 from pydantic import BaseModel
@@ -35,20 +36,35 @@
 from seer.configuration import AppConfig
 from seer.dependency_injection import inject, injected
 
+# Track initialization state globally
+_db_initialized = False
 
 @inject
 def initialize_database(
     config: AppConfig = injected,
     app: Flask = injected,
 ):
+    global _db_initialized
+
+    # Only initialize if not already done
+    if _db_initialized:
+        return
+
+    # If there's already an instance registered, we can reuse it
+    if 'sqlalchemy' in current_app.extensions:
+        return
+
     app.config["SQLALCHEMY_DATABASE_URI"] = config.DATABASE_URL
     app.config["SQLALCHEMY_ENGINE_OPTIONS"] = {"connect_args": {"prepare_threshold": None}}
 
+
     db.init_app(app)
     migrate.init_app(app, db)
 
+
     with app.app_context():
         Session.configure(bind=db.engine)
+        _db_initialized = True
 
 
 class Base(DeclarativeBase):