How can I create custom tool for adding two numbers using ReAct agent?

[Prabhjot410]

I am trying to build custom tool then passing it into ReAct Agent but I am getting the KeyError: tool. Which I think, my tools is not built correctly

def multiply(a: int, b: int) -> int:
    """Multiply two integers and returns the result integer"""
    return a * b


multiply_tool = FunctionTool.from_defaults(fn=multiply)

agent = ReActAgent.from_tools([multiply_tool, add_tool], llm=llm, verbose=True)

[dosubot[bot]]

🤖

Hello again @Prabhjot410! It's good to see you diving into more coding challenges. Let's see what we can do to help you with this one.

It seems like you're trying to create a custom tool for adding two numbers using a ReAct agent, but you're encountering a KeyError because the tool isn't correctly defined. Here's how you can define an add function and create a FunctionTool from it:

def add(a: int, b: int) -> int:
    """Add two integers and returns the result integer"""
    return a + b

from llama_index.tools import FunctionTool

add_tool = FunctionTool.from_defaults(fn=add)

Then, you can initialize the ReAct agent with the add_tool:

from llama_index.agent import ReActAgent
from llama_index.llms import OpenAI

llm = OpenAI(model="gpt-3.5-turbo-instruct")
agent = ReActAgent.from_tools([add_tool, multiply_tool], llm=llm, verbose=True)

Now, the add_tool should be correctly defined and added to the agent's toolset, and you should be able to use it without encountering a KeyError.

If you want to create a more complex custom tool, you can create a class that extends the BaseToolSpec class. This class should define the spec_functions list, the functions themselves, and the get_fn_schema_from_fn_name method. You can find more information about this in the tests/tools/tool_spec/test_base.py file in the LlamaIndex repository.

I hope this helps! If you have any other questions, feel free to ask.

Best,
Dosu

Sources

• docs/examples/agent/react_agent.ipynb
• tests/tools/tool_spec/test_base.py

---

This response is meant to be useful and save you time. It is not meant to be a precise solution, but rather a starting point for your own research.

Help me be more useful! Please leave a 👍 if this is helpful and 👎 if it is irrelevant.

If you want to continue the conversation, start your reply with @dosu-bot.