integrate into database manager and rest api.

Author: Swiftyos

autogpt_platform/backend/backend/executor/database.py

@@ -88,10 +88,42 @@ async def lifespan(self, app: "FastAPI"):
             logger.info(f"[{self.service_name}] ⏳ Connecting to Database...")
             await db.connect()
 
+            # Initialize SQLAlchemy if enabled (for gradual migration from Prisma)
+            if config.enable_sqlalchemy:
+                try:
+                    from backend.data import sqlalchemy as sa
+
+                    engine = sa.create_engine()
+                    sa.initialize(engine)
+                    app.state.db_engine = engine
+                    logger.info(
+                        f"[{self.service_name}] ✓ SQLAlchemy initialized "
+                        f"(pool_size={config.sqlalchemy_pool_size}, "
+                        f"max_overflow={config.sqlalchemy_max_overflow})"
+                    )
+                except Exception as e:
+                    logger.error(
+                        f"[{self.service_name}] Failed to initialize SQLAlchemy: {e}"
+                    )
+                    raise
+
             logger.info(f"[{self.service_name}] ✅ Ready")
             yield
 
             logger.info(f"[{self.service_name}] ⏳ Disconnecting Database...")
+
+            # Dispose SQLAlchemy if it was enabled
+            if config.enable_sqlalchemy:
+                try:
+                    from backend.data import sqlalchemy as sa
+
+                    await sa.dispose()
+                    logger.info(f"[{self.service_name}] ✓ SQLAlchemy disposed")
+                except Exception as e:
+                    logger.warning(
+                        f"[{self.service_name}] Error disposing SQLAlchemy: {e}"
+                    )
+
             await db.disconnect()
 
     async def health_check(self) -> str:

autogpt_platform/backend/backend/server/rest_api.py

@@ -78,6 +78,24 @@ async def lifespan_context(app: fastapi.FastAPI):
 
     await backend.data.db.connect()
 
+    # Initialize SQLAlchemy if enabled (for gradual migration from Prisma)
+    config = backend.util.settings.Config()
+    if config.enable_sqlalchemy:
+        try:
+            from backend.data import sqlalchemy as sa
+
+            engine = sa.create_engine()
+            sa.initialize(engine)
+            app.state.db_engine = engine
+            logger.info(
+                f"✓ AgentServer: SQLAlchemy initialized "
+                f"(pool_size={config.sqlalchemy_pool_size}, "
+                f"max_overflow={config.sqlalchemy_max_overflow})"
+            )
+        except Exception as e:
+            logger.error(f"Failed to initialize SQLAlchemy: {e}")
+            raise
+
     # Configure thread pool for FastAPI sync operation performance
     # CRITICAL: FastAPI automatically runs ALL sync functions in this thread pool:
     # - Any endpoint defined with 'def' (not async def)
@@ -118,6 +136,16 @@ async def lifespan_context(app: fastapi.FastAPI):
     except Exception as e:
         logger.warning(f"Error shutting down cloud storage handler: {e}")
 
+    # Dispose SQLAlchemy if it was enabled
+    if config.enable_sqlalchemy:
+        try:
+            from backend.data import sqlalchemy as sa
+
+            await sa.dispose()
+            logger.info("✓ AgentServer: SQLAlchemy disposed")
+        except Exception as e:
+            logger.warning(f"Error disposing SQLAlchemy: {e}")
+
     await backend.data.db.disconnect()
 
 

autogpt_platform/backend/backend/util/settings.py

@@ -274,6 +274,12 @@ class Config(UpdateTrackingModel["Config"], BaseSettings):
     )
 
     # SQLAlchemy Configuration
+    enable_sqlalchemy: bool = Field(
+        default=False,
+        description="Enable SQLAlchemy database connections. Set to true to enable gradual migration from Prisma to SQLAlchemy. "
+        "When disabled, only Prisma is used. When enabled, both ORMs coexist during transition.",
+    )
+
     sqlalchemy_pool_size: int = Field(
         default=10,
         ge=1,