prediction-market-agent-tooling 0.23.0__py3-none-any.whl → 0.24.0__py3-none-any.whl

prediction_market_agent_tooling/deploy/agent.py

@@ -84,62 +84,30 @@ class Answer(BaseModel):
 
 
 class DeployableAgent:
-    bet_on_n_markets_per_run: int = 1
-
     def __init__(self) -> None:
         self.langfuse_wrapper = LangfuseWrapper(agent_name=self.__class__.__name__)
         self.load()
 
     def __init_subclass__(cls, **kwargs: t.Any) -> None:
-        if cls.__init__ is not DeployableAgent.__init__:
+        if "DeployableAgent" not in str(
+            cls.__init__
+        ) and "DeployableTraderAgent" not in str(cls.__init__):
             raise TypeError(
-                "Cannot override __init__ method of DeployableAgent class, please override the `load` method to set up the agent."
+                "Cannot override __init__ method of deployable agent class, please override the `load` method to set up the agent."
             )
 
     def load(self) -> None:
         pass
 
-    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 pick_markets(self, markets: t.Sequence[AgentMarket]) -> t.Sequence[AgentMarket]:
-        """
-        Subclasses can implement their own logic instead of this one, or on top of this one.
-        By default, it picks only the first {n_markets_per_run} markets where user didn't bet recently and it's a reasonable question.
-        """
-        picked: list[AgentMarket] = []
-
-        for market in markets:
-            if len(picked) >= self.bet_on_n_markets_per_run:
-                break
-
-            if self.have_bet_on_market_since(market, since=timedelta(hours=24)):
-                continue
-
-            # Do as a last check, as it uses paid OpenAI API.
-            if not is_predictable_binary(market.question):
-                continue
-
-            picked.append(market)
-
-        return picked
-
-    def answer_binary_market(self, market: AgentMarket) -> Answer | None:
-        """
-        Answer the binary market. This method must be implemented by the subclass.
-        """
-        raise NotImplementedError("This method must be implemented by the subclass")
-
     def deploy_local(
         self,
         market_type: MarketType,
         sleep_time: float,
         timeout: float,
-        place_bet: bool,
     ) -> None:
         start_time = time.time()
         while True:
-            self.run(market_type=market_type, _place_bet=place_bet)
+            self.run(market_type=market_type)
             time.sleep(sleep_time)
             if time.time() - start_time > timeout:
                 break
@@ -223,6 +191,51 @@ def {entrypoint_function_name}(request) -> str:
         if cron_schedule:
             schedule_deployed_gcp_function(fname, cron_schedule=cron_schedule)
 
