mirror of https://github.com/0xplaygrounds/rig
32 lines
699 B
Rust
32 lines
699 B
Rust
use rig::providers::gemini;
|
|
use rig::Embed;
|
|
|
|
#[derive(Embed, Debug)]
|
|
struct Greetings {
|
|
#[embed]
|
|
message: String,
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<(), anyhow::Error> {
|
|
// Initialize the Google Gemini client
|
|
// Create OpenAI client
|
|
let client = gemini::Client::from_env();
|
|
|
|
let embeddings = client
|
|
.embeddings(gemini::embedding::EMBEDDING_001)
|
|
.document(Greetings {
|
|
message: "Hello, world!".to_string(),
|
|
})?
|
|
.document(Greetings {
|
|
message: "Goodbye, world!".to_string(),
|
|
})?
|
|
.build()
|
|
.await
|
|
.expect("Failed to embed documents");
|
|
|
|
println!("{:?}", embeddings);
|
|
|
|
Ok(())
|
|
}
|