prediction-market-agent-tooling 0.69.15.dev1136__tar.gz → 0.69.15.dev1138__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prediction-market-agent-tooling
-Version: 0.69.15.dev1136
+Version: 0.69.15.dev1138
 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-0.69.15.dev1138}/prediction_market_agent_tooling/tools/contract.py

@@ -12,6 +12,7 @@ from web3 import Web3
 from web3.constants import CHECKSUM_ADDRESSS_ZERO, HASH_ZERO
 from web3.contract.contract import Contract as Web3Contract
 from web3.exceptions import ContractCustomError, ContractLogicError
+from web3.types import BlockIdentifier
 
 from prediction_market_agent_tooling.chains import POLYGON_CHAIN_ID
 from prediction_market_agent_tooling.config import APIKeys, RPCConfig
@@ -97,6 +98,7 @@ class ContractBaseClass(BaseModel):
         function_name: str,
         function_params: t.Optional[list[t.Any] | dict[str, t.Any]] = None,
         web3: Web3 | None = None,
+        block_identifier: BlockIdentifier | None = None,
     ) -> t.Any:
         """
         Used for reading from the contract.
@@ -108,6 +110,7 @@ class ContractBaseClass(BaseModel):
             contract_abi=self.abi,
             function_name=function_name,
             function_params=function_params,
+            block_identifier=block_identifier,
         )
 
     def send(
@@ -290,8 +293,17 @@ class ContractERC20BaseClass(ContractBaseClass):
             web3=web3,
         )
 
-    def balanceOf(self, for_address: ChecksumAddress, web3: Web3 | None = None) -> Wei:
-        balance = Wei(self.call("balanceOf", [for_address], web3=web3))
+    def balanceOf(
+        self,
+        for_address: ChecksumAddress,
+        web3: Web3 | None = None,
+        block_identifier: BlockIdentifier | None = None,
+    ) -> Wei:
+        balance = Wei(
+            self.call(
+                "balanceOf", [for_address], web3=web3, block_identifier=block_identifier
+            )
+        )
         return balance
 
     def balance_of_in_tokens(

{prediction_market_agent_tooling-0.69.15.dev1136 → prediction_market_agent_tooling-0.69.15.dev1138}/prediction_market_agent_tooling/tools/web3_utils.py

@@ -16,7 +16,14 @@ from safe_eth.safe.safe import SafeV141
 from web3 import Web3
 from web3.constants import HASH_ZERO
 from web3.contract.contract import ContractFunction as Web3ContractFunction
-from web3.types import AccessList, AccessListEntry, Nonce, TxParams, TxReceipt
+from web3.types import (
+    AccessList,
+    AccessListEntry,
+    BlockIdentifier,
+    Nonce,
+    TxParams,
+    TxReceipt,
+)
 
 from prediction_market_agent_tooling.gtypes import (
     ABI,
@@ -106,11 +113,12 @@ def call_function_on_contract(
     contract_abi: ABI,
     function_name: str,
     function_params: Optional[list[Any] | dict[str, Any]] = None,
+    block_identifier: Optional[BlockIdentifier] = None,
 ) -> Any:
     contract = web3.eth.contract(address=contract_address, abi=contract_abi)
     output = contract.functions[function_name](
         *parse_function_params(function_params)
-    ).call()
+    ).call(block_identifier=block_identifier)
     return output
 
 

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

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