Bug Zapper Agent

from dataclasses import dataclass
from pathlib import Path
from typing import Union

@dataclass
class FlagDependencies:
    flag: str
    file_path: Union[str, Path]
    
    def __post_init__(self):
        self.file_path = Path(self.file_path)

from pydantic import BaseModel, Field

class ZapperResult(BaseModel):
    path: str = Field("The path to the file where the error occurs")
    file: str = Field("The name of the file with the error")
    line: int = Field("Which line the error occured in", ge=0)
    what: str = Field("Explanation in natural language of what the error is")
    todo: str = Field("Describe steps to take in natural language in order to fix the error")

from pydantic_ai import Agent
import os

bug_zapper = Agent(
    model_name=os.getenv('MODEL'),
    api_key=os.getenv('API_KEY'),
    deps_type=FlagDependencies,
    result_type=ZapperResult,
    system_prompt=(
        'You are an expert software debugging assistant specializing in programming error analysis. '
        'Your task is to analyze error tracebacks and provide structured, actionable advice. '
    )
)

from agents.agents import bug_zapper, FlagDependencies
import asyncio

async def analyze_error():
    # Create dependencies
    deps = FlagDependencies(
        flag="strict",
        file_path="/path/to/buggy_file.py"
    )
    
    # Run the agent with error traceback
    result = await bug_zapper.run(
        """Traceback (most recent call last):
  File "example.py", line 15, in <module>
    result = divide(10, 0)
  File "example.py", line 5, in divide
    return a / b
ZeroDivisionError: division by zero
""",
        deps=deps
    )
    
    print(f"Error in: {result.data.file}:{result.data.line}")
    print(f"What: {result.data.what}")
    print(f"Fix: {result.data.todo}")

asyncio.run(analyze_error())

MODEL=gpt-4  # AI model to use
API_KEY=your_api_key_here  # API key for the AI service

@bug_zapper.tool
def custom_debugging_tool(ctx: RunContext):
    # Add custom debugging logic
    pass