prediction-market-agent-tooling 0.69.7.dev1098__tar.gz → 0.69.7.dev1100__tar.gz

{prediction_market_agent_tooling-0.69.7.dev1098 → prediction_market_agent_tooling-0.69.7.dev1100}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: prediction-market-agent-tooling
-Version: 0.69.7.dev1098
+Version: 0.69.7.dev1100
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.13

{prediction_market_agent_tooling-0.69.7.dev1098 → prediction_market_agent_tooling-0.69.7.dev1100}/prediction_market_agent_tooling/tools/contract.py

@@ -909,8 +909,8 @@ def contract_implements_function(
 
     # If not found directly and we should check proxies
     if not implements and look_for_proxy_contract:
-        imp_address = uni_implementation_address(contract_address, web3)
-        if imp_address is not None:
+        imp_addresses = uni_implementation_address(contract_address, web3)
+        for imp_address in imp_addresses:
             implements = contract_implements_function(
                 imp_address,
                 function_name=function_name,
@@ -918,30 +918,28 @@ def contract_implements_function(
                 function_arg_types=function_arg_types,
                 look_for_proxy_contract=False,
             )
+            # If one of the implementations has the function, we can terminate early.
+            if implements:
+                break
 
     return implements
 
 
 def uni_implementation_address(
     contract_address: ChecksumAddress, web3: Web3
-) -> ChecksumAddress | None:
+) -> list[ChecksumAddress]:
     """
     There are multiple ways how proxies can be implemented.
-    This function enumerates them and returns the one that succeeds, or None.
+    This function enumerates them and returns the ones that succeed, or an empty list.
     """
-    address = implementation_proxy_address(contract_address, web3)
-    if address is not None:
-        return address
-
-    address = minimal_proxy_address(contract_address, web3)
-    if address is not None:
-        return address
-
-    address = seer_minimal_proxy_address(contract_address, web3)
-    if address is not None:
-        return address
-
-    return None
+    # It's non-intuitive, but for a single contract, multiple of these can return values.
+    # For example in `test_wrapped_erc1155_init_collateral` test, both minimal_proxy and seer_minimal proxy return an address, but different.
+    addresses = [
+        implementation_proxy_address(contract_address, web3),
+        minimal_proxy_address(contract_address, web3),
+        seer_minimal_proxy_address(contract_address, web3),
+    ]
+    return [addr for addr in addresses if addr is not None]
 
 
 def implementation_proxy_address(
@@ -951,7 +949,7 @@ def implementation_proxy_address(
         contract_address, "implementation", web3, look_for_proxy_contract=False
     ):
         return None
-    return ContractProxyOnGnosisChain(address=contract_address).implementation()
+    return ContractProxyOnGnosisChain(address=contract_address).implementation(web3)
 
 
 def minimal_proxy_address(

{prediction_market_agent_tooling-0.69.7.dev1098 → prediction_market_agent_tooling-0.69.7.dev1100}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "prediction-market-agent-tooling"
-version = "0.69.7.dev1098"
+version = "0.69.7.dev1100"
 description = "Tools to benchmark, deploy and monitor prediction market agents."
 authors = ["Gnosis"]
 readme = "README.md"