prediction-market-agent-tooling 0.63.1__tar.gz → 0.63.3__tar.gz

{prediction_market_agent_tooling-0.63.1 → prediction_market_agent_tooling-0.63.3}/PKG-INFO

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

{prediction_market_agent_tooling-0.63.1 → prediction_market_agent_tooling-0.63.3}/prediction_market_agent_tooling/markets/seer/price_manager.py

@@ -1,3 +1,6 @@
+import typing as t
+
+from cachetools import TTLCache, cached
 from web3 import Web3
 
 from prediction_market_agent_tooling.gtypes import (
@@ -20,6 +23,27 @@ from prediction_market_agent_tooling.tools.cow.cow_order import (
 from prediction_market_agent_tooling.tools.hexbytes_custom import HexBytes
 
 
+def _make_cache_key(
+    *args: t.Any,
+    token: ChecksumAddress,
+    collateral_exchange_amount: CollateralToken | None = None,
+) -> str:
+    """
+    Generate a unique cache key based on a token address and optional collateral token.
+    """
+
+    if collateral_exchange_amount is None:
+        return f"{token}-no_collateral"
+
+    return "-".join(
+        [
+            token,
+            collateral_exchange_amount.symbol,
+            str(collateral_exchange_amount.value),
+        ]
+    )
+
+
 class PriceManager:
     def __init__(self, seer_market: SeerMarket, seer_subgraph: SeerSubgraphHandler):
         self.seer_market = seer_market
@@ -65,6 +89,7 @@ class PriceManager:
         else:
             return None
 
+    @cached(TTLCache(maxsize=100, ttl=5 * 60), key=_make_cache_key)
     def get_price_for_token(
         self,
         token: ChecksumAddress,
@@ -78,7 +103,7 @@ class PriceManager:
 
         try:
             buy_token_amount = get_buy_token_amount_else_raise(
-                amount_wei=collateral_exchange_amount.as_wei,
+                sell_amount=collateral_exchange_amount.as_wei,
                 sell_token=self.seer_market.collateral_token_contract_address_checksummed,
                 buy_token=token,
             )

{prediction_market_agent_tooling-0.63.1 → prediction_market_agent_tooling-0.63.3}/prediction_market_agent_tooling/markets/seer/seer.py

@@ -46,7 +46,10 @@ from prediction_market_agent_tooling.tools.contract import (
     init_collateral_token_contract,
     to_gnosis_chain_contract,
 )
-from prediction_market_agent_tooling.tools.cow.cow_order import swap_tokens_waiting
+from prediction_market_agent_tooling.tools.cow.cow_order import (
+    get_buy_token_amount_else_raise,
+    swap_tokens_waiting,
+)
 from prediction_market_agent_tooling.tools.datetime_utc import DatetimeUTC
 from prediction_market_agent_tooling.tools.tokens.auto_deposit import (
     auto_deposit_collateral_token,
@@ -122,6 +125,23 @@ class SeerAgentMarket(AgentMarket):
         amount_outcome_tokens = bet_amount_in_tokens / price
         return OutcomeToken(amount_outcome_tokens)
 
+    def get_sell_value_of_outcome_token(
+        self, outcome: str, amount: OutcomeToken
+    ) -> CollateralToken:
+        if amount == amount.zero():
+            return CollateralToken.zero()
+
+        outcome_index = self.get_outcome_index(outcome=outcome)
+        wrapped_outcome_token = self.wrapped_tokens[outcome_index]
+
+        # We calculate how much collateral we would get back if we sold `amount` of outcome token.
+        value_outcome_token_in_collateral = get_buy_token_amount_else_raise(
+            sell_amount=amount.as_outcome_wei.as_wei,
+            sell_token=wrapped_outcome_token,
+            buy_token=self.collateral_token_contract_address_checksummed,
+        )
+        return value_outcome_token_in_collateral.as_token
+
     def get_outcome_str_from_bool(self, outcome: bool) -> OutcomeStr:
         outcome_translated = SeerOutcomeEnum.from_bool(outcome)
         idx = self.seer_outcomes[outcome_translated]
@@ -134,9 +154,9 @@ class SeerAgentMarket(AgentMarket):
     def get_tiny_bet_amount(self) -> CollateralToken:
         return self.get_in_token(SEER_TINY_BET_AMOUNT)
 
-    def get_position(
+    def get_position_else_raise(
         self, user_id: str, web3: Web3 | None = None
-    ) -> ExistingPosition | None:
+    ) -> ExistingPosition:
         """
         Fetches position from the user in a given market.
         We ignore the INVALID balances since we are only interested in binary outcomes.
@@ -169,6 +189,15 @@ class SeerAgentMarket(AgentMarket):
             amounts_ot=amounts_ot,
         )
 
+    def get_position(
+        self, user_id: str, web3: Web3 | None = None
+    ) -> ExistingPosition | None:
+        try:
+            return self.get_position_else_raise(user_id=user_id, web3=web3)
+        except Exception as e:
+            logger.warning(f"Could not get position for user {user_id}, exception {e}")
+            return None
+
     @staticmethod
     def get_user_id(api_keys: APIKeys) -> str:
         return OmenAgentMarket.get_user_id(api_keys)

{prediction_market_agent_tooling-0.63.1 → prediction_market_agent_tooling-0.63.3}/prediction_market_agent_tooling/tools/_generic_value.py

@@ -231,6 +231,9 @@ class _GenericValue(
     def __bool__(self) -> bool:
         return bool(self.value)
 
+    def __hash__(self) -> int:  # type: ignore[override]
+        return hash(tuple(sorted(self.items())))
+
     @classmethod
     def __get_pydantic_core_schema__(
         cls, source_type: t.Any, handler: GetCoreSchemaHandler

{prediction_market_agent_tooling-0.63.1 → prediction_market_agent_tooling-0.63.3}/prediction_market_agent_tooling/tools/cow/cow_order.py

@@ -127,14 +127,14 @@ def get_quote(
 
 
 def get_buy_token_amount_else_raise(
-    amount_wei: Wei,
+    sell_amount: Wei,
     sell_token: ChecksumAddress,
     buy_token: ChecksumAddress,
     chain: Chain = Chain.GNOSIS,
     env: Envs = "prod",
 ) -> Wei:
     order_quote = get_quote(
-        amount_wei=amount_wei,
+        amount_wei=sell_amount,
         sell_token=sell_token,
         buy_token=buy_token,
         chain=chain,

{prediction_market_agent_tooling-0.63.1 → prediction_market_agent_tooling-0.63.3}/prediction_market_agent_tooling/tools/tokens/usd.py

@@ -50,7 +50,7 @@ def get_single_token_to_usd_rate(token_address: ChecksumAddress) -> USD:
             .as_token.value
         )
     in_wei = get_buy_token_amount_else_raise(
-        amount_wei=CollateralToken(1).as_wei,
+        sell_amount=CollateralToken(1).as_wei,
         sell_token=token_address,
         buy_token=WRAPPED_XDAI_CONTRACT_ADDRESS,
     )
@@ -72,7 +72,7 @@ def get_single_usd_to_token_rate(token_address: ChecksumAddress) -> CollateralTo
             .as_token.value
         )
     in_wei = get_buy_token_amount_else_raise(
-        amount_wei=CollateralToken(1).as_wei,
+        sell_amount=CollateralToken(1).as_wei,
         sell_token=WRAPPED_XDAI_CONTRACT_ADDRESS,
         buy_token=token_address,
     )

{prediction_market_agent_tooling-0.63.1 → prediction_market_agent_tooling-0.63.3}/pyproject.toml

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