2 releases
0.1.1 | Jun 13, 2021 |
---|---|
0.1.0 | Jun 13, 2021 |
#868 in Machine learning
11KB
206 lines
HLTK chatbot
This is a regex powered chatbot made in rust, it is based in NLTK https://github.com/nltk/nltk/blob/develop/nltk/chat/util.py
How to install
add to your Cargo.toml file
[dependencies]
...
human_language_toolkit_chatbot = "0.1.0"
...
You can find the updated version at https://crates.io/crates/human_language_toolkit_chatbot
How to use it
You can read from a json file.
use human_language_toolkit_chatbot::{Chatbot, ChatbotError, CompiledChatbot};
...
let suntsu = Chatbot::from_file(String::from("bots/suntsu.json")).unwrap();
...
Also you can create the bot by programming.
...
let mut pairs: Vec<(String, Vec<String>)> = Vec::new();
pairs.push((
String::from(r"Hello(.*)"),
vec![
String::from("Hello... I'm glad you could drop by today."),
String::from("Hi there... how are you today?"),
String::from("Hello, how are you feeling today?"),
],
));
pairs.push((
String::from(r"(.*) sorry (.*)"),
vec![
String::from("There are many times when no apology is needed."),
String::from("What feelings do you have when you apologize?"),
],
));
...
let fallback = vec![String::from("Sorry I didn't understand")];
let reflections = Chatbot::default_reflections();
let eliza = Chatbot {
pairs,
fallback,
reflections,
};
// You can save your model in a .json file
match eliza.to_file(String::from("bots/eliza.json")) {
Ok(_) => (),
Err(e) => println!("error at {}", e),
}
let eliza = eliza.compile();
You can talk with the bot with the function respond
but you can also use the converse
function to make an infinity conversation.
Let's connect 😋
Dependencies
~3–5MB
~94K SLoC