bittensor-cli 9.0.0rc4__py3-none-any.whl → 9.0.2__py3-none-any.whl

bittensor_cli/cli.py

@@ -26,7 +26,9 @@ from bittensor_cli.src import (
     WalletValidationTypes as WV,
     Constants,
     COLOR_PALETTE,
+    HYPERPARAMS,
 )
+from bittensor_cli.version import __version__, __version_as_int__
 from bittensor_cli.src.bittensor import utils
 from bittensor_cli.src.bittensor.balances import Balance
 from async_substrate_interface.errors import SubstrateRequestException
@@ -72,21 +74,8 @@ except ImportError:
         pass
 
 
-__version__ = "9.0.0rc4"
 
 
-_core_version = re.match(r"^\d+\.\d+\.\d+", __version__).group(0)
-_version_split = _core_version.split(".")
-__version_info__ = tuple(int(part) for part in _version_split)
-_version_int_base = 1000
-assert max(__version_info__) < _version_int_base
-
-__version_as_int__: int = sum(
-    e * (_version_int_base**i) for i, e in enumerate(reversed(__version_info__))
-)
-assert __version_as_int__ < 2**31  # fits in int32
-__new_signature_version__ = 360
-
 _epilog = "Made with [bold red]:heart:[/bold red] by The Openτensor Foundaτion"
 
 np.set_printoptions(precision=8, suppress=True, floatmode="fixed")
