Sienna Meridian Satterwhite
ce96bcfeeb
- Use Model::mistral_small_latest() instead of enum variants - Use Model::mistral_embed() instead of EmbedModel::MistralEmbed - Use serde_json::json!() for tool parameter schemas - Update ChatParams for Option<f32> temperature field - Update streaming test comments for new delta.content type
26 lines
943 B
Rust
26 lines
943 B
Rust
use jrest::expect;
|
|
use mistralai_client::v1::{client::Client, constants::Model};
|
|
|
|
#[test]
|
|
fn test_client_embeddings() {
|
|
let client: Client = Client::new(None, None, None, None).unwrap();
|
|
|
|
let model = Model::mistral_embed();
|
|
let input = vec!["Embed this sentence.", "As well as this one."]
|
|
.iter()
|
|
.map(|s| s.to_string())
|
|
.collect();
|
|
let options = None;
|
|
|
|
let response = client.embeddings(model, input, options).unwrap();
|
|
|
|
expect!(response.object).to_be("list".to_string());
|
|
expect!(response.data.len()).to_be(2);
|
|
expect!(response.data[0].index).to_be(0);
|
|
expect!(response.data[0].object.clone()).to_be("embedding".to_string());
|
|
expect!(response.data[0].embedding.len()).to_be_greater_than(0);
|
|
expect!(response.usage.prompt_tokens).to_be_greater_than(0);
|
|
expect!(response.usage.completion_tokens).to_be(0);
|
|
expect!(response.usage.total_tokens).to_be_greater_than(0);
|
|
}
|