Create and manage agents (Deprecated)

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 create_agent(prompt):
9
    """Create a new agent with the specified prompt."""
10
    url = f"{BASE_URL}/agent/create-agent"
11
    payload = {"prompt": prompt}
12
    
13
    response = requests.post(url, headers=headers, json=payload)
14
    
15
    if response.status_code == 201:
16
        data = response.json()
17
        agent_id = data.get("agent_id")
18
        print(f"Success! Agent created with ID: {agent_id}")
19
        return agent_id
20
    else:
21
        print(f"Error: {response.status_code}, {response.text}")
22
        return None
23
24
# Create an agent to analyze Tesla's stock performance
25
agent_id = create_agent("What was TSLA's close price on Feb 1, 2023 and how does it compare to recent performance?")

1
import time
2
3
def get_agent_status(agent_id):
4
    """Check if an agent has completed its processing."""
5
    url = f"{BASE_URL}/agent/status?agent_id={agent_id}"
6
    
7
    response = requests.get(url, headers=headers)
8
    
9
    if response.status_code == 200:
10
        data = response.json()
11
        status = data.get("status")
12
        print(f"Agent status: {status}")
13
        return status
14
    else:
15
        print(f"Error: {response.status_code}, {response.text}")
16
        return None
17
18
# Poll until the agent is done
19
status = ""
20
while status != "COMPLETE":
21
    status = get_agent_status(agent_id)
22
    if status in ["ERROR", "CANCELLED", "NO_RESULTS_FOUND"]:
23
        print(f"Agent encountered an issue: {status}")
24
        break
25
    if status != "COMPLETE":
26
        print("Agent still working... waiting 10 seconds")
27
        time.sleep(10)

1
def get_agent_output(agent_id):
2
    """Retrieve the results from a completed agent."""
3
    url = f"{BASE_URL}/get-agent-output/{agent_id}"
4
    
5
    response = requests.get(url, headers=headers)
6
    
7
    if response.status_code == 200:
8
        return response.json()
9
    else:
10
        print(f"Error: {response.status_code}, {response.text}")
11
        return None
12
13
# Get the agent's findings
14
output = get_agent_output(agent_id)
15
16
# Process the outputs
17
if output and "outputs" in output:
18
    for item in output["outputs"]:
19
        output_type = item["output"]["output_type"]
20
        
21
        if output_type == "text":
22
            print("Text output:", item["output"]["val"])
23
        elif output_type == "table":
24
            print("Table output with columns:", [col["name"] for col in item["output"]["columns"]])
25
        elif output_type == "graph":
26
            print(f"Graph output: {item['output']['title']} ({item['output']['graph']['graph_type']} graph)")

1
def get_agent(agent_id):
2
    """Get metadata for a specific agent."""
3
    url = f"{BASE_URL}/agent/get-agent/{agent_id}"
4
    
5
    response = requests.get(url, headers=headers)
6
    
7
    if response.status_code == 200:
8
        data = response.json()
9
        print(f"Agent Name: {data.get('agent_name')}")
10
        print(f"Created: {data.get('created_at')}")
11
        print(f"Description: {data.get('agent_description')}")
12
        return data
13
    else:
14
        print(f"Error: {response.status_code}, {response.text}")
15
        return None
16
17
# Get details for a specific agent
18
agent_details = get_agent(agent_id)

1
def get_all_agents():
2
    """List all available agents."""
3
    url = f"{BASE_URL}/agent/get-all-agents"
4
    
5
    response = requests.get(url, headers=headers)
6
    
7
    if response.status_code == 200:
8
        data = response.json()
9
        return data.get("agents", [])
10
    else:
11
        print(f"Error: {response.status_code}, {response.text}")
12
        return []
13
14
# Display all your agents
15
agents = get_all_agents()
16
for agent in agents:
17
    print(f"ID: {agent['agent_id']}")
18
    print(f"Name: {agent['agent_name']}")
19
    print(f"Created: {agent['created_at']}")
20
    print("---")