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

prediction_market_agent_tooling/markets/omen/omen_contracts.py

@@ -15,12 +15,12 @@ from prediction_market_agent_tooling.gtypes import (
     HexBytes,
     HexStr,
     IPFSCIDVersion0,
-    OutcomeWei,
+    OmenOutcomeToken,
     TxParams,
     TxReceipt,
     Wei,
     int_to_hexbytes,
-    xDaiWei,
+    wei_type,
 )
 from prediction_market_agent_tooling.markets.omen.data_models import (
     INVALID_ANSWER_HEX_BYTES,
@@ -128,8 +128,8 @@ class OmenConditionalTokenContract(ContractOnGnosisChain):
 
     def balanceOf(
         self, from_address: ChecksumAddress, position_id: int, web3: Web3 | None = None
-    ) -> OutcomeWei:
-        balance = OutcomeWei(
+    ) -> Wei:
+        balance = wei_type(
             self.call("balanceOf", [from_address, position_id], web3=web3)
         )
         return balance
@@ -169,7 +169,7 @@ class OmenConditionalTokenContract(ContractOnGnosisChain):
         collateral_token_address: ChecksumAddress,
         conditionId: HexBytes,
         index_sets: t.List[int],
-        amount: OutcomeWei,
+        amount: Wei,
         parent_collection_id: HexStr = build_parent_collection_id(),
         web3: Web3 | None = None,
     ) -> TxReceipt:
@@ -296,28 +296,28 @@ class OmenFixedProductMarketMakerContract(ContractOnGnosisChain):
     # Factory contract at https://gnosisscan.io/address/0x9083a2b699c0a4ad06f63580bde2635d26a3eef0.
 
     def balanceOf(self, for_address: ChecksumAddress, web3: Web3 | None = None) -> Wei:
-        balance = Wei(self.call("balanceOf", [for_address], web3=web3))
+        balance: Wei = self.call("balanceOf", [for_address], web3=web3)
         return balance
 
     def calcBuyAmount(
         self, investment_amount: Wei, outcome_index: int, web3: Web3 | None = None
-    ) -> OutcomeWei:
+    ) -> OmenOutcomeToken:
         """
         Returns amount of shares we will get for the given outcome_index for the given investment amount.
         """
-        calculated_shares = OutcomeWei(
-            self.call("calcBuyAmount", [investment_amount, outcome_index], web3=web3)
+        calculated_shares: OmenOutcomeToken = self.call(
+            "calcBuyAmount", [investment_amount, outcome_index], web3=web3
         )
         return calculated_shares
 
     def calcSellAmount(
         self, return_amount: Wei, outcome_index: int, web3: Web3 | None = None
-    ) -> OutcomeWei:
+    ) -> OmenOutcomeToken:
         """
         Returns amount of shares we will sell for the requested wei.
         """
-        calculated_shares = OutcomeWei(
-            self.call("calcSellAmount", [return_amount, outcome_index], web3=web3)
+        calculated_shares: OmenOutcomeToken = self.call(
+            "calcSellAmount", [return_amount, outcome_index], web3=web3
         )
         return calculated_shares
 
@@ -334,7 +334,7 @@ class OmenFixedProductMarketMakerContract(ContractOnGnosisChain):
         api_keys: APIKeys,
         amount_wei: Wei,
         outcome_index: int,
-        min_outcome_tokens_to_buy: OutcomeWei,
+        min_outcome_tokens_to_buy: OmenOutcomeToken,
         tx_params: t.Optional[TxParams] = None,
         web3: Web3 | None = None,
     ) -> TxReceipt:
@@ -355,7 +355,7 @@ class OmenFixedProductMarketMakerContract(ContractOnGnosisChain):
         api_keys: APIKeys,
         amount_wei: Wei,
         outcome_index: int,
-        max_outcome_tokens_to_sell: OutcomeWei,
+        max_outcome_tokens_to_sell: OmenOutcomeToken,
         tx_params: t.Optional[TxParams] = None,
         web3: Web3 | None = None,
     ) -> TxReceipt:
