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