prediction-market-agent-tooling 0.57.17__py3-none-any.whl → 0.57.19__py3-none-any.whl

prediction_market_agent_tooling/config.py

@@ -1,11 +1,13 @@
 import typing as t
 
+from eth_account.signers.local import LocalAccount
 from pydantic import Field
 from pydantic.types import SecretStr
 from pydantic.v1.types import SecretStr as SecretStrV1
 from pydantic_settings import BaseSettings, SettingsConfigDict
 from safe_eth.eth import EthereumClient
 from safe_eth.safe.safe import SafeV141
+from web3 import Account
 
 from prediction_market_agent_tooling.gtypes import (
     ChainID,
@@ -184,6 +186,12 @@ class APIKeys(BaseSettings):
             self.SQLALCHEMY_DB_URL, "SQLALCHEMY_DB_URL missing in the environment."
         )
 
+    def get_account(self) -> LocalAccount:
+        acc: LocalAccount = Account.from_key(
+            self.bet_from_private_key.get_secret_value()
+        )
+        return acc
+
     def model_dump_public(self) -> dict[str, t.Any]:
         return {
             k: v

prediction_market_agent_tooling/markets/omen/omen.py

@@ -69,15 +69,18 @@ from prediction_market_agent_tooling.markets.omen.omen_subgraph_handler import (
 )
 from prediction_market_agent_tooling.tools.balances import get_balances
 from prediction_market_agent_tooling.tools.contract import (
-    ContractDepositableWrapperERC20BaseClass,
-    ContractERC4626BaseClass,
-    auto_deposit_collateral_token,
     init_collateral_token_contract,
     to_gnosis_chain_contract,
 )
 from prediction_market_agent_tooling.tools.custom_exceptions import OutOfFundsError
 from prediction_market_agent_tooling.tools.hexbytes_custom import HexBytes
 from prediction_market_agent_tooling.tools.ipfs.ipfs_handler import IPFSHandler
+from prediction_market_agent_tooling.tools.tokens.auto_deposit import (
+    auto_deposit_collateral_token,
+)
+from prediction_market_agent_tooling.tools.tokens.auto_withdraw import (
+    auto_withdraw_collateral_token,
+)
 from prediction_market_agent_tooling.tools.utils import (
     BPS_CONSTANT,
     DatetimeUTC,
@@ -901,19 +904,12 @@ def omen_sell_outcome_tx(
         max_outcome_tokens_to_sell,
         web3=web3,
     )
-    if auto_withdraw and (
-        isinstance(collateral_token_contract, ContractERC4626BaseClass)
-        or isinstance(
-            collateral_token_contract, ContractDepositableWrapperERC20BaseClass
-        )
-    ):
-        collateral_token_contract.withdraw(
-            api_keys,
-            remove_fraction(
-                amount_wei,
-                0.001,  # Allow 0.1% slippage.
-            ),
-            web3,
+    if auto_withdraw:
+        auto_withdraw_collateral_token(
+            collateral_token_contract=collateral_token_contract,
+            amount_wei=amount_wei,
+            api_keys=api_keys,
+            web3=web3,
         )
 
     return tx_receipt["transactionHash"].hex()

prediction_market_agent_tooling/markets/omen/omen_constants.py

@@ -0,0 +1,8 @@
+from web3 import Web3
+
+WRAPPED_XDAI_CONTRACT_ADDRESS = Web3.to_checksum_address(
+    "0xe91d153e0b41518a2ce8dd3d7944fa863463a97d"
+)
+SDAI_CONTRACT_ADDRESS = Web3.to_checksum_address(
+    "0xaf204776c7245bF4147c2612BF6e5972Ee483701"
+)

prediction_market_agent_tooling/markets/omen/omen_contracts.py

@@ -31,6 +31,10 @@ from prediction_market_agent_tooling.markets.omen.data_models import (
     RealitioLogNewQuestionEvent,
     format_realitio_question,
 )
+from prediction_market_agent_tooling.markets.omen.omen_constants import (
+    SDAI_CONTRACT_ADDRESS,
+    WRAPPED_XDAI_CONTRACT_ADDRESS,
+)
 from prediction_market_agent_tooling.tools.contract import (
     ContractDepositableWrapperERC20OnGnosisChain,
     ContractERC20OnGnosisChain,
@@ -419,18 +423,26 @@ class OmenFixedProductMarketMakerContract(ContractOnGnosisChain):
         )
 
 
-class WrappedxDaiContract(ContractDepositableWrapperERC20OnGnosisChain):
+class GNOContract(ContractERC20OnGnosisChain):
     address: ChecksumAddress = Web3.to_checksum_address(
-        "0xe91d153e0b41518a2ce8dd3d7944fa863463a97d"
+        "0x9c58bacc331c9aa871afd802db6379a98e80cedb"
     )
 
 
-class sDaiContract(ContractERC4626OnGnosisChain):
+class WETHContract(ContractERC20OnGnosisChain):
     address: ChecksumAddress = Web3.to_checksum_address(
-        "0xaf204776c7245bF4147c2612BF6e5972Ee483701"
+        "0x6a023ccd1ff6f2045c3309768ead9e68f978f6e1"
     )
 
 
+class WrappedxDaiContract(ContractDepositableWrapperERC20OnGnosisChain):
+    address: ChecksumAddress = WRAPPED_XDAI_CONTRACT_ADDRESS
+
+
+class sDaiContract(ContractERC4626OnGnosisChain):
+    address: ChecksumAddress = SDAI_CONTRACT_ADDRESS
+
+
 OMEN_DEFAULT_MARKET_FEE_PERC = 0.02  # 2% fee from the buying shares amount.
 REALITY_DEFAULT_FINALIZATION_TIMEOUT = timedelta(days=3)
 

prediction_market_agent_tooling/markets/seer/seer.py

@@ -9,11 +9,13 @@ from prediction_market_agent_tooling.markets.seer.seer_contracts import (
     SeerMarketFactory,
 )
 from prediction_market_agent_tooling.tools.contract import (
-    auto_deposit_collateral_token,
     init_collateral_token_contract,
     to_gnosis_chain_contract,
 )
 from prediction_market_agent_tooling.tools.datetime_utc import DatetimeUTC
+from prediction_market_agent_tooling.tools.tokens.auto_deposit import (
+    auto_deposit_collateral_token,
+)
 from prediction_market_agent_tooling.tools.web3_utils import xdai_to_wei
 
 

prediction_market_agent_tooling/tools/contract.py

@@ -22,11 +22,7 @@ from prediction_market_agent_tooling.gtypes import (
 )
 from prediction_market_agent_tooling.tools.data_models import MessageContainer
 from prediction_market_agent_tooling.tools.hexbytes_custom import HexBytes
-from prediction_market_agent_tooling.tools.utils import (
-    BPS_CONSTANT,
-    DatetimeUTC,
-    should_not_happen,
-)
+from prediction_market_agent_tooling.tools.utils import BPS_CONSTANT, DatetimeUTC
 from prediction_market_agent_tooling.tools.web3_utils import (
     call_function_on_contract,
     send_function_on_contract_tx,
@@ -798,99 +794,6 @@ def init_collateral_token_contract(
         )
 
 
-def auto_deposit_collateral_token(
-    collateral_token_contract: ContractERC20BaseClass,
-    amount_wei: Wei,
-    api_keys: APIKeys,
-    web3: Web3 | None,
-) -> None:
-    if isinstance(collateral_token_contract, ContractERC4626BaseClass):
-        auto_deposit_erc4626(collateral_token_contract, amount_wei, api_keys, web3)
-
-    elif isinstance(
-        collateral_token_contract, ContractDepositableWrapperERC20BaseClass
-    ):
-        auto_deposit_depositable_wrapper_erc20(
-            collateral_token_contract, amount_wei, api_keys, web3
-        )
-
-    elif isinstance(collateral_token_contract, ContractERC20BaseClass):
-        if (
-            collateral_token_contract.balanceOf(
-                for_address=api_keys.bet_from_address, web3=web3
-            )
-            < amount_wei
-        ):
-            raise ValueError(
-                f"Not enough of the collateral token, but it's not a wrapper token that we can deposit automatically."
-            )
-
-    else:
-        should_not_happen("Unsupported ERC20 contract type.")
-
-
-def auto_deposit_depositable_wrapper_erc20(
-    collateral_token_contract: ContractDepositableWrapperERC20BaseClass,
-    amount_wei: Wei,
-    api_keys: APIKeys,
-    web3: Web3 | None,
-) -> None:
-    collateral_token_balance = collateral_token_contract.balanceOf(
-        for_address=api_keys.bet_from_address, web3=web3
-    )
-
-    # If we have enough of the collateral token, we don't need to deposit.
-    if collateral_token_balance >= amount_wei:
-        return
-
-    # If we don't have enough, we need to deposit the difference.
-    left_to_deposit = Wei(amount_wei - collateral_token_balance)
-    collateral_token_contract.deposit(api_keys, left_to_deposit, web3=web3)
-
-
-def auto_deposit_erc4626(
-    collateral_token_contract: ContractERC4626BaseClass,
-    asset_amount_wei: Wei,
-    api_keys: APIKeys,
-    web3: Web3 | None,
-) -> None:
-    for_address = api_keys.bet_from_address
-    collateral_token_balance_in_shares = collateral_token_contract.balanceOf(
-        for_address=for_address, web3=web3
-    )
-    asset_amount_wei_in_shares = collateral_token_contract.convertToShares(
-        asset_amount_wei, web3
-    )
-
-    # If we have enough shares, we don't need to deposit.
-    if collateral_token_balance_in_shares >= asset_amount_wei_in_shares:
-        return
-
-    # If we need to deposit into erc4626, we first need to have enough of the asset token.
-    asset_token_contract = collateral_token_contract.get_asset_token_contract(web3=web3)
-
-    # If the asset token is Depositable Wrapper ERC-20, we can deposit it, in case we don't have enough.
-    if (
-        collateral_token_contract.get_asset_token_balance(for_address, web3)
-        < asset_amount_wei
-    ):
-        if isinstance(asset_token_contract, ContractDepositableWrapperERC20BaseClass):
-            auto_deposit_depositable_wrapper_erc20(
-                asset_token_contract, asset_amount_wei, api_keys, web3
-            )
-        else:
-            raise ValueError(
-                "Not enough of the asset token, but it's not a depositable wrapper token that we can deposit automatically."
-            )
-
-    # Finally, we can deposit the asset token into the erc4626 vault.
-    collateral_token_balance_in_assets = collateral_token_contract.convertToAssets(
-        collateral_token_balance_in_shares, web3
-    )
-    left_to_deposit = Wei(asset_amount_wei - collateral_token_balance_in_assets)
-    collateral_token_contract.deposit_asset_token(left_to_deposit, api_keys, web3)
-
-
 def to_gnosis_chain_contract(
     contract: ContractERC20BaseClass,
 ) -> ContractERC20OnGnosisChain:

prediction_market_agent_tooling/tools/cow/cow_order.py

@@ -0,0 +1,133 @@
+import asyncio
+from datetime import timedelta
+
+import httpx
+from cowdao_cowpy import swap_tokens
+from cowdao_cowpy.common.chains import Chain
+from cowdao_cowpy.common.config import SupportedChainId
+from cowdao_cowpy.common.constants import CowContractAddress
+from cowdao_cowpy.order_book.api import OrderBookApi
+from cowdao_cowpy.order_book.config import Envs, OrderBookAPIConfigFactory
+from cowdao_cowpy.order_book.generated.model import (
+    Address,
+    OrderMetaData,
+    OrderQuoteRequest,
+    OrderQuoteSide1,
+    OrderQuoteSideKindSell,
+    OrderStatus,
+    TokenAmount,
+)
+from eth_account.signers.local import LocalAccount
+from eth_typing.evm import ChecksumAddress
+from web3 import Web3
+from web3.types import Wei
+
+from prediction_market_agent_tooling.config import APIKeys
+from prediction_market_agent_tooling.gtypes import ChecksumAddress, Wei, wei_type
+from prediction_market_agent_tooling.loggers import logger
+from prediction_market_agent_tooling.tools.contract import ContractERC20OnGnosisChain
+from prediction_market_agent_tooling.tools.utils import utcnow
+
+
+def get_order_book_api(env: Envs, chain: Chain) -> OrderBookApi:
+    chain_id = SupportedChainId(chain.value[0])
+    return OrderBookApi(OrderBookAPIConfigFactory.get_config(env, chain_id))
+
+
+def get_buy_token_amount(
+    amount_wei: Wei,
+    sell_token: ChecksumAddress,
+    buy_token: ChecksumAddress,
+    chain: Chain = Chain.GNOSIS,
+    env: Envs = "prod",
+) -> Wei:
+    order_book_api = get_order_book_api(env, chain)
+    order_quote_request = OrderQuoteRequest(
+        sellToken=Address(sell_token),
+        buyToken=Address(buy_token),
+        from_=Address(
+            "0x1234567890abcdef1234567890abcdef12345678"
+        ),  # Just random address, doesn't matter.
+    )
+    order_side = OrderQuoteSide1(
+        kind=OrderQuoteSideKindSell.sell,
+        sellAmountBeforeFee=TokenAmount(str(amount_wei)),
+    )
+    order_quote = asyncio.run(
+        order_book_api.post_quote(order_quote_request, order_side)
+    )
+    return wei_type(order_quote.quote.buyAmount.root)
+
+
+def swap_tokens_waiting(
+    amount_wei: Wei,
+    sell_token: ChecksumAddress,
+    buy_token: ChecksumAddress,
+    api_keys: APIKeys,
+    chain: Chain = Chain.GNOSIS,
+    env: Envs = "prod",
+    web3: Web3 | None = None,
+) -> OrderMetaData:
+    account = api_keys.get_account()
+
+    # Approve the CoW Swap Vault Relayer to get the sell token.
+    ContractERC20OnGnosisChain(address=sell_token).approve(
+        api_keys,
+        Web3.to_checksum_address(CowContractAddress.VAULT_RELAYER.value),
+        amount_wei=amount_wei,
+        web3=web3,
+    )
+
+    # CoW library uses async, so we need to wrap the call in asyncio.run for us to use it.
+    return asyncio.run(
+        swap_tokens_waiting_async(
+            amount_wei, sell_token, buy_token, account, chain, env
+        )
+    )
+
+
+async def swap_tokens_waiting_async(
+    amount_wei: Wei,
+    sell_token: ChecksumAddress,
+    buy_token: ChecksumAddress,
+    account: LocalAccount,
+    chain: Chain,
+    env: Envs,
+    timeout: timedelta = timedelta(seconds=60),
+) -> OrderMetaData:
+    order = await swap_tokens(
+        amount=amount_wei,
+        sell_token=sell_token,
+        buy_token=buy_token,
+        account=account,
+        chain=chain,
+        env=env,
+    )
+    logger.info(f"Order created: {order}")
+    start_time = utcnow()
+
+    while True:
+        async with httpx.AsyncClient() as client:
+            response = await client.get(order.url)
+            order_metadata = OrderMetaData.model_validate(response.json())
+
+        if order_metadata.status == OrderStatus.fulfilled:
+            logger.info(f"Order {order.uid} ({order.url}) completed.")
+            return order_metadata
+
+        elif order_metadata.status in (
+            OrderStatus.cancelled,
+            OrderStatus.expired,
+        ):
+            raise ValueError(f"Order {order.uid} failed. {order.url}")
+
+        if utcnow() - start_time > timeout:
+            raise TimeoutError(
+                f"Timeout waiting for order {order.uid} to be completed. {order.url}"
+            )
+
+        logger.info(
+            f"Order status of {order.uid} ({order.url}): {order_metadata.status}, waiting..."
+        )
+
+        await asyncio.sleep(3.14)

prediction_market_agent_tooling/tools/tokens/auto_deposit.py

@@ -0,0 +1,156 @@
+from web3 import Web3
+
+from prediction_market_agent_tooling.config import APIKeys
+from prediction_market_agent_tooling.gtypes import Wei, wei_type
+from prediction_market_agent_tooling.loggers import logger
+from prediction_market_agent_tooling.tools.contract import (
+    ContractDepositableWrapperERC20BaseClass,
+    ContractERC20BaseClass,
+    ContractERC20OnGnosisChain,
+    ContractERC4626BaseClass,
+)
+from prediction_market_agent_tooling.tools.cow.cow_order import (
+    get_buy_token_amount,
+    swap_tokens_waiting,
+)
+from prediction_market_agent_tooling.tools.tokens.main_token import KEEPING_ERC20_TOKEN
+from prediction_market_agent_tooling.tools.utils import should_not_happen
+from prediction_market_agent_tooling.tools.web3_utils import wei_to_xdai
+
+
+def auto_deposit_collateral_token(
+    collateral_token_contract: ContractERC20BaseClass,
+    amount_wei: Wei,
+    api_keys: APIKeys,
+    web3: Web3 | None = None,
+) -> None:
+    if isinstance(collateral_token_contract, ContractDepositableWrapperERC20BaseClass):
+        # In this case, we can use deposit function directly, no need to go through DEX.
+        auto_deposit_depositable_wrapper_erc20(
+            collateral_token_contract, amount_wei, api_keys, web3
+        )
+
+    elif isinstance(collateral_token_contract, ContractERC4626BaseClass):
+        auto_deposit_erc4626(collateral_token_contract, amount_wei, api_keys, web3)
+
+    elif isinstance(collateral_token_contract, ContractERC20BaseClass):
+        auto_deposit_erc20(collateral_token_contract, amount_wei, api_keys, web3)
+
+    else:
+        should_not_happen("Unsupported ERC20 contract type.")
+
+
+def auto_deposit_depositable_wrapper_erc20(
+    collateral_token_contract: ContractDepositableWrapperERC20BaseClass,
+    amount_wei: Wei,
+    api_keys: APIKeys,
+    web3: Web3 | None,
+) -> None:
+    collateral_token_balance = collateral_token_contract.balanceOf(
+        for_address=api_keys.bet_from_address, web3=web3
+    )
+
+    # If we have enough of the collateral token, we don't need to deposit.
+    if collateral_token_balance >= amount_wei:
+        return
+
+    # If we don't have enough, we need to deposit the difference.
+    left_to_deposit = Wei(amount_wei - collateral_token_balance)
+    logger.info(
+        f"Depositing {wei_to_xdai(left_to_deposit)} {collateral_token_contract.symbol()}."
+    )
+    collateral_token_contract.deposit(api_keys, left_to_deposit, web3=web3)
+
+
+def auto_deposit_erc4626(
+    collateral_token_contract: ContractERC4626BaseClass,
+    asset_amount_wei: Wei,
+    api_keys: APIKeys,
+    web3: Web3 | None,
+) -> None:
+    for_address = api_keys.bet_from_address
+    collateral_token_balance_in_shares = collateral_token_contract.balanceOf(
+        for_address=for_address, web3=web3
+    )
+    asset_amount_wei_in_shares = collateral_token_contract.convertToShares(
+        asset_amount_wei, web3
+    )
+
+    # If we have enough shares, we don't need to deposit.
+    if collateral_token_balance_in_shares >= asset_amount_wei_in_shares:
+        return
+
+    # If we need to deposit into erc4626, we first need to have enough of the asset token.
+    asset_token_contract = collateral_token_contract.get_asset_token_contract(web3=web3)
+
+    if isinstance(asset_token_contract, ContractDepositableWrapperERC20BaseClass):
+        # If the asset token is Depositable Wrapper ERC-20, we don't need to go through DEX.
+        # First, calculate how much of asset token we need to deposit into the vault.
+        collateral_token_balance_in_assets = collateral_token_contract.convertToAssets(
+            collateral_token_balance_in_shares, web3
+        )
+        left_to_deposit = Wei(asset_amount_wei - collateral_token_balance_in_assets)
+        if (
+            collateral_token_contract.get_asset_token_balance(for_address, web3)
+            < left_to_deposit
+        ):
+            # If we don't have enough of asset token to deposit into the vault, deposit that one first.
+            auto_deposit_depositable_wrapper_erc20(
+                asset_token_contract, left_to_deposit, api_keys, web3
+            )
+        # And finally, we can deposit the asset token into the erc4626 vault directly as well, without DEX.
+        collateral_token_contract.deposit_asset_token(left_to_deposit, api_keys, web3)
+
+    else:
+        # Otherwise, we need to go through DEX.
+        auto_deposit_erc20(collateral_token_contract, asset_amount_wei, api_keys, web3)
+
+
+def auto_deposit_erc20(
+    collateral_token_contract: ContractERC20BaseClass,
+    amount_xdai_wei: Wei,
+    api_keys: APIKeys,
+    web3: Web3 | None,
+) -> None:
+    # How much it is in the other token (collateral token).
+    collateral_amount_wei = get_buy_token_amount(
+        amount_xdai_wei,
+        KEEPING_ERC20_TOKEN.address,
+        collateral_token_contract.address,
+    )
+    # How much do we have already in the other token (collateral token).
+    collateral_balance_wei = collateral_token_contract.balanceOf(
+        api_keys.bet_from_address
+    )
+    # Amount of collateral token remaining to get.
+    remaining_to_get_in_collateral_wei = max(
+        0, collateral_amount_wei - collateral_balance_wei
+    )
+    if not remaining_to_get_in_collateral_wei:
+        return
+    # Estimate of how much of the source token we need to sell in order to fill the remaining collateral amount, with 1% slippage to be sure.
+    amount_to_sell_wei = wei_type(
+        (remaining_to_get_in_collateral_wei * amount_xdai_wei)
+        / collateral_amount_wei
+        * 1.01
+    )
+    # If we don't have enough of the source token.
+    if amount_to_sell_wei > ContractERC20OnGnosisChain(
+        address=KEEPING_ERC20_TOKEN.address
+    ).balanceOf(api_keys.bet_from_address):
+        # Try to deposit it, if it's depositable token (like Wrapped xDai, agent could have xDai).
+        if isinstance(KEEPING_ERC20_TOKEN, ContractDepositableWrapperERC20BaseClass):
+            auto_deposit_depositable_wrapper_erc20(
+                KEEPING_ERC20_TOKEN, amount_to_sell_wei, api_keys, web3
+            )
+        else:
+            raise ValueError(
+                "Not enough of the source token to sell to get the desired amount of the collateral token."
+            )
+    swap_tokens_waiting(
+        amount_wei=amount_to_sell_wei,
+        sell_token=KEEPING_ERC20_TOKEN.address,
+        buy_token=collateral_token_contract.address,
+        api_keys=api_keys,
+        web3=web3,
+    )

prediction_market_agent_tooling/tools/tokens/auto_withdraw.py

@@ -0,0 +1,62 @@
+from web3 import Web3
+
+from prediction_market_agent_tooling.config import APIKeys
+from prediction_market_agent_tooling.gtypes import Wei
+from prediction_market_agent_tooling.tools.contract import (
+    ContractERC20BaseClass,
+    ContractERC4626BaseClass,
+)
+from prediction_market_agent_tooling.tools.cow.cow_order import (
+    get_buy_token_amount,
+    swap_tokens_waiting,
+)
+from prediction_market_agent_tooling.tools.tokens.main_token import KEEPING_ERC20_TOKEN
+from prediction_market_agent_tooling.tools.utils import should_not_happen
+from prediction_market_agent_tooling.tools.web3_utils import remove_fraction
+
+
+def auto_withdraw_collateral_token(
+    collateral_token_contract: ContractERC20BaseClass,
+    amount_wei: Wei,
+    api_keys: APIKeys,
+    web3: Web3 | None = None,
+    slippage: float = 0.001,
+) -> None:
+    # Small slippage as exact exchange rate constantly changes and we don't care about small differences.
+    amount_wei = remove_fraction(
+        amount_wei,
+        slippage,
+    )
+
+    if collateral_token_contract.address == KEEPING_ERC20_TOKEN.address:
+        # Do nothing, as this is the token we want to keep.
+        return
+    elif (
+        isinstance(collateral_token_contract, ContractERC4626BaseClass)
+        and collateral_token_contract.get_asset_token_contract().address
+        == KEEPING_ERC20_TOKEN.address
+    ):
+        # If the ERC4626 is backed by KEEPING_ERC20_TOKEN, we can withdraw it directly, no need to go through DEX.
+        collateral_token_contract.withdraw(
+            api_keys,
+            amount_wei,
+            web3=web3,
+        )
+    elif isinstance(collateral_token_contract, ContractERC20BaseClass):
+        # Otherwise, DEX will handle the rest of token swaps.
+        # First, convert `amount_wei` from xDai-based value into the collateral token-based value.
+        collateral_amount_wei = get_buy_token_amount(
+            amount_wei,
+            KEEPING_ERC20_TOKEN.address,
+            collateral_token_contract.address,
+        )
+        # And then sell it.
+        swap_tokens_waiting(
+            amount_wei=collateral_amount_wei,
+            sell_token=collateral_token_contract.address,
+            buy_token=KEEPING_ERC20_TOKEN.address,
+            api_keys=api_keys,
+            web3=web3,
+        )
+    else:
+        should_not_happen("Unsupported ERC20 contract type.")

prediction_market_agent_tooling/tools/tokens/main_token.py

@@ -0,0 +1,13 @@
+from prediction_market_agent_tooling.markets.omen.omen_constants import (
+    WRAPPED_XDAI_CONTRACT_ADDRESS,
+)
+from prediction_market_agent_tooling.tools.contract import (
+    ContractDepositableWrapperERC20OnGnosisChain,
+)
+
+# This is the token where agents will hold their funds,
+# except for a small portion that will be kept in the native token of the network to pay for the fees.
+# If changed, then keep in mind that we assume this token is equal to 1 USD.
+KEEPING_ERC20_TOKEN = ContractDepositableWrapperERC20OnGnosisChain(
+    address=WRAPPED_XDAI_CONTRACT_ADDRESS
+)

{prediction_market_agent_tooling-0.57.17.dist-info → prediction_market_agent_tooling-0.57.19.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: prediction-market-agent-tooling
-Version: 0.57.17
+Version: 0.57.19
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.13
@@ -14,6 +14,7 @@ Provides-Extra: openai
 Provides-Extra: optuna
 Requires-Dist: autoflake (>=2.2.1,<3.0.0)
 Requires-Dist: base58 (>=1.0.2,<2.0)
+Requires-Dist: cowdao-cowpy (>=1.0.0rc1,<2.0.0)
 Requires-Dist: cron-validator (>=1.0.8,<2.0.0)
 Requires-Dist: eth-account (>=0.8.0,<0.12.0)
 Requires-Dist: eth-keys (>=0.6.1,<0.7.0)
@@ -24,6 +25,7 @@ Requires-Dist: google-cloud-functions (>=1.16.0,<2.0.0)
 Requires-Dist: google-cloud-resource-manager (>=1.12.0,<2.0.0)
 Requires-Dist: google-cloud-secret-manager (>=2.18.2,<3.0.0)
 Requires-Dist: hishel (>=0.0.31,<0.0.32)
+Requires-Dist: httpx (>=0.25.2,<1.0.0)
 Requires-Dist: isort (>=5.13.2,<6.0.0)
 Requires-Dist: langchain (>=0.2.6,<0.3.0) ; extra == "langchain"
 Requires-Dist: langchain-community (>=0.0.19)
@@ -47,7 +49,7 @@ Requires-Dist: safe-eth-py (>=6.0.0b41,<7.0.0)
 Requires-Dist: scikit-learn (>=1.3.1,<2.0.0)
 Requires-Dist: sqlmodel (>=0.0.22,<0.0.23)
 Requires-Dist: streamlit (>=1.31.0,<2.0.0)
-Requires-Dist: subgrounds (>=1.9.1,<2.0.0)
+Requires-Dist: subgrounds-gnosis (>=1.9.2,<2.0.0)
 Requires-Dist: tabulate (>=0.9.0,<0.10.0)
 Requires-Dist: tavily-python (>=0.5.0,<0.6.0)
 Requires-Dist: tqdm (>=4.66.2,<5.0.0)

{prediction_market_agent_tooling-0.57.17.dist-info → prediction_market_agent_tooling-0.57.19.dist-info}/RECORD

@@ -22,7 +22,7 @@ prediction_market_agent_tooling/benchmark/__init__.py,sha256=47DEQpj8HBSa-_TImW-
 prediction_market_agent_tooling/benchmark/agents.py,sha256=B1-uWdyeN4GGKMWGK_-CcAFJg1m9Y_XuaeIHPB29QR8,3971
 prediction_market_agent_tooling/benchmark/benchmark.py,sha256=MqTiaaJ3cYiOLUVR7OyImLWxcEya3Rl5JyFYW-K0lwM,17097
 prediction_market_agent_tooling/benchmark/utils.py,sha256=D0MfUkVZllmvcU0VOurk9tcKT7JTtwwOp-63zuCBVuc,2880
-prediction_market_agent_tooling/config.py,sha256=Ffb1ftRiguAxa5cS0nXvKK01HomdqzFDCsp8rFd5taM,7553
+prediction_market_agent_tooling/config.py,sha256=QBrbqG9wN1JFBUt_kCl2ybPamobKP6UhfiaRZK_VjMo,7805
 prediction_market_agent_tooling/deploy/agent.py,sha256=58Ms6wFcRM3yhpdDVT-ohryZhHJNinn1Z4MdrmCMXvM,23627
 prediction_market_agent_tooling/deploy/agent_example.py,sha256=dIIdZashExWk9tOdyDjw87AuUcGyM7jYxNChYrVK2dM,1001
 prediction_market_agent_tooling/deploy/betting_strategy.py,sha256=kMrIE3wMv_IB6nJd_1DmDXDkEZhsXFOgyTd7JZ0gqHI,13068
@@ -52,8 +52,9 @@ prediction_market_agent_tooling/markets/metaculus/data_models.py,sha256=Suxa7xEL
 prediction_market_agent_tooling/markets/metaculus/metaculus.py,sha256=86TIx6cavEWc8Cv4KpZxSvwiSw9oFybXE3YB49pg-CA,4377
 prediction_market_agent_tooling/markets/omen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prediction_market_agent_tooling/markets/omen/data_models.py,sha256=uT8ILKrg2g4jGodPxtolPErk25buNzMYndb01ZL2dYE,28421
-prediction_market_agent_tooling/markets/omen/omen.py,sha256=w6uf_hDpVS2_9LnfmC3ZX9KZSMtXeuUvyOr1fs9nGZU,51880
-prediction_market_agent_tooling/markets/omen/omen_contracts.py,sha256=baXJwk-jSI3-FV-k239oCNOA4oKz6LT47juX8AKpW3A,28297
+prediction_market_agent_tooling/markets/omen/omen.py,sha256=xNHYn5diKjKlkeKoYNr8DrE0gSMsU8poGhxBjw2PZow,51783
+prediction_market_agent_tooling/markets/omen/omen_constants.py,sha256=D9oflYKafLQiHYtB5sScMHqmXyzM8JP8J0yATmc4SQQ,233
+prediction_market_agent_tooling/markets/omen/omen_contracts.py,sha256=EXqBlVivbmW8aBQ65O09X2xkyesHAop49GUl1tUffWA,28648
 prediction_market_agent_tooling/markets/omen/omen_resolving.py,sha256=E1BVDvQd1qYtCxmfC94kJtGkmQqpGPHL3zTkcs5wW6M,9697
 prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py,sha256=mZ0CgKfHj0gLFq9plNpBhNqMuclb8V3qNagWfLYcpUc,38806
 prediction_market_agent_tooling/markets/polymarket/api.py,sha256=UZ4_TG8ceb9Y-qgsOKs8Qiv8zDt957QkT8IX2c83yqo,4800
@@ -62,7 +63,7 @@ prediction_market_agent_tooling/markets/polymarket/data_models_web.py,sha256=VZh
 prediction_market_agent_tooling/markets/polymarket/polymarket.py,sha256=NRoZK71PtH8kkangMqme7twcAXhRJSSabbmOir-UnAI,3418
 prediction_market_agent_tooling/markets/polymarket/utils.py,sha256=DImFxeMg8lTfsEDZ8FavndW38TfUsCkawcVGnucsuGo,2029
 prediction_market_agent_tooling/markets/seer/data_models.py,sha256=DmxkAfusRbupqBDjohSFhsYyW7WjKZ1sQKpm8xeAMi4,2017
-prediction_market_agent_tooling/markets/seer/seer.py,sha256=sPjhWdC3ri3xavD_7dnGAUFyZvPrrUOJoVXNhE8fvqM,3277
+prediction_market_agent_tooling/markets/seer/seer.py,sha256=l87L82M-RHMBJjfdtNXWgGLc979n2Fub9Z702zE8I9c,3351
 prediction_market_agent_tooling/markets/seer/seer_contracts.py,sha256=E7CYAKZiK6cg3dyj1kJuIPKSYYUft98F64shF5S0g4s,2730
 prediction_market_agent_tooling/markets/seer/seer_subgraph_handler.py,sha256=KraQe4GgbdMcYnvpEJqcSHXQqgyDiC7IjkIOHAfma9s,5528
 prediction_market_agent_tooling/monitor/financial_metrics/financial_metrics.py,sha256=fjIgjDIx5MhH5mwf7S0cspLOOSU3elYLhGYoIiM26mU,2746
@@ -83,8 +84,9 @@ prediction_market_agent_tooling/tools/betting_strategies/utils.py,sha256=kpIb-ci
 prediction_market_agent_tooling/tools/caches/db_cache.py,sha256=aafau_n_AUbLIwkyIRiTPgKB0dmM0767mSqyPDLF2A4,10576
 prediction_market_agent_tooling/tools/caches/inmemory_cache.py,sha256=ZW5iI5rmjqeAebu5T7ftRnlkxiL02IC-MxCfDB80x7w,1506
 prediction_market_agent_tooling/tools/caches/serializers.py,sha256=vFDx4fsPxclXp2q0sv27j4al_M_Tj9aR2JJP-xNHQXA,2151
-prediction_market_agent_tooling/tools/contract.py,sha256=VKSH-31Cf3DwtHpH9KZnQr08IxmfyvYvuqkcOK7dUN8,29608
+prediction_market_agent_tooling/tools/contract.py,sha256=JkkrXm75_XUZ33jVlXF7v-Ge9RLSfXQeY523H_kuThQ,26069
 prediction_market_agent_tooling/tools/costs.py,sha256=EaAJ7v9laD4VEV3d8B44M4u3_oEO_H16jRVCdoZ93Uw,954
+prediction_market_agent_tooling/tools/cow/cow_order.py,sha256=ohriN_65BZ69f_BV6bNg_AwZjidip4yi4d2G6Ddy5Qg,4238
 prediction_market_agent_tooling/tools/custom_exceptions.py,sha256=Fh8z1fbwONvP4-j7AmV_PuEcoqb6-QXa9PJ9m7guMcM,93
 prediction_market_agent_tooling/tools/data_models.py,sha256=jDQ7FU0QQhXlcgJh5VZZGwDTYP2OPAqKPHZFewCPAUY,732
 prediction_market_agent_tooling/tools/datetime_utc.py,sha256=8_WackjtjC8zHXrhQFTGQ6e6Fz_6llWoKR4CSFvIv9I,2766
@@ -109,11 +111,14 @@ 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/tavily/tavily_models.py,sha256=5ldQs1pZe6uJ5eDAuP4OLpzmcqYShlIV67kttNFvGS0,342
 prediction_market_agent_tooling/tools/tavily/tavily_search.py,sha256=Kw2mXNkMTYTEe1MBSTqhQmLoeXtgb6CkmHlcAJvhtqE,3809
+prediction_market_agent_tooling/tools/tokens/auto_deposit.py,sha256=o8_ERfPL-ps9FLvH5vgdiSRJQ4dZONJw9KK9sHgeP2I,6390
+prediction_market_agent_tooling/tools/tokens/auto_withdraw.py,sha256=B02maQkl3wNptWFtZdnDosICJE5BfnLUVqxp5uJIaPA,2353
+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.57.17.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.57.17.dist-info/METADATA,sha256=4gih1dpxZvldnv0oRnp5g0e8-dhNYHxjXzKHRdIIRd8,8452
-prediction_market_agent_tooling-0.57.17.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-prediction_market_agent_tooling-0.57.17.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.57.17.dist-info/RECORD,,
+prediction_market_agent_tooling-0.57.19.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.57.19.dist-info/METADATA,sha256=NAaCqa6AaiRsgR3wu3r70MKXHPeDc9UuqmkdjMlL6AI,8546
+prediction_market_agent_tooling-0.57.19.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+prediction_market_agent_tooling-0.57.19.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.57.19.dist-info/RECORD,,