Agents & Roles

from crewai import Agent

agent = Agent(
    # Required: What is this agent's job title?
    role="Senior Data Scientist",

    # Required: What is the agent trying to achieve?
    goal="Analyze datasets and extract actionable insights",

    # Required: Background that shapes the agent's behavior
    backstory="""You are a seasoned data scientist with 10 years
    of experience in statistical analysis and machine learning.
    You are known for finding patterns others miss and
    communicating findings clearly to stakeholders.""",

    # Optional configurations
    verbose=True,              # Show reasoning steps
    allow_delegation=True,    # Can delegate to other agents
    max_iter=15,              # Max reasoning iterations
    max_rpm=10,               # Rate limit (requests/minute)
    memory=True,              # Enable memory
    llm="gpt-4o",              # LLM model to use
    tools=[],                 # List of tools
)

from crewai_tools import (
    SerperDevTool,       # Web search
    ScrapeWebsiteTool,   # Web scraping
    FileReadTool,        # Read files
    DirectoryReadTool,   # Read directories
)

# Search tool (requires SERPER_API_KEY)
search_tool = SerperDevTool()
scrape_tool = ScrapeWebsiteTool()

# Research agent with web tools
researcher = Agent(
    role="Web Research Specialist",
    goal="Find accurate, up-to-date information online",
    backstory="Expert at finding reliable information online.",
    tools=[search_tool, scrape_tool],
    verbose=True
)

# File-reading agent
analyst = Agent(
    role="Data Analyst",
    goal="Analyze data files and extract insights",
    backstory="Expert at reading and analyzing data files.",
    tools=[FileReadTool(), DirectoryReadTool()],
    verbose=True
)

from crewai.tools import tool

@tool("Calculate ROI")
def calculate_roi(investment: float, returns: float) -> str:
    """Calculate Return on Investment given investment and returns."""
    roi = ((returns - investment) / investment) * 100
    return f"ROI: {roi:.2f}%"

@tool("Get Stock Price")
def get_stock_price(symbol: str) -> str:
    """Get the current stock price for a given ticker symbol."""
    # In production, call a real API
    return f"{symbol}: $150.25"

# Agent with custom tools
financial_analyst = Agent(
    role="Financial Analyst",
    goal="Provide accurate financial analysis",
    backstory="Expert in financial modeling and valuation.",
    tools=[calculate_roi, get_stock_price],
    verbose=True
)