fix: generate am from delegation manager (#425)

Layr-Labs · Jan 6, 2025 · a0cf13a · a0cf13a
1 parent fce9e63
commit a0cf13a
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/chainio/clients/avsregistry/bindings.go b/chainio/clients/avsregistry/bindings.go
@@ -3,6 +3,7 @@ package avsregistry
 import (
 	"github.com/Layr-Labs/eigensdk-go/chainio/clients/eth"
 	blsapkregistry "github.com/Layr-Labs/eigensdk-go/contracts/bindings/BLSApkRegistry"
+	contractDelegationManager "github.com/Layr-Labs/eigensdk-go/contracts/bindings/DelegationManager"
 	indexregistry "github.com/Layr-Labs/eigensdk-go/contracts/bindings/IndexRegistry"
 	opstateretriever "github.com/Layr-Labs/eigensdk-go/contracts/bindings/OperatorStateRetriever"
 	regcoordinator "github.com/Layr-Labs/eigensdk-go/contracts/bindings/RegistryCoordinator"
@@ -211,7 +212,14 @@ func NewBindingsFromConfig(
 		if err != nil {
 			return nil, utils.WrapError("Failed to get AvsDirectory address", err)
 		}
-		allocationManagerAddr, err = contractServiceManager.AllocationManager(&bind.CallOpts{})
+
+		delegationManager, err := contractDelegationManager.NewContractDelegationManager(
+			delegationManagerAddr,
+			client)
+		if err != nil {
+			return nil, utils.WrapError("Failed to get DelegationManager contract", err)
+		}
+		allocationManagerAddr, err = delegationManager.AllocationManager(&bind.CallOpts{})
 		if err != nil {
 			return nil, utils.WrapError("Failed to get AllocationManager address", err)
 		}