ask/
ask.rs

1//! Asks Jev three questions about a support ticket, one of each type, then prints the answers
2//! and what they cost.
3//!
4//! ```sh
5//! TYPESAFE_API_KEY=... cargo run --example ask
6//! ```
7//!
8//! This sends the ticket below to TypeSafe's API and is billed to the key.
9
10use serde::Serialize;
11use typesafe_jev::{Choice, Client, Config, Error, Noul, Questions, Score};
12
13/// The state can be any type that serializes to a JSON object, array or string.
14#[derive(Serialize)]
15struct Ticket<'a> {
16    customer_plan: &'a str,
17    message: &'a str,
18}
19
20fn main() -> Result<(), Error> {
21    let client = Client::from_env(Config::default())?;
22
23    let ticket = Ticket {
24        customer_plan: "business",
25        message: "Hi, I've been trying to connect my Stripe account for 3 days and the integration keeps failing. \
26                  I'm losing sales. Please help ASAP.",
27    };
28    let questions = Questions::new()
29        .with(
30            "department",
31            Choice::new(
32                "Which team should handle the `message`?",
33                [
34                    ("billing", "Payment or subscription issues"),
35                    ("technical", "Bugs or integration problems"),
36                    ("sales", "Pricing or account questions"),
37                ],
38            ),
39        )
40        .with(
41            "frustration",
42            Score::new(
43                "How frustrated the customer appears",
44                ["Calm, just stating facts", "Frustrated but civil", "Very angry, strong language"],
45            ),
46        )
47        .with("is_urgent", Noul::new("The message conveys urgency or time-sensitivity"));
48
49    let response = match client.ask(&ticket, &questions) {
50        Ok(response) => response,
51        // Too much state for one request: a real caller would split it and ask again.
52        Err(Error::TokenLimit(message)) => {
53            eprintln!("request too large: {message}");
54            return Ok(());
55        }
56        Err(other) => return Err(other),
57    };
58
59    if let Some(department) = response.choice("department") {
60        println!("department: {} (confidence {:.2})", department.choice, department.confidence);
61        for (option, probability) in &department.probabilities {
62            println!("  {option}: {probability:.2}");
63        }
64    }
65    if let Some(frustration) = response.score("frustration") {
66        println!("frustration: {:.2} of {} (confidence {:.2})", frustration.score, frustration.legend.len() - 1, frustration.confidence);
67    }
68    if let Some(urgent) = response.noul("is_urgent") {
69        println!("urgent: p={:.2}", urgent.noul);
70    }
71
72    let usage = client.usage();
73    println!(
74        "{}: {} request(s), {} retries, {} input tokens, ~${:.6}",
75        response.model,
76        usage.requests(),
77        usage.retries(),
78        usage.input_tokens(),
79        usage.cost_usd()
80    );
81    Ok(())
82}
ask/ask.rs

ask/
ask.rs
