a bg@s<ddlmZmZddlmZddlmZGdddeZdS))AnyIterator)Document)BaseDataFrameLoadercs@eZdZdZddeedfddZeeddd Z Z S) PolarsDataFrameLoaderzLoad `Polars` DataFrame.textpage_content_column) data_framer cs:ddl}t||js&tdt|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__/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/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