andreban/gemini-rs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Makes all usageMetadata fields optional

Browse Source
This commit is contained in:
André Cipriani Bandarra
2024-07-12 16:49:26 +01:00
parent 154b2f353b
commit 74150d16b0
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/types/generate_content.rs
View File

@@ -97,8 +97,8 @@ pub struct SafetyRating {
#[serde(rename_all = "camelCase")] #[serde(rename_all = "camelCase")]
pub struct UsageMetadata { pub struct UsageMetadata {
pub candidates_token_count: Option<u32>, pub candidates_token_count: Option<u32>,
pub prompt_token_count: u32, pub prompt_token_count: Option<u32>,
pub total_token_count: u32, pub total_token_count: Option<u32>,
} }
#[derive(Clone, Debug, Serialize, Deserialize)] #[derive(Clone, Debug, Serialize, Deserialize)]
@@ -163,7 +163,13 @@ impl GenerateContentResponse {
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use super::GenerateContentResponse; use super::{GenerateContentResponse, GenerateContentResponseResult};
#[test]
pub fn parses_empty_metadata_response() {
let input = r#"{"candidates": [{"content": {"role": "model","parts": [{"text": "-"}]}}],"usageMetadata": {}}"#;
serde_json::from_str::<GenerateContentResponseResult>(input).unwrap();
}
#[test] #[test]
pub fn parses_max_tokens_response() { pub fn parses_max_tokens_response() {