diff --git a/prediction_market_agent/agents/known_outcome_agent/deploy.py b/prediction_market_agent/agents/known_outcome_agent/deploy.py index 2d744f0..eb551be 100644 --- 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 index 89b3c90..b6b9405 100644 --- 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 index c903514..b42f96f 100644 --- 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(