ai-parrot 0.2.2__cp311-cp311-manylinux_2_28_x86_64.whl → 0.2.3__cp311-cp311-manylinux_2_28_x86_64.whl

{ai_parrot-0.2.2.dist-info → ai_parrot-0.2.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ai-parrot
-Version: 0.2.2
+Version: 0.2.3
 Summary: Live Chatbots based on Langchain chatbots and Agents     Integrated into Navigator Framework or used into aiohttp applications.
 Home-page: https://github.com/phenobarbital/ai-parrot
 Author: Jesus Lara

{ai_parrot-0.2.2.dist-info → ai_parrot-0.2.3.dist-info}/RECORD

@@ -4,9 +4,9 @@ parrot/exceptions.cpython-311-x86_64-linux-gnu.so,sha256=p2epCb17B2DCUg3-CZizDQI
 parrot/manager.py,sha256=UShLbqfNyPvnGYHzprzQYwu_oj1L-Mhhk2qMRRnsfwo,5187
 parrot/models.py,sha256=GlyWMe2OCgM9dySm2BzK60yKOMHqFth7TYHidTWnioY,12513
 parrot/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-parrot/version.py,sha256=-soyx6MTab6HIEb_c5TDLZ5-rdqnRO3gWdWhzpWVQj8,373
+parrot/version.py,sha256=jffo_3m_M-XJTuobuxGR6dQJrAoGVYBA7Ul7Ldc1jKU,373
 parrot/chatbots/__init__.py,sha256=ypskCnME0xUv6psBEGCEyXCrD0J0ULHSllpVmSxqb4A,200
-parrot/chatbots/abstract.py,sha256=axQkULxAoWGzwa8OZdgVo202DYWLBULW8CrHLqrvbYE,33106
+parrot/chatbots/abstract.py,sha256=sYEgs-SCHuax6f_QiTVNYxqyuGYdsGYLPSFbLhgJnaU,30520
 parrot/chatbots/asktroc.py,sha256=gyWzyvpAnmXwXd-3DEKoIJtAxt6NnP5mUZdZbkFky8s,604
 parrot/chatbots/base.py,sha256=fXqCzxeA2dU2wYlOV7jCMF4S9_8ohqGMdUfm2VQTfQQ,8054
 parrot/chatbots/basic.py,sha256=DIMTPoGc90BRSlokeOdnjlEXAAfZlIFqxXWaMyAX9uk,232
@@ -102,8 +102,8 @@ resources/users/handlers.py,sha256=BGzqBvPY_OaIF_nONWX4b_B5OyyBrdGuSihIsdlFwjk,2
 resources/users/models.py,sha256=glk7Emv7QCi6i32xRFDrGc8UwK23_LPg0XUOJoHnwRU,6799
 settings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 settings/settings.py,sha256=9ueEvyLNurUX-AaIeRPV8GKX1c4YjDLbksUAeqEq6Ck,1854
-ai_parrot-0.2.2.dist-info/LICENSE,sha256=vRKOoa7onTsLNvSzJtGtMaNhWWh8B3YAT733Tlu6M4o,1070
-ai_parrot-0.2.2.dist-info/METADATA,sha256=HQMbR7U-zS_IJUSlQwbU7zkDRgHIkFl3NQjYPQ8YaK8,9336
-ai_parrot-0.2.2.dist-info/WHEEL,sha256=czZm5XKxt3lM6KcAi6fwGhndrA0uZKYAkkvqX1vZAww,113
-ai_parrot-0.2.2.dist-info/top_level.txt,sha256=qHoO4BhYDfeTkyKnciZSQtn5FSLN3Q-P5xCTkyvbuxg,26
-ai_parrot-0.2.2.dist-info/RECORD,,
+ai_parrot-0.2.3.dist-info/LICENSE,sha256=vRKOoa7onTsLNvSzJtGtMaNhWWh8B3YAT733Tlu6M4o,1070
+ai_parrot-0.2.3.dist-info/METADATA,sha256=pODrWxcJrK5wB-ZCkJel6S8kht_mSNp7aEySw_ivigY,9336
+ai_parrot-0.2.3.dist-info/WHEEL,sha256=czZm5XKxt3lM6KcAi6fwGhndrA0uZKYAkkvqX1vZAww,113
+ai_parrot-0.2.3.dist-info/top_level.txt,sha256=qHoO4BhYDfeTkyKnciZSQtn5FSLN3Q-P5xCTkyvbuxg,26
+ai_parrot-0.2.3.dist-info/RECORD,,

parrot/chatbots/abstract.py

@@ -107,20 +107,8 @@ try:
 except (ModuleNotFoundError, ImportError):
     GROQ_ENABLED = False
 
-from ..loaders import (
-    PDFLoader,
-    PDFTablesLoader,
-    GithubLoader,
-    RepositoryLoader,
-    WebLoader,
-    VimeoLoader,
-    YoutubeLoader,
-    PPTXLoader,
-    MSWordLoader
-)
 from .retrievals import RetrievalManager
 from ..conf import (
-    DEFAULT_LLM_MODEL_NAME,
     EMBEDDING_DEVICE,
     MAX_VRAM_AVAILABLE,
     RAM_AVAILABLE,
@@ -802,122 +790,21 @@ class AbstractChatbot(ABC, DBInterface):
                     collection=collection
                 )
 
-    def load_pdf(self, path: Path, source_type: str = 'pdf', **kwargs):
-        loader = PDFLoader(path, source_type=source_type, no_summarization=True, **kwargs)
-        return loader.load()
-
-    def load_github(
-        self,
-        url: str,
-        github_token: str,
-        lang: str = 'python',
-        branch: str = 'master',
-        source_type: str = 'code'
-    ) -> list:
-        git = GithubLoader(
-            url,
-            github_token=github_token,
-            lang=lang,
-            branch=branch,
-            source_type=source_type
-        )
-        return git.load()
-
-    def load_repository(
-        self,
-        path: Path,
-        lang: str = 'python',
-        source_type: str = 'code',
-        **kwargs
-    ) -> list:
-        repo = RepositoryLoader(
-            source_type=source_type,
-            **kwargs
-        )
-        return repo.load(path, lang=lang)
-
-    def process_websites(
-        self,
-        websites: list,
-        source_type: str = 'website',
-        **kwargs
-    ) -> list:
-        loader = WebLoader(
-            urls=websites,
-            source_type=source_type
-        )
-        return loader.load()
-
-    def load_youtube_videos(
-        self,
-        urls: list,
-        video_path: Union[str, Path],
-        source_type: str = 'youtube',
-        priority: int = 'high',
-        language: str = 'en',
-        **kwargs
-    ) -> list:
-        yt = YoutubeLoader(
-            urls=urls,
-            video_path=video_path,
-            source_type=source_type,
-            priority=priority,
-            language=language,
-            llm=self._llm,
-            **kwargs
-        )
-        return yt.load()
-
-    def load_vimeo_videos(
-        self,
-        urls: list,
-        video_path: Union[str, Path],
-        source_type: str = 'vimeo',
-        priority: int = 'high',
-        language: str = 'en',
-        **kwargs
-    ) -> list:
-        yt = VimeoLoader(
-            urls=urls,
-            video_path=video_path,
-            source_type=source_type,
-            priority=priority,
-            language=language,
-            llm=self._llm,
-            **kwargs
-        )
-        return yt.load()
-
-    def load_directory(
-        self,
-        path: Union[str, Path],
-        source_type: str = 'documents',
-    ) -> list:
-        return None
-
-    def load_docx(
+    def clean_history(
         self,
-        path: Path,
-        source_type: str = 'docx',
-        **kwargs
-    ) -> list:
-        return MSWordLoader.from_path(
-            path=path,
-            source_type=source_type,
-            **kwargs
-        )
-
-    def load_pptx(
-        self,
-        path: Path,
-        source_type: str = 'pptx',
-        **kwargs
-    ) -> list:
-        return PPTXLoader.from_path(
-            path=path,
-            source_type=source_type,
-            **kwargs
-        )
+        session_id: str = None
+    ):
+        try:
+            redis_client = RedisChatMessageHistory(
+                    url=REDIS_HISTORY_URL,
+                    session_id=session_id,
+                    ttl=60
+            )
+            redis_client.clear()
+        except Exception as e:
+            self.logger.error(
+                f"Error clearing chat history: {e}"
+            )
 
     def get_memory(
         self,
@@ -925,7 +812,7 @@ class AbstractChatbot(ABC, DBInterface):
         key: str = 'chat_history',
         input_key: str = 'question',
         output_key: str = 'answer',
-        size: int = 30,
+        size: int = 5,
         ttl: int = 86400
     ):
         args = {

parrot/version.py

@@ -3,7 +3,7 @@
 __title__ = "ai-parrot"
 __description__ = "Live Chatbots based on Langchain chatbots and Agents \
     Integrated into Navigator Framework or used into aiohttp applications."
-__version__ = "0.2.2"
+__version__ = "0.2.3"
 __author__ = "Jesus Lara"
 __author_email__ = "jesuslarag@gmail.com"
 __license__ = "MIT"