prediction-market-agent-tooling 0.63.1__py3-none-any.whl → 0.63.3__py3-none-any.whl

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/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/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/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/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.dist-info → prediction_market_agent_tooling-0.63.3.dist-info}/METADATA

@@ -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.dist-info → prediction_market_agent_tooling-0.63.3.dist-info}/RECORD

@@ -62,8 +62,8 @@ prediction_market_agent_tooling/markets/polymarket/data_models_web.py,sha256=LVE
 prediction_market_agent_tooling/markets/polymarket/polymarket.py,sha256=6rc9qulPl90MxXKB55XiiWKLhjfAyG_eUzAlqpq1UIE,3339
 prediction_market_agent_tooling/markets/polymarket/utils.py,sha256=8kTeVjXPcXC6DkDvWYsZQLY7x8DS6CEp_yznSEazsNU,2037
 prediction_market_agent_tooling/markets/seer/data_models.py,sha256=FwTOq9X2iJ7r3ijtE0evl8pMSbFPm4lUwuc9m7YsMVA,6373
-prediction_market_agent_tooling/markets/seer/price_manager.py,sha256=NZdMXRzOiGq8n4YarOQiIGGpYTNm0XUy9x8HZWmS7Es,4836
-prediction_market_agent_tooling/markets/seer/seer.py,sha256=PQ685AE9BjH4QZC_zcTCGt7RNBRI6wKkXoCUvQeMDO0,14897
+prediction_market_agent_tooling/markets/seer/price_manager.py,sha256=EuOe6zCmEEWXz1loXaQQOYC7ZNQ0cDJn0rZadK1dBlc,5460
+prediction_market_agent_tooling/markets/seer/seer.py,sha256=_FzFGTkaOHQ2WTkUa_ba-Tc9zqw4KTmbCTrH4EmvOsU,16035
 prediction_market_agent_tooling/markets/seer/seer_contracts.py,sha256=yufEojZxLa_SQv9xhsG23wLGsyHOWD8Azm7-7E5Zn_8,2714
 prediction_market_agent_tooling/markets/seer/seer_subgraph_handler.py,sha256=KKRI493VNNAY9tR1AjzNraeH76MvDsBV6GsiLZas0_Y,9859
 prediction_market_agent_tooling/markets/seer/subgraph_data_models.py,sha256=0izxS8Mtzonfdl9UqvFVXrdj0hVzieroekXhogfZKCw,1817
@@ -76,7 +76,7 @@ prediction_market_agent_tooling/monitor/monitor.py,sha256=yvSXwV-3DSAm3vsFZnFC9j
 prediction_market_agent_tooling/monitor/monitor_app.py,sha256=-_6w_ZvQ-Ad5qaeuo7NKTXUOOZ_6OrR8jMe25BGOY4k,4615
 prediction_market_agent_tooling/monitor/monitor_settings.py,sha256=Xiozs3AsufuJ04JOe1vjUri-IAMWHjjmc2ugGGiHNH4,947
 prediction_market_agent_tooling/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-prediction_market_agent_tooling/tools/_generic_value.py,sha256=0_EDybTsr3iXqPPqy2U1WMw4st_SlYaCrf2p1MZhED4,10511
+prediction_market_agent_tooling/tools/_generic_value.py,sha256=pD_PI13lpPp1gFoljHwa_Lzlp-u2pu0m-Z7LcxwDM2U,10618
 prediction_market_agent_tooling/tools/balances.py,sha256=Osab21btfJDw2Y-jT_TV-KHGrseCRxcsYeW6WcOMB8E,1050
 prediction_market_agent_tooling/tools/betting_strategies/kelly_criterion.py,sha256=L9yp3m_81i-y85dGZH1rYRv9YXYYt1w3RGroFAAYSbw,4873
 prediction_market_agent_tooling/tools/betting_strategies/market_moving.py,sha256=36ucOuF38bdB4rnSPNjC1qpidBo30Ff4P5Kb-7emKEQ,5469
