Merge pull request #2846 from opentensor/feat/roman/selective-metagraph

Add `SelectiveMetagraph` interface into SDK

Author: basfroman

bittensor/core/async_subtensor.py

@@ -16,17 +16,18 @@
 
 from bittensor.core.chain_data import (
     DelegateInfo,
-    StakeInfo,
+    DynamicInfo,
     MetagraphInfo,
     NeuronInfoLite,
     NeuronInfo,
     ProposalVoteData,
+    StakeInfo,
+    SelectiveMetagraphIndex,
     SubnetHyperparameters,
     SubnetIdentity,
     SubnetInfo,
     WeightCommitInfo,
     decode_account_id,
-    DynamicInfo,
 )
 from bittensor.core.chain_data.chain_identity import ChainIdentity
 from bittensor.core.chain_data.delegate_info import DelegatedInfo
@@ -1470,32 +1471,56 @@ async def get_minimum_required_stake(self):
     async def get_metagraph_info(
         self,
         netuid: int,
+        field_indices: Optional[list["SelectiveMetagraphIndex"]] = None,
         block: Optional[int] = None,
         block_hash: Optional[str] = None,
         reuse_block: bool = False,
     ) -> Optional[MetagraphInfo]:
         """
-        Retrieves the MetagraphInfo dataclass from the node for a single subnet (netuid)
+        Retrieves full or partial metagraph information for the specified subnet (netuid).
 
         Arguments:
-            netuid: The NetUID of the subnet.
+            netuid (int): The NetUID of the subnet to query.
+            field_indices (Optional[list[SelectiveMetagraphIndex]]): An optional list of SelectiveMetagraphIndex values
+                specifying which fields to retrieve. If not provided, all available fields will be returned.
             block: the block number at which to retrieve the hyperparameter. Do not specify if using block_hash or
                 reuse_block
             block_hash: The hash of blockchain block number for the query. Do not specify if using
                 block or reuse_block
             reuse_block: Whether to reuse the last-used block hash. Do not set if using block_hash or block.
 
         Returns:
-            MetagraphInfo dataclass
+            Optional[MetagraphInfo]: A MetagraphInfo object containing the requested subnet data, or None if the subnet
+                with the given netuid does not exist.
+
+        Example:
+            meta_info = await subtensor.get_metagraph_info(netuid=2)
+
+            partial_meta_info = await subtensor.get_metagraph_info(
+                netuid=2,
+                field_indices=[SelectiveMetagraphIndex.Name, SelectiveMetagraphIndex.OwnerHotkeys]
+            )
         """
+        indexes = SelectiveMetagraphIndex.all_indices()
+
+        if field_indices:
+            if isinstance(field_indices, list) and all(
+                isinstance(f, SelectiveMetagraphIndex) for f in field_indices
+            ):
+                indexes = [f.value for f in field_indices]
+            else:
+                raise ValueError(
+                    "`field_indices` must be a list of SelectiveMetagraphIndex items."
+                )
+
         block_hash = await self.determine_block_hash(block, block_hash, reuse_block)
         if not block_hash and reuse_block:
             block_hash = self.substrate.last_block_hash
 
         query = await self.substrate.runtime_call(
             "SubnetInfoRuntimeApi",
-            "get_metagraph",
-            params=[netuid],
+            "get_selective_metagraph",
+            params=[netuid, indexes if 0 in indexes else [0] + indexes],
             block_hash=block_hash,
         )
         if query.value is None:

bittensor/core/chain_data/__init__.py

@@ -16,6 +16,7 @@
     MetagraphInfoEmissions,
     MetagraphInfoPool,
     MetagraphInfoParams,
+    SelectiveMetagraphIndex,
 )
 from .neuron_info import NeuronInfo
 from .neuron_info_lite import NeuronInfoLite
@@ -50,6 +51,7 @@
     ProposalCallData,
     ProposalVoteData,
     ScheduledColdkeySwapInfo,
+    SelectiveMetagraphIndex,
     StakeInfo,
     SubnetHyperparameters,
     SubnetIdentity,