prediction-market-agent-tooling 0.61.1.dev484__py3-none-any.whl → 0.61.1.dev486__py3-none-any.whl

prediction_market_agent_tooling/markets/seer/data_models.py

@@ -11,21 +11,12 @@ from prediction_market_agent_tooling.gtypes import (
     ChecksumAddress,
     HexAddress,
     HexBytes,
-    Probability,
-    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.markets.seer.subgraph_data_models import (
     SeerParentMarket,
-    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.datetime_utc import DatetimeUTC
-from prediction_market_agent_tooling.tools.web3_utils import xdai_to_wei
 
 
 class SeerOutcomeEnum(str, Enum):
@@ -165,93 +156,6 @@ class SeerMarket(BaseModel):
     def created_time(self) -> DatetimeUTC:
         return DatetimeUTC.to_datetime_utc(self.block_timestamp)
 
-    @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.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,
-        collateral_token_contract_address_checksummed: 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()
-            != collateral_token_contract_address_checksummed.lower()
-        ):
-            logger.warning(
-                f"Pool {pool.id.hex()} has collateral mismatch with market. Collateral from pool {collateral_address.hex()}, collateral from market {collateral_token_contract_address_checksummed}, 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
-
-    @property
-    def current_p_yes(self) -> Probability:
-        price_data = {}
-        for idx, wrapped_token in enumerate(self.wrapped_tokens):
-            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)
-
-        price_yes = price_data[self.outcome_as_enums[SeerOutcomeEnum.YES]]
-        price_no = price_data[self.outcome_as_enums[SeerOutcomeEnum.NO]]
-        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)
-
     @property
     def url(self) -> str:
         chain_id = RPCConfig().chain_id

prediction_market_agent_tooling/markets/seer/price_manager.py

@@ -1,5 +1,109 @@
-from prediction_market_agent_tooling.gtypes import ChecksumAddress
+from web3 import Web3
+
+from prediction_market_agent_tooling.gtypes import ChecksumAddress, xdai_type
+from prediction_market_agent_tooling.gtypes import Probability
+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, seer_market: SeerMarket, seer_subgraph: SeerSubgraphHandler):
+        self.seer_market = seer_market
+        self.seer_subgraph = seer_subgraph
+
+    def current_p_yes(self) -> Probability:
+        price_data = {}
+        for idx, wrapped_token in enumerate(self.seer_market.wrapped_tokens):
+            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)
+
+        price_yes = price_data[self.seer_market.outcome_as_enums[SeerOutcomeEnum.YES]]
+        price_no = price_data[self.seer_market.outcome_as_enums[SeerOutcomeEnum.NO]]
+        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 = SeerSubgraphHandler().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_contract_address_checksummed.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_contract_address_checksummed}, returning 0."
+            )
+            return 0
+
+        price_in_collateral_units = (
+            pool.token0Price
+            if self._pool_token0_matches_token(token=token, pool=pool)
+            else pool.token1Price
+        )
+        return price_in_collateral_units

prediction_market_agent_tooling/markets/seer/seer.py

