mirror of
https://github.com/hwchase17/langchain.git
synced 2025-09-20 01:54:14 +00:00
Harrison/arxiv tool (#3186)
Co-authored-by: leo-gan <leo.gan.57@gmail.com>
This commit is contained in:
1
langchain/tools/arxiv/__init__.py
Normal file
1
langchain/tools/arxiv/__init__.py
Normal file
@@ -0,0 +1 @@
|
||||
"""Arxiv API toolkit."""
|
24
langchain/tools/arxiv/tool.py
Normal file
24
langchain/tools/arxiv/tool.py
Normal file
@@ -0,0 +1,24 @@
|
||||
"""Tool for the Wikipedia API."""
|
||||
|
||||
from langchain.tools.base import BaseTool
|
||||
from langchain.utilities.arxiv import ArxivAPIWrapper
|
||||
|
||||
|
||||
class ArxivQueryRun(BaseTool):
|
||||
"""Tool that adds the capability to search using the Arxiv API."""
|
||||
|
||||
name = "Arxiv"
|
||||
description = (
|
||||
"A wrapper around Arxiv. "
|
||||
"Useful for getting summary of articles from arxiv.org. "
|
||||
"Input should be a search query."
|
||||
)
|
||||
api_wrapper: ArxivAPIWrapper
|
||||
|
||||
def _run(self, query: str) -> str:
|
||||
"""Use the Arxiv tool."""
|
||||
return self.api_wrapper.run(query)
|
||||
|
||||
async def _arun(self, query: str) -> str:
|
||||
"""Use the Arxiv tool asynchronously."""
|
||||
raise NotImplementedError("ArxivAPIWrapper does not support async")
|
Reference in New Issue
Block a user