@@ -124,7 +113,7 @@ class Options:
     )
     mnemonic = typer.Option(
         None,
-        help="Mnemonic used to regenerate your key. For example: horse cart dog ...",
+        help='Mnemonic used to regenerate your key. For example: "horse cart dog ..."',
     )
     seed = typer.Option(
         None, help="Seed hex string used to regenerate your key. For example: 0x1234..."
@@ -141,22 +130,15 @@ class Options:
     use_password = typer.Option(
         True,
         help="Set this to `True` to protect the generated Bittensor key with a password.",
-        is_flag=True,
-        flag_value=False,
     )
     public_hex_key = typer.Option(None, help="The public key in hex format.")
     ss58_address = typer.Option(
         None, "--ss58", "--ss58-address", help="The SS58 address of the coldkey."
     )
-    overwrite_coldkey = typer.Option(
-        False,
-        help="Overwrite the old coldkey with the newly generated coldkey.",
-        prompt=True,
-    )
-    overwrite_hotkey = typer.Option(
+    overwrite = typer.Option(
         False,
-        help="Overwrite the old hotkey with the newly generated hotkey.",
-        prompt=True,
+        "--overwrite/--no-overwrite",
+        help="Overwrite the existing wallet file with the new one.",
     )
     network = typer.Option(
         None,
@@ -663,7 +645,7 @@ class CLIManager:
         self.config_app.command("set")(self.set_config)
         self.config_app.command("get")(self.get_config)
         self.config_app.command("clear")(self.del_config)
-        self.config_app.command("metagraph", hidden=True)(self.metagraph_config)
+        # self.config_app.command("metagraph", hidden=True)(self.metagraph_config)
 
         # wallet commands
         self.wallet_app.command(
@@ -811,6 +793,12 @@ class CLIManager:
         self.subnets_app.command(
             "price", rich_help_panel=HELP_PANELS["SUBNETS"]["INFO"]
         )(self.subnets_price)
+        self.subnets_app.command(
+            "set-identity", rich_help_panel=HELP_PANELS["SUBNETS"]["IDENTITY"]
+        )(self.subnets_set_identity)
+        self.subnets_app.command(
+            "get-identity", rich_help_panel=HELP_PANELS["SUBNETS"]["IDENTITY"]
+        )(self.subnets_get_identity)
 
         # weights commands
         self.weights_app.command(
@@ -970,13 +958,13 @@ class CLIManager:
             finally:
                 if initiated is False:
                     asyncio.create_task(cmd).cancel()
-                if exit_early is True:
+                if (
+                    exit_early is True
+                ):  # temporarily to handle multiple run commands in one session
                     try:
                         raise typer.Exit()
                     except Exception as e:  # ensures we always exit cleanly
-                        if not isinstance(
-                            e, (typer.Exit, RuntimeError)
-                        ):  # temporarily to handle multiple run commands in one session
+                        if not isinstance(e, (typer.Exit, RuntimeError)):
                             err_console.print(f"An unknown error has occurred: {e}")
 
         return self.asyncio_runner(_run())
@@ -1563,7 +1551,7 @@ class CLIManager:
         """
         Displays all the wallets and their corresponding hotkeys that are located in the wallet path specified in the config.
 
-        The output display shows each wallet and its associated `ss58` addresses for the coldkey public key and any hotkeys. The output is presented in a hierarchical tree format, with each wallet as a root node and any associated hotkeys as child nodes. The `ss58` address is displayed for each coldkey and hotkey that is not encrypted and exists on the device.
+        The output display shows each wallet and its associated `ss58` addresses for the coldkey public key and any hotkeys. The output is presented in a hierarchical tree format, with each wallet as a root node and any associated hotkeys as child nodes. The `ss58` address (or an `<ENCRYPTED>` marker, for encrypted hotkeys) is displayed for each coldkey and hotkey that exists on the device.
 
         Upon invocation, the command scans the wallet directory and prints a list of all the wallets, indicating whether the
         public keys are available (`?` denotes unavailable or encrypted keys).
@@ -1648,7 +1636,7 @@ class CLIManager:
                 "Netuids must be a comma-separated list of ints, e.g., `--netuids 1,2,3,4`.",
             )
 
-        ask_for = [WO.NAME, WO.PATH] if not all_wallets else [WO.PATH]
+        ask_for = [WO.NAME] if not all_wallets else []
         validate = WV.WALLET if not all_wallets else WV.NONE
         wallet = self.wallet_ask(
             wallet_name, wallet_path, wallet_hotkey, ask_for=ask_for, validate=validate
@@ -1696,9 +1684,11 @@ class CLIManager:
             None,
             "--amount",
             "-a",
-            prompt=True,
             help="Amount (in TAO) to transfer.",
         ),
+        transfer_all: bool = typer.Option(
+            False, "--all", prompt=False, help="Transfer all available balance."
+        ),
         wallet_name: str = Options.wallet_name,
         wallet_path: str = Options.wallet_path,
         wallet_hotkey: str = Options.wallet_hotkey,
@@ -1734,20 +1724,25 @@ class CLIManager:
             wallet_name,
             wallet_path,
             wallet_hotkey,
-            ask_for=[WO.NAME, WO.PATH],
+            ask_for=[WO.NAME],
             validate=WV.WALLET,
         )
-
-        # For Rao games - temporarily commented out
-        # effective_network = get_effective_network(self.config, network)
-        # if is_rao_network(effective_network):
-        #     print_error("This command is disabled on the 'rao' network.")
-        #     raise typer.Exit()
-
         subtensor = self.initialize_chain(network)
+        if transfer_all and amount:
+            print_error("Cannot specify an amount and '--all' flag.")
+            raise typer.Exit()
+        elif transfer_all:
+            amount = 0
+        elif not amount:
+            amount = FloatPrompt.ask("Enter amount (in TAO) to transfer.")
         return self._run_command(
             wallets.transfer(
-                wallet, subtensor, destination_ss58_address, amount, prompt
+                wallet,
+                subtensor,
+                destination_ss58_address,
+                amount,
+                transfer_all,
+                prompt,
             )
         )
 
@@ -1933,6 +1928,7 @@ class CLIManager:
             "--max-successes",
             help="Set the maximum number of times to successfully run the faucet for this command.",
         ),
+        prompt: bool = Options.prompt,
     ):
         """
         Obtain test TAO tokens by performing Proof of Work (PoW).
@@ -1971,6 +1967,7 @@ class CLIManager:
                 output_in_place,
                 verbose,
                 max_successes,
+                prompt,
             )
         )
 
@@ -1984,6 +1981,7 @@ class CLIManager:
         json: Optional[str] = Options.json,
         json_password: Optional[str] = Options.json_password,
         use_password: Optional[bool] = Options.use_password,
+        overwrite: bool = Options.overwrite,
         quiet: bool = Options.quiet,
         verbose: bool = Options.verbose,
     ):
@@ -2030,6 +2028,7 @@ class CLIManager:
                 json,
                 json_password,
                 use_password,
+                overwrite,
             )
         )
 
