Email Assistant with CrewAI

from crewai import Agent, Crew, Task
from langchain_openai import ChatOpenAI
from langchain_core.tools import tool

@tool
def search_emails(query: str, limit: int = 5) -> str:
    """Search the inbox for emails matching a query."""
    results = email_client.search(query, max_results=limit)
    return "\n\n".join(
        f"From: {e.sender}\nSubject: {e.subject}\nDate: {e.date}\nPreview: {e.body[:200]}"
        for e in results
    )

@tool
def draft_email(to: str, subject: str, body: str) -> str:
    """Create an email draft."""
    draft_id = email_client.create_draft(to=to, subject=subject, body=body)
    return f"Draft created (ID: {draft_id}). Review before sending."

agent = Agent(
    role="Specialist",
    goal="You are an email assistant. Search emails to find context, then help draft professional replies. Always create drafts — never send directly.",
    tools=[search_emails, draft_email],
    llm=ChatOpenAI(model="gpt-4o"),
)

task = Task(
    description="Find the latest email from the marketing team and draft a reply confirming the deadline",
    expected_output="Detailed response",
    agent=agent,
)

crew = Crew(agents=[agent], tasks=[task])
result = crew.kickoff()
print(result.raw)