Discord Notifications

from peargent import create_agent
from peargent.tools import discord_tool 
from peargent.models import gemini

# Create an agent with Discord notification capability
agent = create_agent(
    name="DiscordAssistant",
    description="A helpful assistant that sends Discord notifications",
    persona=(
        "You are a professional notification assistant. When asked to send Discord messages, "
        "create clear, well-formatted content with appropriate embeds. Use rich formatting "
        "when beneficial for readability. Always confirm successful delivery or report any "
        "errors encountered."
    ),
    model=gemini("gemini-2.5-flash-lite"),
    tools=[discord_tool] 
)

# Use the agent to send a notification
response = agent.run(
    "Send a Discord message saying our deployment completed successfully. "
    "Include details: version 2.0, duration 3 minutes, deployed by CI/CD."
)
print(response)

from peargent.tools import discord_tool

# Send a message with a basic embed
result = discord_tool.run({
    "content": "System status update:",
    "embed": { 
        "title": "System Alert",
        "description": "All systems are operational.",
        "color": 0x00FF00  # Green color
    }
})

if result["success"]:
    print("✅ Embed message sent successfully!")

from peargent.tools import discord_tool

# Send a rich embed with multiple fields
result = discord_tool.run({
    "content": "📊 Server metrics:",
    "embed": { 
        "title": "Server Status Report",
        "description": "Current system metrics",
        "color": 0x5865F2,  # Discord blurple
        "fields": [ 
            {"name": "CPU Usage", "value": "45%", "inline": True},
            {"name": "Memory", "value": "60%", "inline": True},
            {"name": "Disk Space", "value": "75%", "inline": True},
            {"name": "Uptime", "value": "99.9%", "inline": True}
        ],
        "footer": {"text": "Last updated: 2026-01-10"}
    }
})

if result["success"]:
    print("✅ Rich embed sent successfully!")

from peargent.tools import discord_tool

# Use template variables for dynamic content
result = discord_tool.run({
    "content": "Hello {{ name }}! Your task **{{ task }}** is now complete.", 
    "template_vars": { 
        "name": "Team",
        "task": "Data Processing"
    }
})

if result["success"]:
    print("✅ Templated message sent successfully!")

from peargent.tools import discord_tool

# Apply templates to both content and embed
result = discord_tool.run({
    "content": "Deployment to {{ environment }} completed!", 
    "template_vars": { 
        "environment": "Production",
        "status": "Success",
        "version": "v2.3.0",
        "duration": "2m 45s",
        "deployer": "CI/CD Pipeline"
    },
    "embed": {
        "title": "Deployment Status: {{ status }}", 
        "description": "Version {{ version }} deployed successfully", 
        "color": 0x00FF00,  # Green for success
        "fields": [
            {"name": "Version", "value": "{{ version }}", "inline": True}, 
            {"name": "Duration", "value": "{{ duration }}", "inline": True}, 
            {"name": "Deployed By", "value": "{{ deployer }}", "inline": True} 
        ]
    }
})

if result["success"]:
    print("✅ Templated embed sent successfully!")

from peargent.tools import discord_tool

# Override webhook username and avatar
result = discord_tool.run({
    "content": "Custom bot reporting for duty! 🤖",
    "username": "Peargent Bot", 
    "avatar_url": "https://example.com/avatar.png"
})

if result["success"]:
    print("✅ Custom message sent successfully!")

from peargent.tools import discord_tool

# Send embed with image and thumbnail
result = discord_tool.run({
    "content": "🖼️ Image showcase:",
    "embed": {
        "title": "Image Showcase",
        "description": "Check out this beautiful image!",
        "color": 0xFF5733,  # Orange
        "image": {"url": "https://example.com/large-image.png"}, 
        "thumbnail": {"url": "https://example.com/thumbnail.png"} 
    }
})

if result["success"]:
    print("✅ Embed with image sent successfully!")

from peargent.tools import discord_tool
from datetime import datetime, timezone

# Send alert with timestamp
result = discord_tool.run({
    "content": "**URGENT: System Alert**",
    "embed": {
        "title": "⚠️ High CPU Usage Detected",
        "description": "Server CPU usage has exceeded 90% threshold",
        "color": 0xFFCC00,  # Yellow/amber for warning
        "fields": [
            {"name": "Server", "value": "prod-server-01", "inline": True},
            {"name": "Current Usage", "value": "94%", "inline": True},
            {"name": "Threshold", "value": "90%", "inline": True},
            {"name": "Recommendation", "value": "Scale up resources or investigate processes", "inline": False}
        ],
        "footer": {"text": "Monitoring System"}, 
        "timestamp": datetime.now(timezone.utc).isoformat() 
    }
})

if result["success"]:
    print("🚨 Alert notification sent successfully!")

from peargent.tools import discord_tool
from datetime import datetime, timezone

