Basic

"""
Gmail Agent that can read, draft and send emails using the Gmail.
"""

from agno.agent import Agent
from agno.models.openai import OpenAIResponses
from agno.tools.google.gmail import GmailTools
from pydantic import BaseModel, Field

# ---------------------------------------------------------------------------
# Create Agent
# ---------------------------------------------------------------------------


class FindEmailOutput(BaseModel):
    message_id: str = Field(..., description="The message id of the email")
    thread_id: str = Field(..., description="The thread id of the email")
    references: str = Field(..., description="The references of the email")
    in_reply_to: str = Field(..., description="The in-reply-to of the email")
    subject: str = Field(..., description="The subject of the email")
    body: str = Field(..., description="The body of the email")


# Example 1: Include specific Gmail functions for reading only
read_only_agent = Agent(
    name="Gmail Reader Agent",
    model=OpenAIResponses(id="gpt-5.5"),
    tools=[
        GmailTools(
            include_tools=[
                "search_emails",
                "get_emails_by_thread",
                "mark_email_as_read",
                "mark_email_as_unread",
                "list_custom_labels",
            ]
        )
    ],
    description="You are a Gmail reading specialist that can search, read and label emails.",
    instructions=[
        "You can search and read Gmail messages but cannot send or draft emails.",
        "You can mark emails as read or unread for processing workflows.",
        "You can list all available labels in the user's Gmail account.",
        "Summarize email contents and extract key details and dates.",
        "Show the email contents in a structured markdown format.",
    ],
    markdown=True,
    output_schema=FindEmailOutput,
)

# Example 2: Exclude dangerous functions (sending emails)
safe_gmail_agent = Agent(
    name="Safe Gmail Agent",
    model=OpenAIResponses(id="gpt-5.5"),
    tools=[GmailTools(exclude_tools=["send_email", "send_email_reply"])],
    description="You are a Gmail agent with safe operations only.",
    instructions=[
        "You can read and draft emails but cannot send them.",
        "Show the email contents in a structured markdown format.",
    ],
    markdown=True,
    output_schema=FindEmailOutput,
)

# Example 3: Label Management Specialist Agent
label_manager_agent = Agent(
    name="Gmail Label Manager",
    model=OpenAIResponses(id="gpt-5.5"),
    tools=[
        GmailTools(
            include_tools=[
                "list_custom_labels",
                "apply_label",
                "remove_label",
                "delete_custom_label",
                "search_emails",
                "get_emails_by_context",
            ]
        )
    ],
    description="You are a Gmail label management specialist that helps organize emails with labels.",
    instructions=[
        "You specialize in Gmail label management operations.",
        "You can list existing custom labels, apply labels to emails, remove labels, and delete labels.",
        "Always be careful when deleting labels - confirm with the user first.",
        "When applying or removing labels, search for relevant emails first.",
        "Provide clear feedback on label operations performed.",
    ],
    markdown=True,
)

# Example 4: Full Gmail functionality (default)
agent = Agent(
    name="Full Gmail Agent",
    model=OpenAIResponses(id="gpt-5.5"),
    tools=[GmailTools()],
    description="You are an expert Gmail Agent that can read, draft, send and label emails using Gmail.",
    instructions=[
        "Based on user query, you can read, draft, send and label emails using Gmail.",
        "While showing email contents, you can summarize the email contents, extract key details and dates.",
        "Show the email contents in a structured markdown format.",
        "Attachments can be added to the email",
        "When you need to modify an email, make sure to find its message_id and thread_id in order to do modification operations.",
    ],
    markdown=True,
    output_schema=FindEmailOutput,
)

# Example 5: Draft a reply to a conversation thread
thread_reply_agent = Agent(
    name="Thread Reply Agent",
    model=OpenAIResponses(id="gpt-5.5"),
    tools=[GmailTools()],
    description="You are a Gmail agent that finds conversations and drafts threaded replies.",
    instructions=[
        "Search for the requested thread, load full context, then draft a reply.",
        "Always create a draft -- never send directly.",
        "Summarize the thread context so the user knows what the reply addresses.",
    ],
    markdown=True,
)


# BASIC GMAIL OPERATIONS EXAMPLES

# Example 1: Find the last email from a specific sender
email = "<replace_with_email_address>"

# ---------------------------------------------------------------------------
# Run Agent
# ---------------------------------------------------------------------------
if __name__ == "__main__":
    response = agent.print_response(
        f"Find the last email from {email} along with the message id, references and in-reply-to",
        markdown=True,
        stream=True,
        output_schema=FindEmailOutput,
    )

    # Example 2: Mark an email as read/unread (useful for processing workflows)
    # Note: You would typically get the message_id from a search operation first

    # Mark as read (removes UNREAD label)
    agent.print_response(
        f"""Mark the last email received from {email} as unread.""",
        markdown=True,
        stream=True,
    )

    # Example 3: Send a new email with attachments
    # agent.print_response(
    #     f"""Send an email to {email} with subject 'Subject'
    #     and body 'Body' and Attach the file 'tmp/attachment.pdf'""",
    #     markdown=True,
    #     stream=True,
    # )

    # LABEL MANAGEMENT EXAMPLES

    # Example 4.1: List all custom labels
    label_manager_agent.print_response(
        "List all my custom labels in Gmail.",
        markdown=True,
        stream=True,
    )

    # Example 4.2: Apply labels to organize emails
    label_manager_agent.print_response(
        "Apply the 'Newsletters' label to emails from 'newsletter@company.com'. Process the last 5 emails.",
        markdown=True,
        stream=True,
    )

    # Example 4.3: Remove labels from emails
    label_manager_agent.print_response(
        "Remove the 'Urgent' label from emails containing 'resolved' in the subject. Process up to 5 emails.",
        markdown=True,
        stream=True,
    )

    # Example 5: Draft a reply to a conversation thread
    # thread_reply_agent.print_response(
    #     "Find the latest thread about 'project update' and draft a reply asking about next steps",
    #     stream=True,
    # )