Fix embedding response

This commit is contained in:
Dongri Jin
2023-05-08 06:44:45 +09:00
parent 9486ce5015
commit 0f2194ebb8
2 changed files with 27 additions and 3 deletions

19
examples/embedding.rs Normal file
View File

@ -0,0 +1,19 @@
use openai_api_rs::v1::api::Client;
use openai_api_rs::v1::embedding::EmbeddingRequest;
use std::env;
#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
let client = Client::new(env::var("OPENAI_API_KEY").unwrap().to_string());
let req = EmbeddingRequest {
model: "text-embedding-ada-002".to_string(),
input: "story time".to_string(),
user: Option::None,
};
let result = client.embedding(req).await?;
println!("{:?}", result.data);
Ok(())
}
// OPENAI_API_KEY=xxxx cargo run --package openai-api-rs --example embedding