prediction-market-agent-tooling 0.69.14__tar.gz → 0.69.15.dev1136__tar.gz

{prediction_market_agent_tooling-0.69.14 → prediction_market_agent_tooling-0.69.15.dev1136}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prediction-market-agent-tooling
-Version: 0.69.14
+Version: 0.69.15.dev1136
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 License-File: LICENSE
 Author: Gnosis

prediction_market_agent_tooling-0.69.15.dev1136/prediction_market_agent_tooling/abis/processor.abi.json

@@ -0,0 +1,16 @@
+[
+  {
+    "constant": true,
+    "inputs": [],
+    "name": "processor",
+    "outputs": [
+      {
+        "name": "impl",
+        "type": "address"
+      }
+    ],
+    "payable": false,
+    "stateMutability": "view",
+    "type": "function"
+  }
+]

{prediction_market_agent_tooling-0.69.14 → prediction_market_agent_tooling-0.69.15.dev1136}/prediction_market_agent_tooling/tools/contract.py

@@ -196,6 +196,22 @@ class ContractProxyBaseClass(ContractBaseClass):
         return Web3.to_checksum_address(address)
 
 
+class ContractProcessorBaseClass(ContractBaseClass):
+    """
+    Contract base class for processor contracts.
+    """
+
+    abi: ABI = abi_field_validator(
+        os.path.join(
+            os.path.dirname(os.path.realpath(__file__)), "../abis/processor.abi.json"
+        )
+    )
+
+    def processor(self, web3: Web3 | None = None) -> ChecksumAddress:
+        address = self.call("processor", web3=web3)
+        return Web3.to_checksum_address(address)
+
+
 class ContractERC20BaseClass(ContractBaseClass):
     """
     Contract base class extended by ERC-20 standard methods.
@@ -948,6 +964,7 @@ def uni_implementation_address(
     # TODO: Fix the above, and afterwards assert that only 1 imp address is returned from this function. Or prove that this could indeed happen (although we are very pretty sure it shouldn't).
     addresses = [
         implementation_proxy_address(contract_address, web3),
+        processor_proxy_address(contract_address, web3),
         minimal_proxy_address(contract_address, web3),
         seer_minimal_proxy_address(contract_address, web3),
         eip_1967_proxy_address(contract_address, web3),
@@ -979,6 +996,29 @@ def implementation_proxy_address(
         return None
 
 
+def processor_proxy_address(
+    contract_address: ChecksumAddress, web3: Web3
+) -> ChecksumAddress | None:
+    if not contract_implements_function(
+        contract_address, "processor", web3, look_for_proxy_contract=False
+    ):
+        return None
+    try:
+        return ContractProcessorBaseClass(address=contract_address).processor(web3)
+    except (ContractCustomError, ContractLogicError, tenacity.RetryError) as e:
+        if isinstance(e, tenacity.RetryError) and not isinstance(
+            e.last_attempt.exception(), (ContractCustomError, ContractLogicError)
+        ):
+            raise
+
+        # For example https://gnosisscan.io/address/0x3221a28ed2b2e955da64d1d299956f277562c95c#code,
+        # it has the processor method, but it's only for admins.
+        logger.warning(
+            f"Failed to get processor for {contract_address=} even though it has the method: {e}"
+        )
+        return None
+
+
 def minimal_proxy_address(
     contract_address: ChecksumAddress, web3: Web3
 ) -> ChecksumAddress | None:

{prediction_market_agent_tooling-0.69.14 → prediction_market_agent_tooling-0.69.15.dev1136}/prediction_market_agent_tooling/tools/cow/models.py

@@ -1,5 +1,5 @@
 from enum import Enum
-from typing import Optional
+from typing import Optional, TypeAlias
 
 from pydantic import BaseModel, ConfigDict
 from sqlmodel import Field, SQLModel
@@ -74,10 +74,13 @@ class OrderStatus(str, Enum):
     expired = "expired"
 
 
+CowOrderUID: TypeAlias = HexBytes
+
+
 class Order(BaseModel):
     model_config = ConfigDict(validate_by_name=True, validate_by_alias=True)
 
-    uid: str
+    uid: CowOrderUID
     quoteId: int | None = None
     validTo: int
     sellAmount: Wei

{prediction_market_agent_tooling-0.69.14 → prediction_market_agent_tooling-0.69.15.dev1136}/pyproject.toml

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