# Send GitHub-style notification
result = discord_tool.run({
    "content": "New GitHub activity:",
    "embed": {
        "title": "GitHub Pull Request",
        "description": "A new pull request has been opened",
        "url": "https://github.com/user/repo/pull/123", 
        "color": 0x238636,  # GitHub green
        "author": { 
            "name": "GitHub Bot",
            "url": "https://github.com",
            "icon_url": "https://github.githubassets.com/images/modules/logos_page/GitHub-Mark.png"
        },
        "fields": [
            {"name": "Repository", "value": "user/awesome-project", "inline": True},
            {"name": "Branch", "value": "feature/new-tool", "inline": True},
            {"name": "Author", "value": "developer123", "inline": True},
            {"name": "Status", "value": "✅ All checks passed", "inline": False},
            {"name": "Changes", "value": "+150 -20", "inline": True},
            {"name": "Files Changed", "value": "5", "inline": True}
        ],
        "footer": {
            "text": "GitHub Notifications",
            "icon_url": "https://github.githubassets.com/images/modules/logos_page/GitHub-Mark.png"
        },
        "timestamp": datetime.now(timezone.utc).isoformat()
    }
})

if result["success"]:
    print("✅ GitHub notification sent successfully!")

from peargent.tools import discord_tool

# Send build success notification with details
result = discord_tool.run({
    "content": "Build completed!",
    "embed": {
        "title": "✅ Build Successful",
        "description": "All tests passed and deployment completed",
        "color": 0x00FF00,  # Green
        "fields": [
            {"name": "Build #", "value": "1234", "inline": True},
            {"name": "Branch", "value": "main", "inline": True},
            {"name": "Commit", "value": "`abc123f`", "inline": True},
            {"name": "Tests", "value": "✅ 150 passed\n⏭️ 0 skipped\n❌ 0 failed", "inline": False}, 
            {"name": "Coverage", "value": "95%", "inline": True},
            {"name": "Build Time", "value": "3m 22s", "inline": True}
        ],
        "thumbnail": {"url": "https://example.com/success-icon.png"}
    }
})

if result["success"]:
    print("✅ Success notification sent successfully!")

from peargent.tools import discord_tool

# Use Jinja2 conditionals (requires Jinja2 installed)
result = discord_tool.run({
    "content": "{% if severity == 'critical' %}🚨 CRITICAL: {% endif %}System {{ status }}", 
    "template_vars": { 
        "severity": "critical",
        "status": "down"
    },
    "embed": {
        "title": "System Status Alert",
        "color": 0xFF0000
    }
})

if result["success"]:
    print("✅ Conditional message sent successfully!")

from peargent.tools import discord_tool

# Use Jinja2 loops (requires Jinja2 installed)
result = discord_tool.run({
    "content": "Daily Report Summary",
    "embed": {
        "title": "Daily Report",
        "description": """**Completed Tasks:**
{% for task in tasks %}
- {{ task.name }}: {{ task.status }}
{% endfor %}""", 
        "color": 0x5865F2
    },
    "template_vars": { 
        "tasks": [
            {"name": "Data backup", "status": "✅ Success"},
            {"name": "Log cleanup", "status": "✅ Success"},
            {"name": "Health check", "status": "✅ Success"}
        ]
    }
})

if result["success"]:
    print("✅ Looped template sent successfully!")

from peargent import create_agent
from peargent.tools import discord_tool 
from peargent.models import gemini

# Create an intelligent Discord notification agent
agent = create_agent(
    name="DevOpsNotifier",
    description="A DevOps notification assistant",
    persona="""
You are a DevOps notification assistant. When asked to send notifications:
1. Create well-formatted Discord messages with appropriate embeds
2. Use colors based on severity: green=success, yellow=warning, red=error
3. Structure information clearly in embed fields
4. Include relevant emojis for visual clarity
5. Add timestamps for time-sensitive alerts
Always use the discord_tool to send messages.
    """,
    model=gemini("gemini-2.5-flash-lite"),
    tools=[discord_tool] 
)

# Scenario 1: Database backup notification
print("Scenario 1: Database Backup")
print("-" * 60)
response = agent.run(
    "Send a Discord notification that the database backup completed successfully. "
    "Include details: database name 'prod-db', size '2.5 GB', duration '5 minutes', "
    "next backup scheduled for '2026-01-11 02:00 UTC'."
)
print(f"Agent Response: {response}\n")

# Scenario 2: Deployment alert
print("Scenario 2: Deployment Alert")
print("-" * 60)
response = agent.run(
    "Notify the team that deployment to production has started. "
    "Version v3.2.0, deploying from main branch, commit hash: abc123f. "
    "Estimated duration: 5 minutes."
)
print(f"Agent Response: {response}\n")

# Scenario 3: Error notification
print("Scenario 3: Error Alert")
print("-" * 60)
response = agent.run(
    "Send an urgent alert about high memory usage on web-server-02. "
    "Current usage: 95%, threshold: 85%. Immediate action required."
)
print(f"Agent Response: {response}")

from peargent.tools import discord_tool

# Send to alerts channel
result = discord_tool.run({
    "content": "🚨 Critical alert!",
    "webhook_url": "https://discord.com/api/webhooks/ID1/TOKEN1"
})

# Send to general channel
result = discord_tool.run({
    "content": "📢 General announcement",
    "webhook_url": "https://discord.com/api/webhooks/ID2/TOKEN2"
})