@@ -411,7 +411,7 @@ class OmenFixedProductMarketMakerContract(ContractOnGnosisChain):
 
     def totalSupply(self, web3: Web3 | None = None) -> Wei:
         # This is the liquidity you seen on the Omen website (but in Wei).
-        total_supply = Wei(self.call("totalSupply", web3=web3))
+        total_supply: Wei = self.call("totalSupply", web3=web3)
         return total_supply
 
     def get_collateral_token_contract(
@@ -466,7 +466,7 @@ class OmenFixedProductMarketMakerFactoryContract(ContractOnGnosisChain):
         initial_funds_wei: Wei,
         collateral_token_address: ChecksumAddress,
         fee: Wei,  # This is actually fee in %, 'where 100% == 1 xDai'.
-        distribution_hint: list[OutcomeWei] | None = None,
+        distribution_hint: list[OmenOutcomeToken] | None = None,
         tx_params: t.Optional[TxParams] = None,
         web3: Web3 | None = None,
     ) -> tuple[
@@ -585,7 +585,7 @@ class OmenRealitioContract(ContractOnGnosisChain):
         api_keys: APIKeys,
         question: str,
         category: str,
-        outcomes: t.Sequence[str],
+        outcomes: list[str],
         language: str,
         arbitrator: Arbitrator,
         opening: DatetimeUTC,
@@ -640,14 +640,14 @@ class OmenRealitioContract(ContractOnGnosisChain):
         api_keys: APIKeys,
         question_id: HexBytes,
         answer: HexBytes,
-        bond: xDaiWei,
-        max_previous: xDaiWei | None = None,
+        bond: Wei,
+        max_previous: Wei | None = None,
         web3: Web3 | None = None,
     ) -> TxReceipt:
         if max_previous is None:
             # If not provided, defaults to 0, which means no checking,
             # same as on Omen website: https://github.com/protofire/omen-exchange/blob/763d9c9d05ebf9edacbc1dbaa561aa5d08813c0f/app/src/services/realitio.ts#L363.
-            max_previous = xDaiWei(0)
+            max_previous = Wei(0)
 
         return self.send_with_value(
             api_keys=api_keys,
@@ -657,7 +657,7 @@ class OmenRealitioContract(ContractOnGnosisChain):
                 answer=answer,
                 max_previous=max_previous,
             ),
-            amount_wei=bond.as_wei,
+            amount_wei=bond,
             web3=web3,
         )
 
@@ -666,9 +666,9 @@ class OmenRealitioContract(ContractOnGnosisChain):
         api_keys: APIKeys,
         question_id: HexBytes,
         answer: str,
-        outcomes: t.Sequence[str],
-        bond: xDaiWei,
-        max_previous: xDaiWei | None = None,
+        outcomes: list[str],
+        bond: Wei,
+        max_previous: Wei | None = None,
         web3: Web3 | None = None,
     ) -> TxReceipt:
         # Normalise the answer to lowercase, to match Enum values as [YES, NO] against outcomes as ["Yes", "No"].
@@ -690,8 +690,8 @@ class OmenRealitioContract(ContractOnGnosisChain):
         self,
         api_keys: APIKeys,
         question_id: HexBytes,
