nvidia_riva 0.1.0

The rust language implementation of gRPC client for nvidia riva.
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

use std::path::PathBuf;

fn main() {
    tonic_build::configure()
        .out_dir(default_proto_dir().unwrap())
        .type_attribute(
            ".",
            "#[allow(unknown_lints, clippy::derive_partial_eq_without_eq)]",
        )
        .compile(
            &[
                "api/common/riva/proto/health.proto",
                "api/common/riva/proto/riva_audio.proto",
                "api/common/riva/proto/riva_asr.proto",
                "api/common/riva/proto/riva_nlp.proto",
                "api/common/riva/proto/riva_nmt.proto",
                "api/common/riva/proto/riva_tts.proto",
            ],
            &["api/common/"],
        )
        .unwrap();
}

pub fn default_proto_dir() -> anyhow::Result<PathBuf> {
    let dir = format!("{}/target", std::env::var("CARGO_MANIFEST_DIR")?).into();
    std::fs::create_dir_all(&dir)?;
    Ok(dir)
}