AI/LLM Monthly Budget Planner

# Budget Monitoring Setup
import requests
from datetime import datetime

class LLMBudgetMonitor:
    def __init__(self, monthly_budget, alert_threshold=0.8):
        self.monthly_budget = monthly_budget
        self.alert_threshold = alert_threshold
        self.current_spend = 0
        self.alerts_enabled = True
        
    def track_api_call(self, model, tokens, cost):
        """Track each API call and cost"""
        self.current_spend += cost
        
        # Check if approaching budget limit
        utilization = self.current_spend / self.monthly_budget
        
        if utilization > self.alert_threshold and self.alerts_enabled:
            self.send_budget_alert(utilization)
            
        if utilization > 1.0:
            self.enforce_budget_limit()
            
    def send_budget_alert(self, utilization):
        """Send alert when approaching budget"""
        message = f"Budget Alert: {utilization*100:.1f}% of monthly budget used"
        # Send to Slack, email, etc.
        print(f"⚠️ {message}")
        
    def enforce_budget_limit(self):
        """Enforce hard budget limits"""
        # Implement rate limiting or service suspension
        raise Exception("Monthly budget exceeded!")
        
    def get_budget_report(self):
        """Generate budget utilization report"""
        return {
            "budget": self.monthly_budget,
            "spent": self.current_spend,
            "remaining": self.monthly_budget - self.current_spend,
            "utilization": (self.current_spend / self.monthly_budget) * 100,
            "projected_monthly": self.project_monthly_spend()
        }
        
    def project_monthly_spend(self):
        """Project total monthly spend based on current rate"""
        today = datetime.now()
        days_passed = today.day
        days_in_month = 30  # Simplified
        
        daily_rate = self.current_spend / days_passed
        return daily_rate * days_in_month

# Usage
monitor = LLMBudgetMonitor(monthly_budget=15000, alert_threshold=0.8)

# Track API calls
monitor.track_api_call("gpt-4", tokens=1000, cost=0.03)
monitor.track_api_call("gpt-3.5-turbo", tokens=5000, cost=0.0075)

# Get report
report = monitor.get_budget_report()
print(f"Budget Status: ${report['spent']:.2f} / ${report['budget']:.2f}")
print(f"Projected Monthly: ${report['projected_monthly']:.2f}")