minor restructuring for response creation

gunthercox · Sep 7, 2019 · 998a0a1 · 998a0a1
1 parent 0666bb5
commit 998a0a1
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/chatterbot/chatterbot.py b/chatterbot/chatterbot.py
@@ -140,8 +140,6 @@ def generate_response(self, input_statement, additional_response_selection_param
 
         :param input_statement: The input statement to be processed.
         """
-        Statement = self.storage.get_object('statement')
-
         results = []
         result = None
         max_confidence = -1
@@ -196,6 +194,10 @@ def __init__(self, statement, count=1):
             if most_common.count > 1:
                 result = most_common.statement
 
+        return self.create_response(input_statement, result)
+
+    def create_response(self, input_statement, result):
+        Statement = self.storage.get_object('statement')
         response = Statement(
             text=result.text,
             in_response_to=input_statement.text,