update return value

Layr-Labs · Sep 11, 2024 · c1fef1b · c1fef1b
1 parent 45b65da
commit c1fef1b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/contracts/core/StakeRootCompendium.sol b/src/contracts/core/StakeRootCompendium.sol
@@ -526,7 +526,7 @@ contract StakeRootCompendium is StakeRootCompendiumStorage {
         uint256 operatorSetIndex,
         uint256 startOperatorIndex,
         uint256 numOperators
-    ) external view returns (OperatorLeaf[] memory) {
+    ) external view returns (IAVSDirectory.OperatorSet memory, address[] memory, OperatorLeaf[] memory) {
         require(
             operatorSetIndex < operatorSets.length,
             "StakeRootCompendium.getOperatorSetLeaves: operator set index out of bounds"
@@ -546,7 +546,7 @@ contract StakeRootCompendium is StakeRootCompendiumStorage {
                 extraData: operatorExtraDatas[operatorSet.avs][operatorSet.operatorSetId][operators[i]]
             });
         }
-        return operatorLeaves;
+        return (operatorSet, operators, operatorLeaves);
     }
 
     /// @inheritdoc IStakeRootCompendium

diff --git a/src/contracts/interfaces/IStakeRootCompendium.sol b/src/contracts/interfaces/IStakeRootCompendium.sol
@@ -111,13 +111,13 @@ interface IStakeRootCompendium {
      * @param operatorSetIndex the index of the operatorSet within the SRC's operatorSets list to calculate the operator set leaves for
      * @param startOperatorIndex the index of the first operator to get the leaves for
      * @param numOperators the number of operators to get the leaves for
-     * @return the operatorSet leaves
+     * @return the operatorSet, the list of operators and the operatorSet leaves
      */
     function getOperatorSetLeaves(
         uint256 operatorSetIndex,
         uint256 startOperatorIndex,
         uint256 numOperators
-    ) external view returns (OperatorLeaf[] memory);
+    ) external view returns (IAVSDirectory.OperatorSet memory, address[] memory, OperatorLeaf[] memory);
 
     /**
      * @notice deposits funds for an operator set