@@ -2040,6 +2039,7 @@ class CLIManager:
         wallet_hotkey: Optional[str] = Options.wallet_hotkey,
         public_key_hex: Optional[str] = Options.public_hex_key,
         ss58_address: Optional[str] = Options.ss58_address,
+        overwrite: bool = Options.overwrite,
         quiet: bool = Options.quiet,
         verbose: bool = Options.verbose,
     ):
@@ -2087,7 +2087,7 @@ class CLIManager:
             rich.print("[red]Error: Invalid SS58 address or public key![/red]")
             raise typer.Exit()
         return self._run_command(
-            wallets.regen_coldkey_pub(wallet, ss58_address, public_key_hex)
+            wallets.regen_coldkey_pub(wallet, ss58_address, public_key_hex, overwrite)
         )
 
     def wallet_regen_hotkey(
@@ -2102,9 +2102,8 @@ class CLIManager:
         use_password: bool = typer.Option(
             False,  # Overriden to False
             help="Set to 'True' to protect the generated Bittensor key with a password.",
-            is_flag=True,
-            flag_value=True,
         ),
+        overwrite: bool = Options.overwrite,
         quiet: bool = Options.quiet,
         verbose: bool = Options.verbose,
     ):
@@ -2120,6 +2119,7 @@ class CLIManager:
         # Example usage:
 
         [green]$[/green] btcli wallet regen_hotkey --seed 0x1234...
+        [green]$[/green] btcli wallet regen-hotkey --mnemonic "word1 word2 ... word12"
 
         [bold]Note[/bold]: This command is essential for users who need to regenerate their hotkey, possibly for security upgrades or key recovery.
         It should be used with caution to avoid accidental overwriting of existing keys.
@@ -2143,6 +2143,7 @@ class CLIManager:
                 json,
                 json_password,
                 use_password,
+                overwrite,
             )
         )
 
@@ -2160,10 +2161,9 @@ class CLIManager:
         use_password: bool = typer.Option(
             False,  # Overriden to False
             help="Set to 'True' to protect the generated Bittensor key with a password.",
-            is_flag=True,
-            flag_value=True,
         ),
         uri: Optional[str] = Options.uri,
+        overwrite: bool = Options.overwrite,
         quiet: bool = Options.quiet,
         verbose: bool = Options.verbose,
     ):
@@ -2204,7 +2204,9 @@ class CLIManager:
         )
         if not uri:
             n_words = get_n_words(n_words)
-        return self._run_command(wallets.new_hotkey(wallet, n_words, use_password, uri))
+        return self._run_command(
+            wallets.new_hotkey(wallet, n_words, use_password, uri, overwrite)
+        )
 
     def wallet_new_coldkey(
         self,
@@ -2219,6 +2221,7 @@ class CLIManager:
         ),
         use_password: Optional[bool] = Options.use_password,
         uri: Optional[str] = Options.uri,
+        overwrite: bool = Options.overwrite,
         quiet: bool = Options.quiet,
         verbose: bool = Options.verbose,
     ):
