prediction-market-agent-tooling 0.63.10.dev565__py3-none-any.whl → 0.64.0__py3-none-any.whl

prediction_market_agent_tooling/deploy/agent.py

@@ -577,7 +577,7 @@ class DeployableTraderAgent(DeployablePredictionAgent):
 
         if market.get_trade_balance(api_keys) < min_required_balance_to_trade:
             raise OutOfFundsError(
-                f"Minimum required balance {min_required_balance_to_trade} for agent is not met."
+                f"Minimum required balance {min_required_balance_to_trade} for agent {api_keys.bet_from_address=} is not met."
             )
 
     def get_betting_strategy(self, market: AgentMarket) -> BettingStrategy:

prediction_market_agent_tooling/markets/omen/cow_contracts.py

@@ -0,0 +1,34 @@
+import os
+
+from web3 import Web3
+
+from prediction_market_agent_tooling.config import APIKeys
+from prediction_market_agent_tooling.gtypes import ABI, HexBytes
+from prediction_market_agent_tooling.tools.contract import (
+    ContractOnGnosisChain,
+    abi_field_validator,
+)
+
+
+class CowGPv2SettlementContract(ContractOnGnosisChain):
+    # Contract ABI taken from https://github.com/cowprotocol/cow-sdk/blob/main/abi/GPv2Settlement.json.
+    abi: ABI = abi_field_validator(
+        os.path.join(
+            os.path.dirname(os.path.realpath(__file__)),
+            "../../abis/gvp2_settlement.abi.json",
+        )
+    )
+
+    def setPreSignature(
+        self,
+        api_keys: APIKeys,
+        orderId: HexBytes,
+        signed: bool,
+        web3: Web3 | None = None,
+    ) -> None:
+        self.send(
+            api_keys=api_keys,
+            function_name="setPreSignature",
+            function_params=[orderId, signed],
+            web3=web3,
+        )

prediction_market_agent_tooling/markets/omen/omen.py

@@ -1353,7 +1353,7 @@ def send_keeping_token_to_eoa_xdai(
 
     else:
         raise OutOfFundsError(
-            f"Current wxDai balance ({current_balances_eoa=}, {current_balances_betting=}) is less than the required minimum wxDai to withdraw {need_to_withdraw}."
+            f"Current wxDai balance ({current_balances_eoa=} for {api_keys.public_key=}, {current_balances_betting=} for {api_keys.bet_from_address=}) is less than the required minimum wxDai to withdraw {need_to_withdraw}."
         )
 
 

prediction_market_agent_tooling/markets/omen/omen_contracts.py

@@ -435,6 +435,24 @@ class WETHContract(ContractERC20OnGnosisChain):
     )
 
 
+class EUReContract(ContractERC20OnGnosisChain):
+    address: ChecksumAddress = Web3.to_checksum_address(
+        "0xcB444e90D8198415266c6a2724b7900fb12FC56E"
+    )
+
+
+class SAFEContract(ContractERC20OnGnosisChain):
+    address: ChecksumAddress = Web3.to_checksum_address(
+        "0x4d18815D14fe5c3304e87B3FA18318baa5c23820"
+    )
+
+
+class COWContract(ContractERC20OnGnosisChain):
+    address: ChecksumAddress = Web3.to_checksum_address(
+        "0x177127622c4A00F3d409B75571e12cB3c8973d3c"
+    )
+
+
 class WrappedxDaiContract(ContractDepositableWrapperERC20OnGnosisChain):
     address: ChecksumAddress = WRAPPED_XDAI_CONTRACT_ADDRESS
 
@@ -893,30 +911,6 @@ class OmenThumbnailMapping(ContractOnGnosisChain):
         )
 
 
-class CowGPv2SettlementContract(ContractOnGnosisChain):
-    # Contract ABI taken from https://github.com/cowprotocol/cow-sdk/blob/main/abi/GPv2Settlement.json.
-    abi: ABI = abi_field_validator(
-        os.path.join(
-            os.path.dirname(os.path.realpath(__file__)),
-            "../../abis/gvp2_settlement.abi.json",
-        )
-    )
-
-    def setPreSignature(
-        self,
-        api_keys: APIKeys,
-        orderId: HexBytes,
-        signed: bool,
-        web3: Web3 | None = None,
-    ) -> None:
-        self.send(
-            api_keys=api_keys,
-            function_name="setPreSignature",
-            function_params=[orderId, signed],
-            web3=web3,
-        )
-
-
 class CollateralTokenChoice(str, Enum):
     wxdai = "wxdai"
     sdai = "sdai"

prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py

