prediction-market-agent-tooling 0.38.0__py3-none-any.whl → 0.39.1__py3-none-any.whl

prediction_market_agent_tooling/markets/agent_market.py

@@ -114,6 +114,20 @@ class AgentMarket(BaseModel):
         """
         raise NotImplementedError("Subclasses must implement this method")
 
+    def get_last_trade_yes_outcome_price(self) -> float | None:
+        # Price on prediction markets are, by definition, equal to the probability of an outcome.
+        # Just making it explicit in this function.
+        if last_trade_p_yes := self.get_last_trade_p_yes():
+            return float(last_trade_p_yes)
+        return None
+
+    def get_last_trade_no_outcome_price(self) -> float | None:
+        # Price on prediction markets are, by definition, equal to the probability of an outcome.
+        # Just making it explicit in this function.
+        if last_trade_p_no := self.get_last_trade_p_no():
+            return float(last_trade_p_no)
+        return None
+
     def get_bet_amount(self, amount: float) -> BetAmount:
         return BetAmount(amount=amount, currency=self.currency)
 
@@ -193,6 +207,13 @@ class AgentMarket(BaseModel):
         """
         raise NotImplementedError("Subclasses must implement this method")
 
+    @classmethod
+    def get_positions_value(cls, positions: list[Position]) -> BetAmount:
+        """
+        Get the total value of all positions held by a user.
+        """
+        raise NotImplementedError("Subclasses must implement this method")
+
     def can_be_traded(self) -> bool:
         if self.is_closed() or not self.has_liquidity():
             return False

prediction_market_agent_tooling/markets/omen/omen.py

@@ -421,6 +421,67 @@ class OmenAgentMarket(AgentMarket):
 
         return positions
 
+    @classmethod
+    def get_positions_value(cls, positions: list[Position]) -> BetAmount:
+        # Two dicts to map from market ids to (1) positions and (2) market.
+        market_ids_positions = {p.market_id: p for p in positions}
+        # Check there is only one position per market.
+        if len(set(market_ids_positions.keys())) != len(positions):
+            raise ValueError(
+                f"Markets for positions ({market_ids_positions.keys()}) are not unique."
+            )
+        markets: list[OmenAgentMarket] = [
+            OmenAgentMarket.from_data_model(m)
+            for m in OmenSubgraphHandler().get_omen_binary_markets(
+                limit=sys.maxsize, id_in=list(market_ids_positions.keys())
+            )
+        ]
+        market_ids_markets = {m.id: m for m in markets}
+
+        # Validate that dict keys are the same.
+        if set(market_ids_positions.keys()) != set(market_ids_markets.keys()):
+            raise ValueError(
+                f"Market ids in {market_ids_positions.keys()} are not the same as in {market_ids_markets.keys()}"
+            )
+
+        # Initialise position value.
+        total_position_value = 0.0
+
+        for market_id in market_ids_positions.keys():
+            position = market_ids_positions[market_id]
+            market = market_ids_markets[market_id]
+
+            yes_tokens = 0.0
+            no_tokens = 0.0
+            if OMEN_TRUE_OUTCOME in position.amounts:
+                yes_tokens = position.amounts[OutcomeStr(OMEN_TRUE_OUTCOME)].amount
+            if OMEN_FALSE_OUTCOME in position.amounts:
+                no_tokens = position.amounts[OutcomeStr(OMEN_FALSE_OUTCOME)].amount
+
+            # Account for the value of positions in resolved markets
+            if market.is_resolved() and market.has_successful_resolution():
+                valued_tokens = yes_tokens if market.boolean_outcome else no_tokens
+                total_position_value += valued_tokens
+
+            # Or if the market is open and trading, get the value of the position
+            elif market.can_be_traded():
+                total_position_value += yes_tokens * market.yes_outcome_price
+                total_position_value += no_tokens * market.no_outcome_price
+
+            # Or if the market is still open but not trading, estimate the value
+            # of the position
+            else:
+                if yes_tokens:
+                    yes_price = check_not_none(
+                        market.get_last_trade_yes_outcome_price()
+                    )
+                    total_position_value += yes_tokens * yes_price
+                if no_tokens:
+                    no_price = check_not_none(market.get_last_trade_no_outcome_price())
+                    total_position_value += no_tokens * no_price
+
+        return BetAmount(amount=total_position_value, currency=Currency.xDai)
+
     @classmethod
     def get_user_url(cls, keys: APIKeys) -> str:
         return f"https://gnosisscan.io/address/{keys.bet_from_address}"

prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py

@@ -166,6 +166,7 @@ class OmenSubgraphHandler(metaclass=SingletonMeta):
         resolved: bool | None = None,
         liquidity_bigger_than: Wei | None = None,
         condition_id_in: list[HexBytes] | None = None,
+        id_in: list[str] | None = None,
         excluded_questions: set[str] | None = None,
     ) -> dict[str, t.Any]:
         where_stms: dict[str, t.Any] = {
@@ -193,6 +194,9 @@ class OmenSubgraphHandler(metaclass=SingletonMeta):
         if condition_id_in is not None:
             where_stms["condition_"]["id_in"] = [x.hex() for x in condition_id_in]
 
+        if id_in is not None:
+            where_stms["id_in"] = id_in
+
         if resolved is not None:
             if resolved:
                 where_stms["resolutionTimestamp_not"] = None
@@ -308,6 +312,7 @@ class OmenSubgraphHandler(metaclass=SingletonMeta):
         creator: t.Optional[HexAddress] = None,
         liquidity_bigger_than: Wei | None = None,
         condition_id_in: list[HexBytes] | None = None,
+        id_in: list[str] | None = None,
         excluded_questions: set[str] | None = None,  # question titles
         sort_by_field: FieldPath | None = None,
         sort_direction: str | None = None,
@@ -326,6 +331,7 @@ class OmenSubgraphHandler(metaclass=SingletonMeta):
             finalized=finalized,
             resolved=resolved,
             condition_id_in=condition_id_in,
+            id_in=id_in,
             excluded_questions=excluded_questions,
             liquidity_bigger_than=liquidity_bigger_than,
         )

