prediction-market-agent-tooling 0.60.0.dev411__py3-none-any.whl → 0.60.0.dev418__py3-none-any.whl

prediction_market_agent_tooling/markets/omen/omen.py

@@ -408,7 +408,7 @@ class OmenAgentMarket(AgentMarket):
 
     @staticmethod
     def get_trade_balance(api_keys: APIKeys, web3: Web3 | None = None) -> xDai:
-        native_token_balance = get_balances(api_keys.public_key, web3=web3).xdai
+        native_token_balance = get_balances(api_keys.bet_from_address, web3=web3).xdai
         return xdai_type(
             wei_to_xdai(
                 KEEPING_ERC20_TOKEN.balanceOf(api_keys.bet_from_address, web3=web3)

prediction_market_agent_tooling/markets/omen/omen_resolving.py

@@ -26,7 +26,11 @@ from prediction_market_agent_tooling.markets.omen.omen_subgraph_handler import (
     OmenSubgraphHandler,
 )
 from prediction_market_agent_tooling.tools.utils import utcnow
-from prediction_market_agent_tooling.tools.web3_utils import ZERO_BYTES, xdai_to_wei
+from prediction_market_agent_tooling.tools.web3_utils import (
+    ZERO_BYTES,
+    wei_to_xdai,
+    xdai_to_wei,
+)
 
 
 def claim_bonds_on_realitio_questions(
@@ -116,7 +120,9 @@ def claim_bonds_on_realitio_question(
     current_balance = realitio_contract.balanceOf(public_key, web3=web3)
     # Keeping balance on Realitio is not useful, so it's recommended to just withdraw it.
     if current_balance > 0 and auto_withdraw:
-        logger.info(f"Withdrawing remaining balance {current_balance=}")
+        logger.info(
+            f"Withdrawing remaining balance {wei_to_xdai(current_balance)} xDai from Realitio."
+        )
         realitio_contract.withdraw(api_keys, web3=web3)
 
 

prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py

@@ -46,10 +46,12 @@ from prediction_market_agent_tooling.tools.web3_utils import (
     byte32_to_ipfscidv0,
 )
 
-# TODO: Agents don't know how to convert value between other tokens, we assume 1 unit = 1xDai = $1 (for example if market would be in wETH, betting 1 unit of wETH would be crazy :D)
-SAFE_COLLATERAL_TOKEN_MARKETS = (
-    WrappedxDaiContract().address,
-    sDaiContract().address,
+SAFE_COLLATERAL_TOKENS = (
+    WrappedxDaiContract(),
+    sDaiContract(),
+)
+SAFE_COLLATERAL_TOKENS_ADDRESSES = tuple(
+    contract.address for contract in SAFE_COLLATERAL_TOKENS
 )
 
 
@@ -326,7 +328,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
         excluded_questions: set[str] | None = None,  # question titles
         collateral_token_address_in: (
             tuple[ChecksumAddress, ...] | None
-        ) = SAFE_COLLATERAL_TOKEN_MARKETS,
+        ) = SAFE_COLLATERAL_TOKENS_ADDRESSES,
         category: str | None = None,
         creator_in: t.Sequence[HexAddress] | None = None,
     ) -> t.List[OmenMarket]:
@@ -392,7 +394,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
         outcomes: list[str] = OMEN_BINARY_MARKET_OUTCOMES,
         collateral_token_address_in: (
             tuple[ChecksumAddress, ...] | None
-        ) = SAFE_COLLATERAL_TOKEN_MARKETS,
+        ) = SAFE_COLLATERAL_TOKENS_ADDRESSES,
         category: str | None = None,
     ) -> t.List[OmenMarket]:
         """

prediction_market_agent_tooling/tools/caches/db_cache.py

@@ -14,6 +14,7 @@ from typing import (
     overload,
 )
 
+import psycopg2
 from pydantic import BaseModel
 from sqlalchemy import Column
 from sqlalchemy.dialects.postgresql import JSONB
@@ -200,12 +201,22 @@ def db_cache(
                 result=computed_result,
                 created_at=utcnow(),
             )
-            with DBManager(
-                api_keys.sqlalchemy_db_url.get_secret_value()
-            ).get_session() as session:
-                logger.info(f"Saving {cache_entry} into database.")
-                session.add(cache_entry)
-                session.commit()
+            # Do not raise an exception if saving to the database fails, just log it and let the agent continue the work.
+            try:
+                with DBManager(
+                    api_keys.sqlalchemy_db_url.get_secret_value()
+                ).get_session() as session:
+                    logger.info(f"Saving {cache_entry} into database.")
+                    session.add(cache_entry)
+                    session.commit()
+            except psycopg2.errors.UntranslatableCharacter as e:
+                logger.warning(
+                    f"Failed to save {cache_entry} into database, ignoring, because: {e}"
+                )
+            except Exception:
+                logger.exception(
+                    f"Failed to save {cache_entry} into database, ignoring."
+                )
 
         return computed_result
 

{prediction_market_agent_tooling-0.60.0.dev411.dist-info → prediction_market_agent_tooling-0.60.0.dev418.dist-info}/METADATA

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

{prediction_market_agent_tooling-0.60.0.dev411.dist-info → prediction_market_agent_tooling-0.60.0.dev418.dist-info}/RECORD

@@ -51,11 +51,11 @@ prediction_market_agent_tooling/markets/metaculus/data_models.py,sha256=FaBCTPPe
 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/data_models.py,sha256=sfaOpNk6oFIzxYQzs9EehqAT_19IxYJy9pns-UTepOc,28934
-prediction_market_agent_tooling/markets/omen/omen.py,sha256=WImTOfqfLsWEJHWxwWiBjor4jcHvafTmclhgeAcuZrg,51617
+prediction_market_agent_tooling/markets/omen/omen.py,sha256=nJGjVHKKWvrVtTRwLLdpZoOIpk9vuEtypl238nazO9c,51623
 prediction_market_agent_tooling/markets/omen/omen_constants.py,sha256=D9oflYKafLQiHYtB5sScMHqmXyzM8JP8J0yATmc4SQQ,233
 prediction_market_agent_tooling/markets/omen/omen_contracts.py,sha256=EXqBlVivbmW8aBQ65O09X2xkyesHAop49GUl1tUffWA,28648
-prediction_market_agent_tooling/markets/omen/omen_resolving.py,sha256=E1BVDvQd1qYtCxmfC94kJtGkmQqpGPHL3zTkcs5wW6M,9697
-prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py,sha256=mZ0CgKfHj0gLFq9plNpBhNqMuclb8V3qNagWfLYcpUc,38806
+prediction_market_agent_tooling/markets/omen/omen_resolving.py,sha256=dMxnPK2DF3aL4oZkBkbxwTcU7EMmi1vnTm0oC5d91yY,9781
+prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py,sha256=dQyz1RR1MlQncb1Slq7tk1Maql-sbb5YYE_sDe26MYA,38711
 prediction_market_agent_tooling/markets/polymarket/api.py,sha256=UZ4_TG8ceb9Y-qgsOKs8Qiv8zDt957QkT8IX2c83yqo,4800
 prediction_market_agent_tooling/markets/polymarket/data_models.py,sha256=Fd5PI5y3mJM8VHExBhWFWEnuuIKxQmIAXgBuoPDvNjw,4341
 prediction_market_agent_tooling/markets/polymarket/data_models_web.py,sha256=VZhVccTApygSKMmy6Au2G02JCJOKJnR_oVeKlaesuSg,12548
@@ -80,7 +80,7 @@ prediction_market_agent_tooling/tools/betting_strategies/market_moving.py,sha256
 prediction_market_agent_tooling/tools/betting_strategies/minimum_bet_to_win.py,sha256=-FUSuQQgjcWSSnoFxnlAyTeilY6raJABJVM2QKkFqAY,438
 prediction_market_agent_tooling/tools/betting_strategies/stretch_bet_between.py,sha256=THMXwFlskvzbjnX_OiYtDSzI8XVFyULWfP2525_9UGc,429
 prediction_market_agent_tooling/tools/betting_strategies/utils.py,sha256=kpIb-ci67Vc1Yqqaa-_S4OUkbhWSIYog4_Iwp69HU_k,97
-prediction_market_agent_tooling/tools/caches/db_cache.py,sha256=aafau_n_AUbLIwkyIRiTPgKB0dmM0767mSqyPDLF2A4,10576
+prediction_market_agent_tooling/tools/caches/db_cache.py,sha256=Q5E5cI5Mn65wq8N9Mv7on_lOnl9JOqzezvZc2ZbWhFo,11119
 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=gmSuANqlfwldFG7hc4uom0q1Z-mo40qfYh5TDUKLFTs,20518
@@ -117,8 +117,8 @@ prediction_market_agent_tooling/tools/tokens/main_token.py,sha256=v5PfD2m9m92bJp
 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=2PXZfGRrDVZD60agVpBN4JkOF0YsNBXgTEH1y-V71uQ,12723
-prediction_market_agent_tooling-0.60.0.dev411.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.60.0.dev411.dist-info/METADATA,sha256=xWyT-Hk2XabfoQDMmGASUG2wSlZ_-PDDpa2B14CridU,8636
-prediction_market_agent_tooling-0.60.0.dev411.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-prediction_market_agent_tooling-0.60.0.dev411.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.60.0.dev411.dist-info/RECORD,,
+prediction_market_agent_tooling-0.60.0.dev418.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.60.0.dev418.dist-info/METADATA,sha256=jQnA7q4cVvDbsuTDW4z_DpdMmeSwrdN6bRn_PFvPeR4,8636
+prediction_market_agent_tooling-0.60.0.dev418.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+prediction_market_agent_tooling-0.60.0.dev418.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.60.0.dev418.dist-info/RECORD,,