prediction-market-agent-tooling 0.63.4__tar.gz → 0.63.6__tar.gz

{prediction_market_agent_tooling-0.63.4 → prediction_market_agent_tooling-0.63.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: prediction-market-agent-tooling
-Version: 0.63.4
+Version: 0.63.6
 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 → prediction_market_agent_tooling-0.63.6}/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 or not self.liquidityParameter:
+                logger.warning(
+                    f"Market {self.url} has invalid {self.outcomeTokenAmounts=}, but isn't open anymore or doesn't have any liquidity ({self.is_open=}, {self.liquidityParameter=}). 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.model_dump()=}"
+                )
+
+        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-0.63.4 → prediction_market_agent_tooling-0.63.6}/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 → prediction_market_agent_tooling-0.63.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "prediction-market-agent-tooling"
-version = "0.63.4"
+version = "0.63.6"
 description = "Tools to benchmark, deploy and monitor prediction market agents."
 authors = ["Gnosis"]
 readme = "README.md"