google: Add latest versions of Gemini 2.5 Pro and Flash Preview (#32183)

Release Notes:

- Added the latest versions of Gemini 2.5 Pro and Flash Preview
This commit is contained in:
Oleksiy Syvokon 2025-06-05 22:30:34 +03:00 committed by GitHub
parent d15d85830a
commit 04cd3fcd23
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 44 additions and 6 deletions

View file

@ -362,7 +362,7 @@ impl LanguageModel for GoogleLanguageModel {
}
fn telemetry_id(&self) -> String {
format!("google/{}", self.model.id())
format!("google/{}", self.model.request_id())
}
fn max_token_count(&self) -> usize {
@ -374,7 +374,7 @@ impl LanguageModel for GoogleLanguageModel {
request: LanguageModelRequest,
cx: &App,
) -> BoxFuture<'static, Result<usize>> {
let model_id = self.model.id().to_string();
let model_id = self.model.request_id().to_string();
let request = into_google(request, model_id.clone(), self.model.mode());
let http_client = self.http_client.clone();
let api_key = self.state.read(cx).api_key.clone();
@ -411,7 +411,11 @@ impl LanguageModel for GoogleLanguageModel {
>,
>,
> {
let request = into_google(request, self.model.id().to_string(), self.model.mode());
let request = into_google(
request,
self.model.request_id().to_string(),
self.model.mode(),
);
let request = self.stream_completion(request, cx);
let future = self.request_limiter.stream(async move {
let response = request