Orchestrate Equipment Monitoring Agents with llama-agents and FastAPI

                      
                     
from fastapi import FastAPI, HTTPException
import os
from llama_agents import EquipmentAgent
from typing import Dict, Any

# Configuration
API_KEY = os.getenv('API_KEY')
BASE_URL = os.getenv('BASE_URL', 'http://localhost:8000')

# Initialize FastAPI app
app = FastAPI()

# Initialize Llama Agent
agent = EquipmentAgent(api_key=API_KEY)

@app.get("/monitor/{equipment_id}")
async def monitor_equipment(equipment_id: str) -> Dict[str, Any]:
    try:
        # Get status from the equipment agent
        status = await agent.get_status(equipment_id)
        return {'success': True, 'data': status}
    except Exception as error:
        # Handle errors gracefully
        raise HTTPException(status_code=500, detail=str(error))

@app.post("/control/{equipment_id}")
async def control_equipment(equipment_id: str, command: str) -> Dict[str, Any]:
    try:
        # Send a control command to the equipment
        result = await agent.send_command(equipment_id, command)
        return {'success': True, 'data': result}
    except Exception as error:
        # Handle errors gracefully
        raise HTTPException(status_code=500, detail=str(error))

if __name__ == '__main__':
    import uvicorn
    uvicorn.run(app, host="0.0.0.0", port=8000)