a
    bg                     @   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                   @   sP   e Zd ZdZd
ee eee ee ee ee dddZe	e
 d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   sT   || _ || _|| _|| _zddlm} W n tyB   tdY n0 |||| _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   ~/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/document_loaders/browserbase.py__init__   s    	
zBrowserbaseLoader.__init__)returnc                 c   sH   | 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   