Release 3.7 subtensor integration

bittensor/_cli/__init__.py

@@ -25,6 +25,7 @@
 from . import cli_impl
 from .commands import *
 from typing import List, Optional
+from .naka_cli_impl import CLI as naka_CLI
 console = bittensor.__console__
 
 # Turn off rich console locals trace.
@@ -50,7 +51,13 @@ def __new__(
         if config == None: 
             config = cli.config(args)
         cli.check_config( config )
-        return cli_impl.CLI( config = config)
+        if config.subtensor:
+            network = config.subtensor.get('network', bittensor.defaults.subtensor.network)
+
+        if network == 'finney':
+            return cli_impl.CLI( config = config)
+        elif network == 'nakamoto':
+            return naka_CLI(config=config)
 
     @staticmethod   
     def config(args: List[str]) -> 'bittensor.config':
@@ -80,6 +87,7 @@ def config(args: List[str]) -> 'bittensor.config':
         MetagraphCommand.add_args( cmd_parsers )
         SetWeightsCommand.add_args( cmd_parsers )
         NewColdkeyCommand.add_args( cmd_parsers )
+        NewHotkeyCommand.add_args( cmd_parsers )
         ListSubnetsCommand.add_args( cmd_parsers )
         RegenHotkeyCommand.add_args( cmd_parsers )
         RegenColdkeyCommand.add_args( cmd_parsers )

bittensor/_cli/commands/list.py

@@ -85,4 +85,5 @@ def add_args( parser: argparse.ArgumentParser ):
             default=False,
         )
         list_parser.add_argument( '--no_version_checking', action='store_true', help='''Set false to stop cli version checking''', default = False )
-        bittensor.wallet.add_args( list_parser )
+        bittensor.wallet.add_args( list_parser )
+        bittensor.subtensor.add_args( list_parser )

bittensor/_cli/commands/misc.py

@@ -57,6 +57,7 @@ def add_args( parser: argparse.ArgumentParser ):
             default='None', 
         )
         help_parser.add_argument( '--no_version_checking', action='store_true', help='''Set false to stop cli version checking''', default = False )
+        bittensor.subtensor.add_args( help_parser )
 
 class UpdateCommand:
     @staticmethod
@@ -86,6 +87,7 @@ def add_args( parser: argparse.ArgumentParser ):
             default=False,
         )
         update_parser.add_argument( '--no_version_checking', action='store_true', help='''Set false to stop cli version checking''', default = False )
+        bittensor.subtensor.add_args( update_parser )
 
 class ListSubnetsCommand:
     @staticmethod
@@ -96,7 +98,7 @@ def run (cli):
 
         rows = []
         total_neurons = 0
-        
+
         for subnet in subnets:
             total_neurons += subnet.max_n
             rows.append((
@@ -109,7 +111,7 @@ def run (cli):
                 str(subnet.validator_sequence_length),
                 str(subnet.tempo),
                 str(subnet.modality),
-                str(list(subnet.connection_requirements.keys())),
+                str([cr[1] for cr in subnet.connection_requirements]),
                 str(subnet.emission_value),
             ))
 

bittensor/_cli/commands/query.py

@@ -28,23 +28,38 @@ def run (cli):
         subtensor = bittensor.subtensor( config = cli.config )
 
         # Verify subnet exists
+        if not hasattr(cli.config, 'netuid'):
+            bittensor.__console__.print(f"[red]Please specify subnet with --netuid.[/red]")
+            sys.exit(1)
+
         if not subtensor.subnet_exists( netuid = cli.config.netuid ):
-            bittensor.__console__.print(f"[red]Subnet {cli.config.netuid} does not exist[/red]")
+            bittensor.__console__.print(f"[red]Subnet {cli.config.netuid} does not exist.[/red]")
             sys.exit(1)
 
         dendrite = bittensor.dendrite( wallet = wallet )
         stats = {}
         for uid in cli.config.uids:
             neuron = subtensor.neuron_for_uid( uid = uid, netuid = cli.config.netuid )
             endpoint = bittensor.endpoint.from_neuron( neuron )
-            _, c, t = dendrite.forward_text( endpoints = endpoint, inputs = 'hello world')
-            latency = "{}".format(t.tolist()[0]) if c.tolist()[0] == 1 else 'N/A'
-            bittensor.__console__.print("\tUid: [bold white]{}[/bold white]\n\tLatency: [bold white]{}[/bold white]\n\tCode: [bold {}]{}[/bold {}]\n\n".format(uid, latency, bittensor.utils.codes.code_to_loguru_color( c.item() ), bittensor.utils.codes.code_to_string( c.item() ), bittensor.utils.codes.code_to_loguru_color( c.item() )), highlight=True)
+            _, c, t = dendrite.text( endpoints = endpoint, inputs = 'hello world', synapses = [bittensor.synapse.TextCausalLMNext()])
+            latency = "{}".format(t[0].tolist()[0]) if c[0].tolist()[0] == 1 else 'N/A'
+            bittensor.__console__.print("\tUid: [bold white]{}[/bold white]\n\tLatency: [bold white]{}[/bold white]\n\tCode: [bold {}]{}[/bold {}]\n\n".format(
+                    uid, 
+                    latency, 
+                    bittensor.utils.codes.code_to_loguru_color( c[0].item() ), 
+                    bittensor.utils.codes.code_to_string( c[0].item() ), 
+                    bittensor.utils.codes.code_to_loguru_color( c[0].item() )
+                ), highlight=True)
             stats[uid] = latency
         print (stats)
