Fix token counting requests in Gemini (#29643)

Release Notes:

- N/A
This commit is contained in:
Michael Sloan
2025-04-30 04:55:07 +00:00
committed by GitHub
parent 8d77efa781
commit edf78e770d
2 changed files with 5 additions and 5 deletions
@@ -327,7 +327,8 @@ impl LanguageModel for GoogleLanguageModel {
request: LanguageModelRequest,
cx: &App,
) -> BoxFuture<'static, Result<usize>> {
let request = into_google(request, self.model.id().to_string());
let model_id = self.model.id().to_string();
let request = into_google(request, model_id.clone());
let http_client = self.http_client.clone();
let api_key = self.state.read(cx).api_key.clone();
@@ -340,6 +341,7 @@ impl LanguageModel for GoogleLanguageModel {
http_client.as_ref(),
&api_url,
&api_key,
&model_id,
google_ai::CountTokensRequest {
contents: request.contents,
},