o
    Zh#                     @   s<   d dl mZmZ d dlmZ d dlmZ G dd deZdS )    )IteratorOptional)Document)
BaseLoaderc                   @   s@   e Zd ZdZ	ddededee fddZdee fd	d
Z	dS )PsychicLoaderzLoad from `Psychic.dev`.Napi_key
account_idconnector_idc                 C   sL   z
ddl m}m} W n ty   tdw ||d| _||| _|| _dS )zInitialize with API key, connector id, and account id.

        Args:
            api_key: The Psychic API key.
            account_id: The Psychic account id.
            connector_id: The Psychic connector id.
        r   )ConnectorIdPsychiczC`psychicapi` package not found, please run `pip install psychicapi`)Z
secret_keyN)Z
psychicapir
   r   ImportErrorpsychicr	   r   )selfr   r   r	   r
   r    r   c/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/document_loaders/psychic.py__init__   s   

zPsychicLoader.__init__returnc                 c   sF    | j j| j| jd}|jD ]}t|d |d |d ddV  qd S )N)r	   r   contenttitleuri)r   source)Zpage_contentmetadata)r   Zget_documentsr	   r   Z	documentsr   )r   Zpsychic_docsdocr   r   r   	lazy_load    s   

zPsychicLoader.lazy_load)N)
__name__
__module____qualname____doc__strr   r   r   r   r   r   r   r   r   r      s    
r   N)typingr   r   Zlangchain_core.documentsr   Z)langchain_community.document_loaders.baser   r   r   r   r   r   <module>   s    