+        dendrite.__del__()
 
     @staticmethod
     def check_config( config: 'bittensor.Config' ):
+
+        if config.subtensor.get('network') == bittensor.defaults.subtensor.network and not config.no_prompt:
+            config.subtensor.network = Prompt.ask("Enter subtensor network", choices=bittensor.__networks__, default = bittensor.defaults.subtensor.network)
+
         if config.wallet.get('name') == bittensor.defaults.wallet.name and not config.no_prompt:
             wallet_name = Prompt.ask("Enter wallet name", default = bittensor.defaults.wallet.name)
             config.wallet.name = str(wallet_name)
@@ -85,6 +100,12 @@ def add_args( parser: argparse.ArgumentParser ):
             help='''Set true to avoid prompting the user.''',
             default=False,
         )
+        query_parser.add_argument(
+            '--netuid',
+            type=int,
+            help='netuid for subnet to serve this neuron on',
+            default=argparse.SUPPRESS,
+        )
         query_parser.add_argument( '--no_version_checking', action='store_true', help='''Set false to stop cli version checking''', default = False )
         bittensor.wallet.add_args( query_parser )
         bittensor.subtensor.add_args( query_parser )

bittensor/_cli/commands/utils.py

@@ -36,28 +36,28 @@ def check_choice( self, value: str ) -> bool:
 
 
 def check_netuid_set( config: 'bittensor.Config', subtensor: 'bittensor.Subtensor', allow_none: bool = False ):
-
-    all_netuids = [str(netuid) for netuid in subtensor.get_subnets()]
-    if len(all_netuids) == 0:
-        console.print(":cross_mark:[red]There are no open networks.[/red]")
-        sys.exit()
-
-    # Make sure netuid is set.
-    if config.get('netuid', 'notset') == 'notset':
-        if not config.no_prompt:
-            netuid = IntListPrompt.ask("Enter netuid", choices=all_netuids, default=str(all_netuids[0]))
+    if subtensor.network =='finney':
+        all_netuids = [str(netuid) for netuid in subtensor.get_subnets()]
+        if len(all_netuids) == 0:
+            console.print(":cross_mark:[red]There are no open networks.[/red]")
+            sys.exit()
+
+        # Make sure netuid is set.
+        if config.get('netuid', 'notset') == 'notset':
+            if not config.no_prompt:
+                netuid = IntListPrompt.ask("Enter netuid", choices=all_netuids, default=str(all_netuids[0]))
+            else:
+                netuid = str(bittensor.defaults.netuid) if not allow_none else 'None'
         else:
-            netuid = str(bittensor.defaults.netuid) if not allow_none else 'None'
-    else:
-        netuid = config.netuid
-
-    if isinstance(netuid, str) and netuid.lower() in ['none'] and allow_none:
-        config.netuid = None
-    else:
-        try:
-            config.netuid = int(netuid)
-        except ValueError:
-            raise ValueError('netuid must be an integer or "None" (if applicable)')
+            netuid = config.netuid
+
+        if isinstance(netuid, str) and netuid.lower() in ['none'] and allow_none:
+            config.netuid = None
+        else:
+            try:
+                config.netuid = int(netuid)
+            except ValueError:
+                raise ValueError('netuid must be an integer or "None" (if applicable)')
 
 
 def check_for_cuda_reg_config( config: 'bittensor.Config' ) -> None:

bittensor/_cli/commands/wallets.py

@@ -118,6 +118,7 @@ def add_args( parser: argparse.ArgumentParser ):
             help='''Overwrite the old coldkey with the newly generated coldkey'''
         )
         bittensor.wallet.add_args( regen_coldkey_parser )
+        bittensor.subtensor.add_args( regen_coldkey_parser )
 
 
 class RegenColdkeypubCommand:
@@ -180,6 +181,7 @@ def add_args( parser: argparse.ArgumentParser ):
             help='''Overwrite the old coldkeypub file with the newly generated coldkeypub'''
         )
         bittensor.wallet.add_args( regen_coldkeypub_parser )
+        bittensor.subtensor.add_args( regen_coldkeypub_parser )
 
 class RegenHotkeyCommand:
 
@@ -282,6 +284,7 @@ def add_args( parser: argparse.ArgumentParser ):
             help='''Overwrite the old hotkey with the newly generated hotkey'''
         )
         bittensor.wallet.add_args( regen_hotkey_parser )
+        bittensor.subtensor.add_args( regen_hotkey_parser )
 
 
 
@@ -339,6 +342,7 @@ def add_args( parser: argparse.ArgumentParser ):
             help='''Overwrite the old hotkey with the newly generated hotkey'''
         )
         bittensor.wallet.add_args( new_hotkey_parser ) 
+        bittensor.subtensor.add_args( new_hotkey_parser )
 
 
 class NewColdkeyCommand:
@@ -393,4 +397,5 @@ def add_args( parser: argparse.ArgumentParser ):
             default=False,
             help='''Overwrite the old coldkey with the newly generated coldkey'''
         )
-        bittensor.wallet.add_args( new_coldkey_parser )
+        bittensor.wallet.add_args( new_coldkey_parser )
+        bittensor.subtensor.add_args( new_coldkey_parser )