chalk-client 0.1.0

Rust client SDK for the Chalk feature store
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
44
45
46
47
48
49
50
51

//! Querying features via gRPC.
//!
//! Run with:
//! ```sh
//! cargo run --example grpc_query
//! ```

use chalk_client::gen::chalk::common::v1::{OnlineQueryRequest, OutputExpr};
use chalk_client::ChalkGrpcClient;
use std::collections::HashMap;

#[tokio::main]
async fn main() -> chalk_client::error::Result<()> {
    let client = ChalkGrpcClient::new().build().await?;
    println!("Connected to environment: {}", client.environment_id());

    let request = OnlineQueryRequest {
        inputs: HashMap::from([(
            "user.id".to_string(),
            prost_types::Value {
                kind: Some(prost_types::value::Kind::NumberValue(1.0)),
            },
        )]),
        outputs: vec![
            OutputExpr {
                expr: Some(chalk_client::gen::chalk::common::v1::output_expr::Expr::FeatureFqn(
                    "user.name".to_string(),
                )),
            },
            OutputExpr {
                expr: Some(chalk_client::gen::chalk::common::v1::output_expr::Expr::FeatureFqn(
                    "user.age".to_string(),
                )),
            },
        ],
        ..Default::default()
    };

    let response = client.query_proto(request).await?;

    if let Some(data) = &response.data {
        for result in &data.results {
            println!("{}: {:?}", result.field, result.value);
        }
    }
    for err in &response.errors {
        eprintln!("error: {}", err.message);
    }

    Ok(())
}