ai-parrot 0.3.11__tar.gz → 0.3.18__tar.gz

ai_parrot-0.3.18/MANIFEST.in

@@ -0,0 +1,18 @@
+include LICENSE
+include README.md
+global-include *.typed
+recursive-include parrot *
+# Exclude directories
+prune videos
+prune examples
+prune docs
+prune documents
+prune etc
+prune env
+prune resources
+prune dist
+
+
+# Optionally, you can also specify any other unwanted directories
+prune tests
+prune *.egg-info

{ai_parrot-0.3.11 → ai_parrot-0.3.18}/Makefile

@@ -15,7 +15,7 @@ install:
 	# pip install --upgrade querysource[analytics]
 	pip install --upgrade querysource
 	# and Parrot:
-	# pip install -e .[google,milvus,groq,analytics]
+	pip install -e .[google,milvus,groq]
 
 develop:
 	# Install Parrot
@@ -31,6 +31,9 @@ develop:
 	pip install --upgrade querysource
 	python -m pip install -Ur requirements/requirements-dev.txt
 
+setup:
+	python -m pip install -Ur requirements/requirements-dev.txt
+
 dev:
 	flit install --symlink
 

{ai_parrot-0.3.11 → ai_parrot-0.3.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ai-parrot
-Version: 0.3.11
+Version: 0.3.18
 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
@@ -80,29 +80,30 @@ Requires-Dist: torchvision==0.19.1
 Requires-Dist: tf-keras==2.17.0
 Requires-Dist: simsimd==4.3.1
 Requires-Dist: opencv-python==4.10.0.84
+Provides-Extra: basic-loaders
+Requires-Dist: youtube-transcript-api==0.6.2; extra == "basic-loaders"
+Requires-Dist: pymupdf==1.24.4; extra == "basic-loaders"
+Requires-Dist: pymupdf4llm==0.0.1; extra == "basic-loaders"
+Requires-Dist: pdf4llm==0.0.6; extra == "basic-loaders"
+Requires-Dist: pytube==15.0.0; extra == "basic-loaders"
+Requires-Dist: pydub==0.25.1; extra == "basic-loaders"
+Requires-Dist: markdownify==0.12.1; extra == "basic-loaders"
+Requires-Dist: yt_dlp==2024.4.9; extra == "basic-loaders"
+Requires-Dist: moviepy==1.0.3; extra == "basic-loaders"
+Requires-Dist: rapidocr-onnxruntime==1.3.15; extra == "basic-loaders"
+Requires-Dist: pytesseract==0.3.10; extra == "basic-loaders"
+Requires-Dist: python-docx==1.1.0; extra == "basic-loaders"
+Requires-Dist: python-pptx==0.6.23; extra == "basic-loaders"
+Requires-Dist: docx2txt==0.8; extra == "basic-loaders"
+Requires-Dist: mammoth==1.7.1; extra == "basic-loaders"
 Provides-Extra: loaders
 Requires-Dist: unstructured==0.14.3; extra == "loaders"
 Requires-Dist: unstructured-client==0.18.0; extra == "loaders"
-Requires-Dist: youtube-transcript-api==0.6.2; extra == "loaders"
-Requires-Dist: pymupdf==1.24.4; extra == "loaders"
-Requires-Dist: pymupdf4llm==0.0.1; extra == "loaders"
-Requires-Dist: pdf4llm==0.0.6; extra == "loaders"
 Requires-Dist: PyPDF2==3.0.1; extra == "loaders"
 Requires-Dist: pdfminer.six==20231228; extra == "loaders"
 Requires-Dist: pdfplumber==0.11.0; extra == "loaders"
 Requires-Dist: GitPython==3.1.42; extra == "loaders"
 Requires-Dist: opentelemetry-sdk==1.24.0; extra == "loaders"
-Requires-Dist: rapidocr-onnxruntime==1.3.15; extra == "loaders"
-Requires-Dist: pytesseract==0.3.10; extra == "loaders"
-Requires-Dist: python-docx==1.1.0; extra == "loaders"
-Requires-Dist: python-pptx==0.6.23; extra == "loaders"
-Requires-Dist: docx2txt==0.8; extra == "loaders"
-Requires-Dist: pytube==15.0.0; extra == "loaders"
-Requires-Dist: pydub==0.25.1; extra == "loaders"
-Requires-Dist: markdownify==0.12.1; extra == "loaders"
-Requires-Dist: yt_dlp==2024.4.9; extra == "loaders"
-Requires-Dist: moviepy==1.0.3; extra == "loaders"
-Requires-Dist: mammoth==1.7.1; extra == "loaders"
 Requires-Dist: paddlepaddle==2.6.1; extra == "loaders"
 Requires-Dist: paddlepaddle_gpu==2.6.1; extra == "loaders"
 Requires-Dist: paddleocr==2.8.1; extra == "loaders"

ai_parrot-0.3.18/ai_parrot.egg-info/SOURCES.txt

@@ -0,0 +1,226 @@
+.flake8
+.gitignore
+.isort.cfg
+.pylintrc
+INSTALL
+LICENSE
+MANIFEST.in
+Makefile
+README.md
+SECURITY.md
+app.py
+mypy.ini
+pyproject.toml
+pytest.ini
+run.py
+setup.py
+tox.ini
+.github/dependabot.yml
+.github/workflows/codeql-analysis.yml
+.github/workflows/release.yml
+parrot/__init__.py
+parrot/conf.py
+parrot/exceptions.c
+parrot/exceptions.cpython-311-x86_64-linux-gnu.so
+parrot/exceptions.pyx
+parrot/manager.py
+parrot/models.py
+parrot/py.typed
+parrot/version.py
+parrot/__pycache__/__init__.cpython-311.pyc
+parrot/__pycache__/conf.cpython-311.pyc
+parrot/__pycache__/manager.cpython-311.pyc
+parrot/__pycache__/models.cpython-311.pyc
+parrot/__pycache__/version.cpython-311.pyc
+parrot/chatbots/__init__.py
+parrot/chatbots/abstract.py
+parrot/chatbots/asktroc.py
+parrot/chatbots/base.py
+parrot/chatbots/basic.py
+parrot/chatbots/bose.py
+parrot/chatbots/cody.py
+parrot/chatbots/copilot.py
+parrot/chatbots/dataframe.py
+parrot/chatbots/hragents.py
+parrot/chatbots/odoo.py
+parrot/chatbots/__pycache__/__init__.cpython-311.pyc
+parrot/chatbots/__pycache__/abstract.cpython-311.pyc
+parrot/chatbots/__pycache__/askbrett.cpython-311.pyc
+parrot/chatbots/__pycache__/asktroc.cpython-311.pyc
+parrot/chatbots/__pycache__/base.cpython-311.pyc
+parrot/chatbots/__pycache__/basic.cpython-311.pyc
+parrot/chatbots/__pycache__/bose.cpython-311.pyc
+parrot/chatbots/__pycache__/cody.cpython-311.pyc
+parrot/chatbots/__pycache__/copilot.cpython-311.pyc
+parrot/chatbots/__pycache__/dataframe.cpython-311.pyc
+parrot/chatbots/__pycache__/hragents.cpython-311.pyc
+parrot/chatbots/__pycache__/oddie.cpython-311.pyc
+parrot/chatbots/__pycache__/odoo.cpython-311.pyc
+parrot/chatbots/retrievals/__init__.py
+parrot/chatbots/retrievals/constitutional.py
+parrot/chatbots/retrievals/__pycache__/__init__.cpython-311.pyc
+parrot/crew/__init__.py
+parrot/crew/__pycache__/__init__.cpython-311.pyc
+parrot/crew/tools/__init__.py
+parrot/crew/tools/bing.py
+parrot/crew/tools/config.py
+parrot/crew/tools/duckgo.py
+parrot/crew/tools/file.py
+parrot/crew/tools/google.py
+parrot/crew/tools/gtrends.py
+parrot/crew/tools/md2pdf.py
+parrot/crew/tools/rag.py
+parrot/crew/tools/search.py
+parrot/crew/tools/url.py
+parrot/crew/tools/__pycache__/__init__.cpython-311.pyc
+parrot/crew/tools/__pycache__/bing.cpython-311.pyc
+parrot/crew/tools/__pycache__/config.cpython-311.pyc
+parrot/crew/tools/__pycache__/duckgo.cpython-311.pyc
+parrot/crew/tools/__pycache__/file.cpython-311.pyc
+parrot/crew/tools/__pycache__/google.cpython-311.pyc
+parrot/crew/tools/__pycache__/md2pdf.cpython-311.pyc
+parrot/crew/tools/__pycache__/rag.cpython-311.pyc
+parrot/crew/tools/__pycache__/url.cpython-311.pyc
+parrot/crew/tools/__pycache__/weather.cpython-311.pyc
+parrot/handlers/__init__.py
+parrot/handlers/bots.py
+parrot/handlers/chat.py
+parrot/handlers/__pycache__/__init__.cpython-311.pyc
+parrot/handlers/__pycache__/bots.cpython-311.pyc
+parrot/handlers/__pycache__/chat.cpython-311.pyc
+parrot/interfaces/__init__.py
+parrot/interfaces/database.py
+parrot/interfaces/__pycache__/__init__.cpython-311.pyc
+parrot/interfaces/__pycache__/database.cpython-311.pyc
+parrot/llms/__init__.py
+parrot/llms/abstract.py
+parrot/llms/anthropic.py
+parrot/llms/google.py
+parrot/llms/groq.py
+parrot/llms/hf.py
+parrot/llms/openai.py
+parrot/llms/pipes.py
+parrot/llms/vertex.py
+parrot/llms/__pycache__/__init__.cpython-311.pyc
+parrot/llms/__pycache__/abstract.cpython-311.pyc
+parrot/llms/__pycache__/anthropic.cpython-311.pyc
+parrot/llms/__pycache__/google.cpython-311.pyc
+parrot/llms/__pycache__/groq.cpython-311.pyc
+parrot/llms/__pycache__/hf.cpython-311.pyc
+parrot/llms/__pycache__/openai.cpython-311.pyc
+parrot/llms/__pycache__/pipes.cpython-311.pyc
+parrot/llms/__pycache__/vertex.cpython-311.pyc
+parrot/loaders/__init__.py
+parrot/loaders/abstract.py
+parrot/loaders/audio.py
+parrot/loaders/basepdf.py
+parrot/loaders/basevideo.py
+parrot/loaders/csv.py
+parrot/loaders/dir.py
+parrot/loaders/excel.py
+parrot/loaders/github.py
+parrot/loaders/image.py
+parrot/loaders/json.py
+parrot/loaders/pdf.py
+parrot/loaders/pdfchapters.py
+parrot/loaders/pdffn.py
+parrot/loaders/pdfimages.py
+parrot/loaders/pdfmark.py
+parrot/loaders/pdftables.py
+parrot/loaders/ppt.py
+parrot/loaders/qa.py
+parrot/loaders/repo.py
+parrot/loaders/rtd.py
+parrot/loaders/txt.py
+parrot/loaders/video.py
+parrot/loaders/videolocal.py
+parrot/loaders/vimeo.py
+parrot/loaders/web.py
+parrot/loaders/web_base.py
+parrot/loaders/word.py
+parrot/loaders/youtube.py
+parrot/loaders/__pycache__/__init__.cpython-311.pyc
+parrot/loaders/__pycache__/abstract.cpython-311.pyc
+parrot/loaders/__pycache__/basepdf.cpython-311.pyc
+parrot/loaders/__pycache__/basevideo.cpython-311.pyc
+parrot/loaders/__pycache__/dir.cpython-311.pyc
+parrot/loaders/__pycache__/excel.cpython-311.pyc
+parrot/loaders/__pycache__/github.cpython-311.pyc
+parrot/loaders/__pycache__/image.cpython-311.pyc
+parrot/loaders/__pycache__/json.cpython-311.pyc
+parrot/loaders/__pycache__/pdf.cpython-311.pyc
+parrot/loaders/__pycache__/pdfchapters.cpython-311.pyc
+parrot/loaders/__pycache__/pdffn.cpython-311.pyc
+parrot/loaders/__pycache__/pdfimages.cpython-311.pyc
+parrot/loaders/__pycache__/pdfmark.cpython-311.pyc
+parrot/loaders/__pycache__/pdftables.cpython-311.pyc
+parrot/loaders/__pycache__/ppt.cpython-311.pyc
+parrot/loaders/__pycache__/qa.cpython-311.pyc
+parrot/loaders/__pycache__/repo.cpython-311.pyc
+parrot/loaders/__pycache__/rtd.cpython-311.pyc
+parrot/loaders/__pycache__/tables.cpython-311.pyc
+parrot/loaders/__pycache__/txt.cpython-311.pyc
+parrot/loaders/__pycache__/video.cpython-311.pyc
+parrot/loaders/__pycache__/videolocal.cpython-311.pyc
+parrot/loaders/__pycache__/vimeo.cpython-311.pyc
+parrot/loaders/__pycache__/web.cpython-311.pyc
+parrot/loaders/__pycache__/web_base.cpython-311.pyc
+parrot/loaders/__pycache__/word.cpython-311.pyc
+parrot/loaders/__pycache__/youtube.cpython-311.pyc
+parrot/loaders/handlers/__init__.py
+parrot/loaders/handlers/data.py
+parrot/loaders/handlers/__pycache__/__init__.cpython-311.pyc
+parrot/loaders/handlers/__pycache__/data.cpython-311.pyc
+parrot/loaders/utils/__init__.py
+parrot/loaders/utils/models.py
+parrot/loaders/utils/__pycache__/__init__.cpython-311.pyc
+parrot/loaders/utils/__pycache__/models.cpython-311.pyc
+parrot/stores/__init__.py
+parrot/stores/abstract.py
+parrot/stores/milvus.py
+parrot/stores/qdrant.py
+parrot/stores/__pycache__/__init__.cpython-311.pyc
+parrot/stores/__pycache__/abstract.cpython-311.pyc
+parrot/stores/__pycache__/base.cpython-311.pyc
+parrot/stores/__pycache__/milvus.cpython-311.pyc
+parrot/stores/__pycache__/qdrant.cpython-311.pyc
+parrot/tools/__init__.py
+parrot/tools/abstract.py
+parrot/tools/asknews.py
+parrot/tools/bing.py
+parrot/tools/duck.py
+parrot/tools/execute.py
+parrot/tools/google.py
+parrot/tools/stack.py
+parrot/tools/weather.py
+parrot/tools/wikipedia.py
+parrot/tools/zipcode.py
+parrot/tools/__pycache__/__init__.cpython-311.pyc
+parrot/tools/__pycache__/abstract.cpython-311.pyc
+parrot/tools/__pycache__/asknews.cpython-311.pyc
+parrot/tools/__pycache__/bing.cpython-311.pyc
+parrot/tools/__pycache__/duck.cpython-311.pyc
+parrot/tools/__pycache__/execute.cpython-311.pyc
+parrot/tools/__pycache__/google.cpython-311.pyc
+parrot/tools/__pycache__/stack.cpython-311.pyc
+parrot/tools/__pycache__/weather.cpython-311.pyc
+parrot/tools/__pycache__/wikipedia.cpython-311.pyc
+parrot/tools/__pycache__/zipcode.cpython-311.pyc
+parrot/utils/__init__.py
+parrot/utils/toml.py
+parrot/utils/types.cpp
+parrot/utils/types.cpython-311-x86_64-linux-gnu.so
+parrot/utils/types.pyx
+parrot/utils/uv.py
+parrot/utils/__pycache__/__init__.cpython-311.pyc
+parrot/utils/__pycache__/toml.cpython-311.pyc
+parrot/utils/parsers/__init__.py
+parrot/utils/parsers/toml.c
+parrot/utils/parsers/toml.cpython-311-x86_64-linux-gnu.so
+parrot/utils/parsers/toml.pyx
+parrot/utils/parsers/__pycache__/__init__.cpython-311.pyc
+requirements/requirements-dev.txt
+settings/__init__.py
+settings/settings.py
+templates/.compiled
+templates/README.md

{ai_parrot-0.3.11 → ai_parrot-0.3.18}/parrot/chatbots/base.py

@@ -89,6 +89,9 @@ Whether you need help with a specific question or just want to have a conversati
 - OpenWeatherMap: Get weather information about a location.
 - yahoo_finance_news: Retrieve the latest financial news from Yahoo Finance.
 - python_repl_ast: A Python shell. Use this to execute python commands. Input should be a valid python command. When using this tool, sometimes output is abbreviated - make sure it does not look abbreviated before using it in your answer.
+- executable_python_repl_ast: A Python shell. Use this to execute python commands. Input should be a valid python command. When using this tool, whenever you generate a visual output (like charts with matplotlib), instead of using plt.show(), render the image as a base64-encoded HTML string. Do this by saving the plot to a buffer and encoding it in base64, then return the result as a JSON object formatted as follows: "image": "format": "png", "base64": "base64-encoded-string".
+
+
 - youtube_search: Search for videos on YouTube based on specific keywords.
 
 

{ai_parrot-0.3.11 → ai_parrot-0.3.18}/parrot/chatbots/copilot.py

@@ -4,17 +4,43 @@ from .base import BaseAgent
 from ..tools import (
     ZipcodeAPIToolkit,
     WikipediaTool,
-    WikidataTool,
+    # WikidataTool,
     GoogleSearchTool,
     GoogleLocationFinder,
     BingSearchTool,
-    AskNewsTool,
+    # AskNewsTool,
     DuckDuckGoSearchTool,
     YouTubeSearchTool,
     OpenWeatherMapTool,
     StackExchangeTool,
 )
+from ..tools.execute import ExecutablePythonREPLTool
 
+# ZipCode API Toolkit
+zpt = ZipcodeAPIToolkit()
+zpt_tools = zpt.get_tools()
+
+wk1 = WikipediaTool()
+# wk12 = WikidataTool()
+
+g1 = GoogleSearchTool()
+g2 = GoogleLocationFinder()
+
+b = BingSearchTool()
+d = DuckDuckGoSearchTool()
+# ask = AskNewsTool()
+
+yt = YouTubeSearchTool()
+stackexchange = StackExchangeTool()
+weather = OpenWeatherMapTool()
+
+tooling = [
+    wk1,
+    g1, g2,
+    b, d, yt,
+    weather,
+    stackexchange
+] + zpt_tools
 
 class CopilotAgent(BaseAgent):
     """CopilotAgent Agent.
@@ -30,17 +56,24 @@ class CopilotAgent(BaseAgent):
         **kwargs
     ):
         super().__init__(name, llm, tools, prompt_template, **kwargs)
+        if not tools:
+            tools = tooling
         self.tools = [
                 PythonAstREPLTool(
                     name='python_repl_ast',
                     globals={},
                     locals={}
+                ),
+                ExecutablePythonREPLTool(
+                    name='executable_python_repl_ast',
+                    globals={},
+                    locals={}
                 )
             ] + list(tools)
         self.prompt = self.get_prompt(
             self.prompt_template
         )
-        # print('PROMPT > ', self.prompt)
+        print('PROMPT > ', self.prompt)
 
     @classmethod
     def default_tools(cls) -> list:

{ai_parrot-0.3.11 → ai_parrot-0.3.18}/parrot/conf.py

@@ -12,6 +12,8 @@ logging.getLogger(name='h5py').setLevel(logging.INFO)
 logging.getLogger(name='tensorflow').setLevel(logging.INFO)
 logging.getLogger(name='selenium.webdriver').setLevel(logging.WARNING)
 logging.getLogger(name='selenium').setLevel(logging.INFO)
+logging.getLogger(name='matplotlib').setLevel(logging.WARNING)
+logging.getLogger(name='PIL').setLevel(logging.INFO)
 
 
 # Static directory