{prediction_market_agent_tooling-0.38.0.dist-info → prediction_market_agent_tooling-0.39.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prediction-market-agent-tooling
-Version: 0.38.0
+Version: 0.39.1
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.12
@@ -35,7 +35,7 @@ Requires-Dist: safe-cli (>=1.0.0,<2.0.0)
 Requires-Dist: safe-eth-py (>=6.0.0b14,<7.0.0)
 Requires-Dist: scikit-learn (>=1.3.1,<2.0.0)
 Requires-Dist: streamlit (>=1.31.0,<2.0.0)
-Requires-Dist: subgrounds (>=1.8.1,<2.0.0)
+Requires-Dist: subgrounds (>=1.9.1,<2.0.0)
 Requires-Dist: tabulate (>=0.9.0,<0.10.0)
 Requires-Dist: tqdm (>=4.66.2,<5.0.0)
 Requires-Dist: typer (>=0.9.0,<1.0.0)

{prediction_market_agent_tooling-0.38.0.dist-info → prediction_market_agent_tooling-0.39.1.dist-info}/RECORD

@@ -21,7 +21,7 @@ prediction_market_agent_tooling/deploy/gcp/kubernetes_models.py,sha256=qYIHRxQLa
 prediction_market_agent_tooling/deploy/gcp/utils.py,sha256=oyW0jgrUT2Tr49c7GlpcMsYNQjoCSOcWis3q-MmVAhU,6089
 prediction_market_agent_tooling/gtypes.py,sha256=lbV2nsPyhMIRI9olx0_6A06jwTWKYBPGMxyiGVFysag,2467
 prediction_market_agent_tooling/loggers.py,sha256=ua9rynYmsbOJZjxPIFxRBooomeN08zuLSJ7lxZMDS7w,3133
-prediction_market_agent_tooling/markets/agent_market.py,sha256=SMvkXct_RgHXqF-fVq3ooTIQ_99MG77kVlvS3rM8Ozo,7019
+prediction_market_agent_tooling/markets/agent_market.py,sha256=OduMFTRPHLIi_YYpzFHW6cKloMgVvoez9HbdB3F9OkA,7943
 prediction_market_agent_tooling/markets/categorize.py,sha256=yTd-lDMPW4ESDSzmxeLLBuzLX0FggOF7Vbswh7295o0,941
 prediction_market_agent_tooling/markets/data_models.py,sha256=uODY3aoFp8YYeLAUcrzMk1yU8pIKsTLobB9xIEGTmKs,1170
 prediction_market_agent_tooling/markets/manifold/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -32,10 +32,10 @@ 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=fS19WoDG_N8WIa5FY-YRF21Np46y6d93QBJOpNKXi7Q,33939
+prediction_market_agent_tooling/markets/omen/omen.py,sha256=yigcdBM7bT8rinj_gADFADXRNJCZFCMHedQJ5eeQZk4,36786
 prediction_market_agent_tooling/markets/omen/omen_contracts.py,sha256=-EEmN4vA8Fn52ewqQ7ZXZAFnfP4EQ4YISkeDHb9oeLk,21722
 prediction_market_agent_tooling/markets/omen/omen_resolving.py,sha256=g77QsQ5WnSI2rzBlX87L_EhWMwobkyXyfRhHQmpAdzo,9012
-prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py,sha256=2y89FJlDrJ_hpeNTFZU1W30xCpt8QXjssEW8bNZnEpU,23596
+prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py,sha256=6TC-uTq6DEQdXkbsZ0jI-g9zbqoO7UT3RCLFN1mQtAk,23772
 prediction_market_agent_tooling/markets/polymarket/api.py,sha256=HXmA1akA0qDj0m3e-GEvWG8x75pm6BX4H7YJPQcST7I,4767
 prediction_market_agent_tooling/markets/polymarket/data_models.py,sha256=9CJzakyEcsn6DQBK2nOXjOMzTZBLAmK_KqevXvW17DI,4292
 prediction_market_agent_tooling/markets/polymarket/data_models_web.py,sha256=f8SRQy0Rn-gIHSEMrJJAI8H3J7l8lzOLj3aCMe0vJv8,11324
@@ -69,8 +69,8 @@ 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/utils.py,sha256=-G22UEbCRm59bm1RWFdeF55hRsaxgwZVAHvK32-Ye1g,6190
 prediction_market_agent_tooling/tools/web3_utils.py,sha256=nKRHmdLnWSKd3wpo-cysXGvhhrJ2Yf69sN2FFQfSt6s,10578
-prediction_market_agent_tooling-0.38.0.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.38.0.dist-info/METADATA,sha256=BCdK9YuigO1Rdphbyjb_MfbI1HgJJkQsnH53ZAF6opg,7634
-prediction_market_agent_tooling-0.38.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-prediction_market_agent_tooling-0.38.0.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.38.0.dist-info/RECORD,,
+prediction_market_agent_tooling-0.39.1.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.39.1.dist-info/METADATA,sha256=MP61P9hHdpxiOXEhSj03oZIXr9qTScvcR1SM7o4z1C4,7634
+prediction_market_agent_tooling-0.39.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+prediction_market_agent_tooling-0.39.1.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.39.1.dist-info/RECORD,,