@@ -88,7 +88,7 @@ prediction_market_agent_tooling/tools/caches/inmemory_cache.py,sha256=ZW5iI5rmjq
 prediction_market_agent_tooling/tools/caches/serializers.py,sha256=vFDx4fsPxclXp2q0sv27j4al_M_Tj9aR2JJP-xNHQXA,2151
 prediction_market_agent_tooling/tools/contract.py,sha256=1ZFp_VoqTjM8cqOfAhco2Ht0DTqakjhZpuZUrAXr28Q,21332
 prediction_market_agent_tooling/tools/costs.py,sha256=EaAJ7v9laD4VEV3d8B44M4u3_oEO_H16jRVCdoZ93Uw,954
-prediction_market_agent_tooling/tools/cow/cow_order.py,sha256=jWBRF5UdXjlWNn3X8BxcMXSuU3fegoZzz-NXB7sSIt8,6959
+prediction_market_agent_tooling/tools/cow/cow_order.py,sha256=Kk-J3N3wTbmgdYs_XGp3nXbSi3-A17lQ61j60D_lPGg,6961
 prediction_market_agent_tooling/tools/custom_exceptions.py,sha256=Fh8z1fbwONvP4-j7AmV_PuEcoqb6-QXa9PJ9m7guMcM,93
 prediction_market_agent_tooling/tools/datetime_utc.py,sha256=8_WackjtjC8zHXrhQFTGQ6e6Fz_6llWoKR4CSFvIv9I,2766
 prediction_market_agent_tooling/tools/db/db_manager.py,sha256=GtzHH1NLl8HwqC8Z7s6eTlIQXuV0blxfaV2PeQrBnfQ,3013
@@ -117,12 +117,12 @@ prediction_market_agent_tooling/tools/tokens/auto_deposit.py,sha256=MWIRHYCN-7o2
 prediction_market_agent_tooling/tools/tokens/auto_withdraw.py,sha256=22g0SIVmLlgITpdt3kPhJOw0sU4OPeBuYk_7xCrQr9U,2491
 prediction_market_agent_tooling/tools/tokens/main_token.py,sha256=1rbwpdCusPgQIVFuo3m00nBZ_b2lCAoFVm67i-YDcEw,812
 prediction_market_agent_tooling/tools/tokens/token_utils.py,sha256=fhs-FH9m9IbzGa-30R3ZleSKLeKfLEDoJ7F5Om285Vk,1369
-prediction_market_agent_tooling/tools/tokens/usd.py,sha256=Qq8ofVCCMX-eo8mDlHv4gL9DV_bqHFpd63Wv5yE7cjY,3113
+prediction_market_agent_tooling/tools/tokens/usd.py,sha256=yuW8iPPtcpP4eLH2nORMDAfztcq0Nv2ascSrCquF1f8,3115
 prediction_market_agent_tooling/tools/transaction_cache.py,sha256=K5YKNL2_tR10Iw2TD9fuP-CTGpBbZtNdgbd0B_R7pjg,1814
 prediction_market_agent_tooling/tools/utils.py,sha256=1xsyBBJfiEdSoMlceB2F8o2sCb6Z8-qNz11pEJFrdyE,6566
 prediction_market_agent_tooling/tools/web3_utils.py,sha256=eYCc1iWAVtqDKUPTwnMUHuYolPdwh_OTiM3-AdRgDp4,12198
-prediction_market_agent_tooling-0.63.1.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.63.1.dist-info/METADATA,sha256=cpCiRflWwNgAuRkyel1R4e_w_FxZMHpp_o9hjvDUHws,8689
-prediction_market_agent_tooling-0.63.1.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-prediction_market_agent_tooling-0.63.1.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.63.1.dist-info/RECORD,,
+prediction_market_agent_tooling-0.63.3.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.63.3.dist-info/METADATA,sha256=0RZTVpeWNV3EurtBs_MQ55JxJVW1EmHzuxKQsVpAHHU,8689
+prediction_market_agent_tooling-0.63.3.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+prediction_market_agent_tooling-0.63.3.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.63.3.dist-info/RECORD,,