o Zh@s<ddlmZmZddlmZddlmZGdddeZdS))AnyIterator)Document)BaseDataFrameLoadercsBeZdZdZdddedeffddZdeefd d Z Z S) PolarsDataFrameLoaderzLoad `Polars` DataFrame.textpage_content_column data_framer cs:ddl}t||jstdt|tj||ddS)zInitialize with dataframe object. Args: data_frame: Polars DataFrame object. page_content_column: Name of the column containing the page content. Defaults to "text". rNz.Expected data_frame to be a pl.DataFrame, got r)Zpolars isinstanceZ DataFrame ValueErrortypesuper__init__)selfr r pl __class__l/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/document_loaders/polars_dataframe.pyr s   zPolarsDataFrameLoader.__init__returnccs>|jjddD]}||j}||jt||dVqdS)z!Lazy load records from dataframe.T)named)Z page_contentmetadataN)r Z iter_rowsr popr)rrowrrrr lazy_loads   zPolarsDataFrameLoader.lazy_load) __name__ __module__ __qualname____doc__rstrrrrr __classcell__rrrrrsrN)typingrrZlangchain_core.documentsrZ.langchain_community.document_loaders.dataframerrrrrrs