use futures::Stream;
use crate::protocols::openai::stream_aggregator::{StreamAggregable, aggregate_stream};
use crate::types::Annotated;
use super::NvImagesResponse;
impl StreamAggregable for NvImagesResponse {
fn empty() -> Self {
Self::empty()
}
fn merge(&mut self, next: Self) {
self.inner.data.extend(next.inner.data);
}
}
impl NvImagesResponse {
pub async fn from_annotated_stream(
stream: impl Stream<Item = Annotated<NvImagesResponse>>,
) -> Result<NvImagesResponse, String> {
aggregate_stream(stream).await
}
}