Define specific error for event stream ended

This commit is contained in:
2025-04-06 08:12:58 +01:00
parent 8ff379d040
commit d1678bdc37
2 changed files with 7 additions and 0 deletions

View File

@@ -68,6 +68,9 @@ impl<T: TokenProvider + Clone> GeminiClient<T> {
let mapped = event_source.filter_map(|event| {
let event = match event {
Ok(event) => event,
Err(reqwest_eventsource::Error::StreamEnded) => {
return Some(Err(Error::EventSourceClosedError))
}
Err(e) => return Some(Err(e.into())),
};