Skip to content

Commit f23430b

Browse files
pause task master simple (#246)
1 parent e34909d commit f23430b

File tree

3 files changed

+15
-1
lines changed

3 files changed

+15
-1
lines changed

controller/task_master/manager.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,3 +29,9 @@ def delete_task(org_id: str, task_id: str) -> requests.Response:
2929
f"{TASK_MASTER_URL}/task/queue",
3030
json={"orgId": str(org_id), "taskId": str(task_id)},
3131
)
32+
33+
34+
def pause_task_queue(task_queue_pause: bool) -> requests.Response:
35+
return requests.post(
36+
f"{TASK_MASTER_URL}/task/queue/pause?task_queue_pause={task_queue_pause}"
37+
)

fast_api/routes/misc.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
from controller.misc import manager as misc
1717
from controller.monitor import manager as controller_manager
1818
from controller.model_provider import manager as model_provider_manager
19+
from controller.task_master import manager as task_master_manager
1920
from submodules.model import enums
2021
from submodules.model.global_objects import customer_button as customer_button_db_go
2122
import util.user_activity
@@ -154,6 +155,13 @@ def cancel_all_running_tasks(request: Request):
154155
return pack_json_result({"data": {"cancelAllRunningTasks": {"ok": True}}})
155156

156157

158+
@router.get("/pause-task-queue")
159+
def pause_task_queue(request: Request, task_queue_pause: bool):
160+
auth.check_admin_access(request.state.info)
161+
task_master_manager.pause_task_queue(task_queue_pause)
162+
return SILENT_SUCCESS_RESPONSE
163+
164+
157165
@router.get("/all-users-activity")
158166
def get_all_users_activity(request: Request):
159167
auth.check_admin_access(request.state.info)

submodules/model

0 commit comments

Comments
 (0)