From 655efc36789be931af2adc865d7b1e89d9af30c1 Mon Sep 17 00:00:00 2001 From: ibraheem-opentensor Date: Fri, 14 Mar 2025 12:33:45 -0700 Subject: [PATCH 01/12] init --- .../src/bittensor/subtensor_interface.py | 59 +++++++++++++++++++ 1 file changed, 59 insertions(+) diff --git a/bittensor_cli/src/bittensor/subtensor_interface.py b/bittensor_cli/src/bittensor/subtensor_interface.py index acb5fa870..18ea8e452 100644 --- a/bittensor_cli/src/bittensor/subtensor_interface.py +++ b/bittensor_cli/src/bittensor/subtensor_interface.py @@ -1435,3 +1435,62 @@ async def get_owned_hotkeys( ) return [decode_account_id(hotkey[0]) for hotkey in owned_hotkeys or []] + + +async def get_stake_fee( + self, + origin_hotkey_ss58: Optional[str], + origin_netuid: Optional[int], + origin_coldkey_ss58: str, + destination_hotkey_ss58: Optional[str], + destination_netuid: Optional[int], + destination_coldkey_ss58: str, + amount: int, + block_hash: Optional[str] = None, +) -> Balance: + """ + Calculates the fee for a staking operation. + + :param origin_hotkey_ss58: SS58 address of source hotkey (None for new stake) + :param origin_netuid: Netuid of source subnet (None for new stake) + :param origin_coldkey_ss58: SS58 address of source coldkey + :param destination_hotkey_ss58: SS58 address of destination hotkey (None for removing stake) + :param destination_netuid: Netuid of destination subnet (None for removing stake) + :param destination_coldkey_ss58: SS58 address of destination coldkey + :param amount: Amount of stake to transfer in RAO + :param block_hash: Optional block hash at which to perform the calculation + + :return: The calculated stake fee as a Balance object + + Example scenarios: + - Adding stake: origin_hotkey=None, origin_netuid=None + - Removing stake: destination_hotkey=None, destination_netuid=None + - Moving stake between subnets: Both origin and destination parameters provided + """ + + origin = ( + (origin_hotkey_ss58, origin_netuid) + if origin_hotkey_ss58 is not None and origin_netuid is not None + else None + ) + + destination = ( + (destination_hotkey_ss58, destination_netuid) + if destination_hotkey_ss58 is not None and destination_netuid is not None + else None + ) + + result = await self.query_runtime_api( + runtime_api="StakeInfoRuntimeApi", + method="get_stake_fee", + params=[ + origin, + origin_coldkey_ss58, + destination, + destination_coldkey_ss58, + amount, + ], + block_hash=block_hash, + ) + + return Balance.from_rao(result if result is not None else 0) From 67994978dc07d95b8167416f74a1aa689dbe6314 Mon Sep 17 00:00:00 2001 From: ibraheem-opentensor Date: Fri, 14 Mar 2025 12:33:59 -0700 Subject: [PATCH 02/12] add api examples --- .../src/bittensor/subtensor_interface.py | 28 +++++++++++++------ 1 file changed, 20 insertions(+), 8 deletions(-) diff --git a/bittensor_cli/src/bittensor/subtensor_interface.py b/bittensor_cli/src/bittensor/subtensor_interface.py index 18ea8e452..9c0bb6cd3 100644 --- a/bittensor_cli/src/bittensor/subtensor_interface.py +++ b/bittensor_cli/src/bittensor/subtensor_interface.py @@ -1462,18 +1462,30 @@ async def get_stake_fee( :return: The calculated stake fee as a Balance object - Example scenarios: - - Adding stake: origin_hotkey=None, origin_netuid=None - - Removing stake: destination_hotkey=None, destination_netuid=None - - Moving stake between subnets: Both origin and destination parameters provided + When to use None: + + 1. Adding new stake (default fee): + - origin_hotkey_ss58 = None + - origin_netuid = None + - All other fields required + + 2. Removing stake (default fee): + - destination_hotkey_ss58 = None + - destination_netuid = None + - All other fields required + + For all other operations, no None values - provide all parameters: + 3. Moving between subnets + 4. Moving between hotkeys + 5. Moving between coldkeys """ origin = ( - (origin_hotkey_ss58, origin_netuid) - if origin_hotkey_ss58 is not None and origin_netuid is not None + (origin_hotkey_ss58, origin_netuid) + if origin_hotkey_ss58 is not None and origin_netuid is not None else None ) - + destination = ( (destination_hotkey_ss58, destination_netuid) if destination_hotkey_ss58 is not None and destination_netuid is not None @@ -1492,5 +1504,5 @@ async def get_stake_fee( ], block_hash=block_hash, ) - + return Balance.from_rao(result if result is not None else 0) From 17cbe783d15ad076283c9bf0ec72b6368af28690 Mon Sep 17 00:00:00 2001 From: ibraheem-opentensor Date: Fri, 14 Mar 2025 16:02:54 -0700 Subject: [PATCH 03/12] get_stake_fee added --- .../src/bittensor/subtensor_interface.py | 133 +++++++++--------- 1 file changed, 65 insertions(+), 68 deletions(-) diff --git a/bittensor_cli/src/bittensor/subtensor_interface.py b/bittensor_cli/src/bittensor/subtensor_interface.py index 9c0bb6cd3..c82b71d54 100644 --- a/bittensor_cli/src/bittensor/subtensor_interface.py +++ b/bittensor_cli/src/bittensor/subtensor_interface.py @@ -36,6 +36,8 @@ decode_hex_identity_dict, validate_chain_endpoint, u16_normalized_float, + encode_account_id, + decode_account_id, ) SubstrateClass = ( @@ -1436,73 +1438,68 @@ async def get_owned_hotkeys( return [decode_account_id(hotkey[0]) for hotkey in owned_hotkeys or []] + async def get_stake_fee( + self, + origin_hotkey_ss58: Optional[str], + origin_netuid: Optional[int], + origin_coldkey_ss58: str, + destination_hotkey_ss58: Optional[str], + destination_netuid: Optional[int], + destination_coldkey_ss58: str, + amount: int, + block_hash: Optional[str] = None, + ) -> Balance: + """ + Calculates the fee for a staking operation. -async def get_stake_fee( - self, - origin_hotkey_ss58: Optional[str], - origin_netuid: Optional[int], - origin_coldkey_ss58: str, - destination_hotkey_ss58: Optional[str], - destination_netuid: Optional[int], - destination_coldkey_ss58: str, - amount: int, - block_hash: Optional[str] = None, -) -> Balance: - """ - Calculates the fee for a staking operation. - - :param origin_hotkey_ss58: SS58 address of source hotkey (None for new stake) - :param origin_netuid: Netuid of source subnet (None for new stake) - :param origin_coldkey_ss58: SS58 address of source coldkey - :param destination_hotkey_ss58: SS58 address of destination hotkey (None for removing stake) - :param destination_netuid: Netuid of destination subnet (None for removing stake) - :param destination_coldkey_ss58: SS58 address of destination coldkey - :param amount: Amount of stake to transfer in RAO - :param block_hash: Optional block hash at which to perform the calculation - - :return: The calculated stake fee as a Balance object - - When to use None: - - 1. Adding new stake (default fee): - - origin_hotkey_ss58 = None - - origin_netuid = None - - All other fields required - - 2. Removing stake (default fee): - - destination_hotkey_ss58 = None - - destination_netuid = None - - All other fields required - - For all other operations, no None values - provide all parameters: - 3. Moving between subnets - 4. Moving between hotkeys - 5. Moving between coldkeys - """ + :param origin_hotkey_ss58: SS58 address of source hotkey (None for new stake) + :param origin_netuid: Netuid of source subnet (None for new stake) + :param origin_coldkey_ss58: SS58 address of source coldkey + :param destination_hotkey_ss58: SS58 address of destination hotkey (None for removing stake) + :param destination_netuid: Netuid of destination subnet (None for removing stake) + :param destination_coldkey_ss58: SS58 address of destination coldkey + :param amount: Amount of stake to transfer in RAO + :param block_hash: Optional block hash at which to perform the calculation + + :return: The calculated stake fee as a Balance object + + When to use None: + + 1. Adding new stake (default fee): + - origin_hotkey_ss58 = None + - origin_netuid = None + - All other fields required + + 2. Removing stake (default fee): + - destination_hotkey_ss58 = None + - destination_netuid = None + - All other fields required + + For all other operations, no None values - provide all parameters: + 3. Moving between subnets + 4. Moving between hotkeys + 5. Moving between coldkeys + """ + + origin = None + if origin_hotkey_ss58 is not None and origin_netuid is not None: + origin = (origin_hotkey_ss58, origin_netuid) + + destination = None + if destination_hotkey_ss58 is not None and destination_netuid is not None: + destination = (destination_hotkey_ss58, destination_netuid) + + result = await self.query_runtime_api( + runtime_api="StakeInfoRuntimeApi", + method="get_stake_fee", + params=[ + origin, + origin_coldkey_ss58, + destination, + destination_coldkey_ss58, + amount, + ], + block_hash=block_hash, + ) - origin = ( - (origin_hotkey_ss58, origin_netuid) - if origin_hotkey_ss58 is not None and origin_netuid is not None - else None - ) - - destination = ( - (destination_hotkey_ss58, destination_netuid) - if destination_hotkey_ss58 is not None and destination_netuid is not None - else None - ) - - result = await self.query_runtime_api( - runtime_api="StakeInfoRuntimeApi", - method="get_stake_fee", - params=[ - origin, - origin_coldkey_ss58, - destination, - destination_coldkey_ss58, - amount, - ], - block_hash=block_hash, - ) - - return Balance.from_rao(result if result is not None else 0) + return Balance.from_rao(result if result is not None else 0) From 3921aa1d7eaeb7e6df77d096ac4b0a553b9eee7e Mon Sep 17 00:00:00 2001 From: ibraheem-opentensor Date: Sun, 16 Mar 2025 15:06:54 -0700 Subject: [PATCH 04/12] Dynamic fee in stake remove --- bittensor_cli/src/commands/stake/remove.py | 84 +++++++++++++++++++--- 1 file changed, 73 insertions(+), 11 deletions(-) diff --git a/bittensor_cli/src/commands/stake/remove.py b/bittensor_cli/src/commands/stake/remove.py index 07718e405..d62ff75ce 100644 --- a/bittensor_cli/src/commands/stake/remove.py +++ b/bittensor_cli/src/commands/stake/remove.py @@ -194,9 +194,25 @@ async def unstake( ) continue # Skip to the next subnet - useful when single amount is specified for all subnets - received_amount, slippage_pct, slippage_pct_float = _calculate_slippage( - subnet_info=subnet_info, amount=amount_to_unstake_as_balance + stake_fee = await subtensor.get_stake_fee( + origin_hotkey_ss58=staking_address_ss58, + origin_netuid=netuid, + origin_coldkey_ss58=wallet.coldkeypub.ss58_address, + destination_hotkey_ss58=None, + destination_netuid=None, + destination_coldkey_ss58=wallet.coldkeypub.ss58_address, + amount=amount_to_unstake_as_balance.rao, ) + + try: + received_amount, slippage_pct, slippage_pct_float = _calculate_slippage( + subnet_info=subnet_info, + amount=amount_to_unstake_as_balance, + stake_fee=stake_fee, + ) + except ValueError: + continue + total_received_amount += received_amount max_float_slippage = max(max_float_slippage, slippage_pct_float) @@ -220,6 +236,7 @@ async def unstake( str(amount_to_unstake_as_balance), # Amount to Unstake str(subnet_info.price.tao) + f"({Balance.get_unit(0)}/{Balance.get_unit(netuid)})", # Rate + str(stake_fee), # Fee str(received_amount), # Received Amount slippage_pct, # Slippage Percent ] @@ -411,6 +428,11 @@ async def unstake_all( justify="center", style=COLOR_PALETTE["POOLS"]["RATE"], ) + table.add_column( + f"Fee ({Balance.unit})", + justify="center", + style=COLOR_PALETTE["STAKE"]["STAKE_AMOUNT"], + ) table.add_column( f"Recieved ({Balance.unit})", justify="center", @@ -432,9 +454,22 @@ async def unstake_all( hotkey_display = hotkey_names.get(stake.hotkey_ss58, stake.hotkey_ss58) subnet_info = all_sn_dynamic_info.get(stake.netuid) stake_amount = stake.stake - received_amount, slippage_pct, slippage_pct_float = _calculate_slippage( - subnet_info=subnet_info, amount=stake_amount + stake_fee = await subtensor.get_stake_fee( + origin_hotkey_ss58=stake.hotkey_ss58, + origin_netuid=stake.netuid, + origin_coldkey_ss58=wallet.coldkeypub.ss58_address, + destination_hotkey_ss58=None, + destination_netuid=None, + destination_coldkey_ss58=wallet.coldkeypub.ss58_address, + amount=stake_amount.rao, ) + try: + received_amount, slippage_pct, slippage_pct_float = _calculate_slippage( + subnet_info=subnet_info, amount=stake_amount, stake_fee=stake_fee + ) + except ValueError: + continue + max_slippage = max(max_slippage, slippage_pct_float) total_received_value += received_amount @@ -444,6 +479,7 @@ async def unstake_all( str(stake_amount), str(float(subnet_info.price)) + f"({Balance.get_unit(0)}/{Balance.get_unit(stake.netuid)})", + str(stake_fee), str(received_amount), slippage_pct, ) @@ -791,12 +827,15 @@ async def _unstake_all_extrinsic( # Helpers -def _calculate_slippage(subnet_info, amount: Balance) -> tuple[Balance, str, float]: +def _calculate_slippage( + subnet_info, amount: Balance, stake_fee: Balance +) -> tuple[Balance, str, float]: """Calculate slippage and received amount for unstaking operation. Args: subnet_info: Subnet information containing price data amount: Amount being unstaked + stake_fee: Stake fee to include in slippage calculation Returns: tuple containing: @@ -804,15 +843,32 @@ def _calculate_slippage(subnet_info, amount: Balance) -> tuple[Balance, str, flo - slippage_pct: Formatted string of slippage percentage - slippage_pct_float: Float value of slippage percentage """ - received_amount, _, slippage_pct_float = subnet_info.alpha_to_tao_with_slippage( - amount - ) + received_amount, _, _ = subnet_info.alpha_to_tao_with_slippage(amount) + received_amount -= stake_fee + + if received_amount < Balance.from_tao(0): + print_error("Not enough Alpha to pay the transaction fee.") + raise ValueError if subnet_info.is_dynamic: + # Ideal amount w/o slippage + ideal_amount = subnet_info.alpha_to_tao(amount) + ideal_amount -= stake_fee + + # Total slippage including fees + total_slippage = ideal_amount - received_amount + slippage_pct_float = ( + 100 * (float(total_slippage.tao) / float(ideal_amount.tao)) + if ideal_amount.tao != 0 + else 0 + ) slippage_pct = f"{slippage_pct_float:.4f} %" else: - slippage_pct_float = 0 - slippage_pct = "[red]N/A[/red]" + # Root will only have fee-based slippage + slippage_pct_float = ( + 100 * float(stake_fee.tao) / float(amount.tao) if amount.tao != 0 else 0 + ) + slippage_pct = f"{slippage_pct_float:.4f} %" return received_amount, slippage_pct, slippage_pct_float @@ -1174,6 +1230,11 @@ def _create_unstake_table( justify="center", style=COLOR_PALETTE["POOLS"]["RATE"], ) + table.add_column( + f"Fee ({Balance.get_unit(0)})", + justify="center", + style=COLOR_PALETTE["STAKE"]["STAKE_AMOUNT"], + ) table.add_column( f"Received ({Balance.get_unit(0)})", justify="center", @@ -1227,7 +1288,8 @@ def _print_table_and_slippage( - [bold white]Hotkey[/bold white]: The ss58 address or identity of the hotkey you are unstaking from. - [bold white]Amount to Unstake[/bold white]: The stake amount you are removing from this key. - [bold white]Rate[/bold white]: The rate of exchange between TAO and the subnet's stake. - - [bold white]Received[/bold white]: The amount of free balance TAO you will receive on this subnet after slippage. + - [bold white]Fee[/bold white]: The transaction fee for this unstake operation. + - [bold white]Received[/bold white]: The amount of free balance TAO you will receive on this subnet after slippage and fees. - [bold white]Slippage[/bold white]: The slippage percentage of the unstake operation. (0% if the subnet is not dynamic i.e. root).""" safe_staking_description = """ From f5ef40fa92ddc641f45ef77b252c9ee59d02c56a Mon Sep 17 00:00:00 2001 From: ibraheem-opentensor Date: Sun, 16 Mar 2025 15:24:11 -0700 Subject: [PATCH 05/12] Dynamic pricing in movements --- bittensor_cli/src/commands/stake/move.py | 49 +++++++++++++++++++++--- 1 file changed, 43 insertions(+), 6 deletions(-) diff --git a/bittensor_cli/src/commands/stake/move.py b/bittensor_cli/src/commands/stake/move.py index 70afaa453..403dc9668 100644 --- a/bittensor_cli/src/commands/stake/move.py +++ b/bittensor_cli/src/commands/stake/move.py @@ -32,13 +32,14 @@ async def display_stake_movement_cross_subnets( origin_hotkey: str, destination_hotkey: str, amount_to_move: Balance, + stake_fee: Balance, ) -> tuple[Balance, float, str, str]: """Calculate and display slippage information""" if origin_netuid == destination_netuid: subnet = await subtensor.subnet(origin_netuid) received_amount_tao = subnet.alpha_to_tao(amount_to_move) - received_amount_tao -= MIN_STAKE_FEE + received_amount_tao -= stake_fee if received_amount_tao < Balance.from_tao(0): print_error("Not enough Alpha to pay the transaction fee.") @@ -46,7 +47,7 @@ async def display_stake_movement_cross_subnets( received_amount = subnet.tao_to_alpha(received_amount_tao) slippage_pct_float = ( - 100 * float(MIN_STAKE_FEE) / float(MIN_STAKE_FEE + received_amount_tao) + 100 * float(stake_fee) / float(stake_fee + received_amount_tao) if received_amount_tao != 0 else 0 ) @@ -67,7 +68,7 @@ async def display_stake_movement_cross_subnets( received_amount_tao, _, _ = dynamic_origin.alpha_to_tao_with_slippage( amount_to_move ) - received_amount_tao -= MIN_STAKE_FEE + received_amount_tao -= stake_fee received_amount, _, _ = dynamic_destination.tao_to_alpha_with_slippage( received_amount_tao ) @@ -135,6 +136,11 @@ async def display_stake_movement_cross_subnets( justify="center", style=COLOR_PALETTE["POOLS"]["TAO_EQUIV"], ) + table.add_column( + "Fee (τ)", + justify="center", + style=COLOR_PALETTE["STAKE"]["STAKE_AMOUNT"], + ) table.add_column( "slippage", justify="center", @@ -149,13 +155,11 @@ async def display_stake_movement_cross_subnets( str(amount_to_move), price_str, str(received_amount), + str(stake_fee), str(slippage_pct), ) console.print(table) - # console.print( - # f"[dim]A fee of {MIN_STAKE_FEE} applies.[/dim]" - # ) # Display slippage warning if necessary if slippage_pct_float > 5: @@ -513,6 +517,16 @@ async def move_stake( ) return False + stake_fee = await subtensor.get_stake_fee( + origin_hotkey_ss58=origin_hotkey, + origin_netuid=origin_netuid, + origin_coldkey_ss58=wallet.coldkeypub.ss58_address, + destination_hotkey_ss58=destination_hotkey, + destination_netuid=destination_netuid, + destination_coldkey_ss58=wallet.coldkeypub.ss58_address, + amount=amount_to_move_as_balance.rao, + ) + # Slippage warning if prompt: try: @@ -523,6 +537,7 @@ async def move_stake( origin_hotkey=origin_hotkey, destination_hotkey=destination_hotkey, amount_to_move=amount_to_move_as_balance, + stake_fee=stake_fee, ) except ValueError: return False @@ -686,6 +701,16 @@ async def transfer_stake( ) return False + stake_fee = await subtensor.get_stake_fee( + origin_hotkey_ss58=origin_hotkey, + origin_netuid=origin_netuid, + origin_coldkey_ss58=wallet.coldkeypub.ss58_address, + destination_hotkey_ss58=origin_hotkey, + destination_netuid=dest_netuid, + destination_coldkey_ss58=dest_coldkey_ss58, + amount=amount_to_transfer.rao, + ) + # Slippage warning if prompt: try: @@ -696,6 +721,7 @@ async def transfer_stake( origin_hotkey=origin_hotkey, destination_hotkey=origin_hotkey, amount_to_move=amount_to_transfer, + stake_fee=stake_fee, ) except ValueError: return False @@ -844,6 +870,16 @@ async def swap_stake( ) return False + stake_fee = await subtensor.get_stake_fee( + origin_hotkey_ss58=hotkey_ss58, + origin_netuid=origin_netuid, + origin_coldkey_ss58=wallet.coldkeypub.ss58_address, + destination_hotkey_ss58=hotkey_ss58, + destination_netuid=destination_netuid, + destination_coldkey_ss58=wallet.coldkeypub.ss58_address, + amount=amount_to_swap.rao, + ) + # Slippage warning if prompt: try: @@ -854,6 +890,7 @@ async def swap_stake( origin_hotkey=hotkey_ss58, destination_hotkey=hotkey_ss58, amount_to_move=amount_to_swap, + stake_fee=stake_fee, ) except ValueError: return False From 76fae6a24a18a8d1df81b0f883042dbbbfabae16 Mon Sep 17 00:00:00 2001 From: ibraheem-opentensor Date: Sun, 16 Mar 2025 15:26:09 -0700 Subject: [PATCH 06/12] Removes unused imports --- bittensor_cli/src/bittensor/subtensor_interface.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/bittensor_cli/src/bittensor/subtensor_interface.py b/bittensor_cli/src/bittensor/subtensor_interface.py index c82b71d54..ea47fd1e4 100644 --- a/bittensor_cli/src/bittensor/subtensor_interface.py +++ b/bittensor_cli/src/bittensor/subtensor_interface.py @@ -36,8 +36,6 @@ decode_hex_identity_dict, validate_chain_endpoint, u16_normalized_float, - encode_account_id, - decode_account_id, ) SubstrateClass = ( From 3960c7118c64b4d2bf71dd3101f4ed4815084296 Mon Sep 17 00:00:00 2001 From: ibraheem-opentensor Date: Mon, 17 Mar 2025 09:37:51 -0700 Subject: [PATCH 07/12] Adds dynamic fee to stake add --- bittensor_cli/src/commands/stake/add.py | 52 ++++++++++++++++++++----- 1 file changed, 42 insertions(+), 10 deletions(-) diff --git a/bittensor_cli/src/commands/stake/add.py b/bittensor_cli/src/commands/stake/add.py index 6d156f522..2451d51b2 100644 --- a/bittensor_cli/src/commands/stake/add.py +++ b/bittensor_cli/src/commands/stake/add.py @@ -282,10 +282,24 @@ async def stake_extrinsic( return False remaining_wallet_balance -= amount_to_stake - # Calculate slippage - received_amount, slippage_pct, slippage_pct_float, rate = ( - _calculate_slippage(subnet_info, amount_to_stake) + stake_fee = await subtensor.get_stake_fee( + origin_hotkey_ss58=None, + origin_netuid=None, + origin_coldkey_ss58=wallet.coldkeypub.ss58_address, + destination_hotkey_ss58=hotkey[1], + destination_netuid=netuid, + destination_coldkey_ss58=wallet.coldkeypub.ss58_address, + amount=amount_to_stake.rao, ) + + # Calculate slippage + try: + received_amount, slippage_pct, slippage_pct_float, rate = ( + _calculate_slippage(subnet_info, amount_to_stake, stake_fee) + ) + except ValueError: + return False + max_slippage = max(slippage_pct_float, max_slippage) # Add rows for the table @@ -296,6 +310,7 @@ async def stake_extrinsic( str(rate) + f" {Balance.get_unit(netuid)}/{Balance.get_unit(0)} ", # rate str(received_amount.set_unit(netuid)), # received + str(stake_fee), # fee str(slippage_pct), # slippage ] @@ -531,6 +546,11 @@ def _define_stake_table( justify="center", style=COLOR_PALETTE["POOLS"]["TAO_EQUIV"], ) + table.add_column( + "Fee (τ)", + justify="center", + style=COLOR_PALETTE["STAKE"]["STAKE_AMOUNT"], + ) table.add_column( "Slippage", justify="center", style=COLOR_PALETTE["STAKE"]["SLIPPAGE_PERCENT"] ) @@ -585,29 +605,41 @@ def _print_table_and_slippage(table: Table, max_slippage: float, safe_staking: b def _calculate_slippage( - subnet_info, amount: Balance + subnet_info, amount: Balance, stake_fee: Balance ) -> tuple[Balance, str, float, str]: """Calculate slippage when adding stake. Args: subnet_info: Subnet dynamic info amount: Amount being staked + stake_fee: Transaction fee for the stake operation Returns: tuple containing: - - received_amount: Amount received after slippage + - received_amount: Amount received after slippage and fees - slippage_str: Formatted slippage percentage string - slippage_float: Raw slippage percentage value + - rate: Exchange rate string """ - received_amount, _, slippage_pct_float = subnet_info.tao_to_alpha_with_slippage( - amount - ) + amount_after_fee = amount - stake_fee + + if amount_after_fee < 0: + print_error("You don't have enough balance to cover the stake fee.") + raise ValueError() + + received_amount, _, _ = subnet_info.tao_to_alpha_with_slippage(amount_after_fee) + if subnet_info.is_dynamic: + ideal_amount = subnet_info.tao_to_alpha(amount) + total_slippage = ideal_amount - received_amount + slippage_pct_float = 100 * (total_slippage.tao / ideal_amount.tao) slippage_str = f"{slippage_pct_float:.4f} %" rate = f"{(1 / subnet_info.price.tao or 1):.4f}" else: - slippage_pct_float = 0 - slippage_str = f"[{COLOR_PALETTE['STAKE']['SLIPPAGE_TEXT']}]N/A[/{COLOR_PALETTE['STAKE']['SLIPPAGE_TEXT']}]" + slippage_pct_float = ( + 100 * float(stake_fee.tao) / float(amount.tao) if amount.tao != 0 else 0 + ) + slippage_str = f"{slippage_pct_float:.4f} %" rate = "1" return received_amount, slippage_str, slippage_pct_float, rate From 9ac84e4791f2f87e69ecd6ef281973349a1c73b2 Mon Sep 17 00:00:00 2001 From: ibraheem-opentensor Date: Mon, 17 Mar 2025 09:39:27 -0700 Subject: [PATCH 08/12] Updates e2e tests to devnet ready --- .github/workflows/e2e-subtensor-tests.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/e2e-subtensor-tests.yml b/.github/workflows/e2e-subtensor-tests.yml index d66eadbee..9d2e97cc1 100644 --- a/.github/workflows/e2e-subtensor-tests.yml +++ b/.github/workflows/e2e-subtensor-tests.yml @@ -61,7 +61,7 @@ jobs: - name: Setup subtensor repo working-directory: ${{ github.workspace }}/subtensor - run: git checkout testnet + run: git checkout devnet-ready - name: Install Python dependencies run: python3 -m pip install -e . pytest From e9e0e7003488db0eaa612fab994b136c1f5cfb5a Mon Sep 17 00:00:00 2001 From: ibraheem-opentensor Date: Mon, 17 Mar 2025 09:57:51 -0700 Subject: [PATCH 09/12] Update remove slippage calc --- bittensor_cli/src/commands/stake/remove.py | 1 - 1 file changed, 1 deletion(-) diff --git a/bittensor_cli/src/commands/stake/remove.py b/bittensor_cli/src/commands/stake/remove.py index d62ff75ce..ff7daa1e3 100644 --- a/bittensor_cli/src/commands/stake/remove.py +++ b/bittensor_cli/src/commands/stake/remove.py @@ -853,7 +853,6 @@ def _calculate_slippage( if subnet_info.is_dynamic: # Ideal amount w/o slippage ideal_amount = subnet_info.alpha_to_tao(amount) - ideal_amount -= stake_fee # Total slippage including fees total_slippage = ideal_amount - received_amount From e1a751f9eda94ee9aa1de108ecd7f56fa6a19d04 Mon Sep 17 00:00:00 2001 From: ibraheem-opentensor Date: Mon, 17 Mar 2025 12:52:38 -0700 Subject: [PATCH 10/12] Bumps async substrate --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index d7e557dae..4c9f3d3b6 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -16,7 +16,7 @@ requires-python = ">=3.9,<3.13" dependencies = [ "wheel", "async-property==0.2.2", - "async-substrate-interface>=1.0.7", + "async-substrate-interface>=1.0.8", "aiohttp~=3.10.2", "backoff~=2.2.1", "GitPython>=3.0.0", From ac1f249191a91f10d3579a96bd633ebc86869a7f Mon Sep 17 00:00:00 2001 From: ibraheem-opentensor Date: Mon, 17 Mar 2025 15:38:40 -0700 Subject: [PATCH 11/12] Removes check for none --- bittensor_cli/src/bittensor/subtensor_interface.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bittensor_cli/src/bittensor/subtensor_interface.py b/bittensor_cli/src/bittensor/subtensor_interface.py index ea47fd1e4..dba786369 100644 --- a/bittensor_cli/src/bittensor/subtensor_interface.py +++ b/bittensor_cli/src/bittensor/subtensor_interface.py @@ -1500,4 +1500,4 @@ async def get_stake_fee( block_hash=block_hash, ) - return Balance.from_rao(result if result is not None else 0) + return Balance.from_rao(result) From d762c9aaa1f22606b5056eda3e5ed4f16dedf283 Mon Sep 17 00:00:00 2001 From: ibraheem-opentensor Date: Mon, 17 Mar 2025 17:33:18 -0700 Subject: [PATCH 12/12] Dummy commit --- bittensor_cli/src/commands/stake/remove.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bittensor_cli/src/commands/stake/remove.py b/bittensor_cli/src/commands/stake/remove.py index ff7daa1e3..898af9cf8 100644 --- a/bittensor_cli/src/commands/stake/remove.py +++ b/bittensor_cli/src/commands/stake/remove.py @@ -839,7 +839,7 @@ def _calculate_slippage( Returns: tuple containing: - - received_amount: Balance after slippage + - received_amount: Balance after slippage deduction - slippage_pct: Formatted string of slippage percentage - slippage_pct_float: Float value of slippage percentage """