ai-parrot 0.2.6__cp311-cp311-manylinux_2_28_x86_64.whl → 0.2.8__cp311-cp311-manylinux_2_28_x86_64.whl

{ai_parrot-0.2.6.dist-info → ai_parrot-0.2.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ai-parrot
-Version: 0.2.6
+Version: 0.2.8
 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
@@ -95,7 +95,7 @@ Requires-Dist: huggingface-hub ==0.23.5
 Requires-Dist: llama-index ==0.10.20
 Requires-Dist: llama-cpp-python ==0.2.56
 Requires-Dist: asyncdb[all] >=2.7.10
-Requires-Dist: querysource[analytics] >=3.10.1
+Requires-Dist: querysource >=3.10.1
 Requires-Dist: yfinance ==0.2.40
 Requires-Dist: youtube-search ==2.1.2
 Requires-Dist: wikipedia ==1.4.0

{ai_parrot-0.2.6.dist-info → ai_parrot-0.2.8.dist-info}/RECORD

@@ -1,10 +1,10 @@
 parrot/__init__.py,sha256=eTkAkHeJ5BBDG2fxrXA4M37ODBJoS1DQYpeBAWL2xeI,387
 parrot/conf.py,sha256=-9bVGC7Rf-6wpIg6-ojvU4S_G1wBLUCVDt46KEGHEhM,4257
 parrot/exceptions.cpython-311-x86_64-linux-gnu.so,sha256=gDwsnUlOlwphVU97XaqG5e7BJs_PWPKdwgwDsjyVZIg,361200
-parrot/manager.py,sha256=cgiF0JS-canOaBVIYIt-zSJRtOQLZqe2eMPQjGonlGc,5939
+parrot/manager.py,sha256=NhzXoWxSgtoWHpmYP8cV2Ujq_SlvCbQYQBaohAeL2TM,5935
 parrot/models.py,sha256=RsVQCqhSXBKRPcu-BCga9Y1wyvENFXDCuq3_ObIKvAo,13452
 parrot/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-parrot/version.py,sha256=-9AyAV0Vls7nJmBAukBA__wxWp776DSD1ap_87547xE,373
+parrot/version.py,sha256=O1rXCpzREQ3kbfdsa89Mq2HsyQX8wAYv0TSambdhebc,373
 parrot/chatbots/__init__.py,sha256=ypskCnME0xUv6psBEGCEyXCrD0J0ULHSllpVmSxqb4A,200
 parrot/chatbots/abstract.py,sha256=CmDn3k4r9uKImOZRN4L9zxLbCdC-1MPUAorDlfZT-kA,26421
 parrot/chatbots/asktroc.py,sha256=gyWzyvpAnmXwXd-3DEKoIJtAxt6NnP5mUZdZbkFky8s,604
@@ -12,12 +12,12 @@ parrot/chatbots/base.py,sha256=cRw7k5FRKOfLdXQJeQvACVE5ZgE1NUWf3IY7OsSsxuo,12912
 parrot/chatbots/basic.py,sha256=DIMTPoGc90BRSlokeOdnjlEXAAfZlIFqxXWaMyAX9uk,232
 parrot/chatbots/bose.py,sha256=z8rm8G_tAwHjDUodXfrAKnhaMzufQyf-GrhxwHeHle4,757
 parrot/chatbots/cody.py,sha256=Z0LNiNtZjEe7bA3hwexclBZK5zEF9m2ODVmrzZjC3Bw,623
-parrot/chatbots/copilot.py,sha256=iwwH3qMqmnNwoVvWKyyrfTi46gAIjlkAPI4yqUGGOEA,1756
+parrot/chatbots/copilot.py,sha256=WDDNhjyH-xvOOA831E9qmPpGXGCpFbzO3vKj9zbfudg,2038
 parrot/chatbots/dataframe.py,sha256=CfZiLKIwnaku52nl2PNjciqRlH8m2lM4buO6xI7P408,3914
 parrot/chatbots/hragents.py,sha256=PyNIBJ2OH5CtfVydccgpY50V6GI3cLKuVdOMaa7sQz0,574
 parrot/chatbots/oddie.py,sha256=RMbANmJZP1_vLVGKRNBKmA8otyAiWPkvpA0rJ0U3tZk,796
 parrot/chatbots/odoo.py,sha256=RMbANmJZP1_vLVGKRNBKmA8otyAiWPkvpA0rJ0U3tZk,796
-parrot/chatbots/retrievals/__init__.py,sha256=FV3RmHSeMWAG2kJkh5cDM8bR6dT9PzGfv6Y8qyLh3kQ,19590
+parrot/chatbots/retrievals/__init__.py,sha256=Lx-bNLAbMYwi-eeh5ln_IvqvwKrnjB1tI9ieWKhQRXM,19589
 parrot/chatbots/retrievals/constitutional.py,sha256=x7kFR0KpmkR0Wz6bXNOeU3Hesn_EXc6eOSd4TdXxb8o,600
 parrot/crew/__init__.py,sha256=oWLEWNNdYHB2AZUDUZVlSsdMOJiWamvM-jVZ8SzoWQE,77
 parrot/crew/tools/__init__.py,sha256=n2nXEVl8VeWmdI_NyGbiVTAq_S51ebNExuyRM7u9SeI,569
@@ -87,7 +87,7 @@ parrot/tools/abstract.py,sha256=pVSZw8MDpbVcQ-CHaGwP6CpqXHIs8hH8Oy1AqUuMmrw,1706
 parrot/tools/asknews.py,sha256=hEpPJMyNBVfj2maHbqnumn3VkY45oFvrjkE3Rq8EdGA,1039
 parrot/tools/bing.py,sha256=BtmFD66OIuCaOue5U2_yIqtjWf24IhEgNOX1LAVvHtA,464
 parrot/tools/duck.py,sha256=UAAZzlF-Q0sZh0_IcS96dwSgCuBPdeepkwRrMM5cJPY,1920
-parrot/tools/google.py,sha256=NjijcUWH6Crk5Uty_x3FstjDTGZV8JXfBFDQEtMHhac,6236
+parrot/tools/google.py,sha256=5dLqm93LVTmAMd1l9sGfbVs-eoyNSIrnZNcfPgERSaE,6330
 parrot/tools/stack.py,sha256=M-VRWjIDa18bl5p88dSKtxMj4Kn21YB76to0u6yXA30,942
 parrot/tools/weather.py,sha256=4v9Ft5lkVzb9Pg7afNs7BK5T3WEcsZbHPlBrF9oXSo8,2541
 parrot/tools/wikipedia.py,sha256=oadBTRAupu2dKThEORSHqqVs4u0G9lWOltbP6vSZgPE,1999
@@ -103,8 +103,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.6.dist-info/LICENSE,sha256=vRKOoa7onTsLNvSzJtGtMaNhWWh8B3YAT733Tlu6M4o,1070
-ai_parrot-0.2.6.dist-info/METADATA,sha256=7Kji1fBpiTAyv3ZtEf2Ykm66fWDWucpx2hzliTml3ts,10421
-ai_parrot-0.2.6.dist-info/WHEEL,sha256=tFO7F0mawMNWa_NzTDA1ygqZBeMykVNIr04O5Zxk1TE,113
-ai_parrot-0.2.6.dist-info/top_level.txt,sha256=qHoO4BhYDfeTkyKnciZSQtn5FSLN3Q-P5xCTkyvbuxg,26
-ai_parrot-0.2.6.dist-info/RECORD,,
+ai_parrot-0.2.8.dist-info/LICENSE,sha256=vRKOoa7onTsLNvSzJtGtMaNhWWh8B3YAT733Tlu6M4o,1070
+ai_parrot-0.2.8.dist-info/METADATA,sha256=ahe9w-iqJWGMbc0ereeS3suSkWCHSmZntnjdN-XnRr8,10410
+ai_parrot-0.2.8.dist-info/WHEEL,sha256=tFO7F0mawMNWa_NzTDA1ygqZBeMykVNIr04O5Zxk1TE,113
+ai_parrot-0.2.8.dist-info/top_level.txt,sha256=qHoO4BhYDfeTkyKnciZSQtn5FSLN3Q-P5xCTkyvbuxg,26
+ai_parrot-0.2.8.dist-info/RECORD,,

parrot/chatbots/copilot.py

@@ -45,29 +45,38 @@ class CopilotAgent(BaseAgent):
     @classmethod
     def default_tools(cls) -> list:
         # ZipCode API Toolkit
-        zpt = ZipcodeAPIToolkit()
-        zpt_tools = zpt.get_tools()
+        tools = []
+        zpt_tools = []
+        try:
+            zpt = ZipcodeAPIToolkit()
+            zpt_tools = zpt.get_tools()
 
-        wk1 = WikipediaTool()
-        wk12 = WikidataTool()
+        except Exception as e:
+            print('ERROR LOADING ZIPCODE TOOLS')
 
-        g1 = GoogleSearchTool()
-        g2 = GoogleLocationFinder()
+            wk1 = WikipediaTool()
+            wk12 = WikidataTool()
 
-        b = BingSearchTool()
-        d = DuckDuckGoSearchTool()
-        ask = AskNewsTool()
+            g1 = GoogleSearchTool()
+            g2 = GoogleLocationFinder()
 
-        yt = YouTubeSearchTool()
-        stackexchange = StackExchangeTool()
-        weather = OpenWeatherMapTool()
+            b = BingSearchTool()
+            d = DuckDuckGoSearchTool()
+            ask = AskNewsTool()
 
-        tools = [
-            wk1,
-            wk12, g1, g2,
-            b, d, ask, yt,
-            weather,
-            stackexchange
-        ] + zpt_tools
+            yt = YouTubeSearchTool()
+            stackexchange = StackExchangeTool()
+            weather = OpenWeatherMapTool()
+
+            tools = [
+                wk1,
+                wk12, g1, g2,
+                b, d, ask, yt,
+                weather,
+                stackexchange
+            ] + zpt_tools
+
+        except Exception as e:
+            print('TOOL Error > ', e)
 
         return tools

parrot/chatbots/retrievals/__init__.py

@@ -85,25 +85,25 @@ class RetrievalManager:
         self.request = request
 
     def __enter__(self):
-        # self.client_id: str = str(uuid.uuid4())
-        # self.client, self.client_id = self.store.connect()
+        self.client_id: str = str(uuid.uuid4())
+        self.client, _ = self.store.connect(alias=self.client_id)
         return self
 
     async def __aenter__(self):
-        # self.client_id: str = str(uuid.uuid4())
-        # self.client, self.client_id = self.store.connect()
+        self.client_id: str = str(uuid.uuid4())
+        self.client, _ = self.store.connect(alias=self.client_id)
         return self
 
     async def __aexit__(self, exc_type, exc_value, traceback):
-        #self.client.close()  # closing database connection
+        self.client.close()  # closing database connection
         # closing the connection:
-        # self.store.close(alias=self.client_id)
+        self.store.close(alias=self.client_id)
         pass
 
     def __exit__(self, exc_type, exc_value, traceback):
-        # self.client.close()  # closing database connection
+        self.client.close()  # closing database connection
         # closing the connection:
-        # self.store.close(alias=self.client_id)
+        self.store.close(alias=self.client_id)
         pass
 
     def create_memory(

parrot/manager.py

@@ -87,7 +87,7 @@ class ChatbotManager:
                     except Exception as e:
                         print('AQUI >>> ', e)
                         self.logger.error(
-                            f"Failed to configure Agent '{chatbot.name}': {e}"
+                            f"Failed to configure Agent '{bot.name}': {e}"
                         )
                 self.add_chatbot(chatbot)
         self.logger.info(

parrot/tools/google.py

@@ -107,6 +107,7 @@ 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
@@ -133,6 +134,7 @@ class GoogleLocationFinder(BaseTool):
             self.base_url,
             params=params
         )
+        print('LOCATION FINDER > ', response)
         if response.status_code == 200:
             result = response.json()
             if result['status'] == 'OK':

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.6"
+__version__ = "0.2.8"
 __author__ = "Jesus Lara"
 __author_email__ = "jesuslarag@gmail.com"
 __license__ = "MIT"