prediction-market-agent-tooling 0.66.2__tar.gz → 0.66.3__tar.gz

{prediction_market_agent_tooling-0.66.2 → prediction_market_agent_tooling-0.66.3}/PKG-INFO

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

{prediction_market_agent_tooling-0.66.2 → prediction_market_agent_tooling-0.66.3}/prediction_market_agent_tooling/deploy/betting_strategy.py

@@ -40,6 +40,9 @@ class GuaranteedLossError(RuntimeError):
 
 
 class BettingStrategy(ABC):
+    def __init__(self, take_profit: bool = True) -> None:
+        self.take_profit = take_profit
+
     @abstractmethod
     def calculate_trades(
         self,
@@ -209,10 +212,22 @@ class BettingStrategy(ABC):
             )
 
             diff_amount = target_amount - existing_amount
+
             if diff_amount == 0:
                 continue
 
             trade_type = TradeType.SELL if diff_amount < 0 else TradeType.BUY
+
+            # We work with positions, so imagine following scenario: Agent invested $10 when probs were 50:50,
+            # now the probs are 99:1 and his initial $10 is worth $100.
+            # If `take_profit` is set to False, agent won't sell the $90 to get back to the $10 position.
+            if (
+                not self.take_profit
+                and target_amount > 0
+                and trade_type == TradeType.SELL
+            ):
+                continue
+
             trade = Trade(
                 amount=abs(diff_amount),
                 outcome=outcome,
@@ -231,7 +246,8 @@ class BettingStrategy(ABC):
 
 
 class MultiCategoricalMaxAccuracyBettingStrategy(BettingStrategy):
-    def __init__(self, max_position_amount: USD):
+    def __init__(self, max_position_amount: USD, take_profit: bool = True):
+        super().__init__(take_profit=take_profit)
         self.max_position_amount = max_position_amount
 
     @property
@@ -336,7 +352,13 @@ class MaxExpectedValueBettingStrategy(MultiCategoricalMaxAccuracyBettingStrategy
 
 
 class KellyBettingStrategy(BettingStrategy):
-    def __init__(self, max_position_amount: USD, max_price_impact: float | None = None):
+    def __init__(
+        self,
+        max_position_amount: USD,
+        max_price_impact: float | None = None,
+        take_profit: bool = True,
+    ):
+        super().__init__(take_profit=take_profit)
         self.max_position_amount = max_position_amount
         self.max_price_impact = max_price_impact
 
@@ -492,7 +514,12 @@ class KellyBettingStrategy(BettingStrategy):
 
 
 class MaxAccuracyWithKellyScaledBetsStrategy(BettingStrategy):
-    def __init__(self, max_position_amount: USD):
+    def __init__(
+        self,
+        max_position_amount: USD,
+        take_profit: bool = True,
+    ):
+        super().__init__(take_profit)
         self.max_position_amount = max_position_amount
 
     @property

{prediction_market_agent_tooling-0.66.2 → prediction_market_agent_tooling-0.66.3}/prediction_market_agent_tooling/markets/omen/omen_subgraph_handler.py

@@ -53,7 +53,7 @@ from prediction_market_agent_tooling.tools.web3_utils import (
     unwrap_generic_value,
 )
 
-SAFE_COLLATERAL_TOKENS = (
+SAFE_COLLATERAL_TOKENS = [
     WrappedxDaiContract(),
     sDaiContract(),
     GNOContract(),
@@ -62,10 +62,10 @@ SAFE_COLLATERAL_TOKENS = (
     SAFEContract(),
     COWContract(),
     MetriSuperGroup(),
-)
-SAFE_COLLATERAL_TOKENS_ADDRESSES = tuple(
+]
+SAFE_COLLATERAL_TOKENS_ADDRESSES = [
     contract.address for contract in SAFE_COLLATERAL_TOKENS
-)
+]
 
 
 class OmenSubgraphHandler(BaseSubgraphHandler):
@@ -237,7 +237,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
         liquidity_bigger_than: Wei | None,
         condition_id_in: list[HexBytes] | None,
         id_in: list[str] | None,
-        collateral_token_address_in: tuple[ChecksumAddress, ...] | None,
+        collateral_token_address_in: t.Sequence[ChecksumAddress] | None,
         category: str | None,
         include_categorical_markets: bool = False,
         include_scalar_markets: bool = False,
@@ -348,7 +348,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
         created_after: DatetimeUTC | None = None,
         excluded_questions: set[str] | None = None,  # question titles
         collateral_token_address_in: (
-            tuple[ChecksumAddress, ...] | None
+            t.Sequence[ChecksumAddress] | None
         ) = SAFE_COLLATERAL_TOKENS_ADDRESSES,
         category: str | None = None,
         creator_in: t.Sequence[HexAddress] | None = None,
@@ -416,7 +416,7 @@ class OmenSubgraphHandler(BaseSubgraphHandler):
         sort_by_field: FieldPath | None = None,
         sort_direction: str | None = None,
         collateral_token_address_in: (
-            tuple[ChecksumAddress, ...] | None
+            t.Sequence[ChecksumAddress] | None
         ) = SAFE_COLLATERAL_TOKENS_ADDRESSES,
         category: str | None = None,
         include_categorical_markets: bool = True,

{prediction_market_agent_tooling-0.66.2 → prediction_market_agent_tooling-0.66.3}/pyproject.toml

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