src/tools.py

from langchain_core.tools import tool
from langchain_community.tools import DuckDuckGoSearchRun, WikipediaQueryRun
from langchain_community.utilities import DuckDuckGoSearchAPIWrapper, WikipediaAPIWrapper

duckduckapi = DuckDuckGoSearchAPIWrapper(max_results=10)
duckduckgo_search = DuckDuckGoSearchRun(api_wrapper=duckduckapi)

wikipediapi = WikipediaAPIWrapper(top_k_results=5)
wikipedia_search = WikipediaQueryRun(api_wrapper=wikipediapi)

TOOLMAP = {"duckduckgo_search": duckduckgo_search, "wikipedia_search": wikipedia_search}