Schedule and automate agents

1
import requests
2
3
BASE_URL = "https://alfa.boosted.ai/client"
4
API_KEY = "YOUR_API_KEY_HERE"
5
6
headers = {"x-api-key": API_KEY, "Content-Type": "application/json"}
7
8
def enable_agent_automation(agent_id):
9
    """Set an agent to automatically update on a schedule."""
10
    url = f"{BASE_URL}/agent/enable-automation"
11
    payload = {"agent_id": agent_id}
12
    
13
    response = requests.post(url, headers=headers, json=payload)
14
    
15
    if response.status_code == 200:
16
        data = response.json()
17
        next_run = data.get("next_run", "Unknown")
18
        print(f"Automation enabled. Next run scheduled for: {next_run}")
19
        return True
20
    else:
21
        print(f"Error: {response.status_code}, {response.text}")
22
        return False
23
24
# Enable automatic updates for our agent
25
enable_agent_automation("your-agent-id")

1
def set_agent_schedule(agent_id, schedule_description, timezone=None):
2
    """Set a custom schedule for an agent using natural language."""
3
    url = f"{BASE_URL}/agent/set-schedule"
4
    payload = {
5
        "agent_id": agent_id,
6
        "user_schedule_description": schedule_description
7
    }
8
    
9
    if timezone:
10
        payload["agent_timezone"] = timezone
11
    
12
    response = requests.post(url, headers=headers, json=payload)
13
    
14
    if response.status_code == 200:
15
        data = response.json()
16
        print(f"Schedule updated: {data['schedule']['generated_schedule_description']}")
17
        print(f"Next run will use timezone: {data['schedule']['timezone']}")
18
        return data
19
    else:
20
        print(f"Error: {response.status_code}, {response.text}")
21
        return None
22
23
# Set a custom schedule
24
set_agent_schedule("your-agent-id", "Weekly on Friday at 3pm", "US/Pacific")

1
def disable_agent_automation(agent_id):
2
    """Disable automatic updates for an agent."""
3
    url = f"{BASE_URL}/agent/disable-automation"
4
    payload = {"agent_id": agent_id}
5
    
6
    response = requests.post(url, headers=headers, json=payload)
7
    
8
    if response.status_code == 200:
9
        data = response.json()
10
        if data.get("success"):
11
            print("Automation successfully disabled")
12
        return data.get("success", False)
13
    else:
14
        print(f"Error: {response.status_code}, {response.text}")
15
        return False
16
17
# Disable automation for an agent
18
disable_agent_automation("your-agent-id")

1
def rerun_agent(agent_id):
2
    """Manually trigger an agent to run again."""
3
    url = f"{BASE_URL}/agent/rerun-agent"
4
    payload = {"agent_id": agent_id}
5
    
6
    response = requests.post(url, headers=headers, json=payload)
7
    
8
    if response.status_code == 200:
9
        data = response.json()
10
        success = data.get("success", False)
11
        if success:
12
            print("Agent rerun successfully initiated")
13
        return success
14
    else:
15
        print(f"Error: {response.status_code}, {response.text}")
16
        return False
17
18
# Rerun our agent to get fresh data
19
rerun_agent("your-agent-id")