From 6392118cb684cf7f88cd03d9a5d9f08d0adb4870 Mon Sep 17 00:00:00 2001 From: iff Date: Thu, 12 Dec 2024 16:17:04 +0100 Subject: [PATCH] feat: multiple AI suggestion --- module-request-ai/src/main.rs | 6 ++++-- module-request-ai/src/requests.rs | 4 ++-- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/module-request-ai/src/main.rs b/module-request-ai/src/main.rs index 1a921ab..72026b9 100644 --- a/module-request-ai/src/main.rs +++ b/module-request-ai/src/main.rs @@ -29,8 +29,10 @@ fn main() -> Result<(), std::io::Error> { let note = fill(&suggest.note, termwidth()); eprintln!("{}\n{}\n", warn, note); - let command = suggest.command; - print!("{}<_PR_BR>", command); + let suggestions = suggest.commands; + for suggestion in suggestions { + print!("{}<_PR_BR>", suggestion); + } } Ok(()) } diff --git a/module-request-ai/src/requests.rs b/module-request-ai/src/requests.rs index 533ff42..fa0ec50 100644 --- a/module-request-ai/src/requests.rs +++ b/module-request-ai/src/requests.rs @@ -24,7 +24,7 @@ struct Messages { #[derive(Serialize, Deserialize)] pub struct AISuggest { - pub command: String, + pub commands: Vec, pub note: String, } @@ -52,7 +52,7 @@ pub fn ai_suggestion(last_command: &str, error_msg: &str) -> Option { let user_locale = { let locale = std::env::var("_PR_AI_LOCALE") - .unwrap_or_else(|_| get_locale().unwrap_or("en".to_string())); + .unwrap_or_else(|_| get_locale().unwrap_or("en-us".to_string())); if locale.len() < 2 { "en-US".to_string() } else {