src/tools/user_tools/arxiv_tool.py

import importlib

__all__ = ['ArxivTool']


class ArxivTool():
    dependencies = ["arxiv==2.1.3"]

    inputSchema = {
        "name": "ArxivTool",
        "description": "Searches arXiv for academic papers based on a query.",
        "parameters": {
            "type": "object",
            "properties": {
                "query": {
                    "type": "string",
                    "description": "Search query for papers on arXiv.",
                    "examples":[
                        "superconductors gem5",
                        "machine learning in healthcare",
                        "quantum computing algorithms"
                    ]
                },
                "max_results": {
                    "type": "integer",
                    "description": "Maximum number of papers to retrieve. Default is 5.",
                    "default": 5
                }
            },
            "required": ["query"],
        }
    }

    def run(self, **kwargs):
        query = kwargs.get("query")
        max_results = kwargs.get("max_results", 5)

        if not query:
            return {
                "status": "error",
                "message": "Missing required parameter: 'query'",
                "output": None
            }

        try:
            arxiv = importlib.import_module("arxiv")
            client = arxiv.Client()

            search = arxiv.Search(
                query=query,
                max_results=max_results,
            )

            papers = []
            for result in client.results(search):
                papers.append({
                    "title": result.title,
                    "authors": [author.name for author in result.authors],
                    "published": result.published.isoformat(),
                    "summary": result.summary.strip(),
                    "pdf_url": result.pdf_url,
                })

            return {
                "status": "success",
                "message": f"Found {len(papers)} paper(s) on arXiv",
                "output": papers,
            }

        except Exception as e:
            return {
                "status": "error",
                "message": f"arXiv search failed: {str(e)}",
                "output": None,
            }