Source code for sugaroid.brain.reset

from chatterbot.logic import LogicAdapter
from sugaroid.core.statement import SugaroidStatement
from sugaroid.brain.ooo import Emotion
from sugaroid.brain.preprocessors import normalize


[docs]class ResetAdapter(LogicAdapter): """ Resets the Sugaroid global variables """ def __init__(self, chatbot, **kwargs): super().__init__(chatbot, **kwargs)
[docs] def can_process(self, statement): normalized = normalize(str(statement).lower()) if "reset" in normalized and "admin" in normalized: return True else: return False
[docs] def process(self, statement, additional_response_selection_parameters=None): self.chatbot.reset_variables() selected_statement = SugaroidStatement( "Reset of chatbot variables. SUCCESS", chatbot=True ) selected_statement.confidence = 0.95 emotion = Emotion.neutral selected_statement.emotion = emotion return selected_statement