diff --git a/cobo_custody/client/mpc_client.py b/cobo_custody/client/mpc_client.py index 4eb61e5..a7c0091 100644 --- a/cobo_custody/client/mpc_client.py +++ b/cobo_custody/client/mpc_client.py @@ -333,6 +333,10 @@ def babylon_replace_staking_fee(self, request_id: str, related_request_id: str, params = {"request_id": request_id, "related_request_id": related_request_id, "fee_rate": fee_rate, "max_staking_fee": max_staking_fee} return self.request("POST", "/v1/custody/mpc/babylon/replace_staking_fee/", params) + def babylon_drop_staking(self, request_id: str, related_request_id: str, fee_rate: float, max_staking_fee: int = None): + params = {"request_id": request_id, "related_request_id": related_request_id, "fee_rate": fee_rate, "max_staking_fee": max_staking_fee} + return self.request("POST", "/v1/custody/mpc/babylon/drop_staking/", params) + def babylon_broadcast_staking_transaction(self, request_id: str): params = {"request_id": request_id} return self.request("POST", "/v1/custody/mpc/babylon/broadcast_staking_transaction/", params) diff --git a/tests/testcase/test_mpc_client.py b/tests/testcase/test_mpc_client.py index 5ef6cbf..7653c1f 100644 --- a/tests/testcase/test_mpc_client.py +++ b/tests/testcase/test_mpc_client.py @@ -1,5 +1,6 @@ import unittest import time + from cobo_custody.signer.local_signer import LocalSigner from cobo_custody.client.mpc_client import MPCClient