Generate Python Tool Specs from Code

Python File Tool loads Python files and automatically extracts function names, arguments, and descriptions to create custom Tool Specs. The tool provides three methods: function_definitions to get all function definitions from the file, get_function to retrieve a specific function definition, and get_functions to get a list of functions from the file.

You provide a path to a Python file. The tool extracts function definitions, including names, arguments, and descriptions, and returns them as Tool Specs. The tool provides three methods: function_definitions to get all function definitions from the file, get_function to retrieve a specific function definition, and get_functions to get a list of functions from the file.

The source material shows Python File Tool used with LlamaIndex's FunctionAgent and OpenAI LLMs in the provided example.

from llama_index.tools.python_file import PythonFileToolSpec
from llama_index.core.agent.workflow import FunctionAgent
from llama_index.llms.openai import OpenAI

pyfile = PythonFileToolSpec("./numpy_linalg.py")

agent = FunctionAgent(
    tools=pyfile.to_tool_list(),
    llm=OpenAI(model="gpt-4.1"),
)

print(
    await agent.run(
        """Load the eig, transpose and solve functions from the python file,
and then write a function definition using only builtin python types (List, float, Tuple)
with a short 5-10 line doc string tool prompts for the functions that only has a small description and arguments
"""
    )
)

This loader is designed to be used as a way to load data as a Tool in a Agent.