use hypertyper::{HTTPClientFactory, HTTPResult};
use usaidwat::ai::Auth;
use usaidwat::ai::client::APIRequest;
use usaidwat::ai::client::openai::{OpenAIModel, OpenAIRequest, OpenAIResponse};
use usaidwat::ai::service::{APIService, HTTPService};
#[tokio::test]
async fn it_sends_a_post_request_using_gpt4o() {
let auth =
Auth::from_env("OPENAI_API_KEY").expect("Could not create auth. Is $OPENAI_API_KEY set?");
let request = OpenAIRequest::default()
.model(OpenAIModel::Gpt4o)
.input("write a haiku about ai");
let factory = HTTPClientFactory::new(env!("CARGO_PKG_NAME"), env!("CARGO_PKG_VERSION"));
let service = HTTPService::new(factory);
let response: HTTPResult<OpenAIResponse> = service
.post("https://api.openai.com/v1/responses", &auth, &request)
.await;
let resp = response.expect("could not make OpenAI API request");
assert_eq!(resp.output().count(), 1);
assert_eq!(resp.output().next().unwrap().content().count(), 1);
}
#[tokio::test]
async fn it_sends_a_post_request_using_gpt5nano() {
let auth =
Auth::from_env("OPENAI_API_KEY").expect("Could not create auth. Is $OPENAI_API_KEY set?");
let request = OpenAIRequest::default()
.model(OpenAIModel::Gpt5nano)
.input("write a haiku about ai");
let factory = HTTPClientFactory::new(env!("CARGO_PKG_NAME"), env!("CARGO_PKG_VERSION"));
let service = HTTPService::new(factory);
let response: HTTPResult<OpenAIResponse> = service
.post("https://api.openai.com/v1/responses", &auth, &request)
.await;
let resp = response.expect("could not make OpenAI API request");
assert_eq!(resp.output().count(), 2);
let output = resp.output().nth(1).expect(&format!(
"could not get second element of output: {:?}",
resp
));
assert_eq!(output.content().count(), 1);
}