Create agents.py

agents.py

@@ -0,0 +1,96 @@
+# agents.py
+from crewai import Agent, Crew, Process, Task
+from crewai_tools import SerperDevTool
+
+def create_research_crew(topic: str):
+    """Create the research crew with SerperDev tool integration"""
+    try:
+        # Initialize SerperDev tool
+        search_tool = SerperDevTool()
+
+        # Create agents
+        researcher = Agent(
+            role='Research Analyst',
+            goal=f'Conduct exhaustive market research about {topic} with comprehensive analysis',
+            backstory="""Senior market research analyst with 15+ years of experience in detailed industry analysis.""",
+            tools=[search_tool],
+            verbose=True
+        )
+        
+        analyst = Agent(
+            role='Data Analyst',
+            goal='Transform research data into actionable insights with visualizations',
+            backstory="""Expert data analyst specializing in predictive analytics and trend analysis.""",
+            tools=[search_tool],
+            verbose=True
+        )
+        
+        writer = Agent(
+            role='Report Writer',
+            goal='Create comprehensive market research reports with compelling narratives',
+            backstory="""Professional writer specializing in creating clear, engaging market research reports.""",
+            verbose=True
+        )
+
+        # Create tasks
+        research_task = create_research_task(topic, researcher)
+        analysis_task = create_analysis_task(analyst, research_task)
+        report_task = create_report_task(writer, [research_task, analysis_task])
+
+        return Crew(
+            agents=[researcher, analyst, writer],
+            tasks=[research_task, analysis_task, report_task],
+            verbose=True,
+            process=Process.sequential
+        )
+    except Exception as e:
+        raise Exception(f"Error initializing research crew: {str(e)}")
+
+def create_research_task(topic: str, agent):
+    """Create the research task"""
+    return Task(
+        description=f"""
+        Conduct exhaustive market research on {topic} covering:
+        1. Market Overview (size, growth, segmentation)
+        2. Competitive Analysis (players, shares, positioning)
+        3. Market Dynamics (drivers, challenges, trends)
+        4. Economic Factors (macro influences, indicators)
+        
+        Provide specific numerical data and sources.
+        """,
+        agent=agent,
+        expected_output="Comprehensive research data with verified sources"
+    )
+
+def create_analysis_task(agent, research_task):
+    """Create the analysis task"""
+    return Task(
+        description="""
+        Analyze the research findings providing:
+        1. Strategic Market Analysis
+        2. Advanced Data Analysis
+        3. Competitive Intelligence
+        4. Trend Analysis
+        
+        Include specific metrics and visualization-ready data.
+        """,
+        agent=agent,
+        expected_output="Detailed analysis with visualization-ready metrics",
+        context=[research_task]
+    )
+
+def create_report_task(agent, context_tasks):
+    """Create the report task"""
+    return Task(
+        description="""
+        Create a professional report with:
+        1. Executive Summary (2-3 pages)
+        2. Detailed Analysis (10+ pages)
+        3. Sources and Citations
+        
+        Format as JSON with exec_summary, detailed_report, and metrics sections.
+        """,
+        agent=agent,
+        expected_output="JSON formatted report with executive summary and details",
+        context=context_tasks
+    )