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:
parent
d15d85830a
commit
04cd3fcd23
3 changed files with 44 additions and 6 deletions
|
@ -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
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue