Set default bet size for various deployed agents to '1' (#433)

gnosis · Aug 29, 2024 · 0903d72 · 0903d72
1 parent 3a098b3
commit 0903d72
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 2 deletions.
diff --git a/prediction_market_agent/agents/known_outcome_agent/deploy.py b/prediction_market_agent/agents/known_outcome_agent/deploy.py
@@ -1,4 +1,7 @@
 from prediction_market_agent_tooling.deploy.agent import DeployableTraderAgent
+from prediction_market_agent_tooling.deploy.betting_strategy import (
+    MaxAccuracyBettingStrategy,
+)
 from prediction_market_agent_tooling.loggers import logger
 from prediction_market_agent_tooling.markets.agent_market import AgentMarket
 from prediction_market_agent_tooling.markets.data_models import ProbabilisticAnswer
@@ -15,6 +18,7 @@
 class DeployableKnownOutcomeAgent(DeployableTraderAgent):
     model = "gpt-4-1106-preview"
     min_liquidity = 5
+    strategy = MaxAccuracyBettingStrategy(bet_amount=1)
 
     def load(self) -> None:
         self.markets_with_known_outcomes: dict[str, Result] = {}

diff --git a/prediction_market_agent/agents/prophet_agent/deploy.py b/prediction_market_agent/agents/prophet_agent/deploy.py
@@ -1,5 +1,9 @@
 from prediction_market_agent_tooling.deploy.agent import DeployableTraderAgent
-from prediction_market_agent_tooling.deploy.betting_strategy import KellyBettingStrategy
+from prediction_market_agent_tooling.deploy.betting_strategy import (
+    BettingStrategy,
+    KellyBettingStrategy,
+    MaxAccuracyBettingStrategy,
+)
 from prediction_market_agent_tooling.loggers import logger
 from prediction_market_agent_tooling.markets.agent_market import AgentMarket
 from prediction_market_agent_tooling.markets.data_models import ProbabilisticAnswer
@@ -18,6 +22,7 @@
 class DeployableTraderAgentER(DeployableTraderAgent):
     agent: PredictionProphetAgent | OlasAgent
     bet_on_n_markets_per_run = 1
+    strategy: BettingStrategy = MaxAccuracyBettingStrategy(bet_amount=1)
 
     @property
     def model(self) -> str | None:

diff --git a/prediction_market_agent/agents/think_thoroughly_agent/deploy.py b/prediction_market_agent/agents/think_thoroughly_agent/deploy.py
@@ -1,5 +1,9 @@
 from prediction_market_agent_tooling.deploy.agent import DeployableTraderAgent
-from prediction_market_agent_tooling.deploy.betting_strategy import KellyBettingStrategy
+from prediction_market_agent_tooling.deploy.betting_strategy import (
+    BettingStrategy,
+    KellyBettingStrategy,
+    MaxAccuracyBettingStrategy,
+)
 from prediction_market_agent_tooling.markets.agent_market import AgentMarket
 from prediction_market_agent_tooling.markets.data_models import ProbabilisticAnswer
 from prediction_market_agent_tooling.markets.markets import MarketType
@@ -15,6 +19,7 @@ class DeployableThinkThoroughlyAgentBase(DeployableTraderAgent):
     agent_class: type[ThinkThoroughlyBase]
     model: str
     bet_on_n_markets_per_run = 1
+    strategy: BettingStrategy = MaxAccuracyBettingStrategy(bet_amount=1)
 
     def load(self) -> None:
         self.agent = self.agent_class(