diff --git a/crates/google_ai/src/google_ai.rs b/crates/google_ai/src/google_ai.rs index cd9fb181d8..4ed22717fc 100644 --- a/crates/google_ai/src/google_ai.rs +++ b/crates/google_ai/src/google_ai.rs @@ -1,7 +1,7 @@ mod supported_countries; use anyhow::{Result, anyhow, bail}; -use futures::{AsyncBufReadExt, AsyncReadExt, Stream, StreamExt, io::BufReader, stream::BoxStream}; +use futures::{AsyncBufReadExt, AsyncReadExt, StreamExt, io::BufReader, stream::BoxStream}; use http_client::{AsyncBody, HttpClient, Method, Request as HttpRequest}; use serde::{Deserialize, Serialize}; @@ -455,24 +455,3 @@ impl std::fmt::Display for Model { write!(f, "{}", self.id()) } } - -pub fn extract_text_from_events( - events: impl Stream>, -) -> impl Stream> { - events.filter_map(|event| async move { - match event { - Ok(event) => event.candidates.and_then(|candidates| { - candidates.into_iter().next().and_then(|candidate| { - candidate.content.parts.into_iter().next().and_then(|part| { - if let Part::TextPart(TextPart { text }) = part { - Some(Ok(text)) - } else { - None - } - }) - }) - }), - Err(error) => Some(Err(error)), - } - }) -}