prediction-market-agent-tooling 0.37.0__py3-none-any.whl → 0.38.0__py3-none-any.whl

prediction_market_agent_tooling/abis/omen_thumbnailmapping.abi.json

@@ -0,0 +1,52 @@
+[
+  {
+    "type": "function",
+    "name": "get",
+    "inputs": [
+      {
+        "name": "marketAddress",
+        "type": "address",
+        "internalType": "address"
+      }
+    ],
+    "outputs": [
+      {
+        "name": "",
+        "type": "bytes32",
+        "internalType": "bytes32"
+      }
+    ],
+    "stateMutability": "view"
+  },
+  {
+    "type": "function",
+    "name": "remove",
+    "inputs": [
+      {
+        "name": "marketAddress",
+        "type": "address",
+        "internalType": "address"
+      }
+    ],
+    "outputs": [],
+    "stateMutability": "nonpayable"
+  },
+  {
+    "type": "function",
+    "name": "set",
+    "inputs": [
+      {
+        "name": "marketAddress",
+        "type": "address",
+        "internalType": "address"
+      },
+      {
+        "name": "image_hash",
+        "type": "bytes32",
+        "internalType": "bytes32"
+      }
+    ],
+    "outputs": [],
+    "stateMutability": "nonpayable"
+  }
+]

prediction_market_agent_tooling/gtypes.py

@@ -34,6 +34,7 @@ Mana = NewType("Mana", float)  # Manifold's "currency"
 USDC = NewType("USDC", float)
 DatetimeWithTimezone = NewType("DatetimeWithTimezone", datetime)
 ChainID = NewType("ChainID", int)
+IPFSCIDVersion0 = NewType("IPFSCIDVersion0", str)
 
 
 def usd_type(amount: Union[str, int, float]) -> USD:

prediction_market_agent_tooling/markets/omen/omen_contracts.py

@@ -13,6 +13,7 @@ from prediction_market_agent_tooling.gtypes import (
     HexAddress,
     HexBytes,
     HexStr,
+    IPFSCIDVersion0,
     OmenOutcomeToken,
     TxParams,
     TxReceipt,
@@ -26,7 +27,12 @@ from prediction_market_agent_tooling.tools.contract import (
     ContractOnGnosisChain,
     abi_field_validator,
 )
-from prediction_market_agent_tooling.tools.web3_utils import xdai_to_wei
+from prediction_market_agent_tooling.tools.web3_utils import (
+    ZERO_BYTES,
+    byte32_to_ipfscidv0,
+    ipfscidv0_to_byte32,
+    xdai_to_wei,
+)
 
 
 class OmenOracleContract(ContractOnGnosisChain):
@@ -610,3 +616,61 @@ class OmenRealitioContract(ContractOnGnosisChain):
         web3: Web3 | None = None,
     ) -> TxReceipt:
         return self.send(api_keys=api_keys, function_name="withdraw", web3=web3)
+
+
+class OmenThumbnailMapping(ContractOnGnosisChain):
+    # 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/omen_thumbnailmapping.abi.json",
+        )
+    )
+    address: ChecksumAddress = Web3.to_checksum_address(
+        "0x5D8B7B619EcdE05B8A94C0a0E99E0A0727A0e2e7"
+    )
+
+    def get(
+        self,
+        market_address: ChecksumAddress,
+        web3: Web3 | None = None,
+    ) -> IPFSCIDVersion0 | None:
+        hash_bytes = HexBytes(
+            self.call("get", function_params=[market_address], web3=web3)
+        )
+        return byte32_to_ipfscidv0(hash_bytes) if hash_bytes != ZERO_BYTES else None
+
+    def get_url(
+        self,
+        market_address: ChecksumAddress,
+        web3: Web3 | None = None,
+    ) -> str | None:
+        hash_ = self.get(market_address, web3)
+        return f"https://ipfs.io/ipfs/{hash_}" if hash_ is not None else None
+
+    def set(
+        self,
+        api_keys: APIKeys,
+        market_address: ChecksumAddress,
+        image_hash: IPFSCIDVersion0,
+        web3: Web3 | None = None,
+    ) -> TxReceipt:
+        return self.send(
+            api_keys=api_keys,
+            function_name="set",
+            function_params=[market_address, ipfscidv0_to_byte32(image_hash)],
+            web3=web3,
+        )
+
+    def remove(
+        self,
+        api_keys: APIKeys,
+        market_address: ChecksumAddress,
+        web3: Web3 | None = None,
+    ) -> TxReceipt:
+        return self.send(
+            api_keys=api_keys,
+            function_name="remove",
+            function_params=[market_address],
+            web3=web3,
+        )

