prediction-market-agent-tooling 0.58.5__py3-none-any.whl → 0.59.0.dev399__py3-none-any.whl

prediction_market_agent_tooling/tools/contract.py

@@ -18,16 +18,12 @@ from prediction_market_agent_tooling.gtypes import (
     TxParams,
     TxReceipt,
     Wei,
-    xDai,
 )
-from prediction_market_agent_tooling.tools.data_models import MessageContainer
-from prediction_market_agent_tooling.tools.hexbytes_custom import HexBytes
-from prediction_market_agent_tooling.tools.utils import BPS_CONSTANT, DatetimeUTC
+from prediction_market_agent_tooling.tools.utils import DatetimeUTC
 from prediction_market_agent_tooling.tools.web3_utils import (
     call_function_on_contract,
     send_function_on_contract_tx,
     send_function_on_contract_tx_using_safe,
-    wei_to_xdai,
 )
 
 
@@ -568,170 +564,6 @@ class DebuggingContract(ContractOnGnosisChain):
         )
 
 
-class SimpleTreasuryContract(ContractOnGnosisChain, OwnableContract):
-    # Contract ABI taken from built https://github.com/gnosis/labs-contracts.
-    abi: ABI = abi_field_validator(
-        os.path.join(
-            os.path.dirname(os.path.realpath(__file__)),
-            "../abis/simpletreasury.abi.json",
-        )
-    )
-
-    address: ChecksumAddress = Web3.to_checksum_address(
-        "0x624ad0db52e6b18afb4d36b8e79d0c2a74f3fc8a"
-    )
-
-    def required_nft_balance(self, web3: Web3 | None = None) -> int:
-        min_num_of_nfts: int = self.call("requiredNFTBalance", web3=web3)
-        return min_num_of_nfts
-
-    def set_required_nft_balance(
-        self,
-        api_keys: APIKeys,
-        new_required_balance: int,
-        web3: Web3 | None = None,
-    ) -> TxReceipt:
-        return self.send(
-            api_keys=api_keys,
-            function_name="setRequiredNFTBalance",
-            function_params=[new_required_balance],
-            web3=web3,
-        )
-
-    def nft_contract(self, web3: Web3 | None = None) -> ContractERC721BaseClass:
-        nft_contract_address: ChecksumAddress = self.call("nftContract", web3=web3)
-        contract = ContractERC721BaseClass(address=nft_contract_address)
-        return contract
-
-    def withdraw(
-        self,
-        api_keys: APIKeys,
-        web3: Web3 | None = None,
-    ) -> TxReceipt:
-        return self.send(
-            api_keys=api_keys,
-            function_name="withdraw",
-            web3=web3,
-        )
-
-
-class AgentCommunicationContract(ContractOnGnosisChain, OwnableContract):
-    # Contract ABI taken from built https://github.com/gnosis/labs-contracts.
-    abi: ABI = abi_field_validator(
-        os.path.join(
-            os.path.dirname(os.path.realpath(__file__)),
-            "../abis/agentcommunication.abi.json",
-        )
-    )
-
-    address: ChecksumAddress = Web3.to_checksum_address(
-        "0xe9dd78FF297DbaAbe5D0E45aE554a4B561935DE9"
-    )
-
-    def minimum_message_value(self, web3: Web3 | None = None) -> xDai:
-        value: Wei = self.call("minimumValueForSendingMessageInWei", web3=web3)
-        return wei_to_xdai(value)
-
-    def ratio_given_to_treasury(self, web3: Web3 | None = None) -> float:
-        bps: int = self.call("pctToTreasuryInBasisPoints", web3=web3)
-        return bps / BPS_CONSTANT
-
-    def count_unseen_messages(
-        self,
-        agent_address: ChecksumAddress,
-        web3: Web3 | None = None,
-    ) -> int:
-        unseen_message_count: int = self.call(
-            "countMessages", function_params=[agent_address], web3=web3
-        )
-        return unseen_message_count
-
-    def get_at_index(
-        self,
-        agent_address: ChecksumAddress,
-        idx: int,
-        web3: Web3 | None = None,
-    ) -> MessageContainer:
-        message_container_raw: t.Tuple[t.Any] = self.call(
-            "getAtIndex", function_params=[agent_address, idx], web3=web3
-        )
-        return MessageContainer.from_tuple(message_container_raw)
-
-    def set_treasury_address(
-        self,
-        api_keys: APIKeys,
-        new_treasury_address: ChecksumAddress,
-        web3: Web3 | None = None,
-    ) -> TxReceipt:
-        return self.send(
-            api_keys=api_keys,
-            function_name="setTreasuryAddress",
-            function_params=[new_treasury_address],
-            web3=web3,
-        )
-
-    def set_minimum_value_for_sending_message(
-        self,
-        api_keys: APIKeys,
-        new_minimum_value: Wei,
-        web3: Web3 | None = None,
-    ) -> TxReceipt:
-        return self.send(
-            api_keys=api_keys,
-            function_name="adjustMinimumValueForSendingMessage",
-            function_params=[new_minimum_value],
-            web3=web3,
-        )
-
-    def pop_message(
-        self,
-        api_keys: APIKeys,
-        agent_address: ChecksumAddress,
-        index: int = 0,
-        web3: Web3 | None = None,
-    ) -> MessageContainer:
-        """
-        Retrieves and removes message at specified index from the agent's message queue.
-
-        This method first retrieves the message at the front of the queue without removing it,
-        allowing us to return the message content directly. The actual removal of the message
-        from the queue is performed by sending a transaction to the contract, which executes
-        the `popMessageAtIndex` function. The transaction receipt is not used to obtain the message
-        content, as it only contains event data, not the returned struct.
-        """
-
-        # Peek the element before popping.
-        message_container = self.get_at_index(
-            agent_address=agent_address, idx=index, web3=web3
-        )
-
-        # Next, pop that element and discard the transaction receipt.
-        self.send(
-            api_keys=api_keys,
-            function_name="popMessageAtIndex",
-            function_params=[agent_address, index],
-            web3=web3,
-        )
-
-        return message_container
-
-    def send_message(
-        self,
-        api_keys: APIKeys,
-        agent_address: ChecksumAddress,
-        message: HexBytes,
-        amount_wei: Wei,
-        web3: Web3 | None = None,
-    ) -> TxReceipt:
-        return self.send_with_value(
-            api_keys=api_keys,
-            function_name="sendMessage",
-            amount_wei=amount_wei,
-            function_params=[agent_address, message],
-            web3=web3,
-        )
-
-
 def contract_implements_function(
     contract_address: ChecksumAddress,
     function_name: str,

{prediction_market_agent_tooling-0.58.5.dist-info → prediction_market_agent_tooling-0.59.0.dev399.dist-info}/METADATA

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

{prediction_market_agent_tooling-0.58.5.dist-info → prediction_market_agent_tooling-0.59.0.dev399.dist-info}/RECORD

@@ -1,4 +1,3 @@
-prediction_market_agent_tooling/abis/agentcommunication.abi.json,sha256=Go6blKVORPuViDkxw_vqFzcpRH5i2qHje9R1wILDqM8,5817
 prediction_market_agent_tooling/abis/debuggingcontract.abi.json,sha256=KdkCWmif_WO421RFKhc03kGJiccFqsxsZNrgCd8EH_4,571
 prediction_market_agent_tooling/abis/depositablewrapper_erc20.abi.json,sha256=m0Wk3uQyLM8apWRRvX4d3u1d77bWVuXfV38D-aV48t0,4612
 prediction_market_agent_tooling/abis/erc20.abi.json,sha256=b8t6tPUhjBo5dHFq8ipsK-ih_g29dOTbWI0GBZiEWL8,3685
@@ -17,7 +16,6 @@ prediction_market_agent_tooling/abis/ownable.abi.json,sha256=DeTy_7VmsMhFl7jwI8M
 prediction_market_agent_tooling/abis/ownable_erc721.abi.json,sha256=9sxm588MAQmqCV_S0D3eYC7l9grbeALsd0Da_AHxdEI,8506
 prediction_market_agent_tooling/abis/proxy.abi.json,sha256=h24GXZ6Q0bSZlwh7zOv0EiDvbqUz_PHtWfKHTyPJ1w4,644
 prediction_market_agent_tooling/abis/seer_market_factory.abi.json,sha256=g7RVxZVUWlTXIgTV2W6kO4twQM909Qv58zAr7Dk4XIc,13553
-prediction_market_agent_tooling/abis/simpletreasury.abi.json,sha256=7-l7rntLkcFKkoN8hXywO0-h-mUZxOZlyQg1Jk7yTB0,2877
 prediction_market_agent_tooling/benchmark/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prediction_market_agent_tooling/benchmark/agents.py,sha256=B1-uWdyeN4GGKMWGK_-CcAFJg1m9Y_XuaeIHPB29QR8,3971
 prediction_market_agent_tooling/benchmark/benchmark.py,sha256=MqTiaaJ3cYiOLUVR7OyImLWxcEya3Rl5JyFYW-K0lwM,17097
@@ -85,12 +83,11 @@ prediction_market_agent_tooling/tools/betting_strategies/utils.py,sha256=kpIb-ci
 prediction_market_agent_tooling/tools/caches/db_cache.py,sha256=aafau_n_AUbLIwkyIRiTPgKB0dmM0767mSqyPDLF2A4,10576
 prediction_market_agent_tooling/tools/caches/inmemory_cache.py,sha256=ZW5iI5rmjqeAebu5T7ftRnlkxiL02IC-MxCfDB80x7w,1506
 prediction_market_agent_tooling/tools/caches/serializers.py,sha256=vFDx4fsPxclXp2q0sv27j4al_M_Tj9aR2JJP-xNHQXA,2151
-prediction_market_agent_tooling/tools/contract.py,sha256=JkkrXm75_XUZ33jVlXF7v-Ge9RLSfXQeY523H_kuThQ,26069
+prediction_market_agent_tooling/tools/contract.py,sha256=gmSuANqlfwldFG7hc4uom0q1Z-mo40qfYh5TDUKLFTs,20518
 prediction_market_agent_tooling/tools/costs.py,sha256=EaAJ7v9laD4VEV3d8B44M4u3_oEO_H16jRVCdoZ93Uw,954
 prediction_market_agent_tooling/tools/cow/cow_manager.py,sha256=WK6Uk722VotjLHtxDPHxvwBrWVb3rvTegg_3w58ehwU,3869
 prediction_market_agent_tooling/tools/cow/cow_order.py,sha256=ohriN_65BZ69f_BV6bNg_AwZjidip4yi4d2G6Ddy5Qg,4238
 prediction_market_agent_tooling/tools/custom_exceptions.py,sha256=Fh8z1fbwONvP4-j7AmV_PuEcoqb6-QXa9PJ9m7guMcM,93
-prediction_market_agent_tooling/tools/data_models.py,sha256=jDQ7FU0QQhXlcgJh5VZZGwDTYP2OPAqKPHZFewCPAUY,732
 prediction_market_agent_tooling/tools/datetime_utc.py,sha256=8_WackjtjC8zHXrhQFTGQ6e6Fz_6llWoKR4CSFvIv9I,2766
 prediction_market_agent_tooling/tools/db/db_manager.py,sha256=GtzHH1NLl8HwqC8Z7s6eTlIQXuV0blxfaV2PeQrBnfQ,3013
 prediction_market_agent_tooling/tools/google_utils.py,sha256=t3_UEEvKX3L0biSIQ560GdRbllQ6eprhK_upE243A-0,3185
@@ -120,8 +117,8 @@ prediction_market_agent_tooling/tools/tokens/main_token.py,sha256=5iHO7-iehSlXuu
 prediction_market_agent_tooling/tools/transaction_cache.py,sha256=K5YKNL2_tR10Iw2TD9fuP-CTGpBbZtNdgbd0B_R7pjg,1814
 prediction_market_agent_tooling/tools/utils.py,sha256=jLG4nbEoIzzJiZ4RgMx4Q969Zdl0p0s63p8uET_0Fuw,6440
 prediction_market_agent_tooling/tools/web3_utils.py,sha256=e7lqqQddVZaa905rhBb6L1fC3o39Yr-PDJsJjHFBeRE,12523
-prediction_market_agent_tooling-0.58.5.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.58.5.dist-info/METADATA,sha256=rPCTjBMlKpKZHpIpANRKiO-FIZRUqMA_P0DUcps9pcI,8629
-prediction_market_agent_tooling-0.58.5.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-prediction_market_agent_tooling-0.58.5.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.58.5.dist-info/RECORD,,
+prediction_market_agent_tooling-0.59.0.dev399.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.59.0.dev399.dist-info/METADATA,sha256=2hdm2FPByw4iA2un7-to2JRnrvo-pOV6PHz8lkG61zE,8636
+prediction_market_agent_tooling-0.59.0.dev399.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+prediction_market_agent_tooling-0.59.0.dev399.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.59.0.dev399.dist-info/RECORD,,

prediction_market_agent_tooling/abis/agentcommunication.abi.json

@@ -1,224 +0,0 @@
-[
-  {
-    "inputs": [
-      {
-        "internalType": "address payable",
-        "name": "_treasury",
-        "type": "address"
-      },
-      {
-        "internalType": "uint256",
-        "name": "_pctToTreasuryInBasisPoints",
-        "type": "uint256"
-      }
-    ],
-    "stateMutability": "nonpayable",
-    "type": "constructor"
-  },
-  { "inputs": [], "name": "MessageNotSentByAgent", "type": "error" },
-  {
-    "inputs": [
-      { "internalType": "address", "name": "owner", "type": "address" }
-    ],
-    "name": "OwnableInvalidOwner",
-    "type": "error"
-  },
-  {
-    "inputs": [
-      { "internalType": "address", "name": "account", "type": "address" }
-    ],
-    "name": "OwnableUnauthorizedAccount",
-    "type": "error"
-  },
-  {
-    "anonymous": false,
-    "inputs": [
-      {
-        "indexed": true,
-        "internalType": "address",
-        "name": "sender",
-        "type": "address"
-      },
-      {
-        "indexed": true,
-        "internalType": "address",
-        "name": "agentAddress",
-        "type": "address"
-      },
-      {
-        "indexed": false,
-        "internalType": "bytes",
-        "name": "message",
-        "type": "bytes"
-      },
-      {
-        "indexed": false,
-        "internalType": "uint256",
-        "name": "value",
-        "type": "uint256"
-      }
-    ],
-    "name": "LogMessage",
-    "type": "event"
-  },
-  {
-    "anonymous": false,
-    "inputs": [
-      {
-        "indexed": true,
-        "internalType": "address",
-        "name": "previousOwner",
-        "type": "address"
-      },
-      {
-        "indexed": true,
-        "internalType": "address",
-        "name": "newOwner",
-        "type": "address"
-      }
-    ],
-    "name": "OwnershipTransferred",
-    "type": "event"
-  },
-  {
-    "inputs": [
-      { "internalType": "uint256", "name": "newValue", "type": "uint256" }
-    ],
-    "name": "adjustMinimumValueForSendingMessage",
-    "outputs": [],
-    "stateMutability": "nonpayable",
-    "type": "function"
-  },
-  {
-    "inputs": [
-      { "internalType": "address", "name": "agentAddress", "type": "address" }
-    ],
-    "name": "countMessages",
-    "outputs": [{ "internalType": "uint256", "name": "", "type": "uint256" }],
-    "stateMutability": "view",
-    "type": "function"
-  },
-  {
-    "inputs": [
-      { "internalType": "address", "name": "agentAddress", "type": "address" },
-      { "internalType": "uint256", "name": "idx", "type": "uint256" }
-    ],
-    "name": "getAtIndex",
-    "outputs": [
-      {
-        "components": [
-          { "internalType": "address", "name": "sender", "type": "address" },
-          { "internalType": "address", "name": "recipient", "type": "address" },
-          { "internalType": "bytes", "name": "message", "type": "bytes" },
-          { "internalType": "uint256", "name": "value", "type": "uint256" }
-        ],
-        "internalType": "struct DoubleEndedStructQueue.MessageContainer",
-        "name": "",
-        "type": "tuple"
-      }
-    ],
-    "stateMutability": "view",
-    "type": "function"
-  },
-  {
-    "inputs": [],
-    "name": "minimumValueForSendingMessageInWei",
-    "outputs": [{ "internalType": "uint256", "name": "", "type": "uint256" }],
-    "stateMutability": "view",
-    "type": "function"
-  },
-  {
-    "inputs": [],
-    "name": "owner",
-    "outputs": [{ "internalType": "address", "name": "", "type": "address" }],
-    "stateMutability": "view",
-    "type": "function"
-  },
-  {
-    "inputs": [],
-    "name": "pctToTreasuryInBasisPoints",
-    "outputs": [{ "internalType": "uint256", "name": "", "type": "uint256" }],
-    "stateMutability": "view",
-    "type": "function"
-  },
-  {
-    "inputs": [
-      { "internalType": "address", "name": "agentAddress", "type": "address" },
-      { "internalType": "uint256", "name": "idx", "type": "uint256" }
-    ],
-    "name": "popMessageAtIndex",
-    "outputs": [
-      {
-        "components": [
-          { "internalType": "address", "name": "sender", "type": "address" },
-          { "internalType": "address", "name": "recipient", "type": "address" },
-          { "internalType": "bytes", "name": "message", "type": "bytes" },
-          { "internalType": "uint256", "name": "value", "type": "uint256" }
-        ],
-        "internalType": "struct DoubleEndedStructQueue.MessageContainer",
-        "name": "",
-        "type": "tuple"
-      }
-    ],
-    "stateMutability": "nonpayable",
-    "type": "function"
-  },
-  {
-    "inputs": [{ "internalType": "address", "name": "", "type": "address" }],
-    "name": "queues",
-    "outputs": [
-      { "internalType": "uint128", "name": "_begin", "type": "uint128" },
-      { "internalType": "uint128", "name": "_end", "type": "uint128" }
-    ],
-    "stateMutability": "view",
-    "type": "function"
-  },
-  {
-    "inputs": [],
-    "name": "renounceOwnership",
-    "outputs": [],
-    "stateMutability": "nonpayable",
-    "type": "function"
-  },
-  {
-    "inputs": [
-      { "internalType": "address", "name": "agentAddress", "type": "address" },
-      { "internalType": "bytes", "name": "message", "type": "bytes" }
-    ],
-    "name": "sendMessage",
-    "outputs": [],
-    "stateMutability": "payable",
-    "type": "function"
-  },
-  {
-    "inputs": [
-      {
-        "internalType": "address payable",
-        "name": "_treasury",
-        "type": "address"
-      }
-    ],
-    "name": "setTreasuryAddress",
-    "outputs": [],
-    "stateMutability": "nonpayable",
-    "type": "function"
-  },
-  {
-    "inputs": [
-      { "internalType": "address", "name": "newOwner", "type": "address" }
-    ],
-    "name": "transferOwnership",
-    "outputs": [],
-    "stateMutability": "nonpayable",
-    "type": "function"
-  },
-  {
-    "inputs": [],
-    "name": "treasury",
-    "outputs": [
-      { "internalType": "address payable", "name": "", "type": "address" }
-    ],
-    "stateMutability": "view",
-    "type": "function"
-  }
-]

prediction_market_agent_tooling/abis/simpletreasury.abi.json

@@ -1,156 +0,0 @@
-[
-  {
-    "inputs": [
-      {
-        "internalType": "address",
-        "name": "_nftContract",
-        "type": "address"
-      }
-    ],
-    "stateMutability": "nonpayable",
-    "type": "constructor"
-  },
-  {
-    "inputs": [
-      {
-        "internalType": "address",
-        "name": "owner",
-        "type": "address"
-      }
-    ],
-    "name": "OwnableInvalidOwner",
-    "type": "error"
-  },
-  {
-    "inputs": [
-      {
-        "internalType": "address",
-        "name": "account",
-        "type": "address"
-      }
-    ],
-    "name": "OwnableUnauthorizedAccount",
-    "type": "error"
-  },
-  {
-    "anonymous": false,
-    "inputs": [
-      {
-        "indexed": true,
-        "internalType": "address",
-        "name": "previousOwner",
-        "type": "address"
-      },
-      {
-        "indexed": true,
-        "internalType": "address",
-        "name": "newOwner",
-        "type": "address"
-      }
-    ],
-    "name": "OwnershipTransferred",
-    "type": "event"
-  },
-  {
-    "anonymous": false,
-    "inputs": [
-      {
-        "indexed": true,
-        "internalType": "address",
-        "name": "holder",
-        "type": "address"
-      },
-      {
-        "indexed": false,
-        "internalType": "uint256",
-        "name": "amount",
-        "type": "uint256"
-      }
-    ],
-    "name": "WithdrawnByNFTHolder",
-    "type": "event"
-  },
-  {
-    "inputs": [],
-    "name": "nftContract",
-    "outputs": [
-      {
-        "internalType": "contract IERC721",
-        "name": "",
-        "type": "address"
-      }
-    ],
-    "stateMutability": "view",
-    "type": "function"
-  },
-  {
-    "inputs": [],
-    "name": "owner",
-    "outputs": [
-      {
-        "internalType": "address",
-        "name": "",
-        "type": "address"
-      }
-    ],
-    "stateMutability": "view",
-    "type": "function"
-  },
-  {
-    "inputs": [],
-    "name": "renounceOwnership",
-    "outputs": [],
-    "stateMutability": "nonpayable",
-    "type": "function"
-  },
-  {
-    "inputs": [],
-    "name": "requiredNFTBalance",
-    "outputs": [
-      {
-        "internalType": "uint256",
-        "name": "",
-        "type": "uint256"
-      }
-    ],
-    "stateMutability": "view",
-    "type": "function"
-  },
-  {
-    "inputs": [
-      {
-        "internalType": "uint256",
-        "name": "_newBalance",
-        "type": "uint256"
-      }
-    ],
-    "name": "setRequiredNFTBalance",
-    "outputs": [],
-    "stateMutability": "nonpayable",
-    "type": "function"
-  },
-  {
-    "inputs": [
-      {
-        "internalType": "address",
-        "name": "newOwner",
-        "type": "address"
-      }
-    ],
-    "name": "transferOwnership",
-    "outputs": [],
-    "stateMutability": "nonpayable",
-    "type": "function"
-  },
-  {
-    "inputs": [],
-    "name": "withdraw",
-    "outputs": [],
-    "stateMutability": "nonpayable",
-    "type": "function"
-  },
-  {
-    "stateMutability": "payable",
-    "type": "receive"
-  }
-]

prediction_market_agent_tooling/tools/data_models.py

@@ -1,23 +0,0 @@
-import typing as t
-
-from pydantic import BaseModel, ConfigDict, Field
-
-from prediction_market_agent_tooling.gtypes import ChecksumAddress, HexBytes, Wei
-
-
-# Taken from https://github.com/gnosis/labs-contracts/blob/main/src/NFT/DoubleEndedStructQueue.sol
-class MessageContainer(BaseModel):
-    model_config = ConfigDict(populate_by_name=True)
-    sender: ChecksumAddress
-    recipient: ChecksumAddress = Field(alias="agentAddress")
-    message: HexBytes
-    value: Wei
-
-    @staticmethod
-    def from_tuple(values: tuple[t.Any, ...]) -> "MessageContainer":
-        return MessageContainer(
-            sender=values[0],
-            recipient=values[1],
-            message=HexBytes(values[2]),
-            value=values[3],
-        )