o
    Zh_                     @   s,   d dl mZ d dlmZ G dd deZdS )    )Any)BaseDataFrameLoaderc                       s,   e Zd ZdZddedef fddZ  ZS )XorbitsLoaderzLoad `Xorbits` DataFrame.text
data_framepage_content_columnc              
      sf   zddl m} W n ty } ztd|d}~ww t||js)tdt| t j||d dS )aJ  Initialize with dataframe object.

        Requirements:
            Must have xorbits installed. You can install with `pip install xorbits`.

        Args:
            data_frame: Xorbits DataFrame object.
            page_content_column: Name of the column containing the page content.
              Defaults to "text".
        r   NzACannot import xorbits, please install with 'pip install xorbits'.zLExpected data_frame to be a xorbits.pandas.DataFrame,                   got )r   )	Zxorbits.pandasZpandasImportError
isinstanceZ	DataFrame
ValueErrortypesuper__init__)selfr   r   pde	__class__ c/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/document_loaders/xorbits.pyr   	   s"   zXorbitsLoader.__init__)r   )__name__
__module____qualname____doc__r   strr   __classcell__r   r   r   r   r      s     r   N)typingr   Z.langchain_community.document_loaders.dataframer   r   r   r   r   r   <module>   s    