Method get_running_service

crates/chat-cli/src/mcp_client/client.rs:610–624

(&mut self)

impl InitializedMcpClient {
    pub async fn get_running_service(&mut self) -> Result<&RunningService, McpClientError> {
        match self {
            InitializedMcpClient::Pending(handle) if handle.is_finished() => {
                let running_service = handle.await??;
                *self = InitializedMcpClient::Ready(running_service);
                let InitializedMcpClient::Ready(running_service) = self else {
                    unreachable!()
                };

                Ok(running_service)
            },
            InitializedMcpClient::Ready(running_service) => Ok(running_service),
            InitializedMcpClient::Pending(_) => Err(McpClientError::NotReady),
        }
    }
}

#[cfg(test)]