
    {h                     R    d dl mZmZmZ d dlmZ d dlmZ d dlm	Z	  G d de      Z
y)    )IteratorListOptional)Document)
BaseLoader)BraveSearchWrapperc                   P    e Zd ZdZd
dededee   fdZdee	   fdZ
dee	   fd	Zy)BraveSearchLoaderz Load with `Brave Search` engine.Nqueryapi_keysearch_kwargsc                 6    || _         || _        |xs i | _        y)zInitializes the BraveLoader.

        Args:
            query: The query to search for.
            api_key: The API key to use.
            search_kwargs: The search kwargs to use.
        N)r   r   r   )selfr   r   r   s       k/var/www/html/engine/venv/lib/python3.12/site-packages/langchain_community/document_loaders/brave_search.py__init__zBraveSearchLoader.__init__   s     
*0b    returnc                 z    t        | j                  | j                        }|j                  | j                        S )N)r   r   )r   r   r   download_documentsr   )r   brave_clients     r   loadzBraveSearchLoader.load   s3    )LL,,
 ..tzz::r   c              #   >   K   | j                         D ]  }|  y wN)r   )r   docs     r   	lazy_loadzBraveSearchLoader.lazy_load   s      99; 	CI	s   r   )__name__
__module____qualname____doc__strr   dictr   r   r   r   r   r    r   r   r
   r
   	   sD    *
1c 
1C 
1 
1;d8n ;8H- r   r
   N)typingr   r   r   langchain_core.documentsr   )langchain_community.document_loaders.baser   *langchain_community.utilities.brave_searchr   r
   r"   r   r   <module>r'      s     + + - @ I
 r   