
    dh                     B    S SK JrJr  S SKJr  S SKJr   " S S\5      rg)    )AnyIterator)Document)BaseDataFrameLoaderc                   T   ^  \ rS rSrSrSS.S\S\4U 4S jjjrS\\	   4S	 jr
S
rU =r$ )PolarsDataFrameLoader   zLoad `Polars` DataFrame.textpage_content_column
data_framer   c                   > SSK n[        XR                  5      (       d  [        S[	        U5       35      e[
        TU ]  XS9  g)zInitialize with dataframe object.

Args:
    data_frame: Polars DataFrame object.
    page_content_column: Name of the column containing the page content.
      Defaults to "text".
r   Nz.Expected data_frame to be a pl.DataFrame, got r   )polars
isinstance	DataFrame
ValueErrortypesuper__init__)selfr   r   pl	__class__s       m/var/www/html/shao/venv/lib/python3.13/site-packages/langchain_community/document_loaders/polars_dataframe.pyr   PolarsDataFrameLoader.__init__   sF     	*ll33@jAQ@RS  	M    returnc              #      #    U R                   R                  SS9 H7  nXR                     nUR                  U R                  5        [	        X!S9v   M9     g7f)z!Lazy load records from dataframe.T)named)page_contentmetadataN)r   	iter_rowsr   popr   )r   rowr
   s      r   	lazy_loadPolarsDataFrameLoader.lazy_load   sN      ??,,4,8C//0DGGD,,-;; 9s   AA )__name__
__module____qualname____firstlineno____doc__r   strr   r   r   r$   __static_attributes____classcell__)r   s   @r   r   r      s;    "FL N3 N N N <8H- < <r   r   N)typingr   r   langchain_core.documentsr   .langchain_community.document_loaders.dataframer   r   r&   r   r   <module>r2      s      - N</ <r   