prediction-market-agent-tooling 0.58.0__tar.gz → 0.58.2__tar.gz

{prediction_market_agent_tooling-0.58.0 → prediction_market_agent_tooling-0.58.2}/PKG-INFO

@@ -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 → prediction_market_agent_tooling-0.58.2}/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-0.58.2/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-0.58.0 → prediction_market_agent_tooling-0.58.2}/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 → prediction_market_agent_tooling-0.58.2}/pyproject.toml

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