diff --git a/rdagent/components/coder/CoSTEER/__init__.py b/rdagent/components/coder/CoSTEER/__init__.py index 51d30a72..1923119a 100644 --- a/rdagent/components/coder/CoSTEER/__init__.py +++ b/rdagent/components/coder/CoSTEER/__init__.py @@ -17,6 +17,7 @@ from rdagent.core.evolving_agent import EvolvingStrategy, RAGEvoAgent from rdagent.core.exception import CoderError from rdagent.core.experiment import Experiment from rdagent.log import rdagent_logger as logger +from rdagent.oai.backend.base import RD_Agent_TIMER_wrapper class CoSTEER(Developer[Experiment]): @@ -107,6 +108,9 @@ class CoSTEER(Developer[Experiment]): if (datetime.now() - start_datetime).seconds > self.max_seconds: logger.info(f"Reached max time limit {self.max_seconds} seconds, stop evolving") break + if RD_Agent_TIMER_wrapper.timer.started and RD_Agent_TIMER_wrapper.timer.is_timeout(): + logger.info("Global timer is timeout, stop evolving") + break if self.with_feedback and self.filter_final_evo: evo_exp = self._exp_postprocess_by_feedback(evo_exp, self.evolve_agent.evolving_trace[-1].feedback)