Bump PMAT version to 0.9.3, simplify calculate_bet_amount for Known…

…OutcomeAgent (#51)

* Bump PMAT version to 0.9.3, simplify

* Update poetry.lock

* Skip markets with small liquidity

prediction_market_agent/agents/known_outcome_agent/deploy.py

@@ -7,14 +7,13 @@
 from prediction_market_agent_tooling.deploy.constants import OWNER_KEY
 from prediction_market_agent_tooling.gtypes import SecretStr, private_key_type
 from prediction_market_agent_tooling.markets.agent_market import AgentMarket
-from prediction_market_agent_tooling.markets.data_models import BetAmount, Currency
+from prediction_market_agent_tooling.markets.data_models import BetAmount
 from prediction_market_agent_tooling.markets.markets import MarketType
 from prediction_market_agent_tooling.markets.omen.omen import OmenAgentMarket
 from prediction_market_agent_tooling.tools.utils import (
     check_not_none,
     get_current_git_commit_sha,
     get_current_git_url,
-    should_not_happen,
 )
 
 from prediction_market_agent.agents.known_outcome_agent.known_outcome_agent import (
@@ -28,14 +27,19 @@ def market_is_saturated(market: AgentMarket) -> bool:
 
 
 class DeployableKnownOutcomeAgent(DeployableAgent):
-    model = "gpt-4-1106-preview"
+    model = "gpt-4-turbo-preview"
 
     def load(self) -> None:
         self.markets_with_known_outcomes: dict[str, Result] = {}
 
     def pick_markets(self, markets: list[AgentMarket]) -> list[AgentMarket]:
         picked_markets: list[AgentMarket] = []
         for market in markets:
+            if not isinstance(market, OmenAgentMarket):
+                raise NotImplementedError(
+                    "This agent only supports predictions on Omen markets"
+                )
+
             print(f"Looking at market {market.id=} {market.question=}")
 
             # Assume very high probability markets are already known, and have
@@ -45,6 +49,10 @@ def pick_markets(self, markets: list[AgentMarket]) -> list[AgentMarket]:
                 print(
                     f"Skipping market {market.id=} {market.question=}, because it is already saturated."
                 )
+            elif market.get_liquidity_in_xdai() < 5:
+                print(
+                    f"Skipping market {market.id=} {market.question=}, because it has insufficient liquidity."
+                )
             else:
                 picked_markets.append(market)
 
@@ -83,17 +91,7 @@ def answer_binary_market(self, market: AgentMarket) -> bool | None:
 
     def calculate_bet_amount(self, answer: bool, market: AgentMarket) -> BetAmount:
         if isinstance(market, OmenAgentMarket):
-            if market.currency != Currency.xDai:
-                should_not_happen()
-            return BetAmount(
-                # On markets without liquidity, bet just a small amount for benchmarking.
-                amount=(
-                    Decimal(1.0)
-                    if market.get_liquidity_in_xdai() > 5
-                    else market.get_tiny_bet_amount().amount
-                ),
-                currency=Currency.xDai,
-            )
+            return BetAmount(amount=(Decimal(1.0)), currency=market.currency)
         else:
             raise NotImplementedError("This agent only supports xDai markets")
 

prediction_market_agent/agents/known_outcome_agent/known_outcome_agent.py

@@ -216,7 +216,7 @@ def get_known_outcome(model: str, question: str, max_tries: int) -> Answer:
         ).format_messages(date_str=date_str, question=question)
         print(f"Invoking LLM for {search_prompt=}")
         search_query = str(llm.invoke(search_prompt).content).strip('"')
-        print(f"Searchig for {search_query=}")
+        print(f"Searching for {search_query=}")
         search_results = web_search(query=search_query, max_results=5)
         if not search_results:
             raise ValueError("No search results found.")

pyproject.toml

@@ -35,7 +35,7 @@ poetry = "^1.7.1"
 poetry-plugin-export = "^1.6.0"
 functions-framework = "^3.5.0"
 cron-validator = "^1.0.8"
-prediction-market-agent-tooling = "^0.9.1"
+prediction-market-agent-tooling = "^0.9.3"
 pydantic-settings = "^2.1.0"
 autoflake = "^2.2.1"
 isort = "^5.13.2"