technofab/pay-respects
1
0
Fork 0
mirror of https://github.com/TECHNOFAB11/pay-respects.git synced 2025-12-12 06:20:09 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: additional AI configuration

Browse source
This commit is contained in:
iff 2024-11-20 21:16:14 +01:00
parent 927e7d4263
commit b4e432cfd8
3 changed files with 19 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/main.rs
View file

@ -45,7 +45,9 @@ fn main() {
#[cfg(feature = "request-ai")]
{
std::env::set_var("_PR_LOCALE", &locale);
if std::env::var("_PR_AI_LOCALE").is_err() {
std::env::set_var("_PR_AI_LOCALE", &locale);
}
}
args::handle_args();

18
src/requests.rs
View file

@ -23,6 +23,10 @@ pub struct AISuggest {
}
pub fn ai_suggestion(last_command: &str, error_msg: &str) -> Option<AISuggest> {
if std::env::var("_PR_AI_DISABLE").is_ok() {
return None;
}
let error_msg = if error_msg.len() > 300 {
&error_msg[..300]
} else {
@ -70,23 +74,19 @@ pub fn ai_suggestion(last_command: &str, error_msg: &str) -> Option<AISuggest> {
Err(_) => "llama3-8b-8192".to_string(),
};
let user_locale = std::env::var("_PR_LOCALE").unwrap_or("en-US".to_string());
let set_locale = if user_locale != "en-US" {
format!(
"Provide the note in the language for the locale {}\n",
user_locale
)
let user_locale = std::env::var("_PR_AI_LOCALE").unwrap_or("en-US".to_string());
let set_locale = if !user_locale.starts_with("en") {
format!(". Use language for locale {}", user_locale)
} else {
"".to_string()
};
let ai_prompt = format!(
r#"
You run the command `{last_command}` and get the following error message: `{error_msg}`. What would you run next? Answer in the following JSON format without any extra text:
The command `{last_command}` returns the following error message: `{error_msg}`. Provide a command to fix it. Answer in the following JSON format without any extra text:
```
{{"command":"suggestion","note":"why it may fix the error"}}
{{"command":"suggestion","note":"why it may fix the error{set_locale}"}}
```
{set_locale}If you can't provide a good suggestion, reply the command field with `None` and a explanation in note
"#
);