feat: detect ots capabilities

banteg · Apr 12, 2024 · a070592 · a070592
1 parent 131d61b
commit a070592
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/src/ape_geth/query.py b/src/ape_geth/query.py
@@ -30,12 +30,8 @@ def perform_query(self, query: QueryType) -> Iterator:  # type: ignore[override]
 
     @estimate_query.register
     def estimate_contract_creation_query(self, query: ContractCreationQuery) -> Optional[int]:
-        if provider := self.network_manager.active_provider:
-            if not isinstance(provider, EthereumNodeProvider):
-                return None
-            elif uri := provider.http_uri:
-                return 225 if uri.startswith("http://") else 600
-
+        if self.provider._ots_api_level is not None:
+            return 300
         return None
 
     @perform_query.register