prediction-market-agent-tooling 0.58.0__py3-none-any.whl → 0.58.2__py3-none-any.whl

prediction_market_agent_tooling/deploy/gcp/utils.py

@@ -1,6 +1,7 @@
 import json
 import subprocess
 import sys
+from functools import cache
 
 from google.cloud.functions_v2.services.function_service.client import (
     FunctionServiceClient,
@@ -197,6 +198,7 @@ def gcp_function_is_active(fname: str) -> bool:
     return get_gcp_function(fname).state == Function.State.ACTIVE
 
 
+@cache
 def gcp_get_secret_value(name: str, version: str = "latest") -> str:
     client = SecretManagerServiceClient()
     return client.access_secret_version(

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

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

{prediction_market_agent_tooling-0.58.0.dist-info → prediction_market_agent_tooling-0.58.2.dist-info}/RECORD

@@ -29,7 +29,7 @@ prediction_market_agent_tooling/deploy/betting_strategy.py,sha256=kMrIE3wMv_IB6n
 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
 prediction_market_agent_tooling/deploy/gcp/kubernetes_models.py,sha256=OsPboCFGiZKsvGyntGZHwdqPlLTthITkNF5rJFvGgU8,2582
-prediction_market_agent_tooling/deploy/gcp/utils.py,sha256=dzAvSQWRc2pndHFyKhVgrTtDm91K-csgusEaFRyfSow,5664
+prediction_market_agent_tooling/deploy/gcp/utils.py,sha256=WI2ycX1X-IlTRoNoG4ggFlRwPL28kwM9VGDFD2fePLo,5699
 prediction_market_agent_tooling/deploy/trade_interval.py,sha256=Xk9j45alQ_vrasGvsNyuW70XHIQ7wfvjoxNR3F6HYCw,1155
 prediction_market_agent_tooling/gtypes.py,sha256=G9KOKqYcxoKLv5Tfto4g5zq46FeIKxGl4RTArLIJn3I,2563
 prediction_market_agent_tooling/jobs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -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.0.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.58.0.dist-info/METADATA,sha256=E_PM7SHleOyz3e52aCzYfAdmKRrfRZqR05PzbQLZ6kk,8629
-prediction_market_agent_tooling-0.58.0.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-prediction_market_agent_tooling-0.58.0.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.58.0.dist-info/RECORD,,
+prediction_market_agent_tooling/tools/web3_utils.py,sha256=e7lqqQddVZaa905rhBb6L1fC3o39Yr-PDJsJjHFBeRE,12523
+prediction_market_agent_tooling-0.58.2.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.58.2.dist-info/METADATA,sha256=6a2-ucKTXSn19qa7dDHal28aNeopkwoEJB9jOdfW-9A,8629
+prediction_market_agent_tooling-0.58.2.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+prediction_market_agent_tooling-0.58.2.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.58.2.dist-info/RECORD,,