formatter_agent.py

from pydantic import BaseModel, Field
from agents import Agent
from llm_model import llm_model

INSTRUCTIONS = """You are able to generate a nicely formatted HTML email based on a detailed report.
You will be provided with a detailed report. Generate the HTML for the email, providing the 
report converted into clean, well presented HTML with an appropriate subject line."""


class SendEmailData(BaseModel):
    subject: str = Field(description="The nice subject of the email based on the report.")
    html_report: str = Field(description="The final report in HTML format")


formatter_agent = Agent(
    name="FormatterAgent",
    instructions=INSTRUCTIONS,
    model=llm_model,
    output_type=SendEmailData,
)