
    Ng                     Z    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
dS )    )IteratorListOptional)Document)
BaseLoader)BraveSearchWrapperc                   d    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dS )BraveSearchLoaderz Load with `Brave Search` engine.Nqueryapi_keysearch_kwargsc                 4    || _         || _        |pi | _        dS )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       m/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/document_loaders/brave_search.py__init__zBraveSearchLoader.__init__   s%     
*0b    returnc                 l    t          | j        | j                  }|                    | j                  S )N)r   r   )r   r   r   download_documentsr   )r   brave_clients     r   loadzBraveSearchLoader.load   s9    )L,
 
 
 ..tz:::r   c              #   @   K   |                                  D ]}|V  d S N)r   )r   docs     r   	lazy_loadzBraveSearchLoader.lazy_load   s2      99;; 	 	CIIII	 	r   r   )__name__
__module____qualname____doc__strr   dictr   r   r   r   r   r    r   r   r
   r
   	   s        **
1 
1c 
1C 
1 
1 
1 
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 I I I I I    
     r   