Code Review Agent with Pydantic AI

from pydantic_ai import Agent, RunContext

agent = Agent(
    "openai:gpt-4o",
    system_prompt="You are a code reviewer. Find relevant files, read them, search for patterns, and provide a structured review covering security, performance, and best practices.",
)

@agent.tool
async def read_file(ctx: RunContext, path: str) -> str:
    """Read a file from the project."""
    return Path(path).read_text()

@agent.tool
async def find_files(ctx: RunContext, pattern: str) -> str:
    """Find files matching a glob pattern."""
    files = list(Path(".").rglob(pattern))
    return "\n".join(str(f) for f in files[:20])

@agent.tool
async def search_code(ctx: RunContext, pattern: str, path: str = ".") -> str:
    """Search for a regex pattern in source files."""
    import subprocess
    result = subprocess.run(
        ["grep", "-rn", pattern, path, "--include=*.py"],
        capture_output=True, text=True,
    )
    return result.stdout[:3000] or "No matches found."

result = await agent.run("Review the Python files in src/ for common security issues")
print(result.output)