tools/arxiv_search.py

from langchain_community.retrievers import ArxivRetriever
from langchain_core.tools.retriever import create_retriever_tool

retriever = ArxivRetriever(
    arxiv_search=None,
    arxiv_exceptions=None,
    load_max_docs=2,
    get_full_documents=True,
)

arxiv_tool = create_retriever_tool(
    retriever=retriever,
    name="arxiv_search",
    description="Useful for searching academic papers on arXiv. Input should be a search query."
)

if __name__ == "__main__":
    print(arxiv_tool.invoke("What is the ImageBind model?"))