feat: support both

ApeWorX · Dec 2, 2024 · a9ab8ce · a9ab8ce
1 parent 109fdc7
commit a9ab8ce
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/ape_alchemy/provider.py b/ape_alchemy/provider.py
@@ -18,7 +18,11 @@
 from web3.exceptions import ContractLogicError as Web3ContractLogicError
 from web3.exceptions import ExtraDataLengthError
 from web3.gas_strategies.rpc import rpc_gas_price_strategy
-from web3.middleware import ExtraDataToPOAMiddleware
+try:
+    from web3.middleware import ExtraDataToPOAMiddleware  # type: ignore
+except ImportError:
+    from web3.middleware import geth_poa_middleware as ExtraDataToPOAMiddleware  # type: ignore
+
 from web3.middleware.validation import MAX_EXTRADATA_LENGTH
 from web3.types import RPCEndpoint