Method next

crates/chat-cli/src/agent/rts/mod.rs:607–645

(&mut self)

    }

    async fn next(&mut self) -> Result<Option<ChatResponseStream>, RecvError> {
        if let Some(ev) = self.peek.take() {
            return Ok(Some(ev));
        }

        trace!("Attempting to recv next event");
        let start = Instant::now();
        let result = self.response.recv().await;
        let duration = Instant::now().duration_since(start);
        match result {
            Ok(ev) => {
                trace!(?ev, "Received new event");

                if !self.message_start_pushed {
                    self.buf
                        .push(StreamResult::Ok(StreamEvent::MessageStart(MessageStartEvent {
                            role: Role::Assistant,
                        })));
                    self.message_start_pushed = true;
                }

                // Track metadata about the chunk.
                self.time_to_first_chunk
                    .get_or_insert_with(|| self.request_start_time.elapsed());
                self.time_between_chunks.push(duration);
                self.received_response_size += ev.as_ref().map(|e| e.len()).unwrap_or_default();

                Ok(ev)
            },
            Err(err) => {
                error!(?err, "failed to receive the next event");
                if duration.as_secs() >= 59 {
                    Err(RecvError::Timeout { source: err, duration })
                } else {
                    Err(RecvError::Other { source: err })
                }
            },
        }
    }

    fn recv_error_to_stream_error(&self, err: RecvError) -> StreamError {
        match err {