llama-index-tools-code-interpreter 0.0.1__tar.gz

llama_index_tools_code_interpreter-0.0.1/PKG-INFO

@@ -0,0 +1,51 @@
+Metadata-Version: 2.1
+Name: llama-index-tools-code-interpreter
+Version: 0.0.1
+Summary: llama-index tools code_interpreter integration
+License: MIT
+Author: Your Name
+Author-email: you@example.com
+Requires-Python: >=3.8.1,<3.12
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Requires-Dist: llama-index-core (>=0.9.32,<0.10.0)
+Description-Content-Type: text/markdown
+
+# Code Interpreter Tool
+
+This tool can be used to run python scripts and capture the results of stdout and stderr
+
+WARNING: This tool provides the Agent access to the `subprocess.run` command.
+Arbitrary code execution is possible on the machine running this tool.
+This tool is not recommended to be used in a production setting, and would require heavy sandboxing or virtual machines
+
+## Usage
+
+Here's an example usage of the CodeInterpreterToolSpec.
+
+```python
+from llama_index.tools.code_interpreter import CodeInterpreterToolSpec
+from llama_index.agent import OpenAIAgent
+
+code_spec = CodeInterpreterToolSpec()
+
+agent = OpenAIAgent.from_tools(code_spec.to_tool_list())
+
+# Prime the agent to use the tool
+agent.chat(
+    "Can you help me write some python code to pass to the code_interpreter tool"
+)
+agent.chat(
+    "write a python function to calculate volume of a sphere with radius 4.3cm"
+)
+```
+
+The tools available are:
+
+`code_interpreter`: A tool to evalute a python script
+
+This loader is designed to be used as a way to load data as a Tool in a Agent. See [here](https://github.com/emptycrown/llama-hub/tree/main) for examples.
+

llama_index_tools_code_interpreter-0.0.1/README.md

@@ -0,0 +1,34 @@
+# Code Interpreter Tool
+
+This tool can be used to run python scripts and capture the results of stdout and stderr
+
+WARNING: This tool provides the Agent access to the `subprocess.run` command.
+Arbitrary code execution is possible on the machine running this tool.
+This tool is not recommended to be used in a production setting, and would require heavy sandboxing or virtual machines
+
+## Usage
+
+Here's an example usage of the CodeInterpreterToolSpec.
+
+```python
+from llama_index.tools.code_interpreter import CodeInterpreterToolSpec
+from llama_index.agent import OpenAIAgent
+
+code_spec = CodeInterpreterToolSpec()
+
+agent = OpenAIAgent.from_tools(code_spec.to_tool_list())
+
+# Prime the agent to use the tool
+agent.chat(
+    "Can you help me write some python code to pass to the code_interpreter tool"
+)
+agent.chat(
+    "write a python function to calculate volume of a sphere with radius 4.3cm"
+)
+```
+
+The tools available are:
+
+`code_interpreter`: A tool to evalute a python script
+
+This loader is designed to be used as a way to load data as a Tool in a Agent. See [here](https://github.com/emptycrown/llama-hub/tree/main) for examples.

llama_index_tools_code_interpreter-0.0.1/llama_index/tools/code_interpreter/__init__.py

@@ -0,0 +1,6 @@
+"""init.py"""
+from llama_index.tools.code_interpreter.base import (
+    CodeInterpreterToolSpec,
+)
+
+__all__ = ["CodeInterpreterToolSpec"]

llama_index_tools_code_interpreter-0.0.1/llama_index/tools/code_interpreter/base.py

@@ -0,0 +1,36 @@
+"""Code Interpreter tool spec."""
+
+import subprocess
+import sys
+
+from llama_index.core.tools.tool_spec.base import BaseToolSpec
+
+
+class CodeInterpreterToolSpec(BaseToolSpec):
+    """Code Interpreter tool spec.
+
+    WARNING: This tool provides the Agent access to the `subprocess.run` command.
+    Arbitrary code execution is possible on the machine running this tool.
+    This tool is not recommended to be used in a production setting, and would require heavy sandboxing or virtual machines
+
+    """
+
+    spec_functions = ["code_interpreter"]
+
+    def code_interpreter(self, code: str):
+        """
+        A function to execute python code, and return the stdout and stderr
+
+        You should import any libraries that you wish to use. You have access to any libraries the user has installed.
+
+        The code passed to this functuon is executed in isolation. It should be complete at the time it is passed to this function.
+
+        You should interpret the output and errors returned from this function, and attempt to fix any problems.
+        If you cannot fix the error, show the code to the user and ask for help
+
+        It is not possible to return graphics or other complicated data from this function. If the user cannot see the output, save it to a file and tell the user.
+        """
+        result = subprocess.run(
+            [sys.executable, "-c", code], stdout=subprocess.PIPE, stderr=subprocess.PIPE
+        )
+        return f"StdOut:\n{result.stdout}\nStdErr:\n{result.stderr}"

llama_index_tools_code_interpreter-0.0.1/pyproject.toml

@@ -0,0 +1,49 @@
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"
+
+[tool.codespell]
+check-filenames = true
+check-hidden = true
+# Feel free to un-skip examples, and experimental, you will just need to
+# work through many typos (--write-changes and --interactive will help)
+skip = "*.csv,*.html,*.json,*.jsonl,*.pdf,*.txt,*.ipynb"
+
+[tool.mypy]
+disallow_untyped_defs = true
+# Remove venv skip when integrated with pre-commit
+exclude = ["_static", "build", "examples", "notebooks", "venv"]
+ignore_missing_imports = true
+python_version = "3.8"
+
+[tool.poetry]
+name = "llama-index-tools-code-interpreter"
+version = "0.0.1"
+description = "llama-index tools code_interpreter integration"
+authors = ["Your Name <you@example.com>"]
+license = "MIT"
+readme = "README.md"
+packages = [{include = "llama_index/"}]
+
+[tool.poetry.dependencies]
+python = ">=3.8.1,<3.12"
+llama-index-core = "^0.9.32"
+
+[tool.poetry.group.dev.dependencies]
+black = {extras = ["jupyter"], version = "<=23.9.1,>=23.7.0"}
+codespell = {extras = ["toml"], version = ">=v2.2.6"}
+ipython = "8.10.0"
+jupyter = "^1.0.0"
+mypy = "0.991"
+pre-commit = "3.2.0"
+pylint = "2.15.10"
+pytest = "7.2.1"
+pytest-mock = "3.11.1"
+ruff = "0.0.292"
+tree-sitter-languages = "^1.8.0"
+types-Deprecated = ">=0.1.0"
+types-PyYAML = "^6.0.12.12"
+types-protobuf = "^4.24.0.4"
+types-redis = "4.5.5.0"
+types-requests = "2.28.11.8" # TODO: unpin when mypy>0.991
+types-setuptools = "67.1.0.0"