tools.py

from llama_index.tools.wikipedia import WikipediaToolSpec
from llama_index.tools.duckduckgo import DuckDuckGoSearchToolSpec
from llama_index.core.tools.tool_spec.load_and_search import LoadAndSearchToolSpec

def add(a: int, b: int) -> int:
    """Add two numbers"""
    return a + b

def subtract(a: int, b: int) -> int:
    """Subtract two numbers"""
    return a - b

def multiply(a: int, b: int) -> int:
    """Multiply two numbers"""
    return a * b

def divide(a: int, b: int) -> int:
    """Divide two numbers"""
    return a / b

wiki_tool_spec = WikipediaToolSpec().to_tool_list()[1]
search_tool_spec = DuckDuckGoSearchToolSpec()

wiki_tool = LoadAndSearchToolSpec.from_defaults(wiki_tool_spec).to_tool_list()
search_tool = search_tool_spec.duckduckgo_full_search