Source code for langchain_community.document_loaders.brave_search
from typing import Iterator, List, Optional
from langchain_core.documents import Document
from langchain_community.document_loaders.base import BaseLoader
from langchain_community.utilities.brave_search import BraveSearchWrapper
[docs]class BraveSearchLoader(BaseLoader):
"""使用`Brave Search`搜索引擎加载。"""
[docs] def __init__(self, query: str, api_key: str, search_kwargs: Optional[dict] = None):
"""初始化BraveLoader。
参数:
query: 要搜索的查询。
api_key: 要使用的API密钥。
search_kwargs: 要使用的搜索参数。
"""
self.query = query
self.api_key = api_key
self.search_kwargs = search_kwargs or {}
[docs] def load(self) -> List[Document]:
brave_client = BraveSearchWrapper(
api_key=self.api_key,
search_kwargs=self.search_kwargs,
)
return brave_client.download_documents(self.query)
[docs] def lazy_load(self) -> Iterator[Document]:
for doc in self.load():
yield doc