-        bond: xDaiWei,
-        max_previous: xDaiWei | None = None,
+        bond: Wei,
+        max_previous: Wei | None = None,
         web3: Web3 | None = None,
     ) -> TxReceipt:
         return self.submitAnswer(
@@ -709,7 +709,7 @@ class OmenRealitioContract(ContractOnGnosisChain):
         question_id: HexBytes,
         history_hashes: list[HexBytes],
         addresses: list[ChecksumAddress],
-        bonds: list[xDaiWei],
+        bonds: list[Wei],
         answers: list[HexBytes],
         tx_params: t.Optional[TxParams] = None,
         web3: Web3 | None = None,
@@ -732,8 +732,8 @@ class OmenRealitioContract(ContractOnGnosisChain):
         self,
         from_address: ChecksumAddress,
         web3: Web3 | None = None,
-    ) -> xDaiWei:
-        balance = xDaiWei(self.call("balanceOf", [from_address], web3=web3))
+    ) -> Wei:
+        balance = wei_type(self.call("balanceOf", [from_address], web3=web3))
         return balance
 
     def withdraw(
@@ -896,11 +896,9 @@ class OmenThumbnailMapping(ContractOnGnosisChain):
 class CollateralTokenChoice(str, Enum):
     wxdai = "wxdai"
     sdai = "sdai"
-    gno = "gno"
 
 
 COLLATERAL_TOKEN_CHOICE_TO_ADDRESS = {
     CollateralTokenChoice.wxdai: WrappedxDaiContract().address,
     CollateralTokenChoice.sdai: sDaiContract().address,
-    CollateralTokenChoice.gno: GNOContract().address,
 }

prediction_market_agent_tooling/markets/omen/omen_resolving.py

@@ -5,8 +5,8 @@ from prediction_market_agent_tooling.gtypes import (
     ChecksumAddress,
     HexAddress,
     HexBytes,
+    Wei,
     xDai,
-    xDaiWei,
 )
 from prediction_market_agent_tooling.loggers import logger
 from prediction_market_agent_tooling.markets.data_models import Resolution
@@ -32,7 +32,12 @@ from prediction_market_agent_tooling.tools.tokens.main_token import (
     MINIMUM_NATIVE_TOKEN_IN_EOA_FOR_FEES,
 )
 from prediction_market_agent_tooling.tools.utils import utcnow
-from prediction_market_agent_tooling.tools.web3_utils import ZERO_BYTES
+from prediction_market_agent_tooling.tools.web3_utils import (
+    ZERO_BYTES,
+    wei_to_xdai,
+    xdai_to_wei,
+    xdai_type,
+)
 
 
 def claim_bonds_on_realitio_questions(
@@ -88,7 +93,7 @@ def claim_bonds_on_realitio_question(
 
     history_hashes: list[HexBytes] = []
     addresses: list[ChecksumAddress] = []
-    bonds: list[xDaiWei] = []
+    bonds: list[Wei] = []
     answers: list[HexBytes] = []
 
     # Caller must provide the answer history, in reverse order.
@@ -123,7 +128,7 @@ def claim_bonds_on_realitio_question(
     # 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.as_xdai} xDai from Realitio."
+            f"Withdrawing remaining balance {wei_to_xdai(current_balance)} xDai from Realitio."
         )
         realitio_contract.withdraw(api_keys, web3=web3)
 
@@ -145,7 +150,9 @@ def finalize_markets(
         # If we don't have enough of xDai for bond, try to get it from the keeping token.
         send_keeping_token_to_eoa_xdai(
             api_keys=api_keys,
-            min_required_balance=realitio_bond + MINIMUM_NATIVE_TOKEN_IN_EOA_FOR_FEES,
+            min_required_balance=xdai_type(
+                realitio_bond + MINIMUM_NATIVE_TOKEN_IN_EOA_FOR_FEES
+            ),
             web3=web3,
         )
 
@@ -228,7 +235,7 @@ def omen_submit_answer_market_tx(
         question_id=market.question.id,
         answer=resolution.value,
         outcomes=market.question.outcomes,
-        bond=bond.as_xdai_wei,
+        bond=xdai_to_wei(bond),
         web3=web3,
     )
 
@@ -247,7 +254,7 @@ def omen_submit_invalid_answer_market_tx(
     realitio_contract.submit_answer_invalid(
         api_keys=api_keys,
         question_id=market.question.id,
-        bond=bond.as_xdai_wei,
+        bond=xdai_to_wei(bond),
         web3=web3,
     )
 

prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py

@@ -11,6 +11,7 @@ from prediction_market_agent_tooling.gtypes import (
     HexAddress,
     HexBytes,
     Wei,
+    wei_type,
 )
 from prediction_market_agent_tooling.markets.agent_market import FilterBy, SortBy
 from prediction_market_agent_tooling.markets.base_subgraph_handler import (
@@ -23,8 +24,6 @@ from prediction_market_agent_tooling.markets.omen.data_models import (
     OmenMarket,
     OmenPosition,
     OmenUserPosition,
-    OutcomeStr,
-    OutcomeWei,
     RealityAnswer,
     RealityQuestion,
     RealityResponse,
@@ -45,7 +44,6 @@ from prediction_market_agent_tooling.tools.utils import (
 from prediction_market_agent_tooling.tools.web3_utils import (
     ZERO_BYTES,
     byte32_to_ipfscidv0,
-    unwrap_generic_value,
 )
 
 SAFE_COLLATERAL_TOKENS = (
@@ -211,7 +209,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
         self,
         creator: HexAddress | None,
         creator_in: t.Sequence[HexAddress] | None,
-        outcomes: t.Sequence[OutcomeStr],
+        outcomes: list[str],
         created_after: DatetimeUTC | None,
         question_opened_before: DatetimeUTC | None,
         question_opened_after: DatetimeUTC | None,
@@ -349,7 +347,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
             # because even closed markets don't need to be resolved yet (e.g. if someone forgot to finalize the question on reality).
             opened_after = utcnow()
             # Even if the market isn't closed yet, liquidity can be withdrawn to 0, which essentially closes the market.
-            liquidity_bigger_than = Wei(0)
+            liquidity_bigger_than = wei_type(0)
         elif filter_by == FilterBy.NONE:
             pass
         else:
@@ -393,7 +391,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
         id_in: list[str] | None = None,
         sort_by_field: FieldPath | None = None,
         sort_direction: str | None = None,
-        outcomes: t.Sequence[OutcomeStr] = OMEN_BINARY_MARKET_OUTCOMES,
+        outcomes: list[str] = OMEN_BINARY_MARKET_OUTCOMES,
         collateral_token_address_in: (
             tuple[ChecksumAddress, ...] | None
         ) = SAFE_COLLATERAL_TOKENS_ADDRESSES,
@@ -436,7 +434,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
             first=(
                 limit if limit else sys.maxsize
             ),  # if not limit, we fetch all possible markets
-            where=unwrap_generic_value(where_stms),
+            where=where_stms,
             **optional_params,
         )
 
@@ -491,7 +489,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
             where_stms["conditionIds_contains"] = [condition_id.hex()]
 
         positions = self.conditional_tokens_subgraph.Query.positions(
-            first=sys.maxsize, where=unwrap_generic_value(where_stms)
+            first=sys.maxsize, where=where_stms
         )
         fields = self._get_fields_for_positions(positions)
         result = self.sg.query_json(fields)
@@ -502,7 +500,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
         self,
         better_address: ChecksumAddress,
         position_id_in: list[HexBytes] | None = None,
-        total_balance_bigger_than: OutcomeWei | None = None,
+        total_balance_bigger_than: Wei | None = None,
     ) -> list[OmenUserPosition]:
         where_stms: dict[str, t.Any] = {
             "user": better_address.lower(),
@@ -516,7 +514,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
             where_stms["position_"]["positionId_in"] = [x.hex() for x in position_id_in]
 
         positions = self.conditional_tokens_subgraph.Query.userPositions(
-            first=sys.maxsize, where=unwrap_generic_value(where_stms)
+            first=sys.maxsize, where=where_stms
         )
         fields = self._get_fields_for_user_positions(positions)
         result = self.sg.query_json(fields)
@@ -570,9 +568,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
                 < to_int_timestamp(market_resolved_before)
             )
         if collateral_amount_more_than is not None:
-            where_stms.append(
-                trade.collateralAmount > collateral_amount_more_than.value
-            )
+            where_stms.append(trade.collateralAmount > collateral_amount_more_than)
 
         # These values can not be set to `None`, but they can be omitted.
         optional_params = {}
@@ -583,7 +579,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
 
         trades = self.trades_subgraph.Query.fpmmTrades(
             first=limit if limit else sys.maxsize,
-            where=unwrap_generic_value(where_stms),
+            where=where_stms,
             **optional_params,
         )
         fields = self._get_fields_for_bets(trades)
@@ -822,7 +818,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
             first=(
                 limit if limit else sys.maxsize
             ),  # if not limit, we fetch all possible
-            where=unwrap_generic_value(where_stms),
+            where=where_stms,
         )
         fields = self._get_fields_for_reality_questions(questions)
         result = self.sg.query_json(fields)
@@ -836,9 +832,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
             answer.question.questionId == question_id.hex(),
         ]
 
-        answers = self.realityeth_subgraph.Query.answers(
-            where=unwrap_generic_value(where_stms)
-        )
+        answers = self.realityeth_subgraph.Query.answers(where=where_stms)
         fields = self._get_fields_for_answers(answers)
         result = self.sg.query_json(fields)
         items = self._parse_items_from_json(result)
@@ -885,7 +879,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
             first=(
                 limit if limit else sys.maxsize
             ),  # if not limit, we fetch all possible
-            where=unwrap_generic_value(where_stms),
+            where=where_stms,
         )
         fields = self._get_fields_for_responses(responses)
         result = self.sg.query_json(fields)
@@ -944,7 +938,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
 
         prediction_added = (
             self.omen_agent_result_mapping_subgraph.Query.predictionAddeds(
-                where=unwrap_generic_value(where_stms),
+                where=where_stms,
                 orderBy="blockNumber",
                 orderDirection="asc",
             )

prediction_market_agent_tooling/markets/polymarket/data_models.py

@@ -1,6 +1,6 @@
 from pydantic import BaseModel
 
-from prediction_market_agent_tooling.gtypes import USDC, OutcomeStr, Probability
+from prediction_market_agent_tooling.gtypes import USDC, Probability, usdc_type
 from prediction_market_agent_tooling.markets.data_models import Resolution
 from prediction_market_agent_tooling.markets.polymarket.data_models_web import (
     POLYMARKET_FALSE_OUTCOME,
@@ -22,7 +22,7 @@ class PolymarketRewards(BaseModel):
 
 class PolymarketToken(BaseModel):
     token_id: str
-    outcome: OutcomeStr
+    outcome: str
     winner: bool
 
 
@@ -111,7 +111,7 @@ class PolymarketPriceResponse(BaseModel):
 
     @property
     def price_dec(self) -> USDC:
-        return USDC(self.price)
+        return usdc_type(self.price)
 
 
 class Prices(BaseModel):

prediction_market_agent_tooling/markets/polymarket/data_models_web.py

@@ -1,5 +1,5 @@
 """
-Autogenerated using `datamodel-codegen` from a single Polymarket response. Then adjusted, fixed.
+Autogenerated using `datamodel-codegen` from a single Polymarket response. Then adjusted, fixed. 
 Keep in mind that not all fields were used so far, so there might be some bugs.
 
 These models are based on what Polymarket's website returns in the response, and `prediction_market_agent_tooling/markets/polymarket/data_models.py` are based on what their API returns.
@@ -11,7 +11,7 @@ import typing as t
 import requests
 from pydantic import BaseModel, field_validator
 
-from prediction_market_agent_tooling.gtypes import USDC, HexAddress, OutcomeStr
+from prediction_market_agent_tooling.gtypes import USDC, HexAddress
 from prediction_market_agent_tooling.loggers import logger
 from prediction_market_agent_tooling.markets.data_models import Resolution
 from prediction_market_agent_tooling.tools.utils import DatetimeUTC
@@ -47,7 +47,7 @@ class Market1(BaseModel):
     question: str
     image: str
     volume: USDC | None = None
-    outcomes: t.Sequence[OutcomeStr]
+    outcomes: list[str]
     outcomePrices: list[USDC]
     active: bool
     archived: bool
@@ -92,7 +92,7 @@ class Market(BaseModel):
     fee: str | None = None
     lowerBound: t.Any | None = None
     upperBound: t.Any | None = None
-    outcomes: t.Sequence[OutcomeStr]
+    outcomes: list[str]
     image: str | None = None
     icon: str | None = None
     imageOptimized: t.Any | None = None

prediction_market_agent_tooling/markets/polymarket/polymarket.py

@@ -1,12 +1,12 @@
 import typing as t
 
-from prediction_market_agent_tooling.gtypes import USD, Token
 from prediction_market_agent_tooling.markets.agent_market import (
     AgentMarket,
     FilterBy,
     MarketFees,
     SortBy,
 )
+from prediction_market_agent_tooling.markets.data_models import BetAmount, Currency
 from prediction_market_agent_tooling.markets.polymarket.api import (
     get_polymarket_binary_markets,
 )
@@ -24,6 +24,7 @@ class PolymarketAgentMarket(AgentMarket):
     Polymarket's market class that can be used by agents to make predictions.
     """
 
+    currency: t.ClassVar[Currency] = Currency.USDC
     base_url: t.ClassVar[str] = POLYMARKET_BASE_URL
 
     # Based on https://docs.polymarket.com/#fees, there are currently no fees, except for transactions fees.
@@ -48,10 +49,11 @@ class PolymarketAgentMarket(AgentMarket):
             outcome_token_pool=None,
         )
 
-    def get_tiny_bet_amount(self) -> Token:
+    @classmethod
+    def get_tiny_bet_amount(cls) -> BetAmount:
         raise NotImplementedError("TODO: Implement to allow betting on Polymarket.")
 
-    def place_bet(self, outcome: bool, amount: USD) -> str:
+    def place_bet(self, outcome: bool, amount: BetAmount) -> str:
         raise NotImplementedError("TODO: Implement to allow betting on Polymarket.")
 
     @staticmethod

prediction_market_agent_tooling/markets/seer/data_models.py

@@ -12,22 +12,22 @@ from prediction_market_agent_tooling.gtypes import (
     ChecksumAddress,
     HexAddress,
     HexBytes,
-    OutcomeStr,
     Probability,
-    Token,
-    Web3Wei,
+    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.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: t.Sequence[OutcomeStr]
+    outcomes: list[str]
     # Only relevant for scalar markets
     question_start: str = Field(alias="questionStart", default="")
     question_end: str = Field(alias="questionEnd", default="")
@@ -41,7 +41,7 @@ class CreateCategoricalMarketsParams(BaseModel):
     lang: str
     lower_bound: int = Field(alias="lowerBound", default=0)
     upper_bound: int = Field(alias="upperBound", default=0)
-    min_bond: Web3Wei = Field(..., alias="minBond")
+    min_bond: Wei = Field(..., alias="minBond")
     opening_time: int = Field(..., alias="openingTime")
     token_names: list[str] = Field(..., alias="tokenNames")
 
@@ -97,7 +97,7 @@ class SeerMarket(BaseModel):
     id: HexBytes
     creator: HexAddress
     title: str = Field(alias="marketName")
-    outcomes: t.Sequence[OutcomeStr]
+    outcomes: list[str]
     wrapped_tokens: list[HexAddress] = Field(alias="wrappedTokens")
     parent_outcome: int = Field(alias="parentOutcome")
     parent_market: t.Optional[SeerParentMarket] = Field(
@@ -207,7 +207,7 @@ class SeerMarket(BaseModel):
         return Probability(price_yes)
 
     def _get_price_for_token(self, token: ChecksumAddress) -> float:
-        collateral_exchange_amount = Token(1).as_wei
+        collateral_exchange_amount = xdai_to_wei(xdai_type(1))
         try:
             quote = CowManager().get_quote(
                 collateral_token=self.collateral_token_contract_address_checksummed,
@@ -218,7 +218,7 @@ class SeerMarket(BaseModel):
             logger.warning(f"Could not get quote for {token=}, returning price 0. {e=}")
             return 0
 
-        return collateral_exchange_amount.value / float(quote.quote.buyAmount.root)
+        return collateral_exchange_amount / float(quote.quote.buyAmount.root)
 
     @property
     def url(self) -> str:
@@ -238,6 +238,9 @@ class SeerPool(BaseModel):
     liquidity: int
     token0: SeerToken
     token1: SeerToken
+    token0Price: float
+    token1Price: float
+    sqrtPrice: int
 
 
 class NewMarketEvent(BaseModel):