@@ -13,6 +13,7 @@ from prediction_market_agent_tooling.gtypes import (
     wei_type,
     xDai,
     xdai_type,
+    Probability,
 )
 from prediction_market_agent_tooling.loggers import logger
 from prediction_market_agent_tooling.markets.agent_market import (
@@ -33,16 +34,19 @@ from prediction_market_agent_tooling.markets.market_fees import MarketFees
 from prediction_market_agent_tooling.markets.omen.omen import OmenAgentMarket
 from prediction_market_agent_tooling.markets.omen.omen_contracts import sDaiContract
 from prediction_market_agent_tooling.markets.seer.data_models import (
-    NewMarketEvent,
     SeerMarket,
     SeerOutcomeEnum,
 )
+from prediction_market_agent_tooling.markets.seer.price_manager import PriceManager
 from prediction_market_agent_tooling.markets.seer.seer_contracts import (
     SeerMarketFactory,
 )
 from prediction_market_agent_tooling.markets.seer.seer_subgraph_handler import (
     SeerSubgraphHandler,
 )
+from prediction_market_agent_tooling.markets.seer.subgraph_data_models import (
+    NewMarketEvent,
+)
 from prediction_market_agent_tooling.tools.balances import get_balances
 from prediction_market_agent_tooling.tools.contract import (
     ContractERC20OnGnosisChain,
@@ -166,6 +170,33 @@ class SeerAgentMarket(AgentMarket):
     def verify_operational_balance(api_keys: APIKeys) -> bool:
         return OmenAgentMarket.verify_operational_balance(api_keys=api_keys)
 
+    @staticmethod
+    def from_data_model_with_subgraph(
+        model: SeerMarket, seer_subgraph: SeerSubgraphHandler
+    ) -> "SeerAgentMarket":
+        p = PriceManager(seer_market=model, seer_subgraph=seer_subgraph)
+        current_p_yes = p.current_p_yes()
+
+        return SeerAgentMarket(
+            id=model.id.hex(),
+            question=model.title,
+            creator=model.creator,
+            created_time=model.created_time,
+            outcomes=model.outcomes,
+            collateral_token_contract_address_checksummed=model.collateral_token_contract_address_checksummed,
+            condition_id=model.condition_id,
+            url=model.url,
+            close_time=model.close_time,
+            wrapped_tokens=[Web3.to_checksum_address(i) for i in model.wrapped_tokens],
+            fees=MarketFees.get_zero_fees(),
+            outcome_token_pool=None,
+            resolution=model.get_resolution_enum(),
+            volume=None,
+            # current_p_yes=model.current_p_yes,
+            current_p_yes=current_p_yes,
+            seer_outcomes=model.outcome_as_enums,
+        )
+
     @staticmethod
     def from_data_model(model: SeerMarket) -> "SeerAgentMarket":
         return SeerAgentMarket(
@@ -183,7 +214,7 @@ class SeerAgentMarket(AgentMarket):
             outcome_token_pool=None,
             resolution=model.get_resolution_enum(),
             volume=None,
-            current_p_yes=model.current_p_yes,
+            current_p_yes=Probability(123),  # ToDo dont use this method
             seer_outcomes=model.outcome_as_enums,
         )
 
@@ -195,13 +226,16 @@ class SeerAgentMarket(AgentMarket):
         created_after: t.Optional[DatetimeUTC] = None,
         excluded_questions: set[str] | None = None,
     ) -> t.Sequence["SeerAgentMarket"]:
+        seer_subgraph = SeerSubgraphHandler()
+        markets = seer_subgraph.get_binary_markets(
+            limit=limit, sort_by=sort_by, filter_by=filter_by
+        )
+
         return [
-            SeerAgentMarket.from_data_model(m)
-            for m in SeerSubgraphHandler().get_binary_markets(
-                limit=limit,
-                sort_by=sort_by,
-                filter_by=filter_by,
+            SeerAgentMarket.from_data_model_with_subgraph(
+                model=m, seer_subgraph=seer_subgraph
             )
+            for m in markets
         ]
 
     def has_liquidity_for_outcome(self, outcome: bool) -> bool:
@@ -352,4 +386,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.seer_market)
+    return Web3.to_checksum_address(new_market_event.market)

prediction_market_agent_tooling/markets/seer/seer_contracts.py

@@ -5,7 +5,7 @@ from web3.types import TxReceipt
 
 from prediction_market_agent_tooling.config import APIKeys
 from prediction_market_agent_tooling.gtypes import ABI, ChecksumAddress, xDai
-from prediction_market_agent_tooling.markets.seer.data_models import (
+from prediction_market_agent_tooling.markets.seer.subgraph_data_models import (
     CreateCategoricalMarketsParams,
 )
 from prediction_market_agent_tooling.tools.contract import (

{prediction_market_agent_tooling-0.61.1.dev484.dist-info → prediction_market_agent_tooling-0.61.1.dev486.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: prediction-market-agent-tooling
-Version: 0.61.1.dev484
+Version: 0.61.1.dev486
 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.dev484.dist-info → prediction_market_agent_tooling-0.61.1.dev486.dist-info}/RECORD

@@ -61,10 +61,10 @@ 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=jLfCiwkw4OZyqw5RV3kEimmWWeBBF6SMmn4Oxzlm5pQ,9101
-prediction_market_agent_tooling/markets/seer/price_manager.py,sha256=k0I8xLFs0cXEDrTiOtQUxVPaAVYchfz-D8cijeONlXc,176
-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/data_models.py,sha256=ECuCDqRcs9vB9rKIec399HxsL915oT7VGSwztcMfGIw,5347
+prediction_market_agent_tooling/markets/seer/price_manager.py,sha256=fUszpmwUW2iJw5My3FJO2Lim2yXUuaZWQ9aNKX39-d0,4392
+prediction_market_agent_tooling/markets/seer/seer.py,sha256=kBF7ALzlh0HsAgbdTa3A9XayyXrddtgi-dMWquLZX8Y,14365
+prediction_market_agent_tooling/markets/seer/seer_contracts.py,sha256=TXGDbv7CI08X8N5re1SAm-X_BaXj8m4ceyzkS_ktQok,2739
 prediction_market_agent_tooling/markets/seer/seer_subgraph_handler.py,sha256=lcaKbFHxigMYwi4TCePyTb5HXJhGoCu5dJPFWTj5E28,9047
 prediction_market_agent_tooling/markets/seer/subgraph_data_models.py,sha256=RIt2oO-PbykzbYN11Qltt8tjjYTty-PfjH3Q4D2VEoA,1644
 prediction_market_agent_tooling/monitor/financial_metrics/financial_metrics.py,sha256=fjIgjDIx5MhH5mwf7S0cspLOOSU3elYLhGYoIiM26mU,2746
@@ -119,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.dev484.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.61.1.dev484.dist-info/METADATA,sha256=IpD8pQ3IicG0EyoDNg33hCkmVHBpXGKbNg-_FpdXve0,8636
-prediction_market_agent_tooling-0.61.1.dev484.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-prediction_market_agent_tooling-0.61.1.dev484.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.61.1.dev484.dist-info/RECORD,,
+prediction_market_agent_tooling-0.61.1.dev486.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.61.1.dev486.dist-info/METADATA,sha256=7XZFv83NtgtmTfYh1aq17OY1M69UzMcLnyhdkuMmNB0,8636
+prediction_market_agent_tooling-0.61.1.dev486.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+prediction_market_agent_tooling-0.61.1.dev486.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.61.1.dev486.dist-info/RECORD,,