Content Generation Agent with Pydantic AI

from pydantic_ai import Agent, RunContext

agent = Agent(
    "openai:gpt-4o",
    system_prompt="You are a content writer. Research the topic first, then write engaging content. Save the final output using save_content.",
)

@agent.tool
async def web_search(ctx: RunContext, 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"])

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

result = await agent.run("Write a blog post about the benefits of AI agents in customer service")
print(result.output)