diff --git a/prompting/agent.py b/prompting/agent.py index 65f1bdb5e..143861021 100644 --- a/prompting/agent.py +++ b/prompting/agent.py @@ -27,8 +27,6 @@ from transformers import Pipeline -RETRY_LIMIT = 3 - class HumanAgent(vLLM_LLM): "Agent that impersonates a human user and makes queries based on its goal." @@ -83,14 +81,7 @@ def __init__( if begin_conversation: bt.logging.info("🤖 Generating challenge query...") # initiates the conversation with the miner - for i in range(RETRY_LIMIT): - self.challenge = self.challenge_time() - if not self.challenge: - bt.logging.error("❌ Generated an empty challenge. Retrying...") - else: - break - if not self.challenge: - bt.logging.error("Max retries reached. Skipping task.") + self.challenge = self.create_challenge() def create_challenge(self) -> str: """Creates the opening question of the conversation which is based on the task query but dressed in the persona of the user."""