From 7240442a8685a15bc54af48528e5b0efe35241a9 Mon Sep 17 00:00:00 2001
From: Jeremy Pinto <jerpint@gmail.com>
Date: Thu, 29 Jun 2023 15:09:05 -0400
Subject: [PATCH] quickfix: use answer_text instead of answer_generator from
 now on (#113)

* quickfix: use answer_text instead of answer_generator from now on
---
 buster/busterbot.py       | 2 +-
 buster/completers/base.py | 2 +-
 tests/test_chatbot.py     | 2 +-
 tests/test_read_write.py  | 2 +-
 4 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/buster/busterbot.py b/buster/busterbot.py
index 81ad5ef..f263475 100644
--- 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
index ddcdc34..0a74e1d 100644
--- 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
index 9db2c18..0f5103f 100644
--- 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
index 1e1c745..772639c 100644
--- 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(),
     )