udpdate error msg

Author: evanliu048

crates/chat-cli/src/api_client/clients/streaming_client.rs

@@ -175,14 +175,18 @@ impl StreamingClient {
                             && e.raw_response().is_some_and(|resp| resp.status().as_u16() == 500)
                             && e.as_service_error().is_some_and(|err| {
                                 err.meta().message()
-                == Some("Encountered unexpectedly high load when processing the request, please try again.")
+                                == Some("Encountered unexpectedly high load when processing the request, please try again.")
                             });
                         if is_quota_breach {
                             Err(ApiClientError::QuotaBreach("quota has reached its limit"))
                         } else if is_context_window_overflow {
                             Err(ApiClientError::ContextWindowOverflow)
                         } else if is_model_unavailable {
-                            Err(ApiClientError::ModelOverloadedError())
+                            let request_id = e
+                                .as_service_error()
+                                .and_then(|err| err.meta().request_id())
+                                .map(|s| s.to_string());
+                            Err(ApiClientError::ModelOverloadedError(request_id))
                         } else {
                             Err(e.into())
                         }

crates/chat-cli/src/api_client/error.rs

@@ -69,7 +69,7 @@ pub enum ApiClientError {
     #[error(
         "The model you've selected is temporarily unavailable. Please use '/model' to select a different model and try again."
     )]
-    ModelOverloadedError(),
+    ModelOverloadedError(Option<String>),
 }
 
 #[cfg(test)]

crates/chat-cli/src/cli/chat/mod.rs

@@ -1086,6 +1086,23 @@ impl ChatContext {
                         crate::api_client::ApiClientError::QuotaBreach(msg) => {
                             print_err!(msg, err);
                         },
+                        crate::api_client::ApiClientError::ModelOverloadedError(request_id) => {
+                            queue!(
+                                self.output,
+                                style::SetAttribute(Attribute::Bold),
+                                style::SetForegroundColor(Color::Red),
+                            )?;
+                            let message = "The model you've selected is temporarily unavailable. Please use '/model' to select a different model and try again.";
+                            queue!(
+                                self.output,
+                                style::Print(&format!("{}\nRequest ID: {}", message, request_id))
+                            )?;
+                            execute!(
+                                self.output,
+                                style::SetAttribute(Attribute::Reset),
+                                style::SetForegroundColor(Color::Reset),
+                            )?;
+                        },
                         _ => {
                             print_default_error!(err);
                         },