1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
use std::error::Error;
use rust_ai::openai::Embedding;
#[tokio::main]
async fn main() -> Result<(), Box<dyn Error>> {
std::env::set_var("RUST_LOG", "debug");
std::env::set_var("RUST_BACKTRACE", "1");
log4rs::init_file("log4rs.yml", Default::default()).unwrap();
let mut embedding = Embedding::default();
embedding.set_input("Bard vs ChatGPT");
let result = embedding.embedding().await?;
println!("{:?}", result.data.get(0).unwrap().embedding.clone());
Ok(())
}