Yehowshua/ZIm
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Return language model events when using Google model via zed.dev (#27831)

Browse source 
Release Notes:

- N/A
This commit is contained in:
Bennet Bo Fenner 2025-04-01 10:58:17 +02:00 • committed by GitHub
parent 8be5ed22f6
commit 5509e0141a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 12 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

28
crates/language_models/src/provider/cloud.rs
View file

@ -37,9 +37,7 @@ use strum::IntoEnumIterator;
use ui::{TintColor, prelude::*};
use crate::AllLanguageModelSettings;
use crate::provider::anthropic::{
count_anthropic_tokens, into_anthropic, map_to_language_model_completion_events,
};
use crate::provider::anthropic::{count_anthropic_tokens, into_anthropic};
use crate::provider::google::into_google;
use crate::provider::open_ai::{count_open_ai_tokens, into_open_ai};
@ -648,9 +646,11 @@ impl LanguageModel for CloudLanguageModel {
},
)
.await?;
Ok(map_to_language_model_completion_events(Box::pin(
response_lines(response).map_err(AnthropicError::Other),
)))
Ok(
crate::provider::anthropic::map_to_language_model_completion_events(
Box::pin(response_lines(response).map_err(AnthropicError::Other)),
),
)
});
async move { Ok(future.await?.boxed()) }.boxed()
}
@ -698,17 +698,13 @@ impl LanguageModel for CloudLanguageModel {
},
)
.await?;
Ok(google_ai::extract_text_from_events(response_lines(
response,
)))
Ok(
crate::provider::google::map_to_language_model_completion_events(Box::pin(
response_lines(response),
)),
)
});
async move {
Ok(future
.await?
.map(|result| result.map(LanguageModelCompletionEvent::Text))
.boxed())
}
.boxed()
async move { Ok(future.await?.boxed()) }.boxed()
}
}
}