prediction-market-agent-tooling 0.63.4__py3-none-any.whl → 0.63.5__py3-none-any.whl

prediction_market_agent_tooling/markets/omen/data_models.py

@@ -1,6 +1,6 @@
 import typing as t
 
-from pydantic import BaseModel, ConfigDict, Field, computed_field
+from pydantic import BaseModel, ConfigDict, Field, computed_field, model_validator
 from web3 import Web3
 
 from prediction_market_agent_tooling.gtypes import (
@@ -17,6 +17,7 @@ from prediction_market_agent_tooling.gtypes import (
     xDai,
     xDaiWei,
 )
+from prediction_market_agent_tooling.loggers import logger
 from prediction_market_agent_tooling.markets.data_models import (
     Bet,
     Resolution,
@@ -35,6 +36,7 @@ from prediction_market_agent_tooling.tools.utils import (
     should_not_happen,
     utcnow,
 )
+from prediction_market_agent_tooling.tools.web3_utils import is_valid_wei
 
 OMEN_TRUE_OUTCOME = OutcomeStr("Yes")
 OMEN_FALSE_OUTCOME = OutcomeStr("No")
@@ -235,6 +237,27 @@ class OmenMarket(BaseModel):
     condition: Condition
     question: Question
 
+    @model_validator(mode="after")
+    def _model_validator(self) -> "OmenMarket":
+        if not all(is_valid_wei(number.value) for number in self.outcomeTokenAmounts):
+            # Sometimes we receive markets with outcomeTokenAmounts as `model.outcomeTokenAmounts=[OutcomeWei(24662799387878572), OutcomeWei(-24750000000000000)]`, which should be impossible.
+            # Current huntch is that it's a weird transitional status.
+            # Try to set them to zeros if market isn't open anymore (that's expected behaviour in such case),
+            # otherwise raise an error to investigate further.
+            if not self.is_open:
+                logger.warning(
+                    f"Market {self.url} has invalid {self.outcomeTokenAmounts=}, but isn't open anymore. Setting them to zeros."
+                )
+                self.outcomeTokenAmounts = [OutcomeWei(0) for _ in self.outcomes]
+                self.outcomeTokenMarginalPrices = None
+
+            else:
+                raise ValueError(
+                    f"Market {self.url} has invalid {self.outcomeTokenAmounts=}: {self=}"
+                )
+
+        return self
+
     @property
     def openingTimestamp(self) -> int:
         # This field is also available on this model itself, but for some reason it's typed to be optional,

prediction_market_agent_tooling/markets/omen/omen.py

@@ -86,10 +86,7 @@ from prediction_market_agent_tooling.tools.utils import (
     calculate_sell_amount_in_collateral,
     check_not_none,
 )
-from prediction_market_agent_tooling.tools.web3_utils import (
-    get_receipt_block_timestamp,
-    is_valid_wei,
-)
+from prediction_market_agent_tooling.tools.web3_utils import get_receipt_block_timestamp
 
 OMEN_DEFAULT_REALITIO_BOND_VALUE = xDai(0.01)
 OMEN_TINY_BET_AMOUNT = USD(0.00001)
@@ -339,11 +336,6 @@ class OmenAgentMarket(AgentMarket):
 
     @staticmethod
     def from_data_model(model: OmenMarket) -> "OmenAgentMarket":
-        if not all(is_valid_wei(number.value) for number in model.outcomeTokenAmounts):
-            raise ValueError(
-                f"Market {model.url} has invalid {model.outcomeTokenAmounts=}."
-            )
-
         return OmenAgentMarket(
             id=model.id,
             question=model.title,

{prediction_market_agent_tooling-0.63.4.dist-info → prediction_market_agent_tooling-0.63.5.dist-info}/METADATA

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

{prediction_market_agent_tooling-0.63.4.dist-info → prediction_market_agent_tooling-0.63.5.dist-info}/RECORD

@@ -50,8 +50,8 @@ 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/data_models.py,sha256=URvplDcTLeL6vfpkSgi8ln2iEKxt8w5qf6mTZkSyrCo,29189
-prediction_market_agent_tooling/markets/omen/omen.py,sha256=L2JcFdbjLGu2hF-mYnO-EUqo2vDKFUXstsoncSEVxdg,52167
+prediction_market_agent_tooling/markets/omen/data_models.py,sha256=T08irohOXQnNeL45mIhGYgpJTShHVgXvQyYnLX4YLmU,30477
+prediction_market_agent_tooling/markets/omen/omen.py,sha256=Jf2qSJJn0UUISpi24xYRwoVycGuYE42kZ2z1HRGl43w,51927
 prediction_market_agent_tooling/markets/omen/omen_constants.py,sha256=D9oflYKafLQiHYtB5sScMHqmXyzM8JP8J0yATmc4SQQ,233
 prediction_market_agent_tooling/markets/omen/omen_contracts.py,sha256=bCC9A7ZTJxMDJcPbl3jof6HcAGGHv1BrFq3RRWbkQ_c,28739
 prediction_market_agent_tooling/markets/omen/omen_resolving.py,sha256=Cyi9Ci-Z-K8WCZWVLs9oSuJC6qRobi_CpqDCno_5F-0,10238
@@ -121,8 +121,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=1xsyBBJfiEdSoMlceB2F8o2sCb6Z8-qNz11pEJFrdyE,6566
 prediction_market_agent_tooling/tools/web3_utils.py,sha256=zRq-eeBGWt8uUGN9G_WfjmJ0eVvO8aWE9S0Pz_Y6AOA,12342
-prediction_market_agent_tooling-0.63.4.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.63.4.dist-info/METADATA,sha256=_zWTgyTGpx-YPI1eYzV8YpbFyuLlM0Jc1wG9-2UuSI0,8689
-prediction_market_agent_tooling-0.63.4.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-prediction_market_agent_tooling-0.63.4.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.63.4.dist-info/RECORD,,
+prediction_market_agent_tooling-0.63.5.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.63.5.dist-info/METADATA,sha256=IaLaaqZx0vS4t15q4zcpbXt_leP49eUYv8K7rGUh-p8,8689
+prediction_market_agent_tooling-0.63.5.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+prediction_market_agent_tooling-0.63.5.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.63.5.dist-info/RECORD,,