diff --git a/README.md b/README.md index 856711f..2e622e4 100644 --- a/README.md +++ b/README.md @@ -1,2 +1 @@ -# G.1.R - GARBAGE INTERNET ROBOT - +# G.1.R - GARBAGE INTERNET ROBOT \ No newline at end of file diff --git a/src/modules/ai.rs b/src/modules/ai.rs index e223225..beae3e2 100644 --- a/src/modules/ai.rs +++ b/src/modules/ai.rs @@ -9,6 +9,7 @@ struct Config { nick: String, channels: Vec, openai: String, + model: String, accents: String, personalities: String, } @@ -49,7 +50,7 @@ async fn ai(user_message: &str, username: &str, channel: &str) -> Vec { let chat_request = CreateCompletionRequestArgs::default() .prompt(user_message) .max_tokens(40_u16) - .model("text-davinci-003") + .model(config.model) .build() .unwrap(); let chat_response = client @@ -61,6 +62,21 @@ async fn ai(user_message: &str, username: &str, channel: &str) -> Vec { //modify regex for varible username ie G1R g1r GIR gir but as handle nick for bots let response_text = &chat_response.choices.first().unwrap().text; let regex = Regex::new(r#""|[gG][1iI][rR]:\s*|[mM][eE]:?\s"#).unwrap(); + //let nick = &config.nick; + //let regex_str = format!( + // r#""|[{}{}{}]|\b[gG][1iI][rR]:\s*|\b[mM][eE]:?\s"#, + // nick.to_lowercase(), + // nick.to_uppercase(), + // nick.chars().map(|c| match c { /// regex magic nick removal in progress + // 'a' => '4', + // 'e' => '3', + // 'i' => '1', + // 'o' => '0', + // 's' => '5', + // _ => c, + // }).collect::(), + //); + //let regex = Regex::new(®ex_str).unwrap(); let response_text = regex.replace_all(response_text, "").trim().to_string(); let response_lines = response_text.split("\n").filter(|line| !line.trim().is_empty()); let mut responses = Vec::new();