dstack-server 0.3.2

HTTP API + MCP server for dstack
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

mod api;

use clap::Parser;
use tracing_subscriber::EnvFilter;

#[derive(Parser)]
#[command(name = "dstack-serve", about = "dstack HTTP API server")]
struct Args {
    /// Port to bind to
    #[arg(long, default_value = "3500")]
    port: u16,

    /// Bind address
    #[arg(long, default_value = "127.0.0.1")]
    bind: String,
}

#[tokio::main]
async fn main() -> anyhow::Result<()> {
    tracing_subscriber::fmt()
        .with_env_filter(EnvFilter::from_default_env().add_directive("info".parse()?))
        .init();

    let args = Args::parse();
    let cfg = dstack::config::Config::load()?;
    cfg.load_env();

    let app = api::router(cfg);
    let addr = format!("{}:{}", args.bind, args.port);

    tracing::info!("dstack server listening on {}", addr);
    let listener = tokio::net::TcpListener::bind(&addr).await?;
    axum::serve(listener, app).await?;

    Ok(())
}