o
    Zh                     @   s@   d dl mZmZmZ d dlmZ d dlmZ G dd deZdS )    )IteratorOptionalSequence)Document)
BaseLoaderc                   @   sd   e Zd ZdZ					ddee dedee dee dee d	ee fd
dZde	e
 fddZdS )BrowserbaseLoaderzLoad pre-rendered web pages using a headless browser hosted on Browserbase.

    Depends on `browserbase` package.
    Get your API key from https://browserbase.com
    FNurlstext_contentapi_key
project_id
session_idproxyc                 C   sP   || _ || _|| _|| _zddlm} W n ty   tdw |||| _d S )Nr   )BrowserbasezOYou must run `pip install --upgrade browserbase` to use the Browserbase loader.)r   r	   r   r   browserbaser   ImportError)selfr   r	   r
   r   r   r   r    r   g/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/document_loaders/browserbase.py__init__   s   	zBrowserbaseLoader.__init__returnc                 c   sJ    | j | j| j| j| j}t|D ]\}}t|d| j| idV  qdS )zLoad pages from URLsurl)Zpage_contentmetadataN)r   Z	load_urlsr   r	   r   r   	enumerater   )r   Zpagesipager   r   r   	lazy_load)   s   

zBrowserbaseLoader.lazy_load)FNNNN)__name__
__module____qualname____doc__r   strboolr   r   r   r   r   r   r   r   r   r      s*    	
r   N)	typingr   r   r   Zlangchain_core.documentsr   Z)langchain_community.document_loaders.baser   r   r   r   r   r   <module>   s    