fix: add login check for sigv4 chat support (#323)

brandonskiser · web-flow · commit ce3cf0400f6f · 2025-06-25T16:07:25.000-07:00
diff --git a/crates/q_cli/src/util/mod.rs b/crates/q_cli/src/util/mod.rs
@@ -291,7 +291,7 @@ pub fn dialoguer_theme() -> ColorfulTheme {
 }
 
 pub async fn assert_logged_in() -> Result<(), Error> {
-    if !fig_util::system_info::in_cloudshell() && !fig_auth::is_logged_in().await {
+    if !(std::env::var("AMAZON_Q_SIGV4").is_ok_and(|v| !v.is_empty()) || fig_auth::is_logged_in().await) {
         bail!(
             "You are not logged in, please log in with {}",
             format!("{CLI_BINARY_NAME} login",).bold()

Original file line number	Diff line number	Diff line change
`@@ -291,7 +291,7 @@ pub fn dialoguer_theme() -> ColorfulTheme {`
`291`	`291`	`}`
`292`	`292`
`293`	`293`	`pub async fn assert_logged_in() -> Result<(), Error> {`
`294`		`- if !fig_util::system_info::in_cloudshell() && !fig_auth::is_logged_in().await {`
	`294`	`+ if !(std::env::var("AMAZON_Q_SIGV4").is_ok_and(\|v\| !v.is_empty()) \|\| fig_auth::is_logged_in().await) {`
`295`	`295`	`bail!(`
`296`	`296`	`"You are not logged in, please log in with {}",`
`297`	`297`	`format!("{CLI_BINARY_NAME} login",).bold()`