Skip to content

Download Arxiv Input

This module defines the download_arxiv_paper tool, which leverages the ArxivPaperDownloader class to fetch and download academic papers from arXiv based on their unique arXiv ID.

DownloadArxivPaperInput

Bases: BaseModel

Input schema for the arXiv paper download tool. (Optional: if you decide to keep Pydantic validation in the future)

Source code in aiagents4pharma/talk2scholars/tools/paper_download/download_arxiv_input.py 
18
19
20
21
22
23
24
25
26
27
class DownloadArxivPaperInput(BaseModel):
    """
    Input schema for the arXiv paper download tool.
    (Optional: if you decide to keep Pydantic validation in the future)
    """

    arxiv_id: str = Field(
        description="The arXiv paper ID used to retrieve the paper details and PDF."
    )
    tool_call_id: Annotated[str, InjectedToolCallId]

download_arxiv_paper(arxiv_id, tool_call_id)

Download an arXiv paper's PDF using its unique arXiv ID.

This function
  1. Creates an ArxivPaperDownloader instance.
  2. Fetches metadata from arXiv using the provided arxiv_id.
  3. Downloads the PDF from the returned link.
  4. Returns a Command object containing the PDF data and a success message.

Parameters:

Name Type Description Default
arxiv_id str

The unique arXiv paper ID.

 required
tool_call_id InjectedToolCallId

A unique identifier for tracking this tool call.

 required

Returns:

Type Description
Command[Any]

Command[Any]: Contains metadata and messages about the success of the operation.
Source code in aiagents4pharma/talk2scholars/tools/paper_download/download_arxiv_input.py 
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
@tool(args_schema=DownloadArxivPaperInput, parse_docstring=True)
def download_arxiv_paper(
    arxiv_id: str,
    tool_call_id: Annotated[str, InjectedToolCallId],
) -> Command[Any]:
    """
    Download an arXiv paper's PDF using its unique arXiv ID.

    This function:
      1. Creates an `ArxivPaperDownloader` instance.
      2. Fetches metadata from arXiv using the provided `arxiv_id`.
      3. Downloads the PDF from the returned link.
      4. Returns a `Command` object containing the PDF data and a success message.

    Args:
        arxiv_id (str): The unique arXiv paper ID.
        tool_call_id (InjectedToolCallId): A unique identifier for tracking this tool call.

    Returns:
        Command[Any]: Contains metadata and messages about the success of the operation.
    """
    downloader = ArxivPaperDownloader()

    # If the downloader fails or the arxiv_id is invalid, this might raise an error
    pdf_data = downloader.download_pdf(arxiv_id)
    # print (pdf_data)

    content = f"Successfully downloaded PDF for arXiv ID {arxiv_id}"

    return Command(
        update={
            "pdf_data": pdf_data,
            "messages": [ToolMessage(content=content, tool_call_id=tool_call_id)],
        }
    )