Leave Balance Tracker Agent

import pandas as pd
from datetime import datetime

def check_leave_balance(emp_id, leave_type):
    balance = pd.read_sql("SELECT balance FROM leave_balances WHERE emp_id = ? AND type = ?", (emp_id, leave_type))
    return balance[0] if balance else 0

def generate_alerts(balances, thresholds):
    alerts = []
    for key, val in balances.items():
        if val < thresholds.get(key, 0):
            alerts.append(key)
    return alerts