@@ -2258,7 +2261,7 @@ class CLIManager:
         if not uri:
             n_words = get_n_words(n_words)
         return self._run_command(
-            wallets.new_coldkey(wallet, n_words, use_password, uri)
+            wallets.new_coldkey(wallet, n_words, use_password, uri, overwrite)
         )
 
     def wallet_check_ck_swap(
@@ -2294,6 +2297,7 @@ class CLIManager:
         n_words: Optional[int] = None,
         use_password: bool = Options.use_password,
         uri: Optional[str] = Options.uri,
+        overwrite: bool = Options.overwrite,
         quiet: bool = Options.quiet,
         verbose: bool = Options.verbose,
     ):
@@ -2337,12 +2341,7 @@ class CLIManager:
         if not uri:
             n_words = get_n_words(n_words)
         return self._run_command(
-            wallets.wallet_create(
-                wallet,
-                n_words,
-                use_password,
-                uri,
-            )
+            wallets.wallet_create(wallet, n_words, use_password, uri, overwrite)
         )
 
     def wallet_balance(
@@ -2873,7 +2872,7 @@ class CLIManager:
         netuid = get_optional_netuid(netuid, all_netuids)
 
         if stake_all and amount:
-            err_console.print(
+            print_error(
                 "Cannot specify an amount and 'stake-all'. Choose one or the other."
             )
             raise typer.Exit()
@@ -2883,14 +2882,14 @@ class CLIManager:
                 raise typer.Exit()
 
         if all_hotkeys and include_hotkeys:
-            err_console.print(
+            print_error(
                 "You have specified hotkeys to include and also the `--all-hotkeys` flag. The flag"
                 "should only be used standalone (to use all hotkeys) or with `--exclude-hotkeys`."
             )
             raise typer.Exit()
 
         if include_hotkeys and exclude_hotkeys:
-            err_console.print(
+            print_error(
                 "You have specified options for both including and excluding hotkeys. Select one or the other."
             )
             raise typer.Exit()
@@ -2960,24 +2959,24 @@ class CLIManager:
             )
 
         if include_hotkeys:
-            included_hotkeys = parse_to_list(
+            include_hotkeys = parse_to_list(
                 include_hotkeys,
                 str,
                 "Hotkeys must be a comma-separated list of ss58s, e.g., `--include-hotkeys 5Grw....,5Grw....`.",
                 is_ss58=True,
             )
         else:
-            included_hotkeys = []
+            include_hotkeys = []
 
         if exclude_hotkeys:
-            excluded_hotkeys = parse_to_list(
+            exclude_hotkeys = parse_to_list(
                 exclude_hotkeys,
                 str,
                 "Hotkeys must be a comma-separated list of ss58s, e.g., `--exclude-hotkeys 5Grw....,5Grw....`.",
                 is_ss58=True,
             )
         else:
-            excluded_hotkeys = []
+            exclude_hotkeys = []
 
         # TODO: Ask amount for each subnet explicitly if more than one
         if not stake_all and not amount:
@@ -3017,8 +3016,8 @@ class CLIManager:
                 amount,
                 prompt,
                 all_hotkeys,
-                included_hotkeys,
-                excluded_hotkeys,
+                include_hotkeys,
+                exclude_hotkeys,
                 safe_staking,
                 rate_tolerance,
                 allow_partial_stake,
@@ -3126,32 +3125,32 @@ class CLIManager:
         if interactive and any(
             [hotkey_ss58_address, include_hotkeys, exclude_hotkeys, all_hotkeys]
         ):
-            err_console.print(
+            print_error(
                 "Interactive mode cannot be used with hotkey selection options like --include-hotkeys, --exclude-hotkeys, --all-hotkeys, or --hotkey."
             )
             raise typer.Exit()
 
         if unstake_all and unstake_all_alpha:
-            err_console.print("Cannot specify both unstake-all and unstake-all-alpha.")
+            print_error("Cannot specify both unstake-all and unstake-all-alpha.")
             raise typer.Exit()
 
         if not interactive and not unstake_all and not unstake_all_alpha:
             netuid = get_optional_netuid(netuid, all_netuids)
             if all_hotkeys and include_hotkeys:
-                err_console.print(
+                print_error(
                     "You have specified hotkeys to include and also the `--all-hotkeys` flag. The flag"
                     " should only be used standalone (to use all hotkeys) or with `--exclude-hotkeys`."
                 )
                 raise typer.Exit()
 
             if include_hotkeys and exclude_hotkeys:
-                err_console.print(
+                print_error(
                     "You have specified both including and excluding hotkeys options. Select one or the other."
                 )
                 raise typer.Exit()
 
             if unstake_all and amount:
-                err_console.print(
+                print_error(
                     "Cannot specify both a specific amount and 'unstake-all'. Choose one or the other."
                 )
                 raise typer.Exit()
@@ -3212,11 +3211,23 @@ class CLIManager:
                 else:
                     print_error("Invalid hotkey ss58 address.")
                     raise typer.Exit()
-            else:
-                hotkey_or_ss58 = Prompt.ask(
-                    "Enter the [blue]hotkey[/blue] name or [blue]ss58 address[/blue] to unstake all from",
-                    default=self.config.get("wallet_hotkey") or defaults.wallet.hotkey,
+            elif all_hotkeys:
+                wallet = self.wallet_ask(
+                    wallet_name,
+                    wallet_path,
+                    wallet_hotkey,
+                    ask_for=[WO.NAME, WO.PATH],
                 )
+            else:
+                if not hotkey_ss58_address and not wallet_hotkey:
+                    hotkey_or_ss58 = Prompt.ask(
+                        "Enter the [blue]hotkey[/blue] name or [blue]ss58 address[/blue] to unstake all from [dim](or enter 'all' to unstake from all hotkeys)[/dim]",
+                        default=self.config.get("wallet_hotkey")
+                        or defaults.wallet.hotkey,
+                    )
+                else:
+                    hotkey_or_ss58 = hotkey_ss58_address or wallet_hotkey
+
                 if is_valid_ss58_address(hotkey_or_ss58):
                     hotkey_ss58_address = hotkey_or_ss58
                     wallet = self.wallet_ask(
@@ -3225,6 +3236,14 @@ class CLIManager:
                         wallet_hotkey,
                         ask_for=[WO.NAME, WO.PATH],
                     )
+                elif hotkey_or_ss58 == "all":
+                    all_hotkeys = True
+                    wallet = self.wallet_ask(
+                        wallet_name,
+                        wallet_path,
+                        wallet_hotkey,
+                        ask_for=[WO.NAME, WO.PATH],
+                    )
                 else:
                     wallet_hotkey = hotkey_or_ss58
                     wallet = self.wallet_ask(
@@ -3240,6 +3259,9 @@ class CLIManager:
                     subtensor=self.initialize_chain(network),
                     hotkey_ss58_address=hotkey_ss58_address,
                     unstake_all_alpha=unstake_all_alpha,
+                    all_hotkeys=all_hotkeys,
+                    include_hotkeys=include_hotkeys,
+                    exclude_hotkeys=exclude_hotkeys,
                     prompt=prompt,
                 )
             )
@@ -3265,24 +3287,20 @@ class CLIManager:
             )
 
         if include_hotkeys:
-            included_hotkeys = parse_to_list(
+            include_hotkeys = parse_to_list(
                 include_hotkeys,
                 str,
                 "Hotkeys must be a comma-separated list of ss58s or names, e.g., `--include-hotkeys hk1,hk2`.",
                 is_ss58=False,
             )
-        else:
-            included_hotkeys = []
 
         if exclude_hotkeys:
-            excluded_hotkeys = parse_to_list(
+            exclude_hotkeys = parse_to_list(
                 exclude_hotkeys,
                 str,
                 "Hotkeys must be a comma-separated list of ss58s or names, e.g., `--exclude-hotkeys hk3,hk4`.",
                 is_ss58=False,
             )
-        else:
-            excluded_hotkeys = []
 
         return self._run_command(
             remove_stake.unstake(
@@ -3290,8 +3308,8 @@ class CLIManager:
                 subtensor=self.initialize_chain(network),
                 hotkey_ss58_address=hotkey_ss58_address,
                 all_hotkeys=all_hotkeys,
-                include_hotkeys=included_hotkeys,
-                exclude_hotkeys=excluded_hotkeys,
+                include_hotkeys=include_hotkeys,
+                exclude_hotkeys=exclude_hotkeys,
                 amount=amount,
                 prompt=prompt,
                 interactive=interactive,
@@ -3745,7 +3763,6 @@ class CLIManager:
             [],
             "--proportions",
             "--prop",
-            "-p",
             help="Enter the stake weight proportions for the child hotkeys (sum should be less than or equal to 1)",
             prompt=False,
         ),
@@ -3753,9 +3770,10 @@ class CLIManager:
         wait_for_finalization: bool = Options.wait_for_finalization,
         quiet: bool = Options.quiet,
         verbose: bool = Options.verbose,
+        prompt: bool = Options.prompt,
     ):
         """
-        Set child hotkeys on specified subnets.
+        Set child hotkeys on a specified subnet (or all). Overrides currently set children.
 
         Users can specify the 'proportion' to delegate to child hotkeys (ss58 address). The sum of proportions cannot be greater than 1.
 
@@ -3792,7 +3810,7 @@ class CLIManager:
             wallet_name,
             wallet_path,
             wallet_hotkey,
-            ask_for=[WO.NAME, WO.PATH, WO.HOTKEY],
+            ask_for=[WO.NAME, WO.HOTKEY],
             validate=WV.WALLET_AND_HOTKEY,
         )
         return self._run_command(
@@ -3804,6 +3822,7 @@ class CLIManager:
                 proportions=proportions,
                 wait_for_finalization=wait_for_finalization,
                 wait_for_inclusion=wait_for_inclusion,
+                prompt=prompt,
             )
         )
 
@@ -3829,9 +3848,10 @@ class CLIManager:
         wait_for_finalization: bool = Options.wait_for_finalization,
         quiet: bool = Options.quiet,
         verbose: bool = Options.verbose,
+        prompt: bool = Options.prompt,
     ):
         """
-        Remove all children hotkeys on a specified subnet.
+        Remove all children hotkeys on a specified subnet (or all).
 
         This command is used to remove delegated authority from all child hotkeys, removing their position and influence on the subnet.
 
@@ -3844,7 +3864,7 @@ class CLIManager:
             wallet_name,
             wallet_path,
             wallet_hotkey,
-            ask_for=[WO.NAME, WO.PATH, WO.HOTKEY],
+            ask_for=[WO.NAME, WO.HOTKEY],
             validate=WV.WALLET_AND_HOTKEY,
         )
         if all_netuids and netuid:
@@ -3863,6 +3883,7 @@ class CLIManager:
                 netuid,
                 wait_for_inclusion,
                 wait_for_finalization,
+                prompt=prompt,
             )
         )
 
@@ -3918,7 +3939,7 @@ class CLIManager:
             wallet_name,
             wallet_path,
             wallet_hotkey,
-            ask_for=[WO.NAME, WO.PATH, WO.HOTKEY],
+            ask_for=[WO.NAME, WO.HOTKEY],
             validate=WV.WALLET_AND_HOTKEY,
         )
         if all_netuids and netuid:
@@ -3953,7 +3974,7 @@ class CLIManager:
         param_name: str = typer.Option(
             "", "--param", "--parameter", help="The subnet hyperparameter to set"
         ),
-        param_value: str = typer.Option(
+        param_value: Optional[str] = typer.Option(
             "", "--value", help="Value to set the hyperparameter to."
         ),
         quiet: bool = Options.quiet,
@@ -3970,13 +3991,13 @@ class CLIManager:
         """
         self.verbosity_handler(quiet, verbose)
 
-        hyperparams = self._run_command(
-            sudo.get_hyperparameters(self.initialize_chain(network), netuid),
-            exit_early=False,
-        )
-
-        if not hyperparams:
-            raise typer.Exit()
+        if not param_name or not param_value:
+            hyperparams = self._run_command(
+                sudo.get_hyperparameters(self.initialize_chain(network), netuid),
+                exit_early=False,
+            )
+            if not hyperparams:
+                raise typer.Exit()
 
         if not param_name:
             hyperparam_list = [field.name for field in fields(SubnetHyperparameters)]
@@ -3991,10 +4012,23 @@ class CLIManager:
             )
             param_name = hyperparam_list[choice - 1]
 
-        if not param_value:
-            param_value = Prompt.ask(
-                f"Enter the new value for [{COLOR_PALETTE['GENERAL']['SUBHEADING']}]{param_name}[/{COLOR_PALETTE['GENERAL']['SUBHEADING']}] in the VALUE column format"
+        if param_name in ["alpha_high", "alpha_low"]:
+            param_name = "alpha_values"
+            low_val = FloatPrompt.ask(
+                "Enter the new value for [dark_orange]alpha_low[/dark_orange]"
+            )
+            high_val = FloatPrompt.ask(
+                "Enter the new value for [dark_orange]alpha_high[/dark_orange]"
             )
+            param_value = f"{low_val},{high_val}"
+
+        if not param_value:
+            if HYPERPARAMS.get(param_name):
+                param_value = Prompt.ask(
+                    f"Enter the new value for [{COLOR_PALETTE['GENERAL']['SUBHEADING']}]{param_name}[/{COLOR_PALETTE['GENERAL']['SUBHEADING']}] in the VALUE column format"
+                )
+            else:
+                param_value = None
 
         wallet = self.wallet_ask(
             wallet_name, wallet_path, wallet_hotkey, ask_for=[WO.NAME, WO.PATH]
@@ -4060,7 +4094,9 @@ class CLIManager:
         [green]$[/green] btcli sudo proposals
         """
         self.verbosity_handler(quiet, verbose)
-        return self._run_command(sudo.proposals(self.initialize_chain(network)))
+        return self._run_command(
+            sudo.proposals(self.initialize_chain(network), verbose)
+        )
 
     def sudo_senate_vote(
         self,
@@ -4438,6 +4474,7 @@ class CLIManager:
             validate=WV.WALLET_AND_HOTKEY,
         )
         identity = prompt_for_subnet_identity(
+            current_identity={},
             subnet_name=subnet_name,
             github_repo=github_repo,
             subnet_contact=subnet_contact,
@@ -4468,6 +4505,118 @@ class CLIManager:
                     verbose=verbose,
                 )
 
+    def subnets_get_identity(
+        self,
+        network: Optional[list[str]] = Options.network,
+        netuid: int = Options.netuid,
+        quiet: bool = Options.quiet,
+        verbose: bool = Options.verbose,
+    ):
+        """
+        Get the identity information for a subnet.
+
+        This command displays the identity information of a subnet including name, GitHub repo, contact details, etc.
+
+        [green]$[/green] btcli subnets get-identity --netuid 1
+        """
+        self.verbosity_handler(quiet, verbose)
+        return self._run_command(
+            subnets.get_identity(
+                self.initialize_chain(network),
+                netuid,
+            )
+        )
+
+    def subnets_set_identity(
+        self,
+        wallet_name: str = Options.wallet_name,
+        wallet_path: str = Options.wallet_path,
+        wallet_hotkey: str = Options.wallet_hotkey,
+        network: Optional[list[str]] = Options.network,
+        netuid: int = Options.netuid,
+        subnet_name: Optional[str] = typer.Option(
+            None, "--subnet-name", "--name", help="Name of the subnet"
+        ),
+        github_repo: Optional[str] = typer.Option(
+            None, "--github-repo", "--repo", help="GitHub repository URL"
+        ),
+        subnet_contact: Optional[str] = typer.Option(
+            None,
+            "--subnet-contact",
+            "--contact",
+            "--email",
+            help="Contact email for subnet",
+        ),
+        subnet_url: Optional[str] = typer.Option(
+            None, "--subnet-url", "--url", help="Subnet URL"
+        ),
+        discord: Optional[str] = typer.Option(
+            None, "--discord-handle", "--discord", help="Discord handle"
+        ),
+        description: Optional[str] = typer.Option(
+            None, "--description", help="Description"
+        ),
+        additional_info: Optional[str] = typer.Option(
+            None, "--additional-info", help="Additional information"
+        ),
+        prompt: bool = Options.prompt,
+        quiet: bool = Options.quiet,
+        verbose: bool = Options.verbose,
+    ):
+        """
+        Set or update the identity information for a subnet.
+
+        This command allows subnet owners to set or update identity information like name, GitHub repo, contact details, etc.
+
+        [bold]Common Examples:[/bold]
+
+        1. Interactive subnet identity setting:
+        [green]$[/green] btcli subnets set-identity --netuid 1
+
+        2. Set subnet identity with specific values:
+        [green]$[/green] btcli subnets set-identity --netuid 1 --subnet-name MySubnet --github-repo https://github.com/myorg/mysubnet --subnet-contact team@mysubnet.net
+        """
+        self.verbosity_handler(quiet, verbose)
+        wallet = self.wallet_ask(
+            wallet_name,
+            wallet_path,
+            wallet_hotkey,
+            ask_for=[WO.NAME],
+            validate=WV.WALLET,
+        )
+
+        current_identity = self._run_command(
+            subnets.get_identity(
+                self.initialize_chain(network),
+                netuid,
+                f"Current Subnet {netuid}'s Identity",
+            ),
+            exit_early=False,
+        )
+        if current_identity is None:
+            raise typer.Exit()
+
+        identity = prompt_for_subnet_identity(
+            current_identity=current_identity,
+            subnet_name=subnet_name,
+            github_repo=github_repo,
+            subnet_contact=subnet_contact,
+            subnet_url=subnet_url,
+            discord=discord,
+            description=description,
+            additional=additional_info,
+        )
+
+        return self._run_command(
+            subnets.set_identity(
+                wallet,
+                self.initialize_chain(network),
+                netuid,
+                identity,
+                prompt,
+            )
+        )
+
     def subnets_pow_register(
         self,
         wallet_name: Optional[str] = Options.wallet_name,
@@ -4515,6 +4664,7 @@ class CLIManager:
             "-tbp",
             help="Set the number of threads per block for CUDA.",
         ),
+        prompt: bool = Options.prompt,
     ):
         """
         Register a neuron (a subnet validator or a subnet miner) using Proof of Work (POW).
@@ -4552,6 +4702,7 @@ class CLIManager:
                 use_cuda,
                 dev_id,
                 threads_per_block,
+                prompt=prompt,
             )
         )
 
@@ -4582,7 +4733,7 @@ class CLIManager:
             wallet_name,
             wallet_path,
             wallet_hotkey,
-            ask_for=[WO.NAME, WO.PATH, WO.HOTKEY],
+            ask_for=[WO.NAME, WO.HOTKEY],
             validate=WV.WALLET_AND_HOTKEY,
         )
         return self._run_command(
@@ -4717,6 +4868,7 @@ class CLIManager:
         ),
         quiet: bool = Options.quiet,
         verbose: bool = Options.verbose,
+        prompt: bool = Options.prompt,
     ):
         """
         Reveal weights for a specific subnet.
@@ -4788,6 +4940,7 @@ class CLIManager:
                 weights,
                 salt,
                 __version_as_int__,
+                prompt=prompt,
             )
         )
 
@@ -4813,6 +4966,7 @@ class CLIManager:
         ),
         quiet: bool = Options.quiet,
         verbose: bool = Options.verbose,
+        prompt: bool = Options.prompt,
     ):
         """
 
@@ -4883,6 +5037,7 @@ class CLIManager:
                 weights,
                 salt,
                 __version_as_int__,
+                prompt=prompt,
             )
         )