prediction-market-agent-tooling 0.39.4__py3-none-any.whl → 0.40.0__py3-none-any.whl

prediction_market_agent_tooling/config.py

@@ -58,14 +58,14 @@ class APIKeys(BaseSettings):
             "BET_FROM_PRIVATE_KEY missing in the environment.",
         )
 
+    @property
+    def public_key(self) -> ChecksumAddress:
+        return private_key_to_public_key(self.bet_from_private_key)
+
     @property
     def bet_from_address(self) -> ChecksumAddress:
         """If the SAFE is available, we always route transactions via SAFE. Otherwise we use the EOA."""
-        return (
-            self.SAFE_ADDRESS
-            if self.SAFE_ADDRESS
-            else private_key_to_public_key(self.bet_from_private_key)
-        )
+        return self.SAFE_ADDRESS if self.SAFE_ADDRESS else self.public_key
 
     @property
     def openai_api_key(self) -> SecretStr:

prediction_market_agent_tooling/deploy/agent.py

@@ -22,7 +22,7 @@ from prediction_market_agent_tooling.deploy.gcp.utils import (
     gcp_function_is_active,
     gcp_resolve_api_keys_secrets,
 )
-from prediction_market_agent_tooling.gtypes import Probability
+from prediction_market_agent_tooling.gtypes import Probability, xdai_type
 from prediction_market_agent_tooling.loggers import logger
 from prediction_market_agent_tooling.markets.agent_market import (
     AgentMarket,
@@ -36,6 +36,7 @@ from prediction_market_agent_tooling.markets.markets import (
 )
 from prediction_market_agent_tooling.markets.omen.omen import (
     redeem_from_all_user_positions,
+    withdraw_wxdai_to_xdai_to_keep_balance,
 )
 from prediction_market_agent_tooling.monitor.langfuse.langfuse_wrapper import (
     LangfuseWrapper,
@@ -260,9 +261,14 @@ class DeployableTraderAgent(DeployableAgent):
         """
         Executes actions that occur before bets are placed.
         """
+        api_keys = APIKeys()
         if market_type == MarketType.OMEN:
             # Omen is specific, because the user (agent) needs to manually withdraw winnings from the market.
-            redeem_from_all_user_positions(APIKeys())
+            redeem_from_all_user_positions(api_keys)
+            # Exchange wxdai back to xdai if the balance is getting low, so we can keep paying for fees.
+            withdraw_wxdai_to_xdai_to_keep_balance(
+                api_keys, min_required_balance=xdai_type(1), withdraw_multiplier=2
+            )
 
     def process_bets(self, market_type: MarketType) -> None:
         """

prediction_market_agent_tooling/markets/data_models.py

@@ -2,7 +2,7 @@ from datetime import datetime
 from enum import Enum
 from typing import TypeAlias
 
-from pydantic import BaseModel
+from pydantic import BaseModel, computed_field
 
 from prediction_market_agent_tooling.gtypes import OutcomeStr
 
@@ -41,6 +41,7 @@ class ResolvedBet(Bet):
     resolved_time: datetime
     profit: ProfitAmount
 
+    @computed_field  # type: ignore[misc]
     @property
     def is_correct(self) -> bool:
         return self.outcome == self.market_outcome

prediction_market_agent_tooling/markets/manifold/data_models.py

@@ -150,7 +150,7 @@ class ManifoldBet(BaseModel):
     id: str
     fees: ManifoldBetFees
     isCancelled: t.Optional[bool] = None
-    loanAmount: Mana
+    loanAmount: Mana | None
     orderAmount: t.Optional[Mana] = None
     fills: t.Optional[list[ManifoldBetFills]] = None
     createdTime: datetime

prediction_market_agent_tooling/markets/omen/omen.py

@@ -1000,3 +1000,40 @@ def get_binary_market_p_yes_history(market: OmenAgentMarket) -> list[Probability
             )
 
     return history
+
+
+def withdraw_wxdai_to_xdai_to_keep_balance(
+    api_keys: APIKeys,
+    min_required_balance: xDai,
+    withdraw_multiplier: float = 1.0,
+    web3: Web3 | None = None,
+) -> None:
+    """
+    Keeps xDai balance above the minimum required balance by withdrawing wxDai to xDai.
+    Optionally, the amount to withdraw can be multiplied by the `withdraw_multiplier`, which can be useful to keep a buffer.
+    """
+    # xDai needs to be in our wallet where we pay transaction fees, so do not check for Safe's balance.
+    current_balances = get_balances(api_keys.public_key, web3)
+
+    if current_balances.xdai >= min_required_balance:
+        logger.info(
+            f"Current xDai balance {current_balances.xdai} is more or equal than the required minimum balance {min_required_balance}."
+        )
+        return
+
+    need_to_withdraw = xDai(
+        (min_required_balance - current_balances.xdai) * withdraw_multiplier
+    )
+
+    if current_balances.wxdai < need_to_withdraw:
+        raise ValueError(
+            f"Current wxDai balance {current_balances.wxdai} is less than the required minimum wxDai to withdraw {need_to_withdraw}."
+        )
+
+    collateral_token_contract = OmenCollateralTokenContract()
+    collateral_token_contract.withdraw(
+        api_keys=api_keys, amount_wei=xdai_to_wei(need_to_withdraw), web3=web3
+    )
+    logger.info(
+        f"Withdrew {need_to_withdraw} wxDai to keep the balance above the minimum required balance {min_required_balance}."
+    )

prediction_market_agent_tooling/markets/omen/omen_contracts.py

@@ -630,6 +630,10 @@ class OmenThumbnailMapping(ContractOnGnosisChain):
         "0xe0cf08311F03850497B0ed6A2cf067f1750C3eFc"
     )
 
+    @staticmethod
+    def construct_ipfs_url(ipfs_hash: IPFSCIDVersion0) -> str:
+        return f"https://ipfs.io/ipfs/{ipfs_hash}"
+
     def get(
         self,
         market_address: ChecksumAddress,
@@ -646,7 +650,7 @@ class OmenThumbnailMapping(ContractOnGnosisChain):
         web3: Web3 | None = None,
     ) -> str | None:
         hash_ = self.get(market_address, web3)
-        return f"https://ipfs.io/ipfs/{hash_}" if hash_ is not None else None
+        return self.construct_ipfs_url(hash_) if hash_ is not None else None
 
     def set(
         self,

prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py

@@ -2,8 +2,11 @@ import sys
 import typing as t
 from datetime import datetime
 
+import requests
 import tenacity
 from eth_typing import ChecksumAddress
+from PIL import Image
+from PIL.Image import Image as ImageType
 from subgrounds import FieldPath, Subgrounds
 
 from prediction_market_agent_tooling.config import APIKeys
@@ -20,9 +23,15 @@ from prediction_market_agent_tooling.markets.omen.data_models import (
     RealityAnswer,
     RealityQuestion,
 )
+from prediction_market_agent_tooling.markets.omen.omen_contracts import (
+    OmenThumbnailMapping,
+)
 from prediction_market_agent_tooling.tools.singleton import SingletonMeta
 from prediction_market_agent_tooling.tools.utils import to_int_timestamp, utcnow
-from prediction_market_agent_tooling.tools.web3_utils import ZERO_BYTES
+from prediction_market_agent_tooling.tools.web3_utils import (
+    ZERO_BYTES,
+    byte32_to_ipfscidv0,
+)
 
 
 class OmenSubgraphHandler(metaclass=SingletonMeta):
@@ -36,6 +45,11 @@ class OmenSubgraphHandler(metaclass=SingletonMeta):
 
     REALITYETH_GRAPH_URL = "https://gateway-arbitrum.network.thegraph.com/api/{graph_api_key}/subgraphs/id/E7ymrCnNcQdAAgLbdFWzGE5mvr5Mb5T9VfT43FqA7bNh"
 
+    # TODO: Switch to arbitrum subgraph once it's published.
+    OMEN_IMAGE_MAPPING_GRAPH_URL = (
+        "https://api.studio.thegraph.com/query/63564/omen-thumbnailmapping/v0.0.3"
+    )
+
     INVALID_ANSWER = "ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff"
 
     def __init__(self) -> None:
@@ -66,6 +80,11 @@ class OmenSubgraphHandler(metaclass=SingletonMeta):
                 graph_api_key=keys.graph_api_key.get_secret_value()
             )
         )
+        self.omen_image_mapping_subgraph = self.sg.load_subgraph(
+            self.OMEN_IMAGE_MAPPING_GRAPH_URL.format(
+                graph_api_key=keys.graph_api_key.get_secret_value()
+            )
+        )
 
     def _get_fields_for_bets(self, bets_field: FieldPath) -> list[FieldPath]:
         markets = bets_field.fpmm
@@ -609,3 +628,23 @@ class OmenSubgraphHandler(metaclass=SingletonMeta):
                 f"Incorrect number of markets fetched {len(markets)}, expected 1."
             )
         return markets[0]
+
+    def get_market_image_url(self, market_id: HexAddress) -> str | None:
+        image = self.omen_image_mapping_subgraph.Query.omenThumbnailMapping(
+            id=market_id.lower()
+        )
+        fields = [image.id, image.image_hash]
+        result = self.sg.query_json(fields)
+        items = self._parse_items_from_json(result)
+        if not items:
+            return None
+        parsed = byte32_to_ipfscidv0(HexBytes(items[0]["image_hash"]))
+        return OmenThumbnailMapping.construct_ipfs_url(parsed)
+
+    def get_market_image(self, market_id: HexAddress) -> ImageType | None:
+        image_url = self.get_market_image_url(market_id)
+        return (
+            Image.open(requests.get(image_url, stream=True).raw)
+            if image_url is not None
+            else None
+        )

{prediction_market_agent_tooling-0.39.4.dist-info → prediction_market_agent_tooling-0.40.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prediction-market-agent-tooling
-Version: 0.39.4
+Version: 0.40.0
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.12

{prediction_market_agent_tooling-0.39.4.dist-info → prediction_market_agent_tooling-0.40.0.dist-info}/RECORD

@@ -12,8 +12,8 @@ prediction_market_agent_tooling/benchmark/__init__.py,sha256=47DEQpj8HBSa-_TImW-
 prediction_market_agent_tooling/benchmark/agents.py,sha256=HPIFJvackW110ch3UkktbxhU48gMRVo4gQse84Klhdc,4000
 prediction_market_agent_tooling/benchmark/benchmark.py,sha256=xiHKzZx5GHSsDerFHMZ9j_LXAXnSaITSvv67iPe3MEU,21095
 prediction_market_agent_tooling/benchmark/utils.py,sha256=iS1BzyXcCMfMm1Rx--1QCH0pHvBTblTndcDQFbTUJ2s,2897
-prediction_market_agent_tooling/config.py,sha256=pTeV4mbvPmrS_MzmiBklg05_gtsR3pa9Id5HCS-AzKQ,4519
-prediction_market_agent_tooling/deploy/agent.py,sha256=h33uU97x3DjBy8INONtS1Hc5B-NnNDV6MENeeCp0qoA,10185
+prediction_market_agent_tooling/config.py,sha256=kIluGnzhBWIs6DIEoBXbRcIEKaEDU39kVAa3lbIOWlY,4562
+prediction_market_agent_tooling/deploy/agent.py,sha256=TXuQ0L2-dG0tIrJ7srxD3Hbabe_2gILfdBQI-tkl9Og,10522
 prediction_market_agent_tooling/deploy/agent_example.py,sha256=tqXVA2HSFK3pdZ49tMmta8aKdJFT8JN8WeJ1akjrpBk,909
 prediction_market_agent_tooling/deploy/constants.py,sha256=M5ty8URipYMGe_G-RzxRydK3AFL6CyvmqCraJUrLBnE,82
 prediction_market_agent_tooling/deploy/gcp/deploy.py,sha256=CYUgnfy-9XVk04kkxA_5yp0GE9Mw5caYqlFUZQ2j3ks,3739
@@ -23,19 +23,19 @@ prediction_market_agent_tooling/gtypes.py,sha256=lbV2nsPyhMIRI9olx0_6A06jwTWKYBP
 prediction_market_agent_tooling/loggers.py,sha256=ua9rynYmsbOJZjxPIFxRBooomeN08zuLSJ7lxZMDS7w,3133
 prediction_market_agent_tooling/markets/agent_market.py,sha256=ZpBpNLqY2QXTAdmkCr8-L3yjN70H_y-KiCptx3OFQgM,8099
 prediction_market_agent_tooling/markets/categorize.py,sha256=yTd-lDMPW4ESDSzmxeLLBuzLX0FggOF7Vbswh7295o0,941
-prediction_market_agent_tooling/markets/data_models.py,sha256=6J8tyq-ZJI2HknqJolDCMfWXVXZsdBq3EhVQeIiT6bY,1418
+prediction_market_agent_tooling/markets/data_models.py,sha256=IYI8fne-OA18ZP1lcXySEo9rcJHYv3wZrUvJziSaHew,1476
 prediction_market_agent_tooling/markets/manifold/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prediction_market_agent_tooling/markets/manifold/api.py,sha256=m6qOzDiyQfxj62Eo_SzzQLkX4ijpi8KtQKGd4CpKAsk,7307
-prediction_market_agent_tooling/markets/manifold/data_models.py,sha256=Iw-ajDPYhtfcTfN45nJN_kZShfWcWe9mgzexSWnQCCQ,5295
+prediction_market_agent_tooling/markets/manifold/data_models.py,sha256=xoCqxk9cO6MZ_px_SEGYZ8hHyhGh-X6QDC8dDhyF3rk,5302
 prediction_market_agent_tooling/markets/manifold/manifold.py,sha256=EwRL06E2Y_ZAzr8efwS5yD6p6rnykrcBhqmNDUGZ8Aw,4075
 prediction_market_agent_tooling/markets/manifold/utils.py,sha256=cPPFWXm3vCYH1jy7_ctJZuQH9ZDaPL4_AgAYzGWkoow,513
 prediction_market_agent_tooling/markets/markets.py,sha256=w05Oo7yCA2axpCw69Q9y4i9Gcdpht0u5bZGbWqld3rU,2964
 prediction_market_agent_tooling/markets/omen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prediction_market_agent_tooling/markets/omen/data_models.py,sha256=EXtjmcujx68Xu50BVkYXvLuf_Asx5o65RvFS3ZS6HGs,14405
-prediction_market_agent_tooling/markets/omen/omen.py,sha256=dMun3MvkvEVrBmN2NwCCb1Uq3kd9nePe1FhFBgHHcj0,36868
-prediction_market_agent_tooling/markets/omen/omen_contracts.py,sha256=LlslAHfDkGytOje_ZMJDURAv2geAjInDAOnlteNFI0A,21722
+prediction_market_agent_tooling/markets/omen/omen.py,sha256=NtbfiaUQsN0zi7CYbBcQbasc32Dx7OmtIVMv-aXZtbk,38349
+prediction_market_agent_tooling/markets/omen/omen_contracts.py,sha256=JGXCO9MIVr-DGyoH2sxReAw7ZDTC_ev0UxDpq1QBv5Q,21854
 prediction_market_agent_tooling/markets/omen/omen_resolving.py,sha256=g77QsQ5WnSI2rzBlX87L_EhWMwobkyXyfRhHQmpAdzo,9012
-prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py,sha256=35cRqXuU4DvP5weYKcFNpJXhbaC1cfNBSfOYeO2HagE,23833
+prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py,sha256=cwimJjzWw7GNXYj_79ztgJawlwJQzf4Q9BPAtqByRFY,25248
 prediction_market_agent_tooling/markets/polymarket/api.py,sha256=HXmA1akA0qDj0m3e-GEvWG8x75pm6BX4H7YJPQcST7I,4767
 prediction_market_agent_tooling/markets/polymarket/data_models.py,sha256=9CJzakyEcsn6DQBK2nOXjOMzTZBLAmK_KqevXvW17DI,4292
 prediction_market_agent_tooling/markets/polymarket/data_models_web.py,sha256=f8SRQy0Rn-gIHSEMrJJAI8H3J7l8lzOLj3aCMe0vJv8,11324
@@ -69,8 +69,8 @@ prediction_market_agent_tooling/tools/singleton.py,sha256=CiIELUiI-OeS7U7eeHEt0r
 prediction_market_agent_tooling/tools/streamlit_user_login.py,sha256=NXEqfjT9Lc9QtliwSGRASIz1opjQ7Btme43H4qJbzgE,3010
 prediction_market_agent_tooling/tools/utils.py,sha256=-G22UEbCRm59bm1RWFdeF55hRsaxgwZVAHvK32-Ye1g,6190
 prediction_market_agent_tooling/tools/web3_utils.py,sha256=nKRHmdLnWSKd3wpo-cysXGvhhrJ2Yf69sN2FFQfSt6s,10578
-prediction_market_agent_tooling-0.39.4.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.39.4.dist-info/METADATA,sha256=krPaT1l3JTNUQFg-BDEz7Mg-_w6G0CytIYt69rgSCWc,7634
-prediction_market_agent_tooling-0.39.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-prediction_market_agent_tooling-0.39.4.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.39.4.dist-info/RECORD,,
+prediction_market_agent_tooling-0.40.0.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.40.0.dist-info/METADATA,sha256=Jok34BfwM81xbX25ug6pvRUb1a0ch8-vXpHJyNTGQs0,7634
+prediction_market_agent_tooling-0.40.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+prediction_market_agent_tooling-0.40.0.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.40.0.dist-info/RECORD,,