Method load_task

internagent/stage.py:178–280

Load task and create MAS session

(self)

            )

    async def load_task(self):
        """Load task and create MAS session"""
        self.logger.info(f"Creating research session for: {self.args.task_dir}")

        await self.interface.startup()

        # Use args.prompt_path if available (points to launch directory copy), otherwise fall back to task_dir
        task_desc_path = getattr(self.args, 'prompt_path', None) or osp.join(self.args.task_dir, "prompt.json")
        if not osp.exists(task_desc_path):
            raise FileNotFoundError(f"Task description not found: {task_desc_path}")
        self.logger.info(f"Using prompt file: {task_desc_path}")

        # Check if experience library exists and evolve prompt if it does
        # Use base_output_dir if available (set by launch_discovery.py), otherwise use default
        base_output_dir = getattr(self.args, 'base_output_dir', None) or osp.join("results", self.args.task_name)
        library_path = osp.join(base_output_dir, "experience_library.json")

        # Get evolution_interval from config (default: 1, meaning evolve every round)
        evolution_interval = self.config.get('memory', {}).get('long_memory', {}).get('prompt_evolver', {}).get('evolution_interval', 1)

        # Only evolve prompt when:
        # 1. round_num > 1 (first round has no experience yet)
        # 2. round_num % evolution_interval == 0 (respects the configured interval)
        # Note: With interval=1, evolves at rounds 2, 3, 4, ...
        # With interval=2, evolves at rounds 2, 4, 6, ...
        should_evolve = (self.round_num > 1) and ((self.round_num - 1) % evolution_interval == 0)

        if should_evolve:
            self.logger.info(f"Round {self.round_num}: Prompt evolution scheduled (interval={evolution_interval})")
        else:
            self.logger.info(f"Round {self.round_num}: Skipping prompt evolution (interval={evolution_interval})")

        if should_evolve and osp.exists(library_path) and LONG_MEMORY_AVAILABLE and PromptEvolver is not None:
            self.logger.info("Experience library found, evolving prompt...")
            try:
                # Create a backup of the prompt before evolution (in the same directory as prompt file)
                prompt_dir = osp.dirname(task_desc_path)
                backup_path = osp.join(prompt_dir, f"prompt_backup_round{self.round_num}.json")
                import shutil
                shutil.copy(task_desc_path, backup_path)
                self.logger.info(f"Backed up prompt to: {backup_path}")

                # Load config for PromptEvolver
                config = {}
                if self.args.config and osp.exists(self.args.config):
                    try:
                        with open(self.args.config, 'r') as f:
                            if self.args.config.endswith(('.yaml', '.yml')):
                                config = yaml.safe_load(f)
                            else:
                                config = json.load(f)
                    except Exception as e:
                        self.logger.warning(f"Failed to load config: {e}")

                # Initialize PromptEvolver with IdeaGraph
                prompt_evolver = PromptEvolver(self.args, self.logger, config, idea_graph=self.idea_graph)

                # Get domain from current prompt