use std::io::Read;
fn main() {
let mut input_string = String::new();
std::io::stdin().read_line(&mut input_string).unwrap();
let role = match input_string.as_str() {
"system" => Role::System,
"user" => Role::User,
"assistant" => Role::Assistant,
_ => panic!("Invalid role"),
};
println!("Role: {:?}", role);
}
#[derive(Debug)]
enum Role {
System,
User,
Assistant,
}