forked from Archives/langchain
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
19 lines
585 B
Python
19 lines
585 B
Python
from typing import List
|
|
|
|
from langchain.schema import BaseRetriever, Document
|
|
from langchain.utilities.arxiv import ArxivAPIWrapper
|
|
|
|
|
|
class ArxivRetriever(BaseRetriever, ArxivAPIWrapper):
|
|
"""
|
|
It is effectively a wrapper for ArxivAPIWrapper.
|
|
It wraps load() to get_relevant_documents().
|
|
It uses all ArxivAPIWrapper arguments without any change.
|
|
"""
|
|
|
|
def get_relevant_documents(self, query: str) -> List[Document]:
|
|
return self.load(query=query)
|
|
|
|
async def aget_relevant_documents(self, query: str) -> List[Document]:
|
|
raise NotImplementedError
|