Task Automation Agent with Pydantic AI

from pydantic_ai import Agent, RunContext

agent = Agent(
    "openai:gpt-4o",
    system_prompt="You are a project management assistant. Help organize tasks: create new ones, track progress, and update statuses. Be proactive about suggesting task breakdowns.",
)

@agent.tool
async def create_task(ctx: RunContext, title: str, description: str, assignee: str = "") -> str:
    """Create a new task in the project."""
    task_id = await project.create_task(title=title, description=description, assignee=assignee)
    return f"Task created: #{task_id} - {title}"

@agent.tool
async def list_tasks(ctx: RunContext, status: str = "open") -> str:
    """List tasks filtered by status (open, in_progress, done)."""
    tasks = await project.list_tasks(status=status)
    return "\n".join(f"#{t.id} [{t.status}] {t.title} -> {t.assignee}" for t in tasks)

@agent.tool
async def update_task(ctx: RunContext, task_id: int, status: str) -> str:
    """Update task status."""
    await project.update_task(task_id, status=status)
    return f"Task #{task_id} updated to {status}"

result = await agent.run("Break down 'Launch new website' into subtasks and assign them to the team")
print(result.output)