Add think thoroughly agent to run_agent.py (#72)

gnosis · Apr 12, 2024 · b753f9a · b753f9a
1 parent 9f76096
commit b753f9a
Show file tree

Hide file tree

Showing 6 changed files with 8 additions and 8 deletions.
diff --git a/prediction_market_agent/agents/known_outcome_agent/benchmark.py b/prediction_market_agent/agents/known_outcome_agent/benchmark.py
@@ -1,4 +1,3 @@
-# type: ignore
 import time
 import typing as t
 from datetime import timedelta

diff --git a/...s/crewai_subsequential_agent/benchmark.py → ...gents/think_thoroughly_agent/benchmark.py b/...s/crewai_subsequential_agent/benchmark.py → ...gents/think_thoroughly_agent/benchmark.py
@@ -25,7 +25,7 @@
 )
 from prediction_market_agent_tooling.tools.utils import utcnow
 
-from prediction_market_agent.agents.crewai_subsequential_agent.crewai_agent_subquestions import (
+from prediction_market_agent.agents.think_thoroughly_agent.think_thoroughly_agent import (
     CrewAIAgentSubquestions,
 )
 

diff --git a/...ents/crewai_subsequential_agent/deploy.py → ...t/agents/think_thoroughly_agent/deploy.py b/...ents/crewai_subsequential_agent/deploy.py → ...t/agents/think_thoroughly_agent/deploy.py
@@ -6,17 +6,13 @@
 from prediction_market_agent_tooling.markets.markets import MarketType
 from prediction_market_agent_tooling.tools.utils import should_not_happen
 
-from prediction_market_agent.agents.crewai_subsequential_agent.crewai_agent_subquestions import (
+from prediction_market_agent.agents.think_thoroughly_agent.think_thoroughly_agent import (
     CrewAIAgentSubquestions,
 )
 from prediction_market_agent.agents.utils import market_is_saturated
 
 
 class DeployableThinkThoroughlyAgent(DeployableAgent):
-    # For cheaper credits at this experimental stage
-    def __init__(self) -> None:
-        super().__init__()
-
     def pick_markets(self, markets: t.Sequence[AgentMarket]) -> t.Sequence[AgentMarket]:
         # We simply pick 5 random markets to bet on
         picked_markets: list[AgentMarket] = []

diff --git a/...nts/crewai_subsequential_agent/prompts.py → .../agents/think_thoroughly_agent/prompts.py b/...nts/crewai_subsequential_agent/prompts.py → .../agents/think_thoroughly_agent/prompts.py
diff --git a/...ential_agent/crewai_agent_subquestions.py → ...horoughly_agent/think_thoroughly_agent.py b/...ential_agent/crewai_agent_subquestions.py → ...horoughly_agent/think_thoroughly_agent.py
@@ -6,7 +6,7 @@
 from loguru import logger
 from pydantic import BaseModel
 
-from prediction_market_agent.agents.crewai_subsequential_agent.prompts import (
+from prediction_market_agent.agents.think_thoroughly_agent.prompts import (
     CREATE_OUTCOMES_FROM_SCENARIO_OUTPUT,
     CREATE_OUTCOMES_FROM_SCENARIO_PROMPT,
     FINAL_DECISION_PROMPT,

diff --git a/prediction_market_agent/run_agent.py b/prediction_market_agent/run_agent.py
@@ -15,15 +15,20 @@
 from prediction_market_agent.agents.known_outcome_agent.deploy import (
     DeployableKnownOutcomeAgent,
 )
+from prediction_market_agent.agents.think_thoroughly_agent.deploy import (
+    DeployableThinkThoroughlyAgent,
+)
 
 
 class RunnableAgent(str, Enum):
     coinflip = "coinflip"
+    think_thoroughly = "think_thoroughly"
     knownoutcome = "knownoutcome"
 
 
 RUNNABLE_AGENTS = {
     RunnableAgent.coinflip: DeployableCoinFlipAgent,
+    RunnableAgent.think_thoroughly: DeployableThinkThoroughlyAgent,
     RunnableAgent.knownoutcome: DeployableKnownOutcomeAgent,
 }