prediction_market_agent_tooling/tools/web3_utils.py

@@ -1,5 +1,7 @@
+import binascii
 from typing import Any, Optional, TypeVar
 
+import base58
 import tenacity
 from eth_account import Account
 from eth_typing import URI
@@ -16,6 +18,7 @@ from prediction_market_agent_tooling.gtypes import (
     HexAddress,
     HexBytes,
     HexStr,
+    IPFSCIDVersion0,
     PrivateKey,
     xDai,
     xdai_type,
@@ -287,3 +290,22 @@ def send_xdai_to(
         web3, tx_params_new, from_private_key, timeout
     )
     return receipt_tx
+
+
+def ipfscidv0_to_byte32(cid: IPFSCIDVersion0) -> HexBytes:
+    """
+    Convert ipfscidv0 to 32 bytes.
+    Modified from https://github.com/emg110/ipfs2bytes32/blob/main/python/ipfs2bytes32.py
+    """
+    decoded = base58.b58decode(cid)
+    sliced_decoded = decoded[2:]
+    return HexBytes(binascii.b2a_hex(sliced_decoded).decode("utf-8"))
+
+
+def byte32_to_ipfscidv0(hex: HexBytes) -> IPFSCIDVersion0:
+    """
+    Convert 32 bytes hex to ipfscidv0.
+    Modified from https://github.com/emg110/ipfs2bytes32/blob/main/python/ipfs2bytes32.py
+    """
+    completed_binary_str = b"\x12 " + hex
+    return IPFSCIDVersion0(base58.b58encode(completed_binary_str).decode("utf-8"))

{prediction_market_agent_tooling-0.37.0.dist-info → prediction_market_agent_tooling-0.38.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prediction-market-agent-tooling
-Version: 0.37.0
+Version: 0.38.0
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.12
@@ -11,6 +11,7 @@ Provides-Extra: google
 Provides-Extra: langchain
 Provides-Extra: openai
 Requires-Dist: autoflake (>=2.2.1,<3.0.0)
+Requires-Dist: base58 (>=2.1.1,<3.0.0)
 Requires-Dist: cron-validator (>=1.0.8,<2.0.0)
 Requires-Dist: eth-account (>=0.8.0,<0.9.0)
 Requires-Dist: eth-typing (>=3.0.0,<4.0.0)

{prediction_market_agent_tooling-0.37.0.dist-info → prediction_market_agent_tooling-0.38.0.dist-info}/RECORD

@@ -6,6 +6,7 @@ prediction_market_agent_tooling/abis/omen_fpmm_factory.abi.json,sha256=KBEb-COXM
 prediction_market_agent_tooling/abis/omen_kleros.abi.json,sha256=QPMXrYA5UCo_wVUqmzwCxes1bfvfzqzif3L_mJcx6j4,7315
 prediction_market_agent_tooling/abis/omen_oracle.abi.json,sha256=YPZ-FLvd4PA9pYd6d5mQI1TD11JnsE0LGxH4XcZZFCA,1775
 prediction_market_agent_tooling/abis/omen_realitio.abi.json,sha256=7HmFkBF_rq83UTaH2kRRsEfc_WZuf4n-qvkB4nhvweo,15953
+prediction_market_agent_tooling/abis/omen_thumbnailmapping.abi.json,sha256=u1-3B8FB3Ys9KVJCH-lw9ArkicdxbNMf34dV-VEGMMU,930
 prediction_market_agent_tooling/abis/wxdai.abi.json,sha256=m3qC06Yug-pToI0lSFe1f8e6gKMIulnV3MA2K0X51hI,6055
 prediction_market_agent_tooling/benchmark/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prediction_market_agent_tooling/benchmark/agents.py,sha256=HPIFJvackW110ch3UkktbxhU48gMRVo4gQse84Klhdc,4000
@@ -18,7 +19,7 @@ prediction_market_agent_tooling/deploy/constants.py,sha256=M5ty8URipYMGe_G-RzxRy
 prediction_market_agent_tooling/deploy/gcp/deploy.py,sha256=CYUgnfy-9XVk04kkxA_5yp0GE9Mw5caYqlFUZQ2j3ks,3739
 prediction_market_agent_tooling/deploy/gcp/kubernetes_models.py,sha256=qYIHRxQLac3yxtZ8ChikiPG9O1aUQucHW0muTSm1nto,2627
 prediction_market_agent_tooling/deploy/gcp/utils.py,sha256=oyW0jgrUT2Tr49c7GlpcMsYNQjoCSOcWis3q-MmVAhU,6089
-prediction_market_agent_tooling/gtypes.py,sha256=xGSJXw12hzp8LwvQ956l01GiZMWd07MZTYqo8CXVeLY,2417
+prediction_market_agent_tooling/gtypes.py,sha256=lbV2nsPyhMIRI9olx0_6A06jwTWKYBPGMxyiGVFysag,2467
 prediction_market_agent_tooling/loggers.py,sha256=ua9rynYmsbOJZjxPIFxRBooomeN08zuLSJ7lxZMDS7w,3133
 prediction_market_agent_tooling/markets/agent_market.py,sha256=SMvkXct_RgHXqF-fVq3ooTIQ_99MG77kVlvS3rM8Ozo,7019
 prediction_market_agent_tooling/markets/categorize.py,sha256=yTd-lDMPW4ESDSzmxeLLBuzLX0FggOF7Vbswh7295o0,941
@@ -32,7 +33,7 @@ prediction_market_agent_tooling/markets/markets.py,sha256=w05Oo7yCA2axpCw69Q9y4i
 prediction_market_agent_tooling/markets/omen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prediction_market_agent_tooling/markets/omen/data_models.py,sha256=EXtjmcujx68Xu50BVkYXvLuf_Asx5o65RvFS3ZS6HGs,14405
 prediction_market_agent_tooling/markets/omen/omen.py,sha256=fS19WoDG_N8WIa5FY-YRF21Np46y6d93QBJOpNKXi7Q,33939
-prediction_market_agent_tooling/markets/omen/omen_contracts.py,sha256=eDS8vN4Klv_-Y1wwfIeLDt3twhl9U_AJjPQov0JScb0,19888
+prediction_market_agent_tooling/markets/omen/omen_contracts.py,sha256=-EEmN4vA8Fn52ewqQ7ZXZAFnfP4EQ4YISkeDHb9oeLk,21722
 prediction_market_agent_tooling/markets/omen/omen_resolving.py,sha256=g77QsQ5WnSI2rzBlX87L_EhWMwobkyXyfRhHQmpAdzo,9012
 prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py,sha256=2y89FJlDrJ_hpeNTFZU1W30xCpt8QXjssEW8bNZnEpU,23596
 prediction_market_agent_tooling/markets/polymarket/api.py,sha256=HXmA1akA0qDj0m3e-GEvWG8x75pm6BX4H7YJPQcST7I,4767
@@ -67,9 +68,9 @@ prediction_market_agent_tooling/tools/safe.py,sha256=h0xOO0eNtitClf0fPkn-0oTc6A_
 prediction_market_agent_tooling/tools/singleton.py,sha256=CiIELUiI-OeS7U7eeHEt0rnVhtQGzwoUdAgn_7u_GBM,729
 prediction_market_agent_tooling/tools/streamlit_user_login.py,sha256=NXEqfjT9Lc9QtliwSGRASIz1opjQ7Btme43H4qJbzgE,3010
 prediction_market_agent_tooling/tools/utils.py,sha256=-G22UEbCRm59bm1RWFdeF55hRsaxgwZVAHvK32-Ye1g,6190
-prediction_market_agent_tooling/tools/web3_utils.py,sha256=cboATXNmEdn5RmPbVblHOwOdUMKBYrUK3GiI6i6Vzxo,9855
-prediction_market_agent_tooling-0.37.0.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.37.0.dist-info/METADATA,sha256=CaelrPuw-JpZ4VVfKQTy8v8pdKTeWPLrqnEfhqRFHJg,7595
-prediction_market_agent_tooling-0.37.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-prediction_market_agent_tooling-0.37.0.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.37.0.dist-info/RECORD,,
+prediction_market_agent_tooling/tools/web3_utils.py,sha256=nKRHmdLnWSKd3wpo-cysXGvhhrJ2Yf69sN2FFQfSt6s,10578
+prediction_market_agent_tooling-0.38.0.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.38.0.dist-info/METADATA,sha256=BCdK9YuigO1Rdphbyjb_MfbI1HgJJkQsnH53ZAF6opg,7634
+prediction_market_agent_tooling-0.38.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+prediction_market_agent_tooling-0.38.0.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.38.0.dist-info/RECORD,,