prediction-market-agent-tooling 0.34.0__py3-none-any.whl → 0.35.0__py3-none-any.whl

prediction_market_agent_tooling/markets/omen/omen.py

@@ -10,6 +10,7 @@ from prediction_market_agent_tooling.gtypes import (
     ChecksumAddress,
     HexAddress,
     HexStr,
+    OmenOutcomeToken,
     OutcomeStr,
     Probability,
     Wei,
@@ -54,7 +55,10 @@ 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.hexbytes_custom import HexBytes
-from prediction_market_agent_tooling.tools.utils import check_not_none
+from prediction_market_agent_tooling.tools.utils import (
+    calculate_sell_amount_in_collateral,
+    check_not_none,
+)
 from prediction_market_agent_tooling.tools.web3_utils import (
     add_fraction,
     remove_fraction,
@@ -80,6 +84,8 @@ class OmenAgentMarket(AgentMarket):
     finalized_time: datetime | None
     created_time: datetime
     close_time: datetime
+    outcome_token_amounts: list[OmenOutcomeToken]
+    fee: float  # proportion, from 0 to 1
 
     INVALID_MARKET_ANSWER: HexStr = HexStr(
         "0xFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF"
@@ -155,19 +161,50 @@ class OmenAgentMarket(AgentMarket):
             web3=web3,
         )
 
+    def calculate_sell_amount_in_collateral(
+        self, amount: TokenAmount, outcome: bool
+    ) -> xDai:
+        if len(self.outcome_token_amounts) != 2:
+            raise ValueError(
+                f"Market {self.id} has {len(self.outcome_token_amounts)} "
+                f"outcomes. This method only supports binary markets."
+            )
+        sell_index = self.yes_index if outcome else self.no_index
+        other_index = self.no_index if outcome else self.yes_index
+        collateral = calculate_sell_amount_in_collateral(
+            shares_to_sell=amount.amount,
+            holdings=wei_to_xdai(Wei(self.outcome_token_amounts[sell_index])),
+            other_holdings=wei_to_xdai(Wei(self.outcome_token_amounts[other_index])),
+            fee=self.fee,
+        )
+        return xDai(collateral)
+
     def sell_tokens(
-        self, outcome: bool, amount: TokenAmount, auto_withdraw: bool = False
+        self,
+        outcome: bool,
+        amount: TokenAmount,
+        auto_withdraw: bool = False,
+        api_keys: APIKeys | None = None,
+        web3: Web3 | None = None,
     ) -> None:
         if not self.can_be_traded():
             raise ValueError(
                 f"Market {self.id} is not open for trading. Cannot sell tokens."
             )
+
+        # Convert from token (i.e. share) number to xDai value of tokens, as
+        # this is the expected unit of the argument in the smart contract.
+        collateral = self.calculate_sell_amount_in_collateral(
+            amount=amount,
+            outcome=outcome,
+        )
         binary_omen_sell_outcome_tx(
-            api_keys=APIKeys(),
-            amount=xDai(amount.amount),
+            amount=collateral,
+            api_keys=api_keys if api_keys is not None else APIKeys(),
             market=self,
             binary_outcome=outcome,
             auto_withdraw=auto_withdraw,
+            web3=web3,
         )
 
     def was_any_bet_outcome_correct(
@@ -245,6 +282,8 @@ class OmenAgentMarket(AgentMarket):
             url=model.url,
             volume=wei_to_xdai(model.collateralVolume),
             close_time=model.close_time,
+            outcome_token_amounts=model.outcomeTokenAmounts,
+            fee=float(wei_to_xdai(model.fee)) if model.fee is not None else 0.0,
         )
 
     @staticmethod

prediction_market_agent_tooling/tools/utils.py

@@ -7,6 +7,7 @@ from typing import Any, NoReturn, Optional, Type, TypeVar, cast
 import pytz
 import requests
 from pydantic import BaseModel, ValidationError
+from scipy.optimize import newton
 from scipy.stats import entropy
 
 from prediction_market_agent_tooling.gtypes import (
@@ -173,3 +174,35 @@ def prob_uncertainty(prob: Probability) -> float:
         - Market's probability is 0.95, so the market is quite certain about YES: prob_uncertainty(0.95) == 0.286
     """
     return float(entropy([prob, 1 - prob], base=2))
+
+
+def calculate_sell_amount_in_collateral(
+    shares_to_sell: float,
+    holdings: float,
+    other_holdings: float,
+    fee: float,
+) -> float:
+    """
+    Computes the amount of collateral that needs to be sold to get `shares`
+    amount of shares. Returns None if the amount can't be computed.
+
+    Taken from https://github.com/protofire/omen-exchange/blob/29d0ab16bdafa5cc0d37933c1c7608a055400c73/app/src/util/tools/fpmm/trading/index.ts#L99
+    Simplified for binary markets.
+    """
+
+    if not (0 <= fee < 1.0):
+        raise ValueError("Fee must be between 0 and 1")
+
+    for v in [shares_to_sell, holdings, other_holdings]:
+        if v <= 0:
+            raise ValueError("All share args must be greater than 0")
+
+    def f(r: float) -> float:
+        R = r / (1 - fee)
+        first_term = other_holdings - R
+        second_term = holdings + shares_to_sell - R
+        third_term = holdings * other_holdings
+        return (first_term * second_term) - third_term
+
+    amount_to_sell = newton(f, 0)
+    return float(amount_to_sell) * 0.999999  # Avoid rounding errors

{prediction_market_agent_tooling-0.34.0.dist-info → prediction_market_agent_tooling-0.35.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prediction-market-agent-tooling
-Version: 0.34.0
+Version: 0.35.0
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.12

{prediction_market_agent_tooling-0.34.0.dist-info → prediction_market_agent_tooling-0.35.0.dist-info}/RECORD

@@ -31,7 +31,7 @@ prediction_market_agent_tooling/markets/manifold/utils.py,sha256=cPPFWXm3vCYH1jy
 prediction_market_agent_tooling/markets/markets.py,sha256=w05Oo7yCA2axpCw69Q9y4i9Gcdpht0u5bZGbWqld3rU,2964
 prediction_market_agent_tooling/markets/omen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 prediction_market_agent_tooling/markets/omen/data_models.py,sha256=EXtjmcujx68Xu50BVkYXvLuf_Asx5o65RvFS3ZS6HGs,14405
-prediction_market_agent_tooling/markets/omen/omen.py,sha256=9le-tvs-zclHtzpvK7bMVqg2lZV-VZh1c5OB_uAdSiY,32042
+prediction_market_agent_tooling/markets/omen/omen.py,sha256=aJ5CCste61B7q0OD3EjLP8H7AjD_RaBBNyUfnZGh5KM,33597
 prediction_market_agent_tooling/markets/omen/omen_contracts.py,sha256=eDS8vN4Klv_-Y1wwfIeLDt3twhl9U_AJjPQov0JScb0,19888
 prediction_market_agent_tooling/markets/omen/omen_resolving.py,sha256=g77QsQ5WnSI2rzBlX87L_EhWMwobkyXyfRhHQmpAdzo,9012
 prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py,sha256=RlaQMDF75lo2fos8LUb_OAdrw_ILLOXQLdejzE7tOmA,23053
@@ -64,10 +64,10 @@ prediction_market_agent_tooling/tools/parallelism.py,sha256=8mgkF5sBwFGS5GMvlpza
 prediction_market_agent_tooling/tools/safe.py,sha256=h0xOO0eNtitClf0fPkn-0oTc6A_bflDTee98V_aiV-A,5195
 prediction_market_agent_tooling/tools/singleton.py,sha256=CiIELUiI-OeS7U7eeHEt0rnVhtQGzwoUdAgn_7u_GBM,729
 prediction_market_agent_tooling/tools/streamlit_user_login.py,sha256=NXEqfjT9Lc9QtliwSGRASIz1opjQ7Btme43H4qJbzgE,3010
-prediction_market_agent_tooling/tools/utils.py,sha256=zkmwPi3YisgZDPCeNwaRbL8sInOYOkvFgFY4Q8PbEo4,5077
+prediction_market_agent_tooling/tools/utils.py,sha256=-G22UEbCRm59bm1RWFdeF55hRsaxgwZVAHvK32-Ye1g,6190
 prediction_market_agent_tooling/tools/web3_utils.py,sha256=cboATXNmEdn5RmPbVblHOwOdUMKBYrUK3GiI6i6Vzxo,9855
-prediction_market_agent_tooling-0.34.0.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.34.0.dist-info/METADATA,sha256=Ih-qmKV0wWypWJv0ABswvSd1VSPOJBQwDiXw714KSx0,7515
-prediction_market_agent_tooling-0.34.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-prediction_market_agent_tooling-0.34.0.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.34.0.dist-info/RECORD,,
+prediction_market_agent_tooling-0.35.0.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.35.0.dist-info/METADATA,sha256=PGIW673s30z0PdKLKcyhroDIjsBJWwzuZ7S-2bqIzyU,7515
+prediction_market_agent_tooling-0.35.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+prediction_market_agent_tooling-0.35.0.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.35.0.dist-info/RECORD,,