diff --git a/chatterbot/storage/django_storage.py b/chatterbot/storage/django_storage.py index b0a6f9775..c4b8fabd9 100644 --- a/chatterbot/storage/django_storage.py +++ b/chatterbot/storage/django_storage.py @@ -55,7 +55,7 @@ def filter(self, **kwargs): if responses: kwargs['in_response__response__text__in'] = [] for response in responses: - kwargs['in_response__response__text__in'].append(response.response.text) + kwargs['in_response__response__text__in'].append(response) else: kwargs['in_response'] = None diff --git a/chatterbot/storage/mongodb.py b/chatterbot/storage/mongodb.py index 0dba194a0..cf267364b 100644 --- a/chatterbot/storage/mongodb.py +++ b/chatterbot/storage/mongodb.py @@ -170,7 +170,7 @@ def filter(self, **kwargs): if 'in_response_to' in kwargs: serialized_responses = [] for response in kwargs['in_response_to']: - serialized_responses.append({'text': response.text}) + serialized_responses.append({'text': response}) query = query.statement_response_list_equals(serialized_responses) del kwargs['in_response_to'] diff --git a/tests/storage_adapter_tests/test_json_file_storage_adapter.py b/tests/storage_adapter_tests/test_json_file_storage_adapter.py index bc7b78396..379aa079b 100644 --- a/tests/storage_adapter_tests/test_json_file_storage_adapter.py +++ b/tests/storage_adapter_tests/test_json_file_storage_adapter.py @@ -250,7 +250,7 @@ def test_filter_in_response_to_no_matches(self): self.adapter.update(self.statement1) results = self.adapter.filter( - in_response_to=[Response("Maybe")] + in_response_to="Maybe" ) self.assertEqual(len(results), 0) diff --git a/tests/storage_adapter_tests/test_mongo_adapter.py b/tests/storage_adapter_tests/test_mongo_adapter.py index 66cd7234f..cb5e4e1c1 100644 --- a/tests/storage_adapter_tests/test_mongo_adapter.py +++ b/tests/storage_adapter_tests/test_mongo_adapter.py @@ -277,9 +277,7 @@ def test_filter_text_no_matches(self): def test_filter_in_response_to_no_matches(self): self.adapter.update(self.statement1) - results = self.adapter.filter( - in_response_to=[Response("Maybe")] - ) + results = self.adapter.filter(in_response_to="Maybe") self.assertEqual(len(results), 0) def test_filter_equal_results(self): diff --git a/tests/storage_adapter_tests/test_storage_adapter.py b/tests/storage_adapter_tests/test_storage_adapter.py index a9c6cd34b..35c077d05 100644 --- a/tests/storage_adapter_tests/test_storage_adapter.py +++ b/tests/storage_adapter_tests/test_storage_adapter.py @@ -20,7 +20,7 @@ def test_count(self): def test_find(self): with self.assertRaises(StorageAdapter.AdapterMethodNotImplementedError): - self.adapter.find("") + self.adapter.find('') def test_filter(self): with self.assertRaises(StorageAdapter.AdapterMethodNotImplementedError): @@ -28,11 +28,11 @@ def test_filter(self): def test_remove(self): with self.assertRaises(StorageAdapter.AdapterMethodNotImplementedError): - self.adapter.remove("") + self.adapter.remove('') def test_update(self): with self.assertRaises(StorageAdapter.AdapterMethodNotImplementedError): - self.adapter.update("") + self.adapter.update('') def test_get_random(self): with self.assertRaises(StorageAdapter.AdapterMethodNotImplementedError): diff --git a/tests_django/test_django_adapter.py b/tests_django/test_django_adapter.py index 011588072..fec6e5c89 100644 --- a/tests_django/test_django_adapter.py +++ b/tests_django/test_django_adapter.py @@ -203,9 +203,7 @@ def test_filter_text_no_matches(self): def test_filter_in_response_to_no_matches(self): self.adapter.update(self.statement1) - results = self.adapter.filter( - in_response_to__contains="Maybe" - ) + results = self.adapter.filter(in_response_to="Maybe") self.assertEqual(len(results), 0) def test_filter_equal_results(self):