Adds more builders

This commit is contained in:
2024-11-27 15:11:01 +00:00
parent 326b3919d1
commit db5a01afef
4 changed files with 120 additions and 26 deletions

View File

@@ -15,7 +15,15 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
);
let prompt = "Tell me the story of the genesis of the universe as a bedtime story.";
let request = GenerateContentRequest::from_prompt(prompt, None);
let request = GenerateContentRequest::builder()
.add_content(
Content::builder()
.role(Role::User)
.add_part(Part::Text(prompt.to_string()))
.build(),
)
.build();
let queue = gemini.stream_generate_content(&request, "gemini-pro").await;
while let Some(response) = queue.pop().await {

View File

@@ -15,7 +15,14 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
);
let prompt = "What is the airspeed of an unladen swallow?";
let request = GenerateContentRequest::builder().with_prompt(prompt).build();
let request = GenerateContentRequest::builder()
.add_content(
Content::builder()
.role(Role::User)
.add_part(Part::Text(prompt.to_string()))
.build(),
)
.build();
let response = gemini.generate_content(&request, "gemini-pro").await?;
println!("Response: {:?}", response.candidates[0].get_text().unwrap());