Method execute

internagent/mas/agents/dr_agent.py:142–189

Execute discovery research task using the DR workflow. Args: context: Dictionary containing: - task: The research task description - file_path: Optional file path for additional context - goal: Research goal inform

(self, context: Dict[str, Any], params: Dict[str, Any])

        return workflow_config
    
    async def execute(self, context: Dict[str, Any], params: Dict[str, Any]) -> Dict[str, Any]:
        """
        Execute discovery research task using the DR workflow.
        
        Args:
            context: Dictionary containing:
                - task: The research task description
                - file_path: Optional file path for additional context
                - goal: Research goal information (optional)
                - iteration: Current iteration number (optional)
            params: Dictionary containing optional configuration overrides
                
        Returns:
            The workflow execution result (format depends on the workflow)
        """
        # Extract task information from context
        task = context.get("task", "")
        if not task:
            # Try alternative keys
            task = context.get("description", "") or context.get("goal", {}).get("description", "")
        
        if not task:
            raise AgentExecutionError("Task description is required for DR agent execution")
        
        # Extract optional file path
        file_path = context.get("file_path", None)
        
        # Check if workflow is initialized
        if self.workflow is None:
            logger.error("DR workflow is not initialized - cannot execute task")
            # Return a simple fallback response
            return f"Background research needed for: {task}\n\nNote: DR workflow is not available. Please provide background context manually."
        
        try:
            logger.info(f"DR Agent executing task: {task}")
            
            # Execute the DR workflow
            result = self.workflow.execute(task=task, file_path=file_path)
            
            logger.info("DR workflow execution completed successfully")
            
            # Return the result directly (user has modified return format)
            return result
            
        except Exception as e:
            logger.error(f"Error during DR workflow execution: {str(e)}")
            # Return a fallback response instead of raising error
            return f"Background research for: {task}\n\nNote: DR workflow execution failed. Please provide background context manually."