From 02166cea128de04cbb6cb53bcc50ebe43a1ca983 Mon Sep 17 00:00:00 2001
From: Evan Griffiths <56087052+evangriffiths@users.noreply.github.com>
Date: Tue, 29 Oct 2024 08:28:37 +0000
Subject: [PATCH] Reduce bet_on_n_markets_per_run for expensive agents (#534)

---
 prediction_market_agent/agents/prophet_agent/deploy.py          | 1 +
 prediction_market_agent/agents/think_thoroughly_agent/deploy.py | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/prediction_market_agent/agents/prophet_agent/deploy.py b/prediction_market_agent/agents/prophet_agent/deploy.py
index a6a0a182..75a5a834 100644
--- a/prediction_market_agent/agents/prophet_agent/deploy.py
+++ b/prediction_market_agent/agents/prophet_agent/deploy.py
@@ -120,6 +120,7 @@ def load(self) -> None:
 
 class DeployablePredictionProphetGPTo1PreviewAgent(DeployableTraderAgentER):
     agent: PredictionProphetAgent
+    bet_on_n_markets_per_run = 2
 
     def get_betting_strategy(self, market: AgentMarket) -> BettingStrategy:
         return KellyBettingStrategy(
diff --git a/prediction_market_agent/agents/think_thoroughly_agent/deploy.py b/prediction_market_agent/agents/think_thoroughly_agent/deploy.py
index 0b0273f8..74ae51a2 100644
--- a/prediction_market_agent/agents/think_thoroughly_agent/deploy.py
+++ b/prediction_market_agent/agents/think_thoroughly_agent/deploy.py
@@ -19,7 +19,7 @@
 class DeployableThinkThoroughlyAgentBase(DeployableTraderAgent):
     agent_class: type[ThinkThoroughlyBase]
     model: str
-    bet_on_n_markets_per_run = 2
+    bet_on_n_markets_per_run = 1
 
     def load(self) -> None:
         self.agent = self.agent_class(