prediction-market-agent-tooling 0.47.0__tar.gz → 0.47.1__tar.gz

{prediction_market_agent_tooling-0.47.0 → prediction_market_agent_tooling-0.47.1}/PKG-INFO

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

{prediction_market_agent_tooling-0.47.0 → prediction_market_agent_tooling-0.47.1}/prediction_market_agent_tooling/deploy/agent.py

@@ -216,6 +216,19 @@ class DeployableTraderAgent(DeployableAgent):
     def have_bet_on_market_since(self, market: AgentMarket, since: timedelta) -> bool:
         return have_bet_on_market_since(keys=APIKeys(), market=market, since=since)
 
+    def check_min_required_balance_to_operate(self, market_type: MarketType) -> None:
+        api_keys = APIKeys()
+        if self.min_required_balance_to_operate is None:
+            return
+        if market_type == MarketType.OMEN and not is_minimum_required_balance(
+            api_keys.public_key,
+            min_required_balance=self.min_required_balance_to_operate,
+        ):
+            raise OutOfFundsError(
+                f"Minimum required balance {self.min_required_balance_to_operate} "
+                f"for agent with address {api_keys.public_key} is not met."
+            )
+
     def pick_markets(
         self, market_type: MarketType, markets: t.Sequence[AgentMarket]
     ) -> t.Sequence[AgentMarket]:
@@ -280,15 +293,7 @@ class DeployableTraderAgent(DeployableAgent):
         if market_type == MarketType.OMEN:
             # Omen is specific, because the user (agent) needs to manually withdraw winnings from the market.
             redeem_from_all_user_positions(api_keys)
-            # Check if we have enough of balance to operate.
-            if self.min_required_balance_to_operate is not None:
-                if not is_minimum_required_balance(
-                    api_keys.public_key,
-                    min_required_balance=self.min_required_balance_to_operate,
-                ):
-                    raise OutOfFundsError(
-                        f"Minimum required balance {self.min_required_balance_to_operate} is not met."
-                    )
+            self.check_min_required_balance_to_operate(market_type)
             # Exchange wxdai back to xdai if the balance is getting low, so we can keep paying for fees.
             if self.min_balance_to_keep_in_native_currency is not None:
                 withdraw_wxdai_to_xdai_to_keep_balance(
@@ -304,6 +309,8 @@ class DeployableTraderAgent(DeployableAgent):
         available_markets = self.get_markets(market_type)
         markets = self.pick_markets(market_type, available_markets)
         for market in markets:
+            # We need to check it again before each market bet, as the balance might have changed.
+            self.check_min_required_balance_to_operate(market_type)
             result = self.answer_binary_market(market)
             if result is None:
                 logger.info(f"Skipping market {market} as no answer was provided")

{prediction_market_agent_tooling-0.47.0 → prediction_market_agent_tooling-0.47.1}/pyproject.toml

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