Betting in more markets

gnosis · Oct 11, 2024 · 4a7d6ac · 4a7d6ac
1 parent 55aee24
commit 4a7d6ac
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/prediction_market_agent/agents/arbitrage_agent/data_models.py b/prediction_market_agent/agents/arbitrage_agent/data_models.py
@@ -14,7 +14,7 @@ class CorrelatedMarketPair(BaseModel):
     related_market: AgentMarket
 
     def __repr__(self) -> str:
-        return f"main_market_question {self.main_market.question=} related_market_question {self.related_market.question=}"
+        return f"main_market {self.main_market.question} related_market_question {self.related_market.question} potential profit {self.potential_profit_per_bet_unit}"
 
     @computed_field  # type: ignore[prop-decorator]
     @property

diff --git a/prediction_market_agent/agents/arbitrage_agent/deploy.py b/prediction_market_agent/agents/arbitrage_agent/deploy.py
@@ -48,6 +48,7 @@ class DeployableArbitrageAgent(DeployableTraderAgent):
     model = "gpt-4o"
     # trade amount will be divided between correlated markets.
     total_trade_amount = BetAmount(amount=0.1, currency=OmenAgentMarket.currency)
+    bet_on_n_markets_per_run = 5
 
     def run(self, market_type: MarketType) -> None:
         if market_type != MarketType.OMEN:
@@ -69,7 +70,7 @@ def get_markets(
     ) -> t.Sequence[AgentMarket]:
         return super().get_markets(
             market_type=market_type,
-            limit=limit,
+            limit=50,
             sort_by=SortBy.HIGHEST_LIQUIDITY,
             # Fetching most liquid markets since more likely they will have related markets
             filter_by=FilterBy.OPEN,