
    dhV                     \    S SK JrJrJr  S SKJr  S SKJr   " S S\5      r " S S\5      r	g)	    )AnyIteratorLiteral)Document)
BaseLoaderc                   B    \ rS rSrSS.S\S\4S jjrS\\   4S jr	S	r
g
)BaseDataFrameLoader   textpage_content_column
data_framer   c                    Xl         X l        g)zInitialize with dataframe object.

Args:
    data_frame: DataFrame object.
    page_content_column: Name of the column containing the page content.
      Defaults to "text".
Nr   r   )selfr   r   s      f/var/www/html/shao/venv/lib/python3.13/site-packages/langchain_community/document_loaders/dataframe.py__init__BaseDataFrameLoader.__init__	   s     %#6     returnc              #      #    U R                   R                  5        H;  u  pUR                  5       nUR                  U R                  5      n[        XCS9v   M=     g7f)z!Lazy load records from dataframe.)page_contentmetadataN)r   iterrowsto_dictpopr   r   )r   _rowr   r   s        r   	lazy_loadBaseDataFrameLoader.lazy_load   sJ      oo..0FA{{}H<< 8 89D@@ 1s   AAr   N)__name__
__module____qualname____firstlineno__r   strr   r   r   r   __static_attributes__ r   r   r	   r	      s+    FL 	73 	7 	7A8H- Ar   r	   c                   J   ^  \ rS rSrSr  S	S\S\S\S   4U 4S jjjrSr	U =r
$ )
DataFrameLoader   zLoad `Pandas` DataFrame.r   r   engine)pandasmodinc                   >  US:X  a  SSK nOUS:X  a  SSKJ n  O[        SU S35      e[	        XR
                  5      (       d  [        S[        U5       35      e[        TU ]!  XS	9  g! [         a  n[        S5      UeSnAff = f)
zInitialize with dataframe object.

Args:
    data_frame: Pandas DataFrame object.
    page_content_column: Name of the column containing the page content.
      Defaults to "text".
r,   r   Nr-   zUnsupported engine z&. Must be one of 'pandas', or 'modin'.zBUnable to import pandas, please install with `pip install pandas`.z.Expected data_frame to be a pd.DataFrame, got r   )	r,   modin.pandas
ValueErrorImportError
isinstance	DataFrametypesuperr   )r   r   r   r+   pde	__class__s         r   r   DataFrameLoader.__init__    s    	!#7") )&1WX  *ll33@jAQ@RS  	M  	T	s!   
A) A) A) )
B3A??Br'   )r   r,   )r!   r"   r#   r$   __doc__r   r%   r   r   r&   __classcell__)r8   s   @r   r)   r)      s?    "
 $*-5	NN !N )*	N Nr   r)   N)
typingr   r   r   langchain_core.documentsr   )langchain_community.document_loaders.baser   r	   r)   r'   r   r   <module>r?      s.    ) ) - @A* A*"N) "Nr   