Method generate_ideas

internagent/stage.py:283–384

Run MAS to generate ideas

(self)

    
    
    async def generate_ideas(self):
        """Run MAS to generate ideas"""
        if self.session_id is None:
            await self.load_task()

        async def status_callback(session_id, old_state, new_state):
            # Agent transition logging is handled by OrchestrationAgent
            pass
        
        while self.status != "completed":
            try:
                full_status = await self.interface.get_session_status(self.session_id)
                self.status = full_status['state']
                iterations = full_status['iterations_completed']
                
                if self.status == "awaiting_feedback":
                    if self.args.offline_feedback:
                        with open(self.args.offline_feedback, "r") as f:
                            feedback = json.load(f)
                        await self.interface.add_feedback(self.session_id, feedback)
                        self.logger.info(f"Feedback added: {feedback}")
                
                elif self.status == "completed":
                    self.logger.info("Idea generation completed")
                    break
                
                elif self.status == "error":
                    raise RuntimeError("Error in MAS session")
                
                self.logger.info(f"Running session {self.session_id}, iteration {iterations}")
                self.status = await self.interface.run_session(
                    self.session_id,
                    status_callback=status_callback
                )
                
            except Exception as e:
                self.logger.error(f"Error in session: {str(e)}")
                raise
        
        top_ideas = await self.interface.get_top_ideas(self.session_id)
        self.logger.info(f"Generated {len(top_ideas)} top ideas")

        # Add ideas to IdeaGraph (Long Memory)
        if self.idea_graph is not None:
            try:
                # First, load all historical ideas to build the complete graph
                self.logger.info("Loading historical ideas into IdeaGraph...")
                self._load_historical_ideas_to_graph()
                # Get graph statistics
                stats = self.idea_graph.get_graph_stats()
                self.logger.info(f"IdeaGraph stats: {stats['num_nodes']} nodes, {stats['num_edges']} edges")

                # Perform clustering if enough ideas
                if stats['num_nodes'] >= 3:
                    self.logger.info("Clustering ideas in graph...")
                    self.idea_graph.cluster_ideas(method="louvain")

                    # Log cluster summary