diff --git a/CHANGELOG.md b/CHANGELOG.md index 39e9362c..3579cedc 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,3 +1,6 @@ +# Unreleased +- Support `gpt_5_1_2025_11_13`, `gpt_5_4_2026_03_05`, `gemini_2_5_pro`, and `gemini_3_1_pro_preview` model versions + # v0.39.0 - Allow `re get comments --reviewed-only` to be combined with dataset filters diff --git a/api/src/resources/dataset.rs b/api/src/resources/dataset.rs index 66b024f8..ee0ca5cd 100644 --- a/api/src/resources/dataset.rs +++ b/api/src/resources/dataset.rs @@ -165,8 +165,16 @@ pub struct IterativeConfig { pub enum GptModelVersion { #[serde(rename = "gpt_4o_2024_05_13")] Gpt4o20240513, + #[serde(rename = "gpt_5_1_2025_11_13")] + Gpt5120251113, + #[serde(rename = "gpt_5_4_2026_03_05")] + Gpt5420260305, #[serde(rename = "gemini_2_5_flash")] GeminiFlash25, + #[serde(rename = "gemini_2_5_pro")] + GeminiPro25, + #[serde(rename = "gemini_3_1_pro_preview")] + Gemini31ProPreview, } #[derive(Eq, Clone, Debug, PartialEq, Serialize, Deserialize)] diff --git a/cli/src/commands/get/comments.rs b/cli/src/commands/get/comments.rs index 9c87f729..ec1058df 100644 --- a/cli/src/commands/get/comments.rs +++ b/cli/src/commands/get/comments.rs @@ -908,7 +908,7 @@ fn get_comments_from_uids( // they are set to None or [] here let comments: Vec<_> = page .into_iter() - .zip(predictions.into_iter()) + .zip(predictions) .map(|(comment, prediction)| AnnotatedComment { comment: comment.comment, labelling: Some(vec![Labelling { diff --git a/cli/src/commands/get/emails.rs b/cli/src/commands/get/emails.rs index 846d32b4..7b8b93ed 100644 --- a/cli/src/commands/get/emails.rs +++ b/cli/src/commands/get/emails.rs @@ -102,7 +102,7 @@ fn download_emails( .try_for_each(|page| { let page = page.context("Operation to get emails has failed.")?; statistics.add_emails(page.len()); - print_resources_as_json(page.into_iter(), &mut writer) + print_resources_as_json(page, &mut writer) })?; log::info!( "Successfully downloaded {} emails.",