prediction-market-agent-tooling 0.62.0.dev475__py3-none-any.whl → 0.62.0.dev491__py3-none-any.whl

prediction_market_agent_tooling/markets/omen/omen.py

@@ -788,7 +788,7 @@ def omen_sell_outcome_tx(
     outcome: str,
     auto_withdraw: bool,
     web3: Web3 | None = None,
-    slippage: float = 0.001,
+    slippage: float = 0.01,
 ) -> str:
     """
     Sells the given xDai value of shares corresponding to the given outcome in
@@ -807,7 +807,6 @@ def omen_sell_outcome_tx(
         else market.get_in_token(amount)
     )
     amount_wei = amount_token.as_wei
-    amount_wei = amount_wei.without_fraction(slippage)
 
     logger.info(
         f"Selling asked {amount.value=} {amount.symbol}, converted to {amount_wei.as_token.value=} {amount_wei.as_token.symbol} for {outcome=} in market {market.url=}."

prediction_market_agent_tooling/tools/tokens/auto_deposit.py

@@ -23,7 +23,7 @@ def auto_deposit_collateral_token(
     collateral_amount_wei_or_usd: Wei | USD,
     api_keys: APIKeys,
     web3: Web3 | None = None,
-    slippage: float = 0.01,
+    surplus: float = 0.01,
 ) -> None:
     collateral_amount_wei = (
         collateral_amount_wei_or_usd
@@ -32,8 +32,8 @@ def auto_deposit_collateral_token(
             collateral_amount_wei_or_usd, collateral_token_contract.address
         ).as_wei
     )
-    # Deposit a bit more, to cover any later changes in the price.
-    collateral_amount_wei = collateral_amount_wei.with_fraction(slippage)
+    # Deposit a bit more to cover any small changes in conversions.
+    collateral_amount_wei = collateral_amount_wei.with_fraction(surplus)
 
     if isinstance(collateral_token_contract, ContractDepositableWrapperERC20BaseClass):
         # In this case, we can use deposit function directly, no need to go through DEX.
@@ -137,14 +137,11 @@ def auto_deposit_erc20(
     )
     if not remaining_to_get_in_collateral_wei:
         return
-    # Get 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 = (
-        get_sell_token_amount(
-            remaining_to_get_in_collateral_wei,
-            sell_token=KEEPING_ERC20_TOKEN.address,
-            buy_token=collateral_token_contract.address,
-        )
-        * 1.01
+    # Get of how much of the source token we need to sell in order to fill the remaining collateral amount.
+    amount_to_sell_wei = get_sell_token_amount(
+        remaining_to_get_in_collateral_wei,
+        sell_token=KEEPING_ERC20_TOKEN.address,
+        buy_token=collateral_token_contract.address,
     )
     # If we don't have enough of the source token.
     if amount_to_sell_wei > ContractERC20OnGnosisChain(

prediction_market_agent_tooling/tools/tokens/auto_withdraw.py

@@ -17,11 +17,7 @@ def auto_withdraw_collateral_token(
     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 = amount_wei.without_fraction(slippage)
-
     if not amount_wei:
         logger.warning(
             f"Amount to withdraw is zero, skipping withdrawal of {collateral_token_contract.symbol_cached(web3)}."

prediction_market_agent_tooling/tools/tokens/usd.py

@@ -5,7 +5,6 @@ from prediction_market_agent_tooling.gtypes import (
     USD,
     ChecksumAddress,
     CollateralToken,
-    OutcomeToken,
     xDai,
 )
 from prediction_market_agent_tooling.markets.omen.omen_constants import (
@@ -27,23 +26,18 @@ def get_xdai_in_usd(amount: xDai) -> USD:
 
 
 def get_usd_in_token(amount: USD, token_address: ChecksumAddress) -> CollateralToken:
-    rate = get_single_token_unit_to_usd_rate(token_address)
-    return CollateralToken(amount.value / rate.value)
+    rate = get_single_usd_to_token_rate(token_address)
+    return CollateralToken(amount.value * rate.value)
 
 
-def get_token_in_usd(
-    amount: CollateralToken | OutcomeToken, token_address: ChecksumAddress
-) -> USD:
-    rate = get_single_token_unit_to_usd_rate(token_address)
+def get_token_in_usd(amount: CollateralToken, token_address: ChecksumAddress) -> USD:
+    rate = get_single_token_to_usd_rate(token_address)
     return USD(amount.value * rate.value)
 
 
 # A short cache to not spam CoW and prevent timeouts, but still have relatively fresh data.
-RATE_CACHE: TTLCache[ChecksumAddress, USD] = TTLCache(maxsize=100, ttl=5 * 60)
-
-
-@cached(RATE_CACHE)
-def get_single_token_unit_to_usd_rate(token_address: ChecksumAddress) -> USD:
+@cached(TTLCache(maxsize=100, ttl=5 * 60))
+def get_single_token_to_usd_rate(token_address: ChecksumAddress) -> USD:
     # (w)xDai is a stable coin against USD, so use it to estimate USD worth.
     if WRAPPED_XDAI_CONTRACT_ADDRESS == token_address:
         return USD(1.0)
@@ -61,3 +55,25 @@ def get_single_token_unit_to_usd_rate(token_address: ChecksumAddress) -> USD:
     )
     in_token = in_wei.as_token
     return USD(in_token.value)
+
+
+# A short cache to not spam CoW and prevent timeouts, but still have relatively fresh data.
+@cached(TTLCache(maxsize=100, ttl=5 * 60))
+def get_single_usd_to_token_rate(token_address: ChecksumAddress) -> CollateralToken:
+    # (w)xDai is a stable coin against USD, so use it to estimate USD worth.
+    if WRAPPED_XDAI_CONTRACT_ADDRESS == token_address:
+        return CollateralToken(1.0)
+    # sDai is ERC4626 with wxDai as asset, we can take the rate directly from there instead of calling CoW.
+    if SDAI_CONTRACT_ADDRESS == token_address:
+        return CollateralToken(
+            ContractERC4626OnGnosisChain(address=SDAI_CONTRACT_ADDRESS)
+            .convertToShares(CollateralToken(1).as_wei)
+            .as_token.value
+        )
+    in_wei = get_buy_token_amount(
+        sell_amount=CollateralToken(1).as_wei,
+        sell_token=WRAPPED_XDAI_CONTRACT_ADDRESS,
+        buy_token=token_address,
+    )
+    in_token = in_wei.as_token
+    return CollateralToken(in_token.value)

{prediction_market_agent_tooling-0.62.0.dev475.dist-info → prediction_market_agent_tooling-0.62.0.dev491.dist-info}/METADATA

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

{prediction_market_agent_tooling-0.62.0.dev475.dist-info → prediction_market_agent_tooling-0.62.0.dev491.dist-info}/RECORD

@@ -51,7 +51,7 @@ prediction_market_agent_tooling/markets/metaculus/data_models.py,sha256=FaBCTPPe
 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=URvplDcTLeL6vfpkSgi8ln2iEKxt8w5qf6mTZkSyrCo,29189
-prediction_market_agent_tooling/markets/omen/omen.py,sha256=L1NaKGkdCVtGVz7RbkBuHaYRyYfuYRthTxmnYWNwgeU,51949
+prediction_market_agent_tooling/markets/omen/omen.py,sha256=_Vt1_w_A6-FHPVgNclwSh2qpABBXXTJ2wKPtcFfrDTQ,51893
 prediction_market_agent_tooling/markets/omen/omen_constants.py,sha256=D9oflYKafLQiHYtB5sScMHqmXyzM8JP8J0yATmc4SQQ,233
 prediction_market_agent_tooling/markets/omen/omen_contracts.py,sha256=bCC9A7ZTJxMDJcPbl3jof6HcAGGHv1BrFq3RRWbkQ_c,28739
 prediction_market_agent_tooling/markets/omen/omen_resolving.py,sha256=Cyi9Ci-Z-K8WCZWVLs9oSuJC6qRobi_CpqDCno_5F-0,10238
@@ -112,16 +112,16 @@ 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=pPs0qZNfJ7G-1ajfz0iaWOBQyiC0TbcShfrW8T39jtg,3859
-prediction_market_agent_tooling/tools/tokens/auto_deposit.py,sha256=ZKGJPoRhch7QVDkZ_5HbBM1si-BKiWyrfuePRh51wqo,6777
-prediction_market_agent_tooling/tools/tokens/auto_withdraw.py,sha256=nV9jSUt1ydceDj03cTwdyrJs8kFGqz9z6YcvTVUaNg4,2682
+prediction_market_agent_tooling/tools/tokens/auto_deposit.py,sha256=MWIRHYCN-7o2uawEBwnBNhlzme1kLyX_5Q2zg7UIegQ,6700
+prediction_market_agent_tooling/tools/tokens/auto_withdraw.py,sha256=22g0SIVmLlgITpdt3kPhJOw0sU4OPeBuYk_7xCrQr9U,2491
 prediction_market_agent_tooling/tools/tokens/main_token.py,sha256=1rbwpdCusPgQIVFuo3m00nBZ_b2lCAoFVm67i-YDcEw,812
 prediction_market_agent_tooling/tools/tokens/token_utils.py,sha256=zwV-jGFkPJu4-IslXOUqnsNQjzh_9CrfkruDQL0dq0c,1381
-prediction_market_agent_tooling/tools/tokens/usd.py,sha256=FmQUfrkLvN8cI_6vw9QNhOXdaqYlw-DxevP39VA8hmQ,2240
+prediction_market_agent_tooling/tools/tokens/usd.py,sha256=l4vPemtF6Zcwh8HNXH3-L5k7OE_XolbklzBsTpFruW8,3116
 prediction_market_agent_tooling/tools/transaction_cache.py,sha256=K5YKNL2_tR10Iw2TD9fuP-CTGpBbZtNdgbd0B_R7pjg,1814
 prediction_market_agent_tooling/tools/utils.py,sha256=1xsyBBJfiEdSoMlceB2F8o2sCb6Z8-qNz11pEJFrdyE,6566
 prediction_market_agent_tooling/tools/web3_utils.py,sha256=eYCc1iWAVtqDKUPTwnMUHuYolPdwh_OTiM3-AdRgDp4,12198
-prediction_market_agent_tooling-0.62.0.dev475.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.62.0.dev475.dist-info/METADATA,sha256=DPsjD7mRc-nEPy68lzMeE5X2cMG-1X2IB_cpOBqk3V4,8696
-prediction_market_agent_tooling-0.62.0.dev475.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
-prediction_market_agent_tooling-0.62.0.dev475.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.62.0.dev475.dist-info/RECORD,,
+prediction_market_agent_tooling-0.62.0.dev491.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.62.0.dev491.dist-info/METADATA,sha256=ke5Gwr94EoF5Ng4zCSmSZ82erv0TIiILRsXBE3IcyT8,8696
+prediction_market_agent_tooling-0.62.0.dev491.dist-info/WHEEL,sha256=XbeZDeTWKc1w7CSIyre5aMDU_-PohRwTQceYnisIYYY,88
+prediction_market_agent_tooling-0.62.0.dev491.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.62.0.dev491.dist-info/RECORD,,