+    def run(self, market_type: MarketType) -> None:
+        raise NotImplementedError("This method must be implemented by the subclass.")
+
+    def get_gcloud_fname(self, market_type: MarketType) -> str:
+        return f"{self.__class__.__name__.lower()}-{market_type}-{datetime.now().strftime('%Y-%m-%d--%H-%M-%S')}"
+
+
+class DeployableTraderAgent(DeployableAgent):
+    bet_on_n_markets_per_run: int = 1
+
+    def __init__(self, place_bet: bool = True) -> None:
+        super().__init__()
+        self.place_bet = place_bet
+
+    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 pick_markets(self, markets: t.Sequence[AgentMarket]) -> t.Sequence[AgentMarket]:
+        """
+        Subclasses can implement their own logic instead of this one, or on top of this one.
+        By default, it picks only the first {n_markets_per_run} markets where user didn't bet recently and it's a reasonable question.
+        """
+        picked: list[AgentMarket] = []
+
+        for market in markets:
+            if len(picked) >= self.bet_on_n_markets_per_run:
+                break
+
+            if self.have_bet_on_market_since(market, since=timedelta(hours=24)):
+                continue
+
+            # Do as a last check, as it uses paid OpenAI API.
+            if not is_predictable_binary(market.question):
+                continue
+
+            picked.append(market)
+
+        return picked
+
+    def answer_binary_market(self, market: AgentMarket) -> Answer | None:
+        """
+        Answer the binary market. This method must be implemented by the subclass.
+        """
+        raise NotImplementedError("This method must be implemented by the subclass")
+
     def calculate_bet_amount(self, answer: Answer, market: AgentMarket) -> BetAmount:
         """
         Calculate the bet amount. By default, it returns the minimum bet amount.
@@ -253,7 +266,7 @@ def {entrypoint_function_name}(request) -> str:
             # Omen is specific, because the user (agent) needs to manually withdraw winnings from the market.
             redeem_from_all_user_positions(private_credentials)
 
-    def process_bets(self, market_type: MarketType, _place_bet: bool = True) -> None:
+    def process_bets(self, market_type: MarketType) -> None:
         """
         Processes bets placed by agents on a given market.
         """
@@ -264,7 +277,7 @@ def {entrypoint_function_name}(request) -> str:
             if result is None:
                 logger.debug(f"Skipping market {market} as no answer was provided")
                 continue
-            if _place_bet:
+            if self.place_bet:
                 amount = self.calculate_bet_amount(result, market)
                 logger.debug(
                     f"Placing bet on {market} with result {result} and amount {amount}"
@@ -277,10 +290,7 @@ def {entrypoint_function_name}(request) -> str:
     def after(self, market_type: MarketType) -> None:
         pass
 
-    def run(self, market_type: MarketType, _place_bet: bool = True) -> None:
+    def run(self, market_type: MarketType) -> None:
         self.before(market_type)
-        self.process_bets(market_type, _place_bet)
+        self.process_bets(market_type)
         self.after(market_type)
-
-    def get_gcloud_fname(self, market_type: MarketType) -> str:
-        return f"{self.__class__.__name__.lower()}-{market_type}-{datetime.now().strftime('%Y-%m-%d--%H-%M-%S')}"

prediction_market_agent_tooling/deploy/agent_example.py

@@ -3,13 +3,13 @@ import typing as t
 
 from prediction_market_agent_tooling.deploy.agent import (
     Answer,
-    DeployableAgent,
+    DeployableTraderAgent,
     Probability,
 )
 from prediction_market_agent_tooling.markets.agent_market import AgentMarket
 
 
-class DeployableCoinFlipAgent(DeployableAgent):
+class DeployableCoinFlipAgent(DeployableTraderAgent):
     def pick_markets(self, markets: t.Sequence[AgentMarket]) -> t.Sequence[AgentMarket]:
         return random.sample(markets, 1)
 
@@ -23,6 +23,6 @@ class DeployableCoinFlipAgent(DeployableAgent):
         )
 
 
-class DeployableAlwaysRaiseAgent(DeployableAgent):
+class DeployableAlwaysRaiseAgent(DeployableTraderAgent):
     def answer_binary_market(self, market: AgentMarket) -> Answer | None:
         raise RuntimeError("I always raise!")

{prediction_market_agent_tooling-0.23.0.dist-info → prediction_market_agent_tooling-0.24.0.dist-info}/METADATA

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

{prediction_market_agent_tooling-0.23.0.dist-info → prediction_market_agent_tooling-0.24.0.dist-info}/RECORD

@@ -12,8 +12,8 @@ prediction_market_agent_tooling/benchmark/agents.py,sha256=HPIFJvackW110ch3Ukktb
 prediction_market_agent_tooling/benchmark/benchmark.py,sha256=xiHKzZx5GHSsDerFHMZ9j_LXAXnSaITSvv67iPe3MEU,21095
 prediction_market_agent_tooling/benchmark/utils.py,sha256=iS1BzyXcCMfMm1Rx--1QCH0pHvBTblTndcDQFbTUJ2s,2897
 prediction_market_agent_tooling/config.py,sha256=FI8feNLmA0QYmN-llvyDh3c6tN2ZGl7Eerw1e27v51A,4757
-prediction_market_agent_tooling/deploy/agent.py,sha256=OxSL1w51GumyBUxN2cx6VPgvf8NWsq3-7wAy9g3hsFA,10018
-prediction_market_agent_tooling/deploy/agent_example.py,sha256=KaTJm43cwbBZR2BmTEyB1ohChAqwYlBUqaaVumsWQe4,891
+prediction_market_agent_tooling/deploy/agent.py,sha256=M6qSZ1mN0AEwyix3WJal7ulifI0o-v89S2K_s_zTBbs,10292
+prediction_market_agent_tooling/deploy/agent_example.py,sha256=tqXVA2HSFK3pdZ49tMmta8aKdJFT8JN8WeJ1akjrpBk,909
 prediction_market_agent_tooling/deploy/constants.py,sha256=M5ty8URipYMGe_G-RzxRydK3AFL6CyvmqCraJUrLBnE,82
 prediction_market_agent_tooling/deploy/gcp/deploy.py,sha256=CYUgnfy-9XVk04kkxA_5yp0GE9Mw5caYqlFUZQ2j3ks,3739
 prediction_market_agent_tooling/deploy/gcp/kubernetes_models.py,sha256=qYIHRxQLac3yxtZ8ChikiPG9O1aUQucHW0muTSm1nto,2627
@@ -65,8 +65,8 @@ prediction_market_agent_tooling/tools/safe.py,sha256=h0xOO0eNtitClf0fPkn-0oTc6A_
 prediction_market_agent_tooling/tools/singleton.py,sha256=CiIELUiI-OeS7U7eeHEt0rnVhtQGzwoUdAgn_7u_GBM,729
 prediction_market_agent_tooling/tools/utils.py,sha256=zkmwPi3YisgZDPCeNwaRbL8sInOYOkvFgFY4Q8PbEo4,5077
 prediction_market_agent_tooling/tools/web3_utils.py,sha256=cboATXNmEdn5RmPbVblHOwOdUMKBYrUK3GiI6i6Vzxo,9855
-prediction_market_agent_tooling-0.23.0.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.23.0.dist-info/METADATA,sha256=gkLdQmTB2vbEO3toPUFru7h7BqsIvLm50aXGbNDK1xM,5563
-prediction_market_agent_tooling-0.23.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-prediction_market_agent_tooling-0.23.0.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.23.0.dist-info/RECORD,,
+prediction_market_agent_tooling-0.24.0.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.24.0.dist-info/METADATA,sha256=k0OnkcYQbBhcrWsBS3hX8LjHzY_j3SJ3PHhEHgiIkJ4,5563
+prediction_market_agent_tooling-0.24.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+prediction_market_agent_tooling-0.24.0.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.24.0.dist-info/RECORD,,