Don't allow Stop in NFT agents (#659)

gnosis · Jan 24, 2025 · 506d4d1 · 506d4d1
1 parent 7883045
commit 506d4d1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/prediction_market_agent/agents/microchain_agent/deploy.py b/prediction_market_agent/agents/microchain_agent/deploy.py
@@ -43,6 +43,7 @@ class DeployableMicrochainAgentAbstract(DeployableAgent, metaclass=abc.ABCMeta):
     max_iterations: int | None = 50
     import_actions_from_memory = 0
     sleep_between_iterations = 0
+    allow_stop: bool = True
     identifier: AgentIdentifier
     functions_config: FunctionsConfig
 
@@ -77,7 +78,7 @@ def build_agent(self, market_type: MarketType) -> Agent:
             market_type=market_type,
             model=self.model,
             unformatted_system_prompt=unformatted_system_prompt,
-            allow_stop=True,
+            allow_stop=self.allow_stop,
             long_term_memory=self.long_term_memory,
             keys=APIKeys(),
             functions_config=self.functions_config,

diff --git a/...iction_market_agent/agents/microchain_agent/nft_treasury_game/deploy_nft_treasury_game.py b/...iction_market_agent/agents/microchain_agent/nft_treasury_game/deploy_nft_treasury_game.py
@@ -38,6 +38,7 @@
 class DeployableAgentNFTGameAbstract(DeployableMicrochainAgentAbstract):
     # Agent configuration
     sleep_between_iterations = 15
+    allow_stop = False
     import_actions_from_memory = 100
     functions_config = FunctionsConfig(
         common_functions=True,