Add builders for GenerateContent and CountToken

This commit is contained in:
2024-11-26 20:27:36 +00:00
parent e6de1d1ce7
commit 326b3919d1
11 changed files with 128 additions and 51 deletions

View File

@@ -15,12 +15,7 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
);
let prompt = "What is the airspeed of an unladen swallow?";
let request = CountTokensRequest {
contents: Content {
role: Some("user".to_string()),
parts: Some(vec![Part::Text(prompt.to_string())]),
},
};
let request = CountTokensRequestBuilder::from_prompt(prompt).build();
let result = gemini.count_tokens(&request, "gemini-pro").await?;
println!("Response: {:?}", result);