diff --git a/src/agentlab/agents/hitl_agent/generic_human_guided_agent.py b/src/agentlab/agents/hitl_agent/generic_human_guided_agent.py index e8d31688..b29a405c 100644 --- a/src/agentlab/agents/hitl_agent/generic_human_guided_agent.py +++ b/src/agentlab/agents/hitl_agent/generic_human_guided_agent.py @@ -245,7 +245,7 @@ def get_action(self, obs): ) self.ui.update_context(hint_labeling_inputs) - response = self.ui.wait_for_response(timeout=600) + response = self.ui.wait_for_response(timeout=None) if response["type"] == "reprompt": new_hints = response["payload"].get("hints", []) diff --git a/src/agentlab/agents/hitl_agent/hitl_agent.py b/src/agentlab/agents/hitl_agent/hitl_agent.py index 9b84793b..c5fc46da 100644 --- a/src/agentlab/agents/hitl_agent/hitl_agent.py +++ b/src/agentlab/agents/hitl_agent/hitl_agent.py @@ -75,7 +75,7 @@ def get_action(self, obs): ) self.ui.update_context(hint_labeling_inputs) - response = self.ui.wait_for_response(timeout=600) + response = self.ui.wait_for_response(timeout=None) if response["type"] == "reprompt": new_hints = response["payload"].get("hints", [])