ai-parrot 0.2.8__tar.gz → 0.3.1__tar.gz

{ai_parrot-0.2.8 → ai_parrot-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ai-parrot
-Version: 0.2.8
+Version: 0.3.1
 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.8 → ai_parrot-0.3.1}/ai_parrot.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ai-parrot
-Version: 0.2.8
+Version: 0.3.1
 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.8 → ai_parrot-0.3.1}/parrot/chatbots/copilot.py

@@ -54,6 +54,7 @@ class CopilotAgent(BaseAgent):
         except Exception as e:
             print('ERROR LOADING ZIPCODE TOOLS')
 
+        try:
             wk1 = WikipediaTool()
             wk12 = WikidataTool()
 

{ai_parrot-0.2.8 → ai_parrot-0.3.1}/parrot/loaders/videolocal.py

@@ -105,3 +105,16 @@ class VideoLocalLoader(BaseVideoLoader):
                     if set(item.parts).isdisjoint(self.skip_directories):
                         documents.extend(self.load_video(item))
         return self.split_documents(documents)
+
+    def extract(self) -> list:
+        documents = []
+        if self.path.is_file():
+            docs = self.load_video(self.path)
+            documents.extend(docs)
+        if self.path.is_dir():
+            # iterate over the files in the directory
+            for ext in self._extension:
+                for item in self.path.glob(f'*{ext}'):
+                    if set(item.parts).isdisjoint(self.skip_directories):
+                        documents.extend(self.load_video(item))
+        return documents

{ai_parrot-0.2.8 → ai_parrot-0.3.1}/parrot/tools/google.py

@@ -107,7 +107,6 @@ class GoogleLocationFinder(BaseTool):
         super().__init__(**kwargs)
         self.google_key = kwargs.get('api_key', GOOGLE_API_KEY)
         self.kwargs = kwargs
-        print(':: LOCATION FINDER STARTED :: ')
 
     def extract_location(self, data):
         city = state = state_code = zipcode = None
@@ -134,7 +133,6 @@ class GoogleLocationFinder(BaseTool):
             self.base_url,
             params=params
         )
-        print('LOCATION FINDER > ', response)
         if response.status_code == 200:
             result = response.json()
             if result['status'] == 'OK':

{ai_parrot-0.2.8 → ai_parrot-0.3.1}/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.8"
+__version__ = "0.3.1"
 __author__ = "Jesus Lara"
 __author_email__ = "jesuslarag@gmail.com"
 __license__ = "MIT"