bug: fix avs directory init

Layr-Labs · Aug 6, 2024 · b86879d · b86879d
1 parent 66c0054
commit b86879d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/chainio/clients/elcontracts/builder.go b/chainio/clients/elcontracts/builder.go
@@ -38,6 +38,7 @@ func BuildClients(
 		elContractBindings.DelegationManager,
 		elContractBindings.StrategyManager,
 		elContractBindings.RewardsCoordinator,
+		elContractBindings.AvsDirectory,
 		elContractBindings.StrategyManagerAddr,
 		elChainReader,
 		client,

diff --git a/chainio/clients/elcontracts/writer.go b/chainio/clients/elcontracts/writer.go
@@ -49,6 +49,7 @@ func NewChainWriter(
 	delegationManager *delegationmanager.ContractDelegationManager,
 	strategyManager *strategymanager.ContractStrategyManager,
 	rewardsCoordinator *rewardscoordinator.ContractIRewardsCoordinator,
+	avsDirectory *avsdirectory.ContractIAVSDirectory,
 	strategyManagerAddr gethcommon.Address,
 	elChainReader Reader,
 	ethClient eth.HttpBackend,
@@ -64,6 +65,7 @@ func NewChainWriter(
 		strategyManager:     strategyManager,
 		strategyManagerAddr: strategyManagerAddr,
 		rewardsCoordinator:  rewardsCoordinator,
+		avsDirectory:        avsDirectory,
 		elChainReader:       elChainReader,
 		logger:              logger,
 		ethClient:           ethClient,
@@ -104,6 +106,7 @@ func BuildELChainWriter(
 		elContractBindings.DelegationManager,
 		elContractBindings.StrategyManager,
 		elContractBindings.RewardsCoordinator,
+		elContractBindings.AvsDirectory,
 		elContractBindings.StrategyManagerAddr,
 		elChainReader,
 		ethClient,
@@ -142,6 +145,7 @@ func NewWriterFromConfig(
 		elContractBindings.DelegationManager,
 		elContractBindings.StrategyManager,
 		elContractBindings.RewardsCoordinator,
+		elContractBindings.AvsDirectory,
 		elContractBindings.StrategyManagerAddr,
 		elChainReader,
 		ethClient,