diff --git a/prediction_market_agent/agents/microchain_agent/deploy.py b/prediction_market_agent/agents/microchain_agent/deploy.py index 3618e507..82b4a1ba 100644 --- 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/prediction_market_agent/agents/microchain_agent/nft_treasury_game/deploy_nft_treasury_game.py b/prediction_market_agent/agents/microchain_agent/nft_treasury_game/deploy_nft_treasury_game.py index f9015ae0..56a45df3 100644 --- a/prediction_market_agent/agents/microchain_agent/nft_treasury_game/deploy_nft_treasury_game.py +++ b/prediction_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,