embedding/
embedding.rs

1use openai_rst::{
2    client::Client,
3    embedding::EmbeddingRequest,
4    models::{EmbeddingsModels, Model},
5};
6
7#[tokio::main]
8async fn main() -> Result<(), Box<dyn std::error::Error>> {
9    let client = Client::from_env().unwrap();
10
11    let mut req = EmbeddingRequest::new(
12        Model::Embedding(EmbeddingsModels::TextEmbeddingAda002),
13        "story time".to_string(),
14    );
15    req.dimensions = Some(10);
16
17    let result = client.embedding(req).await?;
18    println!("{:?}", result.data);
19
20    Ok(())
21}
22
23// OPENAI_API_KEY=xxxx cargo run --package openai-rst --example embedding