diff --git a/fastdeploy/splitwise/internal_adapter_utils.py b/fastdeploy/splitwise/internal_adapter_utils.py index e64e468b186..84f6a7a8c95 100644 --- a/fastdeploy/splitwise/internal_adapter_utils.py +++ b/fastdeploy/splitwise/internal_adapter_utils.py @@ -105,6 +105,15 @@ def _recv_external_module_control_instruct(self): with self.response_lock: self.recv_control_cmd_server.response_for_control_cmd(task_id_str, result) + elif task["cmd"] == "interrupt_requests": + self.engine.resource_manager.add_abort_req_ids(task["req_ids"]) + result = { + "task_id": task_id_str, + "result": {"success": True, "interrupted_req_ids": task["req_ids"]}, + } + with self.response_lock: + self.recv_control_cmd_server.response_for_control_cmd(task_id_str, result) + except Exception as e: logger.error(f"handle_control_cmd got error: {e}, {traceback.format_exc()!s}")