allow less then max hint queries

ollmer · ollmer · commit bb15454cd33a · 2025-09-24T20:24:32.000+02:00
diff --git a/src/agentlab/agents/generic_agent_hinter/generic_agent.py b/src/agentlab/agents/generic_agent_hinter/generic_agent.py
@@ -200,7 +200,7 @@ def _get_queries(self):
         )
 
         queries = ans_dict.get("queries", [])
-        assert len(queries) == self.flags.n_retrieval_queries
+        assert len(queries) <= self.flags.n_retrieval_queries
 
         # TODO: we should probably propagate these chat_messages to be able to see them in xray
         return queries, ans_dict.get("think", None)

Original file line number	Diff line number	Diff line change
`@@ -200,7 +200,7 @@ def _get_queries(self):`
`200`	`200`	`)`
`201`	`201`
`202`	`202`	`queries = ans_dict.get("queries", [])`
`203`		`- assert len(queries) == self.flags.n_retrieval_queries`
	`203`	`+ assert len(queries) <= self.flags.n_retrieval_queries`
`204`	`204`
`205`	`205`	`# TODO: we should probably propagate these chat_messages to be able to see them in xray`
`206`	`206`	`return queries, ans_dict.get("think", None)`