@@ -30,7 +30,12 @@ from prediction_market_agent_tooling.markets.omen.data_models import (
     RealityResponse,
 )
 from prediction_market_agent_tooling.markets.omen.omen_contracts import (
+    COWContract,
+    EUReContract,
+    GNOContract,
     OmenThumbnailMapping,
+    SAFEContract,
+    WETHContract,
     WrappedxDaiContract,
     sDaiContract,
 )
@@ -51,6 +56,11 @@ from prediction_market_agent_tooling.tools.web3_utils import (
 SAFE_COLLATERAL_TOKENS = (
     WrappedxDaiContract(),
     sDaiContract(),
+    GNOContract(),
+    WETHContract(),
+    EUReContract(),
+    SAFEContract(),
+    COWContract(),
 )
 SAFE_COLLATERAL_TOKENS_ADDRESSES = tuple(
     contract.address for contract in SAFE_COLLATERAL_TOKENS

prediction_market_agent_tooling/tools/cow/cow_order.py

@@ -30,7 +30,7 @@ from web3 import Web3
 from prediction_market_agent_tooling.config import APIKeys
 from prediction_market_agent_tooling.gtypes import ChecksumAddress, HexBytes, Wei
 from prediction_market_agent_tooling.loggers import logger
-from prediction_market_agent_tooling.markets.omen.omen_contracts import (
+from prediction_market_agent_tooling.markets.omen.cow_contracts import (
     CowGPv2SettlementContract,
 )
 from prediction_market_agent_tooling.tools.contract import ContractERC20OnGnosisChain

{prediction_market_agent_tooling-0.63.10.dev565.dist-info → prediction_market_agent_tooling-0.64.0.dist-info}/METADATA

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

{prediction_market_agent_tooling-0.63.10.dev565.dist-info → prediction_market_agent_tooling-0.64.0.dist-info}/RECORD

@@ -23,7 +23,7 @@ prediction_market_agent_tooling/benchmark/agents.py,sha256=B1-uWdyeN4GGKMWGK_-Cc
 prediction_market_agent_tooling/benchmark/benchmark.py,sha256=MqTiaaJ3cYiOLUVR7OyImLWxcEya3Rl5JyFYW-K0lwM,17097
 prediction_market_agent_tooling/benchmark/utils.py,sha256=D0MfUkVZllmvcU0VOurk9tcKT7JTtwwOp-63zuCBVuc,2880
 prediction_market_agent_tooling/config.py,sha256=QWybGq-DfjlRKG2G4l_4kJEyH3pmUvmcw4n2d2-z3eQ,10183
-prediction_market_agent_tooling/deploy/agent.py,sha256=tEYX133rSsZ6b9kx-RtZSHxkdN8LZvn0P6CVz6KOJlc,26531
+prediction_market_agent_tooling/deploy/agent.py,sha256=dobqPUQkaDPhsvMmXwibNKu4hSSTXTvmfa3F46ylLBc,26560
 prediction_market_agent_tooling/deploy/agent_example.py,sha256=dIIdZashExWk9tOdyDjw87AuUcGyM7jYxNChYrVK2dM,1001
 prediction_market_agent_tooling/deploy/betting_strategy.py,sha256=p25t7VU7I4hSkSl6SpzI_W55kLbYEySQdBqeschmARY,12918
 prediction_market_agent_tooling/deploy/constants.py,sha256=M5ty8URipYMGe_G-RzxRydK3AFL6CyvmqCraJUrLBnE,82
@@ -52,12 +52,13 @@ prediction_market_agent_tooling/markets/metaculus/api.py,sha256=4TRPGytQQbSdf42D
 prediction_market_agent_tooling/markets/metaculus/data_models.py,sha256=FaBCTPPezXbBwZ9p791CiVgQ4vB696xnMbz9XVXmiVI,3267
 prediction_market_agent_tooling/markets/metaculus/metaculus.py,sha256=86TIx6cavEWc8Cv4KpZxSvwiSw9oFybXE3YB49pg-CA,4377
 prediction_market_agent_tooling/markets/omen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+prediction_market_agent_tooling/markets/omen/cow_contracts.py,sha256=sl1L4cK5nAJwZ2wdhLzqh8p7h_IEValNvLwKUlInKxw,957
 prediction_market_agent_tooling/markets/omen/data_models.py,sha256=dtWbRy9tpasssQP80ZFYxAc_9pnVD-zUpVjenBqurnA,30221
-prediction_market_agent_tooling/markets/omen/omen.py,sha256=Jf2qSJJn0UUISpi24xYRwoVycGuYE42kZ2z1HRGl43w,51927
+prediction_market_agent_tooling/markets/omen/omen.py,sha256=BvTFQ9MZqo9q0wEaE2TzsZ8Q_yPGIZQehehnDIJ3id0,51987
 prediction_market_agent_tooling/markets/omen/omen_constants.py,sha256=D9oflYKafLQiHYtB5sScMHqmXyzM8JP8J0yATmc4SQQ,233
-prediction_market_agent_tooling/markets/omen/omen_contracts.py,sha256=aBhOKild5Ly8SylmvmuEglSbwpJRyod6mkXz41YDuS8,29423
+prediction_market_agent_tooling/markets/omen/omen_contracts.py,sha256=H7jFVLcdbxq6gxWS77YzBFHLnfaItU6MXXVzPzDCkwM,29236
 prediction_market_agent_tooling/markets/omen/omen_resolving.py,sha256=Cyi9Ci-Z-K8WCZWVLs9oSuJC6qRobi_CpqDCno_5F-0,10238
-prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py,sha256=6N6RJ-noC7gkoDBeYdBSXp1QI3u4iftBLu8Dtew9yU4,39017
+prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py,sha256=vwjAsxgBvIxzlYealHJIzEC6Q6hFv6qKssJ8PrsgoFs,39203
 prediction_market_agent_tooling/markets/polymarket/api.py,sha256=UZ4_TG8ceb9Y-qgsOKs8Qiv8zDt957QkT8IX2c83yqo,4800
 prediction_market_agent_tooling/markets/polymarket/data_models.py,sha256=utGN-Lhjsa-RHX5WW_jcqgWXtRkEZn0JemwYZkt3Lng,4344
 prediction_market_agent_tooling/markets/polymarket/data_models_web.py,sha256=LVEsNw2nUx5poiU1m803NNqG5-fs8-MODQRyGLqy4mE,12585
@@ -90,7 +91,7 @@ prediction_market_agent_tooling/tools/caches/inmemory_cache.py,sha256=ZW5iI5rmjq
 prediction_market_agent_tooling/tools/caches/serializers.py,sha256=vFDx4fsPxclXp2q0sv27j4al_M_Tj9aR2JJP-xNHQXA,2151
 prediction_market_agent_tooling/tools/contract.py,sha256=1ZFp_VoqTjM8cqOfAhco2Ht0DTqakjhZpuZUrAXr28Q,21332
 prediction_market_agent_tooling/tools/costs.py,sha256=EaAJ7v9laD4VEV3d8B44M4u3_oEO_H16jRVCdoZ93Uw,954
-prediction_market_agent_tooling/tools/cow/cow_order.py,sha256=nERAci2yNnpCLvotBl9B8gCOeZR058dCa4YwhUcK0PQ,8798
+prediction_market_agent_tooling/tools/cow/cow_order.py,sha256=gakwBMVD287v4FrgJC-qwpc5zZHu9yLfTzcmhht-BXA,8797
 prediction_market_agent_tooling/tools/custom_exceptions.py,sha256=Fh8z1fbwONvP4-j7AmV_PuEcoqb6-QXa9PJ9m7guMcM,93
 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
@@ -123,8 +124,8 @@ prediction_market_agent_tooling/tools/tokens/usd.py,sha256=yuW8iPPtcpP4eLH2nORMD
 prediction_market_agent_tooling/tools/transaction_cache.py,sha256=K5YKNL2_tR10Iw2TD9fuP-CTGpBbZtNdgbd0B_R7pjg,1814
 prediction_market_agent_tooling/tools/utils.py,sha256=AC2a68jwASMWuQi-w8twl8b_M52YwrEJ81abmuEaqMY,6661
 prediction_market_agent_tooling/tools/web3_utils.py,sha256=zRq-eeBGWt8uUGN9G_WfjmJ0eVvO8aWE9S0Pz_Y6AOA,12342
-prediction_market_agent_tooling-0.63.10.dev565.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.63.10.dev565.dist-info/METADATA,sha256=YMqiprzCES9UDZBzP6wIBYJbSO8sWrdYsaMOAaJze8w,8698
-prediction_market_agent_tooling-0.63.10.dev565.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-prediction_market_agent_tooling-0.63.10.dev565.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.63.10.dev565.dist-info/RECORD,,
+prediction_market_agent_tooling-0.64.0.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.64.0.dist-info/METADATA,sha256=KnjNTkpA-CeHrfoyRAXM5nEXMLYvXa-H24R0hBeV1Gk,8690
+prediction_market_agent_tooling-0.64.0.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+prediction_market_agent_tooling-0.64.0.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.64.0.dist-info/RECORD,,