Method complete

core/src/agent_api/run_lifecycle.rs:28–59

(&self, result: std::result::Result<AgentResult, E>)

impl StreamRunWorkerState {
    pub(super) async fn complete<E>(&self, result: std::result::Result<AgentResult, E>)
    where
        E: std::fmt::Display,
    {
        match result {
            Ok(result) => {
                if let Some(persistence) = &self.persistence {
                    persistence.record_result(&result);
                    self.should_auto_save
                        .store(true, std::sync::atomic::Ordering::Release);
                }
            }
            Err(error) => {
                let cancelled = self
                    .cancel_token
                    .lock()
                    .await
                    .as_ref()
                    .map(|t| t.is_cancelled())
                    .unwrap_or(false);
                if cancelled {
                    let _ = self.run_store.mark_cancelled(&self.run_id).await;
                } else {
                    let error_message = error.to_string();
                    let _ = self
                        .run_store
                        .mark_failed(&self.run_id, error_message)
                        .await;
                }
            }
        }
    }
}

#[derive(Clone)]