ml_embedding_basic_usage/basic_usage.rs
1use use_ml_embedding::{EmbeddingDimension, EmbeddingModelName, EmbeddingVectorShape};
2
3fn main() -> Result<(), use_ml_embedding::EmbeddingError> {
4 let model = EmbeddingModelName::new("text-embedding")?;
5 let dimension = EmbeddingDimension::new(384)?;
6 let shape = EmbeddingVectorShape::new(dimension);
7
8 assert_eq!(model.as_str(), "text-embedding");
9 assert_eq!(shape.dimension().get(), 384);
10 Ok(())
11}