prediction-market-agent-tooling 0.23.0__tar.gz → 0.25.0__tar.gz

{prediction_market_agent_tooling-0.23.0 → prediction_market_agent_tooling-0.25.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: prediction-market-agent-tooling
-Version: 0.23.0
+Version: 0.25.0
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.12
@@ -9,7 +9,6 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Provides-Extra: google
 Provides-Extra: langchain
-Provides-Extra: mech-client
 Requires-Dist: autoflake (>=2.2.1,<3.0.0)
 Requires-Dist: cron-validator (>=1.0.8,<2.0.0)
 Requires-Dist: eth-account (>=0.8.0,<0.9.0)
@@ -25,7 +24,6 @@ Requires-Dist: langchain-community (>=0.0.19)
 Requires-Dist: langchain-openai (>=0.0.5,<0.0.6) ; extra == "langchain"
 Requires-Dist: langfuse (>=2.27.1,<3.0.0)
 Requires-Dist: loguru (>=0.7.2,<0.8.0)
-Requires-Dist: mech-client (>=0.2.13,<0.3.0) ; extra == "mech-client"
 Requires-Dist: numpy (>=1.26.4,<2.0.0)
 Requires-Dist: prompt-toolkit (>=3.0.43,<4.0.0)
 Requires-Dist: pydantic (>=2.6.1,<3.0.0)

{prediction_market_agent_tooling-0.23.0 → prediction_market_agent_tooling-0.25.0}/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-0.23.0 → prediction_market_agent_tooling-0.25.0}/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 → prediction_market_agent_tooling-0.25.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "prediction-market-agent-tooling"
-version = "0.23.0"
+version = "0.25.0"
 description = "Tools to benchmark, deploy and monitor prediction market agents."
 authors = ["Gnosis"]
 readme = "README.md"
@@ -40,13 +40,11 @@ safe-eth-py = "^6.0.0b14"
 eth-account = "^0.8.0"
 prompt-toolkit = "^3.0.43"
 safe-cli = "^1.0.0"
-mech-client = { version = "^0.2.13", optional = true }
 langfuse = "^2.27.1"
 
 [tool.poetry.extras]
 langchain = ["langchain", "langchain-openai"]
 google = ["google-api-python-client"]
-mech-client = ["mech-client"]
 
 [tool.poetry.group.dev.dependencies]
 pytest = "*"