update add_stake

Author: ibraheem-abe

bittensor/core/extrinsics/staking.py

@@ -97,47 +97,51 @@ def add_stake_extrinsic(
         return False
 
     try:
+        call_params = {
+            "hotkey": hotkey_ss58,
+            "netuid": netuid,
+            "amount_staked": staking_balance.rao,
+        }
+
         if safe_staking:
             pool = subtensor.subnet(netuid=netuid)
             base_price = pool.price.rao
             price_with_tolerance = base_price * (1 + rate_threshold)
 
-            # For logging purposes
+            # For logging
             base_rate = pool.price.tao
             rate_with_tolerance = base_rate * (1 + rate_threshold)
 
             logging.info(
                 f":satellite: [magenta]Safe Staking to:[/magenta] "
-                f"[blue]netuid: {netuid}, amount: {staking_balance}, tolerance percentage: {rate_threshold*100}%, "
-                f"price limit: {rate_with_tolerance}, original price: {base_rate}, with partial stake: {allow_partial_stake} "
-                f"on {subtensor.network}[/blue] [magenta]...[/magenta]"
+                f"[blue]netuid: [yellow]{netuid}[/yellow], amount: [green]{staking_balance}[/green], "
+                f"tolerance percentage: [yellow]{rate_threshold*100}%[/yellow], "
+                f"price limit: [yellow]{rate_with_tolerance}[/yellow], "
+                f"original price: [green]{base_rate}[/green], "
+                f"with partial stake: [yellow]{allow_partial_stake}[/yellow] "
+                f"on [blue]{subtensor.network}[/blue][/magenta]...[/magenta]"
             )
-            call = subtensor.substrate.compose_call(
-                call_module="SubtensorModule",
-                call_function="add_stake_limit",
-                call_params={
-                    "hotkey": hotkey_ss58,
-                    "netuid": netuid,
-                    "amount_staked": staking_balance.rao,
+
+            call_params.update(
+                {
                     "limit_price": price_with_tolerance,
                     "allow_partial": allow_partial_stake,
-                },
+                }
             )
+            call_function = "add_stake_limit"
         else:
             logging.info(
                 f":satellite: [magenta]Staking to:[/magenta] "
-                f"[blue]netuid: {netuid}, amount: {staking_balance} "
-                f"on {subtensor.network}[/blue] [magenta]...[/magenta]"
-            )
-            call = subtensor.substrate.compose_call(
-                call_module="SubtensorModule",
-                call_function="add_stake",
-                call_params={
-                    "hotkey": hotkey_ss58,
-                    "amount_staked": staking_balance.rao,
-                    "netuid": netuid,
-                },
+                f"[blue]netuid: [yellow]{netuid}[/yellow], amount: [green]{staking_balance}[/green] "
+                f"on [blue]{subtensor.network}[/blue][magenta]...[/magenta]"
             )
+            call_function = "add_stake"
+
+        call = subtensor.substrate.compose_call(
+            call_module="SubtensorModule",
+            call_function=call_function,
+            call_params=call_params,
+        )
 
         staking_response, err_msg = subtensor.sign_and_send_extrinsic(
             call,