use test_log::test;
use tyrell::{ClaudeRequest, ContentType, Model, Role};
#[test(tokio::test)]
async fn test_qa() {
let chat = ClaudeRequest::builder()
.model(Model::Opus3)
.add_message(
Role::User,
vec![ContentType::Text {
text: "who was the 16th president of the USA?".to_string(),
}],
)
.max_tokens(200)
.build()
.unwrap();
let response = chat.call().await.unwrap();
println!("{:#?}", response);
assert_eq!(response.role, Role::Assistant);
assert!(!response.content.is_empty());
}