arXiv

ArxivToolSpec

基础：BaseToolSpec

arXiv 工具规范。

源代码位于 llama-index-integrations/tools/llama-index-tools-arxiv/llama_index/tools/arxiv/base.py

class ArxivToolSpec(BaseToolSpec):
    """arXiv tool spec."""

    spec_functions = ["arxiv_query"]

    def __init__(self, max_results: Optional[int] = 3):
        self.max_results = max_results

    def arxiv_query(self, query: str, sort_by: Optional[str] = "relevance"):
        """
        A tool to query arxiv.org
        ArXiv contains a variety of papers that are useful for answering
        mathematic and scientific questions.

        Args:
            query (str): The query to be passed to arXiv.
            sort_by (str): Either 'relevance' (default) or 'recent'

        """
        import arxiv

        sort = arxiv.SortCriterion.Relevance
        if sort_by == "recent":
            sort = arxiv.SortCriterion.SubmittedDate
        search = arxiv.Search(query, max_results=self.max_results, sort_by=sort)
        results = []
        for result in search.results():
            results.append(
                Document(text=f"{result.pdf_url}: {result.title}\n{result.summary}")
            )
        return results

arxiv_query

arxiv_query(query: str, sort_by: Optional[str] = 'relevance')

一个查询 arxiv.org 的工具。arXiv 包含各种论文，对于回答数学和科学问题非常有用。

参数

名称	类型	描述	默认
query	str	要传递给 arXiv 的查询。	必需
sort_by	str	'relevance'（默认）或 'recent'	'relevance'

源代码位于 llama-index-integrations/tools/llama-index-tools-arxiv/llama_index/tools/arxiv/base.py

def arxiv_query(self, query: str, sort_by: Optional[str] = "relevance"):
    """
    A tool to query arxiv.org
    ArXiv contains a variety of papers that are useful for answering
    mathematic and scientific questions.

    Args:
        query (str): The query to be passed to arXiv.
        sort_by (str): Either 'relevance' (default) or 'recent'

    """
    import arxiv

    sort = arxiv.SortCriterion.Relevance
    if sort_by == "recent":
        sort = arxiv.SortCriterion.SubmittedDate
    search = arxiv.Search(query, max_results=self.max_results, sort_by=sort)
    results = []
    for result in search.results():
        results.append(
            Document(text=f"{result.pdf_url}: {result.title}\n{result.summary}")
        )
    return results