More review comments

gnosis · Mar 20, 2024 · e1e232e · e1e232e
1 parent c7a93d0
commit e1e232e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/prediction_market_agent/agents/known_outcome_agent/deploy.py b/prediction_market_agent/agents/known_outcome_agent/deploy.py
@@ -4,6 +4,7 @@
 from prediction_market_agent_tooling.config import APIKeys
 from prediction_market_agent_tooling.deploy.agent import DeployableAgent
 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.markets import MarketType
@@ -66,17 +67,15 @@ def calculate_bet_amount(self, answer: bool, market: AgentMarket) -> BetAmount:
         market_type=MarketType.OMEN,
         labels={OWNER_KEY: getpass.getuser()},
         secrets={
-            "OPENAI_API_KEY": "EVAN_OPENAI_API_KEY:latest",
             "TAVILY_API_KEY": "GNOSIS_AI_TAVILY_API_KEY:latest",
-            "BET_FROM_PRIVATE_KEY": "EVAN_OMEN_BETTER_0_PKEY:latest",
         },
         memory=1024,
         api_keys=APIKeys(
             BET_FROM_ADDRESS=verify_address(
                 "0xb611A9f02B318339049264c7a66ac3401281cc3c"
             ),
-            BET_FROM_PRIVATE_KEY=None,
-            OPENAI_API_KEY=None,
+            BET_FROM_PRIVATE_KEY=private_key_type("EVAN_OMEN_BETTER_0_PKEY:latest"),
+            OPENAI_API_KEY=SecretStr("EVAN_OPENAI_API_KEY:latest"),
             MANIFOLD_API_KEY=None,
         ),
         cron_schedule="0 */4 * * *",

diff --git a/prediction_market_agent/agents/known_outcome_agent/known_outcome_agent.py b/prediction_market_agent/agents/known_outcome_agent/known_outcome_agent.py
@@ -133,7 +133,6 @@ def summarize_if_required(content: str, model: str, question: str) -> str:
         raise ValueError(f"Unknown model: {model}")
 
     if len(content) > max_length:
-        breakpoint()
         return _summary(content=content, objective=question, separators=["  "])
     else:
         return content