ray-dashboard-sdk 0.4.0

A simple Rust client for the Ray Dashboard REST API.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

use std::env;

use ray_dashboard_sdk::RayDashboardClient;
use tracing::{Level, info};
use tracing_subscriber::FmtSubscriber;

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    // Initialize tracing subscriber
    let subscriber = FmtSubscriber::builder().with_max_level(Level::INFO).finish();
    tracing::subscriber::set_global_default(subscriber)?;

    // Get Ray Dashboard connection info from environment variables
    let host = env::var("RAY_DASHBOARD_HOST").unwrap_or("127.0.0.1".to_string());
    let port = env::var("RAY_DASHBOARD_PORT").unwrap_or("8265".to_string());
    let ray_dashboard_url = format!("http://{}:{}", host, port);

    info!("Connecting to Ray Dashboard at: {}", ray_dashboard_url);

    // Create the client
    let client = RayDashboardClient::new(&ray_dashboard_url)?;

    // Ping the server to check connectivity
    info!("Pinging Ray Dashboard...");
    client.ping().await?;
    info!("Successfully connected to Ray Dashboard");

    // List all jobs
    info!("Listing all jobs");
    let jobs = client.jobs().list().await?;

    if jobs.is_empty() {
        info!("No jobs found");
    } else {
        info!("Found {} job(s)", jobs.len());
        for job in jobs {
            let id = job.submission_id.unwrap();
            info!("Job: {} (status: {:?})", id, job.status);
        }
    }

    Ok(())
}