Bump to 0.24.0 PMAT (#143)

gnosis · May 7, 2024 · dd79d47 · dd79d47
1 parent 2c83bb7
commit dd79d47
Show file tree

Hide file tree

Showing 9 changed files with 76 additions and 75 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/prediction_market_agent/agents/coinflip_agent/deploy.py b/prediction_market_agent/agents/coinflip_agent/deploy.py
@@ -3,13 +3,13 @@
 
 from prediction_market_agent_tooling.deploy.agent import (
     Answer,
-    DeployableAgent,
+    DeployableTraderAgent,
     Probability,
 )
 from prediction_market_agent_tooling.markets.agent_market import AgentMarket
 
 
-class DeployableCoinFlipAgent(DeployableAgent):
+class DeployableCoinFlipAgent(DeployableTraderAgent):
     def pick_markets(self, markets: t.Sequence[AgentMarket]) -> t.Sequence[AgentMarket]:
         return random.sample(markets, 1)
 

diff --git a/prediction_market_agent/agents/known_outcome_agent/deploy.py b/prediction_market_agent/agents/known_outcome_agent/deploy.py
@@ -1,7 +1,7 @@
 import random
 import typing as t
 
-from prediction_market_agent_tooling.deploy.agent import Answer, DeployableAgent
+from prediction_market_agent_tooling.deploy.agent import Answer, DeployableTraderAgent
 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 BetAmount
@@ -15,7 +15,7 @@
 from prediction_market_agent.agents.utils import market_is_saturated
 
 
-class DeployableKnownOutcomeAgent(DeployableAgent):
+class DeployableKnownOutcomeAgent(DeployableTraderAgent):
     model = "gpt-4-1106-preview"
     min_liquidity = 5
 

diff --git a/prediction_market_agent/agents/mech_agent/deploy.py b/prediction_market_agent/agents/mech_agent/deploy.py
@@ -1,7 +1,7 @@
 import typing as t
 
 from prediction_market_agent_tooling.benchmark.utils import OutcomePrediction
-from prediction_market_agent_tooling.deploy.agent import DeployableAgent
+from prediction_market_agent_tooling.deploy.agent import DeployableTraderAgent
 from prediction_market_agent_tooling.markets.agent_market import AgentMarket
 
 from prediction_market_agent.tools.mech.utils import (
@@ -12,7 +12,7 @@
 )
 
 
-class DeployableMechAgentBase(DeployableAgent):
+class DeployableMechAgentBase(DeployableTraderAgent):
     bet_on_n_markets_per_run: int = 5
 
     def load(self) -> None:

diff --git a/prediction_market_agent/agents/microchain_agent/deploy.py b/prediction_market_agent/agents/microchain_agent/deploy.py
@@ -9,7 +9,7 @@ class DeployableMicrochainAgent(DeployableAgent):
     model = "gpt-4-1106-preview"
     n_iterations = 50
 
-    def run(self, market_type: MarketType, _place_bet: bool = True) -> None:
+    def run(self, market_type: MarketType) -> None:
         """
         Override main 'run' method, as the all logic from the helper methods
         is handed over to the agent.

diff --git a/prediction_market_agent/agents/replicate_to_omen_agent/deploy.py b/prediction_market_agent/agents/replicate_to_omen_agent/deploy.py
@@ -31,9 +31,7 @@ class ReplicateSettings(BaseSettings):
 
 
 class DeployableReplicateToOmenAgent(DeployableAgent):
-    def run(
-        self, market_type: MarketType = MarketType.MANIFOLD, _place_bet: bool = True
-    ) -> None:
+    def run(self, market_type: MarketType = MarketType.MANIFOLD) -> None:
         if market_type != MarketType.OMEN:
             raise RuntimeError("Can replicate only into Omen.")
 

diff --git a/prediction_market_agent/agents/think_thoroughly_agent/deploy.py b/prediction_market_agent/agents/think_thoroughly_agent/deploy.py
@@ -1,4 +1,4 @@
-from prediction_market_agent_tooling.deploy.agent import Answer, DeployableAgent
+from prediction_market_agent_tooling.deploy.agent import Answer, DeployableTraderAgent
 from prediction_market_agent_tooling.markets.agent_market import AgentMarket
 from prediction_market_agent_tooling.markets.markets import MarketType
 
@@ -7,7 +7,7 @@
 )
 
 
-class DeployableThinkThoroughlyAgent(DeployableAgent):
+class DeployableThinkThoroughlyAgent(DeployableTraderAgent):
     model: str = "gpt-4-turbo-2024-04-09"
     bet_on_n_markets_per_run = 1
 
@@ -19,7 +19,9 @@ def answer_binary_market(self, market: AgentMarket) -> Answer | None:
 
 
 if __name__ == "__main__":
-    agent = DeployableThinkThoroughlyAgent()
+    agent = DeployableThinkThoroughlyAgent(place_bet=False)
     agent.deploy_local(
-        market_type=MarketType.OMEN, sleep_time=540, timeout=180, place_bet=False
+        market_type=MarketType.OMEN,
+        sleep_time=540,
+        timeout=180,
     )
diff --git a/prediction_market_agent/run_agent.py b/prediction_market_agent/run_agent.py
@@ -1,6 +1,6 @@
 """
 Entrypoint for running the agent in GKE.
-If the agent adheres to PMAT standard (subclasses DeployableAgent), 
+If the agent adheres to PMAT standard (subclasses deployable agent), 
 simply add the agent to the `RunnableAgent` enum and then `RUNNABLE_AGENTS` dict.
 
 Can also be executed locally, simply by running `python prediction_market_agent/run_agent.py <agent> <market_type>`.

diff --git a/pyproject.toml b/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 = { version = "^0.23.0", extras = ["langchain", "google", "mech-client"] }
+prediction-market-agent-tooling = { version = "^0.24.0", extras = ["langchain", "google", "mech-client"] }
 pydantic-settings = "^2.1.0"
 autoflake = "^2.2.1"
 isort = "^5.13.2"