prediction-market-agent-tooling 0.58.1__py3-none-any.whl → 0.58.3.dev385__py3-none-any.whl

prediction_market_agent_tooling/tools/omen/sell_positions.py

@@ -0,0 +1,59 @@
+from datetime import timedelta
+
+from prediction_market_agent_tooling.config import APIKeys
+from prediction_market_agent_tooling.loggers import logger
+from prediction_market_agent_tooling.markets.omen.omen import OmenAgentMarket
+from prediction_market_agent_tooling.markets.omen.omen_subgraph_handler import (
+    OmenSubgraphHandler,
+)
+from prediction_market_agent_tooling.tools.balances import get_balances
+from prediction_market_agent_tooling.tools.utils import utcnow
+
+
+def sell_all(
+    api_keys: APIKeys,
+    closing_later_than_days: int,
+    auto_withdraw: bool = False,
+) -> None:
+    """
+    Helper function to sell all existing outcomes on Omen that would resolve later than in X days.
+    """
+    better_address = api_keys.bet_from_address
+    bets = OmenSubgraphHandler().get_bets(
+        better_address=better_address,
+        market_opening_after=utcnow() + timedelta(days=closing_later_than_days),
+    )
+    bets_total_usd = sum(b.collateral_amount_usd for b in bets)
+    unique_market_urls = set(b.fpmm.url for b in bets)
+    starting_balance = get_balances(better_address)
+    new_balance = starting_balance
+
+    logger.info(
+        f"For {better_address}, found the following {len(bets)} bets on {len(unique_market_urls)} unique markets worth of {bets_total_usd} USD: {unique_market_urls}"
+    )
+
+    for bet in bets:
+        agent_market = OmenAgentMarket.from_data_model(bet.fpmm)
+        outcome = agent_market.outcomes[bet.outcomeIndex]
+        current_token_balance = agent_market.get_token_balance(better_address, outcome)
+
+        if current_token_balance.amount <= OmenAgentMarket.get_tiny_bet_amount().amount:
+            logger.info(
+                f"Skipping bet on {bet.fpmm.url} because the actual balance is unreasonably low {current_token_balance.amount}."
+            )
+            continue
+
+        old_balance = new_balance
+        agent_market.sell_tokens(
+            bet.boolean_outcome,
+            current_token_balance,
+            auto_withdraw=auto_withdraw,
+            api_keys=api_keys,
+        )
+        new_balance = get_balances(better_address)
+
+        logger.info(
+            f"Sold bet on {bet.fpmm.url} for {new_balance.wxdai - old_balance.wxdai} xDai."
+        )
+
+    logger.info(f"Obtained back {new_balance.wxdai - starting_balance.wxdai} wxDai.")

prediction_market_agent_tooling/tools/web3_utils.py

@@ -351,6 +351,13 @@ def byte32_to_ipfscidv0(hex: HexBytes) -> IPFSCIDVersion0:
     return IPFSCIDVersion0(base58.b58encode(completed_binary_str).decode("utf-8"))
 
 
+@tenacity.retry(
+    wait=tenacity.wait_chain(*[tenacity.wait_fixed(n) for n in range(1, 10)]),
+    stop=tenacity.stop_after_attempt(5),
+    after=lambda x: logger.debug(
+        f"get_receipt_block_timestamp failed, {x.attempt_number=}."
+    ),
+)
 def get_receipt_block_timestamp(receipt_tx: TxReceipt, web3: Web3) -> int:
     block_number = receipt_tx["blockNumber"]
     block = web3.eth.get_block(block_number)

{prediction_market_agent_tooling-0.58.1.dist-info → prediction_market_agent_tooling-0.58.3.dev385.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: prediction-market-agent-tooling
-Version: 0.58.1
+Version: 0.58.3.dev385
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.13
@@ -39,7 +39,7 @@ Requires-Dist: optuna (>=4.1.0,<5.0.0) ; extra == "optuna"
 Requires-Dist: pinatapy-vourhey (>=0.2.0,<0.3.0)
 Requires-Dist: prompt-toolkit (>=3.0.43,<4.0.0)
 Requires-Dist: proto-plus (>=1.0.0,<2.0.0)
-Requires-Dist: protobuf (>=4.0.0,<5.0.0)
+Requires-Dist: protobuf (>=5.0.0,<6.0.0)
 Requires-Dist: psycopg2-binary (>=2.9.9,<3.0.0)
 Requires-Dist: pydantic (>=2.6.1,<3.0.0)
 Requires-Dist: pydantic-settings (>=2.4.0,<3.0.0)

{prediction_market_agent_tooling-0.58.1.dist-info → prediction_market_agent_tooling-0.58.3.dev385.dist-info}/RECORD

@@ -102,6 +102,7 @@ prediction_market_agent_tooling/tools/is_predictable.py,sha256=qVd6zqay2Dg2fyeAu
 prediction_market_agent_tooling/tools/langfuse_.py,sha256=jI_4ROxqo41CCnWGS1vN_AeDVhRzLMaQLxH3kxDu3L8,1153
 prediction_market_agent_tooling/tools/langfuse_client_utils.py,sha256=83T31s-YbsnNBLyYCjmBI2BBKUEqJUuYFa0uCdkoqy8,5901
 prediction_market_agent_tooling/tools/omen/reality_accuracy.py,sha256=M1SF7iSW1gVlQSTskdVFTn09uPLST23YeipVIWj54io,2236
+prediction_market_agent_tooling/tools/omen/sell_positions.py,sha256=JIfTMKF0bHyQ9qXNfyr5ul_eu4hBuydlRVGJET8Uw90,2308
 prediction_market_agent_tooling/tools/parallelism.py,sha256=6Gou0hbjtMZrYvxjTDFUDZuxmE2nqZVbb6hkg1hF82A,1022
 prediction_market_agent_tooling/tools/relevant_news_analysis/data_models.py,sha256=95l84aztFaxcRLLcRQ46yKJbIlOEuDAbIGLouyliDzA,1316
 prediction_market_agent_tooling/tools/relevant_news_analysis/relevant_news_analysis.py,sha256=CddJem7tk15NAudJDwmuL8znTycbR-YI8kTNtd3LzG8,5474
@@ -116,9 +117,9 @@ prediction_market_agent_tooling/tools/tokens/auto_withdraw.py,sha256=B02maQkl3wN
 prediction_market_agent_tooling/tools/tokens/main_token.py,sha256=5iHO7-iehSlXuue6ocVrr4IsklVjm7QHIwln4BsebJA,573
 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=wqUDCed3iNrn1Wao1iwGN6tzIrhpzrTRj319wlveJEo,12275
-prediction_market_agent_tooling-0.58.1.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.58.1.dist-info/METADATA,sha256=TN65w8MLP9CCXVec1tociFt0mAYyiHbiDWNE_MzU9ds,8629
-prediction_market_agent_tooling-0.58.1.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-prediction_market_agent_tooling-0.58.1.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.58.1.dist-info/RECORD,,
+prediction_market_agent_tooling/tools/web3_utils.py,sha256=e7lqqQddVZaa905rhBb6L1fC3o39Yr-PDJsJjHFBeRE,12523
+prediction_market_agent_tooling-0.58.3.dev385.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.58.3.dev385.dist-info/METADATA,sha256=BUNlEFZXKyo10bb9d49r_eyTZgC8tRHExC6OukNMZFU,8636
+prediction_market_agent_tooling-0.58.3.dev385.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+prediction_market_agent_tooling-0.58.3.dev385.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.58.3.dev385.dist-info/RECORD,,