quickfix: use answer_text instead of answer_generator from now on (#113)

* quickfix: use answer_text instead of answer_generator from now on
jerpint · Jun 29, 2023 · 7240442 · 7240442
1 parent 332a722
commit 7240442
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/buster/busterbot.py b/buster/busterbot.py
@@ -102,7 +102,7 @@ def process_input(self, user_input: str, source: str = None) -> Completion:
                 error=False,
                 user_input=user_input,
                 matched_documents=pd.DataFrame(),
-                answer_generator=irrelevant_question_message,
+                answer_text=irrelevant_question_message,
                 answer_relevant=False,
                 question_relevant=False,
                 validator=self.validator,

diff --git a/buster/completers/base.py b/buster/completers/base.py
@@ -237,7 +237,7 @@ def get_completion(
             # because we are proceeding with a completion, we assume the question is relevant.
             completion = self.completion_class(
                 user_input=user_input,
-                answer_generator=self.no_documents_message,
+                answer_text=self.no_documents_message,
                 error=False,
                 matched_documents=matched_documents,
                 question_relevant=question_relevant,

diff --git a/tests/test_chatbot.py b/tests/test_chatbot.py
@@ -82,7 +82,7 @@ def complete(self):
 
     def get_completion(self, user_input, matched_documents, validator, *arg, **kwarg) -> Completion:
         return Completion(
-            answer_generator=self.expected_answer,
+            answer_text=self.expected_answer,
             error=False,
             user_input=user_input,
             matched_documents=matched_documents,

diff --git a/tests/test_read_write.py b/tests/test_read_write.py
@@ -29,7 +29,7 @@ def test_read_write_completion():
     c = Completion(
         user_input="What is the meaning of life?",
         error=False,
-        answer_generator="This is my actual answer",
+        answer_text="This is my actual answer",
         matched_documents=matched_documents,
         validator=MockValidator(),
     )