File size: 825 Bytes
f9e5853
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
from crewai import Agent, Task

class ReportGeneratorAgent(Agent):
    def __init__(self, llm):
        super().__init__(
            llm=llm,
            role="report generator",
            backstory="I generate a final comprehensive report without a bias",
            goal="Create a detailed report summarizing the research findings."
        )

    def execute_task(self, task: Task, context: dict = None, tools: list = None):
        """Generate a comprehensive report."""
        
        summaries =  context
        report = self.generate_report(summaries)
        return report

    def generate_report(self, summaries):
        """Generate a comprehensive research report."""
        return self.llm.call([
                        {"role": "user", "content": summaries}
                    ])