ChatterBot 1.2.1__py3-none-any.whl → 1.2.3__py3-none-any.whl

chatterbot/storage/storage_adapter.py

@@ -1,6 +1,4 @@
 import logging
-from chatterbot import languages
-from chatterbot.tagging import PosLemmaTagger
 
 
 class StorageAdapter(object):
@@ -17,11 +15,9 @@ class StorageAdapter(object):
         """
         self.logger = kwargs.get('logger', logging.getLogger(__name__))
 
-        Tagger = kwargs.get('tagger', PosLemmaTagger)
-
-        self.tagger = Tagger(language=kwargs.get(
-            'tagger_language', languages.ENG
-        ))
+        self.raise_on_missing_search_text = kwargs.get(
+            'raise_on_missing_search_text', True
+        )
 
     def get_model(self, model_name):
         """
@@ -116,6 +112,12 @@ class StorageAdapter(object):
             this parameter, then the statement will be included in the
             result set.
             Defaults to None
+
+        :param search_in_response_to: If the ``search_in_response_to`` field
+            of a statement contains a word that is in the string provided to
+            this parameter, then the statement will be included in the
+            result set.
+            Defaults to None
         """
         raise self.AdapterMethodNotImplementedError(
             'The `filter` method is not implemented by this adapter.'

chatterbot/trainers.py

@@ -93,7 +93,7 @@ class ListTrainer(Trainer):
         statements_to_create = []
 
         # Run the pipeline in bulk to improve performance
-        documents = self.chatbot.storage.tagger.as_nlp_pipeline(conversation)
+        documents = self.chatbot.tagger.as_nlp_pipeline(conversation)
 
         # for text in enumerate(conversation):
         for document in tqdm(documents, desc='List Trainer', disable=not self.show_training_progress):
@@ -143,7 +143,7 @@ class ChatterBotCorpusTrainer(Trainer):
             for conversation in corpus:
 
                 # Run the pipeline in bulk to improve performance
-                documents = self.chatbot.storage.tagger.as_nlp_pipeline(conversation)
+                documents = self.chatbot.tagger.as_nlp_pipeline(conversation)
 
                 previous_statement_text = None
                 previous_statement_search_text = ''
@@ -344,7 +344,7 @@ class UbuntuCorpusTrainer(Trainer):
                     previous_statement_text = None
                     previous_statement_search_text = ''
 
-                    documents = self.chatbot.storage.tagger.as_nlp_pipeline([
+                    documents = self.chatbot.tagger.as_nlp_pipeline([
                         (
                             row[3],
                             {

chatterbot/vectorstores.py

@@ -0,0 +1,74 @@
+"""
+Redis vector store.
+"""
+from __future__ import annotations
+
+from typing import Any, List, Sequence
+
+from langchain_core.documents import Document
+from redisvl.redis.utils import convert_bytes
+from redisvl.query import FilterQuery
+
+from langchain_core.documents import Document
+from langchain_redis.vectorstores import RedisVectorStore as LangChainRedisVectorStore
+
+
+class RedisVectorStore(LangChainRedisVectorStore):
+    """
+    Redis vector store integration.
+    """
+
+    def query_search(
+        self,
+        k=4,
+        filter=None,
+        sort_by=None,
+    ) -> List[Document]:
+        """
+        Return docs based on the provided query.
+
+        k: int, default=4
+            Number of documents to return.
+        filter: str, default=None
+            A filter expression to apply to the query.
+        sort_by: str, default=None
+            A field to sort the results by.
+
+        returns:
+            A list of Documents most matching the query.
+        """
+        from chatterbot import ChatBot
+
+        return_fields = [
+            self.config.content_field
+        ]
+        return_fields += [
+            field.name
+            for field in self._index.schema.fields.values()
+            if field.name
+            not in [self.config.embedding_field, self.config.content_field]
+        ]
+
+        query = FilterQuery(
+            return_fields=return_fields,
+            num_results=k,
+            filter_expression=filter,
+            sort_by=sort_by,
+        )
+
+        try:
+            results = self._index.query(query)
+        except Exception as e:
+            raise ChatBot.ChatBotException(f'Error querying index: {query}') from e
+
+        if results:
+            with self._index.client.pipeline(transaction=False) as pipe:
+                for document in results:
+                    pipe.hgetall(document['id'])
+                full_documents = convert_bytes(pipe.execute())
+        else:
+            full_documents = []
+
+        return self._prepare_docs_full(
+            True, results, full_documents, True
+        )

{ChatterBot-1.2.1.dist-info → chatterbot-1.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ChatterBot
-Version: 1.2.1
+Version: 1.2.3
 Summary: ChatterBot is a machine learning, conversational dialog engine
 Author: Gunther Cox
 License: Copyright (c) 2016 - 2025, Gunther Cox
@@ -53,11 +53,18 @@ Requires-Dist: flake8; extra == "test"
 Requires-Dist: coverage; extra == "test"
 Requires-Dist: nose; extra == "test"
 Requires-Dist: sphinx<8.2,>=5.3; extra == "test"
+Requires-Dist: sphinx-sitemap>=2.6.0; extra == "test"
 Provides-Extra: dev
 Requires-Dist: pint>=0.8.1; extra == "dev"
-Requires-Dist: pymongo<4.12,>=4.11; extra == "dev"
 Requires-Dist: pyyaml<7.0,>=6.0; extra == "dev"
 Requires-Dist: chatterbot-corpus>=1.2.2; extra == "dev"
+Provides-Extra: redis
+Requires-Dist: redis[hiredis]; extra == "redis"
+Requires-Dist: langchain-redis; extra == "redis"
+Requires-Dist: langchain-huggingface; extra == "redis"
+Requires-Dist: sentence-transformers; extra == "redis"
+Provides-Extra: mongodb
+Requires-Dist: pymongo<4.12,>=4.11; extra == "mongodb"
 
 ![ChatterBot: Machine learning in Python](https://i.imgur.com/b3SCmGT.png)
 
@@ -70,7 +77,6 @@ to be trained to speak any language.
 
 [![Package Version](https://img.shields.io/pypi/v/chatterbot.svg)](https://pypi.python.org/pypi/chatterbot/)
 [![Python 3.9](https://img.shields.io/badge/python-3.9-blue.svg)](https://www.python.org/downloads/release/python-360/)
-[![Django 2.0](https://img.shields.io/badge/Django-2.0-blue.svg)](https://docs.djangoproject.com/en/2.1/releases/2.0/)
 [![Coverage Status](https://img.shields.io/coveralls/gunthercox/ChatterBot.svg)](https://coveralls.io/r/gunthercox/ChatterBot)
 [![Code Climate](https://codeclimate.com/github/gunthercox/ChatterBot/badges/gpa.svg)](https://codeclimate.com/github/gunthercox/ChatterBot)
 [![Join the chat at https://gitter.im/chatterbot/Lobby](https://badges.gitter.im/chatterbot/Lobby.svg)](https://gitter.im/chatterbot/Lobby?utm_source=badge&utm_medium=badge&utm_content=badge)

{ChatterBot-1.2.1.dist-info → chatterbot-1.2.3.dist-info}/RECORD

@@ -1,8 +1,8 @@
-chatterbot/__init__.py,sha256=Edfsbjoii_j0I4OOI1p1JLwfVRpwUW_8NhjfoNuvZRI,158
+chatterbot/__init__.py,sha256=PhC2oXazQN3HNYXvWb33IAqEzwyt5QtqcfESq8eg3sg,158
 chatterbot/__main__.py,sha256=nk19D56TlPT9Zdqkq4qZZrOnLKEc4YTwUVWmXYwSyHg,207
 chatterbot/adapters.py,sha256=LJ_KqLpHKPdYAFpMGK63RVH4weV5X0Zh5uGyan6qdVU,878
-chatterbot/chatterbot.py,sha256=WhV4sSa8psrm6DebtAewIEwkz3MrnjWRftQeZMQso2w,9328
-chatterbot/comparisons.py,sha256=bSVTsCbFEiyVv0Rg60addtrxgjbG3QlJa3murgDWkws,6145
+chatterbot/chatterbot.py,sha256=YLKLkQ-XI4Unr3rbzjpGIupOqenuevm21tAnx-yFFgQ,10400
+chatterbot/comparisons.py,sha256=8-qLFWC1Z7tZ3iPUpyY6AD9l-whSo3QE1Rno_SzIp-I,6570
 chatterbot/components.py,sha256=ld3Xam8olBClvE5QqcFYggE7Q7tODCFek7BO7lhfyeU,1782
 chatterbot/constants.py,sha256=c_KPQKc82CHX6H3maeyTYqWatx6j-N-8HJhmejoVi60,1875
 chatterbot/conversation.py,sha256=Y-WOxPN7I3igRyAEe5py1sfS6JIYPdbwjVlY3kM8Ys8,3175
@@ -12,11 +12,12 @@ chatterbot/filters.py,sha256=vDSDJz2FM10xT6ybs7qJiqy4X5I4gTEfwEnjBGUxZ9g,847
 chatterbot/languages.py,sha256=XSenfc5FxHk_JWG5gGHsZvjvrPBbCaVCm_OU-BeER_M,32784
 chatterbot/parsing.py,sha256=vS-w70cMkjq4YEpDOv_pXWhAI6Zj06WYDAcMDhYDj0M,23174
 chatterbot/preprocessors.py,sha256=aI4v987dZc7GOKhO43i0i73EX748hehYSpzikFHpEXs,1271
-chatterbot/response_selection.py,sha256=9E7CJKlC3UCHTGvEmYvfE9cEHOltJeU77z9NfRzmeB8,2950
-chatterbot/search.py,sha256=Bx6j_NIdp7YDnJvunE7rmk9ma37AKYw96R0iu2i4chc,5141
+chatterbot/response_selection.py,sha256=aYeZ54jpGIcQnI-1-TDcua_f1p3PiM5_iMg4hF5ZaIU,2951
+chatterbot/search.py,sha256=S4MFL1JzPqT-pv7tCgd-MIqf0T9Ia_KOLoNgzdoCP4Y,7035
 chatterbot/tagging.py,sha256=GLY9wg_rvn6pSYVML-HcxkIo_3BZ3SAyj-q1oNZY8pI,2584
-chatterbot/trainers.py,sha256=4u6RDRPpAnecTEAOrGcDvMTjEn8Kxn8slM4UnovDvNk,13339
+chatterbot/trainers.py,sha256=U1yh0_V7FFL51MeQe1P1Q59weceDbkHh_2kDiDYpSEc,13315
 chatterbot/utils.py,sha256=ckQXvsjp2FO9GcWxziY67JovN7mShnE4RlzdYarQY5k,3277
+chatterbot/vectorstores.py,sha256=-S1NB8PrZzoFIu95n2W7N4UaXuCUpyDUXIGYFebjv08,2056
 chatterbot/ext/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chatterbot/ext/django_chatterbot/__init__.py,sha256=iWzmBzpAsYwkwi1faxAPFY9L1bbL97RgVXK2uqULIMc,92
 chatterbot/ext/django_chatterbot/abstract_models.py,sha256=PkuBGS0uv2toL3jGE1U6HJLCLXoKwpRNohm9JbVT_y4,3303
@@ -48,19 +49,20 @@ chatterbot/ext/django_chatterbot/migrations/__init__.py,sha256=47DEQpj8HBSa-_TIm
 chatterbot/ext/sqlalchemy_app/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 chatterbot/ext/sqlalchemy_app/models.py,sha256=pjU4e2BUSitw_IAkrk4iFQ9pZRU35y5MomvX7aiBFCw,2492
 chatterbot/logic/__init__.py,sha256=28-5swBCPfSVMl8xB5C8frOKZ2oj28rQfenbd9E4r-4,531
-chatterbot/logic/best_match.py,sha256=iueuuK6WTTywqskwC1CuWD2uHgA65Hz30h8tce_H1bU,4619
+chatterbot/logic/best_match.py,sha256=8TNW0uZ_Uq-XPfaZUMUZDVH6KzDT65j59xblxQBv-dQ,4820
 chatterbot/logic/logic_adapter.py,sha256=5kNEirh5fiF5hhSMFXD7bIkKwXHmrSsSS4qDm-6xry0,4694
 chatterbot/logic/mathematical_evaluation.py,sha256=GPDKUwNFajERof2R-MkPGi2jJRP-rKAGm_f0V9JHDHE,2282
-chatterbot/logic/specific_response.py,sha256=_VeJaa3kun0J7cVzLOlTYK1tBpth0B6UWms7QwtcNpY,1082
+chatterbot/logic/specific_response.py,sha256=o17YIeu9DzucO8MXMP3kwNIBb1b8br60bbAhSE7AZWc,2386
 chatterbot/logic/time_adapter.py,sha256=mxdoQGeC5IjREH4PU5iHYOIPEvnYnzgysocR8xMYWXc,2406
-chatterbot/logic/unit_conversion.py,sha256=DT50HHE3njUo_ttDSU8S-fwBylarhDF3l_McRLSX6Ic,5823
-chatterbot/storage/__init__.py,sha256=IymIHfeisvULQzUYsQSiUBbWIZ1m5EzyMVI082tTw5w,369
-chatterbot/storage/django_storage.py,sha256=b_hJkBm0ZNgBB16HjJaNYVFEPs0AApRjOZpuiGNDaXk,6990
-chatterbot/storage/mongodb.py,sha256=s6rzn0m_eu4kkXeb80vVCyHyZrdrVW_Zf8PlttUHQlk,8962
-chatterbot/storage/sql_storage.py,sha256=X3PKKYcS4tiBWmuvxNMTLxK0shUQlchW0UPB1Mb1rqI,13144
-chatterbot/storage/storage_adapter.py,sha256=QwY3cGVpZLxkmww0OnPGZbdOykuOZT7WSKYHq84TgI0,5956
-ChatterBot-1.2.1.dist-info/LICENSE,sha256=5b04U8mi0wp5gJMYlKi49EalnD9Q2nwY_6UEI_Avgu4,1476
-ChatterBot-1.2.1.dist-info/METADATA,sha256=dF7H-ZhaTdjCYSDFhsChT8h38fFQfykFmXvB6s1v1n0,8311
-ChatterBot-1.2.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-ChatterBot-1.2.1.dist-info/top_level.txt,sha256=W2TzAbAJ-eBXTIKZZhVlkrh87msJNmBQpyhkrHqjSrE,11
-ChatterBot-1.2.1.dist-info/RECORD,,
+chatterbot/logic/unit_conversion.py,sha256=-ENMLqZqtZx0riUi0guda2oJECST0M7pZG4cSIv3ieM,5898
+chatterbot/storage/__init__.py,sha256=ADw0WQe0YKr1UIDQLaxwf0mHDnuKW_CSzgz11K4TM-4,465
+chatterbot/storage/django_storage.py,sha256=S5S4GipD7FyNJy4RWu5-S8sLPuSJIObwTtqTpnJu-ok,6159
+chatterbot/storage/mongodb.py,sha256=Ozvdvcjb3LGZxcvbSQGzwP9VloYQbmsa2FaKunFpMyU,7934
+chatterbot/storage/redis.py,sha256=FKROrzZ-7WXZ8ZoK0dKmTDdS45TxL04XOSeu0p3Jrak,12675
+chatterbot/storage/sql_storage.py,sha256=VVYZvclG_74IN-MrG0edc-RQ2gUO6gRQyCWWSO0MmCk,13082
+chatterbot/storage/storage_adapter.py,sha256=fvyb-qNiB0HMJ0siVMCWUIY--6d-C47N1_kKZVFZAv4,6110
+chatterbot-1.2.3.dist-info/LICENSE,sha256=5b04U8mi0wp5gJMYlKi49EalnD9Q2nwY_6UEI_Avgu4,1476
+chatterbot-1.2.3.dist-info/METADATA,sha256=xnofLrmf6knmhcwBVcodzvxpZQ-eb4tbLB970dXQG8I,8503
+chatterbot-1.2.3.dist-info/WHEEL,sha256=52BFRY2Up02UkjOa29eZOS2VxUrpPORXg1pkohGGUS8,91
+chatterbot-1.2.3.dist-info/top_level.txt,sha256=W2TzAbAJ-eBXTIKZZhVlkrh87msJNmBQpyhkrHqjSrE,11
+chatterbot-1.2.3.dist-info/RECORD,,

{ChatterBot-1.2.1.dist-info → chatterbot-1.2.3.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.8.0)
+Generator: setuptools (76.0.0)
 Root-Is-Purelib: true
 Tag: py3-none-any