diff --git a/sopel_modules/SpiceBot/AI.py b/sopel_modules/SpiceBot/AI.py index 220c681..14e5782 100644 --- a/sopel_modules/SpiceBot/AI.py +++ b/sopel_modules/SpiceBot/AI.py @@ -76,11 +76,11 @@ class SpiceBot_AI(): def load_bot_values(self): - # current_bot_db = botdb.get_bot_value('botai') or None - # if current_bot_db: - # for predicate in list(current_bot_db.keys()): - # predval = current_bot_db[predicate] - # self.aiml_kernel.setBotPredicate(predicate, predval) + current_bot_db = botdb.get_plugin_value('SpiceBot_AI', 'brain') or None + if current_bot_db: + for predicate in list(current_bot_db.keys()): + predval = current_bot_db[predicate] + self.aiml_kernel.setBotPredicate(predicate, predval) # sopel nick self.aiml_kernel.setBotPredicate("nick", botconfig.nick) @@ -242,8 +242,8 @@ class SpiceBot_AI(): def save_brain(self): self.aiml_kernel.saveBrain(botconfig.config.aibrain) - # botsessiondata = self.aiml_kernel._botPredicates - # botdb.set_bot_value('botai', botsessiondata) + botsessiondata = self.aiml_kernel._botPredicates + botdb.set_plugin_value('SpiceBot_AI', 'brain', botsessiondata) botai = SpiceBot_AI()