From f3aaca139ba196055c99edb7cab403d9aa3fbc5a Mon Sep 17 00:00:00 2001
From: Zamil Majdy <zamil.majdy@agpt.co>
Date: Thu, 10 Apr 2025 22:08:35 +0200
Subject: [PATCH] feat(platform): Move scheduler from RestService pod to
 WebsocketService pod to make it singleton

---
 autogpt_platform/backend/backend/rest.py     | 3 +--
 autogpt_platform/backend/backend/ws.py       | 2 ++
 autogpt_platform/docker-compose.platform.yml | 2 +-
 3 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/autogpt_platform/backend/backend/rest.py b/autogpt_platform/backend/backend/rest.py
index a20df1091d..0fb1eed875 100644
--- a/autogpt_platform/backend/backend/rest.py
+++ b/autogpt_platform/backend/backend/rest.py
@@ -1,5 +1,5 @@
 from backend.app import run_processes
-from backend.executor import DatabaseManager, Scheduler
+from backend.executor import DatabaseManager
 from backend.notifications.notifications import NotificationManager
 from backend.server.rest_api import AgentServer
 
@@ -11,7 +11,6 @@ def main():
     run_processes(
         NotificationManager(),
         DatabaseManager(),
-        Scheduler(),
         AgentServer(),
     )
 
diff --git a/autogpt_platform/backend/backend/ws.py b/autogpt_platform/backend/backend/ws.py
index 3b15a60eb0..ed5dc88b15 100644
--- a/autogpt_platform/backend/backend/ws.py
+++ b/autogpt_platform/backend/backend/ws.py
@@ -1,4 +1,5 @@
 from backend.app import run_processes
+from backend.executor import Scheduler
 from backend.server.ws_api import WebsocketServer
 
 
@@ -6,6 +7,7 @@ def main():
     """
     Run all the processes required for the AutoGPT-server WebSocket API.
     """
+    run_processes(Scheduler())
     run_processes(WebsocketServer())
 
 
diff --git a/autogpt_platform/docker-compose.platform.yml b/autogpt_platform/docker-compose.platform.yml
index 1197b6def0..0e50711b84 100644
--- a/autogpt_platform/docker-compose.platform.yml
+++ b/autogpt_platform/docker-compose.platform.yml
@@ -96,7 +96,6 @@ services:
     ports:
       - "8006:8006"
       - "8007:8007"
-      - "8003:8003" # execution scheduler
     networks:
       - app-network
 
@@ -180,6 +179,7 @@ services:
 
     ports:
       - "8001:8001"
+      - "8003:8003" # execution scheduler
     networks:
       - app-network