Content Generation Agent with CrewAI

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

@tool
def web_search(query: str) -> str:
    """Research a topic before writing content."""
    from tavily import TavilyClient
    client = TavilyClient()
    results = client.search(query, max_results=3)
    return "\n\n".join(r["content"] for r in results["results"])

@tool
def save_content(filename: str, content: str) -> str:
    """Save generated content to a file."""
    Path(f"output/{filename}").write_text(content)
    return f"Saved to output/{filename}"

agent = Agent(
    role="Specialist",
    goal="You are a content writer. Research the topic first, then write engaging content. Save the final output using save_content.",
    tools=[web_search, save_content],
    llm=ChatOpenAI(model="gpt-4o"),
)

task = Task(
    description="Write a blog post about the benefits of AI agents in customer service",
    expected_output="Detailed response",
    agent=agent,
)

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