Feat: Ask Amount for Each Subnet When Staking to Multiple

[Mobile-Crest]

Ask Amount for Each Subnet When Staking to Multiple

Summary

Implements per-subnet amount prompting when staking to multiple netuids, improving user control and transparency over stake distribution.

Changes

CLI (bittensor_cli/cli.py)

• When multiple netuids are provided (e.g., --netuids 1,2,3), the CLI now prompts for an amount for each subnet individually
• Each prompt displays the remaining balance after previous stakes
• Single netuid behavior remains unchanged
• Empty/None netuids (all subnets) continues to ask for a single amount per netuid

Stake Add Function (src/commands/stake/add.py)

• Updated function signature to accept amount: Union[float, list[float]]
• Added logic to handle both single amounts and per-netuid amount lists
• Maintains backward compatibility with existing single-amount behavior

Example Usage

Before:

btcli stake add --netuids 1,2,3
# Prompted once: "Amount to stake (TAO τ)"
# Same amount applied to all subnets

After:

btcli stake add --netuids 1,2,3
# Prompted for each:
# "Amount to stake to netuid 1 (TAO τ) (remaining balance: 100.0)"
# "Amount to stake to netuid 2 (TAO τ) (remaining balance: 90.0)"
# "Amount to stake to netuid 3 (TAO τ) (remaining balance: 80.0)"

Contribution by Gittensor, learn more at https://gittensor.io/

[Mobile-Crest]

@thewhaleking
can you please check this pull request when you get a chance? thanks

[ibraheem-abe]

@Mobile-Crest Some e2e coverage will be good for this. Maybe add this behaviour in the current add_stake test we have

[Mobile-Crest]

@Mobile-Crest Some e2e coverage will be good for this. Maybe add this behaviour in the current add_stake test we have

is it possible to e2e test user prompt?

[thewhaleking]

This is very silly to have --amount AND --amounts that do nearly-identical-but-different things. Look at how similar items are done throughout this file.

[Mobile-Crest]

This is very silly to have --amount AND --amounts that do nearly-identical-but-different things. Look at how similar items are done throughout this file.

we already have --netuid and --netuids,
I thought they pair each other

[thewhaleking]

This is very silly to have --amount AND --amounts that do nearly-identical-but-different things. Look at how similar items are done throughout this file.

we already have --netuid and --netuids, I thought they pair each other

In stake add, --netuid is simply an alias of --netuids. They are a single arg, much like --wallet-path and -p.

[Mobile-Crest]

This is very silly to have --amount AND --amounts that do nearly-identical-but-different things. Look at how similar items are done throughout this file.

we already have --netuid and --netuids, I thought they pair each other

In stake add, --netuid is simply an alias of --netuids. They are a single arg, much like --wallet-path and -p.

then should I make amount an alias of amounts arg?
if multiple netuids and single amount, apply same amount to those subnets
if multiple netuids and multiple amounts, apply corresponding amount, but item length must be same