prediction-market-agent-tooling 0.62.0.dev475__tar.gz → 0.62.0.dev490__tar.gz

{prediction_market_agent_tooling-0.62.0.dev475 → prediction_market_agent_tooling-0.62.0.dev490}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: prediction-market-agent-tooling
-Version: 0.62.0.dev475
+Version: 0.62.0.dev490
 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 → prediction_market_agent_tooling-0.62.0.dev490}/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 → prediction_market_agent_tooling-0.62.0.dev490}/pyproject.toml

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