28 lines
891 B
Rust
28 lines
891 B
Rust
use gemini_rs::prelude::*;
|
|
|
|
#[tokio::main]
|
|
async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|
let authentication_manager = gcp_auth::provider().await?;
|
|
let api_endpoint = std::env::var("API_ENDPOINT")?;
|
|
let project_id = std::env::var("PROJECT_ID")?;
|
|
let location_id = std::env::var("LOCATION_ID")?;
|
|
|
|
let gemini = GeminiClient::new(
|
|
authentication_manager,
|
|
api_endpoint,
|
|
project_id,
|
|
location_id,
|
|
);
|
|
|
|
let prompt = vec![Content::builder()
|
|
.role(Role::User)
|
|
.add_text_part("What is the airspeed of an unladen swallow?")
|
|
.build()];
|
|
|
|
let request = GenerateContentRequest::builder().contents(prompt).build();
|
|
let response = gemini.generate_content(&request, "gemini-pro").await?;
|
|
println!("Response: {:?}", response.candidates[0].get_text().unwrap());
|
|
|
|
Ok(())
|
|
}
|