prediction-market-agent-tooling 0.61.1.dev477__py3-none-any.whl → 0.61.1.dev483__py3-none-any.whl

prediction_market_agent_tooling/markets/seer/data_models.py

@@ -5,7 +5,6 @@ from urllib.parse import urljoin
 
 from pydantic import BaseModel, ConfigDict, Field
 from web3 import Web3
-from web3.constants import ADDRESS_ZERO
 
 from prediction_market_agent_tooling.config import RPCConfig
 from prediction_market_agent_tooling.gtypes import (
@@ -13,37 +12,13 @@ from prediction_market_agent_tooling.gtypes import (
     HexAddress,
     HexBytes,
     Probability,
-    Wei,
-    xdai_type,
 )
-from prediction_market_agent_tooling.loggers import logger
 from prediction_market_agent_tooling.markets.data_models import Resolution
-from prediction_market_agent_tooling.tools.cow.cow_manager import CowManager
+from prediction_market_agent_tooling.markets.seer.price_manager import PriceManager
+from prediction_market_agent_tooling.markets.seer.subgraph_data_models import (
+    SeerParentMarket,
+)
 from prediction_market_agent_tooling.tools.datetime_utc import DatetimeUTC
-from prediction_market_agent_tooling.tools.web3_utils import xdai_to_wei
-
-
-class CreateCategoricalMarketsParams(BaseModel):
-    model_config = ConfigDict(populate_by_name=True)
-
-    market_name: str = Field(..., alias="marketName")
-    outcomes: list[str]
-    # Only relevant for scalar markets
-    question_start: str = Field(alias="questionStart", default="")
-    question_end: str = Field(alias="questionEnd", default="")
-    outcome_type: str = Field(alias="outcomeType", default="")
-
-    # Not needed for non-conditional markets.
-    parent_outcome: int = Field(alias="parentOutcome", default=0)
-    parent_market: HexAddress = Field(alias="parentMarket", default=ADDRESS_ZERO)
-
-    category: str
-    lang: str
-    lower_bound: int = Field(alias="lowerBound", default=0)
-    upper_bound: int = Field(alias="upperBound", default=0)
-    min_bond: Wei = Field(..., alias="minBond")
-    opening_time: int = Field(..., alias="openingTime")
-    token_names: list[str] = Field(..., alias="tokenNames")
 
 
 class SeerOutcomeEnum(str, Enum):
@@ -84,10 +59,6 @@ class SeerOutcomeEnum(str, Enum):
             raise ValueError(f"Unknown outcome: {self}")
 
 
-class SeerParentMarket(BaseModel):
-    id: HexBytes
-
-
 SEER_BASE_URL = "https://app.seer.pm"
 
 
@@ -110,6 +81,7 @@ class SeerMarket(BaseModel):
     has_answers: bool | None = Field(alias="hasAnswers")
     payout_reported: bool = Field(alias="payoutReported")
     payout_numerators: list[int] = Field(alias="payoutNumerators")
+    outcomes_supply: int = Field(alias="outcomesSupply")
 
     @property
     def has_valid_answer(self) -> bool:
@@ -188,65 +160,10 @@ class SeerMarket(BaseModel):
 
     @property
     def current_p_yes(self) -> Probability:
-        price_data = {}
-        for idx in range(len(self.outcomes)):
-            wrapped_token = self.wrapped_tokens[idx]
-            price = self._get_price_for_token(
-                token=Web3.to_checksum_address(wrapped_token)
-            )
-            price_data[idx] = price
-
-        if sum(price_data.values()) == 0:
-            logger.warning(
-                f"Could not get p_yes for market {self.id.hex()}, all price quotes are 0."
-            )
-            return Probability(0)
-
-        yes_idx = self.outcome_as_enums[SeerOutcomeEnum.YES]
-        price_yes = price_data[yes_idx] / sum(price_data.values())
-        return Probability(price_yes)
-
-    def _get_price_for_token(self, token: ChecksumAddress) -> float:
-        collateral_exchange_amount = xdai_to_wei(xdai_type(1))
-        try:
-            quote = CowManager().get_quote(
-                collateral_token=self.collateral_token_contract_address_checksummed,
-                buy_token=token,
-                sell_amount=collateral_exchange_amount,
-            )
-        except Exception as e:
-            logger.warning(f"Could not get quote for {token=}, returning price 0. {e=}")
-            return 0
-
-        return collateral_exchange_amount / float(quote.quote.buyAmount.root)
+        p = PriceManager(self)
+        return p.current_market_p_yes
 
     @property
     def url(self) -> str:
         chain_id = RPCConfig().chain_id
         return urljoin(SEER_BASE_URL, f"markets/{chain_id}/{self.id.hex()}")
-
-
-class SeerToken(BaseModel):
-    id: HexBytes
-    name: str
-    symbol: str
-
-
-class SeerPool(BaseModel):
-    model_config = ConfigDict(populate_by_name=True)
-    id: HexBytes
-    liquidity: int
-    token0: SeerToken
-    token1: SeerToken
-    token0Price: float
-    token1Price: float
-    sqrtPrice: int
-
-
-class NewMarketEvent(BaseModel):
-    market: HexAddress
-    marketName: str
-    parentMarket: HexAddress
-    conditionId: HexBytes
-    questionId: HexBytes
-    questionsIds: list[HexBytes]

prediction_market_agent_tooling/markets/seer/price_manager.py

@@ -0,0 +1,124 @@
+from web3 import Web3
+
+from prediction_market_agent_tooling.gtypes import ChecksumAddress
+from prediction_market_agent_tooling.gtypes import Probability, xdai_type
+from prediction_market_agent_tooling.loggers import logger
+from prediction_market_agent_tooling.markets.seer.data_models import (
+    SeerMarket,
+    SeerOutcomeEnum,
+)
+from prediction_market_agent_tooling.markets.seer.seer_subgraph_handler import (
+    SeerSubgraphHandler,
+)
+from prediction_market_agent_tooling.markets.seer.subgraph_data_models import SeerPool
+from prediction_market_agent_tooling.tools.caches.inmemory_cache import (
+    persistent_inmemory_cache,
+)
+from prediction_market_agent_tooling.tools.cow.cow_manager import CowManager
+from prediction_market_agent_tooling.tools.web3_utils import xdai_to_wei
+
+
+class PriceManager:
+    def __init__(self, market: SeerMarket):
+        self.seer_market = market
+        self.subgraph_handler = SeerSubgraphHandler()
+
+    def fetch_outcome_price_for_seer_market(self) -> float:
+        price_data = {}
+        for idx in range(len(self.seer_market.outcomes)):
+            wrapped_token = self.seer_market.wrapped_tokens[idx]
+            price = self.get_price_for_token(
+                token=Web3.to_checksum_address(wrapped_token)
+            )
+            price_data[idx] = price
+
+        if sum(price_data.values()) == 0:
+            logger.warning(
+                f"Could not get p_yes for market {self.seer_market.id.hex()}, all price quotes are 0."
+            )
+            return Probability(0)
+
+        yes_idx = self.seer_market.outcome_as_enums[SeerOutcomeEnum.YES]
+        price_yes = price_data[yes_idx] / sum(price_data.values())
+        return Probability(price_yes)
+
+    def current_market_p_yes(self) -> Probability:
+        price_data = {}
+        for idx in range(len(self.seer_market.outcomes)):
+            wrapped_token = self.seer_market.wrapped_tokens[idx]
+            price = self.get_price_for_token(
+                token=Web3.to_checksum_address(wrapped_token)
+            )
+            price_data[idx] = price
+
+        if sum(price_data.values()) == 0:
+            logger.warning(
+                f"Could not get p_yes for market {self.seer_market.id.hex()}, all price quotes are 0."
+            )
+            return Probability(0)
+
+        yes_idx = self.seer_market.outcome_as_enums[SeerOutcomeEnum.YES]
+        no_idx = self.seer_market.outcome_as_enums[SeerOutcomeEnum.NO]
+        price_yes = price_data[yes_idx]
+        price_no = price_data[no_idx]
+        if price_yes and not price_no:
+            # We simply return p_yes since it's probably a bug that p_no wasn't found.
+            return Probability(price_yes)
+        elif price_no and not price_yes:
+            # We return the complement of p_no (and ignore invalid).
+            return Probability(1.0 - price_no)
+        else:
+            # If all prices are available, we normalize price_yes by the other prices for the final probability.
+            price_yes = price_yes / sum(price_data.values())
+            return Probability(price_yes)
+
+    @persistent_inmemory_cache
+    def get_price_for_token(self, token: ChecksumAddress) -> float:
+        collateral_exchange_amount = xdai_to_wei(xdai_type(1))
+        try:
+            quote = CowManager().get_quote(
+                collateral_token=self.seer_market.collateral_token_contract_address_checksummed,
+                buy_token=token,
+                sell_amount=collateral_exchange_amount,
+            )
+        except Exception as e:
+            logger.warning(
+                f"Could not get quote for {token=} from Cow, exception {e=}. Falling back to pools. "
+            )
+            price = self.get_token_price_from_pools(token=token)
+            return price
+
+        return collateral_exchange_amount / float(quote.quote.buyAmount.root)
+
+    @staticmethod
+    def _pool_token0_matches_token(token: ChecksumAddress, pool: SeerPool) -> bool:
+        return pool.token0.id.hex().lower() == token.lower()
+
+    def get_token_price_from_pools(self, token: ChecksumAddress) -> float:
+        pool = self.subgraph_handler.get_pool_by_token(token_address=token)
+
+        if not pool:
+            logger.warning(f"Could not find a pool for {token=}, returning 0.")
+            return 0
+        # Check if other token is market's collateral (sanity check).
+
+        collateral_address = (
+            pool.token0.id
+            if self._pool_token0_matches_token(token=token, pool=pool)
+            else pool.token1.id
+        )
+        if (
+            collateral_address.hex().lower()
+            != self.seer_market.collateral_token.lower()
+        ):
+            logger.warning(
+                f"Pool {pool.id.hex()} has collateral mismatch with market. Collateral from pool {collateral_address.hex()}, collateral from market {self.seer_market.collateral_token}, returning 0."
+            )
+            return 0
+
+        price_in_collateral_units = (
+            pool.token0Price
+            if self._pool_token0_matches_token(pool)
+            else pool.token1Price
+        )
+        return price_in_collateral_units

prediction_market_agent_tooling/markets/seer/seer.py

@@ -352,4 +352,4 @@ def extract_market_address_from_tx(
         .process_receipt(tx_receipt)
     )
     new_market_event = NewMarketEvent(**event_logs[0]["args"])
-    return Web3.to_checksum_address(new_market_event.market)
+    return Web3.to_checksum_address(new_market_event.seer_market)

prediction_market_agent_tooling/markets/seer/seer_subgraph_handler.py

@@ -5,6 +5,7 @@ from typing import Any
 from subgrounds import FieldPath
 from web3.constants import ADDRESS_ZERO
 
+from prediction_market_agent_tooling.gtypes import ChecksumAddress
 from prediction_market_agent_tooling.markets.agent_market import FilterBy, SortBy
 from prediction_market_agent_tooling.markets.base_subgraph_handler import (
     BaseSubgraphHandler,
@@ -49,6 +50,7 @@ class SeerSubgraphHandler(BaseSubgraphHandler):
             markets_field.creator,
             markets_field.conditionId,
             markets_field.marketName,
+            markets_field.outcomesSupply,
             markets_field.parentOutcome,
             markets_field.outcomes,
             markets_field.payoutReported,
@@ -124,8 +126,12 @@ class SeerSubgraphHandler(BaseSubgraphHandler):
             case SortBy.CLOSING_SOONEST:
                 sort_direction = "asc"
                 sort_by_field = self.seer_subgraph.Market.openingTs
-            # ToDo - Implement liquidity conditions by looking up Swapr subgraph.
-            case SortBy.NONE | SortBy.HIGHEST_LIQUIDITY | SortBy.LOWEST_LIQUIDITY:
+            case SortBy.HIGHEST_LIQUIDITY | SortBy.LOWEST_LIQUIDITY:
+                sort_direction = (
+                    "desc" if sort_by == SortBy.HIGHEST_LIQUIDITY else "asc"
+                )
+                sort_by_field = self.seer_subgraph.Market.outcomesSupply
+            case SortBy.NONE:
                 sort_direction = None
                 sort_by_field = None
             case _:
@@ -187,7 +193,7 @@ class SeerSubgraphHandler(BaseSubgraphHandler):
         return binary_markets
 
     def get_market_by_id(self, market_id: HexBytes) -> SeerMarket:
-        markets_field = self.seer_subgraph.Query.market(id=market_id.hex().lower())
+        markets_field = self.seer_subgraph.Query.seer_market(id=market_id.hex().lower())
         fields = self._get_fields_for_markets(markets_field)
         markets = self.do_query(fields=fields, pydantic_model=SeerMarket)
         if len(markets) != 1:
@@ -212,17 +218,19 @@ class SeerSubgraphHandler(BaseSubgraphHandler):
         ]
         return fields
 
-    def get_swapr_pools_for_market(self, market: SeerMarket) -> list[SeerPool]:
+    def get_pool_by_token(self, token_address: ChecksumAddress) -> SeerPool | None:
         # We iterate through the wrapped tokens and put them in a where clause so that we hit the subgraph endpoint just once.
         wheres = []
-        for wrapped_token in market.wrapped_tokens:
-            wheres.extend(
-                [
-                    {"token0": wrapped_token.lower()},
-                    {"token1": wrapped_token.lower()},
-                ]
-            )
+        wheres.extend(
+            [
+                {"token0": token_address.lower()},
+                {"token1": token_address.lower()},
+            ]
+        )
         pools_field = self.swapr_algebra_subgraph.Query.pools(where={"or": wheres})
         fields = self._get_fields_for_pools(pools_field)
         pools = self.do_query(fields=fields, pydantic_model=SeerPool)
-        return pools
+        # We assume there is only one pool for outcomeToken/sDAI.
+        if pools:
+            return pools[0]
+        return None

prediction_market_agent_tooling/markets/seer/subgraph_data_models.py

@@ -0,0 +1,56 @@
+from pydantic import BaseModel, ConfigDict, Field
+
+from prediction_market_agent_tooling.gtypes import HexBytes, HexAddress
+
+
+class SeerToken(BaseModel):
+    id: HexBytes
+    name: str
+    symbol: str
+
+
+class SeerPool(BaseModel):
+    model_config = ConfigDict(populate_by_name=True)
+    id: HexBytes
+    liquidity: int
+    token0: SeerToken
+    token1: SeerToken
+    token0Price: float
+    token1Price: float
+    sqrtPrice: int
+
+
+class NewMarketEvent(BaseModel):
+    market: HexAddress
+    marketName: str
+    parentMarket: HexAddress
+    conditionId: HexBytes
+    questionId: HexBytes
+    questionsIds: list[HexBytes]
+
+
+class CreateCategoricalMarketsParams(BaseModel):
+    model_config = ConfigDict(populate_by_name=True)
+
+    market_name: str = Field(..., alias="marketName")
+    outcomes: list[str]
+    # Only relevant for scalar markets
+    question_start: str = Field(alias="questionStart", default="")
+    question_end: str = Field(alias="questionEnd", default="")
+    outcome_type: str = Field(alias="outcomeType", default="")
+
+    # Not needed for non-conditional markets.
+    parent_outcome: int = Field(alias="parentOutcome", default=0)
+    parent_market: HexAddress = Field(alias="parentMarket", default=ADDRESS_ZERO)
+
+    category: str
+    lang: str
+    lower_bound: int = Field(alias="lowerBound", default=0)
+    upper_bound: int = Field(alias="upperBound", default=0)
+    min_bond: Wei = Field(..., alias="minBond")
+    opening_time: int = Field(..., alias="openingTime")
+    token_names: list[str] = Field(..., alias="tokenNames")
+
+
+class SeerParentMarket(BaseModel):
+    id: HexBytes

{prediction_market_agent_tooling-0.61.1.dev477.dist-info → prediction_market_agent_tooling-0.61.1.dev483.dist-info}/METADATA

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

{prediction_market_agent_tooling-0.61.1.dev477.dist-info → prediction_market_agent_tooling-0.61.1.dev483.dist-info}/RECORD

@@ -61,10 +61,12 @@ prediction_market_agent_tooling/markets/polymarket/data_models.py,sha256=Fd5PI5y
 prediction_market_agent_tooling/markets/polymarket/data_models_web.py,sha256=VZhVccTApygSKMmy6Au2G02JCJOKJnR_oVeKlaesuSg,12548
 prediction_market_agent_tooling/markets/polymarket/polymarket.py,sha256=NRoZK71PtH8kkangMqme7twcAXhRJSSabbmOir-UnAI,3418
 prediction_market_agent_tooling/markets/polymarket/utils.py,sha256=8kTeVjXPcXC6DkDvWYsZQLY7x8DS6CEp_yznSEazsNU,2037
-prediction_market_agent_tooling/markets/seer/data_models.py,sha256=67hUD-Bs_bJd_j0iuxXTHR0lcWNF3Tkk-fnCi6kSSoE,8268
-prediction_market_agent_tooling/markets/seer/seer.py,sha256=r21sXj_4_oIG2L1N5l56vEYGI_q2RGem_6G-Sixkbck,12954
+prediction_market_agent_tooling/markets/seer/data_models.py,sha256=EDzVMgBvum400VQl4IUvdnchBszF6M_4LWxMWkeTIHY,5576
+prediction_market_agent_tooling/markets/seer/price_manager.py,sha256=tuozfXp2dwgUQXvD4G4UAOzs-rIm6vgevbi804vv8-0,5102
+prediction_market_agent_tooling/markets/seer/seer.py,sha256=Anxp66UUzIIKt5wgLoqtUjUM-5HQ0wC5f5rDiYwrRRg,12959
 prediction_market_agent_tooling/markets/seer/seer_contracts.py,sha256=E7CYAKZiK6cg3dyj1kJuIPKSYYUft98F64shF5S0g4s,2730
-prediction_market_agent_tooling/markets/seer/seer_subgraph_handler.py,sha256=-XpRmK5gmXCmf1h82PMkNf_AizFdzI90m9P3OzsVfnw,8684
+prediction_market_agent_tooling/markets/seer/seer_subgraph_handler.py,sha256=2M6vmUxMKQU61VNCp4Ykican6XH1LOp257--GjCJZx8,8974
+prediction_market_agent_tooling/markets/seer/subgraph_data_models.py,sha256=uPTCYTg_lS91Pb7JCU-neVEgAQe3zSOtkttlhQGlcBI,1599
 prediction_market_agent_tooling/monitor/financial_metrics/financial_metrics.py,sha256=fjIgjDIx5MhH5mwf7S0cspLOOSU3elYLhGYoIiM26mU,2746
 prediction_market_agent_tooling/monitor/markets/manifold.py,sha256=TS4ERwTfQnot8dhekNyVNhJYf5ysYsjF-9v5_kM3aVI,3334
 prediction_market_agent_tooling/monitor/markets/metaculus.py,sha256=LOnyWWBFdg10-cTWdb76nOsNjDloO8OfMT85GBzRCFI,1455
@@ -117,8 +119,8 @@ prediction_market_agent_tooling/tools/tokens/main_token.py,sha256=7JPgVF4RbiFzLD
 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=3wfqNxvMn44ivweFRoeKNVb9QRtFd7kFtp7VUY5juEE,12862
-prediction_market_agent_tooling-0.61.1.dev477.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.61.1.dev477.dist-info/METADATA,sha256=cTdPU6VSva5PSgofTUucAuFXtbSnDug4fvTtvi-s1s8,8636
-prediction_market_agent_tooling-0.61.1.dev477.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-prediction_market_agent_tooling-0.61.1.dev477.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.61.1.dev477.dist-info/RECORD,,
+prediction_market_agent_tooling-0.61.1.dev483.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.61.1.dev483.dist-info/METADATA,sha256=078bYVh1iho-ehDfULc_ZXtRaQc71KnHdmJErDU_B5k,8636
+prediction_market_agent_tooling-0.61.1.dev483.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+prediction_market_agent_tooling-0.61.1.dev483.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.61.1.dev483.dist-info/RECORD,,